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