// genji.config.ts
import { defineConfig, LanguageModel } from "@xieyezi/genji-i18n";

export default defineConfig({
  model: LanguageModel.GPT3_5,
  entryLocale: "zh-CN",
  entrySuffix: ".zh-CN.md",
  outputLocales: ["en-US", "ja-JP"],
  entry: ["./index.zh-CN.md"],
  outputCustom: (locale, { getDefaultSuffix }) => {
    if (locale === "en-US") return ".md";
    return getDefaultSuffix(locale);
  }
});

export interface GenjiI18nConfig {
  /**
   * @description ChatGPT model to use
   */
  model?: LanguageModel;
  /**
   * @description Number of concurrently pending promises returned
   */
  concurrency?: number;
  /**
   * @description Provide some context for a more accurate translation
   */
  reference?: string;
  /**
   * @description Split locale JSON by token
   */
  splitToken?: number;
  /**
   * @description Sampling temperature to use
   */
  temperature?: number;
  /**
   * @description The entry file or folder, support glob
   */
  entry: string[];
  /**
   * @description The language that will use as translation ref
   */
  entryLocale: string;
  /**
   * @description Markdown Suffix
   */
  entrySuffix?: string;
  /**
   * @description The markdown that will ignore, support glob
   */
  exclude?: string[];
  /**
   * @description All languages that need to be translated
   */
  outputLocales: string[];
  /**
   * @description IS use json mode
   */
  experimental?: {
    jsonMode?: boolean;
  };
  /**
   * @description Markdown extension generator function
   */
  outputCustom?: (
    locale: string,
    config: {
      fileContent: string;
      filePath: string;
      getDefaultSuffix: (locale: string) => string;
    }
  ) => string;
}