feat: connect pixellab (#144)

Co-authored-by: CheerChen <meetcheerego@gmail.com>

Author: a1528zhang

src/providers/pixellab/actions.ts

@@ -0,0 +1,25 @@
+import type { ProviderDefinition } from "../../core/types.ts";
+
+import { pixellabActions } from "./actions.ts";
+
+const service = "pixellab";
+
+export const provider: ProviderDefinition = {
+  service,
+  displayName: "PixelLab",
+  description:
+    "Generate, edit, animate, and manage pixel-art images, UI assets, characters, and objects with PixelLab.",
+  categories: ["AI", "Design & Media"],
+  authTypes: ["api_key"],
+  auth: [
+    {
+      type: "api_key",
+      label: "API Token",
+      placeholder: "PIXELLAB_API_TOKEN",
+      description:
+        "PixelLab API token sent as a Bearer token. Create or manage tokens in the PixelLab dashboard: https://www.pixellab.ai/",
+    },
+  ],
+  homepageUrl: "https://www.pixellab.ai/",
+  actions: pixellabActions,
+};

src/providers/pixellab/definition.ts

@@ -0,0 +1,35 @@
+import { describe, expect, it, vi } from "vitest";
+import { createGuardedFetch } from "../../core/guarded-fetch.ts";
+import { credentialValidators } from "./executors.ts";
+
+describe("PixelLab credential validator egress", () => {
+  it("skips reserved Fake-IP DNS results for the fixed PixelLab API host", async () => {
+    const transport = vi.fn(async () => Response.json([])) as unknown as typeof fetch;
+    const defaultFetcher = createGuardedFetch({
+      fetch: transport,
+      lookup: async () => [{ address: "198.18.0.130", family: 4 }],
+    });
+
+    await expect(
+      credentialValidators.apiKey!({ apiKey: "test-token", values: {} }, { fetcher: defaultFetcher }),
+    ).resolves.toMatchObject({ grantedScopes: [] });
+
+    expect(transport).toHaveBeenCalledOnce();
+  });
+
+  it("still blocks redirects from PixelLab to reserved targets", async () => {
+    const transport = vi.fn(
+      async () =>
+        new Response(null, {
+          status: 302,
+          headers: { location: "http://169.254.169.254/latest/meta-data/" },
+        }),
+    ) as unknown as typeof fetch;
+
+    await expect(
+      credentialValidators.apiKey!({ apiKey: "test-token", values: {} }, { fetcher: transport }),
+    ).rejects.toThrow(/redirect location/u);
+
+    expect(transport).toHaveBeenCalledOnce();
+  });
+});

src/providers/pixellab/executors.test.ts

@@ -0,0 +1,31 @@
+import type { CredentialValidators, ProviderExecutors } from "../../core/types.ts";
+
+import { createProviderFetch, defineApiKeyProviderExecutors } from "../provider-runtime.ts";
+import { pixellabCharacterActionHandlers } from "./runtime-character.ts";
+import { pixellabImageExtraActionHandlers } from "./runtime-image-extra.ts";
+import { pixellabImageActionHandlers } from "./runtime-image.ts";
+import { pixellabObjectActionHandlers } from "./runtime-object.ts";
+import { pixellabUiActionHandlers } from "./runtime-ui.ts";
+import { pixellabActionHandlers, validatePixellabCredential } from "./runtime.ts";
+
+const service = "pixellab";
+
+export const executors: ProviderExecutors = defineApiKeyProviderExecutors(
+  service,
+  {
+    ...pixellabActionHandlers,
+    ...pixellabImageActionHandlers,
+    ...pixellabImageExtraActionHandlers,
+    ...pixellabUiActionHandlers,
+    ...pixellabCharacterActionHandlers,
+    ...pixellabObjectActionHandlers,
+  },
+  { skipDnsValidation: true },
+);
+
+export const credentialValidators: CredentialValidators = {
+  apiKey(input, { fetcher, signal }) {
+    const credentialFetch = createProviderFetch({ fetch: fetcher, skipDnsValidation: true });
+    return validatePixellabCredential(input.apiKey, credentialFetch, signal);
+  },
+};