wvg/popup.js
2024-05-03 02:35:48 +09:00

85 lines
3.5 KiB
JavaScript

let psshs=chrome.extension.getBackgroundPage().psshs;
let requests=chrome.extension.getBackgroundPage().requests;
let pageURL=chrome.extension.getBackgroundPage().pageURL;
async function guess(){
//Be patient!
document.body.style.cursor = "wait";
document.getElementById("guess").disabled=true
//Init Pyodide
let pyodide = await loadPyodide();
await pyodide.loadPackage(["certifi-2024.2.2-py3-none-any.whl","charset_normalizer-3.3.2-py3-none-any.whl","construct-2.8.8-py2.py3-none-any.whl","idna-3.6-py3-none-any.whl","packaging-23.2-py3-none-any.whl","protobuf-4.24.4-cp312-cp312-emscripten_3_1_52_wasm32.whl","pycryptodome-3.20.0-cp35-abi3-emscripten_3_1_52_wasm32.whl","pymp4-1.4.0-py3-none-any.whl","pyodide_http-0.2.1-py3-none-any.whl","pywidevine-1.8.0-py3-none-any.whl","requests-2.31.0-py3-none-any.whl","urllib3-2.2.1-py3-none-any.whl"].map(e=>"wheels/"+e))
//Configure Guesser
pyodide.globals.set("pssh", document.getElementById('pssh').value);
pyodide.globals.set("licUrl", requests[userInputs['license']]['url']);
pyodide.globals.set("licHeaders", requests[userInputs['license']]['headers']);
pyodide.globals.set("licBody", requests[userInputs['license']]['body']);
let pre=await fetch('python/pre.py').then(res=>res.text())
let after=await fetch('python/after.py').then(res=>res.text())
let scheme=await fetch(`python/schemes/${document.getElementById("scheme").value}.py`).then(res=>res.text())
//Get result
let result = await pyodide.runPythonAsync([pre, scheme, after].join("\n"));
document.getElementById('result').value=result;
//Save history
let historyData={
PSSH: document.getElementById('pssh').value,
KEYS: result.split("\n").slice(0,-1)
}
chrome.storage.local.set({[pageURL]: historyData}, function () {});
//All Done!
document.body.style.cursor = "auto";
document.getElementById("guess").disabled=false
}
function copyResult(){
this.select();
navigator.clipboard.writeText(this.value);
}
window.corsFetch = (u, m, h, b) => {
return new Promise((resolve, reject) => {
chrome.tabs.query({ url:pageURL }, (tabs) => {
chrome.tabs.sendMessage(tabs[0].id, {type:"FETCH", u:u, m:m, h:h, b:b}, (res) => {
resolve(res)
})
})
})
}
async function autoSelect(){
let selectRules = await fetch("selectRules.conf").then((r)=>r.text());
selectRules = selectRules.replace(/\n^$/gm, "");
selectRules=selectRules.split("\n").map(function(row){return row.split("$$");});
for(var item of selectRules){
let search = requests.map(r => r['url']).findIndex(e => e.includes(item[0]));
if(search>=0){
if(item[1]) document.getElementById("scheme").value = item[1];
requestList.children[search].click();
break;
}
}
if(psshs.length==1){
document.getElementById('pssh').value=psshs[0];
}
if(requests.length==1){
requestList.children[0].click();
}
}
if(psshs.length!=0){
document.addEventListener('DOMContentLoaded', function() {
document.getElementById('noEME').style.display='none';
document.getElementById('home').style.display='block';
document.getElementById('guess').addEventListener("click", guess);
document.getElementById('result').addEventListener("click", copyResult);
drawList(psshs,'psshSearch','psshList','pssh');
drawList(requests.map(r => r['url']),'requestSearch','requestList','license');
autoSelect();
});
}