import type {Scene} from 'cesium';

interface ScreenshotOptions {
  offsetX: number;
  offsetY: number;
  width: number;
  height: number;
}

/**
 * @param scene
 * @param options
 */
export function takeScreenshot(
  scene: Scene,
  options: ScreenshotOptions,
): Promise<string> {
  return new Promise((resolve, reject) => {
    // preserveDrawingBuffers is false so we render on demand and immediately read the buffer
    const remover = scene.postRender.addEventListener(() => {
      remover();
      try {
        let url;

        if (options) {
          const smallerCanvas = document.createElement('canvas');

          smallerCanvas.width = options.width;
          smallerCanvas.height = options.height;
          smallerCanvas
            .getContext('2d')
            .drawImage(
              scene.canvas,
              options.offsetX,
              options.offsetY,
              options.width,
              options.height,
              0,
              0,
              options.width,
              options.height,
            );
          url = smallerCanvas.toDataURL();
        } else {
          url = scene.canvas.toDataURL();
        }
        resolve(url);
      } catch (e) {
        reject(e);
      }
    });

    scene.requestRender();
  });
}