From 9b40f88db3cd0a4b3ed57356d95c2c3d590753ef Mon Sep 17 00:00:00 2001
From: Kuba Orlik <kontakt@kuba-orlik.name>
Date: Sun, 7 Nov 2021 19:27:03 +0100
Subject: [PATCH] Option to show only entries with either cookies or history

---
 options.tsx        | 12 ++++++++++++
 request-cluster.ts | 11 +++++++++++
 2 files changed, 23 insertions(+)

diff --git a/options.tsx b/options.tsx
index 9f242a8..c6e1765 100644
--- a/options.tsx
+++ b/options.tsx
@@ -5,6 +5,8 @@ export default function Options({
   setMinValueLength,
   cookiesOnly,
   setCookiesOnly,
+  cookiesOrOriginOnly,
+  setCookiesOrOriginOnly,
 }) {
   return (
     <fieldset>
@@ -26,6 +28,16 @@ export default function Options({
         onChange={(e) => setCookiesOnly(e.target.checked)}
       />
       <label htmlFor="cookiesOnly">Pokazuj tylko dane z cookiesów</label>
+      <br />
+      <input
+        type="checkbox"
+        id="cookiesOrOriginOnly"
+        value={cookiesOrOriginOnly}
+        onChange={(e) => setCookiesOrOriginOnly(e.target.checked)}
+      />
+      <label htmlFor="cookiesOrOriginOnly">
+        Pokazuj tylko dane z cookiesów lub z częścią historii przeglądania
+      </label>
     </fieldset>
   );
 }
diff --git a/request-cluster.ts b/request-cluster.ts
index e19f042..bf4ea05 100644
--- a/request-cluster.ts
+++ b/request-cluster.ts
@@ -26,6 +26,7 @@ export class RequestCluster extends EventEmitter {
   getStolenData(filter: {
     minValueLength: number;
     cookiesOnly: boolean;
+    cookiesOrOriginOnly: boolean;
   }): MergedStolenDataEntry[] {
     return this.requests
       .map((request) => request.stolenData)
@@ -34,6 +35,12 @@ export class RequestCluster extends EventEmitter {
         return entry.value.length >= filter.minValueLength;
       })
       .filter((entry) => !filter.cookiesOnly || entry.source === "cookie")
+      .filter(
+        (entry) =>
+          !filter.cookiesOrOriginOnly ||
+          entry.source === "cookie" ||
+          entry.classification === "history"
+      )
       .sort((entryA, entryB) => (entryA.name > entryB.name ? -1 : 1))
       .filter((element, index, array) => {
         // remove duplicates by name/value
@@ -103,4 +110,8 @@ export class RequestCluster extends EventEmitter {
       .map((request) => request.getMarkedEntries())
       .reduce(reduceConcat, []);
   }
+
+  exposesOrigin() {
+    return this.requests.some((request) => request.exposesOrigin());
+  }
 }