autocrew 0.1.0

package/src/modules/cover/prompt-builder.ts

@@ -0,0 +1,212 @@
+/**
+ * Cover Prompt Builder — generates 3 differentiated cover prompt sets
+ * from a content topic.
+ *
+ * Each set produces a different visual style while sharing the same
+ * core subject matter and title text.
+ */
+
+// --- Types ---
+
+export type CoverStyle = "cinematic" | "minimalist" | "bold-impact";
+
+export interface CoverPromptSet {
+  label: "A" | "B" | "C";
+  style: CoverStyle;
+  /** Full English image generation prompt */
+  imagePrompt: string;
+  /** Chinese title text for the cover (2-8 chars) */
+  titleText: string;
+  /** Layout description for reference */
+  layoutHint: string;
+}
+
+export interface PromptBuilderInput {
+  /** Content title */
+  title: string;
+  /** Content body (first ~200 chars used for context) */
+  body: string;
+  /** Target platform */
+  platform?: string;
+  /** Whether personal IP reference photos are available */
+  hasReferencePhotos: boolean;
+  /** Optional custom title override (user-specified cover title) */
+  customTitle?: string;
+}
+
+// --- Title extraction ---
+
+/**
+ * Extract a short, punchy cover title (2-8 Chinese chars) from the content title.
+ * Strips filler words and picks the most impactful segment.
+ */
+export function extractCoverTitle(title: string, customTitle?: string): string {
+  if (customTitle && customTitle.length >= 2 && customTitle.length <= 8) {
+    return customTitle;
+  }
+
+  // Remove common filler patterns
+  let cleaned = title
+    .replace(/[【】《》「」『』""'']/g, "")
+    .replace(/[!！?？。，,.:：;；\s]+/g, " ")
+    .trim();
+
+  // If already short enough, use as-is
+  if (cleaned.length <= 8) return cleaned;
+
+  // Try to find a punchy segment: split by common delimiters
+  const segments = cleaned.split(/[，,：:！!？?|｜—\-\/、]/);
+  const best = segments
+    .map((s) => s.trim())
+    .filter((s) => s.length >= 2 && s.length <= 8)
+    .sort((a, b) => b.length - a.length)[0];
+
+  if (best) return best;
+
+  // Fallback: take first 6 chars
+  return cleaned.slice(0, 6);
+}
+
+// --- Subject analysis ---
+
+type SubjectType = "person" | "concept" | "event";
+
+function analyzeSubject(title: string, body: string): SubjectType {
+  const text = title + " " + body.slice(0, 200);
+  const personKeywords = /我|你|他|她|创始人|CEO|老板|博主|达人|明星|网红|自己|个人|人物/;
+  const eventKeywords = /事件|新闻|发布|上线|爆|热|突发|刚刚|今天|昨天|最新/;
+
+  if (personKeywords.test(text)) return "person";
+  if (eventKeywords.test(text)) return "event";
+  return "concept";
+}
+
+// --- Mood analysis ---
+
+function analyzeMood(title: string, body: string): string {
+  const text = title + " " + body.slice(0, 200);
+
+  if (/危|险|警|崩|暴|怒|恐|慌|焦虑|失败/.test(text)) return "tense, dramatic";
+  if (/赚|钱|财|富|增长|暴涨|翻倍/.test(text)) return "ambitious, golden";
+  if (/美|好|幸福|温暖|治愈|舒适|放松/.test(text)) return "warm, soft";
+  if (/科技|AI|未来|数字|智能|技术/.test(text)) return "futuristic, cool-toned";
+  if (/干货|方法|技巧|攻略|教程|秘诀/.test(text)) return "clean, professional";
+  return "cinematic, atmospheric";
+}
+
+// --- Style templates ---
+
+const STYLE_CONFIGS: Record<CoverStyle, {
+  styleDesc: string;
+  lightingDesc: string;
+  colorDesc: string;
+  layoutForPerson: string;
+  layoutForConcept: string;
+  layoutForEvent: string;
+}> = {
+  cinematic: {
+    styleDesc: "cinematic movie poster style, photorealistic, film grain texture",
+    lightingDesc: "dramatic chiaroscuro lighting with strong shadows and highlights, Rembrandt lighting",
+    colorDesc: "deep contrast, desaturated with selective color accent",
+    layoutForPerson: "person positioned in the lower 2/3 of the frame, looking slightly off-camera, title text in bold sans-serif at the top 1/3 with dark gradient overlay",
+    layoutForConcept: "strong visual metaphor centered in frame, title text overlaid at upper 1/3 with semi-transparent dark band",
+    layoutForEvent: "most dramatic moment frozen in time, title text at top with heavy dark vignette ensuring readability",
+  },
+  minimalist: {
+    styleDesc: "minimalist editorial style, clean composition, large negative space",
+    lightingDesc: "soft diffused studio lighting, even and clean",
+    colorDesc: "muted palette with one bold accent color, high-key or low-key depending on mood",
+    layoutForPerson: "person small in frame with vast negative space, title text large and dominant occupying 40% of frame",
+    layoutForConcept: "single iconic object or symbol centered, surrounded by clean space, title text as the primary visual element",
+    layoutForEvent: "abstract representation of the event, geometric shapes, title text centered and oversized",
+  },
+  "bold-impact": {
+    styleDesc: "bold high-impact visual, saturated colors, dynamic composition",
+    lightingDesc: "high-contrast dramatic lighting with neon or colored light accents",
+    colorDesc: "vibrant saturated colors, complementary color scheme, eye-catching",
+    layoutForPerson: "close-up or medium shot with intense expression, title text integrated into the composition with bold color block behind it",
+    layoutForConcept: "explosive or dynamic visual with motion blur or energy effects, title text large and bold with drop shadow",
+    layoutForEvent: "action-packed composition with diagonal lines and movement, title text at an angle or with perspective effect",
+  },
+};
+
+// --- Core prompt builder ---
+
+function buildImagePrompt(
+  titleText: string,
+  subject: SubjectType,
+  mood: string,
+  style: CoverStyle,
+  hasReferencePhotos: boolean,
+): string {
+  const config = STYLE_CONFIGS[style];
+  const layout = subject === "person"
+    ? config.layoutForPerson
+    : subject === "event"
+      ? config.layoutForEvent
+      : config.layoutForConcept;
+
+  const parts: string[] = [
+    // Core visual
+    `Vertical 3:4 portrait orientation cover image.`,
+    config.styleDesc + ".",
+    `Mood: ${mood}.`,
+    config.lightingDesc + ".",
+    config.colorDesc + ".",
+
+    // Layout
+    `Composition: ${layout}.`,
+
+    // Title text on image
+    `The image MUST include the Chinese text "${titleText}" as a prominent visual element.`,
+    `Text style: bold sans-serif font, large size, high contrast against background, clearly readable.`,
+    `Text must be sharp, correctly spelled, and not distorted.`,
+
+    // Reference photo handling
+    ...(hasReferencePhotos && subject === "person"
+      ? ["Feature the person from the reference photo as the main subject, maintaining their likeness."]
+      : []),
+
+    // Prohibitions
+    "No watermarks, no logos, no URLs.",
+    "No white or light solid color backgrounds.",
+    "No cartoon or illustration style — photorealistic only.",
+    "No blurry, warped, or misspelled text.",
+  ];
+
+  return parts.join(" ");
+}
+
+// --- Public API ---
+
+/**
+ * Generate 3 cover prompt sets (A/B/C) from content metadata.
+ */
+export function buildCoverPrompts(input: PromptBuilderInput): CoverPromptSet[] {
+  const titleText = extractCoverTitle(input.title, input.customTitle);
+  const subject = analyzeSubject(input.title, input.body);
+  const mood = analyzeMood(input.title, input.body);
+
+  const styles: Array<{ label: "A" | "B" | "C"; style: CoverStyle }> = [
+    { label: "A", style: "cinematic" },
+    { label: "B", style: "minimalist" },
+    { label: "C", style: "bold-impact" },
+  ];
+
+  return styles.map(({ label, style }) => {
+    const config = STYLE_CONFIGS[style];
+    const layout = subject === "person"
+      ? config.layoutForPerson
+      : subject === "event"
+        ? config.layoutForEvent
+        : config.layoutForConcept;
+
+    return {
+      label,
+      style,
+      imagePrompt: buildImagePrompt(titleText, subject, mood, style, input.hasReferencePhotos),
+      titleText,
+      layoutHint: layout,
+    };
+  });
+}

