src/Typecheck.hs

@@ -78,6 +78,22 @@ typeCheckExpression (BinaryOperation op expr1 expr2) symtab classes =
         Or -> checkLogicalOperation op expr1' expr2' type1 type2
         NameResolution -> resolveNameResolution expr1' expr2 symtab classes
 
+typeCheckExpression (UnaryOperation op expr) symtab classes =
+  let expr' = typeCheckExpression expr symtab classes
+      type' = getTypeFromExpr expr'
+   in case op of
+        Not ->
+          if type' == "boolean"
+            then
+              TypedExpression "boolean" (UnaryOperation op expr')
+            else
+              error "Logical NOT operation requires an operand of type boolean"
+        Minus ->
+          if type' == "int" || type' == "char"
+            then
+              TypedExpression type' (UnaryOperation op expr')
+            else
+              error "Unary minus operation requires an operand of type int or char"
 
 typeCheckExpression (StatementExpressionExpression stmtExpr) symtab classes =
   let stmtExpr' = typeCheckStatementExpression stmtExpr symtab classes
@@ -147,59 +163,39 @@ typeCheckStatementExpression (MethodCall expr methodName args) symtab classes =
                 Nothing -> error $ "Class for object type '" ++ objType ++ "' not found."
         _ -> error "Invalid object type for method call. Object must have a class type."
 
-typeCheckStatementExpression (UnaryOperation op expr) symtab classes =
+typeCheckStatementExpression (PostIncrement expr) symtab classes =
   let expr' = typeCheckExpression expr symtab classes
       type' = getTypeFromExpr expr'
-   in case op of
-        Not ->
-          if type' == "boolean"
+   in if type' == "int" || type' == "char"
         then
-              TypedStatementExpression "boolean" (UnaryOperation op expr')
-            else
-              error "Logical NOT operation requires an operand of type boolean"
-        Minus ->
-          if type' == "int" 
-            then
-              TypedStatementExpression "int" (UnaryOperation op expr')
-          else if type' == "char" 
-            then
-              TypedStatementExpression "char" (UnaryOperation op expr')
-            else
-              error "Unary minus operation requires an operand of type int or char"
-        PostIncrement ->
-          if type' == "int"
-            then
-              TypedStatementExpression "int" (UnaryOperation op expr')
-          else if type' == "char"
-            then
-              TypedStatementExpression "char" (UnaryOperation op expr')
+          TypedStatementExpression type' (PostIncrement expr')
         else
           error "Post-increment operation requires an operand of type int or char"
-        PostDecrement ->
-          if type' == "int"
+
+typeCheckStatementExpression (PostDecrement expr) symtab classes =
+  let expr' = typeCheckExpression expr symtab classes
+      type' = getTypeFromExpr expr'
+   in if type' == "int" || type' == "char"
         then
-              TypedStatementExpression "int" (UnaryOperation op expr')
-          else if type' == "char"
-            then
-              TypedStatementExpression "char" (UnaryOperation op expr')
+          TypedStatementExpression type' (PostDecrement expr')
         else
           error "Post-decrement operation requires an operand of type int or char"
-        PreIncrement ->
-          if type' == "int"
+
+typeCheckStatementExpression (PreIncrement expr) symtab classes =
+  let expr' = typeCheckExpression expr symtab classes
+      type' = getTypeFromExpr expr'
+   in if type' == "int" || type' == "char"
         then
-              TypedStatementExpression "int" (UnaryOperation op expr')
-          else if type' == "char"
-            then
-              TypedStatementExpression "char" (UnaryOperation op expr')
+          TypedStatementExpression type' (PreIncrement expr')
         else
           error "Pre-increment operation requires an operand of type int or char"
-        PreDecrement ->
-          if type' == "int"
+
+typeCheckStatementExpression (PreDecrement expr) symtab classes =
+  let expr' = typeCheckExpression expr symtab classes
+      type' = getTypeFromExpr expr'
+   in if type' == "int" || type' == "char"
         then
-              TypedStatementExpression "int" (UnaryOperation op expr')
-          else if type' == "char"
-            then
-              TypedStatementExpression "char" (UnaryOperation op expr')
+          TypedStatementExpression type' (PreDecrement expr')
         else
           error "Pre-decrement operation requires an operand of type int or char"