@scriptc/compiler 0.4.2 → 0.4.4

package/dist/ir/validate.js

@@ -4,12 +4,13 @@
  * an ICE that points at source is a gift to whoever debugs it.
  */
 import { validByteNumberToken } from "./byte-numbers.js";
-import { arrayOf, BYTES_ELEMENT_NAME, BOOL, BYTES_U8, bytesOf, canAdaptDynFuncTo, canDynCheckTo, canConvertToDyn, DYN, DYN_HANDLE_KINDS, F64, ffiClassType, ffiSourceParamTypes, isDirentShape, isDynTypedRefType, isFfiCallbackParam, isFfiContextParam, isFfiReleaseParam, isJsonSafeType, isJsonStringifySafeType, isRefCounted, isSortValuesElement, isSupportedArrayElem, isSupportedIndexValue, isSupportedMapKey, isSupportedMapValue, isSupportedSetElem, isUnitType, REF_TRUTHY_KINDS, RUNTIME_EMITTER_CLASS, RUNTIME_ERROR_CLASSES, RUNTIME_STREAM_CLASSES, shapeHasAccessorSlots, STRING, typeEquals, typeKey, maybePromiseUnionArmsOk, VOID, } from "./ir.js";
+import { arrayOf, iterableArmsElem, BYTES_ELEMENT_NAME, BOOL, BYTES_U8, bytesOf, canAdaptDynFuncTo, canDynCheckTo, canConvertToDyn, DYN, DYN_HANDLE_KINDS, F64, ffiClassType, ffiSourceParamTypes, isDirentShape, isDynTypedRefType, isFfiCallbackParam, isFfiContextParam, isFfiReleaseParam, isJsonSafeType, isJsonStringifySafeType, isRefCounted, isSortValuesElement, isSupportedArrayElem, isSupportedIndexValue, isSupportedMapKey, isSupportedMapValue, isSupportedSetElem, isUnitType, REF_TRUTHY_KINDS, RUNTIME_EMITTER_CLASS, RUNTIME_ERROR_CLASSES, RUNTIME_STREAM_CLASSES, shapeHasAccessorSlots, STRING, typeEquals, typeKey, maybePromiseUnionArmsOk, unionContainerArmsOk, VOID, } from "./ir.js";
 import { DYN_CLASS_PROPERTIES, callSiteReturnType, classValueConstructor } from "./ir.js";
 import { unionWideningTags } from "./analysis.js";
 import { alwaysReturns } from "./control-flow.js";
 import { dynamicOperationOf, functionTier } from "./tiers.js";
 import { LIB_FN_SIGS, pipelinePumpedRuns } from "./builtin-signatures.js";
