From 3e44e17ea83486699c9aa5aeb65a2a4904ba4ee9 Mon Sep 17 00:00:00 2001
From: flawedworld <38294951+flawedworld@users.noreply.github.com>
Date: Mon, 1 Mar 2021 19:55:57 +0000
Subject: Add ability to remove non-stock key via WebUSB.

---
 static/js/web-install.js | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'static/js/web-install.js')

diff --git a/static/js/web-install.js b/static/js/web-install.js
index 9571b49f..9b88502e 100644
--- a/static/js/web-install.js
+++ b/static/js/web-install.js
@@ -207,6 +207,19 @@ async function flashRelease(setProgress) {
     return `Flashed ${latestZip} to device.`;
 }
 
+async function eraseNonStockKey(setProgress) {
+    await ensureConnected(setProgress);
+
+    setProgress("Erasing key...");
+    try {
+        await device.runCommand("erase:avb_custom_key");
+    } catch (error) {
+        console.log(error);
+        throw error;
+    }
+    return "Key erased.";
+}
+
 async function lockBootloader(setProgress) {
     await ensureConnected(setProgress);
 
@@ -281,6 +294,7 @@ if ("usb" in navigator) {
     addButtonHook("download-release", downloadRelease);
     addButtonHook("flash-release", flashRelease);
     addButtonHook("lock-bootloader", lockBootloader);
+    addButtonHook("remove-custom-key", eraseNonStockKey);
 } else {
     console.log("WebUSB unavailable");
 }
-- 
cgit v1.3.1