@algosail/lang 0.2.11 → 0.5.0

package/lib/codegen/emit-builtin.js

@@ -0,0 +1,371 @@
+/**
+ * L5 этап 7: комбинаторы `builtin_word` RFC-builtins-0.1 §4.4–4.5 (dip, keep, bi, tri, spread, both).
+ * Перестановки §4.1–4.2 — в emit-body.js (`BUILTIN_SHUFFLE_EXPR`).
+ *
+ * Расширение: новый builtin в RFC → `builtins-set.js`, `builtin-signatures.js`,
+ * при необходимости `bind-values.js`, затем маппинг здесь или в shuffle-таблице.
+ */
+
+/**
+ * @typedef {{
+ *   idToJs: Map<string, string>
+ *   lines: string[]
+ *   genName: () => string
+ *   genQuoteName: () => string
+ *   quoteMetaByValueId: Map<string, import('./emit-body.js').QuoteEmitMeta>
+ *   slotExprByName: Map<string, string>
+ *   strict: boolean
+ *   wordLabel: string
+ *   entryParamCounter: { n: number }
+ *   resolveQualifiedCallee?: (moduleAlias: string, wordName: string) => string
+ *   callerAsync: boolean
+ *   calleeAsyncResolver?: import('./emit-body.js').CalleeAsyncResolver | null
+ *   sumEliminatorNames?: Set<string> | null
+ * }} CombinatorEmitEnv
+ */
+
+export const COMBINATOR_BUILTIN_NAMES = new Set([
+  'dip',
+  'keep',
+  'bi',
+  'tri',
+  'spread',
+  'both'
+])
+
+/**
+ * Изолированное исполнение тела quotation: как inline-ветка `call` (RFC-compile §6.3).
+ *
+ * @param {Record<string, unknown>[]} innerSteps
+ * @param {string[]} innerEntryStackIds
+ * @param {string[]} innerArgIds id значений снизу вверх под quotation
+ * @param {string[]} topResultIds id верхней части пост-стека под результат inner (дно→вершина)
+ * @param {CombinatorEmitEnv} env
+ * @param {(steps: Record<string, unknown>[], e: object) => string[]} processIrSteps
+ */
+export function mapInlineQuoteResults (
+  innerSteps,
+  innerEntryStackIds,
+  innerArgIds,
+  topResultIds,
+  env,
+  processIrSteps
+) {
+  if (innerArgIds.length !== innerEntryStackIds.length) {
+    throw new Error(
+      `emit-builtin[${env.wordLabel}]: несовпадение длины innerArgIds и innerEntryStackIds`
+    )
+  }
+  const merged = new Map(env.idToJs)
+  for (let i = 0; i < innerEntryStackIds.length; i++) {
+    const aid = innerArgIds[i]
+    const j = env.idToJs.get(aid)
+    if (j == null) {
+      throw new Error(`emit-builtin[${env.wordLabel}]: нет JS для inner arg ${aid}`)
+    }
+    merged.set(innerEntryStackIds[i], j)
+  }
+  const innerEnv = {
+    idToJs: merged,
+    lines: env.lines,
+    genName: env.genName,
+    genQuoteName: env.genQuoteName,
+    quoteMetaByValueId: env.quoteMetaByValueId,
+    slotExprByName: env.slotExprByName,
+    strict: env.strict,
+    wordLabel: env.wordLabel,
+    entryParamCounter: env.entryParamCounter,
+    resolveQualifiedCallee: env.resolveQualifiedCallee,
+    callerAsync: env.callerAsync,
+    calleeAsyncResolver: env.calleeAsyncResolver,
+    sumEliminatorNames: env.sumEliminatorNames ?? null,
+    initialStackIds:
+      innerEntryStackIds.length > 0 ? [...innerEntryStackIds] : undefined
+  }
+  const innerFinalIds = processIrSteps(innerSteps, innerEnv)
+  if (innerFinalIds.length !== topResultIds.length) {
+    throw new Error(
+      `emit-builtin[${env.wordLabel}]: внутри quotation ожидалось ${topResultIds.length} выходов, есть ${innerFinalIds.length}`
+    )
+  }
+  for (let i = 0; i < topResultIds.length; i++) {
+    const rid = topResultIds[i]
+    const innerId = innerFinalIds[i]
+    const j = merged.get(innerId)
+    if (j == null) {
+      throw new Error(
+        `emit-builtin[${env.wordLabel}]: нет JS для внутреннего id ${innerId}`
+      )
+    }
+    env.idToJs.set(rid, j)
+  }
+}
+
+/**
+ * Quote как значение (после вызова слова): изолированный вызов как у `call` invoke (RFC-builtins §4.6 Iso).
+ *
+ * @param {string} quoteId
+ * @param {string[]} innerArgIds
+ * @param {string[]} topTargetIds
+ * @param {number} innerOutLen
+ * @param {CombinatorEmitEnv} env
+ * @param {boolean} calleeAsyncInvoke
+ */
+function mapInvokeCombinatorQuote (
+  quoteId,
+  innerArgIds,
+  topTargetIds,
+  innerOutLen,
+  env,
+  calleeAsyncInvoke
+) {
+  const quoteJs = env.idToJs.get(quoteId)
+  if (quoteJs == null) {
+    throw new Error(
+      `emit-builtin[${env.wordLabel}]: invoke — нет JS для quote id ${quoteId}`
+    )
+  }
+  const callArgExprs = innerArgIds.map((id) => {
+    const j = env.idToJs.get(id)
+    if (j == null) {
+      throw new Error(`emit-builtin[${env.wordLabel}]: invoke — нет JS для arg ${id}`)
+    }
+    return j
+  })
+  const paren = callArgExprs.length ? `(${callArgExprs.join(', ')})` : '()'
+  let callExpr = `${quoteJs}${paren}`
+  const useAwait = env.callerAsync && calleeAsyncInvoke
+  if (useAwait) {
+    callExpr = `await ${callExpr}`
+  }
+  const m = innerOutLen
+  if (m !== topTargetIds.length) {
+    throw new Error(
+      `emit-builtin[${env.wordLabel}]: invoke — несовпадение m и topTargetIds (${m} vs ${topTargetIds.length})`
+    )
+  }
+  if (m === 0) {
+    env.lines.push(`${callExpr};`)
+  } else if (m === 1) {
+    const nm = env.genName()
+    env.lines.push(`const ${nm} = ${callExpr};`)
+    env.idToJs.set(topTargetIds[0], nm)
+  } else {
+    const names = topTargetIds.map(() => env.genName())
+    env.lines.push(`const [${names.join(', ')}] = ${callExpr};`)
+    for (let i = 0; i < m; i++) {
+      env.idToJs.set(topTargetIds[i], names[i])
+    }
+  }
+}
+
+/**
+ * @param {object} part
+ * @param {string} quoteId
+ * @param {string[]} innerArgIds
+ * @param {string[]} topTargetIds
+ * @param {CombinatorEmitEnv} env
+ * @param {(steps: Record<string, unknown>[], e: object) => string[]} processIrSteps
+ * @param {Record<string, unknown>} node
+ */
+function runCombinatorQuoteBranch (
+  part,
+  quoteId,
+  innerArgIds,
+  topTargetIds,
+  env,
+  processIrSteps,
+  node
+) {
+  const k = /** @type {{ kind?: string }} */ (part).kind
+  if (k === 'invoke') {
+    const p = /** @type {{ innerInLen: number, innerOutLen: number }} */ (part)
+    if (innerArgIds.length !== p.innerInLen) {
+      throw new Error(
+        `emit-builtin[${env.wordLabel}]: invoke — ожидалось ${p.innerInLen} argIds`
+      )
+    }
+    const calleeAsyncInvoke = node.calleeAsync === true
+    mapInvokeCombinatorQuote(
+      quoteId,
+      innerArgIds,
+      topTargetIds,
+      p.innerOutLen,
+      env,
+      calleeAsyncInvoke
+    )
+    return
+  }
+  const inl = /** @type {{ innerSteps: Record<string, unknown>[], innerEntryStackIds: string[] }} */ (
+    part
+  )
+  mapInlineQuoteResults(
+    inl.innerSteps,
+    inl.innerEntryStackIds,
+    innerArgIds,
+    topTargetIds,
+    env,
+    processIrSteps
+  )
+}
+
+/**
+ * @param {string} name
+ * @param {Record<string, unknown>} node
+ * @param {{ argIds: string[], resultIds: string[] }} vb
+ * @param {number} L
+ * @param {CombinatorEmitEnv} env
+ * @param {(steps: Record<string, unknown>[], e: object) => string[]} processIrSteps
+ * @returns {boolean}
+ */
+export function tryEmitCombinatorBuiltin (
+  name,
+  node,
+  vb,
+  L,
+  env,
+  processIrSteps
+) {
+  if (!COMBINATOR_BUILTIN_NAMES.has(name)) {
+    return false
+  }
+
+  const cmeta = node.combinatorInlineMeta
+  if (cmeta == null || typeof cmeta !== 'object') {
+    throw new Error(`emit-builtin[${env.wordLabel}]: комбинатор без combinatorInlineMeta`)
+  }
+  const parts = /** @type {{ parts?: object[] }} */ (cmeta).parts
+  if (!Array.isArray(parts)) {
+    throw new Error(`emit-builtin[${env.wordLabel}]: combinatorInlineMeta без parts`)
+  }
+
+  const baseLen = L - vb.argIds.length
+  const topOutIds = vb.resultIds.slice(baseLen)
+  const a = vb.argIds
+
+  switch (name) {
+    case 'dip': {
+      if (a.length !== 3 || topOutIds.length !== 2 || parts.length !== 1) {
+        throw new Error(`emit-builtin[${env.wordLabel}]: dip — неверные привязки`)
+      }
+      runCombinatorQuoteBranch(
+        parts[0],
+        a[2],
+        [a[0]],
+        [topOutIds[0]],
+        env,
+        processIrSteps,
+        node
+      )
+      break
+    }
+    case 'keep': {
+      if (a.length !== 3 || topOutIds.length !== 2 || parts.length !== 1) {
+        throw new Error(`emit-builtin[${env.wordLabel}]: keep — неверные привязки`)
+      }
+      runCombinatorQuoteBranch(
+        parts[0],
+        a[2],
+        [a[0], a[1]],
+        [topOutIds[0]],
+        env,
+        processIrSteps,
+        node
+      )
+      break
+    }
+    case 'bi': {
+      if (a.length !== 4 || topOutIds.length !== 3 || parts.length !== 2) {
+        throw new Error(`emit-builtin[${env.wordLabel}]: bi — неверные привязки`)
+      }
+      runCombinatorQuoteBranch(
+        parts[0],
+        a[2],
+        [a[1]],
+        [topOutIds[1]],
+        env,
+        processIrSteps,
+        node
+      )
+      runCombinatorQuoteBranch(
+        parts[1],
+        a[3],
+        [a[1]],
+        [topOutIds[2]],
+        env,
+        processIrSteps,
+        node
+      )
+      break
+    }
+    case 'tri': {
+      if (a.length !== 5 || topOutIds.length !== 4 || parts.length !== 3) {
+        throw new Error(`emit-builtin[${env.wordLabel}]: tri — неверные привязки`)
+      }
+      for (let k = 0; k < 3; k++) {
+        runCombinatorQuoteBranch(
+          parts[k],
+          a[2 + k],
+          [a[1]],
+          [topOutIds[1 + k]],
+          env,
+          processIrSteps,
+          node
+        )
+      }
+      break
+    }
+    case 'spread': {
+      if (a.length !== 5 || topOutIds.length !== 3 || parts.length !== 2) {
+        throw new Error(`emit-builtin[${env.wordLabel}]: spread — неверные привязки`)
+      }
+      runCombinatorQuoteBranch(
+        parts[0],
+        a[3],
+        [a[1]],
+        [topOutIds[1]],
+        env,
+        processIrSteps,
+        node
+      )
+      runCombinatorQuoteBranch(
+        parts[1],
+        a[4],
+        [a[2]],
+        [topOutIds[2]],
+        env,
+        processIrSteps,
+        node
+      )
+      break
+    }
+    case 'both': {
+      if (a.length !== 4 || topOutIds.length !== 3 || parts.length !== 1) {
+        throw new Error(`emit-builtin[${env.wordLabel}]: both — неверные привязки`)
+      }
+      runCombinatorQuoteBranch(
+        parts[0],
+        a[3],
+        [a[1]],
+        [topOutIds[1]],
+        env,
+        processIrSteps,
+        node
+      )
+      runCombinatorQuoteBranch(
+        parts[0],
+        a[3],
+        [a[2]],
+        [topOutIds[2]],
+        env,
+        processIrSteps,
+        node
+      )
+      break
+    }
+    default:
+      return false
+  }
+
+  return true
+}