signal-desktop/ts/util/handleVideoAttachment.ts

// Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only

import { blobToArrayBuffer } from 'blob-util';

import * as log from '../logging/log';
import { makeVideoScreenshot } from '../types/VisualAttachment';
import { IMAGE_PNG, stringToMIMEType } from '../types/MIME';
import type { InMemoryAttachmentDraftType } from '../types/Attachment';
import { fileToBytes } from './fileToBytes';

export async function handleVideoAttachment(
  file: File,
  options?: { generateScreenshot: boolean }
): Promise<InMemoryAttachmentDraftType> {
  const objectUrl = URL.createObjectURL(file);
  if (!objectUrl) {
    throw new Error('Failed to create object url for video!');
  }
  try {
    const data = await fileToBytes(file);
    const attachment: InMemoryAttachmentDraftType = {
      contentType: stringToMIMEType(file.type),
      data,
      fileName: file.name,
      path: file.name,
      pending: false,
      size: data.byteLength,
    };

    if (options?.generateScreenshot) {
      const screenshotContentType = IMAGE_PNG;

      const screenshotBlob = await makeVideoScreenshot({
        objectUrl,
        contentType: screenshotContentType,
        logger: log,
      });
      attachment.screenshotData = new Uint8Array(
        await blobToArrayBuffer(screenshotBlob)
      );
      attachment.screenshotContentType = screenshotContentType;
    }

    return attachment;
  } finally {
    URL.revokeObjectURL(objectUrl);
  }
}