stage1 使用切片表达字符串而非C字符串 提供属性系统

stage1/spl_ast2ir.c

@@ -38,6 +38,26 @@ static spl_ir_node_ref_t ast2ir_intrin_call(spl_ast2ir_t *a, spl_ast_node_t *n,
 static spl_ir_node_ref_t ast2ir_intrin_field(spl_ast2ir_t *a, spl_ast_node_t *n, spl_intrin_t in);
 static spl_ir_node_ref_t ast2ir_intrin_len(spl_ast2ir_t *a, spl_ast_node_t *n);
 
+static void copy_attrs_to_ir_node(spl_ast2ir_t *a, spl_ast_node_t *n, spl_ir_node_ref_t ir_ref) {
+    if (!ir_ref || !n || vec_size(n->attrs) == 0)
+        return;
+    spl_ir_node_t *irn = spl_ir_node(&a->ir->ir, a->ir->current_fn, ir_ref);
+    if (!irn)
+        return;
+    for (usize i = 0; i < vec_size(n->attrs); ++i)
+        vec_push(irn->attrs, vec_at(n->attrs, i));
+}
+
+static void copy_attrs_to_ir_func(spl_ast2ir_t *a, spl_ast_node_t *n, spl_ir_func_ref_t fn_ref) {
+    if (!fn_ref || !n || vec_size(n->attrs) == 0)
+        return;
+    spl_ir_func_t *irf = spl_ir_func(&a->ir->ir, fn_ref);
+    if (!irf)
+        return;
+    for (usize i = 0; i < vec_size(n->attrs); ++i)
+        vec_push(irf->attrs, vec_at(n->attrs, i));
+}
+
 static spl_ir_node_ref_t ast2ir_intrin_call(spl_ast2ir_t *a, spl_ast_node_t *n, spl_intrin_t in,
                                             spl_ast_node_ref_vec_t *args) {
     switch (in) {
@@ -59,12 +79,37 @@ static spl_ir_node_ref_t ast2ir_intrin_call(spl_ast2ir_t *a, spl_ast_node_t *n,
         spl_ir_node_ref_t br = spl_ir_builder_control_br(a->ir, notc);
         spl_ir_builder_block_end(a->ir);
 
-        usize trap_blk = spl_ir_builder_block_begin(a->ir);
+        usize panic_blk = spl_ir_builder_block_begin(a->ir);
+
+        const char *fname = n->dbg.fname ? n->dbg.fname : "?";
+        const char *user_msg = NULL;
+        if (vec_size(*args) == 2) {
+            spl_ast_node_t *mn = spl_ast_node(a->sema->ast, vec_at(*args, 1));
+            if (mn && mn->kind == SPL_AST_EXPR_STRING_LIT && mn->string_lit.data)
+                user_msg = mn->string_lit.data;
+        }
+
+        char *msg;
+        int need;
+        if (user_msg) {
+            need = snprintf(NULL, 0, "%s:%d:%d: assert failed: %s", fname, n->dbg.line, n->dbg.col,
+                            user_msg);
+            msg = malloc((usize)need + 1);
+            snprintf(msg, (usize)need + 1, "%s:%d:%d: assert failed: %s", fname, n->dbg.line,
+                     n->dbg.col, user_msg);
+        } else {
+            need = snprintf(NULL, 0, "%s:%d:%d: assert failed", fname, n->dbg.line, n->dbg.col);
+            msg = malloc((usize)need + 1);
+            snprintf(msg, (usize)need + 1, "%s:%d:%d: assert failed", fname, n->dbg.line,
+                     n->dbg.col);
+        }
+
+        spl_ir_builder_dbg_panic(a->ir, msg, (usize)need);
         spl_ir_builder_control_trap(a->ir);
         spl_ir_builder_block_end(a->ir);
 
         usize cont_blk = spl_ir_builder_block_begin(a->ir);
-        spl_ir_builder_control_br_fixup(a->ir, br, trap_blk, cont_blk);
+        spl_ir_builder_control_br_fixup(a->ir, br, panic_blk, cont_blk);
         return 0;
     }
     case SPL_INTRIN_BREAKPOINT:
@@ -259,6 +304,8 @@ static spl_ir_node_ref_t emit_expr(spl_ast2ir_t *a, spl_ast_node_ref_t ref) {
     if (!n || n->expected_lang_tid == 0)
         return val;
 
+    copy_attrs_to_ir_node(a, n, val);
+
     spl_lang_type_id_t from = n->resolved_lang_tid;
     spl_lang_type_id_t to = n->expected_lang_tid;
     if (!from || from == to)
@@ -284,8 +331,27 @@ static spl_ir_node_ref_t emit_expr_raw(spl_ast2ir_t *a, spl_ast_node_ref_t ref)
         return spl_ir_builder_type_const_float(a->ir, node_ir(a, n), n->float_lit.val);
     case SPL_AST_EXPR_CHAR_LIT:
         return spl_ir_builder_type_const_char(a->ir, node_ir(a, n), n->char_lit.val);
-    case SPL_AST_EXPR_STRING_LIT:
-        return spl_ir_builder_type_const_cstr(a->ir, node_ir(a, n), n->string_lit.data);
+    case SPL_AST_EXPR_STRING_LIT: {
+        usize len = n->string_lit.len;
+        spl_lang_type_id_t u8_lang = spl_lang_make_int(a->sema->type, 8, 0);
+        spl_type_id_t u8_ir = to_ir_type(a, u8_lang);
+        spl_type_id_t ptr_u8_ir = spl_type_builder_ptr(a->sema->type, u8_ir);
+        spl_lang_type_id_t usize_lang = spl_lang_make_int(a->sema->type, sizeof(void *) * 8, 0);
+        spl_type_id_t usize_ir = to_ir_type(a, usize_lang);
+        spl_lang_type_id_t arr_lang = spl_lang_make_array(a->sema->type, u8_lang, len);
+        spl_type_id_t arr_ir = to_ir_type(a, arr_lang);
+
+        spl_ir_node_ref_t raw =
+            spl_ir_builder_type_const_bytes(a->ir, arr_ir, n->string_lit.data, n->string_lit.len);
+        spl_ir_node_ref_t ptr = spl_ir_builder_mem_global_alloc(a->ir, ptr_u8_ir, raw);
+        spl_ir_node_ref_t len_c = spl_ir_builder_type_const_int(a->ir, usize_ir, len);
+
+        spl_type_id_t slice_ir = node_ir(a, n);
+        spl_ir_node_ref_t node = spl_ir_builder_agg_construct(a->ir, slice_ir);
+        spl_ir_builder_agg_construct_field(a->ir, node, ptr);
+        spl_ir_builder_agg_construct_field(a->ir, node, len_c);
+        return node;
+    }
     case SPL_AST_EXPR_BOOL_LIT:
         return spl_ir_builder_type_const_bool(a->ir, node_ir(a, n), n->bool_lit.val);
     case SPL_AST_EXPR_NULL_LIT:
@@ -935,10 +1001,11 @@ static void emit_fn(spl_ast2ir_t *a, spl_ast_node_t *n) {
     if (!fn_ir)
         return;
 
-    spl_ir_attr_t attr = {.kind = SPL_IR_ATTR_NONE};
-    spl_ir_func_ref_t fn_ref = spl_ir_builder_fn_begin(a->ir, n->fn_decl.name, fn_ir, attr);
+    spl_ir_func_ref_t fn_ref = spl_ir_builder_fn_begin(a->ir, n->fn_decl.name, fn_ir);
     map_put(a->def2ir_map, n->resolved_def_id, (spl_ir_node_ref_t)fn_ref);
 
+    copy_attrs_to_ir_func(a, n, fn_ref);
+
     spl_lang_type_id_t saved_ret = a->cur_ret_lang;
     a->cur_ret_lang = fn_ln->extra.fn.ret;