yukigo-haskell-parser 0.1.0

package/dist/typechecker/checker.d.ts

@@ -0,0 +1,80 @@
+import { AST, Function, UnguardedBody, GuardedBody, Visitor } from "@yukigo/ast";
+import { CoreHM } from "./core.js";
+export interface TypeVar {
+    type: "TypeVar";
+    id: number;
+    name?: string;
+    constraints: string[];
+}
+export interface TypeConstructor {
+    type: "TypeConstructor";
+    name: string;
+    args: Type[];
+}
+export interface FunctionType {
+    type: "TypeConstructor";
+    name: "->";
+    args: [Type, Type];
+}
+export interface ListType {
+    type: "TypeConstructor";
+    name: "List";
+    args: [Type];
+}
+export interface TupleType {
+    type: "TypeConstructor";
+    name: "Tuple";
+    args: Type[];
+}
+export interface TypeScheme {
+    type: "TypeScheme";
+    quantifiers: number[];
+    body: Type;
+    constraints: Map<number, string[]>;
+}
+export type Type = TypeVar | TypeConstructor;
+export type Environment = Map<string, TypeScheme>;
+export type Substitution = Map<number, Type>;
+export type Result<T> = {
+    success: true;
+    value: T;
+} | {
+    success: false;
+    error: string;
+};
+export declare const booleanType: TypeConstructor;
+export declare const numberType: TypeConstructor;
+export declare const stringType: TypeConstructor;
+export declare class FunctionRegistrarVisitor implements Visitor<void> {
+    private env;
+    private signatureMap;
+    private coreHM;
+    constructor(env: Environment, signatureMap: Map<string, TypeScheme>, coreHM: CoreHM);
+    visitFunction(node: Function): void;
+}
+export declare class FunctionCheckerVisitor implements Visitor<void> {
+    private environments;
+    private signatureMap;
+    private coreHM;
+    private errors;
+    constructor(environments: Environment[], signatureMap: Map<string, TypeScheme>, coreHM: CoreHM, errors: string[]);
+    visitFunction(node: Function): void;
+}
+export declare class TypeChecker {
+    private signatureMap;
+    private coreHM;
+    private errors;
+    constructor();
+    check(ast: AST): string[];
+    inferencePass(ast: AST): void;
+}
+export declare function showType(t: Type): string;
+export declare function getReturnType(type: Type): Type;
+export declare function getArgumentTypes(type: Type): Type[];
+export declare function getArity(type: Type): number;
+export declare function isUnguardedBody(body: UnguardedBody | GuardedBody[]): body is UnguardedBody;
+export declare function isFunctionType(t: Type): t is FunctionType;
+export declare function isListType(t: Type): t is ListType;
+export declare function isTupleType(t: Type): t is TupleType;
+export declare function functionType(params: Type, returnType: Type): FunctionType;
+export declare function listType(elementsType: Type): ListType;

package/dist/typechecker/checker.js

