@algosail/lang 0.2.12 → 0.5.1

package/lib/typecheck/unify-type.js

@@ -0,0 +1,665 @@
+/**
+ * L3: унификация IR типов (RFC-typecheck-0.1 §5.5, RFC-0.1 §13 E1304/E1305/E1309).
+ */
+import * as diag from '../parse/diagnostics.js'
+import { quotationOutgoingEffectsEqual } from './effect-decl.js'
+
+/** Встроенные контейнеры List / Dict / Map (совпадает с BUILTIN_TYPE_CTORS в normalize-sig.js). */
+const BUILTIN_CONTAINER_CTORS = new Set(['List', 'Dict', 'Map'])
+
+/**
+ * @param {object} span
+ * @param {string} code
+ * @param {string} message
+ */
+function pushDiag (out, span, code, message) {
+  const d = { code, message }
+  if (span?.start) {
+    d.offset = span.start.offset
+    d.line = span.start.line
+    d.column = span.start.column
+  }
+  out.push(d)
+}
+
+/**
+ * @param {object} t
+ * @returns {string}
+ */
+export function formatIrType (t) {
+  if (!t) return '?'
+  switch (t.kind) {
+    case 'prim':
+      return t.name
+    case 'tvar':
+      return t.name
+    case 'stack_label':
+      return `~${t.name}`
+    case 'app':
+      return `${t.ctor}(${t.args.map(formatIrType).join(' ')})`
+    case 'adt':
+      return t.type
+    case 'opaque':
+      return t.name
+    case 'mod_adt':
+      return `~${t.module}/${t.type}`
+    case 'quote':
+      return 'Quote(...)'
+    case 'named_quote':
+      return `${t.prefix}:Quote(...)`
+    default:
+      return t.kind || '?'
+  }
+}
+
+/**
+ * @param {object} t
+ * @returns {boolean}
+ */
+export function irContainsStackLabel (t) {
+  if (!t || typeof t !== 'object') return false
+  switch (t.kind) {
+    case 'stack_label':
+      return true
+    case 'app':
+      return t.args.some(irContainsStackLabel)
+    case 'quote':
+    case 'named_quote':
+      return irContainsNormSigStackLabels(t.inner)
+    default:
+      return false
+  }
+}
+
+/**
+ * @param {import('./normalize-sig.js').NormalizedSignature} sig
+ * @returns {boolean}
+ */
+export function irSignatureContainsStackLabel (sig) {
+  return irContainsNormSigStackLabels(sig)
+}
+
+function irContainsNormSigStackLabels (sig) {
+  return [...sig.left, ...sig.right].some(irContainsStackLabel)
+}
+
+/**
+ * Сигнатура Quote из шаблона встроенного слова (builtin-signatures): без привязки к узлу исходника.
+ *
+ * @param {object | null} t
+ */
+function isBuiltinQuoteShell (t) {
+  return (
+    !!t &&
+    (t.kind === 'quote' || t.kind === 'named_quote') &&
+    t.source === null
+  )
+}
+
+/**
+ * Одна область имён на сигнатуру; вложенные quotation — новая область для tvar и stack_label.
+ *
+ * @param {import('./normalize-sig.js').NormalizedSignature} sig
+ * @returns {import('./normalize-sig.js').NormalizedSignature}
+ */
+export function shareTvarsInSignature (sig) {
+  return shareTvarsInSignatureWithMap(sig, new Map(), new Map())
+}
+
+/**
+ * @param {import('./normalize-sig.js').NormalizedSignature} sig
+ * @param {Map<string, object>} tvarByName
+ * @param {Map<string, object>} stackLabelByName
+ */
+function shareTvarsInSignatureWithMap (sig, tvarByName, stackLabelByName) {
+  function walkType (t) {
+    if (!t || typeof t !== 'object') return t
+    switch (t.kind) {
+      case 'tvar': {
+        let c = tvarByName.get(t.name)
+        if (!c) {
+          c = { kind: 'tvar', name: t.name, source: t.source, decl: t.decl ?? null }
+          tvarByName.set(t.name, c)
+        }
+        return c
+      }
+      case 'stack_label': {
+        let c = stackLabelByName.get(t.name)
+        if (!c) {
+          c = { kind: 'stack_label', name: t.name, source: t.source }
+          stackLabelByName.set(t.name, c)
+        }
+        return c
+      }
+      case 'app':
+        return { kind: 'app', ctor: t.ctor, args: t.args.map(walkType), source: t.source, decl: t.decl }
+      case 'quote': {
+        return {
+          kind: 'quote',
+          source: t.source,
+          inner: shareTvarsInSignatureWithMap(t.inner, new Map(), new Map())
+        }
+      }
+      case 'named_quote': {
+        return {
+          kind: 'named_quote',
+          prefix: t.prefix,
+          source: t.source,
+          inner: shareTvarsInSignatureWithMap(t.inner, new Map(), new Map())
+        }
+      }
+      default:
+        return t
+    }
+  }
+  const out = {
+    left: sig.left.map(walkType),
+    right: sig.right.map(walkType),
+    effectsAdd: sig.effectsAdd,
+    effectsRemove: sig.effectsRemove
+  }
+  if (sig.adtEliminator != null) out.adtEliminator = sig.adtEliminator
+  return out
+}
+
+/**
+ * Свежая копия сигнатуры для одного вызова: новые tvar и stack_label, согласованные внутри копии.
+ *
+ * @param {import('./normalize-sig.js').NormalizedSignature} sig
+ * @returns {import('./normalize-sig.js').NormalizedSignature}
+ */
+export function freshInstanceSignature (sig) {
+  /** @type {WeakMap<object, object>} */
+  const tvarMap = new WeakMap()
+  /** @type {WeakMap<object, object>} */
+  const stackLabelMap = new WeakMap()
+
+  function cloneType (t) {
+    if (!t || typeof t !== 'object') return t
+    switch (t.kind) {
+      case 'tvar': {
+        let n = tvarMap.get(t)
+        if (!n) {
+          n = { kind: 'tvar', name: t.name, source: t.source, decl: t.decl ?? null }
+          tvarMap.set(t, n)
+        }
+        return n
+      }
+      case 'stack_label': {
+        let n = stackLabelMap.get(t)
+        if (!n) {
+          n = { kind: 'stack_label', name: t.name, source: t.source }
+          stackLabelMap.set(t, n)
+        }
+        return n
+      }
+      case 'app':
+        return {
+          kind: 'app',
+          ctor: t.ctor,
+          args: t.args.map(cloneType),
+          source: t.source,
+          decl: t.decl
+        }
+      case 'quote':
+        return { kind: 'quote', source: t.source, inner: cloneNormSig(t.inner) }
+      case 'named_quote':
+        return {
+          kind: 'named_quote',
+          prefix: t.prefix,
+          source: t.source,
+          inner: cloneNormSig(t.inner)
+        }
+      default:
+        return t
+    }
+  }
+
+  function cloneNormSig (s) {
+    const o = {
+      left: s.left.map(cloneType),
+      right: s.right.map(cloneType),
+      effectsAdd: s.effectsAdd,
+      effectsRemove: s.effectsRemove
+    }
+    if (s.adtEliminator != null) o.adtEliminator = s.adtEliminator
+    return o
+  }
+
+  return cloneNormSig(sig)
+}
+
+/**
+ * @param {object} t
+ * @param {WeakMap<object, object>} subst
+ * @returns {object}
+ */
+export function derefType (t, subst) {
+  let u = t
+  while (
+    u &&
+    (u.kind === 'tvar' || u.kind === 'stack_label') &&
+    subst.has(u)
+  ) {
+    u = subst.get(u)
+  }
+  return u
+}
+
+/**
+ * @param {object} v — tvar
+ * @param {object} t
+ * @param {WeakMap<object, object>} subst
+ * @returns {boolean}
+ */
+function occursIn (v, t, subst) {
+  const w = derefType(t, subst)
+  if (!w) return false
+  if (w.kind === 'tvar' || w.kind === 'stack_label') {
+    if (w === v) return true
+    if (subst.has(w)) return occursIn(v, subst.get(w), subst)
+    return false
+  }
+  if (w.kind === 'app') return w.args.some((a) => occursIn(v, a, subst))
+  if (w.kind === 'quote' || w.kind === 'named_quote') {
+    const s = w.inner
+    return [...s.left, ...s.right].some((x) => occursIn(v, x, subst))
+  }
+  return false
+}
+
+/**
+ * @param {object} a
+ * @param {object} b
+ * @param {WeakMap<object, object>} subst
+ * @param {object[]} diags
+ * @param {object | null} span
+ * @returns {boolean}
+ */
+export function unifyTypes (a, b, subst, diags, span) {
+  a = derefType(a, subst)
+  b = derefType(b, subst)
+  if (a === b) return true
+  if (a.kind === 'tvar') {
+    if (occursIn(a, b, subst)) {
+      pushDiag(diags, span, 'E1305', diag.e1305OccursCheck(a.name))
+      return false
+    }
+    subst.set(a, b)
+    return true
+  }
+  if (b.kind === 'tvar') {
+    if (occursIn(b, a, subst)) {
+      pushDiag(diags, span, 'E1305', diag.e1305OccursCheck(b.name))
+      return false
+    }
+    subst.set(b, a)
+    return true
+  }
+  if (a.kind === 'stack_label' && b.kind === 'stack_label') {
+    if (a.name !== b.name) {
+      pushDiag(
+        diags,
+        span,
+        'E1304',
+        diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+      )
+      return false
+    }
+    if (a !== b) subst.set(a, b)
+    return true
+  }
+  if (a.kind === 'stack_label') {
+    if (occursIn(a, b, subst)) {
+      pushDiag(
+        diags,
+        span,
+        'E1305',
+        diag.e1305OccursCheck(`~${a.name}`)
+      )
+      return false
+    }
+    subst.set(a, b)
+    return true
+  }
+  if (b.kind === 'stack_label') {
+    if (occursIn(b, a, subst)) {
+      pushDiag(
+        diags,
+        span,
+        'E1305',
+        diag.e1305OccursCheck(`~${b.name}`)
+      )
+      return false
+    }
+    subst.set(b, a)
+    return true
+  }
+  if (a.kind !== b.kind) {
+    if (
+      (a.kind === 'quote' && b.kind === 'named_quote') ||
+      (a.kind === 'named_quote' && b.kind === 'quote')
+    ) {
+      return unifyNormSigQuoteIO(a.inner, b.inner, subst, diags, span, {
+        ignoreOutgoingEffects:
+          isBuiltinQuoteShell(a) || isBuiltinQuoteShell(b)
+      })
+    }
+    pushDiag(
+      diags,
+      span,
+      'E1304',
+      diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+    )
+    return false
+  }
+  switch (a.kind) {
+    case 'prim':
+      if (a.name !== b.name) {
+        pushDiag(
+          diags,
+          span,
+          'E1304',
+          diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+        )
+        return false
+      }
+      return true
+    case 'opaque':
+      if (a.name !== b.name) {
+        pushDiag(
+          diags,
+          span,
+          'E1304',
+          diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+        )
+        return false
+      }
+      return true
+    case 'app':
+      if (a.ctor !== b.ctor || a.args.length !== b.args.length) {
+        pushDiag(
+          diags,
+          span,
+          'E1304',
+          diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+        )
+        return false
+      }
+      if (BUILTIN_CONTAINER_CTORS.has(a.ctor)) {
+        for (let i = 0; i < a.args.length; i++) {
+          const nested = []
+          if (!unifyTypes(a.args[i], b.args[i], subst, nested, span)) {
+            if (
+              nested.length === 1 &&
+              nested[0].code === 'E1304'
+            ) {
+              // Порядок (a, b) совпадает с вложенным e1304TypeMismatch для unifyTypes(a.args[i], b.args[i]).
+              pushDiag(
+                diags,
+                span,
+                'E1309',
+                diag.e1309ContainerContract(
+                  a.ctor,
+                  formatIrType(derefType(a.args[i], subst)),
+                  formatIrType(derefType(b.args[i], subst))
+                )
+              )
+            } else {
+              for (const nd of nested) {
+                diags.push(nd)
+              }
+            }
+            return false
+          }
+        }
+        return true
+      }
+      for (let i = 0; i < a.args.length; i++) {
+        if (!unifyTypes(a.args[i], b.args[i], subst, diags, span)) return false
+      }
+      return true
+    case 'adt':
+      if (a.type !== b.type || a.path !== b.path) {
+        pushDiag(
+          diags,
+          span,
+          'E1304',
+          diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+        )
+        return false
+      }
+      return true
+    case 'mod_adt':
+      if (a.module !== b.module || a.type !== b.type) {
+        pushDiag(
+          diags,
+          span,
+          'E1304',
+          diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+        )
+        return false
+      }
+      return true
+    case 'quote':
+      return unifyNormSigQuoteIO(a.inner, b.inner, subst, diags, span, {
+        ignoreOutgoingEffects: isBuiltinQuoteShell(a) || isBuiltinQuoteShell(b)
+      })
+    case 'named_quote':
+      if (a.prefix !== b.prefix) {
+        pushDiag(
+          diags,
+          span,
+          'E1304',
+          diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+        )
+        return false
+      }
+      return unifyNormSigQuoteIO(a.inner, b.inner, subst, diags, span, {
+        ignoreOutgoingEffects: isBuiltinQuoteShell(a) || isBuiltinQuoteShell(b)
+      })
+    default:
+      pushDiag(
+        diags,
+        span,
+        'E1304',
+        diag.e1304TypeMismatch(formatIrType(a), formatIrType(b))
+      )
+      return false
+  }
+}
+
+/**
+ * @param {object} t
+ * @param {WeakMap<object, object>} subst
+ * @returns {object}
+ */
+export function applySubstDeep (t, subst) {
+  const u = derefType(t, subst)
+  if (!u) return u
+  switch (u.kind) {
+    case 'tvar':
+    case 'stack_label':
+      return derefType(u, subst)
+    case 'app':
+      return {
+        kind: 'app',
+        ctor: u.ctor,
+        args: u.args.map((x) => applySubstDeep(x, subst)),
+        source: u.source,
+        decl: u.decl
+      }
+    case 'quote':
+      return {
+        kind: 'quote',
+        source: u.source,
+        inner: applySubstToNormSig(u.inner, subst)
+      }
+    case 'named_quote':
+      return {
+        kind: 'named_quote',
+        prefix: u.prefix,
+        source: u.source,
+        inner: applySubstToNormSig(u.inner, subst)
+      }
+    default:
+      return u
+  }
+}
+
+/**
+ * @param {import('./normalize-sig.js').NormalizedSignature} sig
+ * @param {WeakMap<object, object>} subst
+ * @returns {import('./normalize-sig.js').NormalizedSignature}
+ */
+export function applySubstToNormSig (sig, subst) {
+  const o = {
+    left: sig.left.map((x) => applySubstDeep(x, subst)),
+    right: sig.right.map((x) => applySubstDeep(x, subst)),
+    effectsAdd: sig.effectsAdd,
+    effectsRemove: sig.effectsRemove
+  }
+  if (sig.adtEliminator != null) o.adtEliminator = sig.adtEliminator
+  return o
+}
+
+/**
+ * @param {object[]} stack
+ * @param {WeakMap<object, object>} subst
+ * @returns {object[]}
+ */
+export function applySubstToStack (stack, subst) {
+  return stack.map((x) => applySubstDeep(x, subst))
+}
+
+/**
+ * Унификация вложенных сигнатур Quote (I -> O) (RFC-typecheck-0.1 §5.6).
+ *
+ * @param {import('./normalize-sig.js').NormalizedSignature} ia
+ * @param {import('./normalize-sig.js').NormalizedSignature} ib
+ * @param {WeakMap<object, object>} subst
+ * @param {object[]} diags
+ * @param {object | null} span
+ * @param {{ ignoreOutgoingEffects?: boolean }} [opts]
+ * @returns {boolean}
+ */
+/**
+ * Выходные слоты quotation (RFC-0.1 §6.1.5): имена ~s / ~b — метки региона стека;
+ * шаблон eliminator-а использует ~b, пользовательская цитата — ~s.
+ *
+ * @param {object} a
+ * @param {object} b
+ * @param {WeakMap<object, object>} subst
+ * @param {object[]} diags
+ * @param {object | null} span
+ * @returns {boolean}
+ */
+function unifyQuotationOutputSlots (a, b, subst, diags, span) {
+  a = derefType(a, subst)
+  b = derefType(b, subst)
+  if (a.kind === 'stack_label' && b.kind === 'stack_label' && a.name !== b.name) {
+    if (occursIn(a, b, subst)) {
+      pushDiag(
+        diags,
+        span,
+        'E1305',
+        diag.e1305OccursCheck(`~${a.name}`)
+      )
+      return false
+    }
+    subst.set(a, b)
+    return true
+  }
+  return unifyTypes(a, b, subst, diags, span)
+}
+
+export function unifyNormSigQuoteIO (ia, ib, subst, diags, span, opts) {
+  const ignoreOutgoingEffects = opts?.ignoreOutgoingEffects === true
+  if (ia.left.length !== ib.left.length) {
+    pushDiag(
+      diags,
+      span,
+      'E1303',
+      diag.e1303StackEffect(
+        `${ib.left.length} слотов входа quotation`,
+        `${ia.left.length} слотов`
+      )
+    )
+    return false
+  }
+  for (let i = 0; i < ia.left.length; i++) {
+    if (!unifyTypes(ia.left[i], ib.left[i], subst, diags, span)) return false
+  }
+  if (ia.right.length !== ib.right.length) {
+    pushDiag(
+      diags,
+      span,
+      'E1303',
+      diag.e1303StackEffect(
+        `${ib.right.length} слотов выхода quotation`,
+        `${ia.right.length} слотов`
+      )
+    )
+    return false
+  }
+  for (let i = 0; i < ia.right.length; i++) {
+    if (!unifyQuotationOutputSlots(ia.right[i], ib.right[i], subst, diags, span)) {
+      return false
+    }
+  }
+  if (
+    !ignoreOutgoingEffects &&
+    !quotationOutgoingEffectsEqual(ia, ib)
+  ) {
+    pushDiag(
+      diags,
+      span,
+      'E1304',
+      diag.e1304TypeMismatch(
+        'совпадение маркеров эффектов вложенной quotation_sig',
+        'разные +Async/+Fail/-Async/-Fail'
+      )
+    )
+    return false
+  }
+  return true
+}
+
+/**
+ * Сравнение итогового стека с объявленным выходом (длина и покомпонентная унификация).
+ *
+ * @param {object[]} stackBottomToTop
+ * @param {object[]} expectedRight
+ * @param {WeakMap<object, object>} subst
+ * @param {object[]} diags
+ * @param {object | null} span
+ * @returns {boolean}
+ */
+export function unifyStackWithOutput (stackBottomToTop, expectedRight, subst, diags, span) {
+  if (stackBottomToTop.length !== expectedRight.length) {
+    pushDiag(
+      diags,
+      span,
+      'E1303',
+      diag.e1303StackEffect(
+        `${expectedRight.length} слотов`,
+        `${stackBottomToTop.length} слотов`
+      )
+    )
+    return false
+  }
+  for (let i = 0; i < expectedRight.length; i++) {
+    if (
+      !unifyTypes(
+        stackBottomToTop[i],
+        expectedRight[i],
+        subst,
+        diags,
+        span
+      )
+    ) {
+      return false
+    }
+  }
+  return true
+}