@honua/sdk-esri-compat 0.0.6-alpha.0

package/core/pbf-decoder.js

@@ -0,0 +1,627 @@
+/**
+ * Minimal protobuf wire-format reader and Esri PBF query response decoder.
+ *
+ * Decodes `FeatureCollectionPBuffer` responses (from `f=pbf`) into the same
+ * JSON-compatible shape as `f=json`, making the binary format transparent
+ * to callers.
+ *
+ * No external protobuf library dependency. Implements only the subset of
+ * the wire format needed for the Esri PBF query schema.
+ *
+ * @module
+ */
+// ── Wire format constants ────────────────────────────────────
+const WIRE_VARINT = 0;
+const WIRE_64BIT = 1;
+const WIRE_LENGTH_DELIMITED = 2;
+const WIRE_32BIT = 5;
+// ── Low-level protobuf reader ────────────────────────────────
+/** Reads protobuf wire-format primitives from a byte buffer. */
+class PbfReader {
+    view;
+    bytes;
+    pos;
+    end;
+    constructor(buffer, offset = 0, length) {
+        this.bytes = buffer;
+        this.view = new DataView(buffer.buffer, buffer.byteOffset, buffer.byteLength);
+        this.pos = offset;
+        this.end = length !== undefined ? offset + length : buffer.length;
+    }
+    /** Read an unsigned 32-bit varint. */
+    readVarint() {
+        let result = 0;
+        let shift = 0;
+        while (this.pos < this.end) {
+            const byte = this.bytes[this.pos++];
+            result |= (byte & 0x7f) << shift;
+            if ((byte & 0x80) === 0)
+                return result >>> 0;
+            shift += 7;
+            if (shift > 35)
+                throw new Error("Varint too long");
+        }
+        throw new Error("Unexpected end of buffer reading varint");
+    }
+    /** Read a 64-bit varint as a JavaScript number (safe for values < 2^53). */
+    readVarint64() {
+        let lo = 0;
+        let hi = 0;
+        let shift = 0;
+        while (shift < 28 && this.pos < this.end) {
+            const byte = this.bytes[this.pos++];
+            lo |= (byte & 0x7f) << shift;
+            shift += 7;
+            if ((byte & 0x80) === 0)
+                return lo >>> 0;
+        }
+        // Remaining bits go into hi
+        while (this.pos < this.end) {
+            const byte = this.bytes[this.pos++];
+            hi |= (byte & 0x7f) << (shift - 28);
+            shift += 7;
+            if ((byte & 0x80) === 0)
+                break;
+            if (shift > 63)
+                throw new Error("Varint too long");
+        }
+        return (hi >>> 0) * 0x10000000 + (lo >>> 0);
+    }
+    /** Read a signed 64-bit varint (zigzag-decoded) as a JavaScript number. */
+    readSVarint64() {
+        const n = this.readVarint64();
+        // Zigzag decode: (n >>> 1) ^ -(n & 1)
+        // For numbers that fit in 32 bits this is straightforward.
+        // For larger numbers, we do the math in floating point (safe for ints < 2^53).
+        const half = Math.floor(n / 2);
+        return n % 2 === 0 ? half : -(half + 1);
+    }
+    /** Read a signed 32-bit varint (zigzag-decoded). */
+    readSVarint32() {
+        const n = this.readVarint();
+        return (n >>> 1) ^ -(n & 1);
+    }
+    /** Read a little-endian 64-bit float. */
+    readDouble() {
+        const val = this.view.getFloat64(this.pos, true);
+        this.pos += 8;
+        return val;
+    }
+    /** Read a little-endian 32-bit float. */
+    readFloat() {
+        const val = this.view.getFloat32(this.pos, true);
+        this.pos += 4;
+        return val;
+    }
+    /** Read a UTF-8 string of the given byte length. */
+    readString(byteLength) {
+        const slice = this.bytes.subarray(this.pos, this.pos + byteLength);
+        this.pos += byteLength;
+        return textDecoder.decode(slice);
+    }
+    /** Read a boolean from a varint. */
+    readBool() {
+        return this.readVarint() !== 0;
+    }
+    /** Read a field tag, returning [fieldNumber, wireType]. */
+    readTag() {
+        const tag = this.readVarint();
+        return [tag >>> 3, tag & 0x07];
+    }
+    /** Skip a field value based on wire type. */
+    skip(wireType) {
+        switch (wireType) {
+            case WIRE_VARINT:
+                this.readVarint64();
+                break;
+            case WIRE_64BIT:
+                this.pos += 8;
+                break;
+            case WIRE_LENGTH_DELIMITED: {
+                const len = this.readVarint();
+                this.pos += len;
+                break;
+            }
+            case WIRE_32BIT:
+                this.pos += 4;
+                break;
+            default:
+                throw new Error(`Unknown wire type: ${wireType}`);
+        }
+    }
+    /** Create a sub-reader for a length-delimited field. */
+    subReader(byteLength) {
+        const sub = new PbfReader(this.bytes, this.pos, byteLength);
+        this.pos += byteLength;
+        return sub;
+    }
+    /** Read packed repeated sint64 values. */
+    readPackedSInt64(byteLength) {
+        const result = [];
+        const end = this.pos + byteLength;
+        while (this.pos < end) {
+            result.push(this.readSVarint64());
+        }
+        return result;
+    }
+    /** Read packed repeated uint32 values. */
+    readPackedUInt32(byteLength) {
+        const result = [];
+        const end = this.pos + byteLength;
+        while (this.pos < end) {
+            result.push(this.readVarint());
+        }
+        return result;
+    }
+}
+const textDecoder = new TextDecoder();
+// ── Esri PBF geometry type enum ──────────────────────────────
+const PBF_GEOMETRY_TYPE_NAMES = {
+    0: "esriGeometryPoint",
+    1: "esriGeometryMultipoint",
+    2: "esriGeometryPolyline",
+    3: "esriGeometryPolygon",
+    4: "esriGeometryEnvelope",
+    127: "esriGeometryNull",
+};
+function decodeScale(reader) {
+    let xScale = 0;
+    let yScale = 0;
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 1 && wire === WIRE_64BIT)
+            xScale = reader.readDouble();
+        else if (field === 2 && wire === WIRE_64BIT)
+            yScale = reader.readDouble();
+        else
+            reader.skip(wire);
+    }
+    return { xScale, yScale };
+}
+function decodeTranslate(reader) {
+    let xTranslate = 0;
+    let yTranslate = 0;
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 1 && wire === WIRE_64BIT)
+            xTranslate = reader.readDouble();
+        else if (field === 2 && wire === WIRE_64BIT)
+            yTranslate = reader.readDouble();
+        else
+            reader.skip(wire);
+    }
+    return { xTranslate, yTranslate };
+}
+function decodeTransform(reader) {
+    let xScale = 1;
+    let yScale = 1;
+    let xTranslate = 0;
+    let yTranslate = 0;
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 2 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            const sub = reader.subReader(len);
+            const s = decodeScale(sub);
+            xScale = s.xScale;
+            yScale = s.yScale;
+        }
+        else if (field === 3 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            const sub = reader.subReader(len);
+            const t = decodeTranslate(sub);
+            xTranslate = t.xTranslate;
+            yTranslate = t.yTranslate;
+        }
+        else {
+            reader.skip(wire);
+        }
+    }
+    return { xScale, yScale, xTranslate, yTranslate };
+}
+function decodeField(reader) {
+    let name = "";
+    let fieldType = 0;
+    let alias = "";
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 1 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            name = reader.readString(len);
+        }
+        else if (field === 2 && wire === WIRE_VARINT) {
+            fieldType = reader.readVarint();
+        }
+        else if (field === 3 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            alias = reader.readString(len);
+        }
+        else {
+            reader.skip(wire);
+        }
+    }
+    return { name, fieldType, alias: alias || name };
+}
+function decodeSpatialReference(reader) {
+    let wkid = 0;
+    let latestWkid = 0;
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 1 && wire === WIRE_VARINT)
+            wkid = reader.readVarint();
+        else if (field === 2 && wire === WIRE_VARINT)
+            latestWkid = reader.readVarint();
+        else
+            reader.skip(wire);
+    }
+    return { wkid, latestWkid: latestWkid || wkid };
+}
+function decodeValue(reader) {
+    let value = null;
+    let fieldIndex = -1;
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        switch (field) {
+            case 1: // string_value
+                if (wire === WIRE_LENGTH_DELIMITED) {
+                    const len = reader.readVarint();
+                    value = reader.readString(len);
+                }
+                else
+                    reader.skip(wire);
+                break;
+            case 2: // float_value
+                if (wire === WIRE_32BIT)
+                    value = reader.readFloat();
+                else
+                    reader.skip(wire);
+                break;
+            case 3: // double_value
+                if (wire === WIRE_64BIT)
+                    value = reader.readDouble();
+                else
+                    reader.skip(wire);
+                break;
+            case 4: // sint_value (sint32)
+                if (wire === WIRE_VARINT)
+                    value = reader.readSVarint32();
+                else
+                    reader.skip(wire);
+                break;
+            case 5: // uint_value
+                if (wire === WIRE_VARINT)
+                    value = reader.readVarint();
+                else
+                    reader.skip(wire);
+                break;
+            case 6: // int64_value
+                if (wire === WIRE_VARINT)
+                    value = reader.readVarint64();
+                else
+                    reader.skip(wire);
+                break;
+            case 7: // uint64_value
+                if (wire === WIRE_VARINT)
+                    value = reader.readVarint64();
+                else
+                    reader.skip(wire);
+                break;
+            case 9: // bool_value
+                if (wire === WIRE_VARINT)
+                    value = reader.readBool();
+                else
+                    reader.skip(wire);
+                break;
+            case 10: // null_value
+                if (wire === WIRE_VARINT) {
+                    reader.readVarint(); // consume the bool
+                    value = null;
+                }
+                else
+                    reader.skip(wire);
+                break;
+            case 11: // field_index
+                if (wire === WIRE_VARINT)
+                    fieldIndex = reader.readVarint();
+                else
+                    reader.skip(wire);
+                break;
+            default:
+                reader.skip(wire);
+        }
+    }
+    return { value, fieldIndex };
+}
+function decodeGeometry(reader, transform, layerGeometryType) {
+    let lengths = [];
+    let coords = [];
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 1 && wire === WIRE_VARINT) {
+            reader.readVarint(); // geometryType — we use layerGeometryType instead
+        }
+        else if (field === 2 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            lengths = reader.readPackedUInt32(len);
+        }
+        else if (field === 3 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            coords = reader.readPackedSInt64(len);
+        }
+        else {
+            reader.skip(wire);
+        }
+    }
+    if (coords.length === 0)
+        return null;
+    // Delta-decode and transform coordinates
+    const xScale = transform?.xScale ?? 1;
+    const yScale = transform?.yScale ?? 1;
+    const xTranslate = transform?.xTranslate ?? 0;
+    const yTranslate = transform?.yTranslate ?? 0;
+    // Undo delta encoding: each coordinate pair is (deltaX, deltaY)
+    let prevX = 0;
+    let prevY = 0;
+    const worldCoords = [];
+    for (let i = 0; i < coords.length; i += 2) {
+        prevX += coords[i];
+        prevY += coords[i + 1];
+        worldCoords.push([prevX * xScale + xTranslate, prevY * yScale + yTranslate]);
+    }
+    return buildGeoServicesGeometry(layerGeometryType, worldCoords, lengths);
+}
+function buildGeoServicesGeometry(geometryType, worldCoords, lengths) {
+    switch (geometryType) {
+        case "esriGeometryPoint":
+            return worldCoords.length > 0 ? { x: worldCoords[0][0], y: worldCoords[0][1] } : { x: null, y: null };
+        case "esriGeometryMultipoint":
+            return { points: worldCoords };
+        case "esriGeometryPolyline": {
+            const paths = [];
+            let offset = 0;
+            for (const len of lengths) {
+                paths.push(worldCoords.slice(offset, offset + len));
+                offset += len;
+            }
+            return { paths: paths.length > 0 ? paths : [worldCoords] };
+        }
+        case "esriGeometryPolygon": {
+            const rings = [];
+            let offset = 0;
+            for (const len of lengths) {
+                rings.push(worldCoords.slice(offset, offset + len));
+                offset += len;
+            }
+            return { rings: rings.length > 0 ? rings : [worldCoords] };
+        }
+        default:
+            return { x: worldCoords[0]?.[0] ?? null, y: worldCoords[0]?.[1] ?? null };
+    }
+}
+function decodeFeature(reader, fields, transform, geometryType) {
+    const attributes = {};
+    let geometry = null;
+    const valueEntries = [];
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 1 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            const sub = reader.subReader(len);
+            valueEntries.push(decodeValue(sub));
+        }
+        else if (field === 2 && wire === WIRE_LENGTH_DELIMITED) {
+            const len = reader.readVarint();
+            const sub = reader.subReader(len);
+            geometry = decodeGeometry(sub, transform, geometryType);
+        }
+        else {
+            reader.skip(wire);
+        }
+    }
+    // Map value entries to field names using field indices
+    for (const entry of valueEntries) {
+        if (entry.fieldIndex >= 0 && entry.fieldIndex < fields.length) {
+            attributes[fields[entry.fieldIndex].name] = entry.value;
+        }
+    }
+    const result = { attributes };
+    if (geometry !== null) {
+        result.geometry = geometry;
+    }
+    return result;
+}
+function decodeFeatureResult(reader) {
+    let objectIdFieldName = "";
+    let geometryType = "";
+    let spatialReference = null;
+    let exceededTransferLimit = false;
+    let hasZ = false;
+    let hasM = false;
+    let transform = null;
+    const fields = [];
+    const featureReaders = [];
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        switch (field) {
+            case 1: // objectIdFieldName
+                if (wire === WIRE_LENGTH_DELIMITED) {
+                    const len = reader.readVarint();
+                    objectIdFieldName = reader.readString(len);
+                }
+                else
+                    reader.skip(wire);
+                break;
+            case 7: // geometryType
+                if (wire === WIRE_VARINT) {
+                    const gt = reader.readVarint();
+                    geometryType = PBF_GEOMETRY_TYPE_NAMES[gt] ?? "esriGeometryNull";
+                }
+                else
+                    reader.skip(wire);
+                break;
+            case 8: // spatialReference
+                if (wire === WIRE_LENGTH_DELIMITED) {
+                    const len = reader.readVarint();
+                    spatialReference = decodeSpatialReference(reader.subReader(len));
+                }
+                else
+                    reader.skip(wire);
+                break;
+            case 9: // exceededTransferLimit
+                if (wire === WIRE_VARINT)
+                    exceededTransferLimit = reader.readBool();
+                else
+                    reader.skip(wire);
+                break;
+            case 10: // hasZ
+                if (wire === WIRE_VARINT)
+                    hasZ = reader.readBool();
+                else
+                    reader.skip(wire);
+                break;
+            case 11: // hasM
+                if (wire === WIRE_VARINT)
+                    hasM = reader.readBool();
+                else
+                    reader.skip(wire);
+                break;
+            case 12: // transform
+                if (wire === WIRE_LENGTH_DELIMITED) {
+                    const len = reader.readVarint();
+                    transform = decodeTransform(reader.subReader(len));
+                }
+                else
+                    reader.skip(wire);
+                break;
+            case 13: // fields (repeated)
+                if (wire === WIRE_LENGTH_DELIMITED) {
+                    const len = reader.readVarint();
+                    fields.push(decodeField(reader.subReader(len)));
+                }
+                else
+                    reader.skip(wire);
+                break;
+            case 15: // features (repeated)
+                if (wire === WIRE_LENGTH_DELIMITED) {
+                    const len = reader.readVarint();
+                    // Defer feature decoding until we have all fields and transform
+                    featureReaders.push(reader.subReader(len));
+                }
+                else
+                    reader.skip(wire);
+                break;
+            default:
+                reader.skip(wire);
+        }
+    }
+    // Now decode features with full field/transform context
+    const features = featureReaders.map((fr) => decodeFeature(fr, fields, transform, geometryType));
+    // Build JSON-compatible response matching f=json shape
+    const result = {
+        objectIdFieldName,
+        fields: fields.map((f) => ({
+            name: f.name,
+            type: mapPbfFieldTypeToGeoServices(f.fieldType),
+            alias: f.alias,
+        })),
+        features,
+    };
+    if (geometryType && geometryType !== "esriGeometryNull") {
+        result.geometryType = geometryType;
+    }
+    if (spatialReference) {
+        result.spatialReference = spatialReference;
+    }
+    if (hasZ)
+        result.hasZ = true;
+    if (hasM)
+        result.hasM = true;
+    if (exceededTransferLimit) {
+        result.exceededTransferLimit = true;
+    }
+    return result;
+}
+function mapPbfFieldTypeToGeoServices(fieldType) {
+    switch (fieldType) {
+        case 0:
+            return "esriFieldTypeSmallInteger";
+        case 1:
+            return "esriFieldTypeInteger";
+        case 2:
+            return "esriFieldTypeSingle";
+        case 3:
+            return "esriFieldTypeDouble";
+        case 4:
+            return "esriFieldTypeString";
+        case 5:
+            return "esriFieldTypeDate";
+        case 6:
+            return "esriFieldTypeOID";
+        case 7:
+            return "esriFieldTypeGeometry";
+        case 8:
+            return "esriFieldTypeBlob";
+        case 10:
+            return "esriFieldTypeGUID";
+        case 11:
+            return "esriFieldTypeGlobalID";
+        case 12:
+            return "esriFieldTypeXML";
+        case 13:
+            return "esriFieldTypeBigInteger";
+        default:
+            return "esriFieldTypeString";
+    }
+}
+// ── Public API ───────────────────────────────────────────────
+/**
+ * Decode an Esri FeatureCollectionPBuffer response into a JSON-compatible
+ * query response object.
+ *
+ * The returned object has the same shape as an `f=json` response:
+ * `{ objectIdFieldName, geometryType, spatialReference, fields, features, ... }`
+ *
+ * @param buffer - The raw PBF bytes from a `f=pbf` response.
+ * @returns A JSON-compatible query response object.
+ */
+export function decodePbfQueryResponse(buffer) {
+    const bytes = buffer instanceof Uint8Array ? buffer : new Uint8Array(buffer);
+    const reader = new PbfReader(bytes);
+    let queryResult = {};
+    // FeatureCollectionPBuffer
+    while (reader.pos < reader.end) {
+        const [field, wire] = reader.readTag();
+        if (field === 1 && wire === WIRE_LENGTH_DELIMITED) {
+            // version string — skip
+            const len = reader.readVarint();
+            reader.pos += len;
+        }
+        else if (field === 2 && wire === WIRE_LENGTH_DELIMITED) {
+            // queryResult
+            const len = reader.readVarint();
+            const qrReader = reader.subReader(len);
+            // QueryResult → featureResult (field 1)
+            while (qrReader.pos < qrReader.end) {
+                const [qrField, qrWire] = qrReader.readTag();
+                if (qrField === 1 && qrWire === WIRE_LENGTH_DELIMITED) {
+                    const frLen = qrReader.readVarint();
+                    queryResult = decodeFeatureResult(qrReader.subReader(frLen));
+                }
+                else {
+                    qrReader.skip(qrWire);
+                }
+            }
+        }
+        else {
+            reader.skip(wire);
+        }
+    }
+    return queryResult;
+}
+/**
+ * Check whether a Response has a protobuf content type.
+ */
+export function isPbfResponse(response) {
+    const ct = response.headers.get("content-type") ?? "";
+    return ct.includes("application/x-protobuf") || ct.includes("application/protobuf");
+}