nautilus-orm-codegen 1.0.1

// Runtime file — do not edit manually.

import * as readline from 'readline';
import { EngineProcess } from './_engine.js';
import { IsolationLevel, TransactionClient } from './_transaction.js';
import { errorFromCode, HandshakeError, ProtocolError } from './_errors.js';

export class NautilusClient {
  constructor(schemaPath, options) {
    this.schemaPath = schemaPath;
    this.engine = new EngineProcess(undefined, options?.migrate ?? false);
    this.nextId = 0;
    this.pending = new Map();
    this.partialData = new Map();
    this.rl = null;
    this._delegates = {};
  }

  async connect() {
    if (this.engine.isRunning()) return;

    this.engine.spawn(this.schemaPath);
    this._startReading();
    await this._handshake();

    process.once('exit', () => { this.engine.terminate().catch(() => {}); });
  }

  async disconnect() {
    this.rl?.close();
    this.rl = null;

    await this.engine.terminate();

    const err = new ProtocolError('Client disconnected');
    for (const { reject } of this.pending.values()) reject(err);
    this.pending.clear();
    this.partialData.clear();
  }

  async _rpc(method, params) {
    if (!this.engine.isRunning()) {
      throw new ProtocolError('Engine is not running. Call connect() first.');
    }

    const id      = ++this.nextId;
    const payload = JSON.stringify({ jsonrpc: '2.0', id, method, params }, this._jsonReplacer) + '\n';

    return new Promise((resolve, reject) => {
      this.pending.set(id, { resolve, reject });

      this.engine.stdin.write(payload, (err) => {
        if (err) {
          this.pending.delete(id);
          reject(new ProtocolError(`Write failed: ${err.message}`));
        }
      });
    });
  }

  _jsonReplacer(_key, value) {
    if (value instanceof Date)   return value.toISOString();
    if (value instanceof Buffer) return value.toString('base64');
    return value;
  }

  _startReading() {
    const stdout = this.engine.stdout;
    this.rl = readline.createInterface({ input: stdout, crlfDelay: Infinity });

    this.rl.on('line', (line) => {
      const trimmed = line.trim();
      if (!trimmed) return;

      let response;
      try {
        response = JSON.parse(trimmed);
      } catch {
        console.error('[nautilus-js] Failed to parse response:', trimmed);
        return;
      }

      const id = response.id;
      if (id == null) return;

      const pending = this.pending.get(id);
      if (!pending) return;

      if (response.partial === true) {
        const chunkData = (response.result?.['data']) ?? [];
        if (!this.partialData.has(id)) this.partialData.set(id, []);
        this.partialData.get(id).push(...chunkData);
      } else {
        this.pending.delete(id);

        if (response.error) {
          this.partialData.delete(id);
          pending.reject(errorFromCode(
            response.error.code,
            response.error.message,
            response.error.data,
          ));
        } else {
          let result = response.result;

          if (this.partialData.has(id)) {
            const accumulated = this.partialData.get(id);
            this.partialData.delete(id);
            if (result && Array.isArray(result['data'])) {
              result = { ...result, data: [...accumulated, ...result['data']] };
            } else {
              result = { ...(result ?? {}), data: accumulated };
            }
          }

          pending.resolve(result);
        }
      }
    });

    this.rl.on('close', () => {
      const stderr = this.engine.getStderrOutput().trim();
      const msg    = stderr
        ? `Engine process exited unexpectedly.\nDetails: ${stderr}`
        : 'Engine process exited unexpectedly (no output on stderr).';
      const err = new ProtocolError(msg);
      for (const { reject } of this.pending.values()) reject(err);
      this.pending.clear();
      this.partialData.clear();
    });
  }

  async _handshake() {
    let response;
    try {
      response = await this._rpc('engine.handshake', {
        protocolVersion: 1,
        clientName:      'nautilus-js',
        clientVersion:   '0.1.0',
      });
    } catch (e) {
      await this.disconnect();
      throw new HandshakeError(`Handshake failed: ${e}`);
    }

    const v = response?.['protocolVersion'];
    if (v !== 1) {
      await this.disconnect();
      throw new HandshakeError(
        `Protocol version mismatch: engine uses ${v}, client expects 1`,
      );
    }
  }

  async _startTransaction(timeoutMs = 5000, isolationLevel) {
    const params = { protocolVersion: 1, timeoutMs };
    if (isolationLevel != null) params['isolationLevel'] = isolationLevel;
    const result = await this._rpc('transaction.start', params);
    return result['id'];
  }

  async _commitTransaction(txId) {
    await this._rpc('transaction.commit', { protocolVersion: 1, id: txId });
  }

  async _rollbackTransaction(txId) {
    try {
      await this._rpc('transaction.rollback', { protocolVersion: 1, id: txId });
    } catch { /* best-effort */ }
  }

  async _runTransactionBatch(operations, options) {
    const params = {
      protocolVersion: 1,
      operations,
      timeoutMs: options?.timeout ?? 5000,
    };
    if (options?.isolationLevel != null) params['isolationLevel'] = options.isolationLevel;

    const result = await this._rpc('transaction.batch', params);
    return Array.isArray(result?.['results']) ? result['results'] : [];
  }

  async _runTransactionCallback(fn, options) {
    const txId = await this._startTransaction(
      options?.timeout ?? 5000,
      options?.isolationLevel,
    );
    const tx = new TransactionClient(this, txId);
    try {
      const result = await fn(tx);
      await this._commitTransaction(txId);
      return result;
    } catch (e) {
      await this._rollbackTransaction(txId);
      throw e;
    }
  }
}