Files
liusida-ComfyUI-Login/js/logout.js
T
2024-06-07 10:17:38 +08:00

24 lines
827 B
JavaScript

import { app } from "../../scripts/app.js";
app.registerExtension({
name: "Comfy.Login.Logout",
init() {},
async setup() {
await new Promise(resolve => setTimeout(resolve, 500)); // Delay for 0.5 second before appending the Logout button to the menu, ensuring it is added last.
const menu = document.querySelector(".comfy-menu");
const logoutButton = document.createElement("button");
logoutButton.textContent = "Logout";
logoutButton.onclick = () => {
let workflowData = localStorage.getItem('workflow'); // Save the workflow data
localStorage.clear(); // Clear all items in localStorage
localStorage.setItem('workflow', workflowData); // Restore the workflow data
sessionStorage.clear(); // If you use sessionStorage
window.location.href = "/logout";
}
menu.append(logoutButton);
},
});