Merge pull request #36 from kntkymt/feature/prefix-operator-node-sizeof

CSTにsizeofのNodeを作る
kntkymt · Jan 22, 2024 · a471da8 · a471da8
2 parents 40aa815 + dea1ef1
commit a471da8
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 14 deletions.
diff --git a/Sources/Generator/Generator.swift b/Sources/Generator/Generator.swift
@@ -591,6 +591,12 @@ public final class Generator {
                 } else {
                     throw GenerateError.invalidSyntax(location: node.sourceTokens[0].sourceRange.start)
                 }
+
+            case .sizeof:
+                // FIXME: どうやって式の型を推測する？
+                // 今はとりあえず固定で8
+                result += "    mov x0, #8\n"
+                result += "    str x0, [sp, #-16]!\n"
             }
 
             return result

diff --git a/Sources/Parser/Node/ExpressionNode.swift b/Sources/Parser/Node/ExpressionNode.swift
@@ -36,6 +36,9 @@ public class PrefixOperatorExpressionNode: NodeProtocol {
 
         /// `&`
         case address
+
+        /// `sizeof`
+        case sizeof
     }
 
     // MARK: - Property
@@ -62,6 +65,9 @@ public class PrefixOperatorExpressionNode: NodeProtocol {
         case .reserved(.and):
             return .address
 
+        case .keyword(.sizeof):
+            return .sizeof
+
         default:
             fatalError()
         }

diff --git a/Sources/Parser/Parser.swift b/Sources/Parser/Parser.swift
@@ -614,12 +614,10 @@ public final class Parser {
 
         switch tokens[index].kind {
         case .keyword(.sizeof):
-            try consumeKeywordToken(.sizeof)
-
-            _ = try unary()
-
-            // FIXME: どうやって式の型を判断する？
-            return IntegerLiteralNode(literal: TokenNode(token: Token(kind: .number("8"), sourceRange: SourceRange(start: .startOfFile, end: .startOfFile))))
+            return PrefixOperatorExpressionNode(
+                operator: try consumeKeywordToken(.sizeof),
+                expression: try unary()
+            )
 
         case .reserved(.add):
             return PrefixOperatorExpressionNode(

diff --git a/Tests/ParserTest/OperatorsTest.swift b/Tests/ParserTest/OperatorsTest.swift
@@ -373,17 +373,18 @@ final class OperatorsTest: XCTestCase {
         )
         let node = try Parser(tokens: tokens).stmt()
 
+        XCTAssertEqual(node.sourceTokens, tokens)
+
         XCTAssertEqual(
             node,
             BlockItemNode(
-                item: IntegerLiteralNode(
-                    literal: TokenNode(token: Token(
-                        kind: .number("8"),
-                        sourceRange: SourceRange(
-                            start: SourceLocation(line: 1, column: 1),
-                            end: SourceLocation(line: 1, column: 1)
-                        )
-                    ))
+                item: PrefixOperatorExpressionNode(
+                    operator: TokenNode(token: tokens[0]),
+                    expression: TupleExpressionNode(
+                        parenthesisLeft: TokenNode(token: tokens[1]),
+                        expression: IntegerLiteralNode(literal: TokenNode(token: tokens[2])),
+                        parenthesisRight: TokenNode(token: tokens[3])
+                    )
                 ),
                 semicolon: TokenNode(token: tokens[4])
             )