new: SourceInfo Less Equality

Author: cstml

lambda-buffers-compiler/lambda-buffers-compiler.cabal

@@ -94,9 +94,11 @@ library
   import:          common-imports
   build-depends:
     , containers                  >=0.6.5.1
+    , data-default                >=0.7
     , freer-simple                >=1.2
     , generic-arbitrary
     , generic-lens                >=2.2
+    , generics-sop                >=0.5
     , lambda-buffers-compiler-pb  >=0.1.0.0
     , mtl                         >=2.2
     , parsec                      >=3.1
@@ -119,6 +121,7 @@ library
     LambdaBuffers.Compiler.NamingCheck
     LambdaBuffers.Compiler.ProtoCompat
     LambdaBuffers.Compiler.ProtoCompat.FromProto
+    LambdaBuffers.Compiler.ProtoCompat.SILEq
     LambdaBuffers.Compiler.ProtoCompat.Types
     LambdaBuffers.Compiler.TypeClassCheck
     LambdaBuffers.Compiler.TypeClassCheck.Compat

lambda-buffers-compiler/src/LambdaBuffers/Compiler/ProtoCompat.hs

@@ -3,7 +3,9 @@
 module LambdaBuffers.Compiler.ProtoCompat (
   module FromProto,
   module Types,
+  module SILEq,
 ) where
 
 import LambdaBuffers.Compiler.ProtoCompat.FromProto as FromProto
+import LambdaBuffers.Compiler.ProtoCompat.SILEq as SILEq
 import LambdaBuffers.Compiler.ProtoCompat.Types as Types

lambda-buffers-compiler/src/LambdaBuffers/Compiler/ProtoCompat/SILEq.hs

@@ -0,0 +1,45 @@
+{-# LANGUAGE DefaultSignatures #-}
+{-# LANGUAGE UndecidableInstances #-}
+
+module LambdaBuffers.Compiler.ProtoCompat.SILEq (
+  (=:=),
+  SILEq (sileq),
+) where
+
+import Data.Bifunctor (Bifunctor (bimap))
+import Data.Map qualified as M
+import Data.Set qualified as S
+import Data.Text (Text)
+import Generics.SOP (All2, Generic (..), Proxy (..), hcmap, mapII)
+
+{- | SourceInfo Less Equality.
+ A TypeClass that enables equality without taking into consideration the SourceInfo inside a datatype.
+-}
+class Eq a => SILEq a where
+  sileq :: a -> a
+  default sileq :: (Generic a, All2 SILEq (Code a)) => a -> a
+  sileq = gsileq
+
+gsileq :: (Generic a, All2 SILEq (Code a)) => a -> a
+gsileq = to . hcmap (Proxy :: Proxy SILEq) (mapII sileq) . from
+
+{- | Equality without SourceInfo.
+ If the type does not contain SourceInfo then: (==) = (=:=).
+-}
+(=:=) :: SILEq a => a -> a -> Bool
+(=:=) a b = sileq a == sileq b
+
+instance SILEq a => SILEq [a] where
+  sileq = fmap sileq
+
+instance SILEq Int where
+  sileq = id
+
+instance SILEq Text where
+  sileq = id
+
+instance (Ord k, SILEq k, SILEq v) => SILEq (M.Map k v) where
+  sileq = M.fromList . fmap (bimap sileq sileq) . M.toList
+
+instance (Ord a, SILEq a) => SILEq (S.Set a) where
+  sileq = S.fromList . fmap sileq . S.toList