aimakeall-mcp 0.3.0 → 0.4.1

package/README.md

@@ -16,7 +16,7 @@ Claude Code·Codex 같은 MCP 클라이언트에서 자연어로 AImakeAll 영
   "mcpServers": {
     "aimakeall": {
       "command": "npx",
-      "args": ["-y", "aimakeall-mcp@0.3.0"],
+      "args": ["-y", "aimakeall-mcp@0.4.1"],
       "env": { "AIMAKEALL_PAT": "aio_pat_..." }
     }
   }

package/lib/cloud-tools.mjs

@@ -14,6 +14,7 @@ import {
   fileToDataUrl,
   fileToImagePayload,
   guessMimeType,
+  saveExportBuffer,
   saveMediaBuffer,
   savePayloadHandle,
   SERVER_JSON_BODY_LIMIT_BYTES,
@@ -46,6 +47,26 @@ function decodeHeaderValue(value) {
   }
 }
 
+// 상품 이미지 URL 클라이언트 가드 — 서버가 1차 필터하지만 이 코드는 사용자 PC 에서
+// 돌므로, 악성 페이지가 심은 내부망 URL(127.0.0.1:9876 컴패니언 등) fetch 를 이중 차단.
+function isSafePublicImageUrl(rawUrl) {
+  try {
+    const parsed = new URL(String(rawUrl || ""));
+    if (parsed.protocol !== "http:" && parsed.protocol !== "https:") return false;
+    const host = parsed.hostname.toLowerCase();
+    if (!host || host === "localhost" || host.endsWith(".localhost") || host.endsWith(".local") || host.endsWith(".internal")) return false;
+    if (host.includes(":") || host.startsWith("[")) return false; // IPv6 리터럴은 스킵 (이미지 CDN 에 불필요)
+    if (/^\d{1,3}(\.\d{1,3}){3}$/.test(host)) {
+      const [a, b] = host.split(".").map(Number);
+      if (a === 0 || a === 10 || a === 127 || (a === 100 && b >= 64 && b <= 127) || (a === 169 && b === 254)
+        || (a === 172 && b >= 16 && b <= 31) || (a === 192 && b === 168) || a >= 224) return false;
+    }
+    return true;
+  } catch {
+    return false;
+  }
+}
+
 // 실측 mp3 길이(초) — 파싱 실패 시에만 CBR-128 근사로 폴백.
 function audioDurationSecFromFile(filePath) {
   try {
@@ -165,6 +186,83 @@ export function registerCloudTools(server, config, api) {
     }),
   );
 
