Commit 4ef1c1c705ceb2d89d0bf93d7a2403c165ae9dc0 | zig

author	Veikka Tuominen <git@vexu.eu> 2022-04-15 10:24:18+03:00
committer	Veikka Tuominen <git@vexu.eu> 2022-04-15 10:24:18+03:00
log	4ef1c1c705ceb2d89d0bf93d7a2403c165ae9dc0
tree	b6dd27838bc132121570fe81f4dbbe9b6bcb61f4
parent	d66c61a2cf69665223815a1a12a1f93b30b99571

Sema: allow fieldType on optionals and error unions

1 files changed, 31 insertions(+), 15 deletions(-)

src
- Sema.zig

src/Sema.zig+31-15

 @@ -13071,22 +13071,38 @@ fn fieldType(
 ) CompileError!Air.Inst.Ref {
     const resolved_ty = try sema.resolveTypeFields(block, ty_src, aggregate_ty);
     const target = sema.mod.getTarget();
-    switch (resolved_ty.zigTypeTag()) {
+    var cur_ty = resolved_ty;
-        .Struct => {
+    while (true) {
-            const struct_obj = resolved_ty.castTag(.@"struct").?.data;
+        switch (cur_ty.zigTypeTag()) {
-            const field = struct_obj.fields.get(field_name) orelse
+            .Struct => {
-                return sema.failWithBadStructFieldAccess(block, struct_obj, field_src, field_name);
+                const struct_obj = cur_ty.castTag(.@"struct").?.data;
-            return sema.addType(field.ty);
+                const field = struct_obj.fields.get(field_name) orelse
-        },
+                    return sema.failWithBadStructFieldAccess(block, struct_obj, field_src, field_name);
-        .Union => {
+                return sema.addType(field.ty);
-            const union_obj = resolved_ty.cast(Type.Payload.Union).?.data;
+            },
-            const field = union_obj.fields.get(field_name) orelse
+            .Union => {
-                return sema.failWithBadUnionFieldAccess(block, union_obj, field_src, field_name);
+                const union_obj = cur_ty.cast(Type.Payload.Union).?.data;
-            return sema.addType(field.ty);
+                const field = union_obj.fields.get(field_name) orelse
-        },
+                    return sema.failWithBadUnionFieldAccess(block, union_obj, field_src, field_name);
-        else => return sema.fail(block, ty_src, "expected struct or union; found '{}'", .{
+                return sema.addType(field.ty);
+            },
+            .Optional => {
+                if (cur_ty.castTag(.optional)) |some| {
+                    // Struct/array init through optional requires the child type to not be a pointer.
+                    // If the child of .optional is a pointer it'll error on the next loop.
+                    cur_ty = some.data;
+                    continue;
+                }
+            },
+            .ErrorUnion => {
+                cur_ty = cur_ty.errorUnionPayload();
+                continue;
+            },
+            else => {},
+        }
+        return sema.fail(block, ty_src, "expected struct or union; found '{}'", .{
             resolved_ty.fmt(target),
-        }),
+        });
     }
 }