feat: firefoxAddon and firefoxAddons queries

Author: aklinker1

src/@types/ctx.d.ts

@@ -2,4 +2,7 @@ interface WxtQueueCtx {
   chrome: ReturnType<
     typeof import("../services/chrome-service").createChromeService
   >;
+  firefox: ReturnType<
+    typeof import("../services/firefox-service").createFirefoxService
+  >;
 }

src/apis/firefox-api.ts

@@ -0,0 +1,32 @@
+export function createFirefoxApiClient() {
+  return {
+    getAddon: async (
+      idOrSlugOrGuid: number | string
+    ): Promise<Gql.FirefoxAddon> => {
+      const url = new URL(
+        `https://addons.mozilla.org/api/v5/addons/addon/${idOrSlugOrGuid}`
+      );
+      const res = await fetch(url);
+      if (res.status !== 200)
+        throw Error(
+          `${url.href} failed with status: ${res.status} ${res.statusText}`
+        );
+
+      const json = await res.json();
+
+      return {
+        id: json.id,
+        iconUrl: json.icon_url,
+        lastUpdated: json.last_updated,
+        longDescription: Object.values<string>(json.description)[0],
+        name: Object.values<string>(json.name)[0],
+        rating: json.ratings.average,
+        reviewCount: json.ratings.count,
+        shortDescription: Object.values<string>(json.summary)[0],
+        storeUrl: json.url,
+        version: json.current_version.version,
+        dailyActiveUsers: json.average_daily_users,
+      };
+    },
+  };
+}

src/apis/index.ts

@@ -0,0 +1 @@
+export * from "./firefox-api";

src/crawlers/chrome.ts src/crawlers/chrome-crawler.tssrc/crawlers/chrome.ts renamed to src/crawlers/chrome-crawler.ts

@@ -1 +1 @@
-export * as chrome from "./chrome";
+export * as chrome from "./chrome-crawler";

src/crawlers/index.ts

@@ -1,4 +1,6 @@
 export const rootResolver: Gql.RootResolver = {
   chromeExtension: ({ id }, ctx) => ctx.chrome.getExtension(id),
   chromeExtensions: ({ ids }, ctx) => ctx.chrome.getExtensions(ids),
+  firefoxAddon: ({ id }, ctx) => ctx.firefox.getAddon(id),
+  firefoxAddons: ({ ids }, ctx) => ctx.firefox.getAddons(ids),
 };

src/resolvers/index.ts

@@ -4,13 +4,21 @@ schema {
 
 type Query {
   """
-  Returns information about a chrome extension based on it's ID and language (defaults to "en")
+  Returns information about a Chrome extension based on it's ID and language (defaults to "en").
   """
   chromeExtension(id: String!): ChromeExtension
   """
-  Returns information about a list of chrome extension ids
+  Returns information about a list of Chrome extension ids.
   """
   chromeExtensions(ids: [String!]!): [ChromeExtension]!
+  """
+  Returns information about a Firefox addon based on it's ID/GUID/slug.
+  """
+  firefoxAddon(id: String!): FirefoxAddon
+  """
+  Returns information about a list of Firefox addon ID/GUID/slugs.
+  """
+  firefoxAddons(ids: [String!]!): [FirefoxAddon]!
 }
 
 type ChromeExtension {
@@ -26,3 +34,17 @@ type ChromeExtension {
   rating: Float
   reviewCount: Int
 }
+
+type FirefoxAddon {
+  id: String!
+  name: String!
+  iconUrl: String!
+  storeUrl: String!
+  shortDescription: String!
+  longDescription: String!
+  dailyActiveUsers: Int!
+  version: String!
+  lastUpdated: String!
+  rating: Float
+  reviewCount: Int
+}

src/schema.gql

@@ -4,6 +4,7 @@ import { createGraphql } from "./graphql";
 import playgroundHtmlTemplate from "./public/playground.html";
 import consola from "consola";
 import { createChromeService } from "./services/chrome-service";
+import { createFirefoxService } from "./services/firefox-service";
 
 const playgroundHtml = playgroundHtmlTemplate.replace(
   "{{VERSION}}",
@@ -15,8 +16,10 @@ export function createServer(config?: ServerConfig) {
   if (port == null) port = Number(process.env.PORT ?? "3000");
 
   const chrome = createChromeService();
+  const firefox = createFirefoxService();
   const graphql = createGraphql({
     chrome,
+    firefox,
   });
 
   const httpServer = Bun.serve({

src/server.ts

@@ -0,0 +1,25 @@
+import DataLoader from "dataloader";
+import { createFirefoxApiClient } from "../apis";
+
+export function createFirefoxService() {
+  const firefox = createFirefoxApiClient();
+
+  const loader = new DataLoader<string | number, Gql.FirefoxAddon | undefined>(
+    (ids) => Promise.all(ids.map((id) => firefox.getAddon(id)))
+  );
+
+  return {
+    getAddon: (id: string | number) => loader.load(id),
+    getAddons: async (ids: Array<string | number>) => {
+      const result = await loader.loadMany(ids);
+      return result.map((item) => {
+        if (item == null) return undefined;
+        if (item instanceof Error) {
+          console.warn("Error fetching multiple addons:", item);
+          return undefined;
+        }
+        return item;
+      });
+    },
+  };
+}