complex

Author: xsro

packages/jsmasm/.vscode/launch.json

@@ -20,7 +20,7 @@
                 "<node_internals>/**"
             ],
             "program": "${workspaceFolder}\\dist\\app.js",
-            "preLaunchTask": "tsc: build - tsconfig.json",
+            "preLaunchTask": "tsc: watch - tsconfig.json",
             "outFiles": [
                 "${workspaceFolder}/dist/**/*.js"
             ]

packages/jsmasm/src/app.ts

@@ -24,11 +24,11 @@ END  BEG
 `; // Replace with your file path
 let lexer = new Lexer(input);
 
-// let token: Token;
-// while ((token = lexer.nextToken()).type !== TokenType.EOF) {
-//   console.log(token);
-// }
-// lexer = new Lexer(input);
+let token: Token;
+while ((token = lexer.nextToken()).type !== TokenType.EOF) {
+  console.log(token);
+}
+lexer = new Lexer(input);
 
 const parser = new Parser(lexer);
 const ast = parser.parse();

packages/jsmasm/src/ast.ts

@@ -30,6 +30,7 @@ export interface Location {
     | ProcedureDeclaration
     | ProcedureEnd
     | DataDeclaration
+    | AssumeDirective
     | IncludeDirective;
 
   // 段声明
@@ -100,6 +101,24 @@ export interface Location {
     type: "ExpressionOperand";
     expression: Expression;
   }
+
+  export type DataExpression={
+    type:"DataExpression",
+    value:string
+    | NumericLiteralExpression
+    | DupExpression,
+    location:Location
+  }
+
+  export type DupExpression={
+    type:"DupExpression",
+    value:NumericLiteralExpression[]|string|DupQuestionMark
+    times:number
+  }
+
+  export type DupQuestionMark={
+    type:"DupQuestionMark"
+  }
 
   // 表达式类型
   export type Expression =
@@ -187,7 +206,7 @@ export interface Location {
     type: "DataDeclaration";
     name: string;
     dataType: string; // DB, DW, DD, ...
-    initialValue: Expression | string; // 可以是表达式或字符串
+    value: DataExpression[]; // 可以是表达式或字符串
   }
 
   // 指令
@@ -201,4 +220,9 @@ export interface Location {
   export interface IncludeDirective extends ASTNode {
     type: "IncludeDirective";
     filename: string;
+  }
+
+  export interface AssumeDirective extends ASTNode{
+    type:"AssumeDirective"
+    pair:{register:string,segment:string}[]
   }

packages/jsmasm/src/lexer.ts

@@ -66,9 +66,13 @@ export class Lexer {
       this.input = input;
     }
 
-    peek(): string {
+    private peek(): string {
       return this.input[this.position];
     }
+
+    public nextchar():string{
+      return this.input[this.position+1];
+    }
 
     private advance(): string {
       const char = this.peek();
@@ -87,15 +91,15 @@ export class Lexer {
     }
 
     private isDigit(char: string): boolean {
-      return /[0-9]/.test(char);
+      return /[0-9A-Fa-fHBhb]/.test(char);
     }
 
     private isAlphanumeric(char: string): boolean {
       return this.isAlpha(char) || this.isDigit(char);
     }
 
     private isWhitespace(char: string): boolean {
-      return /\s/.test(char);
+      return /[\s]/.test(char);
     }
 
     private skipWhitespace(): void {
@@ -105,6 +109,10 @@ export class Lexer {
     }
 
     public nextToken(): Token {
+      if (this.peek() === "\n") {
+        this.advance();
+        return {type: TokenType.NEWLINE, value: "\n", location: {line:this.line, column:this.column}}
+      }
       this.skipWhitespace();
 
       if (this.position >= this.input.length) {
@@ -223,11 +231,6 @@ export class Lexer {
         }
       }
 
-      if (char === "\n") {
-          this.advance();
-          return {type: TokenType.NEWLINE, value: "\n", location: {line, column}}
-      }
-  
       return { type: TokenType.ERROR, value: this.advance(), location: { line, column } };
     }
   }

packages/jsmasm/src/parser.ts

@@ -2,6 +2,11 @@
 import { Lexer, Token, TokenType } from "./lexer";
 import * as AST from "./ast";
 
+
+function parseNumber(input: string) {
+  return parseInt(input)
+}
+
 export class Parser {
   private lexer: Lexer;
   private currentToken: Token;
@@ -11,12 +16,12 @@ export class Parser {
   constructor(lexer: Lexer) {
     this.lexer = lexer;
     this.currentToken = this.lexer.nextToken();
-    this.nextToken=this.lexer.nextToken();
+    this.nextToken = this.lexer.nextToken();
   }
 
-  public next(){
-    this.currentToken=this.nextToken;
-    this.nextToken=this.lexer.nextToken();
+  public next() {
+    this.currentToken = this.nextToken;
+    this.nextToken = this.lexer.nextToken();
     return this.currentToken
   }
 
@@ -54,6 +59,45 @@ export class Parser {
     };
   }
 
+  private parseDataDeclaration(): AST.DataDeclaration {
+    const name = this.eat(TokenType.IDENTIFIER).value;
+    const dataType = this.eat(this.currentToken.type).value;
+    let value: AST.DataExpression[] = []
+    while (this.currentToken.type !== TokenType.NEWLINE) {
+      if (this.currentToken.type === TokenType.COMMA) {
+        this.eat(TokenType.COMMA)
+        continue
+      }
+      if (this.currentToken.type === TokenType.STRING) {
+        let a: AST.DataExpression = {
+          type: "DataExpression",
+          value: this.eat(TokenType.STRING).value,
+          location: this.currentToken.location
+        }
+        value.push(a)
+      }
+      if (this.currentToken.type === TokenType.NUMBER) {
+        let a: AST.DataExpression = {
+          type: "DataExpression",
+          value: {
+            "type": "NumericLiteralExpression",
+            value: parseNumber(this.eat(TokenType.NUMBER).value),
+            location: this.currentToken.location
+          },
+          location: this.currentToken.location
+        }
+        value.push(a)
+      }
+    }
+    return {
+      type: "DataDeclaration",
+      name,
+      dataType,
+      value,
+      location: this.currentToken.location,
+    };
+  }
+
   private parseSegmentEnd(): AST.SegmentEnd {
     const name = this.eat(TokenType.IDENTIFIER).value;
     this.eat(TokenType.ENDS);
@@ -76,6 +120,22 @@ export class Parser {
     return { type: "Instruction", mnemonic, operands, location: this.currentToken.location };
   }
 
+  private parseAssume(): AST.AssumeDirective {
+    let out: AST.AssumeDirective = {
+      "type": "AssumeDirective",
+      "pair": [],
+      "location": this.currentToken.location
+    };
+    this.eat(this.currentToken.type)
+    while(this.currentToken.type!==TokenType.NEWLINE){
+      const register=this.eat(this.currentToken.type).value;
+      this.eat(TokenType.COLON)
+      const segment=this.eat(this.currentToken.type).value
+      out.pair.push({register,segment})
+    }
+    return out
+  }
+
   private parseOperand(): AST.Operand {
     if (this.currentToken.type === TokenType.IDENTIFIER) {
       if (this.nextToken.value === ":") {
@@ -131,7 +191,7 @@ export class Parser {
     const name = this.eat(TokenType.DIRECTIVE).value;
     const args: AST.Expression[] = [];
     if (this.currentToken.type === TokenType.STRING) {
-        args.push({type: "StringLiteralExpression", name:"debug!!!!",value: this.eat(TokenType.STRING).value, location: this.currentToken.location});
+      args.push({ type: "StringLiteralExpression", name: "debug!!!!", value: this.eat(TokenType.STRING).value, location: this.currentToken.location });
     }
 
     return { type: "Directive", name, arguments: args, location: this.currentToken.location };
@@ -151,7 +211,12 @@ export class Parser {
     if (this.currentToken.type === TokenType.IDENTIFIER) {
       if (this.nextToken.value === "SEGMENT") {
         return this.parseSegmentDeclaration();
-      } else if (this.nextToken.value === "ENDS") {
+      }
+      else if ([TokenType.DB, TokenType.DW, TokenType.DD].some(a => a === this.nextToken.type)) {
+        return this.parseDataDeclaration();
+      }
+
+      else if (this.nextToken.value === "ENDS") {
         return this.parseSegmentEnd();
       } else if (this.nextToken.value === ":") {
         return this.parseLabel();
@@ -164,6 +229,8 @@ export class Parser {
       } else {
         return this.parseDirective();
       }
+    }else if(this.currentToken.type===TokenType.ASSUME){
+      return this.parseAssume()
     }
     return null;
   }