diff --git a/src/main/java/com/ghost/temple/CookieManager.java b/src/main/java/com/ghost/temple/CookieManager.java index 1552bf9..a316d3a 100644 --- a/src/main/java/com/ghost/temple/CookieManager.java +++ b/src/main/java/com/ghost/temple/CookieManager.java @@ -11,10 +11,17 @@ public class CookieManager { private static final String COOKIE_FILE = "cookies.data"; public static void saveCookies(WebDriver driver) { + Set cookies = driver.manage().getCookies(); + System.out.println("🍪 Saving cookies (" + cookies.size() + " total):"); + for (Cookie c : cookies) { + System.out.println(" - " + c.getName() + " = " + c.getValue() + ", domain=" + c.getDomain()); + } + try (ObjectOutputStream oos = new ObjectOutputStream(new FileOutputStream(COOKIE_FILE))) { - oos.writeObject(driver.manage().getCookies()); + oos.writeObject(cookies); System.out.println("🍪 Cookies saved successfully!"); } catch (IOException e) { + System.err.println("❌ Failed to save cookies:"); e.printStackTrace(); } } @@ -29,13 +36,16 @@ public class CookieManager { try (ObjectInputStream ois = new ObjectInputStream(new FileInputStream(file))) { Set cookies = (Set) ois.readObject(); + System.out.println("🍪 Loading cookies (" + cookies.size() + " total):"); driver.get(url); // must open the domain before adding cookies for (Cookie cookie : cookies) { + System.out.println(" - Adding cookie: " + cookie.getName() + " = " + cookie.getValue() + ", domain=" + cookie.getDomain()); driver.manage().addCookie(cookie); } driver.navigate().refresh(); System.out.println("✅ Cookies loaded successfully!"); } catch (IOException | ClassNotFoundException e) { + System.err.println("❌ Failed to load cookies:"); e.printStackTrace(); } } diff --git a/src/main/java/com/ghost/temple/StorageManager.java b/src/main/java/com/ghost/temple/StorageManager.java index f3c0791..cb8447a 100644 --- a/src/main/java/com/ghost/temple/StorageManager.java +++ b/src/main/java/com/ghost/temple/StorageManager.java @@ -20,14 +20,18 @@ public class StorageManager { String localStorage = (String) js.executeScript("return JSON.stringify(window.localStorage);"); String sessionStorage = (String) js.executeScript("return JSON.stringify(window.sessionStorage);"); + System.out.println("💾 Saving localStorage: " + localStorage); + System.out.println("💾 Saving sessionStorage: " + sessionStorage); + String json = "{\"localStorage\":" + localStorage + ",\"sessionStorage\":" + sessionStorage + "}"; try (FileWriter writer = new FileWriter(STORAGE_FILE)) { writer.write(json); } - System.out.println("✅ localStorage and sessionStorage saved"); + System.out.println("✅ localStorage and sessionStorage saved successfully"); } catch (IOException e) { + System.err.println("❌ Error saving storage:"); e.printStackTrace(); } } @@ -36,6 +40,8 @@ public class StorageManager { public static void loadStorage(WebDriver driver, String url) { try { String json = new String(Files.readAllBytes(Paths.get(STORAGE_FILE)), StandardCharsets.UTF_8); + System.out.println("📂 Loaded storage JSON: " + json); + driver.get(url); // Open page first to set storage JavascriptExecutor js = (JavascriptExecutor) driver; @@ -47,7 +53,7 @@ public class StorageManager { ); driver.navigate().refresh(); // Refresh so storage is applied - System.out.println("✅ localStorage and sessionStorage loaded"); + System.out.println("✅ localStorage and sessionStorage loaded successfully"); } catch (IOException e) { System.out.println("⚠️ Storage file not found, skipping load"); } diff --git a/src/main/java/com/ghost/temple/hooks/Hooks.java b/src/main/java/com/ghost/temple/hooks/Hooks.java index 42fca4d..75a7349 100644 --- a/src/main/java/com/ghost/temple/hooks/Hooks.java +++ b/src/main/java/com/ghost/temple/hooks/Hooks.java @@ -28,7 +28,7 @@ public class Hooks { } StorageManager.saveStorage(DriverManager.getDriver()); // Save local/session storage CookieManager.saveCookies(DriverManager.getDriver()); // Save cookies - //DriverManager.quitDriver(); // Then quit driver + DriverManager.quitDriver(); // Then quit driver } } \ No newline at end of file