import { S3 as S3Client } from '@aws-sdk/client-s3';
import type { S3ClientConfig } from '@aws-sdk/client-s3';
import type { IStorageAdapterV2 } from '~/types/nc-plugin';
import GenericS3 from '~/plugins/GenericS3/GenericS3';

interface VultrObjectStorageInput {
  bucket: string;
  region: string;
  access_key: string;
  hostname: string;
  access_secret: string;
  acl?: string;
}

export default class Vultr extends GenericS3 implements IStorageAdapterV2 {
  name = 'Vultr';

  protected input: VultrObjectStorageInput;

  constructor(input: unknown) {
    super(input as VultrObjectStorageInput);
  }

  protected get defaultParams() {
    return {
      Bucket: this.input.bucket,
      ACL: this.input?.acl || 'public-read',
    };
  }

  public async init(): Promise<any> {
    const s3Options: S3ClientConfig = {
      region: this.input.region,
      credentials: {
        accessKeyId: this.input.access_key,
        secretAccessKey: this.input.access_secret,
      },
      endpoint: this.input.hostname,
    };

    this.s3Client = new S3Client(s3Options);
  }
  override getUploadedPath(path: string): { path?: string; url?: string } {
    const updatedEndpoint = this.input.hostname.startsWith('https://')
      ? this.input.hostname
      : `https://${this.input.hostname}`;
    return {
      url: `${updatedEndpoint}/${this.input.bucket}/${path}`,
    };
  }
}