+  server.tool(
+    "research_product",
+    "상품 링크로 리뷰 기반 시장조사를 합니다: 구매자 불만 TOP3(근거 리뷰 인용)·첫 3초 후킹 패턴·경쟁 광고 공백·핵심 USP·금지 표현. 결과의 planBrief는 plan_commerce_video의 researchBrief로, productImagePaths는 productImagePaths로 그대로 이어가세요. 사이트가 서버 접근을 차단하면(fetchBlocked 안내) 에이전트가 직접 상품 페이지·리뷰를 읽어 pageText/reviewsText로 다시 호출하세요.",
+    {
+      productUrl: z.string().optional().describe("상품 페이지 URL (아마존·쇼핑몰 등)"),
+      productName: z.string().optional().describe("상품명 (URL 없이 텍스트만 넘길 때)"),
+      pageText: z.string().optional().describe("에이전트가 직접 읽은 상품 페이지 텍스트 (봇 차단 폴백)"),
+      reviewsText: z.string().optional().describe("에이전트가 수집한 구매 후기 텍스트 (봇 차단 폴백)"),
+      downloadImages: z.boolean().optional().describe("기본 true — 상품 이미지를 로컬 저장해 plan_commerce_video용 경로 반환"),
+    },
+    wrapCloudHandler(config, async ({ productUrl = "", productName = "", pageText = "", reviewsText = "", downloadImages = true }) => {
+      if (!productUrl.trim() && !pageText.trim() && !reviewsText.trim()) {
+        return textResult("productUrl 또는 pageText/reviewsText 중 하나는 필요합니다.", { isError: true });
+      }
+      const payload = await api.request("/api/tracker/commerce/product-research", {
+        body: { pageText, productName, reviewsText, url: productUrl },
+        method: "POST",
+        timeoutMs: 120_000,
+      });
+      if (!payload?.ok) {
+        return textResult(
+          `리서치 자료를 얻지 못했습니다: ${payload?.error || "자료 없음"}\n${payload?.guidance || ""}`,
+          { isError: true },
+        );
+      }
+      const productImagePaths = [];
+      if (downloadImages) {
+        const EXT_BY_SUBTYPE = { jpeg: "jpg", jpg: "jpg", png: "png", webp: "webp" };
+        for (const imageUrl of (Array.isArray(payload?.product?.imageUrls) ? payload.product.imageUrls : []).slice(0, 5)) {
+          if (!isSafePublicImageUrl(imageUrl)) continue;
+          try {
+            // redirect 미추적 — CDN 이 내부 주소로 리다이렉트시키는 우회 차단.
+            const res = await fetch(imageUrl, { redirect: "manual", signal: AbortSignal.timeout(30_000) });
+            if (!res.ok) continue;
+            const contentType = String(res.headers.get("content-type") || "").toLowerCase();
+            const subtype = contentType.startsWith("image/") ? contentType.slice(6).split(";")[0].trim() : "";
+            const ext = EXT_BY_SUBTYPE[subtype];
+            if (!ext) continue; // 이미지가 아니거나(HTML/SVG 등) 미지원 포맷
+            // 스트리밍 캡 — 전체 버퍼링 전에 8MB 에서 중단.
+            const parts = [];
+            let received = 0;
+            let tooLarge = false;
+            for await (const chunk of res.body) {
+              const piece = Buffer.from(chunk);
+              received += piece.length;
+              if (received > 8 * 1024 * 1024) {
+                tooLarge = true;
+                break;
+              }
+              parts.push(piece);
+            }
+            if (tooLarge || !received) continue;
+            const saved = saveMediaBuffer(config.stateDir, `product-${productImagePaths.length + 1}.${ext}`, Buffer.concat(parts));
+            productImagePaths.push(saved.filePath);
+          } catch {
+            // 이미지 한 장 실패는 무시
+          }
+        }
+      }
+      return jsonResult({
+        fetchBlocked: payload?.fetchBlocked || false,
+        next: productImagePaths.length
+          ? "plan_commerce_video에 productImagePaths와 researchBrief(planBrief)를 그대로 넘기세요."
+          : "상품 이미지를 확보하지 못했습니다 — 에이전트가 상품 사진을 로컬에 저장해 그 경로로 plan_commerce_video를 호출하세요 (researchBrief는 planBrief 사용).",
+        planBrief: payload?.planBrief,
+        product: {
+          imageCount: (Array.isArray(payload?.product?.imageUrls) ? payload.product.imageUrls : []).length,
+          name: payload?.product?.name,
+          price: payload?.product?.price,
+          reviewCount: payload?.product?.reviewCount,
+        },
+        productImagePaths,
+        research: payload?.research,
+      });
+    }),
+  );
+
   server.tool(
     "plan_commerce_video",
     "제품 홍보 영상 기획을 생성합니다. 제품 사진 파일 경로가 최소 1장 필요합니다 (이 PC의 로컬 경로).",
@@ -177,8 +275,9 @@ export function registerCloudTools(server, config, api) {
       tone: z.string().optional().describe("기본 '자동 추천'"),
       categoryId: z.string().optional().describe("기본 ecommerce"),
       sceneCount: z.number().int().min(1).max(12).optional().describe("기본 6"),
+      researchBrief: z.string().optional().describe("research_product 결과의 planBrief — 불만·후킹·경쟁공백이 기획에 반영됨"),
     },
-    wrapCloudHandler(config, async ({ productName, productImagePaths, modelImagePath = "", description = "", targetCustomer = "", tone = "자동 추천", categoryId = "ecommerce", sceneCount = 6 }) => {
+    wrapCloudHandler(config, async ({ productName, productImagePaths, modelImagePath = "", description = "", targetCustomer = "", tone = "자동 추천", categoryId = "ecommerce", sceneCount = 6, researchBrief = "" }) => {
       // 확장자 검증(임의 파일 업로드 차단) + 합산 크기 예산(서버 413 사전 차단).
       for (const filePath of productImagePaths) assertAllowedInputFile(filePath, ALLOWED_IMAGE_EXTS, { kind: "이미지" });
       if (modelImagePath) assertAllowedInputFile(modelImagePath, ALLOWED_IMAGE_EXTS, { kind: "모델 이미지" });
@@ -192,6 +291,7 @@ export function registerCloudTools(server, config, api) {
           operationId: createUsageEventId("commerce-plan"),
           productImages: productImagePaths.map((filePath) => fileToImagePayload(filePath)),
           productName,
+          researchBrief,
           sceneCount,
           selectedCharacters: [],
           skillOverride: null,
@@ -214,7 +314,7 @@ export function registerCloudTools(server, config, api) {
 
   server.tool(
     "generate_scene_image",
-    "씬 이미지를 생성합니다 (기획 결과의 imagePrompt 사용). 반환된 imageUrl을 씬 영상 생성의 입력으로 쓰세요. 캐릭터/앞씬 일관성이 필요하면 앞 씬의 imageUrl을 referenceImageUrls로 전달하세요.",
+    "씬 이미지를 생성합니다 (기획 결과의 imagePrompt 사용). 반환된 imageUrl을 씬 영상 생성의 입력으로 쓰세요. 인물·제품 일관성은 이 참조 방식이 권장 경로입니다: 첫 씬(또는 캐릭터 시트/제품 사진)의 이미지를 referenceImageUrls·referenceImagePaths로 모든 씬에 앵커로 전달하세요.",
     {
       prompt: z.string().describe("이미지 프롬프트 (plan 결과의 imagePrompt)"),
       aspectRatio: z.string().optional().describe("기본 9:16"),
@@ -1001,196 +1101,6 @@ export function registerCloudTools(server, config, api) {
     }),
   );
 
-  // ── 인플루언서(소울) 스튜디오 ────────────────────────────────────────────────
-  server.tool(
-    "create_influencer_character",
-    "참조 사진으로 인플루언서(소울) 캐릭터를 학습 등록합니다. 학습 비용 $2.50(1회성) — 실패 시 자동 재시도 금지(중복 과금). 등록 후 influencer_character_status로 completed까지 10초 간격 폴링하세요 (수 분 소요).",
-    {
-      name: z.string().min(1).describe("캐릭터 이름"),
-      photoPaths: z.array(z.string()).min(5).max(18).describe("동일 인물 사진 로컬 경로 5~18장 (jpg/png/webp)"),
-    },
-    wrapCloudHandler(config, async ({ name, photoPaths }) => {
-      for (const filePath of photoPaths) assertAllowedInputFile(filePath, ALLOWED_IMAGE_EXTS, { kind: "참조 사진" });
-      const publicUrls = [];
-      // 업로드 URL 발급은 사진 수만큼 쓰기 슬롯(20/분)을 쓰므로, 분당 한도에 걸리면
-      // retryAfterSeconds 만큼 대기 후 이어서 진행한다(발급된 URL 은 보존).
-      const issueUploadUrl = async (mime) => {
-        for (let attempt = 0; ; attempt += 1) {
-          try {
-            return await api.request("/api/tracker/higgsfield/files/generate-upload-url", {
-              body: { content_type: mime },
-              method: "POST",
-              timeoutMs: 30_000,
-            });
-          } catch (error) {
-            if (error instanceof AimakeallApiError && error.errorCode === "MACHINE_RATE_LIMITED" && attempt < 3) {
-              await new Promise((resolve) => setTimeout(resolve, Math.max(1, error.retryAfterSeconds || 10) * 1000));
-              continue;
-            }
-            throw error;
-          }
-        }
-      };
-      // presigned PUT(직접) — 컴패니언 불필요.
-      for (const filePath of photoPaths) {
-        const mime = guessMimeType(filePath, "image/jpeg");
-        const issued = await issueUploadUrl(mime);
-        const uploadUrl = String(issued?.upload_url || "");
-        const publicUrl = String(issued?.public_url || "");
-        if (!uploadUrl || !publicUrl) {
-          return textResult("업로드 URL 발급에 실패했습니다. Higgsfield API 키 연결을 확인하세요.", { isError: true });
-        }
-        const put = await fetch(uploadUrl, {
-          body: readFileSync(filePath),
-          headers: { "Content-Type": mime },
-          method: "PUT",
-          signal: AbortSignal.timeout(120_000),
-        });
-        if (!put.ok) {
-          return textResult(`사진 업로드 실패 (HTTP ${put.status}): ${path.basename(filePath)}`, { isError: true });
-        }
-        publicUrls.push(publicUrl);
-      }
-      let payload;
-      try {
-        // 서버의 업스트림 창(90s)보다 길게 잡아, 클라이언트가 먼저 끊어 "성공했는데
-        // 실패로 오인 → 재시도 → 중복 과금($2.50)" 되는 상황을 방지한다.
-        payload = await api.request("/api/tracker/higgsfield/v1/custom-references", {
-          body: { input_images: publicUrls.map((url) => ({ image_url: url, type: "image_url" })), name },
-          method: "POST",
-          timeoutMs: 120_000,
-        });
-      } catch (error) {
-        const reason = String(error?.message || error);
-        return textResult(
-          `캐릭터 학습 등록 요청이 실패했습니다: ${reason}\n재시도 전에 반드시 list_influencer_characters로 캐릭터가 이미 생성됐는지 확인하세요 — 생성돼 있다면 재시도는 중복 과금($2.50)입니다.`,
-          { isError: true },
-        );
-      }
-      const referenceId = String(payload?.id || payload?.reference_id || "");
-      if (!referenceId) return textResult("캐릭터 학습 등록 응답에 ID가 없습니다. list_influencer_characters로 생성 여부를 확인하세요.", { isError: true });
-      return jsonResult({
-        costUsd: payload?.costUsd,
-        next: "influencer_character_status를 10초 간격으로 폴링해 completed를 기다리세요. 이 호출을 재시도하면 중복 과금됩니다.",
-        photoCount: publicUrls.length,
-        referenceId,
-        status: payload?.status,
-      });
-    }),
-  );
-
-  server.tool(
-    "influencer_character_status",
-    "소울 캐릭터 학습 상태를 조회합니다 (terminal: completed | failed).",
-    { referenceId: z.string().min(1) },
-    wrapCloudHandler(config, async ({ referenceId }) => {
-      const payload = await api.request(`/api/tracker/higgsfield/v1/custom-references/${encodeURIComponent(referenceId)}`, {
-        method: "GET",
-        timeoutMs: 30_000,
-      });
-      return jsonResult({
-        name: payload?.name,
-        referenceId,
-        status: payload?.status,
-        thumbnailUrl: payload?.thumbnail_url,
-      });
-    }),
-  );
-
-  server.tool(
-    "list_influencer_characters",
-    "등록된 소울 캐릭터 목록을 조회합니다. (machine 토큰은 서버 경유로 만든 캐릭터만 상세 조회·생성에 사용 가능)",
-    { page: z.number().int().min(1).optional().describe("기본 1") },
-    wrapCloudHandler(config, async ({ page = 1 }) => {
-      const payload = await api.request(`/api/tracker/higgsfield/v1/custom-references/list?page=${page}&page_size=50`, {
-        method: "GET",
-        timeoutMs: 30_000,
-      });
-      return jsonResult({
-        characters: (Array.isArray(payload?.items) ? payload.items : []).map((item) => ({
-          id: item?.id,
-          name: item?.name,
-          status: item?.status,
-        })),
-        page,
-        totalPages: payload?.total_pages,
-      });
-    }),
-  );
-
-  server.tool(
-    "suggest_influencer_prompts",
-    "소울 이미지용 연출·프롬프트 10종을 추천받습니다. 반환 prompt는 부정규칙까지 완성형이므로 generate_influencer_image에 그대로 쓰세요.",
-    {
-      gender: z.enum(["여자", "남자"]),
-      ageRange: z.enum(["10대", "20대", "30대", "40대", "50대", "60대"]),
-      keyword: z.string().min(1).describe("장면 키워드 (예: 카페에서 책 읽는 모습)"),
-    },
-    wrapCloudHandler(config, async ({ gender, ageRange, keyword }) => {
-      const { events } = await api.requestSse("/api/tracker/gemini/soul-prompts", {
-        body: { ageRange, gender, imageModel: "soul", keyword },
-        timeoutMs: 180_000,
-      });
-      const items = events
-        .filter((event) => event?.type === "item" && event.item)
-        .map((event) => ({ directing: event.item.directing, id: event.item.id, prompt: event.item.prompt }));
-      if (!items.length) return textResult("프롬프트 추천 결과가 비어 있습니다.", { isError: true });
-      return jsonResult({ count: items.length, items });
-    }),
-  );
-
-  server.tool(
-    "generate_influencer_image",
-    "학습 완료(completed)된 소울 캐릭터로 인플루언서 이미지를 생성합니다 (720p $0.094/장, 1080p $0.19/장). 반환 requestId를 influencer_image_status로 폴링하세요 (~30초).",
-    {
-      prompt: z.string().min(1).describe("영문 프롬프트 (suggest_influencer_prompts 결과 그대로 권장)"),
-      referenceId: z.string().min(1).describe("completed 상태의 캐릭터 referenceId"),
-      aspectRatio: z.enum(["9:16", "16:9", "4:3", "3:4", "1:1", "2:3", "3:2"]).optional().describe("기본 4:3"),
-      resolution: z.enum(["720p", "1080p"]).optional().describe("기본 720p"),
-      imageReferenceUrl: z.string().optional().describe("구도/씬 참조 이미지 URL (선택)"),
-      seed: z.number().int().optional().describe("재현용 시드 (선택)"),
-    },
-    wrapCloudHandler(config, async ({ prompt, referenceId, aspectRatio = "4:3", resolution = "720p", imageReferenceUrl = "", seed }) => {
-      const payload = await api.request("/api/tracker/higgsfield/higgsfield-ai/soul/character", {
-        body: {
-          aspect_ratio: aspectRatio,
-          batch_size: 1,
-          custom_reference_id: referenceId,
-          custom_reference_strength: 1,
-          enhance_prompt: false,
-          ...(imageReferenceUrl ? { image_reference_url: imageReferenceUrl } : {}),
-          prompt,
-          resolution,
-          ...(seed !== undefined ? { seed } : {}),
-        },
-        method: "POST",
-        timeoutMs: 60_000,
-      });
-      const requestId = String(payload?.request_id || payload?.id || "");
-      if (!requestId) return textResult("이미지 생성 요청 응답에 request_id가 없습니다.", { isError: true });
-      return jsonResult({ costUsd: payload?.costUsd, requestId, status: payload?.status });
-    }),
-  );
-
-  server.tool(
-    "influencer_image_status",
-    "소울 이미지 생성 상태를 조회합니다 (terminal: completed | failed | nsfw | canceled). 결과 이미지는 만료 없는 공개 URL입니다.",
-    { requestId: z.string().min(1) },
-    wrapCloudHandler(config, async ({ requestId }) => {
-      const payload = await api.request(`/api/tracker/higgsfield/requests/${encodeURIComponent(requestId)}/status`, {
-        method: "GET",
-        timeoutMs: 30_000,
-      });
-      return jsonResult({
-        error: payload?.error || payload?.message || undefined,
-        imageUrls: (Array.isArray(payload?.images) ? payload.images : []).map((image) => image?.url).filter(Boolean),
-        progress: payload?.progress,
-        requestId,
-        status: payload?.status,
-      });
-    }),
-  );
-
   // ── 상세페이지(PDP) ─────────────────────────────────────────────────────────
   server.tool(
     "generate_pdp",
@@ -1252,30 +1162,33 @@ export function registerCloudTools(server, config, api) {
         for (const image of Array.isArray(result.lifestyleImages) ? result.lifestyleImages : []) {
           const url = String(image?.url || "");
           if (!url) continue;
+          // HTML 이 실제로 참조하는 컷만 받는다 — 조립에서 빠진 컷을 받아 저장하면
+          // 정리되지 않는 exports/ 에 고아 파일이 영구히 쌓인다.
+          const escaped = escapeHtml(url);
+          if (!html.includes(url) && !html.includes(escaped)) continue;
           try {
             const res = await fetch(url, { signal: AbortSignal.timeout(60_000) });
             if (!res.ok) continue;
             const buf = Buffer.from(await res.arrayBuffer());
             const ext = String(res.headers.get("content-type") || "").includes("png") ? "png" : "jpg";
-            const saved = saveMediaBuffer(config.stateDir, `pdp-image-${image?.idx ?? savedImages.length + 1}.${ext}`, buf);
+            const saved = saveExportBuffer(config.stateDir, `pdp-image-${image?.idx ?? savedImages.length + 1}.${ext}`, buf);
             // HTML 은 같은 디렉터리에 저장되므로 상대경로 치환으로 자립형 유지.
             const local = `./${path.basename(saved.filePath)}`;
-            const before = html;
-            html = html.split(url).join(local).split(escapeHtml(url)).join(local);
-            if (html !== before) savedImages.push(saved.filePath);
+            html = html.split(url).join(local).split(escaped).join(local);
+            savedImages.push(saved.filePath);
           } catch {
             // 다운로드 실패한 컷은 원격 URL 유지
           }
         }
       }
-      const savedHtml = saveMediaBuffer(config.stateDir, "pdp.html", Buffer.from(html, "utf8"), { contentType: "text/html" });
+      const savedHtml = saveExportBuffer(config.stateDir, "pdp.html", Buffer.from(html, "utf8"), { contentType: "text/html" });
       return jsonResult({
         analysis: trimAnalysis(result.analysis),
         heroHeadline: result?.copy?.heroHeadline,
         htmlPath: savedHtml.filePath,
         imageCount: (Array.isArray(result.lifestyleImages) ? result.lifestyleImages : []).filter((image) => image?.url).length,
         localImages: savedImages.length,
-        note: "htmlPath와 로컬 이미지들은 48시간 뒤 자동 정리되는 스테이징 폴더에 있습니다. 보존하려면 같은 폴더의 이미지들과 함께 원하는 위치로 복사하세요.",
+        note: "htmlPath와 로컬 이미지는 자동 정리되지 않는 exports 폴더에 저장됩니다. 옮길 때는 같은 폴더의 이미지들과 함께 이동하세요(상대경로 참조).",
         tagline: result?.copy?.tagline,
       });
     }),

package/lib/config.mjs

@@ -2,7 +2,7 @@ import { homedir } from "node:os";
 import path from "node:path";
 
 // 프록시 버전 — 서버가 X-AImakeAll-MCP-Version 으로 하한을 강제(426)할 수 있다.
-export const MCP_PROXY_VERSION = "0.3.0";
+export const MCP_PROXY_VERSION = "0.4.1";
 
 export const DEFAULT_API_BASE = "https://aimakeall.com";
 export const DEFAULT_COMPANION_URL = "http://127.0.0.1:9876";

package/lib/media-store.mjs

@@ -114,6 +114,17 @@ export function readPayloadHandle(filePath, { stateDir = "" } = {}) {
 }
 
 // 보존 기간이 지난 스테이징 미디어 정리 — 숨김 디렉터리에 GB 단위로 쌓이는 것 방지.
+// 최종 산출물(PDP HTML 등)용 보존 디렉터리 — pruneOldMedia(48h TTL)는 media/ 만
+// 정리하므로 exports/ 는 자동 삭제되지 않는다.
+export function saveExportBuffer(stateDir, name, buffer, { contentType = "" } = {}) {
+  const dir = path.join(stateDir, "exports");
+  mkdirSync(dir, { recursive: true });
+  const safeName = sanitizeOutputFilename(name, "export");
+  const filePath = path.join(dir, `${Date.now()}-${randomUUID().slice(0, 8)}-${safeName}`);
+  writeFileSync(filePath, buffer, { flag: "wx", mode: 0o600 });
+  return { bytes: buffer.length, contentType, filePath };
+}
+
 export function pruneOldMedia(stateDir, { now = Date.now() } = {}) {
   const dir = mediaDir(stateDir);
   let removed = 0;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aimakeall-mcp",
-  "version": "0.3.0",
+  "version": "0.4.1",
   "description": "AImakeAll MCP 서버 — Claude Code/Codex에서 자연어로 영상 기획·생성·렌더·퍼블리시 (렌더는 로컬 컴패니언)",
   "type": "module",
   "bin": {