From d220c22291e4106bd6d9f3ef5d343e1e33b48cae Mon Sep 17 00:00:00 2001
From: Kuba Orlik <kontakt@kuba-orlik.name>
Date: Sat, 6 Nov 2021 20:02:02 +0100
Subject: [PATCH] Add functioning cookiesOnly filter

---
 manifest.json      |  4 ++-
 request-cluster.ts |  8 ++++--
 sidebar.tsx        | 69 +++++++++++++++++++++++++++++++---------------
 3 files changed, 56 insertions(+), 25 deletions(-)

diff --git a/manifest.json b/manifest.json
index b54d864..c1220db 100644
--- a/manifest.json
+++ b/manifest.json
@@ -18,7 +18,9 @@
     "storage",
     "<all_urls>",
     "webRequest",
-    "webRequestBlocking"
+    "webRequestBlocking",
+    "cookies",
+    "privacy"
   ],
   "browser_specific_settings": {
     "gecko": {
diff --git a/request-cluster.ts b/request-cluster.ts
index 50623ba..5ef4a2d 100644
--- a/request-cluster.ts
+++ b/request-cluster.ts
@@ -17,7 +17,7 @@ export class StolenDataEntry {
   ) {
     try {
       this.iab = TCString.decode(value);
-      console.log(this.iab);
+      // console.log(this.iab);
       this.isIAB = true;
     } catch (e) {}
   }
@@ -52,13 +52,17 @@ export class RequestCluster extends EventEmitter {
     return false;
   }
 
-  getStolenData(filter: { minValueLength: number }): StolenDataEntry[] {
+  getStolenData(filter: {
+    minValueLength: number;
+    cookiesOnly: boolean;
+  }): StolenDataEntry[] {
     return this.requests
       .map((request) => request.getAllStolenData())
       .reduce((a, b) => a.concat(b), [])
       .filter((entry) => {
         return entry.value.length >= filter.minValueLength;
       })
+      .filter((entry) => !filter.cookiesOnly || entry.source === "cookie")
       .sort((entry1, entry2) =>
         entry1.getPriority() > entry2.getPriority() ? -1 : 1
       )
diff --git a/sidebar.tsx b/sidebar.tsx
index 763e184..e72fe98 100644
--- a/sidebar.tsx
+++ b/sidebar.tsx
@@ -48,13 +48,14 @@ const TabDropdown = ({
 const StolenDataRow = ({
   tabID,
   shorthost,
-  refreshToken,
   minValueLength,
+  cookiesOnly,
 }: {
   tabID: number;
   shorthost: string;
   refreshToken: number;
   minValueLength: number;
+  cookiesOnly: boolean;
 }) => {
   const cluster = memory.getClustersForTab(tabID)[shorthost];
   const icons: Record<Sources, string> = {
@@ -71,17 +72,19 @@ const StolenDataRow = ({
       </h2>
       <table>
         <tbody>
-          {cluster.getStolenData({ minValueLength }).map((entry) => (
-            <tr>
-              <th style={{ maxWidth: "200px", wordWrap: "break-word" }}>
-                {entry.name}
-              </th>
-              <td>{icons[entry.source]}</td>
-              <td style={{ wordWrap: "anywhere" as any }}>
-                {entry.value} {entry.isIAB ? "!!!!! IAB" : ""}
-              </td>
-            </tr>
-          ))}
+          {cluster
+            .getStolenData({ minValueLength, cookiesOnly })
+            .map((entry) => (
+              <tr>
+                <th style={{ maxWidth: "200px", wordWrap: "break-word" }}>
+                  {entry.name}
+                </th>
+                <td>{icons[entry.source]}</td>
+                <td style={{ wordWrap: "anywhere" as any }}>
+                  {entry.value} {entry.isIAB ? "!!!!! IAB" : ""}
+                </td>
+              </tr>
+            ))}
         </tbody>
       </table>
     </div>
@@ -92,10 +95,12 @@ const StolenData = ({
   pickedTab,
   refreshToken,
   minValueLength,
+  cookiesOnly,
 }: {
   pickedTab: number | null;
   refreshToken: number;
   minValueLength: number;
+  cookiesOnly: boolean;
 }) => {
   const [tab, setTab] = useState<Tab | null>(null);
   useEffect(() => {
@@ -114,21 +119,29 @@ const StolenData = ({
         <h1>
           <img src={tab.favIconUrl} width="20" height="20" /> {tab.title}
         </h1>
-        {clusters.map((cluster) => (
-          <StolenDataRow
-            tabID={pickedTab}
-            shorthost={cluster.id}
-            key={cluster.id}
-            refreshToken={refreshToken}
-            minValueLength={minValueLength}
-          />
-        ))}
+        {clusters
+          .filter((cluster) => !cookiesOnly || cluster.hasCookies())
+          .map((cluster) => (
+            <StolenDataRow
+              tabID={pickedTab}
+              shorthost={cluster.id}
+              key={cluster.id}
+              refreshToken={refreshToken}
+              minValueLength={minValueLength}
+              cookiesOnly={cookiesOnly}
+            />
+          ))}
       </div>
     </div>
   );
 };
 
-const Options = ({ minValueLength, setMinValueLength }) => {
+const Options = ({
+  minValueLength,
+  setMinValueLength,
+  cookiesOnly,
+  setCookiesOnly,
+}) => {
   return (
     <fieldset>
       <h3>Zaawansowane ustawienia</h3>
@@ -141,6 +154,14 @@ const Options = ({ minValueLength, setMinValueLength }) => {
         value={minValueLength}
         onChange={(e) => setMinValueLength(parseInt(e.target.value))}
       />
+      <br />
+      <input
+        type="checkbox"
+        id="cookiesOnly"
+        value={cookiesOnly}
+        onChange={(e) => setCookiesOnly(e.target.checked)}
+      />
+      <label htmlFor="cookiesOnly">Pokazuj tylko dane z cookiesów</label>
     </fieldset>
   );
 };
@@ -148,6 +169,7 @@ const Options = ({ minValueLength, setMinValueLength }) => {
 const Sidebar = () => {
   const [pickedTab, setPickedTab] = useState<number | null>(null);
   const [minValueLength, setMinValueLength] = useState<number | null>(7);
+  const [cookiesOnly, setCookiesOnly] = useState<boolean>(false);
   const counter = useEmitter(memory);
   return (
     <>
@@ -163,11 +185,14 @@ const Sidebar = () => {
       <Options
         minValueLength={minValueLength}
         setMinValueLength={setMinValueLength}
+        cookiesOnly={cookiesOnly}
+        setCookiesOnly={setCookiesOnly}
       />
       <StolenData
         pickedTab={pickedTab}
         refreshToken={counter}
         minValueLength={minValueLength}
+        cookiesOnly={cookiesOnly}
       />
     </>
   );