Use HttpBaseClient for Ahrefs domain rating

apps/web/.env.example

@@ -157,6 +157,7 @@ TWITTER_CLIENT_SECRET=
 TIKTOK_CLIENT_ID=
 TIKTOK_CLIENT_SECRET=
 YOUTUBE_API_KEY=
+AHREFS_API_KEY=
 
 # Scrape Creators
 SCRAPECREATORS_API_KEY=

apps/web/app/(ee)/api/cron/partner-platforms/website/route.ts

@@ -1,9 +1,9 @@
+import { ahrefsClient } from "@/lib/ahrefs/client";
 import { withCron } from "@/lib/cron/with-cron";
 import { prisma } from "@/lib/prisma";
 import { chunk, getDomainWithoutWWW } from "@dub/utils";
 import { PlatformType } from "@prisma/client";
 import { logAndRespond } from "../../utils";
-import { getDomainRating } from "./get-domain-rating";
 
 export const dynamic = "force-dynamic";
 
@@ -49,7 +49,7 @@ export const GET = withCron(async () => {
         }
 
         try {
-          const domainRating = await getDomainRating(target);
+          const domainRating = await ahrefsClient.getDomainRating(target);
 
           await prisma.partnerPlatform.update({
             where: {

apps/web/lib/ahrefs/client.ts

@@ -0,0 +1,37 @@
+import { assertEnv } from "@/lib/assert-env";
+import { HttpBaseClient } from "@/lib/http/base-client";
+import {
+  getDomainRatingInputSchema,
+  getDomainRatingOutputSchema,
+} from "./schema";
+
+class AhrefsClient extends HttpBaseClient {
+  protected readonly vendor = "Ahrefs";
+  protected readonly baseUrl = "https://api.ahrefs.com/v3/public";
+
+  constructor() {
+    super({ timeout: 5_000 });
+  }
+
+  protected buildAuthHeaders() {
+    return {
+      Authorization: `Bearer ${assertEnv("AHREFS_API_KEY")}`,
+    };
+  }
+
+  // GET /v3/public/domain-rating-free
+  async getDomainRating(target: string) {
+    const data = await this.get("/domain-rating-free", {
+      input: {
+        target,
+        output: "json",
+      },
+      inputSchema: getDomainRatingInputSchema,
+      outputSchema: getDomainRatingOutputSchema,
+    });
+
+    return Math.round(data.domain_rating.domain_rating);
+  }
+}
+
+export const ahrefsClient = new AhrefsClient();

apps/web/lib/ahrefs/schema.ts

@@ -0,0 +1,14 @@
+import * as z from "zod/v4";
+
+// GET /v3/public/domain-rating-free query params
+export const getDomainRatingInputSchema = z.object({
+  target: z.string(),
+  output: z.literal("json"),
+});
+
+// GET /v3/public/domain-rating-free response
+export const getDomainRatingOutputSchema = z.object({
+  domain_rating: z.object({
+    domain_rating: z.number(),
+  }),
+});