[feat:root] initial Sugar

Author: uditkarode

axolotl.cabal

@@ -38,6 +38,8 @@ executable axl
                     , Transpiler.Transpiler
                     , Transpiler.Util
                     , Transpiler.Backends.JS.JS
+                   -- Desugarer
+                    , Sugar.Sugar
                    -- Miscellaneous
                     , Paths_axolotl
 

example.axl

@@ -4,7 +4,7 @@
 (print "this is my programming language, axolotl")
 
 (defun get-given-count [(argument-for-no-reason: int)] {
-  (+i argument-for-no-reason 1)
+  (+ argument-for-no-reason 1)
 })
 
 (def text1 "If I have ")
@@ -13,7 +13,7 @@
 (defun arbitrary-function [(initial: int)] {
   (print "expressions are written in braces")
   (print "and the last value is returned!")
-  (print (str text1 (+i initial 2) " apples and I give " (get-given-count 2) text2 (/f (+i initial 2) (get-given-count 4)) " apples!"))
+  (print (str text1 (+ initial 2) " apples and I give " (get-given-count 2) text2 (/ (+ initial 2) (get-given-count 4)) " apples!"))
   420
 })
 

hie.yaml

@@ -60,5 +60,8 @@ cradle:
     - path: "src/Transpiler/Backends/JS/JS.hs"
       component: "axolotl:exe:axl"
 
+    - path: "src/Sugar/Sugar.hs"
+      component: "axolotl:exe:axl"
+
     - path: "src/Paths_axolotl.hs"
       component: "axolotl:exe:axl"

src/Analyser/Analyser.hs

@@ -28,7 +28,6 @@ import Data.Either.Combinators (fromLeft', fromRight, fromRight', isLeft, maybeT
 import qualified Data.HashTable.IO as H
 import Data.Maybe (fromJust, fromMaybe, isJust)
 import Data.Text as T (Text, empty, pack, toLower, unpack)
-import Debug.Trace (trace)
 import Parser.Ast
   ( Expr (ArbitraryBlock, Array, Conditional, FunctionCall, FunctionDef, Nil, Root, VariableDef),
     VDataType (Bool, Function, Inferred, NilType),

src/Parser/Parser.hs

@@ -19,6 +19,7 @@ import Parser.Combinators
     squares,
     strLit,
   )
+import Sugar.Sugar (desugarExpr)
 import Text.Megaparsec (MonadParsec (observing, try), many, single, (<|>))
 import Text.Megaparsec.Char (numberChar)
 
@@ -100,7 +101,7 @@ arraySupportedExpr =
 
 expr :: Parser Expr
 expr =
-  try floatLiteral
+  desugarExpr <$> try floatLiteral
     <|> intLiteral
     <|> charLiteral
     <|> strLiteral
@@ -117,7 +118,7 @@ expr =
     <|> unary
 
 exprs :: Parser [Expr]
-exprs = many expr
+exprs = map desugarExpr <$> many expr
 
 root :: Parser Expr
 root = exprs >>= \x -> pure $ Root x

src/Sugar/Sugar.hs

@@ -0,0 +1,36 @@
+module Sugar.Sugar where
+
+import Analyser.Util (rFoldl)
+import qualified Data.Text as T
+import Parser.Ast (Expr (FloatLiteral, FunctionCall))
+
+mFilter :: [a] -> (a -> Bool) -> [a]
+mFilter a b = filter b a
+
+doesHaveFloat :: [Expr] -> Bool
+doesHaveFloat exprs =
+  not $
+    null
+      ( mFilter
+          exprs
+          ( \a -> do
+              case a of
+                FloatLiteral _ -> True
+                _ -> False
+          )
+      )
+
+desugarExpr :: Expr -> Expr
+desugarExpr e@(FunctionCall name args) = do
+  let mkFc =
+        if doesHaveFloat args
+          then FunctionCall (name <> "f") args
+          else FunctionCall (name <> "i") args
+
+  case name of
+    "+" -> mkFc
+    "-" -> mkFc
+    "*" -> mkFc
+    "/" -> FunctionCall "/f" args
+    _ -> e
+desugarExpr x = x