From 0bfcce5c35f5f4b2d7714fff40788ab09682dccf Mon Sep 17 00:00:00 2001
From: SophieBosio <sophie.bosio@outlook.com>
Date: Mon, 15 Jan 2024 14:51:33 +0100
Subject: [PATCH] Unification module: Missing unify for most cases

---
 src/Interpreter.hs |  7 ++++-
 src/Unification.hs | 75 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 81 insertions(+), 1 deletion(-)

diff --git a/src/Interpreter.hs b/src/Interpreter.hs
index 50facc8..49dc061 100644
--- a/src/Interpreter.hs
+++ b/src/Interpreter.hs
@@ -1,6 +1,7 @@
 module Interpreter where
 
 import Syntax
+import Unification
 
 import Control.Monad.Reader
 
@@ -25,7 +26,7 @@ evaluate (Pattern (Variable x _)) =
 evaluate (Pattern (Pair t0 t1 a)) =
   do v0 <- evaluate t0
      v1 <- evaluate t1
-     return $ (Pattern (Pair v0 v1 a))
+     return $ Pattern (Pair v0 v1 a)
 evaluate (Pattern (Constructor c ps a)) =
   do ts  <- mapM evaluate (Pattern <$> ps) 
      ps' <- mapM (return . strengthenToPattern) ts
@@ -107,6 +108,10 @@ substitute x t v = -- computes t[v/x]
     subs = flip (substitute x) v
     manipulateWith f = strengthenToPattern . f . weakenToTerm
 
+-- substituteWithUnifier :: Unifier a -> Term a -> Term a
+-- substituteWithUnifier xs t =
+--   foldr (\(x, v) t' -> substitute x t' v) t xs
+
 
 -- Utility functions
 bool :: Show a => Term a -> Runtime a Bool
diff --git a/src/Unification.hs b/src/Unification.hs
index 0593294..1670afd 100644
--- a/src/Unification.hs
+++ b/src/Unification.hs
@@ -1 +1,76 @@
 module Unification where
+
+import Syntax
+
+
+-- Abbreviations
+-- Meta is a placeholder for the Pattern/Term constructors
+type Transformation meta a = (meta a -> meta a)
+
+data PatternMatch a =
+    NoMatch
+  | MatchBy (Transformation Pattern a)
+
+errorMessage :: Term a -> Pattern a -> b
+errorMessage p q = error $
+  "couldn't match " ++ show p ++ " against pattern " ++ show q
+
+type Unifier a = Maybe (a -> a)
+
+newtype Substitution meta a = Substitution { unifier :: Unifier (meta a) }
+
+
+-- Export
+patternMatch :: Term a -> Pattern a -> PatternMatch a
+patternMatch p q = maybe NoMatch MatchBy (unifier $ unify p q)
+
+
+-- Unification
+unify :: Term a -> Pattern a -> Substitution Pattern a
+unify (Pattern p1) = unify' p1
+-- unify Rec         X      (T0 a)            a
+-- unify Let         X      (T1 a) (T2 a)     a
+-- unify Application (T1 a) (T2 a)            a
+-- unify Case        (T0 a) [(Alt a, Body a)] a
+--   -- Utilities:
+-- unify Fst         (T0 a)                   a
+-- unify Snd         (T0 a)                   a
+-- unify Plus        (T0 a) (T1 a)            a
+-- unify Minus       (T0 a) (T1 a)            a 
+-- unify Lt          (T0 a) (T1 a)            a
+-- unify Gt          (T0 a) (T1 a)            a
+-- unify Equal       (T0 a) (T1 a)            a
+-- unify Not         (T0 a)                   a
+
+unify' :: Pattern a -> Pattern a -> Substitution Pattern a
+unify' (Variable x _) (Variable y _) | x == y = mempty
+unify' (Variable x _) p              | not $ p `contains` x = p `substitutes` x
+unify' p              (Variable x _) | not $ p `contains` x = p `substitutes` x
+-- unify' (Constructor c ps _) (Constructor c' ps' _)
+--   | c == c' && length ps == length ps'
+--   = 
+-- unify' (Unit _)
+-- unify' (Number n _)
+-- unify' (Boolean b _)
+-- unify' (Pair t0 t1)
+
+
+-- Substitution
+instance Semigroup (Substitution meta a) where
+  s <> s' = Substitution $ (.) <$> unifier s <*> unifier s'
+
+instance Monoid (Substitution meta a) where
+  mempty = Substitution $ return id
+  mappend = (<>)
+
+substitutes :: Pattern a -> X -> Substitution Pattern a
+substitutes = undefined
+
+
+-- Utility functions
+isMatch :: PatternMatch meta -> Bool
+isMatch NoMatch = False
+isMatch _       = True
+
+contains :: Pattern a -> X -> Bool
+contains = undefined