+import { ALWAYS_THROW_LIB_FNS } from "./builtin-effects.js";
 /** The promisified crypto calls' value kinds besides checked-dynamic. */
 const CRYPTO_PROMISE_VALUES = {
     "crypto.randomBytesPromise": "buffer",
@@ -40,6 +41,8 @@ export function validateModule(mod) {
     const moduleLoc = { file: mod.sourceFile, start: 0, end: 0 };
     if (mod.workers !== undefined && typeof mod.workers !== "boolean")
         errors.push({ message: "worker execution mode must be a boolean", loc: moduleLoc });
+    if (mod.declBoundary !== undefined && mod.declBoundary !== true)
+        errors.push({ message: "declBoundary must be true when present", loc: moduleLoc });
     if (mod.lifetimeRegions !== undefined && mod.lifetimeRegions !== true)
         errors.push({ message: "lifetimeRegions must be true when present", loc: moduleLoc });
     if (mod.workers && mod.lib !== undefined)
@@ -678,8 +681,7 @@ export function validateModule(mod) {
             if (arm.kind === "void" ||
                 arm.kind === "union" ||
                 arm.kind === "dyn" ||
-                arm.kind === "date" ||
-                arm.kind === "generator") {
+                (arm.kind === "generator" && iterableArmsElem(u.arms) === null)) {
                 errors.push({ message: `union ${u.id}: arm ${i} is ${arm.kind}`, loc: noLoc });
             }
             if (arm.kind === "promise" && !maybePromiseUnionArmsOk(u.arms)) {
@@ -688,6 +690,9 @@ export function validateModule(mod) {
                     loc: noLoc,
                 });
             }
+            if (arm.kind === "date" && !unionContainerArmsOk(u.arms)) {
+                errors.push({ message: `union ${u.id}: date arm ${i} beside non-unit arms`, loc: noLoc });
+            }
             if (arm.kind === "record" && !recordsById.has(arm.shapeId)) {
                 errors.push({
                     message: `union ${u.id}: arm ${i} references undeclared shape "${arm.shapeId}"`,
@@ -1335,6 +1340,7 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                         e.left.type.kind === "map" ||
                         e.left.type.kind === "set" ||
                         e.left.type.kind === "regex" ||
+                        e.left.type.kind === "date" ||
                         e.left.type.kind === "url" ||
                         e.left.type.kind === "object" ||
                         e.left.type.kind === "record" ||
@@ -1990,6 +1996,9 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                     indexOfNum: { argTypes: [F64, F64], minArgs: 1, result: F64 },
                     lastIndexOfNum: { argTypes: [F64, F64], minArgs: 1, result: F64 },
                     includesNum: { argTypes: [F64, F64], minArgs: 1, result: BOOL },
+                    indexOfElem: { argTypes: [F64, F64], minArgs: 2, result: F64 },
+                    lastIndexOfElem: { argTypes: [F64, F64], minArgs: 2, result: F64 },
+                    includesElem: { argTypes: [F64, F64], minArgs: 2, result: BOOL },
                     fill: { argTypes: [BYTES_U8, F64, F64], minArgs: 1, result: BYTES_U8 },
                     fillNum: { argTypes: [F64, F64, F64], minArgs: 1, result: BYTES_U8 },
                     // Per-element TypedArray fill (any elem — the non-u8 fill path).
@@ -2159,8 +2168,9 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                 if ((e.method === "sortPrimitive" || e.method === "toSortedPrimitive") &&
                     elem.kind !== "string" &&
                     elem.kind !== "f64" &&
-                    elem.kind !== "bool") {
-                    err(`arrIntrinsic ${e.method} requires primitive elements, got ${elem.kind}`, e.loc);
+                    elem.kind !== "bool" &&
+                    elem.kind !== "date") {
+                    err(`arrIntrinsic ${e.method} requires primitive or Date elements, got ${elem.kind}`, e.loc);
                 }
                 if (e.method === "sortValues" || e.method === "sortDense") {
                     const arms = elem.kind === "union" ? unions.get(elem.unionId)?.arms : undefined;
@@ -2226,10 +2236,15 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                     elem.kind !== "f64" &&
                     elem.kind !== "string" &&
                     elem.kind !== "bool" &&
+                    elem.kind !== "date" &&
                     !(elem.kind === "union" &&
                         (unions
                             .get(elem.unionId)
-                            ?.arms.every((a) => a.kind === "f64" || a.kind === "string" || a.kind === "bool" || isUnitType(a)) ??
+                            ?.arms.every((a) => a.kind === "f64" ||
+                            a.kind === "string" ||
+                            a.kind === "bool" ||
+                            a.kind === "date" ||
+                            isUnitType(a)) ??
                             false))) {
                     err(`arrIntrinsic join on ${elem.kind} elements (frontend must reject)`, e.loc);
                 }
@@ -3763,8 +3778,10 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             e.fn === "stream.pipelineDyn" ||
             e.fn === "stream.compose" ||
             e.fn === "stream.pipelineStages" ||
+            e.fn === "stream.composeStages" ||
             e.fn === "stream.pipelineStagesDyn" ||
             e.fn === "sp.pipelineStages" ||
+            e.fn === "sp.pipelineStagesOpts" ||
             e.fn === "sp.pipeline" ||
             e.fn === "sp.pipelineOpts" ||
             e.fn === "sp.pipelineOptsSignal" ||
@@ -3809,6 +3826,9 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             e.fn === "error.assign" ||
             e.fn === "error.hasKey" ||
             e.fn === "error.hasCode" ||
+            e.fn === "error.ownAccessor" ||
+            e.fn === "error.accessorGet" ||
+            e.fn === "error.accessorSet" ||
             e.fn === "error.defineCause" ||
             e.fn === "error.deleteCause") {
             const recv = e.args[0];
@@ -3912,7 +3932,9 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                 err(`libCall dyn.objEntryPairs must return a [string, any] tuple array`, e.loc);
             return true;
         }
-        if (e.fn === "array.namedKeys" || e.fn === "array.hasNamedKey") {
+        if (e.fn === "array.namedKeys" ||
+            e.fn === "array.hasNamedKey" ||
+            e.fn === "array.hasProperties") {
             if (e.args[0]?.type.kind !== "array") {
                 err(`libCall ${e.fn} arg 0: expected an array, got ${e.args[0]?.type.kind}`, e.loc);
             }
@@ -3930,6 +3952,17 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             }
             return true;
         }
+        if (e.fn === "fetch.readableFromWeb") {
+            // Readable.fromWeb: (stream, objectMode, highWaterMark, encoding) → a Readable.
+            if (e.args.length !== 4 ||
+                e.args[0]?.type.kind !== "dyn" ||
+                e.args[1]?.type.kind !== "bool" ||
+                e.args[2]?.type.kind !== "f64" ||
+                e.args[3]?.type.kind !== "string" ||
+                e.type.kind !== "object")
+                err(`libCall ${e.fn} must take (stream, objectMode, highWaterMark, encoding) and return a Readable`, e.loc);
+            return true;
+        }
         if (e.fn === "fetch.readerRead") {
             if (e.type.kind !== "promise" || e.type.inner.kind !== "record") {
                 err(`libCall fetch.readerRead must return a promise of a read-result record`, e.loc);
@@ -4877,6 +4910,35 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
         return false;
     }
     function checkLibSpecialCall(e) {
+        if (e.fn === "promise.fulfilledValue") {
+            const promise = e.args[0]?.type;
+            if (promise?.kind !== "promise" ||
+                e.type.kind === "void" ||
+                isUnitType(e.type) ||
+                promise.inner.kind !== "void")
+                err(`libCall ${e.fn} must read a Promise<void> entry at a non-void type`, e.loc);
+            return true;
+        }
+        if (e.fn === "promise.allTuple") {
+            const [entries, build] = e.args;
+            if (entries?.type.kind !== "array" ||
+                entries.type.elem.kind !== "promise" ||
+                entries.type.elem.inner.kind !== "void" ||
+                build?.kind !== "closure" ||
+                build.captures.length > 0 ||
+                build.type.kind !== "func" ||
+                e.type.kind !== "promise" ||
+                !typeEquals(build.type.ret, e.type.inner))
+                err(`libCall ${e.fn} must join Promise<void> entries into its builder's tuple`, e.loc);
+            return true;
+        }
+        if (e.fn === "als.bindCallback") {
+            // The bound function, at its closure argument's own type.
+            const fn = e.args[0]?.type;
+            if (fn?.kind !== "func" || !typeEquals(e.type, fn))
+                err(`libCall ${e.fn} must return its function argument's type`, e.loc);
+            return true;
+        }
         if (e.fn === "crypto.randomFillView" ||
             e.fn === "crypto.randomFillViewRest" ||
             e.fn === "crypto.getRandomValues") {
@@ -4910,30 +4972,9 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             if (callback?.kind !== "func")
                 err(`libCall ${e.fn} callback must be a function`, e.loc);
         }
-        if (e.fn === "global.undefRead") {
-            // Always throws — the result type is whatever the declared type
-            // of the undefined global mapped to (never materialized).
-            return true;
-        }
-        if (e.fn === "error.nodeThrow" ||
-            e.fn === "error.argTypeThrow" ||
-            e.fn === "error.propTypeThrow" ||
-            e.fn === "fs.mkdtempChk" ||
-            e.fn === "fs.readFileChk" ||
-            e.fn === "fs.opendirChk" ||
-            e.fn === "fs.watchFileChk" ||
-            e.fn === "fs.lchmodChk" ||
-            e.fn === "fs.readChk" ||
-            e.fn === "fs.streamOptsChk" ||
-            e.fn === "net.connectOptsChk" ||
-            e.fn === "tls.caCertsChk") {
-            // Always throws — the result type is the replaced expression's
-            // own (never materialized; the global.undefRead pattern). The
-            // fs Chk ladders qualify: every validation failure throws
-            // Node's typed error, and a full pass throws the trailing
-            // compiler-rendered fence.
+        // Always throws: the result type is never materialized.
+        if (ALWAYS_THROW_LIB_FNS.has(e.fn))
             return true;
-        }
         if (e.fn === "error.new" || e.fn === "error.newOptions") {
             // Which builtin the runtime constructs is named by the result type.
             if (!isBuiltinErrorObject(e.type)) {
@@ -5047,6 +5088,21 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             e.fn === "stream.destroy" ||
             e.fn === "stream.destroyErr" ||
             e.fn === "stream.objectMode" ||
+            e.fn === "stream.decodeStrings" ||
+            e.fn === "stream.optionValues" ||
+            e.fn === "stream.duplexFromGen" ||
+            e.fn === "stream.duplexFromDyn" ||
+            e.fn === "stream.duplexifyReadable" ||
+            e.fn === "stream.duplexifyWritable" ||
+            e.fn === "stream.duplexifyPair" ||
+            e.fn === "stream.duplexFromPromise" ||
+            e.fn === "stream.duplexFromAsyncFn" ||
+            e.fn === "stream.duplexFromGenSignal" ||
+            e.fn === "stream.duplexFromAsyncFnSignal" ||
+            e.fn === "stream.setConstruct" ||
+            e.fn === "stream.replaceConstruct" ||
+            e.fn === "stream.setWritev" ||
+            e.fn === "stream.setWritevDefault" ||
             e.fn === "stream.iteratorClose" ||
             e.fn === "stream.prop" ||
             e.fn === "stream.errored" ||
@@ -5056,8 +5112,10 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             e.fn === "stream.pipelineDyn" ||
             e.fn === "stream.compose" ||
             e.fn === "stream.pipelineStages" ||
+            e.fn === "stream.composeStages" ||
             e.fn === "stream.pipelineStagesDyn" ||
             e.fn === "sp.pipelineStages" ||
+            e.fn === "sp.pipelineStagesOpts" ||
             e.fn === "stream.addAbortSignal" ||
             e.fn === "stream.addAbortSignalFn" ||
             e.fn === "stream.finishedOpts" ||
@@ -5098,6 +5156,36 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                 }
                 return true;
             }
+            if (e.fn === "stream.duplexFromPromise") {
+                // Duplex.from(promise): (promise, box) → a Duplex.
+                if (e.args[0]?.type.kind !== "promise" || e.args[1]?.type.kind !== "func")
+                    err(`libCall ${e.fn} takes a promise and its box helper`, e.loc);
+                if (!isStreamObject(e.type))
+                    err(`libCall ${e.fn} must return a stream class`, e.loc);
+                return true;
+            }
+            if (e.fn === "stream.duplexifyReadable" ||
+                e.fn === "stream.duplexifyWritable" ||
+                e.fn === "stream.duplexifyPair") {
+                // Duplex.from over streams: (r), (w) or (r, w) → a Duplex.
+                const want = e.fn === "stream.duplexifyPair" ? 2 : 1;
+                if (e.args.length !== want || !e.args.every((a) => isStreamObject(a.type)))
+                    err(`libCall ${e.fn} takes ${want} stream argument(s)`, e.loc);
+                if (!isStreamObject(e.type))
+                    err(`libCall ${e.fn} must return a stream class`, e.loc);
+                return true;
+            }
+            if (e.fn === "stream.duplexFromGen" ||
+                e.fn === "stream.duplexFromDyn" ||
+                e.fn === "stream.duplexFromAsyncFn" ||
+                e.fn === "stream.duplexFromGenSignal" ||
+                e.fn === "stream.duplexFromAsyncFnSignal") {
+                // Duplex.from: (stage, source, helpers...) or (body) → a Duplex.
+                if (!isStreamObject(e.type)) {
+                    err(`libCall ${e.fn} must return a stream class, got ${e.type.kind}`, e.loc);
+                }
+                return true;
+            }
             if (e.fn.endsWith(".newDyn")) {
                 // The dyn-options constructor: (optsDyn) → the stream class.
                 if (!isStreamObject(e.type)) {
@@ -5166,12 +5254,14 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             }
             if (e.fn === "stream.pipelineStages" ||
                 e.fn === "stream.pipelineStagesDyn" ||
-                e.fn === "sp.pipelineStages") {
+                e.fn === "sp.pipelineStages" ||
+                e.fn === "sp.pipelineStagesOpts" ||
+                e.fn === "stream.composeStages") {
                 // (n, k, positions, flags, mask, s1..sn, stage1..stagek,
-                // helpers..., box?, (signal, abort)?[, cb]): positions bit j marks
-                // stage j a function, flags bits 1-2 the destination, bit 5 the
-                // signal group; helpers are three per function run a stream
-                // follows.
+                // helpers..., box?, (signal, abort)?[, cb | end, signal?]):
+                // positions bit j marks stage j a function, flags bits 1-2 the
+                // destination, bit 5 the signal group, bit 6 the options' signal;
+                // helpers are three per pumped function run.
                 const [n, k, positions, flags] = e.args
                     .slice(0, 4)
                     .map((a) => (a?.kind === "numLit" ? a.value : -1));
@@ -5183,9 +5273,13 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                         functions++;
                 }
                 const lastIsFunction = Math.floor(positions / 2 ** (total - 1)) % 2 === 1;
-                const helperCount = 3 * pipelinePumpedRuns(total, positions);
+                const helperCount = 3 * pipelinePumpedRuns(total, positions, flags);
                 const groups = helperCount + (destKind === 2 ? 1 : 0) + (flags & 32 ? 2 : 0);
-                const cbArgs = e.fn === "sp.pipelineStages" ? 0 : 1;
+                const cbArgs = e.fn === "sp.pipelineStagesOpts"
+                    ? 1 + (flags & 64 ? 1 : 0)
+                    : e.fn === "sp.pipelineStages" || e.fn === "stream.composeStages"
+                        ? 0
+                        : 1;
                 if (flags < 0 ||
                     n < 0 ||
                     k < 1 ||
@@ -5193,7 +5287,8 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                     total > 52 ||
                     positions >= 2 ** total ||
                     functions !== k ||
-                    (destKind === 2) !== lastIsFunction ||
+                    destKind === 3 ||
+                    (destKind !== 0) !== lastIsFunction ||
                     total < 2 ||
                     e.args.length !== 5 + n + k + groups + cbArgs) {
                     err(`libCall ${e.fn} count/arity mismatch`, e.loc);
@@ -5220,7 +5315,14 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                     if (e.args[at2]?.type.kind !== "dyn" || e.args[at2 + 1]?.type.kind !== "func")
                         err(`libCall ${e.fn} signal and abort must be a dyn and a func`, e.loc);
                 }
-                if (cbArgs) {
+                if (e.fn === "sp.pipelineStagesOpts") {
+                    const tail = e.args.slice(e.args.length - cbArgs);
+                    if (tail[0]?.type.kind !== "bool" || (cbArgs > 1 && tail[1]?.type.kind !== "dyn"))
+                        err(`libCall ${e.fn} end and signal must be a bool and a dyn`, e.loc);
+                    if (e.type.kind !== "promise")
+                        err(`libCall ${e.fn} must return a promise`, e.loc);
+                }
+                else if (cbArgs) {
                     const cbK = e.args[e.args.length - 1]?.type.kind;
                     if (e.fn === "stream.pipelineStages" ? cbK !== "func" : cbK !== "dyn") {
                         err(`libCall ${e.fn} callback has the wrong kind`, e.loc);
@@ -5228,6 +5330,10 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                     if (!isStreamObject(e.type))
                         err(`libCall ${e.fn} must return a stream`, e.loc);
                 }
+                else if (e.fn === "stream.composeStages") {
+                    if (!isStreamObject(e.type))
+                        err(`libCall ${e.fn} must return a stream`, e.loc);
+                }
                 else if (e.type.kind !== "promise") {
                     err(`libCall ${e.fn} must return a promise`, e.loc);
                 }
@@ -5258,6 +5364,8 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
             if (e.fn === "readable.opMap" ||
                 e.fn === "readable.opFilter" ||
                 e.fn === "readable.opFlatMap" ||
+                e.fn === "readable.opFlatMapAsyncGen" ||
+                e.fn === "readable.opFlatMapSyncGen" ||
                 e.fn === "readable.opDrop" ||
                 e.fn === "readable.opTake") {
                 if (!isStreamObject(e.args[0]?.type) || !isStreamObject(e.type)) {
@@ -5265,13 +5373,28 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                 }
                 return true;
             }
-            if (e.fn === "readable.opReduce") {
+            if (e.fn === "readable.opReduce" || e.fn === "readable.opReduceSignal") {
                 const inner = e.type.kind === "promise" ? e.type.inner : null;
                 const kind = e.args[5]?.kind === "numLit" ? e.args[5].value : -1;
                 const want = [DYN, F64, BOOL, STRING][kind];
                 if (!isStreamObject(e.args[0]?.type) || !inner || !want || !typeEquals(inner, want)) {
                     err(`libCall readable.opReduce must return the promise its kind names`, e.loc);
                 }
+                if (e.fn === "readable.opReduceSignal" && e.args[6]?.type.kind !== "func")
+                    err(`libCall ${e.fn} takes ac.abort() as a closure`, e.loc);
+                return true;
+            }
+            if (e.fn === "readable.opOnceOptions") {
+                if (e.args.length !== 0 || e.type.kind !== "dyn")
+                    err(`libCall ${e.fn} takes no arguments`, e.loc);
+                return true;
+            }
+            if (e.fn === "readable.opSignalList" ||
+                e.fn === "readable.opSignalOpts" ||
+                e.fn === "readable.opSignalOf") {
+                // The operator callbacks' { signal }: (dyn) → dyn.
+                if (e.args.length !== 1 || e.args[0]?.type.kind !== "dyn" || e.type.kind !== "dyn")
+                    err(`libCall ${e.fn} takes and returns a checked-dynamic value`, e.loc);
                 return true;
             }
             if (e.fn === "readable.fromDone") {
@@ -5327,6 +5450,10 @@ function validateFunction(fn, functions, ffiByName, classes, records, recordVali
                 e.fn === "readable.setEncoding" ||
                 e.fn === "readable.pushEncoding" ||
                 e.fn === "stream.objectMode" ||
+                e.fn === "stream.decodeStrings" ||
+                e.fn === "stream.optionValues" ||
+                e.fn === "writable.writeEncoding" ||
+                e.fn === "writable.setDefaultEncoding" ||
                 e.fn === "stream.destroy" ||
                 e.fn === "stream.destroyErr" ||
                 e.fn === "stream.iteratorClose") {