zig

commit c7dc56f737ad9eab83b604e132cc3b33bf35e7f0 (tree)
parent c4f5a002535cfd1259d001dbda7923cd635dd0ba
Author: Andrew Kelley <superjoe30@gmail.com>
Date:   Sat,  6 Feb 2016 19:35:33 -0700

add test for ?? return and add const expr eval for ? prefix op

Diffstat:
M
src/analyze.cpp
 | 
18
++++++++++++++++--
M
test/self_hosted.zig
 | 
8
++++++++

2 files changed, 24 insertions(+), 2 deletions(-)
diff --git a/src/analyze.cpp b/src/analyze.cpp
@@ -2399,6 +2399,16 @@ static TypeTableEntry *resolve_expr_const_val_as_null(CodeGen *g, AstNode *node,
     return type;
 }
 
+static TypeTableEntry *resolve_expr_const_val_as_non_null(CodeGen *g, AstNode *node,
+        TypeTableEntry *type, ConstExprValue *other_val)
+{
+    assert(other_val->ok);
+    Expr *expr = get_resolved_expr(node);
+    expr->const_val.ok = true;
+    expr->const_val.data.x_maybe = other_val;
+    return type;
+}
+
 static TypeTableEntry *resolve_expr_const_val_as_c_string_lit(CodeGen *g, AstNode *node, Buf *str) {
     Expr *expr = get_resolved_expr(node);
     expr->const_val.ok = true;
@@ -4376,8 +4386,12 @@ static TypeTableEntry *analyze_prefix_op_expr(CodeGen *g, ImportTableEntry *impo
                     add_node_error(g, expr_node, buf_sprintf("unable to wrap unreachable in maybe type"));
                     return g->builtin_types.entry_invalid;
                 } else {
-                    // TODO eval const expr
-                    return get_maybe_type(g, type_entry);
+                    ConstExprValue *target_const_val = &get_resolved_expr(expr_node)->const_val;
+                    TypeTableEntry *maybe_type = get_maybe_type(g, type_entry);
+                    if (!target_const_val->ok) {
+                        return maybe_type;
+                    }
+                    return resolve_expr_const_val_as_non_null(g, node, maybe_type, target_const_val);
                 }
             }
         case PrefixOpError:
diff --git a/test/self_hosted.zig b/test/self_hosted.zig
@@ -188,3 +188,11 @@ fn do_err_return_in_assignment() -> %void {
 fn make_a_non_err() -> %i32 {
     return 1;
 }
+
+
+
+#attribute("test")
+fn rhs_maybe_unwrap_return() {
+    const x = ?true;
+    const y = x ?? return;
+}