yukigo-haskell-parser 0.1.3 → 0.2.2

package/src/typechecker/core.ts

@@ -1,192 +1,195 @@
-import {
-  charType,
-  Environment,
-  Result,
-  showType,
-  Substitution,
-  Type,
-  TypeConstructor,
-  TypeScheme,
-  TypeVar,
-} from "./checker.js";
-
-export class CoreHM {
-  private nextVarId = 0;
-  constructor(
-    private aliases: Map<string, Type> = new Map(),
-    public typeClasses: Map<string, string[]> = new Map()
-  ) {}
-  public resolve(t: Type): Type {
-    if (t.type === "TypeConstructor" && this.aliases.has(t.name))
-      return this.resolve(this.aliases.get(t.name)!);
-    return t;
-  }
-  public freshVar(constraints: string[] = []): TypeVar {
-    return { type: "TypeVar", id: this.nextVarId++, constraints };
-  }
-
-  public freeTypeVars(t: Type): Map<number, string[]> {
-    const freeVars = new Map<number, string[]>();
-    const collect = (type: Type) => {
-      if (type.type === "TypeVar") {
-        freeVars.set(type.id, type.constraints);
-      } else if (type.type === "TypeConstructor") {
-        type.args.forEach(collect);
-      }
-    };
-    collect(t);
-    return freeVars;
-  }
-  public generalize(env: Environment, t: Type): TypeScheme {
-    const envFreeVars = new Set<number>();
-    for (const scheme of env.values()) {
-      const schemeFreeVars = this.freeTypeVars(scheme.body);
-      for (const id of schemeFreeVars.keys()) {
-        if (!scheme.quantifiers.includes(id)) {
-          envFreeVars.add(id);
-        }
-      }
-    }
-
-    const typeFreeVars = this.freeTypeVars(t);
-
-    const quantifiers: number[] = [];
-    const constraints = new Map<number, string[]>();
-
-    for (const [id, consts] of typeFreeVars.entries()) {
-      if (!envFreeVars.has(id)) {
-        quantifiers.push(id);
-        if (consts.length > 0) {
-          constraints.set(id, consts);
-        }
-      }
-    }
-
-    return { type: "TypeScheme", quantifiers, body: t, constraints };
-  }
-
-  public instantiate(scheme: TypeScheme): Type {
-    const substitutions = new Map<number, Type>();
-    scheme.quantifiers.forEach((id) => {
-      const constraints = scheme.constraints.get(id) || [];
-      substitutions.set(id, this.freshVar(constraints));
-    });
-    return this.applySubst(substitutions, scheme.body);
-  }
-
-  public applySubst(subst: Substitution, t: Type): TypeVar | TypeConstructor {
-    if (t.type === "TypeVar") {
-      const replacement = subst.get(t.id);
-      return replacement ? replacement : t;
-    } else if (t.type === "TypeConstructor") {
-      return {
-        type: "TypeConstructor",
-        name: t.name,
-        args: t.args.map((arg) => this.applySubst(subst, arg)),
-      };
-    }
-
-    throw new Error("Unexpected TypeScheme in applySubst");
-  }
-
-  public composeSubst(s1: Substitution, s2: Substitution): Substitution {
-    const result = new Map(s2);
-    for (const [id, type] of s1) {
-      result.set(id, this.applySubst(s2, type));
-    }
-    return result;
-  }
-
-  public unify(t1: Type, t2: Type): Result<Substitution> {
-    const rt1 = this.resolve(t1);
-    const rt2 = this.resolve(t2);
-    if (rt1.type === "TypeVar") return this.unifyVar(rt1, rt2);
-    if (rt2.type === "TypeVar") return this.unifyVar(rt2, rt1);
-    if (rt1.type === "TypeConstructor" && rt2.type === "TypeConstructor") {
-      const isStringList = (a: TypeConstructor, b: TypeConstructor) =>
-        (a.name === "YuString" && b.name === "List") ||
-        (b.name === "YuString" && a.name === "List");
-
-      if (isStringList(rt1, rt2)) {
-        const listType = rt1.name === "List" ? rt1 : rt2;
-        return this.unify(listType.args[0], charType);
-      }
-
-      if (rt1.name !== rt2.name || rt1.args.length !== rt2.args.length) {
-        return {
-          success: false,
-          error: `Cannot unify ${showType(rt1)} with ${showType(rt2)}`,
-        };
-      }
-      let sub: Substitution = new Map();
-      for (let i = 0; i < rt1.args.length; i++) {
-        const arg1 = this.applySubst(sub, rt1.args[i]);
-        const arg2 = this.applySubst(sub, rt2.args[i]);
-        const argSubRes = this.unify(arg1, arg2);
-        if (!argSubRes.success) return argSubRes;
-        sub = this.composeSubst(sub, argSubRes.value);
-      }
-      return { success: true, value: sub };
-    }
-    return { success: false, error: `Cannot unify non-types` };
-  }
-
-  public unifyVar(v: TypeVar, t: Type): Result<Substitution> {
-    if (t.type === "TypeVar" && t.id === v.id) {
-      return { success: true, value: new Map() };
-    }
-    if (this.occurs(v.id, t)) {
-      return { success: false, error: `Occurs check failed` };
-    }
-
-    for (const constraint of v.constraints) {
-      this.checkConstraint(constraint, t);
-    }
-    if (t.type === "TypeVar") {
-      for (const constraint of t.constraints) {
-        this.checkConstraint(constraint, v);
-      }
-      // Merge constraints
-      const mergedConstraints = [
-        ...new Set([...v.constraints, ...t.constraints]),
-      ];
-      v.constraints = mergedConstraints;
-      t.constraints = mergedConstraints;
-    }
-
-    return { success: true, value: new Map([[v.id, t]]) };
-  }
-
-  public checkConstraint(constraintName: string, t: Type) {
-    const rt = this.resolve(t);
-    if (rt.type === "TypeVar") {
-      if (!rt.constraints.includes(constraintName)) {
-        rt.constraints.push(constraintName);
-      }
-    } else if (rt.type === "TypeConstructor") {
-      let typeName = rt.name;
-      if (
-        typeName === "List" &&
-        rt.args.length === 1 &&
-        rt.args[0].type === "TypeConstructor" &&
-        rt.args[0].name === "YuChar"
-      ) {
-        typeName = "YuString";
-      }
-
-      const instances = this.typeClasses.get(constraintName);
-      if (!instances || !instances.includes(typeName)) {
-        throw new Error(
-          `Type '${showType(rt)}' is not an instance of '${constraintName}'`
-        );
-      }
-    }
-  }
-
-  public occurs(varId: number, t: Type): boolean {
-    if (t.type === "TypeVar") return t.id === varId;
-    if (t.type === "TypeConstructor")
-      return t.args.some((a) => this.occurs(varId, a));
-    return false;
-  }
-}
+import { YUTYPES } from "../utils/types.js";
+import {
+  charType,
+  Environment,
+  isListType,
+  isString,
+  Result,
+  showType,
+  Substitution,
+  Type,
+  TypeConstructor,
+  TypeScheme,
+  TypeVar,
+} from "./checker.js";
+
+export class CoreHM {
+  private nextVarId = 0;
+  constructor(
+    private aliases: Map<string, Type> = new Map(),
+    public typeClasses: Map<string, string[]> = new Map()
+  ) {}
+  public resolve(t: Type): Type {
+    if (t.type === "TypeConstructor" && this.aliases.has(t.name))
+      return this.resolve(this.aliases.get(t.name)!);
+    return t;
+  }
+  public freshVar(constraints: string[] = []): TypeVar {
+    return { type: "TypeVar", id: this.nextVarId++, constraints };
+  }
+
+  public freeTypeVars(t: Type): Map<number, string[]> {
+    const freeVars = new Map<number, string[]>();
+    const collect = (type: Type) => {
+      if (type.type === "TypeVar") {
+        freeVars.set(type.id, type.constraints);
+      } else if (type.type === "TypeConstructor") {
+        type.args.forEach(collect);
+      }
+    };
+    collect(t);
+    return freeVars;
+  }
+  public generalize(env: Environment, t: Type): TypeScheme {
+    const envFreeVars = new Set<number>();
+    for (const scheme of env.values()) {
+      const schemeFreeVars = this.freeTypeVars(scheme.body);
+      for (const id of schemeFreeVars.keys()) {
+        if (!scheme.quantifiers.includes(id)) {
+          envFreeVars.add(id);
+        }
+      }
+    }
+
+    const typeFreeVars = this.freeTypeVars(t);
+
+    const quantifiers: number[] = [];
+    const constraints = new Map<number, string[]>();
+
+    for (const [id, consts] of typeFreeVars.entries()) {
+      if (!envFreeVars.has(id)) {
+        quantifiers.push(id);
+        if (consts.length > 0) {
+          constraints.set(id, consts);
+        }
+      }
+    }
+
+    return { type: "TypeScheme", quantifiers, body: t, constraints };
+  }
+
+  public instantiate(scheme: TypeScheme): Type {
+    const substitutions = new Map<number, Type>();
+    scheme.quantifiers.forEach((id) => {
+      const constraints = scheme.constraints.get(id) || [];
+      substitutions.set(id, this.freshVar(constraints));
+    });
+    return this.applySubst(substitutions, scheme.body);
+  }
+
+  public applySubst(subst: Substitution, t: Type): TypeVar | TypeConstructor {
+    if (t.type === "TypeVar") {
+      const replacement = subst.get(t.id);
+      return replacement ? replacement : t;
+    } else if (t.type === "TypeConstructor") {
+      return {
+        type: "TypeConstructor",
+        name: t.name,
+        args: t.args.map((arg) => this.applySubst(subst, arg)),
+      };
+    }
+
+    throw new Error("Unexpected TypeScheme in applySubst");
+  }
+
+  public composeSubst(s1: Substitution, s2: Substitution): Substitution {
+    const result = new Map(s2);
+    for (const [id, type] of s1) {
+      result.set(id, this.applySubst(s2, type));
+    }
+    return result;
+  }
+
+  public unify(t1: Type, t2: Type): Result<Substitution> {
+    const rt1 = this.resolve(t1);
+    const rt2 = this.resolve(t2);
+    if (rt1.type === "TypeVar") return this.unifyVar(rt1, rt2);
+    if (rt2.type === "TypeVar") return this.unifyVar(rt2, rt1);
+    if (rt1.type === "TypeConstructor" && rt2.type === "TypeConstructor") {
+      const isStringList = (a: TypeConstructor, b: TypeConstructor) =>
+        (isString(a) && isListType(b)) ||
+        (isString(b) && isListType(a));
+
+      if (isStringList(rt1, rt2)) {
+        const listType = rt1.name === YUTYPES.List ? rt1 : rt2;
+        return this.unify(listType.args[0], charType);
+      }
+
+      if (rt1.name !== rt2.name || rt1.args.length !== rt2.args.length) {
+        return {
+          success: false,
+          error: `Cannot unify ${showType(rt1)} with ${showType(rt2)}`,
+        };
+      }
+      let sub: Substitution = new Map();
+      for (let i = 0; i < rt1.args.length; i++) {
+        const arg1 = this.applySubst(sub, rt1.args[i]);
+        const arg2 = this.applySubst(sub, rt2.args[i]);
+        const argSubRes = this.unify(arg1, arg2);
+        if (!argSubRes.success) return argSubRes;
+        sub = this.composeSubst(sub, argSubRes.value);
+      }
+      return { success: true, value: sub };
+    }
+    return { success: false, error: `Cannot unify non-types` };
+  }
+
+  public unifyVar(v: TypeVar, t: Type): Result<Substitution> {
+    if (t.type === "TypeVar" && t.id === v.id) {
+      return { success: true, value: new Map() };
+    }
+    if (this.occurs(v.id, t)) {
+      return { success: false, error: `Occurs check failed` };
+    }
+
+    for (const constraint of v.constraints) {
+      this.checkConstraint(constraint, t);
+    }
+    if (t.type === "TypeVar") {
+      for (const constraint of t.constraints) {
+        this.checkConstraint(constraint, v);
+      }
+      // Merge constraints
+      const mergedConstraints = [
+        ...new Set([...v.constraints, ...t.constraints]),
+      ];
+      v.constraints = mergedConstraints;
+      t.constraints = mergedConstraints;
+    }
+
+    return { success: true, value: new Map([[v.id, t]]) };
+  }
+
+  public checkConstraint(constraintName: string, t: Type) {
+    const rt = this.resolve(t);
+    if (rt.type === "TypeVar") {
+      if (!rt.constraints.includes(constraintName)) {
+        rt.constraints.push(constraintName);
+      }
+    } else if (rt.type === "TypeConstructor") {
+      let typeName = rt.name;
+      if (
+        typeName === YUTYPES.List &&
+        rt.args.length === 1 &&
+        rt.args[0].type === "TypeConstructor" &&
+        rt.args[0].name === YUTYPES.YuChar
+      ) {
+        typeName = YUTYPES.YuString;
+      }
+
+      const instances = this.typeClasses.get(constraintName);
+      if (!instances || !instances.includes(typeName)) {
+        throw new Error(
+          `Type '${showType(rt)}' is not an instance of '${constraintName}'`
+        );
+      }
+    }
+  }
+
+  public occurs(varId: number, t: Type): boolean {
+    if (t.type === "TypeVar") return t.id === varId;
+    if (t.type === "TypeConstructor")
+      return t.args.some((a) => this.occurs(varId, a));
+    return false;
+  }
+}