From 25cc7ab4f50c1b9ac896625f50108b93dd52d9c1 Mon Sep 17 00:00:00 2001
From: Louis Dion-Marcil <maden.ldm@gmail.com>
Date: Wed, 3 Apr 2024 20:58:52 -0400
Subject: [PATCH] Implement "Auto Download", downloads every callback to local
 disk

---
 src/components/header/index.tsx   | 25 +++++++++++++++++++++++++
 src/components/header/styles.scss | 14 ++++++++++++++
 src/lib/index.ts                  |  6 ++++++
 src/lib/localStorage/index.ts     |  1 +
 src/lib/types/storedData.ts       |  1 +
 src/pages/homePage/index.tsx      |  4 ++++
 6 files changed, 51 insertions(+)

diff --git a/src/components/header/index.tsx b/src/components/header/index.tsx
index aac0d86..8d55e3c 100644
--- a/src/components/header/index.tsx
+++ b/src/components/header/index.tsx
@@ -11,7 +11,9 @@ import { ReactComponent as ThemeDarkButtonIcon } from "assets/svg/theme_dark_but
 import { ReactComponent as ThemeSynthButtonIcon } from "assets/svg/theme_synth_button.svg";
 import NotificationsPopup from "components/notificationsPopup";
 import ResetPopup from "components/resetPopup";
+import ToggleBtn from "components/toggleBtn";
 import { handleDataExport } from "lib";
+import { getStoredData, writeStoredData } from "lib/localStorage";
 import { ThemeName, showThemeName } from "theme";
 import "./styles.scss";
 
@@ -69,6 +71,19 @@ const Header = ({
     </button>
   );
 
+  const data = getStoredData();
+  const [inputData, setInputData] = useState<any>({
+    responseExport: data.responseExport
+  });
+
+  const handleToggleBtn = (e: any) => {
+
+    const currentStoredData = getStoredData();
+
+    setInputData({ ...inputData, responseExport: e.target.checked});
+    writeStoredData({ ...currentStoredData, responseExport: e.target.checked});
+  };
+
   return (
     <div id="header" className="header">
       <div>interactsh</div>
@@ -77,6 +92,16 @@ const Header = ({
         <ThemeButton theme="synth" />
         <ThemeButton theme="blue" />
       </div>
+
+      <div>Auto Download</div>
+      <div>
+        <ToggleBtn
+          name="responseExport"
+          onChangeHandler={handleToggleBtn}
+          value={inputData.responseExport}
+        />
+      </div>
+
       <div className="links">
         <button
           type="button"
diff --git a/src/components/header/styles.scss b/src/components/header/styles.scss
index c065e2a..c50a313 100644
--- a/src/components/header/styles.scss
+++ b/src/components/header/styles.scss
@@ -32,6 +32,20 @@
     margin-right: 0.8rem;
     border-right: 1px solid rgba(255, 255, 255, 0.25);
   }
+  & > div:nth-of-type(2) {
+    letter-spacing: -0.02rem;
+    text-align: left;
+    padding-right: 1rem;
+    color: #ffffff;
+    margin-right: 0.8rem;
+    border-right: 1px solid rgba(255, 255, 255, 0.25);
+  }
+  & > div:nth-of-type(3) {
+    text-align: left;
+    color: #ffffff;
+    font-size: 1.4rem;
+    padding-right: 1rem;
+  }
   & > button:nth-of-type(1) {
     margin-left: 0;
     border-radius: 0.6rem;
diff --git a/src/lib/index.ts b/src/lib/index.ts
index 2b16484..8b85b0c 100644
--- a/src/lib/index.ts
+++ b/src/lib/index.ts
@@ -90,6 +90,11 @@ const getData = (key: string) =>
     O.chain(O.fromNullable)
   );
 
+export const handleResponseExport = (item : any) => {
+  const fileName = `${format(Date.now(), "yyyy-MM-dd_hh:mm")}_${item.protocol}_${item['remote-address']}_${item['full-id']}_${item.id}.txt`;
+  downloadData(item['raw-request'], fileName);
+}
+
 export const handleDataExport = () => {
   const values = pipe(
     R.mapWithIndex((key) => ({ key, data: getData(key) }))(localStorage),
@@ -200,6 +205,7 @@ export const register = (
           host,
           correlationIdLength: currentData.correlationIdLength,
           correlationIdNonceLength: currentData.correlationIdNonceLength,
+          responseExport: false,
           increment: 1,
           token,
           telegram: {
diff --git a/src/lib/localStorage/index.ts b/src/lib/localStorage/index.ts
index a25d3ff..40fc1c2 100644
--- a/src/lib/localStorage/index.ts
+++ b/src/lib/localStorage/index.ts
@@ -13,6 +13,7 @@ export const defaultStoredData: StoredData = {
   correlationId: "",
   correlationIdLength: process.env.REACT_APP_CIDL ? +process.env.REACT_APP_CIDL : 20,
   correlationIdNonceLength: process.env.REACT_APP_CIDN ? +process.env.REACT_APP_CIDN : 13,  
+  responseExport: false,
   secretKey: "",
   data: [],
   aesKey: "",
diff --git a/src/lib/types/storedData.ts b/src/lib/types/storedData.ts
index 309b579..06be1c2 100644
--- a/src/lib/types/storedData.ts
+++ b/src/lib/types/storedData.ts
@@ -22,6 +22,7 @@ export const StoredData_ = summon((F) =>
       correlationId: F.string(),
       correlationIdLength: F.number(),
       correlationIdNonceLength: F.number(),
+      responseExport: F.boolean(),
       theme: ThemeName(F),
 
       publicKey: F.string(),
diff --git a/src/pages/homePage/index.tsx b/src/pages/homePage/index.tsx
index b66b88a..51d027e 100644
--- a/src/pages/homePage/index.tsx
+++ b/src/pages/homePage/index.tsx
@@ -20,6 +20,7 @@ import {
   poll,
   decryptAESKey,
   processData,
+  handleResponseExport,
   copyDataToClipboard,
   clearIntervals,
   register,
@@ -213,6 +214,9 @@ const HomePage = () => {
 
           // eslint-disable-next-line array-callback-return
           const formattedString = processedData.map((item: any) => {
+
+            storedData.responseExport && handleResponseExport(item)
+
             const telegramMsg = `<i>[${item['full-id']}]</i> Received <i>${item.protocol.toUpperCase()}</i> interaction from <b><a href="https://ipinfo.io/${item['remote-address']}">${item['remote-address']}</a></b> at <i>${format(new Date(item.timestamp), "yyyy-mm-dd_hh:mm:ss")}</i>`
             storedData.telegram.enabled && notifyTelegram(telegramMsg, storedData.telegram.botToken, storedData.telegram.chatId, 'HTML')
             return {