103 lines
2.7 KiB
C
103 lines
2.7 KiB
C
/* splc0.c — SPL compiler CLI (stage 1, 引导用)
|
|
*
|
|
* splc0 --dump tokens|ast|all <file> dump 前端产物
|
|
* splc0 <in> <out> 编译 (阶段 B 实现)
|
|
*/
|
|
#define __SCC_LOG_IMPL_IMPORT_SRC__
|
|
#include "../stage0/include/utils.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "spl_ast.h"
|
|
#include "spl_lexer.h"
|
|
#include "spl_tok.h"
|
|
|
|
static char *read_file(const char *path, long *out_len) {
|
|
FILE *f = fopen(path, "rb");
|
|
if (!f) {
|
|
perror("fopen");
|
|
return NULL;
|
|
}
|
|
fseek(f, 0, SEEK_END);
|
|
long len = ftell(f);
|
|
fseek(f, 0, SEEK_SET);
|
|
char *buf = malloc((size_t)len + 1);
|
|
if (!buf) {
|
|
fclose(f);
|
|
return NULL;
|
|
}
|
|
fread(buf, 1, (size_t)len, f);
|
|
fclose(f);
|
|
buf[len] = '\0';
|
|
*out_len = len;
|
|
return buf;
|
|
}
|
|
|
|
static const char *const tok_type_names[] = {
|
|
#define X(name, enum_name, dummy) #enum_name,
|
|
KEYWORD_TABLE
|
|
#undef X
|
|
#define X(name, enum_name, dummy) #enum_name,
|
|
TOKEN_TABLE
|
|
#undef X
|
|
};
|
|
|
|
static void dump_tokens(const char *src, const char *fname) {
|
|
spl_tok_vec_t toks = spl_lex(src, fname);
|
|
printf("tokens got (%zu)\n", toks.size);
|
|
for (usize i = 0; i < toks.size; i++) {
|
|
const spl_tok_t *t = &toks.data[i];
|
|
printf("[%s] %.*s (%zu:%zu)\n", tok_type_names[t->type], (int)t->len, t->lexeme, t->line,
|
|
t->col);
|
|
}
|
|
vec_free(toks);
|
|
}
|
|
|
|
static void dump_ast(const char *src, const char *fname) {
|
|
spl_tok_vec_t toks = spl_lex(src, fname);
|
|
spl_ast_t ast;
|
|
spl_ast_init(&ast, &toks);
|
|
spl_ast_prase(&ast);
|
|
spl_ast_valid(&ast);
|
|
spl_ast_dump(&ast, ast.root);
|
|
spl_ast_drop(&ast);
|
|
}
|
|
|
|
static int cmd_dump(const char *flags, const char *path) {
|
|
long len;
|
|
char *src = read_file(path, &len);
|
|
if (!src)
|
|
return 1;
|
|
int do_tokens = strstr(flags, "tokens") != NULL || strcmp(flags, "all") == 0;
|
|
int do_ast = strstr(flags, "ast") != NULL || strcmp(flags, "all") == 0;
|
|
if (do_tokens)
|
|
dump_tokens(src, path);
|
|
if (do_ast)
|
|
dump_ast(src, path);
|
|
free(src);
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
if (argc < 2) {
|
|
LOG_FATAL("Usage: splc0 [--dump <flags>] <in> [out]\n");
|
|
return 1;
|
|
}
|
|
if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-h") == 0) {
|
|
LOG_INFO("splc0 <in> <out> compile (.spl -> .sir, 阶段 B)\n");
|
|
LOG_INFO("splc0 --dump <flags> <file> dump: tokens,ast,all\n");
|
|
return 0;
|
|
}
|
|
if (strcmp(argv[1], "--dump") == 0) {
|
|
if (argc < 4) {
|
|
LOG_INFO("splc0: --dump need <flags> <file>\n");
|
|
return 1;
|
|
}
|
|
return cmd_dump(argv[2], argv[3]);
|
|
}
|
|
LOG_FATAL("splc0: compile todo\n");
|
|
return 1;
|
|
}
|