Spaces:
Running
Running
def generate_download_settings_js(dl_fn, control_ids): | |
js_code = """ | |
async (""" + ", ".join([f'{ctrl[0]}' for ctrl in control_ids]) + """) => { | |
const password = prompt("Please enter a password for encryption", " "); | |
if (!password) { | |
alert("No password provided. Cancelling download."); | |
return; | |
} | |
let settings = {""" + ", ".join([f'"{ctrl[0]}": {ctrl[0]}' for ctrl in control_ids]) + """}; | |
const settingsStr = JSON.stringify(settings); | |
const textEncoder = new TextEncoder(); | |
const encodedSettings = textEncoder.encode(settingsStr); | |
const salt = crypto.getRandomValues(new Uint8Array(16)); | |
const passwordBuffer = textEncoder.encode(password); | |
const keyMaterial = await crypto.subtle.importKey('raw', passwordBuffer, {name: 'PBKDF2'}, false, ['deriveKey']); | |
const key = await crypto.subtle.deriveKey( | |
{name: 'PBKDF2', salt: salt, iterations: 100000, hash: 'SHA-256'}, | |
keyMaterial, | |
{name: 'AES-GCM', length: 256}, | |
false, | |
['encrypt'] | |
); | |
const iv = crypto.getRandomValues(new Uint8Array(12)); | |
const encryptedSettings = await crypto.subtle.encrypt({name: 'AES-GCM', iv: iv}, key, encodedSettings); | |
const blob = new Blob([salt, iv, new Uint8Array(encryptedSettings)], {type: 'application/octet-stream'}); | |
const url = URL.createObjectURL(blob); | |
const a = document.createElement('a'); | |
a.href = url; | |
a.download = '""" + dl_fn + """'; | |
document.body.appendChild(a); | |
a.click(); | |
document.body.removeChild(a); | |
URL.revokeObjectURL(url); | |
}""" | |
return js_code | |
def generate_upload_settings_js(control_ids): | |
js_code = """ | |
async () => { | |
const input = document.createElement('input'); | |
input.type = 'file'; | |
input.onchange = async e => { | |
const file = e.target.files[0]; | |
if (!file) { | |
alert("No file selected."); | |
return; | |
} | |
const password = prompt("Please enter the password for decryption", " "); | |
if (!password) { | |
alert("No password provided. Cancelling upload."); | |
return; | |
} | |
const arrayBuffer = await file.arrayBuffer(); | |
const salt = arrayBuffer.slice(0, 16); | |
const iv = arrayBuffer.slice(16, 28); | |
const encryptedData = arrayBuffer.slice(28); | |
const textEncoder = new TextEncoder(); | |
const passwordBuffer = textEncoder.encode(password); | |
const keyMaterial = await crypto.subtle.importKey('raw', passwordBuffer, {name: 'PBKDF2'}, false, ['deriveKey']); | |
const key = await crypto.subtle.deriveKey( | |
{name: 'PBKDF2', salt: salt, iterations: 100000, hash: 'SHA-256'}, | |
keyMaterial, | |
{name: 'AES-GCM', length: 256}, | |
false, | |
['decrypt'] | |
); | |
try { | |
const decryptedData = await crypto.subtle.decrypt({name: 'AES-GCM', iv: iv}, key, encryptedData); | |
const textDecoder = new TextDecoder(); | |
const settingsStr = textDecoder.decode(decryptedData); | |
const settings = JSON.parse(settingsStr); | |
""" + "\n".join([f'document.querySelector("{ctrl[1]}").value = settings["{ctrl[0]}"];' for ctrl in control_ids]) + """ | |
""" + "\n".join([f'document.querySelector("{ctrl[1]}").dispatchEvent(new InputEvent("input", {{ bubbles: true }}));' for ctrl in control_ids]) + """ | |
} catch (err) { | |
alert("Failed to decrypt. Check your password and try again."); | |
console.error("Decryption failed:", err); | |
} | |
}; | |
input.click(); | |
}""" | |
return js_code |