trace.moe-mcp 1.0.0 → 1.0.2

package/src/api.ts

@@ -1,192 +0,0 @@
-export interface AnilistTitle {
-  native?: string | null;
-  romaji?: string | null;
-  english?: string | null;
-  chinese?: string | null;
-}
-
-export interface AnilistData {
-  id: number;
-  idMal?: number | null;
-  title: AnilistTitle;
-  synonyms?: string[];
-  synonyms_chinese?: string[];
-  isAdult?: boolean;
-  episodes?: number | null;
-}
-
-export interface SearchResultItem {
-  anilist: number | AnilistData;
-  filename: string;
-  episode?: number | string | (number | string)[] | null;
-  episode_start?: number | null;
-  episode_end?: number | null;
-  from: number;
-  to: number;
-  similarity: number;
-  video: string;
-  image: string;
-}
-
-export interface SearchResponse {
-  frameCount: number;
-  error?: string;
-  result: SearchResultItem[];
-  quota?: number;
-  quotaUsed?: number;
-}
-
-export interface UserResponse {
-  id: string;
-  priority: number;
-  concurrency: number;
-  quota: number;
-  quotaUsed: number;
-}
-
-export interface AnilistSearchResultItem {
-  id: number;
-  title: string;
-  similarity: number;
-  anilist: AnilistData;
-}
-
-export class TraceMoeClient {
-  private baseUrl: string;
-  private apiKey: string;
-
-  constructor(options?: { baseUrl?: string; apiKey?: string }) {
-    this.baseUrl = (
-      options?.baseUrl ||
-      process.env.TRACE_MOE_API_HOST ||
-      "https://api.trace.moe"
-    ).replace(/\/$/, "");
-
-    this.apiKey = options?.apiKey || process.env.TRACE_MOE_API_KEY || "";
-  }
-
-  private getHeaders(): Record<string, string> {
-    const headers: Record<string, string> = {
-      "User-Agent": "trace.moe-mcp/1.0.0",
-    };
-    if (this.apiKey) {
-      headers["x-trace-key"] = this.apiKey;
-    }
-    return headers;
-  }
-
-  /**
-   * Search anime scene by sending a 33-element MPEG-7 Color Layout Descriptor vector.
-   */
-  async searchByVector(
-    vector: number[],
-    options?: {
-      anilistInfo?: boolean;
-      anilistID?: number;
-    },
-  ): Promise<SearchResponse> {
-    if (!Array.isArray(vector) || vector.length !== 33) {
-      throw new Error(
-        `Invalid feature vector: expected 33 numbers, got ${Array.isArray(vector) ? vector.length : typeof vector}`,
-      );
-    }
-
-    const queryParams = new URLSearchParams();
-    if (options?.anilistInfo !== false) {
-      queryParams.set("anilistInfo", "2");
-    }
-    if (options?.anilistID !== undefined) {
-      queryParams.set("anilistID", String(options.anilistID));
-    }
-
-    const qs = queryParams.toString();
-    const url = `${this.baseUrl}/search${qs ? `?${qs}` : ""}`;
-
-    const headers = {
-      ...this.getHeaders(),
-      "Content-Type": "application/json",
-    };
-
-    let response: Response | undefined;
-    let retries = 3;
-
-    while (retries > 0) {
-      response = await fetch(url, {
-        method: "POST",
-        headers,
-        body: JSON.stringify({ vector }),
-      });
-
-      if (response.status !== 503 || retries === 1) {
-        break;
-      }
-
-      retries--;
-      await new Promise((resolve) => setTimeout(resolve, 1500));
-    }
-
-    if (!response) {
-      throw new Error("No response received from trace.moe API");
-    }
-
-    if (!response.ok) {
-      const errorText = await response.text().catch(() => "");
-      let parsedError = errorText;
-      try {
-        const json = JSON.parse(errorText);
-        if (json.error) parsedError = json.error;
-      } catch {}
-
-      if (response.status === 402) {
-        throw new Error(`trace.moe search quota exceeded. ${parsedError}`);
-      }
-      if (response.status === 429) {
-        throw new Error(`trace.moe rate limit exceeded. Please try again later. ${parsedError}`);
-      }
-      if (response.status === 503) {
-        throw new Error(`trace.moe server is currently busy or overloaded. ${parsedError}`);
-      }
-      throw new Error(`trace.moe API error (${response.status}): ${parsedError}`);
-    }
-
-    return (await response.json()) as SearchResponse;
-  }
-
-  /**
-   * Search anime by title/synonym using /anilist?q=...
-   */
-  async searchAnilist(query: string): Promise<AnilistSearchResultItem[]> {
-    const url = `${this.baseUrl}/anilist?q=${encodeURIComponent(query)}`;
-    const response = await fetch(url, {
-      method: "GET",
-      headers: this.getHeaders(),
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text().catch(() => "");
-      throw new Error(`trace.moe anilist search failed (${response.status}): ${errorText}`);
-    }
-
-    return (await response.json()) as AnilistSearchResultItem[];
-  }
-
-  /**
-   * Get user quota, priority, concurrency, and usage info from /me
-   */
-  async getMe(): Promise<UserResponse> {
-    const url = `${this.baseUrl}/me`;
-    const response = await fetch(url, {
-      method: "GET",
-      headers: this.getHeaders(),
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text().catch(() => "");
-      throw new Error(`trace.moe /me failed (${response.status}): ${errorText}`);
-    }
-
-    return (await response.json()) as UserResponse;
-  }
-}
-
-export const defaultClient = new TraceMoeClient();

package/src/format.ts

@@ -1,195 +0,0 @@
-import {
-  type AnilistData,
-  type AnilistSearchResultItem,
-  type SearchResponse,
-  type SearchResultItem,
-  type UserResponse,
-} from "./api.ts";
-
-/**
- * Formats seconds into MM:SS or HH:MM:SS format.
- */
-export function formatTime(seconds: number): string {
-  if (isNaN(seconds) || seconds < 0) return "00:00";
-  const sec = Math.floor(seconds);
-  const h = Math.floor(sec / 3600);
-  const m = Math.floor((sec % 3600) / 60);
-  const s = sec % 60;
-
-  const mm = String(m).padStart(2, "0");
-  const ss = String(s).padStart(2, "0");
-
-  if (h > 0) {
-    const hh = String(h).padStart(2, "0");
-    return `${hh}:${mm}:${ss}`;
-  }
-  return `${mm}:${ss}`;
-}
-
-/**
- * Formats a similarity float (0.0 - 1.0) into a percentage string (e.g. "98.5%").
- */
-export function formatSimilarity(similarity: number): string {
-  return `${(similarity * 100).toFixed(1)}%`;
-}
-
-/**
- * Extracts anime titles for all languages (native Japanese, romaji, English, Chinese).
- * Returns primary English/Romaji title and the full list of titles across all languages.
- */
-export function getAnimeTitles(anilist: number | AnilistData): {
-  primary: string;
-  titles: string[];
-  id: number;
-} {
-  if (typeof anilist === "number") {
-    return { primary: `Anilist ID ${anilist}`, titles: [`Anilist ID ${anilist}`], id: anilist };
-  }
-
-  const { chinese, english, native, romaji } = anilist.title || {};
-  const titles: string[] = [];
-
-  if (native && !titles.includes(native)) titles.push(native);
-  if (romaji && !titles.includes(romaji)) titles.push(romaji);
-  if (english && !titles.includes(english)) titles.push(english);
-  if (chinese && !titles.includes(chinese)) titles.push(chinese);
-
-  if (titles.length === 0) {
-    titles.push(`Anilist ID ${anilist.id}`);
-  }
-
-  const primary = english || romaji || native || chinese || `Anilist ID ${anilist.id}`;
-
-  return {
-    primary,
-    titles,
-    id: anilist.id,
-  };
-}
-
-/**
- * Formats episode information into a readable string (e.g. "1/12", "1-2/12", "1", "1-2").
- * Prefers episode_start and episode_end (aligned with Anilist episode count) over filename episode.
- * Returns null if episode is unknown.
- */
-export function formatEpisode(item: SearchResultItem): string | null {
-  let epText: string | null = null;
-
-  if (item.episode_start !== undefined && item.episode_start !== null) {
-    if (
-      item.episode_end !== undefined &&
-      item.episode_end !== null &&
-      item.episode_end !== item.episode_start
-    ) {
-      epText = `${item.episode_start}-${item.episode_end}`;
-    } else {
-      epText = String(item.episode_start);
-    }
-  } else if (item.episode !== undefined && item.episode !== null && item.episode !== "") {
-    if (Array.isArray(item.episode)) {
-      epText = item.episode.join(", ");
-    } else {
-      epText = String(item.episode);
-    }
-  }
-
-  if (!epText) {
-    return null;
-  }
-
-  // If total episode count from Anilist is available, append /total (e.g. "1/12")
-  if (typeof item.anilist === "object" && item.anilist !== null && item.anilist.episodes) {
-    epText = `${epText}/${item.anilist.episodes}`;
-  }
-
-  return epText;
-}
-
-/**
- * Formats a list of search results into markdown.
- */
-export function formatSearchResultsMarkdown(response: SearchResponse): string {
-  if (!response.result || response.result.length === 0) {
-    return "No matching anime scene found on trace.moe.";
-  }
-
-  const lines: string[] = [];
-  lines.push(
-    `### trace.moe Search Results (Found ${response.result.length} matches, compared ${response.frameCount.toLocaleString()} frames)`,
-  );
-  lines.push("");
-
-  response.result.slice(0, 5).forEach((item, index) => {
-    const titleInfo = getAnimeTitles(item.anilist);
-    const ep = formatEpisode(item);
-    const timeRange = `${formatTime(item.from)} - ${formatTime(item.to)}`;
-    const sim = formatSimilarity(item.similarity);
-    const warningEmoji = item.similarity < 0.8 ? " ⚠️" : "";
-
-    lines.push(`#### ${index + 1}. ${titleInfo.primary}${warningEmoji} (${sim} similarity)`);
-    for (const t of titleInfo.titles) {
-      if (t !== titleInfo.primary) {
-        lines.push(`- ${t}`);
-      }
-    }
-    if (ep) {
-      lines.push(`- **Episode**: ${ep}`);
-    }
-    lines.push(`- **Timestamp**: \`${timeRange}\``);
-    lines.push(
-      `- **Anilist**: [https://anilist.co/anime/${titleInfo.id}](https://anilist.co/anime/${titleInfo.id})`,
-    );
-    lines.push(`- **Preview Image**: ${item.image}`);
-    lines.push(`- **Preview Video**: ${item.video}`);
-    lines.push("");
-  });
-
-  return lines.join("\n");
-}
-
-/**
- * Formats anime name search results into markdown.
- */
-export function formatAnilistSearchResultsMarkdown(
-  results: AnilistSearchResultItem[],
-  query: string,
-): string {
-  if (!results || results.length === 0) {
-    return `No anime found matching query: "${query}"`;
-  }
-
-  const lines = [`### Anime Search Results for "${query}" (Found ${results.length} matches):`, ""];
-
-  results.slice(0, 10).forEach((item, idx) => {
-    const titleInfo = getAnimeTitles(item.anilist);
-    const sim = (item.similarity * 100).toFixed(0);
-    lines.push(
-      `#### ${idx + 1}. ${titleInfo.primary} (Match: ${sim}%, Anilist ID: \`${item.id}\`)`,
-    );
-    for (const t of titleInfo.titles) {
-      if (t !== titleInfo.primary) {
-        lines.push(`- ${t}`);
-      }
-    }
-    lines.push(
-      `- **Anilist**: [https://anilist.co/anime/${item.id}](https://anilist.co/anime/${item.id})`,
-    );
-    lines.push("");
-  });
-
-  return lines.join("\n");
-}
-
-/**
- * Formats user quota information into markdown.
- */
-export function formatUserQuotaMarkdown(user: UserResponse): string {
-  const remaining = Math.max(0, user.quota - user.quotaUsed);
-  return `### trace.moe Account Quota & Status
-- **ID**: \`${user.id}\`
-- **Remaining Daily Quota**: **${remaining.toLocaleString()}** / ${user.quota.toLocaleString()}
-- **Searches Used (last 24h)**: ${user.quotaUsed.toLocaleString()}
-- **Concurrency Limit**: ${user.concurrency}
-- **Search Queue Priority**: ${user.priority}
-`;
-}

package/src/image-processor.ts

@@ -1,262 +0,0 @@
-import fs from "node:fs/promises";
-import path from "node:path";
-
-import sharp from "sharp";
-import { ColorLayout } from "trace.moe-id";
-
-sharp.cache(false);
-sharp.concurrency(1);
-
-function getVideoFrameRect(
-  data: Buffer,
-  width: number,
-  height: number,
-  channels = 3,
-  colorTolerance = 10,
-) {
-  function isDark(x: number, y: number) {
-    const i = (y * width + x) * channels;
-    return (
-      data[i] <= colorTolerance && data[i + 1] <= colorTolerance && data[i + 2] <= colorTolerance
-    );
-  }
-
-  function isRowDark(y: number) {
-    let darkPixelCount = 0;
-    for (let x = 0; x < width; x++) {
-      if (isDark(x, y)) darkPixelCount++;
-    }
-    return darkPixelCount > width * 0.95;
-  }
-
-  function isColDark(x: number) {
-    let darkPixelCount = 0;
-    for (let y = 0; y < height; y++) {
-      if (isDark(x, y)) darkPixelCount++;
-    }
-    return darkPixelCount > height * 0.95;
-  }
-
-  let top: number, bottom: number, left: number, right: number;
-
-  const centerY = Math.floor(height / 2);
-  const centerX = Math.floor(width / 2);
-
-  if (!isDark(centerX, centerY)) {
-    top = centerY;
-    bottom = centerY;
-    left = centerX;
-    right = centerX;
-    while (top > 0 && !isRowDark(top - 1)) top--;
-    while (bottom < height - 1 && !isRowDark(bottom + 1)) bottom++;
-    while (left > 0 && !isColDark(left - 1)) left--;
-    while (right < width - 1 && !isColDark(right + 1)) right++;
-  } else {
-    top = 0;
-    bottom = height - 1;
-    left = 0;
-    right = width - 1;
-    while (top < height && isRowDark(top)) top++;
-    while (bottom > top && isRowDark(bottom)) bottom--;
-    while (left < width && isColDark(left)) left++;
-    while (right > left && isColDark(right)) right--;
-  }
-
-  return {
-    x: left,
-    y: top,
-    width: Math.max(1, right - left + 1),
-    height: Math.max(1, bottom - top + 1),
-  };
-}
-
-function getNearestAspectRatio(
-  width: number,
-  height: number,
-  targetAspectRatios: number[],
-  threshold = 0.05,
-) {
-  const aspectRatio = width / height;
-  let bestRatio = null;
-  let minDiff = Infinity;
-
-  for (const targetRatio of targetAspectRatios) {
-    const diff = Math.abs(aspectRatio - targetRatio);
-    if (diff < minDiff) {
-      minDiff = diff;
-      bestRatio = targetRatio;
-    }
-  }
-
-  if (minDiff <= threshold) {
-    return bestRatio;
-  }
-  return null;
-}
-
-function snapRectToNearestAspectRatio(
-  rect: { x: number; y: number; width: number; height: number },
-  maxW: number,
-  maxH: number,
-) {
-  const targetRatios = [4 / 3, 16 / 9, 21 / 9];
-  const currentRatio = rect.width / (rect.height || 1);
-  let R = targetRatios[0];
-  let minDiff = Math.abs(currentRatio - R);
-  for (let i = 1; i < targetRatios.length; i++) {
-    const diff = Math.abs(currentRatio - targetRatios[i]);
-    if (diff < minDiff) {
-      minDiff = diff;
-      R = targetRatios[i];
-    }
-  }
-
-  let w = rect.width;
-  let h = rect.height;
-  if (w / R > h) {
-    w = h * R;
-  } else {
-    h = w / R;
-  }
-
-  if (w < 10) w = 10;
-  if (h < 10) h = 10;
-
-  const cx = rect.x + rect.width / 2;
-  const cy = rect.y + rect.height / 2;
-  let x = cx - w / 2;
-  let y = cy - h / 2;
-
-  if (x < 0) x = 0;
-  if (y < 0) y = 0;
-  if (x + w > maxW) {
-    x = maxW - w;
-    if (x < 0) {
-      x = 0;
-      w = maxW;
-    }
-  }
-  if (y + h > maxH) {
-    y = maxH - h;
-    if (y < 0) {
-      y = 0;
-      h = maxH;
-    }
-  }
-
-  return {
-    x: Math.round(x),
-    y: Math.round(y),
-    width: Math.round(w),
-    height: Math.round(h),
-  };
-}
-
-/**
- * Resizes and optionally crops black borders from an image buffer,
- * returning raw RGB pixel data and dimensions.
- */
-export async function resizeAndCropImage(
-  imageBuffer: Buffer,
-  cutBorders: boolean = true,
-): Promise<{ data: Buffer; width: number; height: number }> {
-  const resized = await sharp(imageBuffer)
-    .resize({ width: 320, height: 320, fit: "inside" })
-    .toBuffer();
-
-  let cropped = sharp(resized);
-
-  if (cutBorders) {
-    try {
-      const { data, info } = await sharp(resized)
-        .removeAlpha()
-        .raw()
-        .toBuffer({ resolveWithObject: true });
-
-      const targetRatios = [4 / 3, 16 / 9, 21 / 9];
-      const matchedRatio = getNearestAspectRatio(info.width, info.height, targetRatios);
-      if (matchedRatio === null) {
-        const detected = getVideoFrameRect(data, info.width, info.height, 3, 10);
-        const snapped = snapRectToNearestAspectRatio(detected, info.width, info.height);
-        cropped = sharp(resized).extract({
-          left: snapped.x,
-          top: snapped.y,
-          width: snapped.width,
-          height: snapped.height,
-        });
-      }
-    } catch {
-      // If border trimming fails, fallback to uncropped resized image
-      cropped = sharp(resized);
-    }
-  }
-
-  const { data, info } = await cropped
-    .flatten({ background: "#000000" })
-    .removeAlpha()
-    .raw()
-    .toBuffer({ resolveWithObject: true });
-
-  return {
-    data,
-    width: info.width,
-    height: info.height,
-  };
-}
-
-/**
- * Extracts the 33-element MPEG-7 Color Layout Vector from an image buffer.
- */
-export async function extractVectorFromBuffer(
-  imageBuffer: Buffer,
-  cutBorders: boolean = true,
-): Promise<number[]> {
-  const { data, width, height } = await resizeAndCropImage(imageBuffer, cutBorders);
-  return ColorLayout.extract({ data, width, height, channels: 3 });
-}
-
-/**
- * Resolves an image input (URL, local file path, or Base64 string) to a Buffer.
- */
-export async function resolveImageInputToBuffer(input: {
-  url?: string;
-  filePath?: string;
-  imageBase64?: string;
-}): Promise<Buffer> {
-  if (input.imageBase64) {
-    // Strip data URL prefix if present (e.g. data:image/jpeg;base64,...)
-    const cleanBase64 = input.imageBase64.replace(/^data:image\/[a-z]+;base64,/, "");
-    return Buffer.from(cleanBase64, "base64");
-  }
-
-  if (input.filePath) {
-    const resolvedPath = path.resolve(input.filePath);
-    return await fs.readFile(resolvedPath);
-  }
-
-  if (input.url) {
-    const response = await fetch(input.url, {
-      headers: {
-        "User-Agent": "trace.moe-mcp/1.0.0",
-      },
-    });
-    if (!response.ok) {
-      throw new Error(`Failed to fetch image from URL: ${response.status} ${response.statusText}`);
-    }
-    const arrayBuffer = await response.arrayBuffer();
-    return Buffer.from(arrayBuffer);
-  }
-
-  throw new Error("No image source provided. Must provide url, filePath, or imageBase64.");
-}
-
-/**
- * High-level helper to process an image input directly into a 33-element vector.
- */
-export async function processImageToVector(
-  input: { url?: string; filePath?: string; imageBase64?: string },
-  cutBorders: boolean = true,
-): Promise<number[]> {
-  const buffer = await resolveImageInputToBuffer(input);
-  return extractVectorFromBuffer(buffer, cutBorders);
-}