Shinichiro Hamaji | b69bf8a | 2015-06-10 14:52:06 +0900 | [diff] [blame] | 1 | // Copyright 2015 Google Inc. All rights reserved |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 15 | package kati |
Fumitoshi Ukai | f2f8456 | 2015-03-30 19:47:45 +0900 | [diff] [blame] | 16 | |
| 17 | import ( |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 18 | "bytes" |
Fumitoshi Ukai | f2f8456 | 2015-03-30 19:47:45 +0900 | [diff] [blame] | 19 | "fmt" |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 20 | "io" |
Shinichiro Hamaji | e7aafb0 | 2015-04-09 18:23:46 +0900 | [diff] [blame] | 21 | "os" |
Fumitoshi Ukai | f2f8456 | 2015-03-30 19:47:45 +0900 | [diff] [blame] | 22 | "os/exec" |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 23 | "path/filepath" |
Shinichiro Hamaji | 13b30d9 | 2015-04-03 14:42:21 +0900 | [diff] [blame] | 24 | "sort" |
Shinichiro Hamaji | e708a9d | 2015-04-03 14:34:35 +0900 | [diff] [blame] | 25 | "strconv" |
Fumitoshi Ukai | f2f8456 | 2015-03-30 19:47:45 +0900 | [diff] [blame] | 26 | "strings" |
Fumitoshi Ukai | 586b02a | 2015-05-08 00:23:10 +0900 | [diff] [blame] | 27 | "time" |
Fumitoshi Ukai | f2f8456 | 2015-03-30 19:47:45 +0900 | [diff] [blame] | 28 | ) |
| 29 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 30 | // mkFunc is a make function. |
Fumitoshi Ukai | f2f8456 | 2015-03-30 19:47:45 +0900 | [diff] [blame] | 31 | // http://www.gnu.org/software/make/manual/make.html#Functions |
Fumitoshi Ukai | e520f26 | 2015-03-31 17:27:03 +0900 | [diff] [blame] | 32 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 33 | // mkFunc is make builtin function. |
| 34 | type mkFunc interface { |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 35 | // Arity is max function's arity. |
| 36 | // ',' will not be handled as argument separator more than arity. |
| 37 | // 0 means varargs. |
| 38 | Arity() int |
| 39 | |
| 40 | // AddArg adds value as an argument. |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 41 | // the first argument will be "(funcname", or "{funcname". |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 42 | AddArg(Value) |
| 43 | |
| 44 | Value |
| 45 | } |
| 46 | |
| 47 | var ( |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 48 | funcMap = map[string]func() mkFunc{ |
| 49 | "patsubst": func() mkFunc { return &funcPatsubst{} }, |
| 50 | "strip": func() mkFunc { return &funcStrip{} }, |
| 51 | "subst": func() mkFunc { return &funcSubst{} }, |
| 52 | "findstring": func() mkFunc { return &funcFindstring{} }, |
| 53 | "filter": func() mkFunc { return &funcFilter{} }, |
| 54 | "filter-out": func() mkFunc { return &funcFilterOut{} }, |
| 55 | "sort": func() mkFunc { return &funcSort{} }, |
| 56 | "word": func() mkFunc { return &funcWord{} }, |
| 57 | "wordlist": func() mkFunc { return &funcWordlist{} }, |
| 58 | "words": func() mkFunc { return &funcWords{} }, |
| 59 | "firstword": func() mkFunc { return &funcFirstword{} }, |
| 60 | "lastword": func() mkFunc { return &funcLastword{} }, |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 61 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 62 | "join": func() mkFunc { return &funcJoin{} }, |
| 63 | "wildcard": func() mkFunc { return &funcWildcard{} }, |
| 64 | "dir": func() mkFunc { return &funcDir{} }, |
| 65 | "notdir": func() mkFunc { return &funcNotdir{} }, |
| 66 | "suffix": func() mkFunc { return &funcSuffix{} }, |
| 67 | "basename": func() mkFunc { return &funcBasename{} }, |
| 68 | "addsuffix": func() mkFunc { return &funcAddsuffix{} }, |
| 69 | "addprefix": func() mkFunc { return &funcAddprefix{} }, |
| 70 | "realpath": func() mkFunc { return &funcRealpath{} }, |
| 71 | "abspath": func() mkFunc { return &funcAbspath{} }, |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 72 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 73 | "if": func() mkFunc { return &funcIf{} }, |
| 74 | "and": func() mkFunc { return &funcAnd{} }, |
| 75 | "or": func() mkFunc { return &funcOr{} }, |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 76 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 77 | "value": func() mkFunc { return &funcValue{} }, |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 78 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 79 | "eval": func() mkFunc { return &funcEval{} }, |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 80 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 81 | "shell": func() mkFunc { return &funcShell{} }, |
| 82 | "call": func() mkFunc { return &funcCall{} }, |
| 83 | "foreach": func() mkFunc { return &funcForeach{} }, |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 84 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 85 | "origin": func() mkFunc { return &funcOrigin{} }, |
| 86 | "flavor": func() mkFunc { return &funcFlavor{} }, |
| 87 | "info": func() mkFunc { return &funcInfo{} }, |
| 88 | "warning": func() mkFunc { return &funcWarning{} }, |
| 89 | "error": func() mkFunc { return &funcError{} }, |
Fumitoshi Ukai | 0b5e18b | 2015-04-03 22:57:17 +0900 | [diff] [blame] | 90 | } |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 91 | ) |
| 92 | |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 93 | func assertArity(name string, req, n int) { |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 94 | if n-1 < req { |
| 95 | panic(fmt.Sprintf("*** insufficient number of arguments (%d) to function `%s'.", n-1, name)) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 96 | } |
| 97 | } |
| 98 | |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 99 | func numericValueForFunc(v string) (int, bool) { |
| 100 | n, err := strconv.Atoi(v) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 101 | if err != nil || n < 0 { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 102 | return n, false |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 103 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 104 | return n, true |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 105 | } |
| 106 | |
Shinichiro Hamaji | ba6b84d | 2015-06-03 18:46:13 +0900 | [diff] [blame] | 107 | func formatCommandOutput(out []byte) []byte { |
| 108 | out = bytes.TrimRight(out, "\n") |
| 109 | out = bytes.Replace(out, []byte{'\n'}, []byte{' '}, -1) |
| 110 | return out |
| 111 | } |
| 112 | |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 113 | type fclosure struct { |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 114 | // args[0] is "(funcname", or "{funcname". |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 115 | args []Value |
| 116 | } |
| 117 | |
| 118 | func (c *fclosure) AddArg(v Value) { |
| 119 | c.args = append(c.args, v) |
Fumitoshi Ukai | 0b5e18b | 2015-04-03 22:57:17 +0900 | [diff] [blame] | 120 | } |
| 121 | |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 122 | func (c *fclosure) String() string { |
| 123 | if len(c.args) == 0 { |
| 124 | panic("no args in func") |
| 125 | } |
| 126 | arg0 := c.args[0].String() |
| 127 | if arg0 == "" { |
| 128 | panic(fmt.Errorf("wrong format of arg0: %q", arg0)) |
| 129 | } |
| 130 | cp := closeParen(arg0[0]) |
| 131 | if cp == 0 { |
| 132 | panic(fmt.Errorf("wrong format of arg0: %q", arg0)) |
| 133 | } |
| 134 | var args []string |
| 135 | for _, arg := range c.args[1:] { |
| 136 | args = append(args, arg.String()) |
| 137 | } |
| 138 | return fmt.Sprintf("$%s %s%c", arg0, strings.Join(args, ","), cp) |
| 139 | } |
Fumitoshi Ukai | 7878113 | 2015-04-10 17:08:40 +0900 | [diff] [blame] | 140 | |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 141 | func (c *fclosure) serialize() serializableVar { |
| 142 | r := serializableVar{Type: "func"} |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 143 | for _, a := range c.args { |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 144 | r.Children = append(r.Children, a.serialize()) |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 145 | } |
| 146 | return r |
| 147 | } |
| 148 | |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 149 | func (c *fclosure) dump(w io.Writer) { |
Fumitoshi Ukai | a78fce0 | 2015-06-25 12:29:09 +0900 | [diff] [blame] | 150 | dumpByte(w, valueTypeFunc) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 151 | for _, a := range c.args { |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 152 | a.dump(w) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 153 | } |
| 154 | } |
| 155 | |
Shinichiro Hamaji | 4125cf4 | 2015-04-03 11:42:28 +0900 | [diff] [blame] | 156 | // http://www.gnu.org/software/make/manual/make.html#Text-Functions |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 157 | type funcSubst struct{ fclosure } |
| 158 | |
| 159 | func (f *funcSubst) Arity() int { return 3 } |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 160 | func (f *funcSubst) Eval(w io.Writer, ev *Evaluator) { |
| 161 | assertArity("subst", 3, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 162 | abuf := newBuf() |
| 163 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 164 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 165 | from := fargs[0] |
| 166 | to := fargs[1] |
| 167 | text := fargs[2] |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 168 | logf("subst from:%q to:%q text:%q", from, to, text) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 169 | w.Write(bytes.Replace(text, from, to, -1)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 170 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 171 | stats.add("funcbody", "subst", t) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 172 | } |
| 173 | |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 174 | type funcPatsubst struct{ fclosure } |
| 175 | |
| 176 | func (f *funcPatsubst) Arity() int { return 3 } |
| 177 | func (f *funcPatsubst) Eval(w io.Writer, ev *Evaluator) { |
| 178 | assertArity("patsubst", 3, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 179 | abuf := newBuf() |
| 180 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 181 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 182 | pat := fargs[0] |
| 183 | repl := fargs[1] |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 184 | ws := newWordScanner(fargs[2]) |
Fumitoshi Ukai | 77411fb | 2015-06-19 15:46:21 +0900 | [diff] [blame] | 185 | space := false |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 186 | for ws.Scan() { |
Fumitoshi Ukai | 77411fb | 2015-06-19 15:46:21 +0900 | [diff] [blame] | 187 | if space { |
| 188 | writeByte(w, ' ') |
| 189 | } |
| 190 | pre, subst, post := substPatternBytes(pat, repl, ws.Bytes()) |
| 191 | w.Write(pre) |
| 192 | if subst != nil { |
| 193 | w.Write(subst) |
| 194 | w.Write(post) |
| 195 | } |
| 196 | space = true |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 197 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 198 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 199 | stats.add("funcbody", "patsubst", t) |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 200 | } |
| 201 | |
| 202 | type funcStrip struct{ fclosure } |
| 203 | |
| 204 | func (f *funcStrip) Arity() int { return 1 } |
| 205 | func (f *funcStrip) Eval(w io.Writer, ev *Evaluator) { |
| 206 | assertArity("strip", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 207 | abuf := newBuf() |
| 208 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 209 | t := time.Now() |
Shinichiro Hamaji | b978e11 | 2015-06-05 11:02:45 +0900 | [diff] [blame] | 210 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | a44b766 | 2015-06-19 11:07:07 +0900 | [diff] [blame] | 211 | space := false |
Shinichiro Hamaji | b978e11 | 2015-06-05 11:02:45 +0900 | [diff] [blame] | 212 | for ws.Scan() { |
Fumitoshi Ukai | a44b766 | 2015-06-19 11:07:07 +0900 | [diff] [blame] | 213 | if space { |
| 214 | writeByte(w, ' ') |
| 215 | } |
| 216 | w.Write(ws.Bytes()) |
| 217 | space = true |
Shinichiro Hamaji | b978e11 | 2015-06-05 11:02:45 +0900 | [diff] [blame] | 218 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 219 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 220 | stats.add("funcbody", "strip", t) |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 221 | } |
| 222 | |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 223 | type funcFindstring struct{ fclosure } |
| 224 | |
| 225 | func (f *funcFindstring) Arity() int { return 2 } |
| 226 | func (f *funcFindstring) Eval(w io.Writer, ev *Evaluator) { |
| 227 | assertArity("findstring", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 228 | abuf := newBuf() |
| 229 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 230 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 231 | find := fargs[0] |
| 232 | text := fargs[1] |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 233 | if bytes.Index(text, find) >= 0 { |
| 234 | w.Write(find) |
| 235 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 236 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 237 | stats.add("funcbody", "findstring", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 238 | } |
| 239 | |
| 240 | type funcFilter struct{ fclosure } |
| 241 | |
| 242 | func (f *funcFilter) Arity() int { return 2 } |
| 243 | func (f *funcFilter) Eval(w io.Writer, ev *Evaluator) { |
| 244 | assertArity("filter", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 245 | abuf := newBuf() |
| 246 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 247 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 248 | var patterns [][]byte |
| 249 | ws := newWordScanner(fargs[0]) |
| 250 | for ws.Scan() { |
| 251 | patterns = append(patterns, ws.Bytes()) |
| 252 | } |
| 253 | ws = newWordScanner(fargs[1]) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 254 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 255 | for ws.Scan() { |
| 256 | text := ws.Bytes() |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 257 | for _, pat := range patterns { |
| 258 | if matchPatternBytes(pat, text) { |
| 259 | sw.Write(text) |
| 260 | } |
| 261 | } |
| 262 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 263 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 264 | stats.add("funcbody", "filter", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 265 | } |
| 266 | |
| 267 | type funcFilterOut struct{ fclosure } |
| 268 | |
| 269 | func (f *funcFilterOut) Arity() int { return 2 } |
| 270 | func (f *funcFilterOut) Eval(w io.Writer, ev *Evaluator) { |
| 271 | assertArity("filter-out", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 272 | abuf := newBuf() |
| 273 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 274 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 275 | var patterns [][]byte |
| 276 | ws := newWordScanner(fargs[0]) |
| 277 | for ws.Scan() { |
| 278 | patterns = append(patterns, ws.Bytes()) |
| 279 | } |
| 280 | ws = newWordScanner(fargs[1]) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 281 | sw := ssvWriter{w: w} |
| 282 | Loop: |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 283 | for ws.Scan() { |
| 284 | text := ws.Bytes() |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 285 | for _, pat := range patterns { |
| 286 | if matchPatternBytes(pat, text) { |
| 287 | continue Loop |
| 288 | } |
| 289 | } |
| 290 | sw.Write(text) |
| 291 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 292 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 293 | stats.add("funcbody", "filter-out", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 294 | } |
| 295 | |
| 296 | type funcSort struct{ fclosure } |
| 297 | |
| 298 | func (f *funcSort) Arity() int { return 1 } |
| 299 | func (f *funcSort) Eval(w io.Writer, ev *Evaluator) { |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 300 | assertArity("sort", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 301 | abuf := newBuf() |
| 302 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 303 | t := time.Now() |
Fumitoshi Ukai | 79cfbab | 2015-06-18 23:28:26 +0900 | [diff] [blame] | 304 | ws := newWordScanner(abuf.Bytes()) |
| 305 | var toks []string |
| 306 | for ws.Scan() { |
| 307 | toks = append(toks, string(ws.Bytes())) |
| 308 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 309 | freeBuf(abuf) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 310 | sort.Strings(toks) |
| 311 | |
| 312 | // Remove duplicate words. |
| 313 | var prev string |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 314 | for _, tok := range toks { |
Fumitoshi Ukai | ce14acb | 2015-06-19 13:54:53 +0900 | [diff] [blame] | 315 | if prev == tok { |
| 316 | continue |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 317 | } |
Fumitoshi Ukai | ce14acb | 2015-06-19 13:54:53 +0900 | [diff] [blame] | 318 | if prev != "" { |
| 319 | writeByte(w, ' ') |
| 320 | } |
| 321 | io.WriteString(w, tok) |
| 322 | prev = tok |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 323 | } |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 324 | stats.add("funcbody", "sort", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 325 | } |
| 326 | |
| 327 | type funcWord struct{ fclosure } |
| 328 | |
| 329 | func (f *funcWord) Arity() int { return 2 } |
| 330 | func (f *funcWord) Eval(w io.Writer, ev *Evaluator) { |
| 331 | assertArity("word", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 332 | abuf := newBuf() |
| 333 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 334 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 335 | v := string(trimSpaceBytes(fargs[0])) |
| 336 | index, ok := numericValueForFunc(v) |
| 337 | if !ok { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 338 | errorExit(ev.filename, ev.lineno, `*** non-numeric first argument to "word" function: %q.`, v) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 339 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 340 | if index == 0 { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 341 | errorExit(ev.filename, ev.lineno, `*** first argument to "word" function must be greater than 0.`) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 342 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 343 | ws := newWordScanner(fargs[1]) |
| 344 | for ws.Scan() { |
| 345 | index-- |
| 346 | if index == 0 { |
| 347 | w.Write(ws.Bytes()) |
| 348 | break |
| 349 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 350 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 351 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 352 | stats.add("funcbody", "word", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 353 | } |
| 354 | |
| 355 | type funcWordlist struct{ fclosure } |
| 356 | |
| 357 | func (f *funcWordlist) Arity() int { return 3 } |
| 358 | func (f *funcWordlist) Eval(w io.Writer, ev *Evaluator) { |
| 359 | assertArity("wordlist", 3, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 360 | abuf := newBuf() |
| 361 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 362 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 363 | v := string(trimSpaceBytes(fargs[0])) |
| 364 | si, ok := numericValueForFunc(v) |
| 365 | if !ok { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 366 | errorExit(ev.filename, ev.lineno, `*** non-numeric first argument to "wordlist" function: %q.`, v) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 367 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 368 | if si == 0 { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 369 | errorExit(ev.filename, ev.lineno, `*** invalid first argument to "wordlist" function: %s`, f.args[1]) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 370 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 371 | v = string(trimSpaceBytes(fargs[1])) |
| 372 | ei, ok := numericValueForFunc(v) |
| 373 | if !ok { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 374 | errorExit(ev.filename, ev.lineno, `*** non-numeric second argument to "wordlist" function: %q.`, v) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 375 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 376 | |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 377 | ws := newWordScanner(fargs[2]) |
| 378 | i := 0 |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 379 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 380 | for ws.Scan() { |
| 381 | i++ |
| 382 | if si <= i && i <= ei { |
| 383 | sw.Write(ws.Bytes()) |
| 384 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 385 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 386 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 387 | stats.add("funcbody", "wordlist", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 388 | } |
| 389 | |
| 390 | type funcWords struct{ fclosure } |
| 391 | |
| 392 | func (f *funcWords) Arity() int { return 1 } |
| 393 | func (f *funcWords) Eval(w io.Writer, ev *Evaluator) { |
| 394 | assertArity("words", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 395 | abuf := newBuf() |
| 396 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 397 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 398 | ws := newWordScanner(abuf.Bytes()) |
| 399 | n := 0 |
| 400 | for ws.Scan() { |
| 401 | n++ |
| 402 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 403 | freeBuf(abuf) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 404 | io.WriteString(w, strconv.Itoa(n)) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 405 | stats.add("funcbody", "words", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 406 | } |
| 407 | |
| 408 | type funcFirstword struct{ fclosure } |
| 409 | |
| 410 | func (f *funcFirstword) Arity() int { return 1 } |
| 411 | func (f *funcFirstword) Eval(w io.Writer, ev *Evaluator) { |
| 412 | assertArity("firstword", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 413 | abuf := newBuf() |
| 414 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 415 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 416 | ws := newWordScanner(abuf.Bytes()) |
| 417 | if ws.Scan() { |
| 418 | w.Write(ws.Bytes()) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 419 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 420 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 421 | stats.add("funcbody", "firstword", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 422 | } |
| 423 | |
| 424 | type funcLastword struct{ fclosure } |
| 425 | |
| 426 | func (f *funcLastword) Arity() int { return 1 } |
| 427 | func (f *funcLastword) Eval(w io.Writer, ev *Evaluator) { |
| 428 | assertArity("lastword", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 429 | abuf := newBuf() |
| 430 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 431 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 432 | ws := newWordScanner(abuf.Bytes()) |
| 433 | var lw []byte |
| 434 | for ws.Scan() { |
| 435 | lw = ws.Bytes() |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 436 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 437 | if lw != nil { |
| 438 | w.Write(lw) |
| 439 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 440 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 441 | stats.add("funcbody", "lastword", t) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 442 | } |
| 443 | |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 444 | // https://www.gnu.org/software/make/manual/html_node/File-Name-Functions.html#File-Name-Functions |
| 445 | |
| 446 | type funcJoin struct{ fclosure } |
| 447 | |
| 448 | func (f *funcJoin) Arity() int { return 2 } |
| 449 | func (f *funcJoin) Eval(w io.Writer, ev *Evaluator) { |
| 450 | assertArity("join", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 451 | abuf := newBuf() |
| 452 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 453 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 454 | ws1 := newWordScanner(fargs[0]) |
| 455 | ws2 := newWordScanner(fargs[1]) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 456 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 457 | for { |
| 458 | if w1, w2 := ws1.Scan(), ws2.Scan(); !w1 && !w2 { |
| 459 | break |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 460 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 461 | sw.Write(ws1.Bytes()) |
| 462 | // Use |w| not to append extra ' '. |
| 463 | w.Write(ws2.Bytes()) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 464 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 465 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 466 | stats.add("funcbody", "join", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 467 | } |
| 468 | |
| 469 | type funcWildcard struct{ fclosure } |
| 470 | |
| 471 | func (f *funcWildcard) Arity() int { return 1 } |
| 472 | func (f *funcWildcard) Eval(w io.Writer, ev *Evaluator) { |
| 473 | assertArity("wildcard", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 474 | abuf := newBuf() |
| 475 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 476 | te := traceEvent.begin("wildcard", tmpval(abuf.Bytes()), traceEventMain) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 477 | if ev.avoidIO && !UseWildcardCache { |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 478 | ev.hasIO = true |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 479 | io.WriteString(w, "$(/bin/ls -d ") |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 480 | w.Write(abuf.Bytes()) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 481 | io.WriteString(w, " 2> /dev/null)") |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 482 | traceEvent.end(te) |
Fumitoshi Ukai | 358c68a | 2015-06-08 13:12:55 +0900 | [diff] [blame] | 483 | freeBuf(abuf) |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 484 | return |
| 485 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 486 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 487 | ws := newWordScanner(abuf.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 488 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 489 | for ws.Scan() { |
Shinichiro Hamaji | ba6b84d | 2015-06-03 18:46:13 +0900 | [diff] [blame] | 490 | pat := string(ws.Bytes()) |
Fumitoshi Ukai | 358c68a | 2015-06-08 13:12:55 +0900 | [diff] [blame] | 491 | wildcard(&sw, pat) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 492 | } |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 493 | traceEvent.end(te) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 494 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 495 | stats.add("funcbody", "wildcard", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 496 | } |
| 497 | |
| 498 | type funcDir struct{ fclosure } |
| 499 | |
| 500 | func (f *funcDir) Arity() int { return 1 } |
| 501 | func (f *funcDir) Eval(w io.Writer, ev *Evaluator) { |
| 502 | assertArity("dir", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 503 | abuf := newBuf() |
| 504 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 505 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 506 | ws := newWordScanner(abuf.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 507 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 508 | for ws.Scan() { |
| 509 | name := string(ws.Bytes()) |
Shinichiro Hamaji | 37626b6 | 2015-04-21 16:59:23 +0900 | [diff] [blame] | 510 | if name == "/" { |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 511 | sw.WriteString(name) |
| 512 | continue |
Shinichiro Hamaji | 37626b6 | 2015-04-21 16:59:23 +0900 | [diff] [blame] | 513 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 514 | sw.WriteString(filepath.Dir(string(name)) + string(filepath.Separator)) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 515 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 516 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 517 | stats.add("funcbody", "dir", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 518 | } |
| 519 | |
| 520 | type funcNotdir struct{ fclosure } |
| 521 | |
| 522 | func (f *funcNotdir) Arity() int { return 1 } |
| 523 | func (f *funcNotdir) Eval(w io.Writer, ev *Evaluator) { |
| 524 | assertArity("notdir", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 525 | abuf := newBuf() |
| 526 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 527 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 528 | ws := newWordScanner(abuf.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 529 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 530 | for ws.Scan() { |
| 531 | name := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 532 | if name == string(filepath.Separator) { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 533 | sw.Write([]byte{}) // separator |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 534 | continue |
| 535 | } |
Shinichiro Hamaji | a135891 | 2015-04-11 12:16:21 +0900 | [diff] [blame] | 536 | sw.WriteString(filepath.Base(name)) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 537 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 538 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 539 | stats.add("funcbody", "notdir", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 540 | } |
| 541 | |
| 542 | type funcSuffix struct{ fclosure } |
| 543 | |
| 544 | func (f *funcSuffix) Arity() int { return 1 } |
| 545 | func (f *funcSuffix) Eval(w io.Writer, ev *Evaluator) { |
| 546 | assertArity("suffix", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 547 | abuf := newBuf() |
| 548 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 549 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 550 | ws := newWordScanner(abuf.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 551 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 552 | for ws.Scan() { |
| 553 | tok := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 554 | e := filepath.Ext(tok) |
| 555 | if len(e) > 0 { |
Shinichiro Hamaji | a135891 | 2015-04-11 12:16:21 +0900 | [diff] [blame] | 556 | sw.WriteString(e) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 557 | } |
| 558 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 559 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 560 | stats.add("funcbody", "suffix", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 561 | } |
| 562 | |
| 563 | type funcBasename struct{ fclosure } |
| 564 | |
| 565 | func (f *funcBasename) Arity() int { return 1 } |
| 566 | func (f *funcBasename) Eval(w io.Writer, ev *Evaluator) { |
| 567 | assertArity("basename", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 568 | abuf := newBuf() |
| 569 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 570 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 571 | ws := newWordScanner(abuf.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 572 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 573 | for ws.Scan() { |
| 574 | tok := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 575 | e := stripExt(tok) |
Shinichiro Hamaji | a135891 | 2015-04-11 12:16:21 +0900 | [diff] [blame] | 576 | sw.WriteString(e) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 577 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 578 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 579 | stats.add("funcbody", "basename", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 580 | } |
| 581 | |
| 582 | type funcAddsuffix struct{ fclosure } |
| 583 | |
| 584 | func (f *funcAddsuffix) Arity() int { return 2 } |
| 585 | func (f *funcAddsuffix) Eval(w io.Writer, ev *Evaluator) { |
| 586 | assertArity("addsuffix", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 587 | abuf := newBuf() |
| 588 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 589 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 590 | suf := fargs[0] |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 591 | ws := newWordScanner(fargs[1]) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 592 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 593 | for ws.Scan() { |
| 594 | sw.Write(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 595 | // Use |w| not to append extra ' '. |
| 596 | w.Write(suf) |
| 597 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 598 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 599 | stats.add("funcbody", "addsuffix", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 600 | } |
| 601 | |
| 602 | type funcAddprefix struct{ fclosure } |
| 603 | |
| 604 | func (f *funcAddprefix) Arity() int { return 2 } |
| 605 | func (f *funcAddprefix) Eval(w io.Writer, ev *Evaluator) { |
| 606 | assertArity("addprefix", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 607 | abuf := newBuf() |
| 608 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 609 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 610 | pre := fargs[0] |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 611 | ws := newWordScanner(fargs[1]) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 612 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 613 | for ws.Scan() { |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 614 | sw.Write(pre) |
| 615 | // Use |w| not to append extra ' '. |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 616 | w.Write(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 617 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 618 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 619 | stats.add("funcbody", "addprefix", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 620 | } |
| 621 | |
| 622 | type funcRealpath struct{ fclosure } |
| 623 | |
| 624 | func (f *funcRealpath) Arity() int { return 1 } |
| 625 | func (f *funcRealpath) Eval(w io.Writer, ev *Evaluator) { |
| 626 | assertArity("realpath", 1, len(f.args)) |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 627 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 628 | io.WriteString(w, "KATI_TODO(realpath)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 629 | ev.hasIO = true |
| 630 | return |
| 631 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 632 | abuf := newBuf() |
| 633 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 634 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 635 | ws := newWordScanner(abuf.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 636 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 637 | for ws.Scan() { |
| 638 | name := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 639 | name, err := filepath.Abs(name) |
| 640 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 641 | logf("abs: %v", err) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 642 | continue |
| 643 | } |
| 644 | name, err = filepath.EvalSymlinks(name) |
| 645 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 646 | logf("realpath: %v", err) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 647 | continue |
| 648 | } |
Shinichiro Hamaji | a135891 | 2015-04-11 12:16:21 +0900 | [diff] [blame] | 649 | sw.WriteString(name) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 650 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 651 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 652 | stats.add("funcbody", "realpath", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 653 | } |
| 654 | |
| 655 | type funcAbspath struct{ fclosure } |
| 656 | |
| 657 | func (f *funcAbspath) Arity() int { return 1 } |
| 658 | func (f *funcAbspath) Eval(w io.Writer, ev *Evaluator) { |
| 659 | assertArity("abspath", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 660 | abuf := newBuf() |
| 661 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 662 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 663 | ws := newWordScanner(abuf.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 664 | sw := ssvWriter{w: w} |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 665 | for ws.Scan() { |
| 666 | name := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 667 | name, err := filepath.Abs(name) |
| 668 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 669 | logf("abs: %v", err) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 670 | continue |
| 671 | } |
Shinichiro Hamaji | a135891 | 2015-04-11 12:16:21 +0900 | [diff] [blame] | 672 | sw.WriteString(name) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 673 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 674 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 675 | stats.add("funcbody", "abspath", t) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 676 | } |
| 677 | |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 678 | // http://www.gnu.org/software/make/manual/make.html#Conditional-Functions |
| 679 | type funcIf struct{ fclosure } |
| 680 | |
| 681 | func (f *funcIf) Arity() int { return 3 } |
| 682 | func (f *funcIf) Eval(w io.Writer, ev *Evaluator) { |
| 683 | assertArity("if", 2, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 684 | abuf := newBuf() |
| 685 | f.args[1].Eval(abuf, ev) |
| 686 | if len(abuf.Bytes()) != 0 { |
| 687 | freeBuf(abuf) |
Fumitoshi Ukai | e27a25d | 2015-04-18 00:31:01 +0900 | [diff] [blame] | 688 | f.args[2].Eval(w, ev) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 689 | return |
| 690 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 691 | freeBuf(abuf) |
Fumitoshi Ukai | e27a25d | 2015-04-18 00:31:01 +0900 | [diff] [blame] | 692 | if len(f.args) > 3 { |
| 693 | f.args[3].Eval(w, ev) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 694 | } |
| 695 | } |
| 696 | |
| 697 | type funcAnd struct{ fclosure } |
| 698 | |
| 699 | func (f *funcAnd) Arity() int { return 0 } |
| 700 | func (f *funcAnd) Eval(w io.Writer, ev *Evaluator) { |
| 701 | assertArity("and", 0, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 702 | abuf := newBuf() |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 703 | var cond []byte |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 704 | for _, arg := range f.args[1:] { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 705 | abuf.Reset() |
| 706 | arg.Eval(abuf, ev) |
| 707 | cond = abuf.Bytes() |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 708 | if len(cond) == 0 { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 709 | freeBuf(abuf) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 710 | return |
| 711 | } |
| 712 | } |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 713 | w.Write(cond) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 714 | freeBuf(abuf) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 715 | } |
| 716 | |
| 717 | type funcOr struct{ fclosure } |
| 718 | |
| 719 | func (f *funcOr) Arity() int { return 0 } |
| 720 | func (f *funcOr) Eval(w io.Writer, ev *Evaluator) { |
| 721 | assertArity("or", 0, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 722 | abuf := newBuf() |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 723 | for _, arg := range f.args[1:] { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 724 | abuf.Reset() |
| 725 | arg.Eval(abuf, ev) |
| 726 | cond := abuf.Bytes() |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 727 | if len(cond) != 0 { |
| 728 | w.Write(cond) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 729 | freeBuf(abuf) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 730 | return |
| 731 | } |
| 732 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 733 | freeBuf(abuf) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 734 | } |
| 735 | |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 736 | // http://www.gnu.org/software/make/manual/make.html#Shell-Function |
| 737 | type funcShell struct{ fclosure } |
| 738 | |
| 739 | func (f *funcShell) Arity() int { return 1 } |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 740 | |
Shinichiro Hamaji | c0095e3 | 2015-05-27 18:49:06 +0900 | [diff] [blame] | 741 | // A hack for Android build. We need to evaluate things like $((3+4)) |
| 742 | // when we emit ninja file, because the result of such expressions |
| 743 | // will be passed to other make functions. |
| 744 | // TODO: Maybe we should modify Android's Makefile and remove this |
| 745 | // workaround. It would be also nice if we can detect things like |
| 746 | // this. |
| 747 | func hasNoIoInShellScript(s []byte) bool { |
| 748 | if len(s) == 0 { |
| 749 | return true |
| 750 | } |
| 751 | if !bytes.HasPrefix(s, []byte("echo $((")) || s[len(s)-1] != ')' { |
| 752 | return false |
| 753 | } |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 754 | logf("has no IO - evaluate now: %s", s) |
Shinichiro Hamaji | c0095e3 | 2015-05-27 18:49:06 +0900 | [diff] [blame] | 755 | return true |
| 756 | } |
| 757 | |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 758 | func (f *funcShell) Eval(w io.Writer, ev *Evaluator) { |
| 759 | assertArity("shell", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 760 | abuf := newBuf() |
| 761 | f.args[1].Eval(abuf, ev) |
Shinichiro Hamaji | c0095e3 | 2015-05-27 18:49:06 +0900 | [diff] [blame] | 762 | if ev.avoidIO && !hasNoIoInShellScript(abuf.Bytes()) { |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 763 | te := traceEvent.begin("shell", tmpval(abuf.Bytes()), traceEventMain) |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 764 | ev.hasIO = true |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 765 | io.WriteString(w, "$(") |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 766 | w.Write(abuf.Bytes()) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 767 | writeByte(w, ')') |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 768 | traceEvent.end(te) |
Fumitoshi Ukai | 6a64323 | 2015-06-08 14:11:23 +0900 | [diff] [blame] | 769 | freeBuf(abuf) |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 770 | return |
| 771 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 772 | arg := abuf.String() |
| 773 | freeBuf(abuf) |
Shinichiro Hamaji | 89551c9 | 2015-04-11 19:33:03 +0900 | [diff] [blame] | 774 | shellVar := ev.LookupVar("SHELL") |
| 775 | // TODO: Should be Eval, not String. |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 776 | cmdline := []string{shellVar.String(), "-c", arg} |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 777 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 778 | logf("shell %q", cmdline) |
Fumitoshi Ukai | 44ae8cf | 2015-06-24 16:44:15 +0900 | [diff] [blame] | 779 | } |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 780 | cmd := exec.Cmd{ |
| 781 | Path: cmdline[0], |
| 782 | Args: cmdline, |
| 783 | Stderr: os.Stderr, |
| 784 | } |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 785 | te := traceEvent.begin("shell", literal(arg), traceEventMain) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 786 | out, err := cmd.Output() |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 787 | shellStats.add(time.Since(te.t)) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 788 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 789 | logf("$(shell %q) failed: %q", arg, err) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 790 | } |
Shinichiro Hamaji | ba6b84d | 2015-06-03 18:46:13 +0900 | [diff] [blame] | 791 | w.Write(formatCommandOutput(out)) |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 792 | traceEvent.end(te) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 793 | } |
| 794 | |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 795 | func (f *funcShell) Compact() Value { |
| 796 | if len(f.args)-1 < 1 { |
| 797 | return f |
| 798 | } |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 799 | if !UseFindCache && !UseShellBuiltins { |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 800 | return f |
| 801 | } |
| 802 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 803 | var exp expr |
Fumitoshi Ukai | 44ae8cf | 2015-06-24 16:44:15 +0900 | [diff] [blame] | 804 | switch v := f.args[1].(type) { |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 805 | case expr: |
| 806 | exp = v |
Fumitoshi Ukai | 44ae8cf | 2015-06-24 16:44:15 +0900 | [diff] [blame] | 807 | default: |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 808 | exp = expr{v} |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 809 | } |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 810 | if UseShellBuiltins { |
Fumitoshi Ukai | 4a70851 | 2015-06-11 17:15:49 +0900 | [diff] [blame] | 811 | // hack for android |
| 812 | for _, sb := range shBuiltins { |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 813 | if v, ok := matchExpr(exp, sb.pattern); ok { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 814 | logf("shell compact apply %s for %s", sb.name, exp) |
Fumitoshi Ukai | 4a70851 | 2015-06-11 17:15:49 +0900 | [diff] [blame] | 815 | return sb.compact(f, v) |
| 816 | } |
Fumitoshi Ukai | 76b609e | 2015-06-10 14:56:49 +0900 | [diff] [blame] | 817 | } |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 818 | logf("shell compact no match: %s", exp) |
Fumitoshi Ukai | 76b609e | 2015-06-10 14:56:49 +0900 | [diff] [blame] | 819 | } |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 820 | return f |
| 821 | } |
| 822 | |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 823 | // https://www.gnu.org/software/make/manual/html_node/Call-Function.html#Call-Function |
| 824 | type funcCall struct{ fclosure } |
| 825 | |
| 826 | func (f *funcCall) Arity() int { return 0 } |
| 827 | |
| 828 | func (f *funcCall) Eval(w io.Writer, ev *Evaluator) { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 829 | abuf := newBuf() |
| 830 | fargs := ev.args(abuf, f.args[1:]...) |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 831 | varname := fargs[0] |
| 832 | variable := string(varname) |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 833 | te := traceEvent.begin("call", literal(variable), traceEventMain) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 834 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 835 | logf("call %q variable %q", f.args[1], variable) |
Fumitoshi Ukai | 5c04ad8 | 2015-06-18 16:14:04 +0900 | [diff] [blame] | 836 | } |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 837 | v := ev.LookupVar(variable) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 838 | // Evalualte all arguments first before we modify the table. |
Shinichiro Hamaji | 39728f1 | 2015-04-11 20:12:23 +0900 | [diff] [blame] | 839 | var args []tmpval |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 840 | // $0 is variable. |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 841 | args = append(args, tmpval(varname)) |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 842 | // TODO(ukai): If variable is the name of a built-in function, |
| 843 | // the built-in function is always invoked (even if a make variable |
| 844 | // by that name also exists). |
| 845 | |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 846 | for i, arg := range fargs[1:] { |
| 847 | // f.args[2]=>args[1] will be $1. |
| 848 | args = append(args, tmpval(arg)) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 849 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 850 | logf("call $%d: %q=>%q", i+1, arg, fargs[i+1]) |
Fumitoshi Ukai | 5c04ad8 | 2015-06-18 16:14:04 +0900 | [diff] [blame] | 851 | } |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 852 | } |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 853 | oldParams := ev.paramVars |
| 854 | ev.paramVars = args |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 855 | |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 856 | var buf bytes.Buffer |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 857 | if LogFlag { |
Fumitoshi Ukai | e27a25d | 2015-04-18 00:31:01 +0900 | [diff] [blame] | 858 | w = io.MultiWriter(w, &buf) |
| 859 | } |
| 860 | v.Eval(w, ev) |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 861 | ev.paramVars = oldParams |
Fumitoshi Ukai | 72598e7 | 2015-06-11 15:53:09 +0900 | [diff] [blame] | 862 | traceEvent.end(te) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 863 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 864 | logf("call %q variable %q return %q", f.args[1], variable, buf.Bytes()) |
Fumitoshi Ukai | 5c04ad8 | 2015-06-18 16:14:04 +0900 | [diff] [blame] | 865 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 866 | freeBuf(abuf) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 867 | } |
| 868 | |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 869 | // http://www.gnu.org/software/make/manual/make.html#Value-Function |
| 870 | type funcValue struct{ fclosure } |
| 871 | |
| 872 | func (f *funcValue) Arity() int { return 1 } |
| 873 | func (f *funcValue) Eval(w io.Writer, ev *Evaluator) { |
| 874 | assertArity("value", 1, len(f.args)) |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 875 | v := ev.LookupVar(f.args[1].String()) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 876 | io.WriteString(w, v.String()) |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 877 | } |
| 878 | |
| 879 | // http://www.gnu.org/software/make/manual/make.html#Eval-Function |
| 880 | type funcEval struct{ fclosure } |
| 881 | |
| 882 | func (f *funcEval) Arity() int { return 1 } |
| 883 | func (f *funcEval) Eval(w io.Writer, ev *Evaluator) { |
| 884 | assertArity("eval", 1, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 885 | abuf := newBuf() |
| 886 | f.args[1].Eval(abuf, ev) |
| 887 | s := abuf.Bytes() |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 888 | logf("eval %q at %s:%d", s, ev.filename, ev.lineno) |
Fumitoshi Ukai | 08eda65 | 2015-06-25 16:27:10 +0900 | [diff] [blame] | 889 | mk, err := parseMakefileBytes(s, ev.filename, ev.lineno) |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 890 | if err != nil { |
| 891 | panic(err) |
| 892 | } |
| 893 | |
| 894 | for _, stmt := range mk.stmts { |
| 895 | ev.eval(stmt) |
| 896 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 897 | freeBuf(abuf) |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 898 | } |
| 899 | |
Fumitoshi Ukai | 9f6b635 | 2015-04-16 16:25:09 +0900 | [diff] [blame] | 900 | func (f *funcEval) Compact() Value { |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 901 | if len(f.args)-1 < 1 { |
| 902 | return f |
| 903 | } |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 904 | switch arg := f.args[1].(type) { |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 905 | case literal, tmpval: |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 906 | case expr: |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 907 | if len(arg) == 1 { |
| 908 | return f |
| 909 | } |
| 910 | switch prefix := arg[0].(type) { |
| 911 | case literal, tmpval: |
| 912 | lhs, op, rhsprefix, ok := parseAssignLiteral(prefix.String()) |
| 913 | if ok { |
| 914 | // $(eval foo = $(bar)) |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 915 | var rhs expr |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 916 | if rhsprefix != literal("") { |
| 917 | rhs = append(rhs, rhsprefix) |
| 918 | } |
| 919 | rhs = append(rhs, arg[1:]...) |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 920 | logf("eval assign %#v => lhs:%q op:%q rhs:%#v", f, lhs, op, rhs) |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 921 | return &funcEvalAssign{ |
| 922 | lhs: lhs, |
| 923 | op: op, |
| 924 | rhs: compactExpr(rhs), |
| 925 | } |
| 926 | } |
| 927 | } |
| 928 | // TODO(ukai): eval -> varassign. e.g $(eval $(foo) := $(x)). |
Fumitoshi Ukai | 769157a | 2015-06-02 15:18:53 +0900 | [diff] [blame] | 929 | return f |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 930 | default: |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 931 | return f |
| 932 | } |
| 933 | arg := f.args[1].String() |
| 934 | arg = stripComment(arg) |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 935 | if arg == "" || strings.TrimSpace(arg) == "" { |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 936 | return &funcNop{expr: f.String()} |
| 937 | } |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 938 | f.args[1] = literal(arg) |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 939 | lhs, op, rhs, ok := parseAssignLiteral(f.args[1].String()) |
| 940 | if ok { |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 941 | return &funcEvalAssign{ |
| 942 | lhs: lhs, |
| 943 | op: op, |
| 944 | rhs: rhs, |
| 945 | } |
| 946 | } |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 947 | return f |
| 948 | } |
| 949 | |
| 950 | func stripComment(arg string) string { |
| 951 | for { |
| 952 | i := strings.Index(arg, "#") |
| 953 | if i < 0 { |
| 954 | return arg |
| 955 | } |
| 956 | eol := strings.Index(arg[i:], "\n") |
| 957 | if eol < 0 { |
| 958 | return arg[:i] |
| 959 | } |
| 960 | arg = arg[:i] + arg[eol+1:] |
| 961 | } |
| 962 | } |
| 963 | |
| 964 | type funcNop struct{ expr string } |
| 965 | |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 966 | func (f *funcNop) String() string { return f.expr } |
| 967 | func (f *funcNop) Eval(io.Writer, *Evaluator) {} |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 968 | func (f *funcNop) serialize() serializableVar { |
| 969 | return serializableVar{ |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 970 | Type: "funcNop", |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 971 | V: f.expr, |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 972 | } |
| 973 | } |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 974 | func (f *funcNop) dump(w io.Writer) { |
Fumitoshi Ukai | a78fce0 | 2015-06-25 12:29:09 +0900 | [diff] [blame] | 975 | dumpByte(w, valueTypeNop) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 976 | } |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 977 | |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 978 | func parseAssignLiteral(s string) (lhs, op string, rhs Value, ok bool) { |
| 979 | eq := strings.Index(s, "=") |
| 980 | if eq < 0 { |
| 981 | return "", "", nil, false |
| 982 | } |
| 983 | // TODO(ukai): factor out parse assign? |
| 984 | lhs = s[:eq] |
| 985 | op = s[eq : eq+1] |
| 986 | if eq >= 1 && (s[eq-1] == ':' || s[eq-1] == '+' || s[eq-1] == '?') { |
| 987 | lhs = s[:eq-1] |
| 988 | op = s[eq-1 : eq+1] |
| 989 | } |
| 990 | lhs = strings.TrimSpace(lhs) |
| 991 | if strings.IndexAny(lhs, ":$") >= 0 { |
| 992 | // target specific var, or need eval. |
| 993 | return "", "", nil, false |
| 994 | } |
| 995 | r := strings.TrimLeft(s[eq+1:], " \t") |
| 996 | rhs = literal(r) |
| 997 | return lhs, op, rhs, true |
| 998 | } |
| 999 | |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1000 | type funcEvalAssign struct { |
| 1001 | lhs string |
| 1002 | op string |
| 1003 | rhs Value |
| 1004 | } |
| 1005 | |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1006 | func (f *funcEvalAssign) String() string { |
| 1007 | return fmt.Sprintf("$(eval %s %s %s)", f.lhs, f.op, f.rhs) |
| 1008 | } |
| 1009 | |
| 1010 | func (f *funcEvalAssign) Eval(w io.Writer, ev *Evaluator) { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1011 | var abuf bytes.Buffer |
| 1012 | f.rhs.Eval(&abuf, ev) |
| 1013 | rhs := trimLeftSpaceBytes(abuf.Bytes()) |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1014 | var rvalue Var |
| 1015 | switch f.op { |
| 1016 | case ":=": |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1017 | // TODO(ukai): compute parsed expr in Compact when f.rhs is |
| 1018 | // literal? e.g. literal("$(foo)") => varref{literal("foo")}. |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 1019 | exp, _, err := parseExpr(rhs, nil, false) |
Fumitoshi Ukai | fa5e922 | 2015-04-17 11:45:20 +0900 | [diff] [blame] | 1020 | if err != nil { |
| 1021 | panic(fmt.Sprintf("eval assign error: %q: %v", f.String(), err)) |
| 1022 | } |
Fumitoshi Ukai | dd248f2 | 2015-06-18 23:04:28 +0900 | [diff] [blame] | 1023 | vbuf := newBuf() |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 1024 | exp.Eval(vbuf, ev) |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1025 | rvalue = &simpleVar{value: vbuf.String(), origin: "file"} |
Fumitoshi Ukai | dd248f2 | 2015-06-18 23:04:28 +0900 | [diff] [blame] | 1026 | freeBuf(vbuf) |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1027 | case "=": |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1028 | rvalue = &recursiveVar{expr: tmpval(rhs), origin: "file"} |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1029 | case "+=": |
| 1030 | prev := ev.LookupVar(f.lhs) |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1031 | if prev.IsDefined() { |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1032 | rvalue = prev.Append(ev, string(rhs)) |
| 1033 | } else { |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1034 | rvalue = &recursiveVar{expr: tmpval(rhs), origin: "file"} |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1035 | } |
| 1036 | case "?=": |
| 1037 | prev := ev.LookupVar(f.lhs) |
| 1038 | if prev.IsDefined() { |
| 1039 | return |
| 1040 | } |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1041 | rvalue = &recursiveVar{expr: tmpval(rhs), origin: "file"} |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1042 | } |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 1043 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1044 | logf("Eval ASSIGN: %s=%q (flavor:%q)", f.lhs, rvalue, rvalue.Flavor()) |
Fumitoshi Ukai | 5c04ad8 | 2015-06-18 16:14:04 +0900 | [diff] [blame] | 1045 | } |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1046 | ev.outVars.Assign(f.lhs, rvalue) |
| 1047 | } |
| 1048 | |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 1049 | func (f *funcEvalAssign) serialize() serializableVar { |
| 1050 | return serializableVar{ |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 1051 | Type: "funcEvalAssign", |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 1052 | Children: []serializableVar{ |
| 1053 | serializableVar{V: f.lhs}, |
| 1054 | serializableVar{V: f.op}, |
| 1055 | f.rhs.serialize(), |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 1056 | }, |
| 1057 | } |
| 1058 | } |
| 1059 | |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 1060 | func (f *funcEvalAssign) dump(w io.Writer) { |
Fumitoshi Ukai | a78fce0 | 2015-06-25 12:29:09 +0900 | [diff] [blame] | 1061 | dumpByte(w, valueTypeAssign) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 1062 | dumpString(w, f.lhs) |
| 1063 | dumpString(w, f.op) |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 1064 | f.rhs.dump(w) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 1065 | } |
| 1066 | |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1067 | // http://www.gnu.org/software/make/manual/make.html#Origin-Function |
| 1068 | type funcOrigin struct{ fclosure } |
| 1069 | |
| 1070 | func (f *funcOrigin) Arity() int { return 1 } |
| 1071 | func (f *funcOrigin) Eval(w io.Writer, ev *Evaluator) { |
| 1072 | assertArity("origin", 1, len(f.args)) |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 1073 | v := ev.LookupVar(f.args[1].String()) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1074 | io.WriteString(w, v.Origin()) |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1075 | } |
| 1076 | |
| 1077 | // https://www.gnu.org/software/make/manual/html_node/Flavor-Function.html#Flavor-Function |
| 1078 | type funcFlavor struct{ fclosure } |
| 1079 | |
| 1080 | func (f *funcFlavor) Arity() int { return 1 } |
| 1081 | func (f *funcFlavor) Eval(w io.Writer, ev *Evaluator) { |
| 1082 | assertArity("flavor", 1, len(f.args)) |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 1083 | v := ev.LookupVar(f.args[1].String()) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1084 | io.WriteString(w, v.Flavor()) |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1085 | } |
| 1086 | |
| 1087 | // http://www.gnu.org/software/make/manual/make.html#Make-Control-Functions |
| 1088 | type funcInfo struct{ fclosure } |
| 1089 | |
| 1090 | func (f *funcInfo) Arity() int { return 1 } |
| 1091 | func (f *funcInfo) Eval(w io.Writer, ev *Evaluator) { |
| 1092 | assertArity("info", 1, len(f.args)) |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1093 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1094 | io.WriteString(w, "KATI_TODO(info)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1095 | ev.hasIO = true |
| 1096 | return |
| 1097 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1098 | abuf := newBuf() |
| 1099 | f.args[1].Eval(abuf, ev) |
| 1100 | fmt.Printf("%s\n", abuf.String()) |
| 1101 | freeBuf(abuf) |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1102 | } |
| 1103 | |
| 1104 | type funcWarning struct{ fclosure } |
| 1105 | |
| 1106 | func (f *funcWarning) Arity() int { return 1 } |
| 1107 | func (f *funcWarning) Eval(w io.Writer, ev *Evaluator) { |
| 1108 | assertArity("warning", 1, len(f.args)) |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1109 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1110 | io.WriteString(w, "KATI_TODO(warning)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1111 | ev.hasIO = true |
| 1112 | return |
| 1113 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1114 | abuf := newBuf() |
| 1115 | f.args[1].Eval(abuf, ev) |
| 1116 | fmt.Printf("%s:%d: %s\n", ev.filename, ev.lineno, abuf.String()) |
| 1117 | freeBuf(abuf) |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1118 | } |
| 1119 | |
| 1120 | type funcError struct{ fclosure } |
| 1121 | |
| 1122 | func (f *funcError) Arity() int { return 1 } |
| 1123 | func (f *funcError) Eval(w io.Writer, ev *Evaluator) { |
| 1124 | assertArity("error", 1, len(f.args)) |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1125 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1126 | io.WriteString(w, "KATI_TODO(error)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1127 | ev.hasIO = true |
| 1128 | return |
| 1129 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1130 | abuf := newBuf() |
| 1131 | f.args[1].Eval(abuf, ev) |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1132 | errorExit(ev.filename, ev.lineno, "*** %s.", abuf.String()) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1133 | freeBuf(abuf) |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1134 | } |
| 1135 | |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1136 | // http://www.gnu.org/software/make/manual/make.html#Foreach-Function |
| 1137 | type funcForeach struct{ fclosure } |
| 1138 | |
| 1139 | func (f *funcForeach) Arity() int { return 3 } |
| 1140 | |
| 1141 | func (f *funcForeach) Eval(w io.Writer, ev *Evaluator) { |
| 1142 | assertArity("foreach", 3, len(f.args)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1143 | abuf := newBuf() |
| 1144 | fargs := ev.args(abuf, f.args[1], f.args[2]) |
| 1145 | varname := string(fargs[0]) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 1146 | ws := newWordScanner(fargs[1]) |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 1147 | text := f.args[3] |
Fumitoshi Ukai | 9b10ecf | 2015-04-15 17:45:50 +0900 | [diff] [blame] | 1148 | restore := ev.outVars.save(varname) |
| 1149 | defer restore() |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1150 | space := false |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 1151 | for ws.Scan() { |
| 1152 | word := ws.Bytes() |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1153 | ev.outVars.Assign(varname, &automaticVar{value: word}) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1154 | if space { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1155 | writeByte(w, ' ') |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1156 | } |
Fumitoshi Ukai | e27a25d | 2015-04-18 00:31:01 +0900 | [diff] [blame] | 1157 | text.Eval(w, ev) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1158 | space = true |
| 1159 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1160 | freeBuf(abuf) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1161 | } |