more wrestling

src/ir/upper/resolve/ident.rs

@@ -0,0 +1,174 @@
+use super::*;
+
+impl MemRes {
+    pub fn validate(
+        &self,
+        fns: &[UFunc],
+        structs: &[UStruct],
+        generics: &[UGeneric],
+        types: &mut Vec<Type>,
+        errs: &mut Vec<ResErr>,
+    ) -> Result<Res, Option<ResErr>> {
+        let no_gargs = || {
+            if self.gargs.len() > 0 {
+                Err(ResErr::GenericCount {
+                    origin: self.origin,
+                    expected: 0,
+                    found: self.gargs.len(),
+                })
+            } else {
+                Ok(())
+            }
+        };
+        Ok(match &self.mem.id {
+            &MemberID::Fn(id) => {
+                validate_gargs(
+                    &fns[id].gargs,
+                    &self.gargs,
+                    generics,
+                    types,
+                    errs,
+                    self.origin,
+                )?;
+                Res::Fn(FnInst {
+                    id,
+                    gargs: self.gargs.clone(),
+                })
+            }
+            &MemberID::Struct(id) => {
+                validate_gargs(
+                    &structs[id].gargs,
+                    &self.gargs,
+                    generics,
+                    types,
+                    errs,
+                    self.origin,
+                )?;
+                Res::Struct(StructInst {
+                    id,
+                    gargs: self.gargs.clone(),
+                })
+            }
+            &MemberID::Var(id) => {
+                no_gargs()?;
+                Res::Var(id)
+            }
+            &MemberID::Module(id) => {
+                no_gargs()?;
+                Res::Module(id)
+            }
+            MemberID::Type(def) => {
+                validate_gargs(&def.gargs, &self.gargs, generics, types, errs, self.origin)?;
+                inst_typedef(def, &self.gargs, types);
+                Res::Type(def.ty)
+            }
+        })
+    }
+}
+
+impl IdentID {
+    pub fn resolve(
+        self,
+        s: &mut Sources,
+        types: &mut Vec<Type>,
+        changed: &mut bool,
+        errs: &mut Vec<ResErr>,
+    ) -> Result<Res, ResolveRes> {
+        let status = &mut s.idents[self].status;
+        // TOOD: there are some clones here that shouldn't be needed
+        Ok(match status {
+            IdentStatus::Res(r) => r.clone(),
+            IdentStatus::Unres { path, base } => {
+                while let Some(mem) = path.pop() {
+                    let res = match base {
+                        ResBase::Unvalidated(u) => {
+                            match u.validate(s.fns, s.structs, s.generics, types, errs) {
+                                Ok(res) => res,
+                                Err(err) => {
+                                    *status = IdentStatus::Failed(err);
+                                    return Err(ResolveRes::Finished);
+                                }
+                            }
+                        }
+                        ResBase::Validated(res) => res.clone(),
+                    };
+                    *base = match (res, mem.ty) {
+                        (Res::Module(id), MemberTy::Member) => {
+                            let Some(m) = s.modules[id].members.get(&mem.name) else {
+                                return Err(ResolveRes::Unfinished);
+                            };
+                            ResBase::Unvalidated(MemRes {
+                                mem: m.clone(),
+                                origin: mem.origin,
+                                gargs: mem.gargs,
+                            })
+                        }
+                        (Res::Var(id), MemberTy::Field) => {
+                            // trait resolution here
+                            let Some(&child) = s.vars[id].children.get(&mem.name) else {
+                                return Err(ResolveRes::Unfinished);
+                            };
+                            ResBase::Unvalidated(MemRes {
+                                mem: Member {
+                                    id: MemberID::Var(child),
+                                },
+                                origin: mem.origin,
+                                gargs: mem.gargs,
+                            })
+                        }
+                        _ => {
+                            *status = IdentStatus::Failed(Some(ResErr::UnknownMember {
+                                origin: mem.origin,
+                                ty: mem.ty,
+                                name: mem.name.clone(),
+                                parent: base.clone(),
+                            }));
+                            return Err(ResolveRes::Finished);
+                        }
+                    };
+                }
+                let res = match base {
+                    ResBase::Unvalidated(u) => {
+                        match u.validate(s.fns, s.structs, s.generics, types, errs) {
+                            Ok(res) => res,
+                            Err(err) => {
+                                *status = IdentStatus::Failed(err);
+                                return Err(ResolveRes::Finished);
+                            }
+                        }
+                    }
+                    ResBase::Validated(res) => res.clone(),
+                };
+                *status = IdentStatus::Res(res.clone());
+                *changed = true;
+                res
+            }
+            IdentStatus::Cooked => return Err(ResolveRes::Finished),
+            IdentStatus::Failed(_) => return Err(ResolveRes::Finished),
+        })
+    }
+}
+
+pub fn validate_gargs(
+    dst: &[GenericID],
+    src: &[TypeID],
+    generics: &[UGeneric],
+    types: &[Type],
+    errs: &mut Vec<ResErr>,
+    origin: Origin,
+) -> Result<(), Option<ResErr>> {
+    if dst.len() != src.len() {
+        return Err(Some(ResErr::GenericCount {
+            origin,
+            expected: dst.len(),
+            found: src.len(),
+        }));
+    }
+    for (dst, src) in dst.iter().zip(src.iter()) {
+        let g = &generics[dst];
+        let t = &types[src];
+        // TODO: validate trait constraints
+    }
+    Ok(())
+}
+