feat: ✨ referrer / origin website of s3 buckets

AlecBlance · Oct 20, 2024 · 3779fe3 · 3779fe3
1 parent 31fa3dc
commit 3779fe3
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/src/components/Bucket.tsx b/src/components/Bucket.tsx
@@ -93,7 +93,12 @@ const Bucket = ({
           </div>
         </AccordionTrigger>
         {isPermPresent && (
-          <AccordionContent className="bg-secondary p-2 text-xs">
+          <AccordionContent className="space-y-2 bg-secondary p-2 text-xs">
+            {info.origin && (
+              <p>
+                <b>Origin:</b> {info.origin}
+              </p>
+            )}
             <div className="flex flex-col gap-y-2">
               {Object.entries(info.permissions).map(([key, value]) => (
                 <div key={key} className="flex flex-col gap-y-1">

diff --git a/src/lib/recorder.ts b/src/lib/recorder.ts
@@ -17,9 +17,6 @@ const getPerms = ($: cheerio.CheerioAPI, hostname: string) => {
   const hasListBucket = $("ListBucketResult");
   let type = "";
   const date = new Date().getTime();
-  console.log("Date", date);
-  console.log("Hostname", hostname);
-
   try {
     if (!hasUri.length && !hasCode.length && hasListBucket)
       throw new Error("No permissions");
@@ -99,7 +96,10 @@ export const recordBuckets = async (
   storage.set({
     buckets: {
       ...buckets,
-      [bucketInfo.type]: [bucketInfo.info, ...buckets[bucketInfo.type]],
+      [bucketInfo.type]: [
+        { ...bucketInfo.info, origin: response.initiator },
+        ...buckets[bucketInfo.type],
+      ],
     },
   });
   if (bucketInfo.type === "good") addNumber(buckets);

diff --git a/src/types/index.ts b/src/types/index.ts
@@ -4,6 +4,7 @@ export interface IBucketInfo {
   permissions: Record<string, string[]>;
   date: number;
   hostname: string;
+  origin: string;
 }
 export type ILastSeen = Record<string, number>;