import type {
  PTableShape,
  PTableVector,
  TableRange,
  UniqueValuesResponse,
  DataQuery,
} from "@milaboratories/pl-model-common";
import type { PFrameInternal } from "@milaboratories/pl-model-middle-layer";

export interface NodeFrameSymbol
  // oxlint-disable-next-line
  extends Symbol {}

export interface NodeTableSymbol
  // oxlint-disable-next-line
  extends Symbol {}

export interface CancellationTokenSymbol
  // oxlint-disable-next-line
  extends Symbol {}

interface Cancellation {
  cancellationTokenCreate: () => CancellationTokenSymbol;
  cancellationTokenCancel: (token: CancellationTokenSymbol) => void;
}

interface HeapProfiler {
  pprofDump: () => Promise<Uint8Array>;
}

interface NodeFrame {
  pFrameCreate: (spillPath: string, logger: PFrameInternal.Logger | undefined) => NodeFrameSymbol;
  pFrameSetDataSource: (
    pFrame: NodeFrameSymbol,
    dataSource: PFrameInternal.PFrameDataSourceV2,
  ) => void;
  pFrameAddColumns: (
    pFrame: NodeFrameSymbol,
    columns: PFrameInternal.AddColumnEntryV2[],
    token: CancellationTokenSymbol | undefined,
  ) => Promise<void>;
  pFrameDispose: (pFrame: NodeFrameSymbol) => void;
  pFrameCreateTable: (
    pFrame: NodeFrameSymbol,
    id: PFrameInternal.PTableId,
    dataQuery: DataQuery,
  ) => NodeTableSymbol;
  pFrameGetUniqueValues: (
    pFrame: NodeFrameSymbol,
    id: PFrameInternal.PTableId,
    request: PFrameInternal.UniqueValuesRequestV2,
    token: CancellationTokenSymbol | undefined,
  ) => Promise<UniqueValuesResponse>;
}

interface NodeTable {
  pTableGetFootprint: (
    boxed: NodeTableSymbol,
    withPredecessors: boolean,
    token: CancellationTokenSymbol | undefined,
  ) => Promise<number>;
  pTableGetShape: (
    boxed: NodeTableSymbol,
    token: CancellationTokenSymbol | undefined,
  ) => Promise<PTableShape>;
  pTableGetData: (
    boxed: NodeTableSymbol,
    columnIndices: number[],
    range: TableRange | undefined,
    token: CancellationTokenSymbol | undefined,
  ) => Promise<PTableVector[]>;
  pTableExport: (
    boxed: NodeTableSymbol,
    path: string,
    headers: [number, string][],
    token: CancellationTokenSymbol | undefined,
  ) => Promise<void>;
  pTableDispose: (boxed: NodeTableSymbol) => void;
}

export interface AddonSymbol
  // oxlint-disable-next-line
  extends Symbol, Cancellation, HeapProfiler, NodeFrame, NodeTable {}