@gizmodata/gizmosql-client 1.2.0

package/dist/flight-client.js

@@ -0,0 +1,352 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FlightClient = void 0;
+const grpc = __importStar(require("@grpc/grpc-js"));
+const apache_arrow_1 = require("apache-arrow");
+const Flight_grpc_pb_1 = require("./generated/proto/Flight_grpc_pb");
+const Flight_pb_1 = require("./generated/proto/Flight_pb");
+const errors_1 = require("./errors");
+const utils_1 = require("./utils");
+class FlightClient {
+    constructor(config) {
+        this.client = null;
+        (0, utils_1.validateConfig)(config);
+        this.config = { plaintext: false, ...config };
+        this.credentials = this.createCredentials();
+        this.metadata = this.createMetadata();
+    }
+    createCredentials() {
+        if (this.config.plaintext) {
+            return grpc.credentials.createInsecure();
+        }
+        if (this.config.tlsSkipVerify) {
+            return grpc.credentials.createSsl(null, null, null, {
+                rejectUnauthorized: false
+            });
+        }
+        return grpc.credentials.createSsl();
+    }
+    createMetadata() {
+        const metadata = new grpc.Metadata();
+        if (this.config.token) {
+            metadata.add("authorization", `Bearer ${this.config.token}`);
+        }
+        else if (this.config.username && this.config.password) {
+            const credential = Buffer.from(`${this.config.username}:${this.config.password}`).toString('base64');
+            metadata.add("authorization", `Basic ${credential}`);
+        }
+        return metadata;
+    }
+    async connect() {
+        try {
+            const address = `${this.config.host}:${this.config.port}`;
+            // Configure gRPC options for large message handling
+            const options = {
+                'grpc.max_receive_message_length': 100 * 1024 * 1024, // 100MB
+                'grpc.max_send_message_length': 100 * 1024 * 1024, // 100MB
+                'grpc.keepalive_time_ms': 30000,
+                'grpc.keepalive_timeout_ms': 5000,
+                'grpc.keepalive_permit_without_calls': 1,
+                'grpc.http2.max_pings_without_data': 0,
+                'grpc.http2.min_ping_interval_without_data_ms': 300000,
+            };
+            this.client = new Flight_grpc_pb_1.FlightServiceClient(address, this.credentials, options);
+            if (this.config.token || (this.config.username && this.config.password)) {
+                await this.authenticate();
+            }
+        }
+        catch (error) {
+            throw new errors_1.ConnectionError(`Failed to connect to ${this.config.host}:${this.config.port}`, error);
+        }
+    }
+    async authenticate() {
+        if (!this.client) {
+            throw new errors_1.ConnectionError('Client not connected');
+        }
+        return new Promise((resolve, reject) => {
+            const call = this.client.handshake(this.metadata);
+            const handshakeRequest = new Flight_pb_1.HandshakeRequest();
+            call.write(handshakeRequest);
+            call.on('metadata', (metadata) => {
+                const authHeader = metadata.get('authorization');
+                if (authHeader && authHeader.length > 0) {
+                    this.metadata.set('authorization', authHeader[0]);
+                }
+            });
+            call.on('data', () => {
+                resolve();
+            });
+            call.on('end', () => {
+                resolve();
+            });
+            call.on('error', (error) => {
+                reject(new errors_1.AuthenticationError((0, utils_1.parseErrorFromGrpc)(error).message));
+            });
+            call.end();
+        });
+    }
+    async getFlightInfo(descriptor) {
+        if (!this.client) {
+            await this.connect();
+        }
+        return new Promise((resolve, reject) => {
+            this.client.getFlightInfo(descriptor, this.metadata, (error, response) => {
+                if (error) {
+                    reject((0, utils_1.parseErrorFromGrpc)(error));
+                }
+                else {
+                    resolve(response);
+                }
+            });
+        });
+    }
+    async doGet(ticket) {
+        if (!this.client) {
+            await this.connect();
+        }
+        return new Promise((resolve, reject) => {
+            const STREAM_TIMEOUT_MS = 60000; // 60 second deadline
+            const options = {
+                deadline: Date.now() + STREAM_TIMEOUT_MS,
+            };
+            const call = this.client.doGet(ticket, this.metadata, options);
+            // Set up flow control for streaming
+            call.pause();
+            call.resume();
+            const flightDataMessages = [];
+            call.on('data', (flightData) => {
+                const dataHeader = flightData.getDataHeader_asU8();
+                const dataBody = flightData.getDataBody_asU8();
+                // Store each FlightData message as a complete unit for Arrow IPC reconstruction
+                flightDataMessages.push({
+                    header: dataHeader,
+                    body: dataBody
+                });
+            });
+            call.on('error', (error) => {
+                reject((0, utils_1.parseErrorFromGrpc)(error));
+            });
+            call.on('end', () => {
+                try {
+                    const result = this.processFlightDataMessages(flightDataMessages);
+                    resolve(result);
+                }
+                catch (error) {
+                    reject(new errors_1.FlightError(`Failed to parse Arrow data: ${error}`));
+                }
+            });
+        });
+    }
+    /**
+     * Processes FlightData messages and reconstructs them into Arrow IPC format.
+     * According to Apache Arrow Flight spec, FlightData contains FlatBuffer messages
+     * that need to be converted to proper Arrow IPC format for parsing.
+     */
+    processFlightDataMessages(messages) {
+        if (messages.length === 0) {
+            return new apache_arrow_1.Table();
+        }
+        const ipcParts = this.convertFlightDataToIPC(messages);
+        return (0, apache_arrow_1.tableFromIPC)(ipcParts);
+    }
+    /**
+     * Converts FlightData messages to Arrow IPC format.
+     * Each FlightData message contains a header (FlatBuffer) and optional body data.
+     */
+    convertFlightDataToIPC(messages) {
+        const ipcParts = [];
+        for (const message of messages) {
+            const { header, body } = message;
+            if (!header || header.length === 0) {
+                continue;
+            }
+            const ipcMessage = this.createIPCMessage(header, body);
+            ipcParts.push(ipcMessage);
+        }
+        return ipcParts;
+    }
+    /**
+     * Creates a single Arrow IPC message from FlightData header and body.
+     * Format: [4 bytes continuation] [4 bytes message length] [message] [padding] [body]
+     */
+    createIPCMessage(header, body) {
+        const messageLength = header.length;
+        const bodyLength = body?.length || 0;
+        const paddingLength = this.calculatePadding(messageLength);
+        const totalLength = 4 + 4 + messageLength + paddingLength + bodyLength;
+        const ipcMessage = new Uint8Array(totalLength);
+        let offset = 0;
+        // Write continuation indicator (0xFFFFFFFF for valid message)
+        offset = this.writeUint32(ipcMessage, offset, 0xFFFFFFFF);
+        // Write message length
+        offset = this.writeUint32(ipcMessage, offset, messageLength);
+        // Write FlatBuffer message (header)
+        ipcMessage.set(header, offset);
+        offset += messageLength;
+        // Skip padding (already zero-filled)
+        offset += paddingLength;
+        // Write body data if present
+        if (body && body.length > 0) {
+            ipcMessage.set(body, offset);
+        }
+        return ipcMessage;
+    }
+    /**
+     * Calculates padding needed to align to 8-byte boundary.
+     */
+    calculatePadding(messageLength) {
+        return (8 - (messageLength % 8)) % 8;
+    }
+    /**
+     * Writes a 32-bit unsigned integer in little-endian format.
+     */
+    writeUint32(buffer, offset, value) {
+        const view = new DataView(buffer.buffer, offset, 4);
+        view.setUint32(0, value, true); // little endian
+        return offset + 4;
+    }
+    async doPut(stream) {
+        if (!this.client) {
+            await this.connect();
+        }
+        return new Promise((resolve, reject) => {
+            const call = this.client.doPut(this.metadata);
+            call.on('error', (error) => {
+                reject((0, utils_1.parseErrorFromGrpc)(error));
+            });
+            call.on('end', () => {
+                resolve();
+            });
+            (async () => {
+                try {
+                    for await (const data of stream) {
+                        call.write(data);
+                    }
+                    call.end();
+                }
+                catch (error) {
+                    reject(new errors_1.FlightError(`Failed to write data: ${error}`));
+                }
+            })();
+        });
+    }
+    async doAction(action) {
+        if (!this.client) {
+            await this.connect();
+        }
+        return new Promise((resolve, reject) => {
+            const call = this.client.doAction(action, this.metadata);
+            const results = [];
+            call.on('data', (result) => {
+                results.push(result);
+            });
+            call.on('error', (error) => {
+                reject((0, utils_1.parseErrorFromGrpc)(error));
+            });
+            call.on('end', () => {
+                resolve(results);
+            });
+        });
+    }
+    async listFlights(criteria) {
+        if (!this.client) {
+            await this.connect();
+        }
+        const criteriaToUse = criteria || new Flight_pb_1.Criteria();
+        return new Promise((resolve, reject) => {
+            const call = this.client.listFlights(criteriaToUse, this.metadata);
+            const flights = [];
+            call.on('data', (flightInfo) => {
+                flights.push(flightInfo);
+            });
+            call.on('error', (error) => {
+                reject((0, utils_1.parseErrorFromGrpc)(error));
+            });
+            call.on('end', () => {
+                resolve(flights);
+            });
+        });
+    }
+    async getSchema(descriptor) {
+        if (!this.client) {
+            await this.connect();
+        }
+        return new Promise((resolve, reject) => {
+            this.client.getSchema(descriptor, this.metadata, (error, response) => {
+                if (error) {
+                    reject((0, utils_1.parseErrorFromGrpc)(error));
+                }
+                else {
+                    try {
+                        const schemaBytes = response.getSchema_asU8();
+                        const schema = apache_arrow_1.Schema.decode(schemaBytes);
+                        resolve(schema);
+                    }
+                    catch (parseError) {
+                        reject(new errors_1.FlightError(`Failed to parse schema: ${parseError}`));
+                    }
+                }
+            });
+        });
+    }
+    async listActions() {
+        if (!this.client) {
+            await this.connect();
+        }
+        const empty = new Flight_pb_1.Empty();
+        return new Promise((resolve, reject) => {
+            const call = this.client.listActions(empty, this.metadata);
+            const actions = [];
+            call.on('data', (actionType) => {
+                actions.push(actionType);
+            });
+            call.on('error', (error) => {
+                reject((0, utils_1.parseErrorFromGrpc)(error));
+            });
+            call.on('end', () => {
+                resolve(actions);
+            });
+        });
+    }
+    async close() {
+        if (this.client) {
+            this.client.close();
+            this.client = null;
+        }
+    }
+}
+exports.FlightClient = FlightClient;
+//# sourceMappingURL=flight-client.js.map

