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 | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 93 | type arityError struct { |
| 94 | narg int |
| 95 | name string |
| 96 | } |
| 97 | |
| 98 | func (e arityError) Error() string { |
| 99 | return fmt.Sprintf("*** insufficient number of arguments (%d) to function `%s'.", e.narg, e.name) |
| 100 | } |
| 101 | |
| 102 | func assertArity(name string, req, n int) error { |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 103 | if n-1 < req { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 104 | return arityError{narg: n - 1, name: name} |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 105 | } |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 106 | return nil |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 107 | } |
| 108 | |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 109 | func numericValueForFunc(v string) (int, bool) { |
| 110 | n, err := strconv.Atoi(v) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 111 | if err != nil || n < 0 { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 112 | return n, false |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 113 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 114 | return n, true |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 115 | } |
| 116 | |
Shinichiro Hamaji | ba6b84d | 2015-06-03 18:46:13 +0900 | [diff] [blame] | 117 | func formatCommandOutput(out []byte) []byte { |
| 118 | out = bytes.TrimRight(out, "\n") |
| 119 | out = bytes.Replace(out, []byte{'\n'}, []byte{' '}, -1) |
| 120 | return out |
| 121 | } |
| 122 | |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 123 | type fclosure struct { |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 124 | // args[0] is "(funcname", or "{funcname". |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 125 | args []Value |
| 126 | } |
| 127 | |
| 128 | func (c *fclosure) AddArg(v Value) { |
| 129 | c.args = append(c.args, v) |
Fumitoshi Ukai | 0b5e18b | 2015-04-03 22:57:17 +0900 | [diff] [blame] | 130 | } |
| 131 | |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 132 | func (c *fclosure) String() string { |
| 133 | if len(c.args) == 0 { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 134 | return "$(func)" |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 135 | } |
| 136 | arg0 := c.args[0].String() |
| 137 | if arg0 == "" { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 138 | return "$(func )" |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 139 | } |
| 140 | cp := closeParen(arg0[0]) |
| 141 | if cp == 0 { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 142 | return "${func }" |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 143 | } |
| 144 | var args []string |
| 145 | for _, arg := range c.args[1:] { |
| 146 | args = append(args, arg.String()) |
| 147 | } |
| 148 | return fmt.Sprintf("$%s %s%c", arg0, strings.Join(args, ","), cp) |
| 149 | } |
Fumitoshi Ukai | 7878113 | 2015-04-10 17:08:40 +0900 | [diff] [blame] | 150 | |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 151 | func (c *fclosure) serialize() serializableVar { |
| 152 | r := serializableVar{Type: "func"} |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 153 | for _, a := range c.args { |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 154 | r.Children = append(r.Children, a.serialize()) |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 155 | } |
| 156 | return r |
| 157 | } |
| 158 | |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 159 | func (c *fclosure) dump(d *dumpbuf) { |
| 160 | d.Byte(valueTypeFunc) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 161 | for _, a := range c.args { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 162 | a.dump(d) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 163 | } |
| 164 | } |
| 165 | |
Shinichiro Hamaji | 4125cf4 | 2015-04-03 11:42:28 +0900 | [diff] [blame] | 166 | // http://www.gnu.org/software/make/manual/make.html#Text-Functions |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 167 | type funcSubst struct{ fclosure } |
| 168 | |
| 169 | func (f *funcSubst) Arity() int { return 3 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 170 | func (f *funcSubst) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 171 | err := assertArity("subst", 3, len(f.args)) |
| 172 | if err != nil { |
| 173 | return err |
| 174 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 175 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 176 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 177 | if err != nil { |
| 178 | return err |
| 179 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 180 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 181 | from := fargs[0] |
| 182 | to := fargs[1] |
| 183 | text := fargs[2] |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 184 | logf("subst from:%q to:%q text:%q", from, to, text) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 185 | w.Write(bytes.Replace(text, from, to, -1)) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 186 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 187 | stats.add("funcbody", "subst", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 188 | return nil |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 189 | } |
| 190 | |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 191 | type funcPatsubst struct{ fclosure } |
| 192 | |
| 193 | func (f *funcPatsubst) Arity() int { return 3 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 194 | func (f *funcPatsubst) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 195 | err := assertArity("patsubst", 3, len(f.args)) |
| 196 | if err != nil { |
| 197 | return err |
| 198 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 199 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 200 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 201 | if err != nil { |
| 202 | return err |
| 203 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 204 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 205 | pat := fargs[0] |
| 206 | repl := fargs[1] |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 207 | ws := newWordScanner(fargs[2]) |
Fumitoshi Ukai | 77411fb | 2015-06-19 15:46:21 +0900 | [diff] [blame] | 208 | space := false |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 209 | for ws.Scan() { |
Fumitoshi Ukai | 77411fb | 2015-06-19 15:46:21 +0900 | [diff] [blame] | 210 | if space { |
| 211 | writeByte(w, ' ') |
| 212 | } |
| 213 | pre, subst, post := substPatternBytes(pat, repl, ws.Bytes()) |
| 214 | w.Write(pre) |
| 215 | if subst != nil { |
| 216 | w.Write(subst) |
| 217 | w.Write(post) |
| 218 | } |
| 219 | space = true |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 220 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 221 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 222 | stats.add("funcbody", "patsubst", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 223 | return nil |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 224 | } |
| 225 | |
| 226 | type funcStrip struct{ fclosure } |
| 227 | |
| 228 | func (f *funcStrip) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 229 | func (f *funcStrip) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 230 | err := assertArity("strip", 1, len(f.args)) |
| 231 | if err != nil { |
| 232 | return err |
| 233 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 234 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 235 | err = f.args[1].Eval(abuf, ev) |
| 236 | if err != nil { |
| 237 | return err |
| 238 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 239 | t := time.Now() |
Shinichiro Hamaji | b978e11 | 2015-06-05 11:02:45 +0900 | [diff] [blame] | 240 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | a44b766 | 2015-06-19 11:07:07 +0900 | [diff] [blame] | 241 | space := false |
Shinichiro Hamaji | b978e11 | 2015-06-05 11:02:45 +0900 | [diff] [blame] | 242 | for ws.Scan() { |
Fumitoshi Ukai | a44b766 | 2015-06-19 11:07:07 +0900 | [diff] [blame] | 243 | if space { |
| 244 | writeByte(w, ' ') |
| 245 | } |
| 246 | w.Write(ws.Bytes()) |
| 247 | space = true |
Shinichiro Hamaji | b978e11 | 2015-06-05 11:02:45 +0900 | [diff] [blame] | 248 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 249 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 250 | stats.add("funcbody", "strip", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 251 | return nil |
Shinichiro Hamaji | 98e910d | 2015-04-11 10:38:44 +0900 | [diff] [blame] | 252 | } |
| 253 | |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 254 | type funcFindstring struct{ fclosure } |
| 255 | |
| 256 | func (f *funcFindstring) Arity() int { return 2 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 257 | func (f *funcFindstring) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 258 | err := assertArity("findstring", 2, len(f.args)) |
| 259 | if err != nil { |
| 260 | return err |
| 261 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 262 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 263 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 264 | if err != nil { |
| 265 | return err |
| 266 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 267 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 268 | find := fargs[0] |
| 269 | text := fargs[1] |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 270 | if bytes.Index(text, find) >= 0 { |
| 271 | w.Write(find) |
| 272 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 273 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 274 | stats.add("funcbody", "findstring", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 275 | return nil |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 276 | } |
| 277 | |
| 278 | type funcFilter struct{ fclosure } |
| 279 | |
| 280 | func (f *funcFilter) Arity() int { return 2 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 281 | func (f *funcFilter) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 282 | err := assertArity("filter", 2, len(f.args)) |
| 283 | if err != nil { |
| 284 | return err |
| 285 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 286 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 287 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 288 | if err != nil { |
| 289 | return err |
| 290 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 291 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 292 | var patterns [][]byte |
| 293 | ws := newWordScanner(fargs[0]) |
| 294 | for ws.Scan() { |
| 295 | patterns = append(patterns, ws.Bytes()) |
| 296 | } |
| 297 | ws = newWordScanner(fargs[1]) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 298 | for ws.Scan() { |
| 299 | text := ws.Bytes() |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 300 | for _, pat := range patterns { |
| 301 | if matchPatternBytes(pat, text) { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 302 | w.writeWord(text) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 303 | } |
| 304 | } |
| 305 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 306 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 307 | stats.add("funcbody", "filter", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 308 | return nil |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 309 | } |
| 310 | |
| 311 | type funcFilterOut struct{ fclosure } |
| 312 | |
| 313 | func (f *funcFilterOut) Arity() int { return 2 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 314 | func (f *funcFilterOut) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 315 | err := assertArity("filter-out", 2, len(f.args)) |
| 316 | if err != nil { |
| 317 | return err |
| 318 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 319 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 320 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 321 | if err != nil { |
| 322 | return err |
| 323 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 324 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 325 | var patterns [][]byte |
| 326 | ws := newWordScanner(fargs[0]) |
| 327 | for ws.Scan() { |
| 328 | patterns = append(patterns, ws.Bytes()) |
| 329 | } |
| 330 | ws = newWordScanner(fargs[1]) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 331 | Loop: |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 332 | for ws.Scan() { |
| 333 | text := ws.Bytes() |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 334 | for _, pat := range patterns { |
| 335 | if matchPatternBytes(pat, text) { |
| 336 | continue Loop |
| 337 | } |
| 338 | } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 339 | w.writeWord(text) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 340 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 341 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 342 | stats.add("funcbody", "filter-out", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 343 | return err |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 344 | } |
| 345 | |
| 346 | type funcSort struct{ fclosure } |
| 347 | |
| 348 | func (f *funcSort) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 349 | func (f *funcSort) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 350 | err := assertArity("sort", 1, len(f.args)) |
| 351 | if err != nil { |
| 352 | return err |
| 353 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 354 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 355 | err = f.args[1].Eval(abuf, ev) |
| 356 | if err != nil { |
| 357 | return err |
| 358 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 359 | t := time.Now() |
Fumitoshi Ukai | 79cfbab | 2015-06-18 23:28:26 +0900 | [diff] [blame] | 360 | ws := newWordScanner(abuf.Bytes()) |
| 361 | var toks []string |
| 362 | for ws.Scan() { |
| 363 | toks = append(toks, string(ws.Bytes())) |
| 364 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 365 | freeBuf(abuf) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 366 | sort.Strings(toks) |
| 367 | |
| 368 | // Remove duplicate words. |
| 369 | var prev string |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 370 | for _, tok := range toks { |
Fumitoshi Ukai | ce14acb | 2015-06-19 13:54:53 +0900 | [diff] [blame] | 371 | if prev == tok { |
| 372 | continue |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 373 | } |
Fumitoshi Ukai | ce14acb | 2015-06-19 13:54:53 +0900 | [diff] [blame] | 374 | if prev != "" { |
| 375 | writeByte(w, ' ') |
| 376 | } |
| 377 | io.WriteString(w, tok) |
| 378 | prev = tok |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 379 | } |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 380 | stats.add("funcbody", "sort", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 381 | return nil |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 382 | } |
| 383 | |
| 384 | type funcWord struct{ fclosure } |
| 385 | |
| 386 | func (f *funcWord) Arity() int { return 2 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 387 | func (f *funcWord) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 388 | err := assertArity("word", 2, len(f.args)) |
| 389 | if err != nil { |
| 390 | return err |
| 391 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 392 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 393 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 394 | if err != nil { |
| 395 | return err |
| 396 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 397 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 398 | v := string(trimSpaceBytes(fargs[0])) |
| 399 | index, ok := numericValueForFunc(v) |
| 400 | if !ok { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 401 | return ev.errorf(`*** non-numeric first argument to "word" function: %q.`, v) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 402 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 403 | if index == 0 { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 404 | return ev.errorf(`*** first argument to "word" function must be greater than 0.`) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 405 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 406 | ws := newWordScanner(fargs[1]) |
| 407 | for ws.Scan() { |
| 408 | index-- |
| 409 | if index == 0 { |
| 410 | w.Write(ws.Bytes()) |
| 411 | break |
| 412 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 413 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 414 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 415 | stats.add("funcbody", "word", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 416 | return err |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 417 | } |
| 418 | |
| 419 | type funcWordlist struct{ fclosure } |
| 420 | |
| 421 | func (f *funcWordlist) Arity() int { return 3 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 422 | func (f *funcWordlist) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 423 | err := assertArity("wordlist", 3, len(f.args)) |
| 424 | if err != nil { |
| 425 | return err |
| 426 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 427 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 428 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 429 | if err != nil { |
| 430 | return err |
| 431 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 432 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 433 | v := string(trimSpaceBytes(fargs[0])) |
| 434 | si, ok := numericValueForFunc(v) |
| 435 | if !ok { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 436 | return ev.errorf(`*** non-numeric first argument to "wordlist" function: %q.`, v) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 437 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 438 | if si == 0 { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 439 | return ev.errorf(`*** invalid first argument to "wordlist" function: %s`, f.args[1]) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 440 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 441 | v = string(trimSpaceBytes(fargs[1])) |
| 442 | ei, ok := numericValueForFunc(v) |
| 443 | if !ok { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 444 | return ev.errorf(`*** non-numeric second argument to "wordlist" function: %q.`, v) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 445 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 446 | |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 447 | ws := newWordScanner(fargs[2]) |
| 448 | i := 0 |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 449 | for ws.Scan() { |
| 450 | i++ |
| 451 | if si <= i && i <= ei { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 452 | w.writeWord(ws.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 453 | } |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 454 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 455 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 456 | stats.add("funcbody", "wordlist", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 457 | return nil |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 458 | } |
| 459 | |
| 460 | type funcWords struct{ fclosure } |
| 461 | |
| 462 | func (f *funcWords) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 463 | func (f *funcWords) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 464 | err := assertArity("words", 1, len(f.args)) |
| 465 | if err != nil { |
| 466 | return err |
| 467 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 468 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 469 | err = f.args[1].Eval(abuf, ev) |
| 470 | if err != nil { |
| 471 | return err |
| 472 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 473 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 474 | ws := newWordScanner(abuf.Bytes()) |
| 475 | n := 0 |
| 476 | for ws.Scan() { |
| 477 | n++ |
| 478 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 479 | freeBuf(abuf) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 480 | io.WriteString(w, strconv.Itoa(n)) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 481 | stats.add("funcbody", "words", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 482 | return nil |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 483 | } |
| 484 | |
| 485 | type funcFirstword struct{ fclosure } |
| 486 | |
| 487 | func (f *funcFirstword) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 488 | func (f *funcFirstword) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 489 | err := assertArity("firstword", 1, len(f.args)) |
| 490 | if err != nil { |
| 491 | return err |
| 492 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 493 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 494 | err = f.args[1].Eval(abuf, ev) |
| 495 | if err != nil { |
| 496 | return err |
| 497 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 498 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 499 | ws := newWordScanner(abuf.Bytes()) |
| 500 | if ws.Scan() { |
| 501 | w.Write(ws.Bytes()) |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 502 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 503 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 504 | stats.add("funcbody", "firstword", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 505 | return nil |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 506 | } |
| 507 | |
| 508 | type funcLastword struct{ fclosure } |
| 509 | |
| 510 | func (f *funcLastword) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 511 | func (f *funcLastword) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 512 | err := assertArity("lastword", 1, len(f.args)) |
| 513 | if err != nil { |
| 514 | return err |
| 515 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 516 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 517 | err = f.args[1].Eval(abuf, ev) |
| 518 | if err != nil { |
| 519 | return err |
| 520 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 521 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 522 | ws := newWordScanner(abuf.Bytes()) |
| 523 | var lw []byte |
| 524 | for ws.Scan() { |
| 525 | lw = ws.Bytes() |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 526 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 527 | if lw != nil { |
| 528 | w.Write(lw) |
| 529 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 530 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 531 | stats.add("funcbody", "lastword", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 532 | return err |
Shinichiro Hamaji | 05b222d | 2015-04-11 11:36:37 +0900 | [diff] [blame] | 533 | } |
| 534 | |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 535 | // https://www.gnu.org/software/make/manual/html_node/File-Name-Functions.html#File-Name-Functions |
| 536 | |
| 537 | type funcJoin struct{ fclosure } |
| 538 | |
| 539 | func (f *funcJoin) Arity() int { return 2 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 540 | func (f *funcJoin) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 541 | err := assertArity("join", 2, len(f.args)) |
| 542 | if err != nil { |
| 543 | return err |
| 544 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 545 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 546 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 547 | if err != nil { |
| 548 | return err |
| 549 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 550 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 551 | ws1 := newWordScanner(fargs[0]) |
| 552 | ws2 := newWordScanner(fargs[1]) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 553 | for { |
| 554 | if w1, w2 := ws1.Scan(), ws2.Scan(); !w1 && !w2 { |
| 555 | break |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 556 | } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 557 | var word []byte |
| 558 | word = append(word, ws1.Bytes()...) |
| 559 | word = append(word, ws2.Bytes()...) |
| 560 | w.writeWord(word) |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 561 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 562 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 563 | stats.add("funcbody", "join", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 564 | return nil |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 565 | } |
| 566 | |
| 567 | type funcWildcard struct{ fclosure } |
| 568 | |
| 569 | func (f *funcWildcard) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 570 | func (f *funcWildcard) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 571 | err := assertArity("wildcard", 1, len(f.args)) |
| 572 | if err != nil { |
| 573 | return err |
| 574 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 575 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 576 | err = f.args[1].Eval(abuf, ev) |
| 577 | if err != nil { |
| 578 | return err |
| 579 | } |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 580 | te := traceEvent.begin("wildcard", tmpval(abuf.Bytes()), traceEventMain) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 581 | if ev.avoidIO && !UseWildcardCache { |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 582 | ev.hasIO = true |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 583 | io.WriteString(w, "$(/bin/ls -d ") |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 584 | w.Write(abuf.Bytes()) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 585 | io.WriteString(w, " 2> /dev/null)") |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 586 | traceEvent.end(te) |
Fumitoshi Ukai | 358c68a | 2015-06-08 13:12:55 +0900 | [diff] [blame] | 587 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 588 | return nil |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 589 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 590 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 591 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 592 | for ws.Scan() { |
Shinichiro Hamaji | ba6b84d | 2015-06-03 18:46:13 +0900 | [diff] [blame] | 593 | pat := string(ws.Bytes()) |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 594 | err = wildcard(w, pat) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 595 | if err != nil { |
| 596 | return err |
| 597 | } |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 598 | } |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 599 | traceEvent.end(te) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 600 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 601 | stats.add("funcbody", "wildcard", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 602 | return nil |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 603 | } |
| 604 | |
| 605 | type funcDir struct{ fclosure } |
| 606 | |
| 607 | func (f *funcDir) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 608 | func (f *funcDir) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 609 | err := assertArity("dir", 1, len(f.args)) |
| 610 | if err != nil { |
| 611 | return err |
| 612 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 613 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 614 | err = f.args[1].Eval(abuf, ev) |
| 615 | if err != nil { |
| 616 | return err |
| 617 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 618 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 619 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 620 | for ws.Scan() { |
Shinichiro Hamaji | 5590685 | 2015-06-29 16:40:33 +0900 | [diff] [blame] | 621 | name := filepath.Dir(string(string(ws.Bytes()))) |
Shinichiro Hamaji | 37626b6 | 2015-04-21 16:59:23 +0900 | [diff] [blame] | 622 | if name == "/" { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 623 | w.writeWordString(name) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 624 | continue |
Shinichiro Hamaji | 37626b6 | 2015-04-21 16:59:23 +0900 | [diff] [blame] | 625 | } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 626 | w.writeWordString(name + string(filepath.Separator)) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 627 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 628 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 629 | stats.add("funcbody", "dir", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 630 | return nil |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 631 | } |
| 632 | |
| 633 | type funcNotdir struct{ fclosure } |
| 634 | |
| 635 | func (f *funcNotdir) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 636 | func (f *funcNotdir) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 637 | err := assertArity("notdir", 1, len(f.args)) |
| 638 | if err != nil { |
| 639 | return err |
| 640 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 641 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 642 | err = f.args[1].Eval(abuf, ev) |
| 643 | if err != nil { |
| 644 | return err |
| 645 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 646 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 647 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 648 | for ws.Scan() { |
| 649 | name := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 650 | if name == string(filepath.Separator) { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 651 | w.writeWord([]byte{}) // separator |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 652 | continue |
| 653 | } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 654 | w.writeWordString(filepath.Base(name)) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 655 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 656 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 657 | stats.add("funcbody", "notdir", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 658 | return nil |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 659 | } |
| 660 | |
| 661 | type funcSuffix struct{ fclosure } |
| 662 | |
| 663 | func (f *funcSuffix) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 664 | func (f *funcSuffix) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 665 | err := assertArity("suffix", 1, len(f.args)) |
| 666 | if err != nil { |
| 667 | return err |
| 668 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 669 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 670 | err = f.args[1].Eval(abuf, ev) |
| 671 | if err != nil { |
| 672 | return err |
| 673 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 674 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 675 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 676 | for ws.Scan() { |
| 677 | tok := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 678 | e := filepath.Ext(tok) |
| 679 | if len(e) > 0 { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 680 | w.writeWordString(e) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 681 | } |
| 682 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 683 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 684 | stats.add("funcbody", "suffix", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 685 | return err |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 686 | } |
| 687 | |
| 688 | type funcBasename struct{ fclosure } |
| 689 | |
| 690 | func (f *funcBasename) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 691 | func (f *funcBasename) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 692 | err := assertArity("basename", 1, len(f.args)) |
| 693 | if err != nil { |
| 694 | return err |
| 695 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 696 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 697 | err = f.args[1].Eval(abuf, ev) |
| 698 | if err != nil { |
| 699 | return err |
| 700 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 701 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 702 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 703 | for ws.Scan() { |
| 704 | tok := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 705 | e := stripExt(tok) |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 706 | w.writeWordString(e) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 707 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 708 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 709 | stats.add("funcbody", "basename", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 710 | return nil |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 711 | } |
| 712 | |
| 713 | type funcAddsuffix struct{ fclosure } |
| 714 | |
| 715 | func (f *funcAddsuffix) Arity() int { return 2 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 716 | func (f *funcAddsuffix) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 717 | err := assertArity("addsuffix", 2, len(f.args)) |
| 718 | if err != nil { |
| 719 | return err |
| 720 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 721 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 722 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 723 | if err != nil { |
| 724 | return err |
| 725 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 726 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 727 | suf := fargs[0] |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 728 | ws := newWordScanner(fargs[1]) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 729 | for ws.Scan() { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 730 | var name []byte |
| 731 | name = append(name, ws.Bytes()...) |
| 732 | name = append(name, suf...) |
| 733 | w.writeWord(name) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 734 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 735 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 736 | stats.add("funcbody", "addsuffix", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 737 | return err |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 738 | } |
| 739 | |
| 740 | type funcAddprefix struct{ fclosure } |
| 741 | |
| 742 | func (f *funcAddprefix) Arity() int { return 2 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 743 | func (f *funcAddprefix) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 744 | err := assertArity("addprefix", 2, len(f.args)) |
| 745 | if err != nil { |
| 746 | return err |
| 747 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 748 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 749 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 750 | if err != nil { |
| 751 | return err |
| 752 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 753 | t := time.Now() |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 754 | pre := fargs[0] |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 755 | ws := newWordScanner(fargs[1]) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 756 | for ws.Scan() { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 757 | var name []byte |
| 758 | name = append(name, pre...) |
| 759 | name = append(name, ws.Bytes()...) |
| 760 | w.writeWord(name) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 761 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 762 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 763 | stats.add("funcbody", "addprefix", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 764 | return err |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 765 | } |
| 766 | |
| 767 | type funcRealpath struct{ fclosure } |
| 768 | |
| 769 | func (f *funcRealpath) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 770 | func (f *funcRealpath) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 771 | err := assertArity("realpath", 1, len(f.args)) |
| 772 | if err != nil { |
| 773 | return err |
| 774 | } |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 775 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 776 | io.WriteString(w, "KATI_TODO(realpath)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 777 | ev.hasIO = true |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 778 | return nil |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 779 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 780 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 781 | err = f.args[1].Eval(abuf, ev) |
| 782 | if err != nil { |
| 783 | return err |
| 784 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 785 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 786 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 787 | for ws.Scan() { |
| 788 | name := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 789 | name, err := filepath.Abs(name) |
| 790 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 791 | logf("abs: %v", err) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 792 | continue |
| 793 | } |
| 794 | name, err = filepath.EvalSymlinks(name) |
| 795 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 796 | logf("realpath: %v", err) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 797 | continue |
| 798 | } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 799 | w.writeWordString(name) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 800 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 801 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 802 | stats.add("funcbody", "realpath", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 803 | return err |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 804 | } |
| 805 | |
| 806 | type funcAbspath struct{ fclosure } |
| 807 | |
| 808 | func (f *funcAbspath) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 809 | func (f *funcAbspath) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 810 | err := assertArity("abspath", 1, len(f.args)) |
| 811 | if err != nil { |
| 812 | return err |
| 813 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 814 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 815 | err = f.args[1].Eval(abuf, ev) |
| 816 | if err != nil { |
| 817 | return err |
| 818 | } |
Fumitoshi Ukai | 9e0c68d | 2015-06-19 13:38:19 +0900 | [diff] [blame] | 819 | t := time.Now() |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 820 | ws := newWordScanner(abuf.Bytes()) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 821 | for ws.Scan() { |
| 822 | name := string(ws.Bytes()) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 823 | name, err := filepath.Abs(name) |
| 824 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 825 | logf("abs: %v", err) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 826 | continue |
| 827 | } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 828 | w.writeWordString(name) |
Shinichiro Hamaji | af61af2 | 2015-04-11 12:13:22 +0900 | [diff] [blame] | 829 | } |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 830 | freeBuf(abuf) |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 831 | stats.add("funcbody", "abspath", t) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 832 | return nil |
Shinichiro Hamaji | c20b84d | 2015-04-11 12:02:51 +0900 | [diff] [blame] | 833 | } |
| 834 | |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 835 | // http://www.gnu.org/software/make/manual/make.html#Conditional-Functions |
| 836 | type funcIf struct{ fclosure } |
| 837 | |
| 838 | func (f *funcIf) Arity() int { return 3 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 839 | func (f *funcIf) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 840 | err := assertArity("if", 2, len(f.args)) |
| 841 | if err != nil { |
| 842 | return err |
| 843 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 844 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 845 | err = f.args[1].Eval(abuf, ev) |
| 846 | if err != nil { |
| 847 | return err |
| 848 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 849 | if len(abuf.Bytes()) != 0 { |
| 850 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 851 | return f.args[2].Eval(w, ev) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 852 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 853 | freeBuf(abuf) |
Fumitoshi Ukai | e27a25d | 2015-04-18 00:31:01 +0900 | [diff] [blame] | 854 | if len(f.args) > 3 { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 855 | return f.args[3].Eval(w, ev) |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 856 | } |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 857 | return nil |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 858 | } |
| 859 | |
| 860 | type funcAnd struct{ fclosure } |
| 861 | |
| 862 | func (f *funcAnd) Arity() int { return 0 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 863 | func (f *funcAnd) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 864 | err := assertArity("and", 0, len(f.args)) |
| 865 | if err != nil { |
| 866 | return nil |
| 867 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 868 | abuf := newBuf() |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 869 | var cond []byte |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 870 | for _, arg := range f.args[1:] { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 871 | abuf.Reset() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 872 | err = arg.Eval(abuf, ev) |
| 873 | if err != nil { |
| 874 | return err |
| 875 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 876 | cond = abuf.Bytes() |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 877 | if len(cond) == 0 { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 878 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 879 | return nil |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 880 | } |
| 881 | } |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 882 | w.Write(cond) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 883 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 884 | return nil |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 885 | } |
| 886 | |
| 887 | type funcOr struct{ fclosure } |
| 888 | |
| 889 | func (f *funcOr) Arity() int { return 0 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 890 | func (f *funcOr) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 891 | err := assertArity("or", 0, len(f.args)) |
| 892 | if err != nil { |
| 893 | return err |
| 894 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 895 | abuf := newBuf() |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 896 | for _, arg := range f.args[1:] { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 897 | abuf.Reset() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 898 | err = arg.Eval(abuf, ev) |
| 899 | if err != nil { |
| 900 | return err |
| 901 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 902 | cond := abuf.Bytes() |
Shinichiro Hamaji | 2216dd6 | 2015-04-11 13:44:39 +0900 | [diff] [blame] | 903 | if len(cond) != 0 { |
| 904 | w.Write(cond) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 905 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 906 | return nil |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 907 | } |
| 908 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 909 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 910 | return nil |
Shinichiro Hamaji | 424baeb | 2015-04-11 12:35:42 +0900 | [diff] [blame] | 911 | } |
| 912 | |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 913 | // http://www.gnu.org/software/make/manual/make.html#Shell-Function |
| 914 | type funcShell struct{ fclosure } |
| 915 | |
| 916 | func (f *funcShell) Arity() int { return 1 } |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 917 | |
Shinichiro Hamaji | c0095e3 | 2015-05-27 18:49:06 +0900 | [diff] [blame] | 918 | // A hack for Android build. We need to evaluate things like $((3+4)) |
| 919 | // when we emit ninja file, because the result of such expressions |
| 920 | // will be passed to other make functions. |
| 921 | // TODO: Maybe we should modify Android's Makefile and remove this |
| 922 | // workaround. It would be also nice if we can detect things like |
| 923 | // this. |
| 924 | func hasNoIoInShellScript(s []byte) bool { |
| 925 | if len(s) == 0 { |
| 926 | return true |
| 927 | } |
| 928 | if !bytes.HasPrefix(s, []byte("echo $((")) || s[len(s)-1] != ')' { |
| 929 | return false |
| 930 | } |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 931 | logf("has no IO - evaluate now: %s", s) |
Shinichiro Hamaji | c0095e3 | 2015-05-27 18:49:06 +0900 | [diff] [blame] | 932 | return true |
| 933 | } |
| 934 | |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 935 | func (f *funcShell) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 936 | err := assertArity("shell", 1, len(f.args)) |
| 937 | if err != nil { |
| 938 | return err |
| 939 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 940 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 941 | err = f.args[1].Eval(abuf, ev) |
| 942 | if err != nil { |
| 943 | return err |
| 944 | } |
Shinichiro Hamaji | c0095e3 | 2015-05-27 18:49:06 +0900 | [diff] [blame] | 945 | if ev.avoidIO && !hasNoIoInShellScript(abuf.Bytes()) { |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 946 | te := traceEvent.begin("shell", tmpval(abuf.Bytes()), traceEventMain) |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 947 | ev.hasIO = true |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 948 | io.WriteString(w, "$(") |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 949 | w.Write(abuf.Bytes()) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 950 | writeByte(w, ')') |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 951 | traceEvent.end(te) |
Fumitoshi Ukai | 6a64323 | 2015-06-08 14:11:23 +0900 | [diff] [blame] | 952 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 953 | return nil |
Shinichiro Hamaji | 8b6487f | 2015-05-26 16:21:47 +0900 | [diff] [blame] | 954 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 955 | arg := abuf.String() |
| 956 | freeBuf(abuf) |
Shinichiro Hamaji | 89551c9 | 2015-04-11 19:33:03 +0900 | [diff] [blame] | 957 | shellVar := ev.LookupVar("SHELL") |
| 958 | // TODO: Should be Eval, not String. |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 959 | cmdline := []string{shellVar.String(), "-c", arg} |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 960 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 961 | logf("shell %q", cmdline) |
Fumitoshi Ukai | 44ae8cf | 2015-06-24 16:44:15 +0900 | [diff] [blame] | 962 | } |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 963 | cmd := exec.Cmd{ |
| 964 | Path: cmdline[0], |
| 965 | Args: cmdline, |
| 966 | Stderr: os.Stderr, |
| 967 | } |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 968 | te := traceEvent.begin("shell", literal(arg), traceEventMain) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 969 | out, err := cmd.Output() |
Fumitoshi Ukai | 9042b99 | 2015-06-23 16:10:27 +0900 | [diff] [blame] | 970 | shellStats.add(time.Since(te.t)) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 971 | if err != nil { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 972 | logf("$(shell %q) failed: %q", arg, err) |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 973 | } |
Shinichiro Hamaji | ba6b84d | 2015-06-03 18:46:13 +0900 | [diff] [blame] | 974 | w.Write(formatCommandOutput(out)) |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 975 | traceEvent.end(te) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 976 | return nil |
Fumitoshi Ukai | b36f387 | 2015-04-10 15:06:38 +0900 | [diff] [blame] | 977 | } |
| 978 | |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 979 | func (f *funcShell) Compact() Value { |
| 980 | if len(f.args)-1 < 1 { |
| 981 | return f |
| 982 | } |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 983 | if !UseFindCache && !UseShellBuiltins { |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 984 | return f |
| 985 | } |
| 986 | |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 987 | var exp expr |
Fumitoshi Ukai | 44ae8cf | 2015-06-24 16:44:15 +0900 | [diff] [blame] | 988 | switch v := f.args[1].(type) { |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 989 | case expr: |
| 990 | exp = v |
Fumitoshi Ukai | 44ae8cf | 2015-06-24 16:44:15 +0900 | [diff] [blame] | 991 | default: |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 992 | exp = expr{v} |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 993 | } |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 994 | if UseShellBuiltins { |
Fumitoshi Ukai | 4a70851 | 2015-06-11 17:15:49 +0900 | [diff] [blame] | 995 | // hack for android |
| 996 | for _, sb := range shBuiltins { |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 997 | if v, ok := matchExpr(exp, sb.pattern); ok { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 998 | logf("shell compact apply %s for %s", sb.name, exp) |
Fumitoshi Ukai | 4a70851 | 2015-06-11 17:15:49 +0900 | [diff] [blame] | 999 | return sb.compact(f, v) |
| 1000 | } |
Fumitoshi Ukai | 76b609e | 2015-06-10 14:56:49 +0900 | [diff] [blame] | 1001 | } |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1002 | logf("shell compact no match: %s", exp) |
Fumitoshi Ukai | 76b609e | 2015-06-10 14:56:49 +0900 | [diff] [blame] | 1003 | } |
Fumitoshi Ukai | 106fb79 | 2015-06-09 10:37:35 +0900 | [diff] [blame] | 1004 | return f |
| 1005 | } |
| 1006 | |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1007 | // https://www.gnu.org/software/make/manual/html_node/Call-Function.html#Call-Function |
| 1008 | type funcCall struct{ fclosure } |
| 1009 | |
| 1010 | func (f *funcCall) Arity() int { return 0 } |
| 1011 | |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1012 | func (f *funcCall) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1013 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1014 | fargs, err := ev.args(abuf, f.args[1:]...) |
| 1015 | if err != nil { |
| 1016 | return err |
| 1017 | } |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 1018 | varname := fargs[0] |
| 1019 | variable := string(varname) |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 1020 | te := traceEvent.begin("call", literal(variable), traceEventMain) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 1021 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1022 | logf("call %q variable %q", f.args[1], variable) |
Fumitoshi Ukai | 5c04ad8 | 2015-06-18 16:14:04 +0900 | [diff] [blame] | 1023 | } |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 1024 | v := ev.LookupVar(variable) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1025 | // Evalualte all arguments first before we modify the table. |
Shinichiro Hamaji | 39728f1 | 2015-04-11 20:12:23 +0900 | [diff] [blame] | 1026 | var args []tmpval |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 1027 | // $0 is variable. |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 1028 | args = append(args, tmpval(varname)) |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 1029 | // TODO(ukai): If variable is the name of a built-in function, |
| 1030 | // the built-in function is always invoked (even if a make variable |
| 1031 | // by that name also exists). |
| 1032 | |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1033 | for i, arg := range fargs[1:] { |
| 1034 | // f.args[2]=>args[1] will be $1. |
| 1035 | args = append(args, tmpval(arg)) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 1036 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1037 | logf("call $%d: %q=>%q", i+1, arg, fargs[i+1]) |
Fumitoshi Ukai | 5c04ad8 | 2015-06-18 16:14:04 +0900 | [diff] [blame] | 1038 | } |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1039 | } |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 1040 | oldParams := ev.paramVars |
| 1041 | ev.paramVars = args |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1042 | |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1043 | var buf bytes.Buffer |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 1044 | if LogFlag { |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1045 | w = &ssvWriter{Writer: io.MultiWriter(w, &buf)} |
Fumitoshi Ukai | e27a25d | 2015-04-18 00:31:01 +0900 | [diff] [blame] | 1046 | } |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1047 | err = v.Eval(w, ev) |
| 1048 | if err != nil { |
| 1049 | return err |
| 1050 | } |
Fumitoshi Ukai | f0a2ba7 | 2015-04-19 00:02:32 +0900 | [diff] [blame] | 1051 | ev.paramVars = oldParams |
Fumitoshi Ukai | 72598e7 | 2015-06-11 15:53:09 +0900 | [diff] [blame] | 1052 | traceEvent.end(te) |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 1053 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1054 | 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] | 1055 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1056 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1057 | return nil |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1058 | } |
| 1059 | |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 1060 | // http://www.gnu.org/software/make/manual/make.html#Value-Function |
| 1061 | type funcValue struct{ fclosure } |
| 1062 | |
| 1063 | func (f *funcValue) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1064 | func (f *funcValue) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1065 | err := assertArity("value", 1, len(f.args)) |
| 1066 | if err != nil { |
| 1067 | return err |
| 1068 | } |
Fumitoshi Ukai | bbb0db5 | 2015-06-30 16:51:27 +0900 | [diff] [blame] | 1069 | abuf := newBuf() |
| 1070 | err = f.args[1].Eval(abuf, ev) |
| 1071 | if err != nil { |
| 1072 | return err |
| 1073 | } |
| 1074 | v := ev.LookupVar(abuf.String()) |
| 1075 | freeBuf(abuf) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1076 | io.WriteString(w, v.String()) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1077 | return nil |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 1078 | } |
| 1079 | |
| 1080 | // http://www.gnu.org/software/make/manual/make.html#Eval-Function |
| 1081 | type funcEval struct{ fclosure } |
| 1082 | |
| 1083 | func (f *funcEval) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1084 | func (f *funcEval) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1085 | err := assertArity("eval", 1, len(f.args)) |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 1086 | if err != nil { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1087 | return err |
| 1088 | } |
| 1089 | abuf := newBuf() |
| 1090 | err = f.args[1].Eval(abuf, ev) |
| 1091 | if err != nil { |
| 1092 | return err |
| 1093 | } |
| 1094 | s := abuf.Bytes() |
| 1095 | logf("eval %q at %s", s, ev.srcpos) |
| 1096 | mk, err := parseMakefileBytes(s, ev.srcpos) |
| 1097 | if err != nil { |
| 1098 | return ev.errorf("%v", err) |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 1099 | } |
| 1100 | |
| 1101 | for _, stmt := range mk.stmts { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1102 | err = ev.eval(stmt) |
| 1103 | if err != nil { |
| 1104 | return err |
| 1105 | } |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 1106 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1107 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1108 | return nil |
Shinichiro Hamaji | 28ea5bc | 2015-04-11 12:41:08 +0900 | [diff] [blame] | 1109 | } |
| 1110 | |
Fumitoshi Ukai | 9f6b635 | 2015-04-16 16:25:09 +0900 | [diff] [blame] | 1111 | func (f *funcEval) Compact() Value { |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1112 | if len(f.args)-1 < 1 { |
| 1113 | return f |
| 1114 | } |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1115 | switch arg := f.args[1].(type) { |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1116 | case literal, tmpval: |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 1117 | case expr: |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1118 | if len(arg) == 1 { |
| 1119 | return f |
| 1120 | } |
| 1121 | switch prefix := arg[0].(type) { |
| 1122 | case literal, tmpval: |
| 1123 | lhs, op, rhsprefix, ok := parseAssignLiteral(prefix.String()) |
| 1124 | if ok { |
| 1125 | // $(eval foo = $(bar)) |
Fumitoshi Ukai | 55c8fa9 | 2015-06-25 15:56:10 +0900 | [diff] [blame] | 1126 | var rhs expr |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1127 | if rhsprefix != literal("") { |
| 1128 | rhs = append(rhs, rhsprefix) |
| 1129 | } |
| 1130 | rhs = append(rhs, arg[1:]...) |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1131 | 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] | 1132 | return &funcEvalAssign{ |
| 1133 | lhs: lhs, |
| 1134 | op: op, |
| 1135 | rhs: compactExpr(rhs), |
| 1136 | } |
| 1137 | } |
| 1138 | } |
| 1139 | // TODO(ukai): eval -> varassign. e.g $(eval $(foo) := $(x)). |
Fumitoshi Ukai | 769157a | 2015-06-02 15:18:53 +0900 | [diff] [blame] | 1140 | return f |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1141 | default: |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1142 | return f |
| 1143 | } |
| 1144 | arg := f.args[1].String() |
| 1145 | arg = stripComment(arg) |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1146 | if arg == "" || strings.TrimSpace(arg) == "" { |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1147 | return &funcNop{expr: f.String()} |
| 1148 | } |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1149 | f.args[1] = literal(arg) |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1150 | lhs, op, rhs, ok := parseAssignLiteral(f.args[1].String()) |
| 1151 | if ok { |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1152 | return &funcEvalAssign{ |
| 1153 | lhs: lhs, |
| 1154 | op: op, |
| 1155 | rhs: rhs, |
| 1156 | } |
| 1157 | } |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1158 | return f |
| 1159 | } |
| 1160 | |
| 1161 | func stripComment(arg string) string { |
| 1162 | for { |
| 1163 | i := strings.Index(arg, "#") |
| 1164 | if i < 0 { |
| 1165 | return arg |
| 1166 | } |
| 1167 | eol := strings.Index(arg[i:], "\n") |
| 1168 | if eol < 0 { |
| 1169 | return arg[:i] |
| 1170 | } |
| 1171 | arg = arg[:i] + arg[eol+1:] |
| 1172 | } |
| 1173 | } |
| 1174 | |
| 1175 | type funcNop struct{ expr string } |
| 1176 | |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1177 | func (f *funcNop) String() string { return f.expr } |
| 1178 | func (f *funcNop) Eval(evalWriter, *Evaluator) error { return nil } |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 1179 | func (f *funcNop) serialize() serializableVar { |
| 1180 | return serializableVar{ |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 1181 | Type: "funcNop", |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1182 | V: f.expr, |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 1183 | } |
| 1184 | } |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1185 | func (f *funcNop) dump(d *dumpbuf) { |
| 1186 | d.Byte(valueTypeNop) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 1187 | } |
Fumitoshi Ukai | da7f255 | 2015-04-16 13:33:37 +0900 | [diff] [blame] | 1188 | |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1189 | func parseAssignLiteral(s string) (lhs, op string, rhs Value, ok bool) { |
| 1190 | eq := strings.Index(s, "=") |
| 1191 | if eq < 0 { |
| 1192 | return "", "", nil, false |
| 1193 | } |
| 1194 | // TODO(ukai): factor out parse assign? |
| 1195 | lhs = s[:eq] |
| 1196 | op = s[eq : eq+1] |
| 1197 | if eq >= 1 && (s[eq-1] == ':' || s[eq-1] == '+' || s[eq-1] == '?') { |
| 1198 | lhs = s[:eq-1] |
| 1199 | op = s[eq-1 : eq+1] |
| 1200 | } |
| 1201 | lhs = strings.TrimSpace(lhs) |
| 1202 | if strings.IndexAny(lhs, ":$") >= 0 { |
| 1203 | // target specific var, or need eval. |
| 1204 | return "", "", nil, false |
| 1205 | } |
| 1206 | r := strings.TrimLeft(s[eq+1:], " \t") |
| 1207 | rhs = literal(r) |
| 1208 | return lhs, op, rhs, true |
| 1209 | } |
| 1210 | |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1211 | type funcEvalAssign struct { |
| 1212 | lhs string |
| 1213 | op string |
| 1214 | rhs Value |
| 1215 | } |
| 1216 | |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1217 | func (f *funcEvalAssign) String() string { |
| 1218 | return fmt.Sprintf("$(eval %s %s %s)", f.lhs, f.op, f.rhs) |
| 1219 | } |
| 1220 | |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1221 | func (f *funcEvalAssign) Eval(w evalWriter, ev *Evaluator) error { |
| 1222 | var abuf buffer |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1223 | err := f.rhs.Eval(&abuf, ev) |
| 1224 | if err != nil { |
| 1225 | return err |
| 1226 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1227 | rhs := trimLeftSpaceBytes(abuf.Bytes()) |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1228 | var rvalue Var |
| 1229 | switch f.op { |
| 1230 | case ":=": |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1231 | // TODO(ukai): compute parsed expr in Compact when f.rhs is |
| 1232 | // literal? e.g. literal("$(foo)") => varref{literal("foo")}. |
Fumitoshi Ukai | e9aa380 | 2015-07-03 11:33:23 +0900 | [diff] [blame] | 1233 | exp, _, err := parseExpr(rhs, nil, parseOp{}) |
Fumitoshi Ukai | fa5e922 | 2015-04-17 11:45:20 +0900 | [diff] [blame] | 1234 | if err != nil { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1235 | return ev.errorf("eval assign error: %q: %v", f.String(), err) |
Fumitoshi Ukai | fa5e922 | 2015-04-17 11:45:20 +0900 | [diff] [blame] | 1236 | } |
Fumitoshi Ukai | dd248f2 | 2015-06-18 23:04:28 +0900 | [diff] [blame] | 1237 | vbuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1238 | err = exp.Eval(vbuf, ev) |
| 1239 | if err != nil { |
| 1240 | return err |
| 1241 | } |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1242 | rvalue = &simpleVar{value: vbuf.String(), origin: "file"} |
Fumitoshi Ukai | dd248f2 | 2015-06-18 23:04:28 +0900 | [diff] [blame] | 1243 | freeBuf(vbuf) |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1244 | case "=": |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1245 | rvalue = &recursiveVar{expr: tmpval(rhs), origin: "file"} |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1246 | case "+=": |
| 1247 | prev := ev.LookupVar(f.lhs) |
Fumitoshi Ukai | d8cec17 | 2015-04-28 00:08:09 +0900 | [diff] [blame] | 1248 | if prev.IsDefined() { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1249 | rvalue, err = prev.Append(ev, string(rhs)) |
| 1250 | if err != nil { |
| 1251 | return err |
| 1252 | } |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1253 | } else { |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1254 | rvalue = &recursiveVar{expr: tmpval(rhs), origin: "file"} |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1255 | } |
| 1256 | case "?=": |
| 1257 | prev := ev.LookupVar(f.lhs) |
| 1258 | if prev.IsDefined() { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1259 | return nil |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1260 | } |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1261 | rvalue = &recursiveVar{expr: tmpval(rhs), origin: "file"} |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1262 | } |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 1263 | if LogFlag { |
Fumitoshi Ukai | 07cf121 | 2015-06-25 17:16:25 +0900 | [diff] [blame] | 1264 | logf("Eval ASSIGN: %s=%q (flavor:%q)", f.lhs, rvalue, rvalue.Flavor()) |
Fumitoshi Ukai | 5c04ad8 | 2015-06-18 16:14:04 +0900 | [diff] [blame] | 1265 | } |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1266 | ev.outVars.Assign(f.lhs, rvalue) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1267 | return nil |
Fumitoshi Ukai | 150c861 | 2015-04-16 15:35:44 +0900 | [diff] [blame] | 1268 | } |
| 1269 | |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 1270 | func (f *funcEvalAssign) serialize() serializableVar { |
| 1271 | return serializableVar{ |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 1272 | Type: "funcEvalAssign", |
Fumitoshi Ukai | a045ccb | 2015-06-25 12:57:25 +0900 | [diff] [blame] | 1273 | Children: []serializableVar{ |
| 1274 | serializableVar{V: f.lhs}, |
| 1275 | serializableVar{V: f.op}, |
| 1276 | f.rhs.serialize(), |
Shinichiro Hamaji | c8bc731 | 2015-04-28 02:48:03 +0900 | [diff] [blame] | 1277 | }, |
| 1278 | } |
| 1279 | } |
| 1280 | |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1281 | func (f *funcEvalAssign) dump(d *dumpbuf) { |
| 1282 | d.Byte(valueTypeAssign) |
| 1283 | d.Str(f.lhs) |
| 1284 | d.Str(f.op) |
| 1285 | f.rhs.dump(d) |
Shinichiro Hamaji | 723f56a | 2015-05-15 17:12:55 +0900 | [diff] [blame] | 1286 | } |
| 1287 | |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1288 | // http://www.gnu.org/software/make/manual/make.html#Origin-Function |
| 1289 | type funcOrigin struct{ fclosure } |
| 1290 | |
| 1291 | func (f *funcOrigin) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1292 | func (f *funcOrigin) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1293 | err := assertArity("origin", 1, len(f.args)) |
| 1294 | if err != nil { |
| 1295 | return err |
| 1296 | } |
Fumitoshi Ukai | b64400e | 2015-06-30 16:06:09 +0900 | [diff] [blame] | 1297 | abuf := newBuf() |
| 1298 | err = f.args[1].Eval(abuf, ev) |
| 1299 | if err != nil { |
| 1300 | return err |
| 1301 | } |
| 1302 | v := ev.LookupVar(abuf.String()) |
| 1303 | freeBuf(abuf) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1304 | io.WriteString(w, v.Origin()) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1305 | return nil |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1306 | } |
| 1307 | |
| 1308 | // https://www.gnu.org/software/make/manual/html_node/Flavor-Function.html#Flavor-Function |
| 1309 | type funcFlavor struct{ fclosure } |
| 1310 | |
| 1311 | func (f *funcFlavor) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1312 | func (f *funcFlavor) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1313 | err := assertArity("flavor", 1, len(f.args)) |
| 1314 | if err != nil { |
| 1315 | return err |
| 1316 | } |
Fumitoshi Ukai | b64400e | 2015-06-30 16:06:09 +0900 | [diff] [blame] | 1317 | abuf := newBuf() |
| 1318 | err = f.args[1].Eval(abuf, ev) |
| 1319 | if err != nil { |
| 1320 | return err |
| 1321 | } |
| 1322 | v := ev.LookupVar(abuf.String()) |
| 1323 | freeBuf(abuf) |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1324 | io.WriteString(w, v.Flavor()) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1325 | return nil |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1326 | } |
| 1327 | |
| 1328 | // http://www.gnu.org/software/make/manual/make.html#Make-Control-Functions |
| 1329 | type funcInfo struct{ fclosure } |
| 1330 | |
| 1331 | func (f *funcInfo) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1332 | func (f *funcInfo) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1333 | err := assertArity("info", 1, len(f.args)) |
| 1334 | if err != nil { |
| 1335 | return err |
| 1336 | } |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1337 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1338 | io.WriteString(w, "KATI_TODO(info)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1339 | ev.hasIO = true |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1340 | return nil |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1341 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1342 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1343 | err = f.args[1].Eval(abuf, ev) |
| 1344 | if err != nil { |
| 1345 | return err |
| 1346 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1347 | fmt.Printf("%s\n", abuf.String()) |
| 1348 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1349 | return nil |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1350 | } |
| 1351 | |
| 1352 | type funcWarning struct{ fclosure } |
| 1353 | |
| 1354 | func (f *funcWarning) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1355 | func (f *funcWarning) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1356 | err := assertArity("warning", 1, len(f.args)) |
| 1357 | if err != nil { |
| 1358 | return err |
| 1359 | } |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1360 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1361 | io.WriteString(w, "KATI_TODO(warning)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1362 | ev.hasIO = true |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1363 | return nil |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1364 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1365 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1366 | err = f.args[1].Eval(abuf, ev) |
| 1367 | if err != nil { |
| 1368 | return err |
| 1369 | } |
| 1370 | fmt.Printf("%s: %s\n", ev.srcpos, abuf.String()) |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1371 | freeBuf(abuf) |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1372 | return nil |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1373 | } |
| 1374 | |
| 1375 | type funcError struct{ fclosure } |
| 1376 | |
| 1377 | func (f *funcError) Arity() int { return 1 } |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1378 | func (f *funcError) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1379 | err := assertArity("error", 1, len(f.args)) |
| 1380 | if err != nil { |
| 1381 | return err |
| 1382 | } |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1383 | if ev.avoidIO { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1384 | io.WriteString(w, "KATI_TODO(error)") |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1385 | ev.hasIO = true |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1386 | return nil |
Shinichiro Hamaji | b41fd50 | 2015-04-29 03:34:07 +0900 | [diff] [blame] | 1387 | } |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1388 | var abuf buffer |
| 1389 | err = f.args[1].Eval(&abuf, ev) |
| 1390 | if err != nil { |
| 1391 | return err |
| 1392 | } |
| 1393 | return ev.errorf("*** %s.", abuf.String()) |
Shinichiro Hamaji | 916f35f | 2015-04-11 12:46:19 +0900 | [diff] [blame] | 1394 | } |
| 1395 | |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1396 | // http://www.gnu.org/software/make/manual/make.html#Foreach-Function |
| 1397 | type funcForeach struct{ fclosure } |
| 1398 | |
| 1399 | func (f *funcForeach) Arity() int { return 3 } |
| 1400 | |
Fumitoshi Ukai | b44b12d | 2015-07-07 14:19:32 +0900 | [diff] [blame^] | 1401 | func (f *funcForeach) Eval(w evalWriter, ev *Evaluator) error { |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1402 | err := assertArity("foreach", 3, len(f.args)) |
| 1403 | if err != nil { |
| 1404 | return err |
| 1405 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1406 | abuf := newBuf() |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1407 | fargs, err := ev.args(abuf, f.args[1], f.args[2]) |
| 1408 | if err != nil { |
| 1409 | return err |
| 1410 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1411 | varname := string(fargs[0]) |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 1412 | ws := newWordScanner(fargs[1]) |
Fumitoshi Ukai | b2670d9 | 2015-04-16 10:28:27 +0900 | [diff] [blame] | 1413 | text := f.args[3] |
Fumitoshi Ukai | 2142617 | 2015-06-30 18:02:45 +0900 | [diff] [blame] | 1414 | ov := ev.LookupVar(varname) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1415 | space := false |
Fumitoshi Ukai | 9da19f6 | 2015-05-08 14:39:08 +0900 | [diff] [blame] | 1416 | for ws.Scan() { |
| 1417 | word := ws.Bytes() |
Fumitoshi Ukai | 7bf992d | 2015-06-25 12:42:19 +0900 | [diff] [blame] | 1418 | ev.outVars.Assign(varname, &automaticVar{value: word}) |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1419 | if space { |
Fumitoshi Ukai | 145598a | 2015-06-19 10:08:17 +0900 | [diff] [blame] | 1420 | writeByte(w, ' ') |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1421 | } |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1422 | err = text.Eval(w, ev) |
| 1423 | if err != nil { |
| 1424 | return err |
| 1425 | } |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1426 | space = true |
| 1427 | } |
Fumitoshi Ukai | b06cd9d | 2015-05-07 12:56:12 +0900 | [diff] [blame] | 1428 | freeBuf(abuf) |
Fumitoshi Ukai | 2142617 | 2015-06-30 18:02:45 +0900 | [diff] [blame] | 1429 | av := ev.LookupVar(varname) |
| 1430 | if _, ok := av.(*automaticVar); ok { |
| 1431 | ev.outVars.Assign(varname, ov) |
| 1432 | } |
Fumitoshi Ukai | 65c7233 | 2015-06-26 21:32:50 +0900 | [diff] [blame] | 1433 | return nil |
Fumitoshi Ukai | d2bcf66 | 2015-04-11 01:02:27 +0900 | [diff] [blame] | 1434 | } |