| ... | ... | @@ -129,7 +129,6 @@ static Buf *fetch_file(FILE *f) { |
| 129 | 129 | |
| 130 | 130 | |
| 131 | 131 | enum TokenId { |
| 132 | | TokenIdDirective, |
| 133 | 132 | TokenIdSymbol, |
| 134 | 133 | TokenIdLParen, |
| 135 | 134 | TokenIdRParen, |
| ... | ... | @@ -141,6 +140,9 @@ enum TokenId { |
| 141 | 140 | TokenIdSemicolon, |
| 142 | 141 | TokenIdNumberLiteral, |
| 143 | 142 | TokenIdPlus, |
| 143 | TokenIdColon, |
| 144 | TokenIdArrow, |
| 145 | TokenIdDash, |
| 144 | 146 | }; |
| 145 | 147 | |
| 146 | 148 | struct Token { |
| ... | ... | @@ -153,14 +155,10 @@ struct Token { |
| 153 | 155 | |
| 154 | 156 | enum TokenizeState { |
| 155 | 157 | TokenizeStateStart, |
| 156 | | TokenizeStateDirective, |
| 157 | | TokenizeStateDirectiveName, |
| 158 | | TokenizeStateIncludeQuote, |
| 159 | | TokenizeStateDirectiveEnd, |
| 160 | | TokenizeStateInclude, |
| 161 | 158 | TokenizeStateSymbol, |
| 162 | | TokenizeStateString, |
| 163 | 159 | TokenizeStateNumber, |
| 160 | TokenizeStateString, |
| 161 | TokenizeStateSawDash, |
| 164 | 162 | }; |
| 165 | 163 | |
| 166 | 164 | struct Tokenize { |
| ... | ... | @@ -171,11 +169,7 @@ struct Tokenize { |
| 171 | 169 | int line; |
| 172 | 170 | int column; |
| 173 | 171 | Token *cur_tok; |
| 174 | | Buf *directive_name; |
| 175 | 172 | Buf *cur_dir_path; |
| 176 | | uint8_t unquote_char; |
| 177 | | int quote_start_pos; |
| 178 | | Buf *include_path; |
| 179 | 173 | ZigList<char *> *include_paths; |
| 180 | 174 | }; |
| 181 | 175 | |
| ... | ... | @@ -217,68 +211,6 @@ static void end_token(Tokenize *t) { |
| 217 | 211 | t->cur_tok = nullptr; |
| 218 | 212 | } |
| 219 | 213 | |
| 220 | | static void put_back(Tokenize *t, int count) { |
| 221 | | t->pos -= count; |
| 222 | | } |
| 223 | | |
| 224 | | static void begin_directive(Tokenize *t) { |
| 225 | | t->state = TokenizeStateDirective; |
| 226 | | begin_token(t, TokenIdDirective); |
| 227 | | assert(!t->directive_name); |
| 228 | | t->directive_name = buf_alloc(); |
| 229 | | } |
| 230 | | |
| 231 | | static bool find_and_include_file(Tokenize *t, char *dir_path, char *file_path) { |
| 232 | | Buf *full_path = buf_sprintf("%s/%s", dir_path, file_path); |
| 233 | | |
| 234 | | FILE *f = fopen(buf_ptr(full_path), "rb"); |
| 235 | | if (!f) |
| 236 | | return false; |
| 237 | | |
| 238 | | Buf *contents = fetch_file(f); |
| 239 | | |
| 240 | | buf_splice_buf(t->buf, t->pos, t->pos, contents); |
| 241 | | |
| 242 | | return true; |
| 243 | | } |
| 244 | | |
| 245 | | static void render_include(Tokenize *t, Buf *target_path, char unquote_char) { |
| 246 | | if (unquote_char == '"') { |
| 247 | | if (find_and_include_file(t, buf_ptr(t->cur_dir_path), buf_ptr(target_path))) |
| 248 | | return; |
| 249 | | } |
| 250 | | for (int i = 0; i < t->include_paths->length; i += 1) { |
| 251 | | char *include_path = t->include_paths->at(i); |
| 252 | | if (find_and_include_file(t, include_path, buf_ptr(target_path))) |
| 253 | | return; |
| 254 | | } |
| 255 | | tokenize_error(t, "include path \"%s\" not found", buf_ptr(target_path)); |
| 256 | | } |
| 257 | | |
| 258 | | static void end_directive(Tokenize *t) { |
| 259 | | end_token(t); |
| 260 | | if (t->include_path) { |
| 261 | | render_include(t, t->include_path, t->unquote_char); |
| 262 | | t->include_path = nullptr; |
| 263 | | } |
| 264 | | t->state = TokenizeStateStart; |
| 265 | | } |
| 266 | | |
| 267 | | static void end_directive_name(Tokenize *t) { |
| 268 | | if (buf_eql_str(t->directive_name, "include")) { |
| 269 | | t->state = TokenizeStateInclude; |
| 270 | | t->directive_name = nullptr; |
| 271 | | } else { |
| 272 | | tokenize_error(t, "invalid directive name: \"%s\"", buf_ptr(t->directive_name)); |
| 273 | | } |
| 274 | | } |
| 275 | | |
| 276 | | static void end_symbol(Tokenize *t) { |
| 277 | | put_back(t, 1); |
| 278 | | end_token(t); |
| 279 | | t->state = TokenizeStateStart; |
| 280 | | } |
| 281 | | |
| 282 | 214 | static ZigList<Token> *tokenize(Buf *buf, ZigList<char *> *include_paths, Buf *cur_dir_path) { |
| 283 | 215 | Tokenize t = {0}; |
| 284 | 216 | t.tokens = allocate<ZigList<Token>>(1); |
| ... | ... | @@ -300,9 +232,6 @@ static ZigList<Token> *tokenize(Buf *buf, ZigList<char *> *include_paths, Buf *c |
| 300 | 232 | t.state = TokenizeStateNumber; |
| 301 | 233 | begin_token(&t, TokenIdNumberLiteral); |
| 302 | 234 | break; |
| 303 | | case '#': |
| 304 | | begin_directive(&t); |
| 305 | | break; |
| 306 | 235 | case '"': |
| 307 | 236 | begin_token(&t, TokenIdStringLiteral); |
| 308 | 237 | t.state = TokenizeStateString; |
| ... | ... | @@ -335,79 +264,20 @@ static ZigList<Token> *tokenize(Buf *buf, ZigList<char *> *include_paths, Buf *c |
| 335 | 264 | begin_token(&t, TokenIdSemicolon); |
| 336 | 265 | end_token(&t); |
| 337 | 266 | break; |
| 267 | case ':': |
| 268 | begin_token(&t, TokenIdColon); |
| 269 | end_token(&t); |
| 270 | break; |
| 338 | 271 | case '+': |
| 339 | 272 | begin_token(&t, TokenIdPlus); |
| 340 | 273 | end_token(&t); |
| 341 | 274 | break; |
| 342 | | default: |
| 343 | | tokenize_error(&t, "invalid character: '%c'", c); |
| 344 | | } |
| 345 | | break; |
| 346 | | case TokenizeStateDirective: |
| 347 | | switch (c) { |
| 348 | | case '\n': |
| 349 | | end_directive_name(&t); |
| 350 | | end_directive(&t); |
| 351 | | break; |
| 352 | | case ' ': |
| 353 | | case '\t': |
| 354 | | case '\f': |
| 355 | | case '\r': |
| 356 | | case 0xb: |
| 357 | | break; |
| 358 | | case SYMBOL_CHAR: |
| 359 | | t.state = TokenizeStateDirectiveName; |
| 360 | | buf_append_char(t.directive_name, c); |
| 361 | | break; |
| 362 | | default: |
| 363 | | tokenize_error(&t, "invalid directive character: '%c'", c); |
| 364 | | break; |
| 365 | | } |
| 366 | | break; |
| 367 | | case TokenizeStateDirectiveName: |
| 368 | | switch (c) { |
| 369 | | case WHITESPACE: |
| 370 | | end_directive_name(&t); |
| 371 | | break; |
| 372 | | case SYMBOL_CHAR: |
| 373 | | buf_append_char(t.directive_name, c); |
| 275 | case '-': |
| 276 | begin_token(&t, TokenIdDash); |
| 277 | t.state = TokenizeStateSawDash; |
| 374 | 278 | break; |
| 375 | 279 | default: |
| 376 | | tokenize_error(&t, "invalid directive name character: '%c'", c); |
| 377 | | break; |
| 378 | | } |
| 379 | | break; |
| 380 | | case TokenizeStateInclude: |
| 381 | | switch (c) { |
| 382 | | case WHITESPACE: |
| 383 | | break; |
| 384 | | case '<': |
| 385 | | case '"': |
| 386 | | t.state = TokenizeStateIncludeQuote; |
| 387 | | t.quote_start_pos = t.pos; |
| 388 | | t.unquote_char = (c == '<') ? '>' : '"'; |
| 389 | | break; |
| 390 | | } |
| 391 | | break; |
| 392 | | case TokenizeStateIncludeQuote: |
| 393 | | if (c == t.unquote_char) { |
| 394 | | t.include_path = buf_slice(t.buf, t.quote_start_pos + 1, t.pos); |
| 395 | | t.state = TokenizeStateDirectiveEnd; |
| 396 | | } |
| 397 | | break; |
| 398 | | case TokenizeStateDirectiveEnd: |
| 399 | | switch (c) { |
| 400 | | case '\n': |
| 401 | | end_directive(&t); |
| 402 | | break; |
| 403 | | case ' ': |
| 404 | | case '\t': |
| 405 | | case '\f': |
| 406 | | case '\r': |
| 407 | | case 0xb: |
| 408 | | break; |
| 409 | | default: |
| 410 | | tokenize_error(&t, "expected whitespace or newline: '%c'", c); |
| 280 | tokenize_error(&t, "invalid character: '%c'", c); |
| 411 | 281 | } |
| 412 | 282 | break; |
| 413 | 283 | case TokenizeStateSymbol: |
| ... | ... | @@ -415,8 +285,10 @@ static ZigList<Token> *tokenize(Buf *buf, ZigList<char *> *include_paths, Buf *c |
| 415 | 285 | case SYMBOL_CHAR: |
| 416 | 286 | break; |
| 417 | 287 | default: |
| 418 | | end_symbol(&t); |
| 419 | | break; |
| 288 | t.pos -= 1; |
| 289 | end_token(&t); |
| 290 | t.state = TokenizeStateStart; |
| 291 | continue; |
| 420 | 292 | } |
| 421 | 293 | break; |
| 422 | 294 | case TokenizeStateString: |
| ... | ... | @@ -434,7 +306,22 @@ static ZigList<Token> *tokenize(Buf *buf, ZigList<char *> *include_paths, Buf *c |
| 434 | 306 | case DIGIT: |
| 435 | 307 | break; |
| 436 | 308 | default: |
| 437 | | end_symbol(&t); |
| 309 | t.pos -= 1; |
| 310 | end_token(&t); |
| 311 | t.state = TokenizeStateStart; |
| 312 | continue; |
| 313 | } |
| 314 | break; |
| 315 | case TokenizeStateSawDash: |
| 316 | switch (c) { |
| 317 | case '>': |
| 318 | t.cur_tok->id = TokenIdArrow; |
| 319 | end_token(&t); |
| 320 | t.state = TokenizeStateStart; |
| 321 | break; |
| 322 | default: |
| 323 | end_token(&t); |
| 324 | t.state = TokenizeStateStart; |
| 438 | 325 | break; |
| 439 | 326 | } |
| 440 | 327 | break; |
| ... | ... | @@ -450,30 +337,17 @@ static ZigList<Token> *tokenize(Buf *buf, ZigList<char *> *include_paths, Buf *c |
| 450 | 337 | switch (t.state) { |
| 451 | 338 | case TokenizeStateStart: |
| 452 | 339 | break; |
| 453 | | case TokenizeStateDirective: |
| 454 | | end_directive(&t); |
| 455 | | break; |
| 456 | | case TokenizeStateDirectiveName: |
| 457 | | end_directive_name(&t); |
| 458 | | end_directive(&t); |
| 459 | | break; |
| 460 | | case TokenizeStateInclude: |
| 461 | | tokenize_error(&t, "missing include path"); |
| 462 | | break; |
| 463 | 340 | case TokenizeStateSymbol: |
| 464 | | end_symbol(&t); |
| 341 | end_token(&t); |
| 465 | 342 | break; |
| 466 | 343 | case TokenizeStateString: |
| 467 | 344 | tokenize_error(&t, "unterminated string"); |
| 468 | 345 | break; |
| 469 | 346 | case TokenizeStateNumber: |
| 470 | | end_symbol(&t); |
| 347 | end_token(&t); |
| 471 | 348 | break; |
| 472 | | case TokenizeStateIncludeQuote: |
| 473 | | tokenize_error(&t, "unterminated include path"); |
| 474 | | break; |
| 475 | | case TokenizeStateDirectiveEnd: |
| 476 | | end_directive(&t); |
| 349 | case TokenizeStateSawDash: |
| 350 | end_token(&t); |
| 477 | 351 | break; |
| 478 | 352 | } |
| 479 | 353 | assert(!t.cur_tok); |
| ... | ... | @@ -482,7 +356,6 @@ static ZigList<Token> *tokenize(Buf *buf, ZigList<char *> *include_paths, Buf *c |
| 482 | 356 | |
| 483 | 357 | static const char * token_name(Token *token) { |
| 484 | 358 | switch (token->id) { |
| 485 | | case TokenIdDirective: return "Directive"; |
| 486 | 359 | case TokenIdSymbol: return "Symbol"; |
| 487 | 360 | case TokenIdLParen: return "LParen"; |
| 488 | 361 | case TokenIdRParen: return "RParen"; |
| ... | ... | @@ -494,6 +367,9 @@ static const char * token_name(Token *token) { |
| 494 | 367 | case TokenIdSemicolon: return "Semicolon"; |
| 495 | 368 | case TokenIdNumberLiteral: return "NumberLiteral"; |
| 496 | 369 | case TokenIdPlus: return "Plus"; |
| 370 | case TokenIdColon: return "Colon"; |
| 371 | case TokenIdArrow: return "Arrow"; |
| 372 | case TokenIdDash: return "Dash"; |
| 497 | 373 | } |
| 498 | 374 | return "(invalid token)"; |
| 499 | 375 | } |
| ... | ... | @@ -507,6 +383,83 @@ static void print_tokens(Buf *buf, ZigList<Token> *tokens) { |
| 507 | 383 | } |
| 508 | 384 | } |
| 509 | 385 | |
| 386 | enum NodeType { |
| 387 | NodeTypeRoot, |
| 388 | }; |
| 389 | |
| 390 | struct AstNode { |
| 391 | enum NodeType type; |
| 392 | ZigList<AstNode *> children; |
| 393 | }; |
| 394 | |
| 395 | enum AstState { |
| 396 | AstStateStart, |
| 397 | }; |
| 398 | |
| 399 | struct BuildAst { |
| 400 | Buf *buf; |
| 401 | AstNode *root; |
| 402 | AstState state; |
| 403 | int line; |
| 404 | int column; |
| 405 | }; |
| 406 | |
| 407 | __attribute__ ((format (printf, 2, 3))) |
| 408 | static void ast_error(BuildAst *b, const char *format, ...) { |
| 409 | int line = b->line + 1; |
| 410 | int column = b->column + 1; |
| 411 | |
| 412 | va_list ap; |
| 413 | va_start(ap, format); |
| 414 | fprintf(stderr, "Error: Line %d, column %d: ", line, column); |
| 415 | vfprintf(stderr, format, ap); |
| 416 | fprintf(stderr, "\n"); |
| 417 | va_end(ap); |
| 418 | exit(EXIT_FAILURE); |
| 419 | } |
| 420 | |
| 421 | static inline bool mem_eql_str(const char *mem, size_t mem_len, const char *str) { |
| 422 | size_t str_len = strlen(str); |
| 423 | if (str_len != mem_len) |
| 424 | return false; |
| 425 | return memcmp(mem, str, mem_len) == 0; |
| 426 | } |
| 427 | |
| 428 | |
| 429 | static AstNode *build_ast(Buf *buf, ZigList<Token> *tokens) { |
| 430 | BuildAst b = {0}; |
| 431 | b.buf = buf; |
| 432 | b.root = allocate<AstNode>(1); |
| 433 | b.root->type = NodeTypeRoot; |
| 434 | |
| 435 | for (int i = 0; i < tokens->length; i += 1) { |
| 436 | Token *token = &tokens->at(i); |
| 437 | const char *token_str = buf_ptr(buf) + token->start_pos; |
| 438 | int token_len = token->end_pos - token->start_pos; |
| 439 | b.line = token->start_line; |
| 440 | b.column = token->start_column; |
| 441 | switch (b.state) { |
| 442 | case AstStateStart: |
| 443 | if (mem_eql_str(token_str, token_len, "fn")) { |
| 444 | zig_panic("TODO fn"); |
| 445 | } else { |
| 446 | Buf msg = {0}; |
| 447 | buf_append_str(&msg, "unexpected symbol: '"); |
| 448 | buf_append_mem(&msg, token_str, token_len); |
| 449 | buf_append_str(&msg, "'"); |
| 450 | ast_error(&b, "%s", buf_ptr(&msg)); |
| 451 | } |
| 452 | break; |
| 453 | } |
| 454 | } |
| 455 | |
| 456 | return b.root; |
| 457 | } |
| 458 | |
| 459 | static void print_ast(AstNode *node) { |
| 460 | zig_panic("TODO"); |
| 461 | } |
| 462 | |
| 510 | 463 | char cur_dir[1024]; |
| 511 | 464 | |
| 512 | 465 | int main(int argc, char **argv) { |
| ... | ... | @@ -559,18 +512,18 @@ int main(int argc, char **argv) { |
| 559 | 512 | |
| 560 | 513 | Buf *in_data = fetch_file(in_f); |
| 561 | 514 | |
| 562 | | fprintf(stderr, "Original source:\n%s\n", buf_ptr(in_data)); |
| 515 | fprintf(stderr, "Original source:\n"); |
| 516 | fprintf(stderr, "----------------\n"); |
| 517 | fprintf(stderr, "%s\n", buf_ptr(in_data)); |
| 563 | 518 | |
| 564 | 519 | ZigList<Token> *tokens = tokenize(in_data, &include_paths, cur_dir_path); |
| 565 | 520 | |
| 566 | 521 | fprintf(stderr, "\nTokens:\n"); |
| 522 | fprintf(stderr, "---------\n"); |
| 567 | 523 | print_tokens(in_data, tokens); |
| 568 | 524 | |
| 569 | | /* |
| 570 | | Buf *preprocessed_source = preprocess(in_data, tokens, &include_paths, cur_dir_path); |
| 571 | | |
| 572 | | fprintf(stderr, "\nPreprocessed source:\n%s\n", buf_ptr(preprocessed_source)); |
| 573 | | */ |
| 525 | AstNode *root = build_ast(in_data, tokens); |
| 526 | print_ast(root); |
| 574 | 527 | |
| 575 | 528 | |
| 576 | 529 | return EXIT_SUCCESS; |