package/dist/flight-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"flight-client.js","sourceRoot":"","sources":["../src/flight-client.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,oDAAsC;AACtC,+CAAyD;AACzD,qEAAuE;AACvE,2DAWqC;AAErC,qCAA6E;AAC7E,mCAA6D;AAE7D,MAAa,YAAY;IAMvB,YAAY,MAA0B;QAL9B,WAAM,GAA+B,IAAI,CAAC;QAMhD,IAAA,sBAAc,EAAC,MAAM,CAAC,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,MAAM,EAAE,CAAC;QAC9C,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC5C,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;IACxC,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CAAC;QAC3C,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE;gBAClD,kBAAkB,EAAE,KAAK;aAC1B,CAAC,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC;IACtC,CAAC;IAEO,cAAc;QACpB,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QACrC,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACtB,QAAQ,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QAC9D,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACxD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACrG,QAAQ,CAAC,GAAG,CAAC,eAAe,EAAE,SAAS,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YAE1D,oDAAoD;YACpD,MAAM,OAAO,GAAG;gBACd,iCAAiC,EAAE,GAAG,GAAG,IAAI,GAAG,IAAI,EAAE,QAAQ;gBAC9D,8BAA8B,EAAE,GAAG,GAAG,IAAI,GAAG,IAAI,EAAK,QAAQ;gBAC9D,wBAAwB,EAAE,KAAK;gBAC/B,2BAA2B,EAAE,IAAI;gBACjC,qCAAqC,EAAE,CAAC;gBACxC,mCAAmC,EAAE,CAAC;gBACtC,8CAA8C,EAAE,MAAM;aACvD,CAAC;YAEF,IAAI,CAAC,MAAM,GAAG,IAAI,oCAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAE1E,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACxE,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,wBAAe,CAAC,wBAAwB,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,KAAc,CAAC,CAAC;QAC5G,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,wBAAe,CAAC,sBAAsB,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAO,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAEnD,MAAM,gBAAgB,GAAG,IAAI,4BAAgB,EAAE,CAAC;YAChD,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAE7B,IAAI,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,QAAuB,EAAE,EAAE;gBAC9C,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;gBACjD,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC,CAAW,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACnB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAClB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;gBAC9B,MAAM,CAAC,IAAI,4BAAmB,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YACrE,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,UAA4B;QAC9C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,MAAO,CAAC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,KAAU,EAAE,QAAoB,EAAE,EAAE;gBACzF,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC;gBACpC,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACpB,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAc;QACxB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,iBAAiB,GAAG,KAAK,CAAC,CAAC,qBAAqB;YAEtD,MAAM,OAAO,GAAG;gBACd,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,iBAAiB;aACzC,CAAC;YAEF,MAAM,IAAI,GAAG,IAAI,CAAC,MAAO,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAEhE,oCAAoC;YACpC,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,EAAE,CAAC;YAEd,MAAM,kBAAkB,GAAoD,EAAE,CAAC;YAE/E,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,UAAsB,EAAE,EAAE;gBACzC,MAAM,UAAU,GAAG,UAAU,CAAC,kBAAkB,EAAE,CAAC;gBACnD,MAAM,QAAQ,GAAG,UAAU,CAAC,gBAAgB,EAAE,CAAC;gBAE/C,gFAAgF;gBAChF,kBAAkB,CAAC,IAAI,CAAC;oBACtB,MAAM,EAAE,UAAU;oBAClB,IAAI,EAAE,QAAQ;iBACf,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;gBAC9B,MAAM,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAClB,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,IAAI,CAAC,yBAAyB,CAAC,kBAAkB,CAAC,CAAC;oBAClE,OAAO,CAAC,MAAM,CAAC,CAAC;gBAClB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,IAAI,oBAAW,CAAC,+BAA+B,KAAK,EAAE,CAAC,CAAC,CAAC;gBAClE,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,yBAAyB,CAAC,QAAyD;QACzF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,oBAAK,EAAE,CAAC;QACrB,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QAEvD,OAAO,IAAA,2BAAY,EAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED;;;OAGG;IACK,sBAAsB,CAAC,QAAyD;QACtF,MAAM,QAAQ,GAAiB,EAAE,CAAC;QAElC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC;YAEjC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,SAAS;YACX,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACvD,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5B,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;OAGG;IACK,gBAAgB,CAAC,MAAkB,EAAE,IAAiB;QAC5D,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC;QACpC,MAAM,UAAU,GAAG,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC;QACrC,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QAC3D,MAAM,WAAW,GAAG,CAAC,GAAG,CAAC,GAAG,aAAa,GAAG,aAAa,GAAG,UAAU,CAAC;QAEvE,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;QAC/C,IAAI,MAAM,GAAG,CAAC,CAAC;QAEf,8DAA8D;QAC9D,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QAE1D,uBAAuB;QACvB,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QAE7D,oCAAoC;QACpC,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/B,MAAM,IAAI,aAAa,CAAC;QAExB,qCAAqC;QACrC,MAAM,IAAI,aAAa,CAAC;QAExB,6BAA6B;QAC7B,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,aAAqB;QAC5C,OAAO,CAAC,CAAC,GAAG,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,MAAkB,EAAE,MAAc,EAAE,KAAa;QACnE,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACpD,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,gBAAgB;QAChD,OAAO,MAAM,GAAG,CAAC,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAiC;QAC3C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAO,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAE/C,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;gBAC9B,MAAM,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAClB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,CAAC,KAAK,IAAI,EAAE;gBACV,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;wBAChC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACnB,CAAC;oBACD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,IAAI,oBAAW,CAAC,yBAAyB,KAAK,EAAE,CAAC,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;QACP,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc;QAC3B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAO,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAa,EAAE,CAAC;YAE7B,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,MAAc,EAAE,EAAE;gBACjC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;gBAC9B,MAAM,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAClB,OAAO,CAAC,OAAO,CAAC,CAAC;YACnB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAmB;QACnC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,IAAI,IAAI,oBAAQ,EAAE,CAAC;QAEjD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAO,CAAC,WAAW,CAAC,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpE,MAAM,OAAO,GAAiB,EAAE,CAAC;YAEjC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,UAAsB,EAAE,EAAE;gBACzC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;gBAC9B,MAAM,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAClB,OAAO,CAAC,OAAO,CAAC,CAAC;YACnB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,UAA4B;QAC1C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,MAAO,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,KAAU,EAAE,QAAa,EAAE,EAAE;gBAC9E,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC;gBACpC,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC;wBACH,MAAM,WAAW,GAAG,QAAQ,CAAC,cAAc,EAAE,CAAC;wBAC9C,MAAM,MAAM,GAAG,qBAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;wBAC1C,OAAO,CAAC,MAAM,CAAC,CAAC;oBAClB,CAAC;oBAAC,OAAO,UAAU,EAAE,CAAC;wBACpB,MAAM,CAAC,IAAI,oBAAW,CAAC,2BAA2B,UAAU,EAAE,CAAC,CAAC,CAAC;oBACnE,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,iBAAK,EAAE,CAAC;QAE1B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAO,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5D,MAAM,OAAO,GAAiB,EAAE,CAAC;YAEjC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,UAAsB,EAAE,EAAE;gBACzC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;gBAC9B,MAAM,CAAC,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBAClB,OAAO,CAAC,OAAO,CAAC,CAAC;YACnB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;IACH,CAAC;CACF;AAxXD,oCAwXC"}

