diff --git a/js.h b/js.h index 33ac52b..b0c6007 100644 --- a/js.h +++ b/js.h @@ -11,6 +11,7 @@ typedef struct js_State js_State; typedef struct js_StringNode js_StringNode; +typedef struct js_Ast js_Ast; typedef int (*js_CFunction)(js_State *J); diff --git a/jsdump.c b/jsdump.c new file mode 100644 index 0000000..c534599 --- /dev/null +++ b/jsdump.c @@ -0,0 +1,570 @@ +#include "js.h" +#include "jsparse.h" + +#include + +static void pstmlist(int d, js_Ast *list); +static void pexpi(int d, int i, js_Ast *exp); +static void pstm(int d, js_Ast *stm); +static void slist(int d, js_Ast *list); +static void sblock(int d, js_Ast *list); + +static inline void pc(int c) +{ + putchar(c); +} + +static inline void ps(const char *s) +{ + fputs(s, stdout); +} + +static inline void in(int d) +{ + while (d-- > 0) + putchar('\t'); +} + +static inline void nl(void) +{ + putchar('\n'); +} + +static void pargs(int d, js_Ast *list) +{ + while (list) { + assert(list->type == AST_LIST); + pexpi(d, 0, list->a); + list = list->b; + if (list) + ps(", "); + } +} + +static void parray(int d, js_Ast *list) +{ + ps("["); + while (list) { + assert(list->type == AST_LIST); + pexpi(d, 0, list->a); + list = list->b; + if (list) + ps(", "); + } + ps("]"); +} + +static void pobject(int d, js_Ast *list) +{ + ps("{"); + while (list) { + js_Ast *kv = list->a; + assert(list->type == AST_LIST); + switch (kv->type) { + case EXP_PROP_VAL: + pexpi(d, 0, kv->a); + ps(": "); + pexpi(d, 0, kv->b); + break; + case EXP_PROP_GET: + ps("get "); + pexpi(d, 0, kv->a); + ps("() {\n"); + pstmlist(d, kv->b); + in(d); ps("}"); + break; + case EXP_PROP_SET: + ps("set "); + pexpi(d, 0, kv->a); + ps("("); + pexpi(d, 0, kv->b); + ps(") {\n"); + pstmlist(d, kv->c); + in(d); ps("}"); + break; + } + list = list->b; + if (list) + ps(", "); + } + ps("}"); +} + +static void pstr(const char *s) +{ + pc('"'); + while (*s) { + if (*s == '"') + ps("\\\""); + else + pc(*s); + ++s; + } + pc('"'); +} + +static void pbin(int d, int i, js_Ast *exp, const char *op) +{ + if (i) pc('('); + pexpi(d, 1, exp->a); + ps(op); + pexpi(d, 1, exp->b); + if (i) pc(')'); +} + +static void puna(int d, int i, js_Ast *exp, const char *pre, const char *suf) +{ + if (i) pc('('); + ps(pre); + pexpi(d, 1, exp->a); + ps(suf); + if (i) pc(')'); +} + +static void pexpi(int d, int i, js_Ast *exp) +{ + switch (exp->type) { + case AST_IDENTIFIER: ps(exp->string); break; + case AST_NUMBER: printf("%.9g", exp->number); break; + case AST_STRING: pstr(exp->string); break; + case AST_REGEXP: pc('/'); ps(exp->string); pc('/'); break; + + case EXP_NULL: ps("null"); break; + case EXP_TRUE: ps("true"); break; + case EXP_FALSE: ps("false"); break; + case EXP_THIS: ps("this"); break; + + case EXP_OBJECT: pobject(d, exp->a); break; + case EXP_ARRAY: parray(d, exp->a); break; + + case EXP_DELETE: puna(d, i, exp, "delete ", ""); break; + case EXP_VOID: puna(d, i, exp, "void ", ""); break; + case EXP_TYPEOF: puna(d, i, exp, "typeof ", ""); break; + case EXP_PREINC: puna(d, i, exp, "++", ""); break; + case EXP_PREDEC: puna(d, i, exp, "--", ""); break; + case EXP_POSTINC: puna(d, i, exp, "", "++"); break; + case EXP_POSTDEC: puna(d, i, exp, "", "--"); break; + case EXP_POS: puna(d, i, exp, "+", ""); break; + case EXP_NEG: puna(d, i, exp, "-", ""); break; + case EXP_BITNOT: puna(d, i, exp, "~", ""); break; + case EXP_LOGNOT: puna(d, i, exp, "!", ""); break; + + case EXP_LOGOR: pbin(d, i, exp, " || "); break; + case EXP_LOGAND: pbin(d, i, exp, " && "); break; + case EXP_BITOR: pbin(d, i, exp, " | "); break; + case EXP_BITXOR: pbin(d, i, exp, " ^ "); break; + case EXP_BITAND: pbin(d, i, exp, " & "); break; + case EXP_EQ: pbin(d, i, exp, " == "); break; + case EXP_NE: pbin(d, i, exp, " != "); break; + case EXP_EQ3: pbin(d, i, exp, " === "); break; + case EXP_NE3: pbin(d, i, exp, " !== "); break; + case EXP_LT: pbin(d, i, exp, " < "); break; + case EXP_GT: pbin(d, i, exp, " > "); break; + case EXP_LE: pbin(d, i, exp, " <= "); break; + case EXP_GE: pbin(d, i, exp, " >= "); break; + case EXP_INSTANCEOF: pbin(d, i, exp, " instanceof "); break; + case EXP_IN: pbin(d, i, exp, " in "); break; + case EXP_SHL: pbin(d, i, exp, " << "); break; + case EXP_SHR: pbin(d, i, exp, " >> "); break; + case EXP_USHR: pbin(d, i, exp, " >>> "); break; + case EXP_ADD: pbin(d, i, exp, " + "); break; + case EXP_SUB: pbin(d, i, exp, " - "); break; + case EXP_MUL: pbin(d, i, exp, " * "); break; + case EXP_DIV: pbin(d, i, exp, " / "); break; + case EXP_MOD: pbin(d, i, exp, " % "); break; + case EXP_ASS: pbin(d, i, exp, " = "); break; + case EXP_ASS_MUL: pbin(d, i, exp, " *= "); break; + case EXP_ASS_DIV: pbin(d, i, exp, " /= "); break; + case EXP_ASS_MOD: pbin(d, i, exp, " %= "); break; + case EXP_ASS_ADD: pbin(d, i, exp, " += "); break; + case EXP_ASS_SUB: pbin(d, i, exp, " -= "); break; + case EXP_ASS_SHL: pbin(d, i, exp, " <<= "); break; + case EXP_ASS_SHR: pbin(d, i, exp, " >>= "); break; + case EXP_ASS_USHR: pbin(d, i, exp, " >>>= "); break; + case EXP_ASS_BITAND: pbin(d, i, exp, " &= "); break; + case EXP_ASS_BITXOR: pbin(d, i, exp, " ^= "); break; + case EXP_ASS_BITOR: pbin(d, i, exp, " |= "); break; + + case EXP_COMMA: pbin(d, 1, exp, ", "); break; + + case EXP_COND: + if (i) pc('('); + pexpi(d, 1, exp->a); + ps(" ? "); + pexpi(d, 1, exp->b); + ps(" : "); + pexpi(d, 1, exp->c); + if (i) pc(')'); + break; + + case EXP_INDEX: + if (i) pc('('); + pexpi(d, 1, exp->a); + pc('['); + pexpi(d, 0, exp->b); + pc(']'); + if (i) pc(')'); + break; + + case EXP_MEMBER: + if (i) pc('('); + pexpi(d, 1, exp->a); + pc('.'); + pexpi(d, 1, exp->b); + if (i) pc(')'); + break; + + case EXP_CALL: + if (i) pc('('); + pexpi(d, 1, exp->a); + pc('('); + pargs(d, exp->b); + pc(')'); + if (i) pc(')'); + break; + + case EXP_NEW: + if (i) pc('('); + ps("new "); + pexpi(d, 1, exp->a); + pc('('); + pargs(d, exp->b); + pc(')'); + if (i) pc(')'); + break; + + case EXP_FUNC: + ps("(function "); + if (exp->a) pexpi(d, 1, exp->a); + pc('('); + pargs(d, exp->b); + ps(") {\n"); + pstmlist(d, exp->c); + in(d); ps("})"); + break; + + default: + ps(""); + break; + } +} + +static void pexp(int d, js_Ast *exp) +{ + pexpi(d, 0, exp); +} + +static void pvar(int d, js_Ast *var) +{ + assert(var->type == EXP_VAR); + pexp(d, var->a); + if (var->b) { + ps(" = "); + pexp(d, var->b); + } +} + +static void pvarlist(int d, js_Ast *list) +{ + while (list) { + assert(list->type == AST_LIST); + pvar(d, list->a); + list = list->b; + if (list) + ps(", "); + } +} + +static void pblock(int d, js_Ast *block) +{ + assert(block->type == STM_BLOCK); + in(d); ps("{\n"); + pstmlist(d, block->a); + in(d); pc('}'); +} + +static void pstmh(int d, js_Ast *stm) +{ + if (stm->type == STM_BLOCK) + pblock(d, stm); + else + pstm(d+1, stm); +} + +static void pcaselist(int d, js_Ast *list) +{ + while (list) { + js_Ast *stm = list->a; + if (stm->type == STM_CASE) { + in(d); ps("case "); pexp(d, stm->a); ps(":\n"); + pstmlist(d, stm->b); + } + if (stm->type == STM_DEFAULT) { + in(d); ps("default:\n"); + pstmlist(d, stm->a); + } + list = list->b; + } +} + +static void pstm(int d, js_Ast *stm) +{ + if (stm->type == STM_BLOCK) { + pblock(d, stm); + return; + } + + in(d); + + switch (stm->type) { + case STM_FUNC: + ps("function "); + pexpi(d, 1, stm->a); + pc('('); + pargs(d, stm->b); + ps(")\n"); + in(d); ps("{\n"); + pstmlist(d, stm->c); + in(d); ps("}"); + break; + + case STM_NOP: + pc(';'); + break; + + case STM_VAR: + ps("var "); + pvarlist(d, stm->a); + ps(";"); + break; + + case STM_IF: + ps("if ("); pexp(d, stm->a); ps(")\n"); + pstmh(d, stm->b); + if (stm->c) { + nl(); in(d); ps("else\n"); + pstmh(d, stm->c); + } + break; + + case STM_DO: + ps("do\n"); + pstmh(d, stm->a); + nl(); + in(d); ps("while ("); pexp(d, stm->b); ps(");"); + break; + + case STM_WHILE: + ps("while ("); pexp(d, stm->a); ps(")\n"); + pstmh(d, stm->b); + break; + + case STM_FOR: + ps("for ("); + pexp(d, stm->a); ps("; "); + pexp(d, stm->b); ps("; "); + pexp(d, stm->c); ps(")\n"); + pstmh(d, stm->d); + break; + case STM_FOR_VAR: + ps("for (var "); + pvarlist(d, stm->a); ps("; "); + pexp(d, stm->b); ps("; "); + pexp(d, stm->c); ps(")\n"); + pstmh(d, stm->d); + break; + case STM_FOR_IN: + ps("for ("); + pexp(d, stm->a); ps(" in "); + pexp(d, stm->b); ps(")\n"); + pstmh(d, stm->c); + break; + case STM_FOR_IN_VAR: + ps("for (var "); + pvarlist(d, stm->a); ps(" in "); + pexp(d, stm->b); ps(")\n"); + pstmh(d, stm->c); + break; + + case STM_CONTINUE: + if (stm->a) { + ps("continue "); pexp(d, stm->a); ps(";"); + } else { + ps("continue;"); + } + break; + + case STM_BREAK: + if (stm->a) { + ps("break "); pexp(d, stm->a); ps(";"); + } else { + ps("break;"); + } + break; + + case STM_RETURN: + if (stm->a) { + ps("return "); pexp(d, stm->a); ps(";"); + } else { + ps("return;"); + } + break; + + case STM_WITH: + ps("with ("); pexp(d, stm->a); ps(")\n"); + pblock(d, stm->b); + break; + + case STM_SWITCH: + ps("switch ("); + pexp(d, stm->a); + ps(")\n"); + in(d); ps("{\n"); + pcaselist(d, stm->b); + in(d); ps("}"); + break; + + case STM_THROW: + ps("throw "); pexp(d, stm->a); ps(";"); + break; + + case STM_TRY: + ps("try\n"); + pstmh(d, stm->a); + if (stm->b && stm->c) { + nl(); in(d); ps("catch ("); pexp(d, stm->b); ps(")\n"); + pstmh(d, stm->c); + } + if (stm->d) { + nl(); in(d); ps("finally\n"); + pstmh(d, stm->d); + } + break; + + case STM_LABEL: + pexp(d, stm->a); ps(": "); pstm(d, stm->b); + break; + + case STM_DEBUGGER: + ps("debugger;"); + break; + + default: + pexp(d, stm); pc(';'); + } +} + +static void pstmlist(int d, js_Ast *list) +{ + while (list) { + assert(list->type == AST_LIST); + pstm(d+1, list->a); + nl(); + list = list->b; + } +} + +void jsP_dumpsyntax(js_State *J, js_Ast *prog) +{ + if (prog->type == AST_LIST) + pstmlist(-1, prog); + else { + pstm(0, prog); + pc('\n'); + } +} + +static const char *stype[] = { + "list", "ident", "number", "string", "regexp", "null", + "true", "false", "this", "array", "object", + "prop_val", "prop_get", "prop_set", "index", + "member", "call", "new", "funexp", "delete", + "void", "typeof", "preinc", "predec", "postinc", + "postdec", "pos", "neg", "bitnot", "lognot", + "logor", "logand", "bitor", "bitxor", "bitand", + "eq", "ne", "eq3", "ne3", "lt", "gt", "le", + "ge", "instanceof", "in", "shl", "shr", "ushr", + "add", "sub", "mul", "div", "mod", "cond", + "ass", "ass_mul", "ass_div", "ass_mod", "ass_add", + "ass_sub", "ass_shl", "ass_shr", "ass_ushr", + "ass_bitand", "ass_bitxor", "ass_bitor", "comma", + "var", "block", "fundec", "nop", "var", "if", + "do", "while", "for", "for_var", "for_in", + "for_in_var", "continue", "break", "return", + "with", "switch", "throw", "try", "label", + "case", "default", "debugger", +}; + +static void snode(int d, js_Ast *node) +{ + void (*afun)(int,js_Ast*) = snode; + void (*bfun)(int,js_Ast*) = snode; + void (*cfun)(int,js_Ast*) = snode; + void (*dfun)(int,js_Ast*) = snode; + + if (!node) { + return; + } + + if (node->type == AST_LIST) { + slist(d, node); + return; + } + + pc('('); + ps(stype[node->type]); + switch (node->type) { + case AST_IDENTIFIER: pc(' '); ps(node->string); break; + case AST_STRING: pc(' '); pstr(node->string); break; + case AST_REGEXP: printf(" /%s/", node->string); break; + case AST_NUMBER: printf(" %.9g", node->number); break; + case STM_BLOCK: afun = sblock; break; + case STM_FUNC: case EXP_FUNC: cfun = sblock; break; + case STM_SWITCH: bfun = sblock; break; + case STM_CASE: bfun = sblock; break; + case STM_DEFAULT: afun = sblock; break; + } + if (node->a) { pc(' '); afun(d, node->a); } + if (node->b) { pc(' '); bfun(d, node->b); } + if (node->c) { pc(' '); cfun(d, node->c); } + if (node->d) { pc(' '); dfun(d, node->d); } + pc(')'); +} + +static void slist(int d, js_Ast *list) +{ + pc('['); + while (list) { + assert(list->type == AST_LIST); + snode(d, list->a); + list = list->b; + if (list) + pc(' '); + } + pc(']'); +} + +static void sblock(int d, js_Ast *list) +{ + ps("[\n"); + in(d+1); + while (list) { + assert(list->type == AST_LIST); + snode(d+1, list->a); + list = list->b; + if (list) { + pc('\n'); + in(d+1); + } + } + pc('\n'); in(d); pc(']'); +} + +void jsP_dumplist(js_State *J, js_Ast *prog) +{ + if (prog->type == AST_LIST) + sblock(0, prog); + else + snode(0, prog); + pc('\n'); +} diff --git a/jsload.c b/jsload.c index 35b49d9..d5153f9 100644 --- a/jsload.c +++ b/jsload.c @@ -5,8 +5,9 @@ static int jsP_loadstring(js_State *J, const char *filename, const char *source) { js_Ast *prog = jsP_parse(J, filename, source); if (prog) { - jsP_foldconstants(J, prog); - jsP_pretty(J, prog); + jsP_optimize(J, prog); + jsP_dumpsyntax(J, prog); + jsP_dumplist(J, prog); jsP_freeparse(J); return 0; } diff --git a/jsoptim.c b/jsoptim.c index f2edd93..9587907 100644 --- a/jsoptim.c +++ b/jsoptim.c @@ -79,7 +79,7 @@ static int foldnumber(js_Ast *node, double *r) return 0; } -void jsP_foldconstants(js_State *J, js_Ast *prog) +void jsP_optimize(js_State *J, js_Ast *prog) { double x; foldnumber(prog, &x); diff --git a/jsparse.h b/jsparse.h index fca1d1f..0391beb 100644 --- a/jsparse.h +++ b/jsparse.h @@ -1,8 +1,6 @@ #ifndef js_parse_h #define js_parse_h -typedef struct js_Ast js_Ast; - struct js_Ast { int type; @@ -124,7 +122,8 @@ enum js_Ast *jsP_parse(js_State *J, const char *filename, const char *source); void jsP_freeparse(js_State *J); -void jsP_foldconstants(js_State *J, js_Ast *prog); -void jsP_pretty(js_State *J, js_Ast *prog); +void jsP_optimize(js_State *J, js_Ast *prog); +void jsP_dumpsyntax(js_State *J, js_Ast *prog); +void jsP_dumplist(js_State *J, js_Ast *prog); #endif diff --git a/jspretty.c b/jspretty.c deleted file mode 100644 index 75eb461..0000000 --- a/jspretty.c +++ /dev/null @@ -1,518 +0,0 @@ -#include "js.h" -#include "jsparse.h" - -static void printast(js_Ast *n, int level); - -static const char *strast(int type) -{ - switch (type) { - case AST_LIST: return "LIST"; - case AST_IDENTIFIER: return "IDENTIFIER"; - case AST_NUMBER: return "NUMBER"; - case AST_STRING: return "STRING"; - case AST_REGEXP: return "REGEXP"; - case EXP_VAR: return "VAR"; - case EXP_NULL: return "NULL"; - case EXP_TRUE: return "TRUE"; - case EXP_FALSE: return "FALSE"; - case EXP_THIS: return "THIS"; - case EXP_ARRAY: return "ARRAY"; - case EXP_OBJECT: return "OBJECT"; - case EXP_PROP_VAL: return "PROP_VAL"; - case EXP_PROP_GET: return "PROP_GET"; - case EXP_PROP_SET: return "PROP_SET"; - case EXP_INDEX: return "INDEX"; - case EXP_MEMBER: return "MEMBER"; - case EXP_NEW: return "new"; - case EXP_CALL: return "CALL"; - case EXP_FUNC: return "function"; - case EXP_COND: return "?:"; - case EXP_COMMA: return ","; - case EXP_DELETE: return "delete"; - case EXP_VOID: return "void"; - case EXP_TYPEOF: return "typeof"; - case EXP_PREINC: return "PRE++"; - case EXP_PREDEC: return "PRE--"; - case EXP_POSTINC: return "POST++"; - case EXP_POSTDEC: return "POST--"; - case EXP_POS: return "+"; - case EXP_NEG: return "-"; - case EXP_BITNOT: return "~"; - case EXP_LOGNOT: return "!"; - case EXP_LOGOR: return "||"; - case EXP_LOGAND: return "&&"; - case EXP_BITOR: return "|"; - case EXP_BITXOR: return "^"; - case EXP_BITAND: return "&"; - case EXP_EQ: return "=="; - case EXP_NE: return "!="; - case EXP_EQ3: return "==="; - case EXP_NE3: return "!=="; - case EXP_LT: return "<"; - case EXP_GT: return ">"; - case EXP_LE: return "<="; - case EXP_GE: return ">="; - case EXP_INSTANCEOF: return "instanceof"; - case EXP_IN: return "in"; - case EXP_SHL: return "<<"; - case EXP_SHR: return ">>"; - case EXP_USHR: return ">>>"; - case EXP_ADD: return "+"; - case EXP_SUB: return "-"; - case EXP_MUL: return "*"; - case EXP_DIV: return "/"; - case EXP_MOD: return "%"; - case EXP_ASS: return "="; - case EXP_ASS_MUL: return "*="; - case EXP_ASS_DIV: return "/="; - case EXP_ASS_MOD: return "%="; - case EXP_ASS_ADD: return "+="; - case EXP_ASS_SUB: return "-="; - case EXP_ASS_SHL: return "<<="; - case EXP_ASS_SHR: return ">>="; - case EXP_ASS_USHR: return ">>>="; - case EXP_ASS_BITAND: return "&="; - case EXP_ASS_BITXOR: return "^="; - case EXP_ASS_BITOR: return "|="; - case STM_BLOCK: return "BLOCK"; - case STM_FUNC: return "function-decl"; - case STM_NOP: return "NOP"; - case STM_VAR: return "var"; - case STM_IF: return "if"; - case STM_DO: return "do-while"; - case STM_WHILE: return "while"; - case STM_FOR: return "for"; - case STM_FOR_VAR: return "for_var"; - case STM_FOR_IN: return "for_in"; - case STM_FOR_IN_VAR: return "for_in_var"; - case STM_CONTINUE: return "continue"; - case STM_BREAK: return "break"; - case STM_RETURN: return "return"; - case STM_WITH: return "with"; - case STM_SWITCH: return "switch"; - case STM_THROW: return "throw"; - case STM_TRY: return "try"; - case STM_LABEL: return "label"; - case STM_CASE: return "case"; - case STM_DEFAULT: return "default"; - case STM_DEBUGGER: return "debugger"; - default: return "(unknown)"; - } -} - -static void indent(int level) -{ - while (level--) - putchar('\t'); -} - -static void printlist(js_Ast *n, int level, const char *sep) -{ - while (n) { - printast(n->a, level); - n = n->b; - if (n) - fputs(sep, stdout); - } -} - -static void printblock(js_Ast *n, int level) -{ - while (n) { - indent(level); - printast(n->a, level); - if (n->a->type < STM_BLOCK) // expression - putchar(';'); - n = n->b; - if (n) - putchar('\n'); - } -} - -static void printstm(js_Ast *n, int level) -{ - if (n->type == STM_BLOCK) { - printf(" {\n"); - printblock(n->a, level + 1); - putchar('\n'); - indent(level); - printf("}"); - } else { - putchar('\n'); - indent(level + 1); - printast(n, level + 1); - if (n->type < STM_BLOCK) // expression - putchar(';'); - } -} - -static void printunary(int level, js_Ast *n, const char *pre, const char *suf) -{ - printf(pre); - printast(n, level); - printf(suf); -} - -static void printbinary(int level, js_Ast *a, js_Ast *b, const char *op) -{ - printf("("); - printast(a, level); - printf(" %s ", op); - printast(b, level); - printf(")"); -} - -static void printast(js_Ast *n, int level) -{ - switch (n->type) { - case AST_IDENTIFIER: printf("%s", n->string); return; - case AST_NUMBER: printf("%g", n->number); return; - case AST_STRING: printf("'%s'", n->string); return; - case AST_REGEXP: printf("/%s/", n->string); return; - case AST_LIST: - putchar('['); - printlist(n, level, " "); - putchar(']'); - break; - - case STM_BLOCK: - putchar('{'); - putchar('\n'); - printblock(n->a, level + 1); - putchar('\n'); - indent(level); - putchar('}'); - break; - - case STM_FOR: - printf("for ("); - printast(n->a, level); printf("; "); - printast(n->b, level); printf("; "); - printast(n->c, level); printf(")"); - printstm(n->d, level); - break; - case STM_FOR_VAR: - printf("for (var "); - printlist(n->a, level, ", "); printf("; "); - printast(n->b, level); printf("; "); - printast(n->c, level); printf(")"); - printstm(n->d, level); - break; - case STM_FOR_IN: - printf("for ("); - printast(n->a, level); printf(" in "); - printast(n->b, level); printf(")"); - printstm(n->c, level); - break; - case STM_FOR_IN_VAR: - printf("for (var "); - printlist(n->a, level, ", "); printf(" in "); - printast(n->b, level); printf(")"); - printstm(n->c, level); - break; - - case STM_NOP: - putchar(';'); - break; - - case STM_VAR: - printf("var "); - printlist(n->a, level, ", "); - putchar(';'); - break; - - case EXP_VAR: - printast(n->a, level); - if (n->b) { - printf(" = "); - printast(n->b, level); - } - break; - - case STM_IF: - printf("if ("); - printast(n->a, level); - printf(")"); - printstm(n->b, level); - if (n->c) { - putchar('\n'); - indent(level); - printf("else"); - printstm(n->c, level); - } - break; - - case STM_DO: - printf("do"); - printstm(n->a, level); - if (n->a->type == STM_BLOCK) { - putchar(' '); - } else { - putchar('\n'); - indent(level); - } - printf("while ("); - printast(n->b, level); - printf(");"); - break; - - case STM_WHILE: - printf("while ("); - printast(n->a, level); - printf(")"); - printstm(n->b, level); - break; - - case STM_CONTINUE: - if (n->a) { - printf("continue "); - printast(n->a, level); - printf(";"); - } else { - printf("continue;"); - } - break; - - case STM_BREAK: - if (n->a) { - printf("break "); - printast(n->a, level); - printf(";"); - } else { - printf("break;"); - } - break; - - case STM_RETURN: - if (n->a) { - printf("return "); - printast(n->a, level); - printf(";"); - } else { - printf("return;"); - } - break; - - case STM_THROW: - printf("throw "); - printast(n->a, level); - printf(";"); - break; - - case STM_SWITCH: - printf("switch ("); - printast(n->a, level); - printf(") {\n"); - printblock(n->b, level); - putchar('\n'); - indent(level); - printf("}"); - break; - - case STM_CASE: - printf("case "); - printast(n->a, level); - printf(":"); - if (n->b) { - printf("\n"); - printblock(n->b, level + 1); - } - break; - - case STM_DEFAULT: - printf("default:"); - if (n->a) { - printf("\n"); - printblock(n->a, level + 1); - } - break; - - case STM_LABEL: - printast(n->a, level); - printf(":"); - printstm(n->b, level - 1); - break; - - case STM_WITH: - printf("with ("); - printast(n->a, level); - printf(")"); - printstm(n->b, level); - break; - - case STM_TRY: - printf("try"); - printstm(n->a, level); - if (n->b && n->c) { - printf(" catch ("); - printast(n->b, level); - printf(")"); - printstm(n->c, level); - } - if (n->d) { - printf(" finally"); - printstm(n->d, level); - } - break; - - case STM_DEBUGGER: - printf("debugger"); - break; - - case STM_FUNC: - printf("function "); - printast(n->a, level); - printf("("); - printlist(n->b, level, ", "); - printf(")\n"); - indent(level); - printf("{\n"); - printblock(n->c, level + 1); - printf("\n"); - indent(level); - printf("}"); - break; - - case EXP_FUNC: - printf("(function "); - if (n->a) - printast(n->a, level); - printf("("); - printlist(n->b, level, ", "); - printf(") {\n"); - printblock(n->c, level + 1); - printf("\n"); - indent(level); - printf("})"); - break; - - case EXP_OBJECT: - printf("{ "); - printlist(n->a, level, ", "); - printf(" }"); - break; - - case EXP_PROP_VAL: - printast(n->a, level); - printf(": "); - printast(n->b, level); - break; - - case EXP_ARRAY: - printf("[ "); - printlist(n->a, level, ", "); - printf(" ]"); - break; - - case EXP_NEW: - printf("(new "); - printast(n->a, level); - printf("("); - printlist(n->b, level, ", "); - printf("))"); - break; - - case EXP_CALL: - printf("("); - printast(n->a, level); - printf("("); - printlist(n->b, level, ", "); - printf("))"); - break; - - case EXP_MEMBER: - printf("("); - printast(n->a, level); - printf("."); - printast(n->b, level); - printf(")"); - break; - - case EXP_INDEX: - printf("("); - printast(n->a, level); - printf("["); - printast(n->b, level); - printf("])"); - break; - - case EXP_COND: - printf("("); - printast(n->a, level); - printf(" ? "); - printast(n->b, level); - printf(" : "); - printast(n->c, level); - printf(")"); - break; - - case EXP_NULL: printf("null"); break; - case EXP_TRUE: printf("true"); break; - case EXP_FALSE: printf("false"); break; - case EXP_THIS: printf("this"); break; - - case EXP_DELETE: printunary(level, n->a, "(delete ", ")"); break; - case EXP_VOID: printunary(level, n->a, "(void ", ")"); break; - case EXP_TYPEOF: printunary(level, n->a, "(typeof ", ")"); break; - case EXP_PREINC: printunary(level, n->a, "(++", ")"); break; - case EXP_PREDEC: printunary(level, n->a, "(--", ")"); break; - case EXP_POSTINC: printunary(level, n->a, "(", "++)"); break; - case EXP_POSTDEC: printunary(level, n->a, "(", "--)"); break; - case EXP_POS: printunary(level, n->a, "(+", ")"); break; - case EXP_NEG: printunary(level, n->a, "(-", ")"); break; - case EXP_BITNOT: printunary(level, n->a, "(~", ")"); break; - case EXP_LOGNOT: printunary(level, n->a, "(!", ")"); break; - - case EXP_COMMA: printbinary(level, n->a, n->b, ","); break; - case EXP_LOGOR: printbinary(level, n->a, n->b, "||"); break; - case EXP_LOGAND: printbinary(level, n->a, n->b, "&&"); break; - case EXP_BITOR: printbinary(level, n->a, n->b, "|"); break; - case EXP_BITXOR: printbinary(level, n->a, n->b, "^"); break; - case EXP_BITAND: printbinary(level, n->a, n->b, "&"); break; - case EXP_EQ: printbinary(level, n->a, n->b, "=="); break; - case EXP_NE: printbinary(level, n->a, n->b, "!="); break; - case EXP_EQ3: printbinary(level, n->a, n->b, "==="); break; - case EXP_NE3: printbinary(level, n->a, n->b, "!=="); break; - case EXP_LT: printbinary(level, n->a, n->b, "<"); break; - case EXP_GT: printbinary(level, n->a, n->b, ">"); break; - case EXP_LE: printbinary(level, n->a, n->b, "<="); break; - case EXP_GE: printbinary(level, n->a, n->b, ">="); break; - case EXP_INSTANCEOF: printbinary(level, n->a, n->b, "instanceof"); break; - case EXP_IN: printbinary(level, n->a, n->b, "in"); break; - case EXP_SHL: printbinary(level, n->a, n->b, "<<"); break; - case EXP_SHR: printbinary(level, n->a, n->b, ">>"); break; - case EXP_USHR: printbinary(level, n->a, n->b, ">>>"); break; - case EXP_ADD: printbinary(level, n->a, n->b, "+"); break; - case EXP_SUB: printbinary(level, n->a, n->b, "-"); break; - case EXP_MUL: printbinary(level, n->a, n->b, "*"); break; - case EXP_DIV: printbinary(level, n->a, n->b, "/"); break; - case EXP_MOD: printbinary(level, n->a, n->b, "%"); break; - case EXP_ASS: printbinary(level, n->a, n->b, "="); break; - case EXP_ASS_MUL: printbinary(level, n->a, n->b, "*="); break; - case EXP_ASS_DIV: printbinary(level, n->a, n->b, "/="); break; - case EXP_ASS_MOD: printbinary(level, n->a, n->b, "%="); break; - case EXP_ASS_ADD: printbinary(level, n->a, n->b, "+="); break; - case EXP_ASS_SUB: printbinary(level, n->a, n->b, "-="); break; - case EXP_ASS_SHL: printbinary(level, n->a, n->b, "<<="); break; - case EXP_ASS_SHR: printbinary(level, n->a, n->b, ">>="); break; - case EXP_ASS_USHR: printbinary(level, n->a, n->b, ">>>="); break; - case EXP_ASS_BITAND: printbinary(level, n->a, n->b, "&="); break; - case EXP_ASS_BITXOR: printbinary(level, n->a, n->b, "^="); break; - case EXP_ASS_BITOR: printbinary(level, n->a, n->b, "|="); break; - - default: - printf("(%s", strast(n->type)); - if (n->a) { putchar(' '); printast(n->a, level); } - if (n->b) { putchar(' '); printast(n->b, level); } - if (n->c) { putchar(' '); printast(n->c, level); } - if (n->d) { putchar(' '); printast(n->d, level); } - putchar(')'); - break; - } -} - -void jsP_pretty(js_State *J, js_Ast *prog) -{ - printblock(prog, 0); - putchar('\n'); -} diff --git a/jsstate.h b/jsstate.h index 05c1fd6..d90769a 100644 --- a/jsstate.h +++ b/jsstate.h @@ -1,8 +1,6 @@ #ifndef js_state_h #define js_state_h -typedef struct js_Ast js_Ast; - struct js_State { jmp_buf jb; /* setjmp buffer for error handling in parser */