Class SchemaEngineWasm

packages/migrate/src/SchemaEngineWasm.ts:46–308

 * - Catch and throw "rich" connector errors
 */
export class SchemaEngineWasm implements SchemaEngine {
  private engine: wasm.SchemaEngineWasm
  private errorRegistry: ErrorRegistry

  // TODO: forward enabled preview features to the Wasm engine
  private enabledPreviewFeatures?: string[]

  // `isRunning` is set to true when the engine is initialized, and set to false when the engine is stopped
  public isRunning = false

  private constructor({ debug, enabledPreviewFeatures, engine, errorRegistry }: SchemaEngineWasmOptions) {
    this.enabledPreviewFeatures = enabledPreviewFeatures
    if (debug) {
      Debug.enable('prisma:schemaEngine*')
    }
    this.engine = engine
    this.errorRegistry = errorRegistry
  }

  static async setup({ adapter, schemaContext, ...rest }: SchemaEngineWasmSetupInput): Promise<SchemaEngineWasm> {
    const debug = (arg: string) => {
      debugStderr(arg)
    }

    // Note: `datamodels` must be either `undefined` or a *non-empty* `LoadedFile[]`.
    const datamodels = schemaContext?.schemaFiles
    const engine = await wasmSchemaEngineLoader.loadSchemaEngine(
      {
        datamodels,
      },
      debug,
      adapter,
    )
    return new SchemaEngineWasm({ ...rest, engine, errorRegistry: adapter.errorRegistry })
  }

  private async runCommand<M extends keyof SchemaEngineMethods>(
    command: M,
    input: SchemaEngineInput<M>,
  ): Promise<Awaited<SchemaEngineOutput<M>>> {
    if (process.env.FORCE_PANIC_SCHEMA_ENGINE && command !== 'debugPanic') return this.debugPanic()

    this.isRunning = true

    debugStdout('[%s] input: %o', command, input)

    try {
      // Don't modify this by extracting the method call into a variable,
      // as it breaks the binding to the WebAssembly instance, and triggers the
      // `TypeError: Cannot read properties of undefined (reading '__wbg_ptr')`
      // error.
      const result = await (this.engine[command] as (_: SchemaEngineInput<M>) => SchemaEngineOutput<M>)(input)
      debugStdout('[%s] result: %o', command, result)
      return result
    } catch (error) {
      const e = error as Error
      debugStdout('[%s] error: %o', command, e)