Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 1 | package main |
| 2 | |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 3 | import "fmt" |
| 4 | |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 5 | type AST interface { |
Fumitoshi Ukai | e34c179 | 2015-03-30 17:53:47 +0900 | [diff] [blame] | 6 | eval(*Evaluator) |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 7 | show() |
| 8 | } |
| 9 | |
| 10 | type ASTBase struct { |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 11 | filename string |
| 12 | lineno int |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 13 | } |
| 14 | |
| 15 | type AssignAST struct { |
| 16 | ASTBase |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 17 | lhs string |
| 18 | rhs string |
| 19 | op string |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 20 | } |
| 21 | |
Fumitoshi Ukai | e34c179 | 2015-03-30 17:53:47 +0900 | [diff] [blame] | 22 | func (ast *AssignAST) eval(ev *Evaluator) { |
| 23 | ev.evalAssign(ast) |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 24 | } |
| 25 | |
Shinichiro Hamaji | 69b7f65 | 2015-03-31 01:01:59 +0900 | [diff] [blame] | 26 | func (ast *AssignAST) evalRHS(ev *Evaluator, lhs string) string { |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 27 | switch ast.op { |
| 28 | case ":=": |
| 29 | return ev.evalExpr(ast.rhs) |
| 30 | case "=": |
| 31 | return ast.rhs |
Shinichiro Hamaji | 69b7f65 | 2015-03-31 01:01:59 +0900 | [diff] [blame] | 32 | case "+=": |
| 33 | prev, _ := ev.getVar(lhs) |
| 34 | return fmt.Sprintf("%s %s", prev, ev.evalExpr(ast.rhs)) |
| 35 | case "?=": |
| 36 | prev, present := ev.getVar(lhs) |
| 37 | if present { |
| 38 | return prev |
| 39 | } |
| 40 | return ev.evalExpr(ast.rhs) |
| 41 | default: |
| 42 | panic(fmt.Sprintf("unknown assign op: %q", ast.op)) |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 43 | } |
| 44 | } |
| 45 | |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 46 | func (ast *AssignAST) show() { |
| 47 | Log("%s=%s", ast.lhs, ast.rhs) |
| 48 | } |
| 49 | |
Shinichiro Hamaji | de82971 | 2015-03-31 18:26:56 +0900 | [diff] [blame^] | 50 | // Note we cannot be sure what this is, until all variables in |expr| |
| 51 | // are expanded. |
| 52 | type MaybeRuleAST struct { |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 53 | ASTBase |
Shinichiro Hamaji | de82971 | 2015-03-31 18:26:56 +0900 | [diff] [blame^] | 54 | expr string |
Shinichiro Hamaji | ae32b78 | 2015-03-31 14:41:19 +0900 | [diff] [blame] | 55 | cmds []string |
Shinichiro Hamaji | 7c4e325 | 2015-03-30 23:04:25 +0900 | [diff] [blame] | 56 | cmdLineno int |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 57 | } |
| 58 | |
Shinichiro Hamaji | de82971 | 2015-03-31 18:26:56 +0900 | [diff] [blame^] | 59 | func (ast *MaybeRuleAST) eval(ev *Evaluator) { |
| 60 | ev.evalMaybeRule(ast) |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 61 | } |
| 62 | |
Shinichiro Hamaji | de82971 | 2015-03-31 18:26:56 +0900 | [diff] [blame^] | 63 | func (ast *MaybeRuleAST) show() { |
| 64 | Log("%s", ast.expr) |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 65 | for _, cmd := range ast.cmds { |
| 66 | Log("\t%s", cmd) |
| 67 | } |
| 68 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 69 | |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 70 | type IncludeAST struct { |
| 71 | ASTBase |
| 72 | expr string |
| 73 | op string |
| 74 | } |
| 75 | |
| 76 | func (ast *IncludeAST) eval(ev *Evaluator) { |
| 77 | ev.evalInclude(ast) |
| 78 | } |
| 79 | |
| 80 | func (ast *IncludeAST) show() { |
| 81 | Log("include %s", ast.expr) |
| 82 | } |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 83 | |
| 84 | type IfAST struct { |
| 85 | ASTBase |
Shinichiro Hamaji | ae32b78 | 2015-03-31 14:41:19 +0900 | [diff] [blame] | 86 | op string |
| 87 | lhs string |
| 88 | rhs string // Empty if |op| is ifdef or ifndef. |
| 89 | trueStmts []AST |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 90 | falseStmts []AST |
| 91 | } |
| 92 | |
| 93 | func (ast *IfAST) eval(ev *Evaluator) { |
| 94 | ev.evalIf(ast) |
| 95 | } |
| 96 | |
| 97 | func (ast *IfAST) show() { |
| 98 | // TODO |
| 99 | Log("if") |
| 100 | } |