const AWS = require('aws-sdk')
const fs = require('node:fs')

const keys = require('../s3.json')

async function uploadVideo(video) {
  const items = keys.videos
  const key = items[Math.floor(Math.random() * items.length)]

  const s3 = new AWS.S3({
    accessKeyId: key.access,
    secretAccessKey: key.secret,
    endpoint: 'https://gateway.storjshare.io'
  })

  const videoFile = fs.createReadStream(video)
  await s3.upload({
    Bucket: 'video',
    Key: video.split('/')[2],
    Body: videoFile,
    ContentType: 'video/webm',
  }).promise()

  return `${key.url}${video.split('/')[2]}`
}

async function uploadImage(id, url) {
  const items = keys.images
  const key = items[Math.floor(Math.random() * items.length)]

  const s3 = new AWS.S3({
    accessKeyId: key.access,
    secretAccessKey: key.secret,
    endpoint: 'https://gateway.storjshare.io'
  })

  const exists = await checkIfFileExists({
    Bucket: 'media', 
    Key: `${id}.webp`
  }, s3)
  
  if (exists) {
    return `${key.url}${id}.webp`
  } else {
    const response = await fetch(url)
    const buffer = Buffer.from(await response.arrayBuffer())

    await s3.upload({
      Bucket: 'media',
      Key: `${id}.webp`,
      Body: buffer,
      ContentType: 'video/webp',
    }).promise()

    return `${key.url}${id}.webp`
  }
}

async function checkIfFileExists(params, s3) {
  try {
    await s3.headObject(params).promise()
    return true
  } catch (err) {
    return false
  }
}

module.exports = { uploadVideo, uploadImage }