// SPDX-FileCopyrightText: 2025 XWiki CryptPad Team <contact@cryptpad.org> and contributors
//
// SPDX-License-Identifier: AGPL-3.0-or-later

define([
    '/common/common-util.js',
    '/components/tweetnacl/nacl-fast.min.js'
], function (Util) {

    var Handler = {};

    var Nacl = window.nacl;
    Handler.formCommandHandlers = function(sframeChan, Utils, nThen, Cryptpad) {
        sframeChan.on('EV_EXPORT_SHEET', function (data) {
            if (!data || !Array.isArray(data.content)) { return; }
            sessionStorage.CP_formExportSheet = JSON.stringify(data);
            var href = Utils.Hash.hashToHref('', 'sheet');
            var a = window.open(href);
            if (!a) { sframeChan.event('EV_POPUP_BLOCKED'); }
            delete sessionStorage.CP_formExportSheet;
        });
        var u8_concat = function (A) {
            var length = 0;
            A.forEach(function (a) { length += a.length; });
            var total = new Uint8Array(length);

            var offset = 0;
            A.forEach(function (a) {
                total.set(a, offset);
                offset += a.length;
            });
            return total;
        };
        var anonProof = function (channel, theirPub, anonKeys) {
            var u8_plain = Util.decodeUTF8(channel);
            var u8_nonce = Nacl.randomBytes(Nacl.box.nonceLength);
            var u8_cipher = Nacl.box(
                u8_plain,
                u8_nonce,
                Util.decodeBase64(theirPub),
                Util.decodeBase64(anonKeys.curvePrivate)
            );
            var u8_bundle = u8_concat([
                u8_nonce, // 24 uint8s
                u8_cipher, // arbitrary length
            ]);
            return {
                key: anonKeys.curvePublic,
                proof: Util.encodeBase64(u8_bundle)
            };
        };
        var deleteLines = false; // "false" to support old forms
        sframeChan.on("Q_FETCH_MY_ANSWERS", function (data, cb) {
            var answers = [];
            var myKeys;
            nThen(function (w) {
                Cryptpad.getFormKeys(w(function (keys) {
                    myKeys = keys;
                }));
                Cryptpad.getFormAnswer({channel: data.channel}, w(function (obj) {
                    if (!obj || obj.error) {
                        if (obj && obj.error === "ENODRIVE") {
                            var answered = JSON.parse(localStorage.CP_formAnswered || "[]");
                            if (answered.indexOf(data.channel) !== -1) {
                                cb({error:'EANSWERED'});
                            } else {
                                cb();
                            }
                            return void w.abort();
                        }
                        w.abort();
                        return void cb(obj);
                    }
                    // Get the latest edit per uid
                    var temp = {};
                    obj.forEach(function (ans) {
                        var uid = ans.uid || '000';
                        temp[uid] = ans;
                    });
                    answers = Object.values(temp);
                }));
                Cryptpad.getPadMetadata({channel: data.channel}, w(function (md) {
                    if (md && md.deleteLines) { deleteLines = true; }
                }));
            }).nThen(function () {
                var n = nThen;
                var err;
                var all = {};
                answers.forEach(function (answer) {
                    n = n(function(waitFor) {
                        var finalKeys = myKeys;
                        if (answer.anonymous) {
                            if (!myKeys.formSeed) {
                                err = 'ANONYMOUS_ERROR';
                                console.error('ANONYMOUS_ERROR', answer);
                                return;
                            }
                            finalKeys = Cryptpad.getAnonymousKeys(myKeys.formSeed, data.channel, Utils);
                        }
                        Cryptpad.getHistoryRange({
                            channel: data.channel,
                            lastKnownHash: answer.hash,
                            toHash: answer.hash,
                        }, waitFor(function (obj) {
                            if (obj && obj.error) { err = obj.error; return; }
                            var messages = obj.messages;
                            if (!messages.length) {
                                // TODO delete from drive.forms?
                                return;
                            }
                            if (obj.lastKnownHash !== answer.hash) { return; }
                            try {
                                var res = Utils.Crypto.Mailbox.openOwnSecretLetter(messages[0].msg, {
                                    validateKey: data.validateKey,
                                    ephemeral_private: Util.decodeBase64(answer.curvePrivate),
                                    my_private: Util.decodeBase64(finalKeys.curvePrivate),
                                    their_public: Util.decodeBase64(data.publicKey)
                                });
                                var parsed = JSON.parse(res.content);
                                parsed._isAnon = answer.anonymous;
                                parsed._time = messages[0].time;
                                if (deleteLines) { parsed._hash = answer.hash; }
                                var uid = parsed._uid || '000';
                                if (all[uid] && !all[uid]._isAnon) { parsed._isAnon = false; }
                                all[uid] = parsed;
                            } catch (e) {
                                err = e;
                            }
                        }));
                    }).nThen;
                });
                n(function () {
                    if (err) { return void cb({error: err}); }
                    cb(all);
                });
            });
        });
        sframeChan.on('Q_FORM_FETCH_ANSWERS', function (data, _cb) {
            var cb = Utils.Util.once(_cb);
            var myKeys = {};
            var myFormKeys;
            var noDriveAnswered = false;
            nThen(function (w) {
                var personalDrive = !Cryptpad.initialTeam || Cryptpad.initialTeam === -1;
                Cryptpad.getAccessKeys(w(function (_keys) {
                    if (!Array.isArray(_keys)) { return; }
                    _keys.some(function (_k) {
                        if ((personalDrive && !_k.id) || Cryptpad.initialTeam === Number(_k.id)) {
                            myKeys = _k;
                            return true;
                        }
                    });
                }));
                Cryptpad.getFormKeys(w(function (keys) {
                    if (!keys.curvePublic && !keys.formSeed) {
                        // No drive mode
                        var answered = JSON.parse(localStorage.CP_formAnswered || "[]");
                        noDriveAnswered = answered.indexOf(data.channel) !== -1;
                    }
                    myFormKeys = keys;
                }));
                Cryptpad.getPadMetadata({channel: data.channel}, w(function (md) {
                    if (md && md.deleteLines) { deleteLines = true; }
                }));
            }).nThen(function () {
                if (myFormKeys.formSeed) {
                    myFormKeys = Cryptpad.getAnonymousKeys(myFormKeys.formSeed, data.channel, Utils);
                }

                let secret, parsed;
                if (data.drive) {
                    parsed = Utils.Hash.parseTypeHash('form', data.href);
                    secret = Utils.Hash.getSecrets('form', data.href, data.password);
                } else {
                    secret = Utils.secret;
                }
                let keys = (secret && secret.keys) || {};
                let formData = Utils.Hash.getFormData(secret);
                const privateKey = parsed?.auditorKey || formData?.form_private;
                const publicKey = formData?.form_public;

                var curvePrivate = privateKey || data.privateKey;
                if (!curvePrivate) { return void cb({error: 'EFORBIDDEN'}); }
                var cryptoKeys = {
                    curvePrivate: curvePrivate,
                    curvePublic: publicKey || data.publicKey,
                    validateKey: data.validateKey
                };

                Cryptpad.getFormResponses({
                    cryptoKeys,
                    channel: data.channel,
                    edPublic: myKeys.edPublic,
                    validateKey: keys.secondaryValidateKey,
                    cantEdit: data.cantEdit,
                    deleteLines
                }, (obj) => {
                    if (obj?.error) { return void cb(obj); }
                    const results = obj.results;

                    // If we have submitted an anonymous answer, retrieve it
                    let myKey;
                    if (myFormKeys.curvePublic && results[myFormKeys.curvePublic]) {
                        myKey = myFormKeys.curvePublic;
                    }

                    cb({
                        noDriveAnswered, myKey, results
                    });
                });
            });
        });
        var noDriveSeed = Utils.Hash.createChannelId();
        sframeChan.on("Q_FORM_SUBMIT", function (data, cb) {
            var box = data.mailbox;
            var myKeys;
            nThen(function (w) {
                Cryptpad.getFormKeys(w(function (keys) {
                    // If formSeed doesn't exists, it means we're probably in noDrive mode.
                    // We can create a seed in localStorage.
                    if (!keys.formSeed) {
                        // No drive mode
                        keys = { formSeed: noDriveSeed };
                    }
                    myKeys = keys;
                }));
            }).nThen(function () {
                var myAnonymousKeys;
                if (data.anonymous) {
                    if (!myKeys.formSeed) { return void cb({ error: "ANONYMOUS_ERROR" }); }
                    myKeys = Cryptpad.getAnonymousKeys(myKeys.formSeed, box.channel, Utils);
                } else {
                    myAnonymousKeys = Cryptpad.getAnonymousKeys(myKeys.formSeed, box.channel, Utils);
                }
                var keys = Utils.secret && Utils.secret.keys;
                myKeys.signingKey = keys.secondarySignKey;

                var ephemeral_keypair = Nacl.box.keyPair();
                var ephemeral_private = Util.encodeBase64(ephemeral_keypair.secretKey);
                myKeys.ephemeral_keypair = ephemeral_keypair;

                if (myAnonymousKeys) {
                    var proof = anonProof(box.channel, box.publicKey, myAnonymousKeys);
                    data.results._proof = proof;
                }

                var crypto = Utils.Crypto.Mailbox.createEncryptor(myKeys);
                var uid = data.results._uid || Utils.Util.uid();
                data.results._uid = uid;
                var text = JSON.stringify(data.results);
                var ciphertext = crypto.encrypt(text, box.publicKey);

                var hash = ciphertext.slice(0,64);
                Cryptpad.anonRpcMsg("WRITE_PRIVATE_MESSAGE", [
                    box.channel,
                    ciphertext
                ], function (err, response) {
                    Cryptpad.storeFormAnswer({
                        uid: uid,
                        channel: box.channel,
                        hash: hash,
                        curvePrivate: ephemeral_private,
                        anonymous: Boolean(data.anonymous)
                    }, function () {
                        var res = data.results;
                        res._isAnon = data.anonymous;
                        res._time = +new Date();
                        if (deleteLines) { res._hash = hash; }
                        cb({
                            error: err,
                            response: response,
                            results: res
                        });
                    });
                });
            });
        });
        sframeChan.on("Q_FORM_DELETE_ALL_ANSWERS", function (data, cb) {
            if (!data || !data.channel) { return void cb({error: 'EINVAL'}); }
            Cryptpad.clearOwnedChannel(data, cb);
        });
        sframeChan.on("Q_FORM_DELETE_ANSWER", function (data, cb) {
            if (!deleteLines) {
                return void cb({error: 'EFORBIDDEN'});
            }
            Cryptpad.deleteFormAnswers(data, cb);
        });
        sframeChan.on("Q_FORM_MUTE", function (data, cb) {
            if (!Utils.secret) { return void cb({error: 'EINVAL'}); }
            Cryptpad.muteChannel(Utils.secret.channel, data.muted, cb);
        });        
    };

    return Handler;
});