Datamuse Words

/**
 * DatamuseWords - generated by OmniStream from Datamuse Words's OpenAPI spec.
 * One typed method per endpoint. A single Omni key reaches the API.
 */
const DEFAULT_BASE = "https://grid.skinvaults.online/v1/proxy/datamuse";

export class DatamuseWordsError extends Error {
  status: number;
  code?: string;
  constructor(status: number, code: string | undefined, message?: string) {
    super(message || `DatamuseWords error ${status}`);
    this.name = "DatamuseWordsError";
    this.status = status;
    this.code = code;
  }
}

export interface DatamuseWordsOptions {
  baseUrl?: string;
  fetch?: typeof fetch;
  timeoutMs?: number;
}

export class DatamuseWords {
  /** @param token Your OmniStream key (one key for every API). */
  constructor(private token: string, private opts: DatamuseWordsOptions = {}) {
    if (!token) throw new Error("DatamuseWords: token is required");
  }
  rateLimit: { remainingMinute?: number; remainingDay?: number } | null = null;

  private async _request(method: string, path: string, { params, body }: { params?: Record<string, unknown>; body?: unknown } = {}): Promise<any> {
    const f = this.opts.fetch ?? globalThis.fetch;
    const url = new URL((this.opts.baseUrl ?? DEFAULT_BASE).replace(/\/+$/, "") + path);
    if (params) for (const [k, v] of Object.entries(params)) if (v != null) url.searchParams.set(k, String(v));
    const headers: Record<string, string> = { "x-omni-key": this.token };
    if (body !== undefined) headers["content-type"] = "application/json";
    const res = await f(url, { method, headers, body: body !== undefined ? JSON.stringify(body) : undefined });
    this.rateLimit = {
      remainingMinute: Number(res.headers.get("x-ratelimit-remaining-minute")) || undefined,
      remainingDay: Number(res.headers.get("x-ratelimit-remaining-day")) || undefined,
    };
    const data = await res.json().catch(() => ({}));
    if (!res.ok) throw new DatamuseWordsError(res.status, data?.error?.code, data?.error?.message);
    return data.data ?? data;
  }

  /** Find words by constraint (meaning, sound, rhyme, etc.) */
  findWords(params: { "ml"?: string; "sl"?: string; "rel_rhy"?: string; "max"?: number }): Promise<any> {
    return this._request("GET", "/words", { params });
  }

  /** Autocomplete suggestions for a prefix */
  suggest(params: { "s": string }): Promise<any> {
    return this._request("GET", "/sug", { params });
  }
}

export default DatamuseWords;