import axios from 'axios';
import FormData from 'form-data';
import fs from 'fs';
import path from 'path';

const API_ENDPOINT = 'https://api.autocaption.io/v1'
const API_KEY = 'YOUR_API_KEY'

const videoName = 'video.mp4'
const videoLang = 'fr'
const videoPath = path.join('./', videoName);

const axiosInstance = axios.create({
baseURL: API_ENDPOINT
});

axiosInstance.defaults.headers.common['x-api-key'] = API_KEY;

const video = fs.createReadStream(videoPath);

// Get signed url to upload video
const getSignedUrlData = async () => {
return axiosInstance.get(`/signedUrl?file=${videoName}`).then(res => res.data);
}

// Upload video to signed url
const uploadVideo = async (url, fields) => {
const formData = new FormData();
for (const [key, value] of Object.entries(fields)) {
    formData.append(key, value);
}
formData.append('file', video);
await axios.post(url, formData, {
    headers: {
        'Content-Type': `multipart/form-data; boundary=${formData._boundary}`
    }
}).catch(err => console.log(err));
}

// Transcript video
const transcriptVideo = async (fileKey, lang) => {
return axiosInstance.post('/transcript', {
    input: fileKey,
    language: lang
}).then(res => res.data);
}

// Get transcript status
const getTranscriptStatus = async (jobId) => {
return axiosInstance.get(`/transcript/${jobId}`).then(res => res.data).catch(err => err.response.data);
}

// Get template by id
const getTemplateById = async (id) => {
return axiosInstance.get(`/templates/${id}`).then(res => res.data);
}

// Generate video
const generateVideo = async (file, config, words) => {
return axiosInstance.post('/generate', {
    input: file,
    config,
    words
}).then(res => res.data);
}

// Get fonts list
const getFontsList = async () => {
return axiosInstance.get(`/fonts?lang=${videoLang}`).then(res => res.data);
}

const main = async () => {
try {
    const { file, url, fields } = await getSignedUrlData();
    await uploadVideo(url, fields);
    console.log('file uploaded')
    const { job_id } = await transcriptVideo(file, videoLang);

    // Initial check without delay
    let transcriptResponse = await getTranscriptStatus(job_id);

    // Loop until the transcript is ready or an error occurs
    while (transcriptResponse.status !== 'COMPLETED') {
        if (transcriptResponse.status == 'FAILED') {
            throw new Error('Transcription failed'); // Throws an error to exit the loop and catch block
        }

        await new Promise(resolve => setTimeout(resolve, 5000)); // Wait for 5 seconds
        transcriptResponse = await getTranscriptStatus(job_id); // Update the response
    }

    console.log('Transcript ready');

    // Get the template DEFAULT
    const { config } = await getTemplateById('0');

    /* 
        - If the template doesn't have a font, get the list of fonts and use the first one
        /!\ Set a value to the config.font otherwise it will throw an error during the video generation
    */
    if (config.font == undefined) {
        const fonts = await getFontsList();
        config.font = fonts[0];
    }

    /* 
        - Instead of a custom template, the default templates don't have the sizes defined, so we need to define them.
        /!\ Set a value to the config.sizes.fontSize otherwise it will throw an error during the video generation
    */
    config.sizes = {
        fontSize: 56,
        strokeSize: 15,
        //...
    };

    // Generate the video and wait the download url - Can take a few minutes
    console.log('Generating video...');
    const { download_url } = await generateVideo(file, config, transcriptResponse.words);
    console.log(download_url);
} catch (error) {
    console.error("An error occurred:", error);
}
};

main();