package/src/modules/cover/ratio-adapter.test.ts

@@ -0,0 +1,122 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+
+// Mock the Gemini adapter and Pro gate before importing
+vi.mock("../../adapters/image/gemini.js", () => ({
+  generateImage: vi.fn(),
+}));
+
+vi.mock("../pro/gate.js", () => ({
+  requirePro: vi.fn(),
+  proGateResponse: vi.fn((name: string, alt: string) => ({
+    ok: false,
+    error: `「${name}」是 Pro 版功能。`,
+    upgradeHint: "运行 autocrew upgrade 了解 Pro 版详情。",
+    freeAlternative: alt,
+  })),
+}));
+
+import { generateMultiRatio, type RatioAdaptInput } from "../cover/ratio-adapter.js";
+import { generateImage } from "../../adapters/image/gemini.js";
+import { requirePro } from "../pro/gate.js";
+
+const mockGenerateImage = vi.mocked(generateImage);
+const mockRequirePro = vi.mocked(requirePro);
+
+const baseInput: RatioAdaptInput = {
+  originalPrompt: "Vertical 3:4 portrait orientation cover image. cinematic style.",
+  apiKey: "test-key",
+  model: "auto",
+  outputDir: "/tmp/test-covers",
+  baseName: "cover-a",
+  dataDir: "/tmp/test-autocrew",
+};
+
+beforeEach(() => {
+  vi.clearAllMocks();
+});
+
+describe("generateMultiRatio", () => {
+  it("returns Pro gate response when user is Free", async () => {
+    mockRequirePro.mockResolvedValue({
+      ok: false,
+      error: "Pro required",
+      upgradeHint: "upgrade",
+    });
+
+    const result = await generateMultiRatio(baseInput);
+    expect("upgradeHint" in result).toBe(true);
+    expect(result.ok).toBe(false);
+  });
+
+  it("generates 16:9 and 4:3 when Pro", async () => {
+    mockRequirePro.mockResolvedValue(null); // Pro user
+    mockGenerateImage.mockResolvedValue({
+      ok: true,
+      imagePath: "/tmp/test-covers/cover-a-16x9.png",
+      model: "gemini-native",
+    });
+
+    const result = await generateMultiRatio(baseInput);
+
+    // Should have called generateImage twice (16:9 and 4:3)
+    expect(mockGenerateImage).toHaveBeenCalledTimes(2);
+
+    // Check that aspect ratios were passed correctly
+    const calls = mockGenerateImage.mock.calls;
+    expect(calls[0][0].aspectRatio).toBe("16:9");
+    expect(calls[1][0].aspectRatio).toBe("4:3");
+  });
+
+  it("adapts prompt for different ratios", async () => {
+    mockRequirePro.mockResolvedValue(null);
+    mockGenerateImage.mockResolvedValue({
+      ok: true,
+      imagePath: "/tmp/test.png",
+      model: "gemini-native",
+    });
+
+    await generateMultiRatio(baseInput);
+
+    const calls = mockGenerateImage.mock.calls;
+    // 16:9 prompt should NOT contain "Vertical 3:4"
+    expect(calls[0][0].prompt).not.toContain("Vertical 3:4");
+    expect(calls[0][0].prompt).toContain("16:9");
+    // 4:3 prompt should NOT contain "Vertical 3:4"
+    expect(calls[1][0].prompt).not.toContain("Vertical 3:4");
+    expect(calls[1][0].prompt).toContain("4:3");
+  });
+
+  it("reports errors for failed generations", async () => {
+    mockRequirePro.mockResolvedValue(null);
+    mockGenerateImage
+      .mockResolvedValueOnce({ ok: true, imagePath: "/tmp/ok.png", model: "gemini-native" })
+      .mockResolvedValueOnce({ ok: false, imagePath: "", model: "gemini-native", error: "API error" });
+
+    const result = await generateMultiRatio(baseInput);
+    // Should not be fully ok since one failed
+    if (!("upgradeHint" in result)) {
+      expect(result.ok).toBe(false);
+      expect(result.errors.length).toBe(1);
+      expect(result.paths["16:9"]).toBeTruthy();
+      expect(result.paths["4:3"]).toBeUndefined();
+    }
+  });
+
+  it("passes reference images through", async () => {
+    mockRequirePro.mockResolvedValue(null);
+    mockGenerateImage.mockResolvedValue({
+      ok: true,
+      imagePath: "/tmp/test.png",
+      model: "gemini-native",
+    });
+
+    await generateMultiRatio({
+      ...baseInput,
+      referenceImagePaths: ["/tmp/photo.jpg"],
+    });
+
+    for (const call of mockGenerateImage.mock.calls) {
+      expect(call[0].referenceImagePaths).toEqual(["/tmp/photo.jpg"]);
+    }
+  });
+});

