import { DashUploadUtils } from '../DashUploadUtils';
import { _invalid, _success, Method } from '../RouteManager';
import ApiManager, { Registration } from './ApiManager';

export default class FireflyManager extends ApiManager {
    getBearerToken = () =>
        fetch('https://ims-na1.adobelogin.com/ims/token/v3', {
            method: 'POST',
            headers: {
                'Content-Type': 'application/x-www-form-urlencoded',
            },
            body: `grant_type=client_credentials&client_id=${process.env._CLIENT_FIREFLY_CLIENT_ID}&client_secret=${process.env._CLIENT_FIREFLY_SECRET}&scope=openid,AdobeID,session,additional_info,read_organizations,firefly_api,ff_apis`,
        }).catch(error => {
            console.error('Error:', error);
            return '';
        });
    askFirefly = (prompt: string = 'a realistic illustration of a cat coding') => {
        const fetched = this.getBearerToken().then(response =>
            response.json().then((data: { access_token: string }) =>
                fetch('https://firefly-api.adobe.io/v3/images/generate', {
                    method: 'POST',
                    headers: [
                        ['Content-Type', 'application/json'],
                        ['Accept', 'application/json'],
                        ['x-api-key', process.env._CLIENT_FIREFLY_CLIENT_ID ?? ''],
                        ['Authorization', `Bearer ${data.access_token}`],
                    ],
                    body: `{  "prompt": "${prompt}" }`,
                })
                    .then(response => response.json().then(json => JSON.stringify((json.outputs?.[0] as { image: { url: string } })?.image)))
                    .catch(error => {
                        console.error('Error:', error);
                        return '';
                    })
            )
        );
        return fetched;
    };
    protected initialize(register: Registration): void {
        register({
            method: Method.POST,
            subscription: '/queryFireflyImage',
            secureHandler: ({ req, res }) =>
                this.askFirefly(req.body.prompt).then(fire =>
                    DashUploadUtils.UploadImage(JSON.parse(fire).url).then(info => {
                        if (info instanceof Error) _invalid(res, info.message);
                        else _success(res, info.accessPaths.agnostic.client);
                    })
                ),
        });
    }
}