jason/ThunderAI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed a possible race condition injecting the script. Thanks to John Bieling.

Browse source
This commit is contained in:
mic 2024-04-10 23:14:01 +02:00
parent 5e4de1c01a
commit 0ecf655245
1 changed files with 34 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
mzta-background.js
View file

@ -79,27 +79,46 @@ messenger.runtime.onMessage.addListener(async (message, sender, sendResponse) =>
});
async function openChatGPT(promptText,action,curr_tabId){
async function openChatGPT(promptText, action, curr_tabId) {
let prefs = await browser.storage.sync.get(prefs_default);
prefs = checkScreenDimensions(prefs);
return browser.windows.create({
let newWindow = await browser.windows.create({
url: "https://chat.openai.com",
type: "popup",
width: prefs.chatgpt_win_width,
height: prefs.chatgpt_win_height
}).then((newWindow) => {
});
console.log("Script started...");
createdWindowID = newWindow.id;
//console.log(promptText);
const tabId = newWindow.tabs[0].id;
browser.tabs.executeScript(tabId,{code: mzta_script})
const createdTab = newWindow.tabs[0];
// Wait for tab loaded.
await new Promise(resolve => {
const tabIsLoaded = tab => {
return tab.status == "complete" && tab.url != "about:blank";
};
const listener = (tabId, changeInfo, updatedTab) => {
if (tabIsLoaded(updatedTab)) {
browser.tabs.onUpdated.removeListener(listener);
resolve();
}
}
// Early exit if loaded already
if (tabIsLoaded(createdTab)) {
resolve();
}
browser.tabs.onUpdated.addListener(listener);
})
browser.tabs.executeScript(createdTab.id, { code: mzta_script, matchAboutBlank: false })
.then(async () => {
console.log("Script injected successfully");
browser.tabs.sendMessage(tabId, {command: "chatgpt_send", prompt: promptText, action: action, tabId: curr_tabId});
}).catch(err => {
browser.tabs.sendMessage(createdTab.id, { command: "chatgpt_send", prompt: promptText, action: action, tabId: curr_tabId });
})
.catch(err => {
console.error("Error injecting the script: ", err);
});
});
}
function checkScreenDimensions(prefs){