package/src/modules/cover/ratio-adapter.ts

@@ -0,0 +1,104 @@
+/**
+ * Ratio Adapter — generates 16:9 and 4:3 versions from a finalized 3:4 cover.
+ *
+ * Strategy: re-generate using the original prompt with the new aspect ratio.
+ * This produces better results than cropping/stretching.
+ *
+ * This is a Pro-only feature (gated by pro/gate.ts).
+ */
+import { generateImage, type GeminiModel, type AspectRatio } from "../../adapters/image/gemini.js";
+import { requirePro, proGateResponse } from "../pro/gate.js";
+import path from "node:path";
+
+export interface RatioAdaptInput {
+  /** Original image prompt used for the 3:4 version */
+  originalPrompt: string;
+  /** Gemini API key */
+  apiKey: string;
+  /** Model to use */
+  model: GeminiModel;
+  /** Reference image paths (same as used for 3:4) */
+  referenceImagePaths?: string[];
+  /** Base directory for saving (e.g. content assets dir) */
+  outputDir: string;
+  /** Base filename without extension (e.g. "cover-A") */
+  baseName: string;
+  /** Data dir for Pro gate check */
+  dataDir?: string;
+}
+
+export interface RatioAdaptResult {
+  ok: boolean;
+  paths: {
+    "16:9"?: string;
+    "4:3"?: string;
+  };
+  errors: string[];
+}
+
+/**
+ * Adapt a prompt to a different aspect ratio.
+ * Replaces "Vertical 3:4 portrait orientation" with the target ratio description.
+ */
+function adaptPromptForRatio(prompt: string, ratio: AspectRatio): string {
+  const ratioDescriptions: Record<string, string> = {
+    "16:9": "Horizontal 16:9 widescreen landscape orientation",
+    "4:3": "Square-ish 4:3 landscape orientation",
+  };
+
+  return prompt
+    .replace(/Vertical 3:4 portrait orientation/i, ratioDescriptions[ratio] || ratio)
+    .replace(/3:4/g, ratio);
+}
+
+/**
+ * Generate 16:9 and 4:3 versions from a finalized 3:4 cover prompt.
+ *
+ * Pro-only: returns upgrade hint if user is on Free plan.
+ */
+export async function generateMultiRatio(
+  input: RatioAdaptInput,
+): Promise<RatioAdaptResult | { ok: false; error: string; upgradeHint: string; freeAlternative: string }> {
+  // Pro gate check
+  const gate = await requirePro("多比例封面生成", input.dataDir);
+  if (gate) {
+    return proGateResponse(
+      "多比例封面生成（16:9 + 4:3）",
+      "3:4 封面已生成。你可以用图片编辑工具手动裁剪为其他比例。",
+    );
+  }
+
+  const ratios: AspectRatio[] = ["16:9", "4:3"];
+  const paths: Record<string, string> = {};
+  const errors: string[] = [];
+
+  for (const ratio of ratios) {
+    const adaptedPrompt = adaptPromptForRatio(input.originalPrompt, ratio);
+    const safeName = ratio.replace(":", "x");
+    const outputPath = path.join(input.outputDir, `${input.baseName}-${safeName}`);
+
+    const result = await generateImage({
+      prompt: adaptedPrompt,
+      aspectRatio: ratio,
+      model: input.model,
+      apiKey: input.apiKey,
+      referenceImagePaths: input.referenceImagePaths,
+      outputPath,
+    });
+
+    if (result.ok) {
+      paths[ratio] = result.imagePath;
+    } else {
+      errors.push(`${ratio}: ${result.error}`);
+    }
+  }
+
+  return {
+    ok: errors.length === 0,
+    paths: {
+      "16:9": paths["16:9"],
+      "4:3": paths["4:3"],
+    },
+    errors,
+  };
+}

