Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 1 | package main |
| 2 | |
| 3 | import ( |
| 4 | "bufio" |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 5 | "bytes" |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 6 | "errors" |
Fumitoshi Ukai | f8efa0a | 2015-03-30 18:10:11 +0900 | [diff] [blame] | 7 | "fmt" |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 8 | "io" |
| 9 | "os" |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame^] | 10 | "strings" |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 11 | ) |
| 12 | |
| 13 | type Makefile struct { |
| 14 | stmts []AST |
| 15 | } |
| 16 | |
| 17 | type parser struct { |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 18 | rd *bufio.Reader |
| 19 | mk Makefile |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 20 | filename string |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 21 | lineno int |
| 22 | elineno int // lineno == elineno unless there is trailing '\'. |
| 23 | unBuf []byte |
| 24 | hasUnBuf bool |
| 25 | done bool |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 26 | } |
| 27 | |
| 28 | func exists(filename string) bool { |
| 29 | f, err := os.Open(filename) |
| 30 | if err != nil { |
| 31 | return false |
| 32 | } |
| 33 | f.Close() |
| 34 | return true |
| 35 | } |
| 36 | |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 37 | func newParser(rd io.Reader, filename string) *parser { |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 38 | return &parser{ |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 39 | rd: bufio.NewReader(rd), |
| 40 | filename: filename, |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 41 | } |
| 42 | } |
| 43 | |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 44 | func (p *parser) readLine() []byte { |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 45 | if p.hasUnBuf { |
| 46 | p.hasUnBuf = false |
| 47 | return p.unBuf |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 48 | } |
| 49 | |
| 50 | p.lineno = p.elineno |
| 51 | line, err := p.rd.ReadBytes('\n') |
| 52 | p.lineno++ |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 53 | p.elineno = p.lineno |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 54 | if err == io.EOF { |
| 55 | p.done = true |
| 56 | } else if err != nil { |
| 57 | panic(err) |
| 58 | } |
| 59 | |
| 60 | if len(line) > 0 { |
| 61 | line = line[0 : len(line)-1] |
| 62 | } |
| 63 | |
| 64 | // TODO: Handle \\ at the end of the line? |
| 65 | for len(line) > 0 && line[len(line)-1] == '\\' { |
| 66 | line = line[:len(line)-1] |
| 67 | nline := p.readLine() |
| 68 | p.elineno++ |
| 69 | line = append(line, nline...) |
| 70 | } |
| 71 | |
| 72 | index := bytes.IndexByte(line, '#') |
| 73 | if index >= 0 { |
| 74 | line = line[:index] |
| 75 | } |
| 76 | |
| 77 | return line |
| 78 | } |
| 79 | |
| 80 | func (p *parser) unreadLine(line []byte) { |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 81 | if p.hasUnBuf { |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 82 | panic("unreadLine twice!") |
| 83 | } |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 84 | p.unBuf = line |
| 85 | p.hasUnBuf = true |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 86 | } |
| 87 | |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 88 | func (p *parser) parseAssign(line []byte, sep, esep int) AST { |
| 89 | Log("parseAssign %s %s", line, line[sep:esep]) |
| 90 | ast := &AssignAST{ |
| 91 | lhs: string(bytes.TrimSpace(line[:sep])), |
| 92 | rhs: string(bytes.TrimLeft(line[esep:], " \t")), |
| 93 | op: string(line[sep:esep]), |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 94 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 95 | ast.filename = p.filename |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 96 | ast.lineno = p.lineno |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 97 | return ast |
| 98 | } |
| 99 | |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 100 | func (p *parser) parseRule(line []byte, sep int) AST { |
| 101 | lhs := string(bytes.TrimSpace(line[:sep])) |
| 102 | rhs := string(bytes.TrimSpace(line[sep+1:])) |
| 103 | ast := &RuleAST{ |
| 104 | lhs: lhs, |
| 105 | rhs: rhs, |
| 106 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 107 | ast.filename = p.filename |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 108 | ast.lineno = p.lineno |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 109 | for { |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 110 | line := p.readLine() |
| 111 | if len(line) == 0 { |
| 112 | break |
| 113 | } else if line[0] == '\t' { |
| 114 | ast.cmds = append(ast.cmds, string(bytes.TrimSpace(line))) |
| 115 | } else { |
| 116 | p.unreadLine(line) |
| 117 | break |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 118 | } |
| 119 | } |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 120 | return ast |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 121 | } |
| 122 | |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame^] | 123 | func (p *parser) parseInclude(line string, oplen int) AST { |
| 124 | ast := &IncludeAST{ |
| 125 | expr: line[oplen+1:], |
| 126 | op: line[:oplen], |
| 127 | } |
| 128 | ast.filename = p.filename |
| 129 | ast.lineno = p.lineno |
| 130 | return ast |
| 131 | } |
| 132 | |
| 133 | func (p *parser) parseLine(line string) AST { |
| 134 | if strings.HasPrefix(line, "include ") { |
| 135 | return p.parseInclude(line, len("include")) |
| 136 | } |
| 137 | if strings.HasPrefix(line, "-include ") { |
| 138 | return p.parseInclude(line, len("-include")) |
| 139 | } |
| 140 | ast := &RawExprAST{expr: line} |
| 141 | ast.filename = p.filename |
| 142 | ast.lineno = p.lineno |
| 143 | return ast |
| 144 | } |
| 145 | |
Fumitoshi Ukai | f8efa0a | 2015-03-30 18:10:11 +0900 | [diff] [blame] | 146 | func (p *parser) parse() (mk Makefile, err error) { |
| 147 | defer func() { |
| 148 | if r := recover(); r != nil { |
| 149 | err = fmt.Errorf("panic: %v", r) |
| 150 | } |
| 151 | }() |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 152 | for !p.done { |
| 153 | line := p.readLine() |
| 154 | |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 155 | var ast AST |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 156 | for i, ch := range line { |
| 157 | switch ch { |
| 158 | case ':': |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 159 | if i+1 < len(line) && line[i+1] == '=' { |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 160 | ast = p.parseAssign(line, i, i+2) |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 161 | } else { |
| 162 | ast = p.parseRule(line, i) |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 163 | } |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 164 | case '=': |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 165 | ast = p.parseAssign(line, i, i+1) |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 166 | case '?': |
| 167 | panic("TODO") |
| 168 | } |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 169 | if ast != nil { |
| 170 | p.mk.stmts = append(p.mk.stmts, ast) |
| 171 | break |
| 172 | } |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 173 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 174 | if ast == nil && len(bytes.TrimSpace(line)) > 0 { |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame^] | 175 | ast = p.parseLine(string(line)) |
| 176 | p.mk.stmts = append(p.mk.stmts, ast) |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 177 | } |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 178 | } |
| 179 | return p.mk, nil |
| 180 | } |
| 181 | |
| 182 | func ParseMakefile(filename string) (Makefile, error) { |
| 183 | f, err := os.Open(filename) |
| 184 | if err != nil { |
| 185 | return Makefile{}, err |
| 186 | } |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 187 | defer f.Close() |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 188 | parser := newParser(f, filename) |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 189 | return parser.parse() |
| 190 | } |
| 191 | |
| 192 | func ParseDefaultMakefile() (Makefile, error) { |
| 193 | candidates := []string{"GNUmakefile", "makefile", "Makefile"} |
| 194 | for _, filename := range candidates { |
| 195 | if exists(filename) { |
| 196 | return ParseMakefile(filename) |
| 197 | } |
| 198 | } |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 199 | return Makefile{}, errors.New("no targets specified and no makefile found.") |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 200 | } |