import { Worker } from 'jest-worker'
import * as path from 'path'
import { execOnce } from '../../../lib/utils'
import ImageData from './image_data'

const getWorker = execOnce(
  () =>
    new Worker(path.resolve(__dirname, 'impl'), {
      enableWorkerThreads: true,
    })
)

export async function decodeBuffer(buffer: Buffer): Promise<ImageData> {
  const worker: typeof import('./impl') = getWorker() as any
  return ImageData.from(await worker.decodeBuffer(buffer))
}

export async function rotate(
  image: ImageData,
  numRotations: number
): Promise<ImageData> {
  const worker: typeof import('./impl') = getWorker() as any
  return ImageData.from(await worker.rotate(image, numRotations))
}

export async function resize(
  image: ImageData,
  { width }: { width: number }
): Promise<ImageData> {
  const worker: typeof import('./impl') = getWorker() as any
  return ImageData.from(await worker.resize(image, { width }))
}

export async function encodeJpeg(
  image: ImageData,
  { quality }: { quality: number }
): Promise<Buffer> {
  const worker: typeof import('./impl') = getWorker() as any
  const o = await worker.encodeJpeg(image, { quality })
  return Buffer.from(o)
}

export async function encodeWebp(
  image: ImageData,
  { quality }: { quality: number }
): Promise<Buffer> {
  const worker: typeof import('./impl') = getWorker() as any
  const o = await worker.encodeWebp(image, { quality })
  return Buffer.from(o)
}

export async function encodePng(image: ImageData): Promise<Buffer> {
  const worker: typeof import('./impl') = getWorker() as any
  const o = await worker.encodePng(image)
  return Buffer.from(o)
}