package/src/modules/filter/sensitive-words.test.ts

@@ -0,0 +1,72 @@
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import fs from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import { scanText, _resetCache } from "../filter/sensitive-words.js";
+
+let testDir: string;
+
+beforeEach(async () => {
+  testDir = await fs.mkdtemp(path.join(os.tmpdir(), "autocrew-sw-test-"));
+  _resetCache();
+});
+
+afterEach(async () => {
+  await fs.rm(testDir, { recursive: true, force: true });
+  _resetCache();
+});
+
+describe("scanText", () => {
+  it("returns ok:true and empty hits for clean text", async () => {
+    const result = await scanText("今天天气不错，出去走走吧。", undefined, testDir);
+    expect(result.ok).toBe(true);
+    expect(result.hits).toHaveLength(0);
+  });
+
+  it("detects political sensitive words", async () => {
+    const result = await scanText("习近平发表讲话", undefined, testDir);
+    expect(result.hits.length).toBeGreaterThan(0);
+    expect(result.hits[0].category).toBeTruthy();
+  });
+
+  it("hitCount matches hits array length", async () => {
+    const result = await scanText("这是一段普通文字", undefined, testDir);
+    expect(result.hitCount).toBe(result.hits.length);
+  });
+
+  it("returns a non-empty summary string", async () => {
+    const result = await scanText("今天天气不错", undefined, testDir);
+    expect(typeof result.summary).toBe("string");
+    expect(result.summary.length).toBeGreaterThan(0);
+  });
+
+  it("loads custom words from dataDir", async () => {
+    const customPath = path.join(testDir, "sensitive-words", "custom.txt");
+    await fs.mkdir(path.dirname(customPath), { recursive: true });
+    await fs.writeFile(customPath, "# comment\n自定义敏感词\n", "utf-8");
+
+    _resetCache();
+    const result = await scanText("这里有自定义敏感词出现", undefined, testDir);
+    const found = result.hits.some((h) => h.word === "自定义敏感词");
+    expect(found).toBe(true);
+  });
+
+  it("handles empty string without error", async () => {
+    const result = await scanText("", undefined, testDir);
+    expect(result.ok).toBe(true);
+    expect(result.hits).toHaveLength(0);
+  });
+
+  it("returns positions array for each hit", async () => {
+    const result = await scanText("习近平习近平", undefined, testDir);
+    if (result.hits.length > 0) {
+      expect(Array.isArray(result.hits[0].positions)).toBe(true);
+      expect(result.hits[0].positions.length).toBeGreaterThan(0);
+    }
+  });
+
+  it("autoFixedText is string or undefined", async () => {
+    const result = await scanText("这是一段文字", undefined, testDir);
+    expect(result.autoFixedText === undefined || typeof result.autoFixedText === "string").toBe(true);
+  });
+});