@@ -0,0 +1,269 @@
+import { Function, Return, } from "@yukigo/ast";
+import { InferenceEngine, PatternVisitor } from "./inference.js";
+import { CoreHM } from "./core.js";
+import { DeclarationCollectorVisitor } from "./DeclarationCollector.js";
+export const booleanType = {
+    type: "TypeConstructor",
+    name: "YuBoolean",
+    args: [],
+};
+export const numberType = {
+    type: "TypeConstructor",
+    name: "YuNumber",
+    args: [],
+};
+export const stringType = {
+    type: "TypeConstructor",
+    name: "YuString",
+    args: [],
+};
+export class FunctionRegistrarVisitor {
+    env;
+    signatureMap;
+    coreHM;
+    constructor(env, signatureMap, coreHM) {
+        this.env = env;
+        this.signatureMap = signatureMap;
+        this.coreHM = coreHM;
+    }
+    visitFunction(node) {
+        const functionName = node.identifier.value;
+        let funcScheme = this.signatureMap.get(functionName);
+        if (!funcScheme) {
+            const funcTypeVar = this.coreHM.freshVar();
+            funcScheme = {
+                type: "TypeScheme",
+                quantifiers: [],
+                body: funcTypeVar,
+                constraints: new Map(),
+            };
+            this.env.set(functionName, funcScheme);
+        }
+        for (const equation of node.equations) {
+            if (isUnguardedBody(equation.body)) {
+                const { statements } = equation.body.sequence;
+                statements
+                    .filter((stmt) => stmt instanceof Function)
+                    .forEach((func) => {
+                    this.env.set(func.identifier.value, funcScheme);
+                });
+            }
+        }
+    }
+}
+export class FunctionCheckerVisitor {
+    environments;
+    signatureMap;
+    coreHM;
+    errors;
+    constructor(environments, signatureMap, coreHM, errors) {
+        this.environments = environments;
+        this.signatureMap = signatureMap;
+        this.coreHM = coreHM;
+        this.errors = errors;
+    }
+    visitFunction(node) {
+        const functionName = node.identifier.value;
+        let funcScheme = this.signatureMap.get(functionName);
+        // Handle function without signature
+        if (!funcScheme) {
+            const firstEq = node.equations[0];
+            this.environments.unshift(new Map());
+            const inferenceEngine = new InferenceEngine(this.signatureMap, this.coreHM, this.environments);
+            const paramTypes = firstEq.patterns.map(() => this.coreHM.freshVar());
+            paramTypes.forEach((type, i) => {
+                try {
+                    new PatternVisitor(this.coreHM, this.signatureMap, type, this.environments, inferenceEngine).visit(firstEq.patterns[i]);
+                }
+                catch (error) {
+                    this.errors.push(`Type error in '${functionName}': ${error.message}`);
+                }
+            });
+            let inferredBodyType;
+            if (isUnguardedBody(firstEq.body)) {
+                const equationStatements = firstEq.body.sequence.statements;
+                equationStatements.forEach((stmt) => stmt.accept(new FunctionCheckerVisitor(this.environments, this.signatureMap, this.coreHM, this.errors)));
+                const returnResult = equationStatements
+                    .find((stmt) => stmt instanceof Return)
+                    .accept(inferenceEngine);
+                if (returnResult.success === false) {
+                    this.errors.push(`Type error in '${functionName}': ${returnResult.error}`);
+                    return;
+                }
+                inferredBodyType = returnResult.value;
+            }
+            else {
+                // Handle guarded body inference if necessary
+                inferredBodyType = this.coreHM.freshVar(); // Placeholder
+            }
+            const fullFuncType = paramTypes.reduceRight((acc, param) => functionType(param, acc), inferredBodyType);
+            // Generalize the inferred type to create a polymorphic type scheme
+            this.environments.shift();
+            funcScheme = this.coreHM.generalize(this.environments[0], fullFuncType);
+            this.signatureMap.set(functionName, funcScheme);
+            this.environments[0].set(functionName, funcScheme);
+        }
+        const expectedArity = getArity(this.coreHM.instantiate(funcScheme));
+        for (const [index, equation] of node.equations.entries()) {
+            if (equation.patterns.length !== expectedArity) {
+                this.errors.push(`Type error in '${functionName}': Arity mismatch in equation number ${index}`);
+                continue;
+            }
+            try {
+                const funcType = this.coreHM.instantiate(funcScheme);
+                const returnType = getReturnType(funcType);
+                const patternTypes = getArgumentTypes(funcType);
+                this.environments.unshift(new Map());
+                const inferenceEngine = new InferenceEngine(this.signatureMap, this.coreHM, this.environments);
+                patternTypes.forEach((argType, i) => {
+                    try {
+                        equation.patterns[i].accept(new PatternVisitor(this.coreHM, this.signatureMap, argType, this.environments, inferenceEngine));
+                    }
+                    catch (error) {
+                        this.errors.push(`Type error in '${functionName}': ${error.message}`);
+                    }
+                });
+                if (isUnguardedBody(equation.body)) {
+                    const equationStatements = equation.body.sequence.statements;
+                    equationStatements.forEach((stmt) => stmt.accept(new FunctionCheckerVisitor(this.environments, this.signatureMap, this.coreHM, this.errors)));
+                    const returnResult = equationStatements
+                        .find((stmt) => stmt instanceof Return)
+                        .accept(inferenceEngine);
+                    if (returnResult.success === false) {
+                        this.errors.push(`Type error in '${functionName}': ${returnResult.error}`);
+                        return;
+                    }
+                    const sub = this.coreHM.unify(returnResult.value, returnType);
+                    if (sub.success === false)
+                        throw Error(sub.error);
+                }
+                else {
+                    // Handles GuardedBody case
+                    for (const guard of equation.body) {
+                        // checks if condition expression in guard is a resolves to YuBoolean
+                        const condition = guard.condition.accept(inferenceEngine);
+                        if (condition.success === false)
+                            throw Error(condition.error);
+                        const conditionSub = this.coreHM.unify(condition.value, booleanType);
+                        if (conditionSub.success === false)
+                            throw Error(conditionSub.error);
+                        const bodyResult = guard.body.accept(inferenceEngine);
+                        if (bodyResult.success === false)
+                            throw Error(bodyResult.error);
+                        const sub = this.coreHM.unify(bodyResult.value, returnType);
+                        if (sub.success === false)
+                            throw Error(sub.error);
+                    }
+                }
+                this.environments.shift();
+            }
+            catch (error) {
+                this.errors.push(`Type error in '${functionName}': ${error.message}`);
+            }
+        }
+    }
+}
+export class TypeChecker {
+    signatureMap;
+    coreHM;
+    errors;
+    constructor() {
+        this.signatureMap = new Map();
+        this.coreHM = new CoreHM();
+        this.errors = [];
+    }
+    check(ast) {
+        const typeAliasMap = new Map();
+        const recordMap = new Map();
+        // Phase 1: Collect declarations
+        const collector = new DeclarationCollectorVisitor(this.errors, typeAliasMap, recordMap, this.signatureMap, this.coreHM);
+        for (const node of ast) {
+            try {
+                node.accept(collector);
+            }
+            catch (error) {
+                this.errors.push(error);
+            }
+        }
+        if (this.errors.length > 0)
+            return this.errors;
+        // Phase 2: Infer and check functions
+        this.inferencePass(ast);
+        return this.errors;
+    }
+    inferencePass(ast) {
+        // Step 1: Register all functions in env
+        const globalEnv = new Map(this.signatureMap);
+        const visitor1 = new FunctionRegistrarVisitor(globalEnv, this.signatureMap, this.coreHM);
+        for (const node of ast) {
+            node.accept(visitor1);
+        }
+        // Step 2: Infer and check each function
+        const visitor2 = new FunctionCheckerVisitor([globalEnv], this.signatureMap, this.coreHM, this.errors);
+        for (const node of ast) {
+            node.accept(visitor2);
+        }
+    }
+}
+export function showType(t) {
+    if (t.type === "TypeVar")
+        return t.name ?? `t${t.id}`;
+    if (isFunctionType(t)) {
+        const a = showType(t.args[0]);
+        const b = showType(t.args[1]);
+        const aDisp = isFunctionType(t.args[0]) ? `(${a})` : a;
+        return `${aDisp} -> ${b}`;
+    }
+    if (isListType(t))
+        return `[${showType(t.args[0])}]`;
+    if (isTupleType(t))
+        return `(${t.args.map(showType.bind(this)).join(", ")})`;
+    return t.args.length
+        ? `${t.name} ${t.args.map(showType.bind(this)).join(" ")}`
+        : t.name;
+}
+export function getReturnType(type) {
+    let t = type;
+    while (isFunctionType(t))
+        t = t.args[1];
+    return t;
+}
+export function getArgumentTypes(type) {
+    const args = [];
+    let t = type;
+    while (isFunctionType(t)) {
+        args.push(t.args[0]);
+        t = t.args[1];
+    }
+    return args;
+}
+export function getArity(type) {
+    return getArgumentTypes(type).length;
+}
+export function isUnguardedBody(body) {
+    return !Array.isArray(body);
+}
+export function isFunctionType(t) {
+    return t.type === "TypeConstructor" && t.name === "->" && t.args.length === 2;
+}
+export function isListType(t) {
+    return t.name === "List";
+}
+export function isTupleType(t) {
+    return t.name === "Tuple";
+}
+export function functionType(params, returnType) {
+    return {
+        type: "TypeConstructor",
+        name: "->",
+        args: [params, returnType],
+    };
+}
+export function listType(elementsType) {
+    return {
+        type: "TypeConstructor",
+        name: "List",
+        args: [elementsType],
+    };
+}
+//# sourceMappingURL=checker.js.map

