import JSEncrypt from "jsencrypt";

interface MessageEvent {
    data: {
        dataList: string[];
        privateKey: string;
    };
}

const decrypt = (enc: string, privateKey: string): string | false => {
    const jsEncrypt = new JSEncrypt();
    jsEncrypt.setPrivateKey(privateKey);

    let ret = "";
    // base64エンコード文字列の末尾にある==を目印に分割
    const encArray = enc.split("==");
    let errorFlag = false;
    encArray.forEach((element: string) => {
        if (!errorFlag && element != "") {
            const decrypted = jsEncrypt.decrypt(element + "==");
            if (decrypted === false) {
                errorFlag = true;
                return; // foreachを抜けるreturn
            }
            ret += decrypted;
        }
    });
    return errorFlag ? false : ret;
};

// メインスレッドからのpostMessage()を受信して復号処理を行うためのイベントリスナーを設定
self.addEventListener("message", (event: MessageEvent) => {
    const dataList = event.data.dataList;
    const resultList: string[] = [];

    for (let i = 0; i < dataList.length; i++) {
        // 進捗をメインスレッドに送信
        self.postMessage({ event: "progress", progress: i });

        const data = dataList[i];
        if (data == "") {
            resultList.push("");
            continue;
        }

        const result = decrypt(data, event.data.privateKey);
        if (result === false) {
            // 復号に失敗した場合はメインスレッドにエラーを送信して終了
            self.postMessage({
                event: "error",
                progress: i,
                message: "復号に失敗しました。暗号キーファイルが正しいことを確認してください。",
            });
            return;
        }
        resultList.push(result);
    }
    // 最後に完了と復号した文字列レコードの配列をメインスレッドに送信
    self.postMessage({ event: "complete", progress: dataList.length, resultList: resultList });
});

import { saveAs } from "file-saver";
import DecryptWorker from './decrypt.worker?worker';

interface WorkerMessage {
    data: {
        event: 'progress' | 'complete' | 'error';
        progress: number;
        resultList?: string[]; // eventがcomplete時のみ
        message?: string; // eventがerror時のみ
    }
}

// ワーカーオブジェクトを生成
const worker = new DecryptWorker();

// ワーカーからのメッセージを受信した際の処理を登録
worker.onmessage = (message: WorkerMessage) => {
    switch (message.data.event) {
        case 'progress':
            // 進捗
            updateProgress(message.data.progress);
            break;
        case 'complete':
            // 完了
            updateProgress(message.data.progress);
            const resultList = message.data.resultList!;

            // 復号データをファイルダウンロードする
            const fileName = 'decrypt.txt';
            saveAs(new Blob([resultList.join("\n")], { type: "text/plain;charset=utf-8" }), fileName);
            break;
        case 'error':
            console.error('worker error : ' + message.data.message!);
            break;
    }
};

// ワーカーで例外が発生した際の処理を登録
worker.onerror = (err: any) => {
    console.log('worker exception!');
};

// ワーカーにメッセージを送信し、処理を開始する
worker.postMessage({ dataList: encryptDataList, privateKey: privateKey });