import { v4 as uuidv4 } from "uuid";
import { Team } from "@server/models";
import { uploadToS3FromUrl } from "@server/utils/s3";
import BaseTask, { TaskPriority } from "./BaseTask";

type Props = {
  /* The teamId to operate on */
  teamId: string;
  /* The original avatarUrl from the SSO provider */
  avatarUrl: string;
};

/**
 * A task that uploads the provided avatarUrl to S3 storage and updates the
 * team's record with the new url.
 */
export default class UploadTeamAvatarTask extends BaseTask<Props> {
  public async perform(props: Props) {
    const team = await Team.findByPk(props.teamId, {
      rejectOnEmpty: true,
    });

    const avatarUrl = await uploadToS3FromUrl(
      props.avatarUrl,
      `avatars/${team.id}/${uuidv4()}`,
      "public-read"
    );

    if (avatarUrl) {
      await team.update({ avatarUrl });
    }
  }

  public get options() {
    return {
      attempts: 3,
      priority: TaskPriority.Normal,
    };
  }
}