import { Transaction } from "sequelize";
import { v4 as uuidv4 } from "uuid";
import { AttachmentPreset } from "@shared/types";
import { Attachment, Event, User } from "@server/models";
import AttachmentHelper from "@server/models/helpers/AttachmentHelper";
import FileStorage from "@server/storage/files";

type BaseProps = {
  id?: string;
  name: string;
  user: User;
  source?: "import";
  preset: AttachmentPreset;
  ip?: string;
  transaction?: Transaction;
};

type UrlProps = BaseProps & {
  url: string;
};

type BufferProps = BaseProps & {
  buffer: Buffer;
  type: string;
};

type Props = UrlProps | BufferProps;

export default async function attachmentCreator({
  id,
  name,
  user,
  source,
  preset,
  ip,
  transaction,
  ...rest
}: Props): Promise<Attachment | undefined> {
  const acl = AttachmentHelper.presetToAcl(preset);
  const key = AttachmentHelper.getKey({
    acl,
    id: uuidv4(),
    name,
    userId: user.id,
  });

  let attachment;

  if ("url" in rest) {
    const { url } = rest;
    const res = await FileStorage.storeFromUrl(url, key, acl);

    if (!res) {
      return;
    }
    attachment = await Attachment.create(
      {
        id,
        key,
        acl,
        size: res.contentLength,
        contentType: res.contentType,
        teamId: user.teamId,
        userId: user.id,
      },
      {
        transaction,
      }
    );
  } else {
    const { buffer, type } = rest;
    await FileStorage.store({
      body: buffer,
      contentType: type,
      contentLength: buffer.length,
      key,
      acl,
    });

    attachment = await Attachment.create(
      {
        id,
        key,
        acl,
        size: buffer.length,
        contentType: type,
        teamId: user.teamId,
        userId: user.id,
      },
      {
        transaction,
      }
    );
  }

  await Event.create(
    {
      name: "attachments.create",
      data: {
        name,
        source,
      },
      modelId: attachment.id,
      teamId: user.teamId,
      actorId: user.id,
      ip,
    },
    {
      transaction,
    }
  );

  return attachment;
}