globals are now variable

1 files changed, 37 insertions, 19 deletions

diff --git a/com.c b/com.c
index ba45975..4252ecd 100644
--- a/com.c
+++ b/com.c
@@ -35,26 +35,44 @@ static void compile_node(State *S, Chunk *ch, AstNode a) {
 			CK(l.len == 2, "print requires exactly 1 argument");
 			compile_node(S, ch, l.vals[1]);
 			chunk_wbc(S, ch, OP_PRINT);
-			break;
-		}
-		CK(l.len == 3, "can only compile binary ops");
-		char opchar = l.vals[0].as.str[0];
-		Op op;
-		switch (opchar) {
-		#define OP(char, code) case char: op = code; break;
-			OP('+', OP_ADD)
-			OP('-', OP_SUB)
-			OP('*', OP_MUL)
-			OP('/', OP_DIV)
-		#undef OP
-			default:
-				printf("unkown op %s\n",l.vals[0].as.str);
-				exit(1);
-				break;
+		} else if (0 == strcmp(l.vals[0].as.str, "set")) {
+			CK(l.len == 3, "set requires exactly 2 arguments");
+			AstNode ident = l.vals[1];
+			CK(ident.ty == AST_IDENT, "set's first argument must be identifier");
+
+			size_t len = strlen(ident.as.str);
+			ObjString *o = objstring_copy(S, ident.as.str, len);
+
+			compile_node(S, ch, l.vals[2]);
+			chunk_wbc(S, ch, OP_SETGLOBAL);
+			chunk_wbc(S, ch, chunk_wconst(S, ch, VAL_OBJ(o)));
+		} else if (0 == strcmp(l.vals[0].as.str, "do")) {
+			for (int i = 1; i < l.len - 1; i++) {
+				compile_node(S, ch, l.vals[i]);
+				chunk_wbc(S, ch, OP_DROP);
+			}
+			compile_node(S, ch, l.vals[l.len - 1]);
+		} else {
+				
+			CK(l.len == 3, "can only compile binary ops");
+			char opchar = l.vals[0].as.str[0];
+			Op op;
+			switch (opchar) {
+			#define OP(char, code) case char: op = code; break;
+				OP('+', OP_ADD)
+				OP('-', OP_SUB)
+				OP('*', OP_MUL)
+				OP('/', OP_DIV)
+			#undef OP
+				default:
+					printf("unkown op %s\n",l.vals[0].as.str);
+					exit(1);
+					break;
+			}
+			compile_node(S, ch, l.vals[1]);
+			compile_node(S, ch, l.vals[2]);
+			chunk_wbc(S, ch, op);
 		}
-		compile_node(S, ch, l.vals[1]);
-		compile_node(S, ch, l.vals[2]);
-		chunk_wbc(S, ch, op);
 		#undef CK
 	}
 	}