#!/usr/bin/env node

import fs from 'node:fs'
import url from 'node:url'
import path from 'node:path'

import { CloudFront } from '@aws-sdk/client-cloudfront'
import { S3 } from '@aws-sdk/client-s3'
import { Upload } from '@aws-sdk/lib-storage'
import mime from 'mime-types'
import readDir from 'recursive-readdir'

import pkg from '../lerna.json' with { type: 'json' }

const __dirname = path.dirname(url.fileURLToPath(import.meta.url))

const region = 'eu-west-1'
const PKG_VERSION = pkg.version
const PRODUCTION_VERSION = 'v9'
const DISTRIBUTION_ID = process.env.DISTRIBUTION_ID
const BUCKET_NAME = 'webdriver.io'
const BUILD_DIR = path.resolve(__dirname, '..', 'website', 'build')
const UPLOAD_OPTIONS = { partSize: 10 * 1024 * 1024, queueSize: 1 }
const IGNORE_FILE_SUFFIX = ['*.rb']

const timestamp = Date.now()
const s3 = new S3({ region })
const files: string[] = await readDir(BUILD_DIR, IGNORE_FILE_SUFFIX)

const version = `v${PKG_VERSION.split('.')[0]}`
const bucketName = version === PRODUCTION_VERSION ? BUCKET_NAME : `${version}.${BUCKET_NAME}`

/**
 * Split array in series
 * @param list array to split
 * @param size size of each chunk
 * @returns array of chunks
 */
function splitInSeries<T>(list: T[], size: number): T[][] {
    const result: T[][] = []

    for (let i = 0; i < list.length; i += size) {
        result.push(list.slice(i, i + size))
    }

    return result
}

/**
 * upload assets
 */
console.log(`Uploading ${BUILD_DIR} to S3 bucket ${bucketName}`)
for (const filesBatch of splitInSeries(files, 10)) {
    await Promise.all(filesBatch.map(async (file) => {
        try {
            const mimeType = mime.lookup(file)
            if (!mimeType) {
                throw new Error(`Couldn't find mime type for ${file}`)
            }

            const res = await new Upload({
                client: s3,
                params: {
                    Bucket: bucketName,
                    Key: file.replace(BUILD_DIR + '/', ''),
                    Body: fs.createReadStream(file),
                    ContentType: mimeType,
                    ACL: 'public-read',
                },
                ...UPLOAD_OPTIONS
            }).done()
            console.log(`${file} uploaded`)
            return res
        } catch (err) {
            console.error(`Couldn't upload file ${file}: ${(err as Error).stack}`)
            throw err
        }
    }))
}

/**
 * invalidate distribution
 */
const distributionId = version === PRODUCTION_VERSION
    ? DISTRIBUTION_ID
    : process.env[`DISTRIBUTION_ID_${version.toUpperCase()}`]
if (distributionId) {
    console.log(`Invalidate objects from distribution ${distributionId}`)
    const cloudfront = new CloudFront({ region })
    const { Invalidation } = await cloudfront.createInvalidation({
        DistributionId: distributionId,
        InvalidationBatch: {
            CallerReference: `${timestamp}`,
            Paths: { Quantity: 1, Items: ['/*'] }
        }
    })
    console.log(`Created new invalidation with ID ${Invalidation?.Id}`)
}

/**
 * delete old assets
 */
const objects = await s3.listObjects({
    Bucket: bucketName
})
if (!objects.Contents) {
    throw new Error('Couldn\'t find any objects')
}
const objectsToDelete = objects.Contents.filter((obj) => obj.LastModified && obj.LastModified.getTime() < timestamp)
console.log(`Found ${objectsToDelete.length} outdated objects to remove...`)

await Promise.all(objectsToDelete.map((obj) => (
    s3.deleteObject({
        Bucket: bucketName,
        Key: obj.Key
    })
)))
console.log('Deleted obsolete items successfully')
console.log('Successfully updated webdriver.io docs')