package/dist/typechecker/checker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"checker.js","sourceRoot":"","sources":["../../src/typechecker/checker.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,QAAQ,EAIR,MAAM,GACP,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACjE,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAEnC,OAAO,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AA+CxE,MAAM,CAAC,MAAM,WAAW,GAAoB;IAC1C,IAAI,EAAE,iBAAiB;IACvB,IAAI,EAAE,WAAW;IACjB,IAAI,EAAE,EAAE;CACT,CAAC;AACF,MAAM,CAAC,MAAM,UAAU,GAAoB;IACzC,IAAI,EAAE,iBAAiB;IACvB,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,EAAE;CACT,CAAC;AACF,MAAM,CAAC,MAAM,UAAU,GAAoB;IACzC,IAAI,EAAE,iBAAiB;IACvB,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,EAAE;CACT,CAAC;AAEF,MAAM,OAAO,wBAAwB;IAEzB;IACA;IACA;IAHV,YACU,GAAgB,EAChB,YAAqC,EACrC,MAAc;QAFd,QAAG,GAAH,GAAG,CAAa;QAChB,iBAAY,GAAZ,YAAY,CAAyB;QACrC,WAAM,GAAN,MAAM,CAAQ;IACrB,CAAC;IACJ,aAAa,CAAC,IAAc;QAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QAC3C,IAAI,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACrD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YAC3C,UAAU,GAAG;gBACX,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,EAAE;gBACf,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,IAAI,GAAG,EAAE;aACvB,CAAC;YACF,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACzC,CAAC;QACD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,MAAM,EAAE,UAAU,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC9C,UAAU;qBACP,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,YAAY,QAAQ,CAAC;qBAC1C,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBAChB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;gBAClD,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;IACH,CAAC;CACF;AACD,MAAM,OAAO,sBAAsB;IAEvB;IACA;IACA;IACA;IAJV,YACU,YAA2B,EAC3B,YAAqC,EACrC,MAAc,EACd,MAAgB;QAHhB,iBAAY,GAAZ,YAAY,CAAe;QAC3B,iBAAY,GAAZ,YAAY,CAAyB;QACrC,WAAM,GAAN,MAAM,CAAQ;QACd,WAAM,GAAN,MAAM,CAAU;IACvB,CAAC;IACJ,aAAa,CAAC,IAAc;QAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QAC3C,IAAI,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACrD,oCAAoC;QACpC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;YACrC,MAAM,eAAe,GAAG,IAAI,eAAe,CACzC,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,YAAY,CAClB,CAAC;YACF,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;YAEtE,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;gBAC7B,IAAI,CAAC;oBACH,IAAI,cAAc,CAChB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,YAAY,EACjB,IAAI,EACJ,IAAI,CAAC,YAAY,EACjB,eAAe,CAChB,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,YAAY,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,gBAAsB,CAAC;YAC3B,IAAI,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClC,MAAM,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAC5D,kBAAkB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAClC,IAAI,CAAC,MAAM,CACT,IAAI,sBAAsB,CACxB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,MAAM,CACZ,CACF,CACF,CAAC;gBACF,MAAM,YAAY,GAAG,kBAAkB;qBACpC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,YAAY,MAAM,CAAC;qBACtC,MAAM,CAAC,eAAe,CAAC,CAAC;gBAC3B,IAAI,YAAY,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;oBACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kBAAkB,YAAY,MAAM,YAAY,CAAC,KAAK,EAAE,CACzD,CAAC;oBACF,OAAO;gBACT,CAAC;gBACD,gBAAgB,GAAG,YAAY,CAAC,KAAK,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACN,6CAA6C;gBAC7C,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,cAAc;YAC3D,CAAC;YAED,MAAM,YAAY,GAAG,UAAU,CAAC,WAAW,CACzC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,CAAC,EACxC,gBAAgB,CACjB,CAAC;YAEF,mEAAmE;YACnE,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC1B,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;YACxE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;YAChD,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC;QACpE,KAAK,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;YACzD,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,KAAK,aAAa,EAAE,CAAC;gBAC/C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,YAAY,wCAAwC,KAAK,EAAE,CAAC,CAAC;gBAChG,SAAS;YACX,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;gBAErD,MAAM,UAAU,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;gBAC3C,MAAM,YAAY,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAEhD,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;gBACrC,MAAM,eAAe,GAAG,IAAI,eAAe,CACzC,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,YAAY,CAClB,CAAC;gBACF,YAAY,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE;oBAClC,IAAI,CAAC;wBACH,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CACzB,IAAI,cAAc,CAChB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,YAAY,EACjB,OAAO,EACP,IAAI,CAAC,YAAY,EACjB,eAAe,CAChB,CACF,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kBAAkB,YAAY,MAAM,KAAK,CAAC,OAAO,EAAE,CACpD,CAAC;oBACJ,CAAC;gBACH,CAAC,CAAC,CAAC;gBACH,IAAI,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBACnC,MAAM,kBAAkB,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;oBAC7D,kBAAkB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAClC,IAAI,CAAC,MAAM,CACT,IAAI,sBAAsB,CACxB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,MAAM,CACZ,CACF,CACF,CAAC;oBACF,MAAM,YAAY,GAAG,kBAAkB;yBACpC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,YAAY,MAAM,CAAC;yBACtC,MAAM,CAAC,eAAe,CAAC,CAAC;oBAC3B,IAAI,YAAY,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;wBACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kBAAkB,YAAY,MAAM,YAAY,CAAC,KAAK,EAAE,CACzD,CAAC;wBACF,OAAO;oBACT,CAAC;oBAED,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;oBAC9D,IAAI,GAAG,CAAC,OAAO,KAAK,KAAK;wBAAE,MAAM,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBACN,2BAA2B;oBAC3B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;wBAClC,qEAAqE;wBAErE,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;wBAC1D,IAAI,SAAS,CAAC,OAAO,KAAK,KAAK;4BAAE,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;wBAE9D,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CACpC,SAAS,CAAC,KAAK,EACf,WAAW,CACZ,CAAC;wBACF,IAAI,YAAY,CAAC,OAAO,KAAK,KAAK;4BAAE,MAAM,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;wBAEpE,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;wBACtD,IAAI,UAAU,CAAC,OAAO,KAAK,KAAK;4BAAE,MAAM,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;wBAEhE,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;wBAC5D,IAAI,GAAG,CAAC,OAAO,KAAK,KAAK;4BAAE,MAAM,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;oBACpD,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC5B,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,YAAY,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACxE,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,WAAW;IACd,YAAY,CAA0B;IACtC,MAAM,CAAS;IACf,MAAM,CAAW;IAEzB;QACE,IAAI,CAAC,YAAY,GAAG,IAAI,GAAG,EAAsB,CAAC;QAClD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;IACD,KAAK,CAAC,GAAQ;QACZ,MAAM,YAAY,GAAG,IAAI,GAAG,EAAgB,CAAC;QAC7C,MAAM,SAAS,GAAG,IAAI,GAAG,EAAgB,CAAC;QAE1C,gCAAgC;QAChC,MAAM,SAAS,GAAG,IAAI,2BAA2B,CAC/C,IAAI,CAAC,MAAM,EACX,YAAY,EACZ,SAAS,EACT,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,MAAM,CACZ,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACzB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC;QAE/C,qCAAqC;QACrC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QAExB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IACD,aAAa,CAAC,GAAQ;QACpB,wCAAwC;QACxC,MAAM,SAAS,GAAG,IAAI,GAAG,CAAqB,IAAI,CAAC,YAAY,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAG,IAAI,wBAAwB,CAC3C,SAAS,EACT,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,MAAM,CACZ,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACxB,CAAC;QACD,wCAAwC;QACxC,MAAM,QAAQ,GAAG,IAAI,sBAAsB,CACzC,CAAC,SAAS,CAAC,EACX,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,MAAM,CACZ,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,QAAQ,CAAC,CAAO;IAC9B,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC;IAEtD,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;QACtB,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvD,OAAO,GAAG,KAAK,OAAO,CAAC,EAAE,CAAC;IAC5B,CAAC;IACD,IAAI,UAAU,CAAC,CAAC,CAAC;QAAE,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IACrD,IAAI,WAAW,CAAC,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;IAE7E,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM;QAClB,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;QAC1D,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAU;IACtC,IAAI,CAAC,GAAS,IAAI,CAAC;IACnB,OAAO,cAAc,CAAC,CAAC,CAAC;QAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,OAAO,CAAC,CAAC;AACX,CAAC;AACD,MAAM,UAAU,gBAAgB,CAAC,IAAU;IACzC,MAAM,IAAI,GAAW,EAAE,CAAC;IACxB,IAAI,CAAC,GAAS,IAAI,CAAC;IACnB,OAAO,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChB,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AACD,MAAM,UAAU,QAAQ,CAAC,IAAU;IACjC,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,IAAmC;IAEnC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,CAAO;IACpC,OAAO,CAAC,CAAC,IAAI,KAAK,iBAAiB,IAAI,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC;AAChF,CAAC;AACD,MAAM,UAAU,UAAU,CAAC,CAAO;IAChC,OAAO,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;AAC3B,CAAC;AACD,MAAM,UAAU,WAAW,CAAC,CAAO;IACjC,OAAO,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAY,EAAE,UAAgB;IACzD,OAAO;QACL,IAAI,EAAE,iBAAiB;QACvB,IAAI,EAAE,IAAI;QACV,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,CAAC;KAC3B,CAAC;AACJ,CAAC;AACD,MAAM,UAAU,QAAQ,CAAC,YAAkB;IACzC,OAAO;QACL,IAAI,EAAE,iBAAiB;QACvB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,CAAC,YAAY,CAAC;KACrB,CAAC;AACJ,CAAC"}

