implemented define

src/index.ts

@@ -73,6 +73,7 @@ async function main() {
     let code = Deno.readTextFileSync(input);
 	const preprocessor = new Preprocessor(includes);
 	code = preprocessor.preprocess(code);
+    // Deno.writeTextFileSync(output + ".pp", code);
     const lexer = new Lexer(code);
     const tokens = lexer.tokenize();
     Deno.writeTextFileSync(output + ".tokens.json", JSON.stringify(tokens, undefined, 4));

src/preprocessor.ts

@@ -1,3 +1,8 @@
+interface Define {
+	name: string;
+	value: string;
+}
+
 export class Preprocessor {
 	include_paths: string[];
 	included_files: string[];
@@ -15,7 +20,8 @@ export class Preprocessor {
 		}
 	}
 
-	preprocess(code: string): string {
+
+	preprocessIncludes(code: string): string {
 		const matches = code.match(/\$include ?<[\w/\.]*.\w*>/g);
 		if (matches) {
 			for (let i = 0; i < matches.length; i++) {
@@ -38,10 +44,46 @@ export class Preprocessor {
 
 				if (!this.included_files.includes(inc)) {
 					this.included_files.push(inc);
-					code += "\n" + this.preprocess(ncode);
+					code += "\n" + this.preprocessIncludes(ncode);
 				}
 			}
 		}
 		return code;
 	}
+
+	preprocessDefines(code: string): string {
+		const matches = code.match(/\$define ([^ ]*) (.*)/g);
+
+		const defines: Define[] = [];
+
+		if (matches) {
+			for (let i = 0; i < matches.length; i++) {
+				code = code.replace(matches[i], "");
+
+				const defineSplit = matches[i].split(" ");
+				defineSplit.shift();
+
+				const defineName = defineSplit.shift() as string;
+				const defineValue = defineSplit.join(" ");
+
+				defines.push({
+					name: defineName,
+					value: defineValue
+				});
+			}
+		}
+
+		for (const define of defines) {
+			code = code.replaceAll(define.name, define.value);
+		}
+
+		return code;
+	}
+
+
+	preprocess(code: string): string {
+		code = this.preprocessIncludes(code);
+		code = this.preprocessDefines(code);
+		return code;
+	}
 }

tests/defines.fl

@@ -0,0 +1,12 @@
+$include <std.fl>
+
+$define STRING_TEST "Hello world!"
+$define INT_TEST 1234
+$define OTHER_TEST printi
+
+function spark(int argc, str[] argv) -> int {
+	prints(STRING_TEST);
+    printi(INT_TEST);
+    OTHER_TEST(INT_TEST);
+	return 0;
+}

tests/defines.fl.expect

@@ -0,0 +1,5 @@
+{
+	"arguments": [],
+	"output": ["Hello world!", "1234", "1234"],
+	"should_fail": false
+}