package/dist/flightsql-client.d.ts

@@ -0,0 +1,48 @@
+import { Schema, Table } from 'apache-arrow';
+import { FlightClient } from './flight-client';
+import { FlightSQLClientConfig, PreparedStatement, TableMetadata } from './types';
+/**
+ * FlightSQL client implementation extending the base Flight client.
+ * Provides SQL-specific operations on top of Apache Arrow Flight protocol.
+ */
+export declare class FlightSQLClient extends FlightClient {
+    private static readonly TYPE_URLS;
+    constructor(config: FlightSQLClientConfig);
+    /**
+     * Wraps a FlightSQL command in a google.protobuf.Any message.
+     * This is required by the FlightSQL specification for proper command serialization.
+     */
+    private packCommand;
+    /**
+     * Executes a SQL query and returns the raw Arrow result.
+     * Provides access to schema information and Arrow batches.
+     */
+    execute(query: string): Promise<Table>;
+    /**
+     * Creates a FlightDescriptor for a FlightSQL command.
+     */
+    private createCommandDescriptor;
+    /**
+     * Gets a ticket from a FlightDescriptor for query execution.
+     */
+    private getQueryTicket;
+    getQuerySchema(query: string): Promise<Schema>;
+    prepare(query: string): Promise<PreparedStatement>;
+    executePrepared(prepared: PreparedStatement): Promise<any[]>;
+    closePrepared(prepared: PreparedStatement): Promise<void>;
+    getCatalogs(): Promise<string[]>;
+    getSchemas(catalog?: string): Promise<Array<{
+        catalog: string;
+        schema: string;
+    }>>;
+    getTables(catalog?: string, dbSchema?: string, tableName?: string, tableTypes?: string[]): Promise<Array<{
+        catalog: string;
+        schema: string;
+        tableName: string;
+        tableType: string;
+    }>>;
+    getTableTypes(): Promise<string[]>;
+    getPrimaryKeys(catalog: string, dbSchema: string, tableName: string): Promise<TableMetadata['primaryKeys']>;
+    getForeignKeys(catalog: string, dbSchema: string, tableName: string): Promise<TableMetadata['foreignKeys']>;
+}
+//# sourceMappingURL=flightsql-client.d.ts.map

package/dist/flightsql-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"flightsql-client.d.ts","sourceRoot":"","sources":["../src/flightsql-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,MAAM,EAAE,KAAK,EAAC,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAC,YAAY,EAAC,MAAM,iBAAiB,CAAC;AAe7C,OAAO,EAAC,qBAAqB,EAAE,iBAAiB,EAAE,aAAa,EAAC,MAAM,SAAS,CAAC;AAGhF;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,YAAY;IAE/C,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAW/B;gBAEU,MAAM,EAAE,qBAAqB;IAIzC;;;OAGG;IACH,OAAO,CAAC,WAAW;IAOnB;;;OAGG;IACG,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;IAiB5C;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAO/B;;OAEG;YACW,cAAc;IAgBtB,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAe9C,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAyBlD,eAAe,CAAC,QAAQ,EAAE,iBAAiB,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IA4B5D,aAAa,CAAC,QAAQ,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAczD,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAyBhC,UAAU,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IA+BjF,SAAS,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,KAAK,CAAC;QAC7G,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC,CAAC;IAkCG,aAAa,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAyBlC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;IAqC3G,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;CAuClH"}