import fs from 'node:fs';

let youtubeData: youtubeDataOptions;
if (fs.existsSync('.data/youtube.data')) {
    youtubeData = JSON.parse(fs.readFileSync('.data/youtube.data').toString());
    youtubeData.file = true;
}

interface youtubeDataOptions {
    cookie?: Object;
    file?: boolean;
}

export function getCookies(): undefined | string {
    let result = '';
    if (!youtubeData?.cookie) return undefined;
    for (const [key, value] of Object.entries(youtubeData.cookie)) {
        result += `${key}=${value};`;
    }
    return result;
}

export function setCookie(key: string, value: string): boolean {
    if (!youtubeData?.cookie) return false;
    key = key.trim();
    value = value.trim();
    Object.assign(youtubeData.cookie, { [key]: value });
    return true;
}

export function uploadCookie() {
    if (youtubeData.cookie && youtubeData.file)
        fs.writeFileSync('.data/youtube.data', JSON.stringify(youtubeData, undefined, 4));
}

export function setCookieToken(options: { cookie: string }) {
    let cook = options.cookie;
    let cookie: Object = {};
    cook.split(';').forEach((x) => {
        const arr = x.split('=');
        if (arr.length <= 1) return;
        const key = arr.shift()?.trim() as string;
        const value = arr.join('=').trim();
        Object.assign(cookie, { [key]: value });
    });
    youtubeData = { cookie };
    youtubeData.file = false;
}

/**
 * Updates cookies locally either in file or in memory.
 *
 * Example
 * ```ts
 * const response = ... // Any https package get function.
 *
 * play.cookieHeaders(response.headers['set-cookie'])
 * ```
 * @param headCookie response headers['set-cookie'] array
 * @returns Nothing
 */
export function cookieHeaders(headCookie: string[]): void {
    if (!youtubeData?.cookie) return;
    headCookie.forEach((x: string) => {
        x.split(';').forEach((z) => {
            const arr = z.split('=');
            if (arr.length <= 1) return;
            const key = arr.shift()?.trim() as string;
            const value = arr.join('=').trim();
            setCookie(key, value);
        });
    });
    uploadCookie();
}