Reset parser more completely after errors

Clear up lexer and code gen state too.

Signed-off-by: Keith Packard <keithp@keithp.com>

snek-code.c

@@ -401,6 +401,14 @@ snek_code_patch_forward(snek_offset_t start, snek_offset_t stop, snek_forward_t
 	}
 }
 
+void
+snek_code_reset(void)
+{
+	snek_compile_size = 0;
+	compile_alloc = 0;
+	snek_compile = NULL;
+}
+
 snek_code_t *
 snek_code_finish(void)
 {
@@ -416,9 +424,7 @@ snek_code_finish(void)
 		snek_code_dump(code);
 #endif
 	}
-	snek_compile_size = 0;
-	compile_alloc = 0;
-	snek_compile = NULL;
+	snek_code_reset();
 	return code;
 }
 

snek-parse.c

@@ -161,12 +161,22 @@ snek_parse(void)
 	for (;;) {
 		/* Reset lex state */
 		snek_current_indent = 0;
+		snek_lex_indent = 0;
 		snek_ignore_nl = 0;
-		snek_abort = false;
+		snek_lex_midline = false;
+		snek_lex_exdent = false;
+
+		/* Reset parse state */
 		snek_parse_middle = false;
 		value_stack_p = 0;
 		for_depth = 0;
 
+		/* Reset codegen state */
+		snek_code_reset();
+
+		/* Reset error state */
+		snek_abort = false;
+
 		parse_return_t ret = parse(NULL);
 
 		switch (ret) {
@@ -185,6 +195,9 @@ snek_parse(void)
 		default:
 			snek_error_syntax(snek_lex_text);
 			{
+				/* Skip input until we get back to
+				 * zero indent
+				 */
 				token_t token;
 				for (;;) {
 					snek_ignore_nl = 0;

snek.h

@@ -432,6 +432,9 @@ snek_code_add_in_enum(snek_id_t id, uint8_t for_depth);
 void
 snek_code_patch_branch(snek_offset_t branch, snek_offset_t target);
 
+void
+snek_code_reset(void);
+
 snek_code_t *
 snek_code_finish(void);