96 lines
No EOL
3.5 KiB
JavaScript
96 lines
No EOL
3.5 KiB
JavaScript
/*
|
|
* ThunderAI [https://micz.it/thunderbird-addon-thunderai/]
|
|
* Copyright (C) 2024 Mic (m@micz.it)
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*
|
|
* This file contains a modified version of the code from the project at https://github.com/boxabirds/chatgpt-frontend-nobuild
|
|
* The original code has been released under the Apache License, Version 2.0.
|
|
*/
|
|
|
|
|
|
|
|
// The controller wires up all the components and workers together,
|
|
// managing the dependencies. A kind of "DI" class.
|
|
const worker = new Worker('model-worker.js', { type: 'module' });
|
|
|
|
const messagesArea = document.querySelector('messages-area');
|
|
messagesArea.init(worker);
|
|
|
|
// Initialize the messageInput component and pass the worker to it
|
|
const messageInput = document.querySelector('message-input');
|
|
messageInput.init(worker);
|
|
messageInput.setMessagesArea(messagesArea);
|
|
|
|
// Data received from the user
|
|
let promptData = null;
|
|
|
|
// ============================== TESTING
|
|
// let browser = {
|
|
// i18n: {
|
|
// getMessage: async function(key) {
|
|
// return key;
|
|
// }
|
|
// },
|
|
// storage: {
|
|
// sync: {
|
|
// get: async function(key) {
|
|
// return 'apitest';
|
|
// }
|
|
// }
|
|
// }
|
|
// }
|
|
// ============================== TESTING - END
|
|
|
|
const params = new URLSearchParams(window.location.search);
|
|
let prefs_api = await browser.storage.sync.get({api_key_chatgpt: '', model_chatgpt: ''});
|
|
// const openaiApiKey = params.get('openapi-key');
|
|
//console.log(">>>>>>>>>>> api_key_chatgpt: " + prefs_api_key.api_key_chatgpt);
|
|
worker.postMessage({ type: 'init', api_key_chatgpt: prefs_api.api_key_chatgpt, model_chatgpt: prefs_api.model_chatgpt});
|
|
messagesArea.appendUserMessage("Will attempt to connect to OpenAI using the API key provided.", "info");
|
|
|
|
// Event listeners for worker messages
|
|
// TODO I'm sure there's a better way to do this
|
|
worker.onmessage = function(event) {
|
|
const { type, payload } = event.data;
|
|
switch (type) {
|
|
case 'messageSent':
|
|
messageInput.handleMessageSent();
|
|
break;
|
|
case 'newToken':
|
|
messagesArea.handleNewToken(payload.token);
|
|
break;
|
|
case 'tokensDone':
|
|
messagesArea.handleTokensDone(promptData);
|
|
messageInput.enableInput();
|
|
break;
|
|
case 'error':
|
|
messagesArea.appendBotMessage(payload,'error');
|
|
break;
|
|
default:
|
|
console.error('[ThunderAI] Unknown event type from API worker:', type);
|
|
}
|
|
};
|
|
|
|
// handling commands from the backgound page
|
|
browser.runtime.onMessage.addListener((message, sender, sendResponse) => {
|
|
//console.log(">>>>>>>>>>>>> controller.js onMessage: " + JSON.stringify(message));
|
|
if (message.command === "chatgpt_send") {
|
|
//send the received prompt to the chatgpt api
|
|
promptData = message;
|
|
messageInput._setMessageInputValue(message.prompt);
|
|
messageInput._handleNewChatMessage();
|
|
}
|
|
}); |