Smart enums implement comparison operators as well.

Author: PawelGerr

samples/Thinktecture.Runtime.Extensions.Samples/SmartEnums/SmartEnumDemos.cs

@@ -74,6 +74,9 @@ private static void DemoForNonValidatableEnum(ILogger logger)
       var comparison = ProductGroup.Apple.CompareTo(ProductGroup.Orange); // -1
       logger.Information("Comparison: {Comparison}", comparison);
 
+      var isBigger = ProductGroup.Apple > ProductGroup.Orange;
+      logger.Information("{Apple} > {Orange} = {IsBigger}", ProductGroup.Apple, ProductGroup.Orange, isBigger);
+
       logger.Information("==== Demo for abstract static members ====");
 
       PrintAllItems<ProductType, string>(logger);

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/DefaultMemberState.cs

@@ -23,6 +23,7 @@ public sealed class DefaultMemberState : IMemberState, IEquatable<DefaultMemberS
    public bool IsFormattable { get; }
    public bool IsComparable { get; }
    public bool IsParsable { get; }
+   public bool HasComparisonOperators { get; }
 
    public DefaultMemberState(string name, ITypeSymbol type, string argumentName, bool isStatic)
    {
@@ -51,6 +52,10 @@ public DefaultMemberState(string name, ITypeSymbol type, string argumentName, bo
          {
             IsParsable = true;
          }
+         else if (@interface.IsIComparisonOperators(_type))
+         {
+            HasComparisonOperators = true;
+         }
       }
    }
 
@@ -79,7 +84,8 @@ public bool Equals(DefaultMemberState? other)
              && IsReferenceType == other.IsReferenceType
              && IsFormattable == other.IsFormattable
              && IsComparable == other.IsComparable
-             && IsParsable == other.IsParsable;
+             && IsParsable == other.IsParsable
+             && HasComparisonOperators == other.HasComparisonOperators;
    }
 
    public override int GetHashCode()
@@ -95,6 +101,7 @@ public override int GetHashCode()
          hashCode = (hashCode * 397) ^ IsFormattable.GetHashCode();
          hashCode = (hashCode * 397) ^ IsComparable.GetHashCode();
          hashCode = (hashCode * 397) ^ IsParsable.GetHashCode();
+         hashCode = (hashCode * 397) ^ HasComparisonOperators.GetHashCode();
 
          return hashCode;
       }

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/SmartEnums/EnumSettings.cs

@@ -7,6 +7,7 @@ public sealed class EnumSettings : IEquatable<EnumSettings>
    public string? KeyPropertyName { get; }
    public bool SkipIComparable { get; }
    public bool SkipIParsable { get; }
+   public bool SkipIComparisonOperators { get; }
    public bool SkipIFormattable { get; }
    public bool SkipToString { get; }
 
@@ -15,6 +16,7 @@ public EnumSettings(AttributeData? attribute)
       KeyPropertyName = attribute?.FindKeyPropertyName().TrimAndNullify();
       SkipIComparable = attribute?.FindSkipIComparable() ?? false;
       SkipIParsable = attribute?.FindSkipIParsable() ?? false;
+      SkipIComparisonOperators = attribute?.FindSkipIComparisonOperators() ?? false;
       SkipIFormattable = attribute?.FindSkipIFormattable() ?? false;
       SkipToString = attribute?.FindSkipToString() ?? false;
    }
@@ -34,6 +36,7 @@ public bool Equals(EnumSettings? other)
       return KeyPropertyName == other.KeyPropertyName
              && SkipIComparable == other.SkipIComparable
              && SkipIParsable == other.SkipIParsable
+             && SkipIComparisonOperators == other.SkipIComparisonOperators
              && SkipIFormattable == other.SkipIFormattable
              && SkipToString == other.SkipToString;
    }
@@ -45,6 +48,7 @@ public override int GetHashCode()
          var hashCode = KeyPropertyName?.GetHashCode() ?? 0;
          hashCode = (hashCode * 397) ^ SkipIComparable.GetHashCode();
          hashCode = (hashCode * 397) ^ SkipIParsable.GetHashCode();
+         hashCode = (hashCode * 397) ^ SkipIComparisonOperators.GetHashCode();
          hashCode = (hashCode * 397) ^ SkipIFormattable.GetHashCode();
          hashCode = (hashCode * 397) ^ SkipToString.GetHashCode();
 

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/SmartEnums/EnumSourceGeneratorState.cs

@@ -156,6 +156,9 @@ public ImmutableArray<IInterfaceCodeGenerator> GetInterfaceCodeGenerators()
       if (!Settings.SkipIParsable && (KeyProperty.IsString() || KeyProperty.IsParsable))
          generators = generators.Add(IsValidatable ? ParsableCodeGenerator.InstanceForValidatableEnum : ParsableCodeGenerator.Instance);
 
+      if (!Settings.SkipIComparisonOperators && KeyProperty.HasComparisonOperators)
+         generators = generators.Add(ComparisonOperatorsCodeGenerator.Default);
+
       return generators;
    }
 }

src/Thinktecture.Runtime.Extensions/EnumGenerationAttribute.cs

@@ -1,3 +1,5 @@
+using System.Numerics;
+
 namespace Thinktecture;
 
 /// <summary>
@@ -33,6 +35,14 @@ public string KeyPropertyName
    /// </summary>
    public bool SkipIParsable { get; set; }
 
+   /// <summary>
+   /// Indication whether the generator should skip the implementation of <see cref="IComparisonOperators{TSelf,TOther,TResult}"/> or not.
+   ///
+   /// This setting has no effect if:
+   /// - the key is not an <see cref="IComparisonOperators{TSelf,TOther,TResult}"/> itself.
+   /// </summary>
+   public bool SkipIComparisonOperators { get; set; }
+
    /// <summary>
    /// Indication whether the generator should skip the implementation of <see cref="IFormattable"/> or not.
    ///