package/dist/typechecker/core.d.ts

@@ -0,0 +1,14 @@
+import { Environment, Result, Substitution, Type, TypeConstructor, TypeScheme, TypeVar } from "./checker.js";
+export declare class CoreHM {
+    private nextVarId;
+    freshVar(constraints?: string[]): TypeVar;
+    freeTypeVars(t: Type): Map<number, string[]>;
+    generalize(env: Environment, t: Type): TypeScheme;
+    instantiate(scheme: TypeScheme): Type;
+    applySubst(subst: Substitution, t: Type): TypeVar | TypeConstructor;
+    composeSubst(s1: Substitution, s2: Substitution): Substitution;
+    unify(t1: Type, t2: Type): Result<Substitution>;
+    unifyVar(v: TypeVar, t: Type): Result<Substitution>;
+    checkConstraint(constraintName: string, t: Type): void;
+    occurs(varId: number, t: Type): boolean;
+}

package/dist/typechecker/core.js

@@ -0,0 +1,142 @@
+import { typeClasses } from "../utils/types.js";
+import { showType, } from "./checker.js";
+export class CoreHM {
+    nextVarId = 0;
+    freshVar(constraints = []) {
+        return { type: "TypeVar", id: this.nextVarId++, constraints };
+    }
+    freeTypeVars(t) {
+        const freeVars = new Map();
+        const collect = (type) => {
+            if (type.type === "TypeVar") {
+                freeVars.set(type.id, type.constraints);
+            }
+            else if (type.type === "TypeConstructor") {
+                type.args.forEach(collect);
+            }
+        };
+        collect(t);
+        return freeVars;
+    }
+    generalize(env, t) {
+        const envFreeVars = new Set();
+        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 = [];
+        const constraints = new Map();
+        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 };
+    }
+    instantiate(scheme) {
+        const substitutions = new Map();
+        scheme.quantifiers.forEach((id) => {
+            const constraints = scheme.constraints.get(id) || [];
+            substitutions.set(id, this.freshVar(constraints));
+        });
+        return this.applySubst(substitutions, scheme.body);
+    }
+    applySubst(subst, t) {
+        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");
+    }
+    composeSubst(s1, s2) {
+        const result = new Map(s2);
+        for (const [id, type] of s1) {
+            result.set(id, this.applySubst(s2, type));
+        }
+        return result;
+    }
+    unify(t1, t2) {
+        if (t1.type === "TypeVar")
+            return this.unifyVar(t1, t2);
+        if (t2.type === "TypeVar")
+            return this.unifyVar(t2, t1);
+        if (t1.type === "TypeConstructor" && t2.type === "TypeConstructor") {
+            if (t1.name !== t2.name || t1.args.length !== t2.args.length) {
+                return {
+                    success: false,
+                    error: `Cannot unify ${showType(t1)} with ${showType(t2)}`,
+                };
+            }
+            let sub = new Map();
+            for (let i = 0; i < t1.args.length; i++) {
+                const arg1 = this.applySubst(sub, t1.args[i]);
+                const arg2 = this.applySubst(sub, t2.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` };
+    }
+    unifyVar(v, t) {
+        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]]) };
+    }
+    checkConstraint(constraintName, t) {
+        if (t.type === "TypeVar") {
+            if (!t.constraints.includes(constraintName)) {
+                t.constraints.push(constraintName);
+            }
+        }
+        else if (t.type === "TypeConstructor") {
+            const instances = typeClasses.get(constraintName);
+            if (!instances || !instances.includes(t.name)) {
+                throw new Error(`Type '${showType(t)}' is not an instance of '${constraintName}'`);
+            }
+        }
+    }
+    occurs(varId, t) {
+        if (t.type === "TypeVar")
+            return t.id === varId;
+        if (t.type === "TypeConstructor")
+            return t.args.some((a) => this.occurs(varId, a));
+        return false;
+    }
+}
+//# sourceMappingURL=core.js.map

package/dist/typechecker/core.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"core.js","sourceRoot":"","sources":["../../src/typechecker/core.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAGL,QAAQ,GAMT,MAAM,cAAc,CAAC;AAEtB,MAAM,OAAO,MAAM;IACT,SAAS,GAAG,CAAC,CAAC;IAEf,QAAQ,CAAC,cAAwB,EAAE;QACxC,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,IAAI,CAAC,SAAS,EAAE,EAAE,WAAW,EAAE,CAAC;IAChE,CAAC;IAEM,YAAY,CAAC,CAAO;QACzB,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAoB,CAAC;QAC7C,MAAM,OAAO,GAAG,CAAC,IAAU,EAAE,EAAE;YAC7B,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC5B,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1C,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;gBAC3C,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC;QACF,OAAO,CAAC,CAAC,CAAC,CAAC;QACX,OAAO,QAAQ,CAAC;IAClB,CAAC;IACM,UAAU,CAAC,GAAgB,EAAE,CAAO;QACzC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QACtC,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;YAClC,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtD,KAAK,MAAM,EAAE,IAAI,cAAc,CAAC,IAAI,EAAE,EAAE,CAAC;gBACvC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;oBACrC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,WAAW,GAAa,EAAE,CAAC;QACjC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAoB,CAAC;QAEhD,KAAK,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;YAClD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;gBACzB,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACrB,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtB,WAAW,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC;IACnE,CAAC;IAEM,WAAW,CAAC,MAAkB;QACnC,MAAM,aAAa,GAAG,IAAI,GAAG,EAAgB,CAAC;QAC9C,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;YAChC,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACrD,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACrD,CAAC;IAEM,UAAU,CAAC,KAAmB,EAAE,CAAO;QAC5C,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACzB,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACpC,OAAO,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;aAAM,IAAI,CAAC,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACxC,OAAO;gBACL,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;aACvD,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IAEM,YAAY,CAAC,EAAgB,EAAE,EAAgB;QACpD,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3B,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,EAAQ,EAAE,EAAQ;QAC7B,IAAI,EAAE,CAAC,IAAI,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACxD,IAAI,EAAE,CAAC,IAAI,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACxD,IAAI,EAAE,CAAC,IAAI,KAAK,iBAAiB,IAAI,EAAE,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACnE,IAAI,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC7D,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,gBAAgB,QAAQ,CAAC,EAAE,CAAC,SAAS,QAAQ,CAAC,EAAE,CAAC,EAAE;iBAC3D,CAAC;YACJ,CAAC;YACD,IAAI,GAAG,GAAiB,IAAI,GAAG,EAAE,CAAC;YAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACzC,IAAI,CAAC,SAAS,CAAC,OAAO;oBAAE,OAAO,SAAS,CAAC;gBACzC,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;QACvC,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAC;IAC7D,CAAC;IAEM,QAAQ,CAAC,CAAU,EAAE,CAAO;QACjC,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;YAC1C,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;QAC7C,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,qBAAqB,EAAE,CAAC;QAC1D,CAAC;QAED,KAAK,MAAM,UAAU,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACzB,KAAK,MAAM,UAAU,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;gBACvC,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YACtC,CAAC;YACD,oBAAoB;YACpB,MAAM,iBAAiB,GAAG;gBACxB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC;aACjD,CAAC;YACF,CAAC,CAAC,WAAW,GAAG,iBAAiB,CAAC;YAClC,CAAC,CAAC,WAAW,GAAG,iBAAiB,CAAC;QACpC,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACxD,CAAC;IAEM,eAAe,CAAC,cAAsB,EAAE,CAAO;QACpD,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC5C,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;aAAM,IAAI,CAAC,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAClD,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9C,MAAM,IAAI,KAAK,CACb,SAAS,QAAQ,CAAC,CAAC,CAAC,4BAA4B,cAAc,GAAG,CAClE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAEM,MAAM,CAAC,KAAa,EAAE,CAAO;QAClC,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS;YAAE,OAAO,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC;QAChD,IAAI,CAAC,CAAC,IAAI,KAAK,iBAAiB;YAC9B,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACnD,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/dist/typechecker/inference.d.ts

@@ -0,0 +1,69 @@
+import { Application, ApplicationPattern, ArithmeticBinaryOperation, ArithmeticUnaryOperation, AsPattern, AssignOperation, ASTNode, BitwiseBinaryOperation, BitwiseUnaryOperation, BooleanPrimitive, CharPrimitive, ComparisonOperation, CompositionExpression, ConsExpression, ConsPattern, ConstructorPattern, DataExpression, FieldExpression, For, FunctorPattern, Generator, If, Lambda, LetInExpression, ListBinaryOperation, ListComprehension, ListPattern, ListPrimitive, ListUnaryOperation, LiteralPattern, LogicalBinaryOperation, LogicalUnaryOperation, NilPrimitive, NumberPrimitive, Otherwise, Pattern, Print, Raise, RangeExpression, Return, StringOperation, StringPrimitive, Switch, SymbolPrimitive, TupleExpression, TuplePattern, TypeCast, UnifyOperation, UnionPattern, VariablePattern, Visitor, WildcardPattern, Yield } from "@yukigo/ast";
+import { Environment, Result, Type, TypeScheme } from "./checker.js";
+import { CoreHM } from "./core.js";
+export declare class PatternVisitor implements Visitor<void> {
+    private coreHM;
+    private signatureMap;
+    private expectedType;
+    private envs;
+    private inferenceEngine;
+    constructor(coreHM: CoreHM, signatureMap: Map<string, TypeScheme>, expectedType: Type, envs: Environment[], inferenceEngine: InferenceEngine);
+    visitVariablePattern(node: VariablePattern): void;
+    visitLiteralPattern(node: LiteralPattern): void;
+    visitApplicationPattern(node: ApplicationPattern): void;
+    visitTuplePattern(node: TuplePattern): void;
+    visitListPattern(node: ListPattern): void;
+    visitFunctorPattern(node: FunctorPattern): Result<Type>;
+    visitAsPattern(node: AsPattern): void;
+    visitWildcardPattern(node: WildcardPattern): Result<Type>;
+    visitUnionPattern(node: UnionPattern): Result<Type>;
+    visitConstructorPattern(node: ConstructorPattern): void;
+    visitConsPattern(node: ConsPattern): void;
+    visit(node: Pattern): void;
+}
+export declare class InferenceEngine implements Visitor<Result<Type>> {
+    private signatureMap;
+    private coreHM;
+    private envs;
+    constructor(signatureMap: Map<string, TypeScheme>, coreHM: CoreHM, envs: Environment[]);
+    visitNumberPrimitive(node: NumberPrimitive): Result<Type>;
+    visitBooleanPrimitive(node: BooleanPrimitive): Result<Type>;
+    visitStringPrimitive(node: StringPrimitive): Result<Type>;
+    visitListPrimitive(node: ListPrimitive): Result<Type>;
+    visitNilPrimitive(node: NilPrimitive): Result<Type>;
+    visitCharPrimitive(node: CharPrimitive): Result<Type>;
+    visitSymbolPrimitive(node: SymbolPrimitive): Result<Type>;
+    visitArithmeticUnaryOperation(node: ArithmeticUnaryOperation): Result<Type>;
+    visitArithmeticBinaryOperation(node: ArithmeticBinaryOperation): Result<Type>;
+    visitListUnaryOperation(node: ListUnaryOperation): Result<Type>;
+    visitListBinaryOperation(node: ListBinaryOperation): Result<Type>;
+    visitComparisonOperation(node: ComparisonOperation): Result<Type>;
+    visitLogicalBinaryOperation(node: LogicalBinaryOperation): Result<Type>;
+    visitLogicalUnaryOperation(node: LogicalUnaryOperation): Result<Type>;
+    visitBitwiseBinaryOperation(node: BitwiseBinaryOperation): Result<Type>;
+    visitBitwiseUnaryOperation(node: BitwiseUnaryOperation): Result<Type>;
+    visitStringOperation(node: StringOperation): Result<Type>;
+    visitUnifyOperation(node: UnifyOperation): Result<Type>;
+    visitAssignOperation(node: AssignOperation): Result<Type>;
+    visitTupleExpr(node: TupleExpression): Result<Type>;
+    visitFieldExpr(node: FieldExpression): Result<Type>;
+    visitDataExpr(node: DataExpression): Result<Type>;
+    visitConsExpr(node: ConsExpression): Result<Type>;
+    visitLetInExpr(node: LetInExpression): Result<Type>;
+    visitOtherwise(node: Otherwise): Result<Type>;
+    visitCompositionExpression(node: CompositionExpression): Result<Type>;
+    visitLambda(node: Lambda): Result<Type>;
+    visitApplication(node: Application): Result<Type>;
+    visitYield(node: Yield): Result<Type>;
+    visitRaise(node: Raise): Result<Type>;
+    visitIf(node: If): Result<Type>;
+    visitReturn(node: Return): Result<Type>;
+    visitTypeCast(node: TypeCast): Result<Type>;
+    visitPrint(node: Print): Result<Type>;
+    visitListComprehension(node: ListComprehension): Result<Type>;
+    visitGenerator(node: Generator): Result<Type>;
+    visitFor(node: For): Result<Type>;
+    visitSwitch(node: Switch): Result<Type>;
+    visitRangeExpression(node: RangeExpression): Result<Type>;
+    visit(node: ASTNode): Result<Type>;
+}