Support walmart.com and walmart.ca

8hobbies · Dec 26, 2024 · 795920e · 795920e
1 parent b0b3e57
commit 795920e
Show file tree

Hide file tree

Showing 2 changed files with 62 additions and 4 deletions.
diff --git a/src/main.test.ts b/src/main.test.ts
@@ -200,8 +200,8 @@ describe("builtin sitedata", () => {
     });
   }
 
-  for (const neweggSiteId of ["Newegg.ca", "Newegg.com"] as const) {
-    const tld = neweggSiteId.split(".").pop();
+  for (const tld of ["ca", "com"] as const) {
+    const neweggSiteId = `Newegg.${tld}` as const;
 
     test(`${neweggSiteId} activating matched`, async () => {
       expect(
@@ -224,8 +224,8 @@ describe("builtin sitedata", () => {
     });
   }
 
-  for (const targetSiteId of ["Target.com"] as const) {
-    const tld = targetSiteId.split(".").pop();
+  for (const tld of ["com"] as const) {
+    const targetSiteId = `Target.${tld}` as const;
 
     test(`${targetSiteId} activating matched`, async () => {
       expect(
@@ -247,4 +247,28 @@ describe("builtin sitedata", () => {
       ).toBe(`https://www.target.${tld}/s`);
     });
   }
+
+  for (const tld of ["ca", "com"] as const) {
+    const walmartSiteId = `Walmart.${tld}` as const;
+
+    test(`${walmartSiteId} activating matched`, async () => {
+      expect(
+        await generateActivatingUrl(
+          `https://www.walmart.${tld}/search?`,
+          builtinSiteData,
+        ),
+      ).toBe(
+        `https://www.walmart.${tld}/search?${siteParams[walmartSiteId].key}=${encodeURIComponent(siteParams[walmartSiteId].value)}`,
+      );
+    });
+
+    test(`${walmartSiteId} disabling matched`, () => {
+      expect(
+        generateDisablingUrl(
+          `https://www.walmart.${tld}/search?facetedValue=`,
+          builtinSiteData,
+        ),
+      ).toBe(`https://www.walmart.${tld}/search`);
+    });
+  }
 });
diff --git a/src/site_data.ts b/src/site_data.ts
@@ -77,6 +77,14 @@ export const siteParams = {
     key: "facetedValue",
     value: "dq4mn",
   },
+  "Walmart.ca": {
+    key: "facet",
+    value: "retailer_type:Walmart",
+  },
+  "Walmart.com": {
+    key: "facet",
+    value: "retailer_type:Walmart",
+  },
 } as const;
 
 export const builtinSiteData = [
@@ -223,4 +231,30 @@ export const builtinSiteData = [
         siteParams["Target.com"].value,
       ),
   },
+  {
+    id: "Walmart.ca",
+    name: "Walmart.ca",
+    urlRegex: /https:\/\/www\.walmart\.ca\/search\?.*/,
+    disablingFunc: (url: string): string | null =>
+      removeUrlParam(url, siteParams["Walmart.ca"].key),
+    activatingFunc: (url: string): string | null =>
+      addUrlParam(
+        url,
+        siteParams["Walmart.ca"].key,
+        siteParams["Walmart.ca"].value,
+      ),
+  },
+  {
+    id: "Walmart.com",
+    name: "Walmart.com",
+    urlRegex: /https:\/\/www\.walmart\.com\/search\?.*/,
+    disablingFunc: (url: string): string | null =>
+      removeUrlParam(url, siteParams["Walmart.com"].key),
+    activatingFunc: (url: string): string | null =>
+      addUrlParam(
+        url,
+        siteParams["Walmart.com"].key,
+        siteParams["Walmart.com"].value,
+      ),
+  },
 ] as const satisfies SiteData;