@things-factory/figure-service 10.1.9 → 10.1.15

package/server/service/figure/figure-tools.ts

@@ -93,6 +93,48 @@ async function figureInDomain(figureId: string, domainId: string): Promise<any>
   return found
 }
 
+/**
+ * figureId 또는 hostContext로부터 대상 Figure row(또는 미저장 초안 가상 row)를 해소한다.
+ */
+async function resolveFigureTarget(figureId: string | undefined, domainId: string, host: any): Promise<any> {
+  if (
+    host?.editorMode === 'new' &&
+    host?.draftSource &&
+    (!figureId || figureId === host.draftSource.type || figureId.startsWith('FIGURE'))
+  ) {
+    return unsavedFigure(host)
+  }
+
+  if (figureId) {
+    const { getRepository, Figure } = getDeps()
+    let found = await getRepository(Figure).findOne({
+      where: { id: figureId, domain: { id: domainId } }
+    })
+    if (!found) {
+      found = await getRepository(Figure).findOne({
+        where: { type: figureId, domain: { id: domainId } }
+      })
+    }
+    if (!found) {
+      if (
+        host?.editorMode === 'new' ||
+        (host?.draftSource &&
+          (host.draftSource.type === figureId ||
+            figureId.startsWith('FIGURE') ||
+            !host.figureId ||
+            host.figureId === figureId))
+      ) {
+        const unsaved = unsavedFigure(host)
+        if (unsaved) return unsaved
+      }
+      throw new Error(`figure not found in this tenant: ${figureId}`)
+    }
+    return found
+  }
+
+  return unsavedFigure(host)
+}
+
 /**
  * 모델러가 아직 저장하지 않은 source를 도구가 읽는 유일한 길.
  *
@@ -101,7 +143,8 @@ async function figureInDomain(figureId: string, domainId: string): Promise<any>
  * 프롬프트로 옮기면 비용만 태우고, DB에 있던 Figure와 다른 것을 고친다고 잘못 말할 수 있다.
  */
 function sourceForTool(row: any, host: any): { source: any; draft: boolean } {
-  const draft = host?.draftSource
+  const isMatch = host?.figureId === row.id || (!row.id && host?.editorMode === 'new') || (!row.id && !host?.figureId)
+  const draft = isMatch ? host?.draftSource : undefined
   if (!draft) {
     return { source: row.source ? JSON.parse(row.source) : null, draft: false }
   }
@@ -120,11 +163,20 @@ function sourceForTool(row: any, host: any): { source: any; draft: boolean } {
   if (checked.errors.length > 0) {
     throw new Error(`the unsaved draft is invalid: ${checked.errors.map((one: any) => one.message).join(' · ')}`)
   }
+
+  let normalizedDraft = draft
   if (draft.type !== row.type) {
-    throw new Error(`the unsaved draft type '${draft.type}' does not match Figure '${row.type}'`)
+    // 템플릿 기본값 'FIGURE', 임시 draftType('FIGURE_...'), 또는 빈 값인 경우
+    // 현재 열려 있는 Figure의 실제 type으로 보정/정규화한다.
+    const isGenericOrTempType = !draft.type || draft.type === 'FIGURE' || draft.type.startsWith('FIGURE_')
+    if (isGenericOrTempType && row.type) {
+      normalizedDraft = { ...draft, type: row.type }
+    } else {
+      throw new Error(`the unsaved draft type '${draft.type}' does not match Figure '${row.type}'`)
+    }
   }
 
-  return { source: draft, draft: true }
+  return { source: normalizedDraft, draft: true }
 }
 
 /**
@@ -134,8 +186,14 @@ function sourceForTool(row: any, host: any): { source: any; draft: boolean } {
  * 도메인에서 Figure를 찾은 뒤, `proposeFigure`가 다시 길이·형식을 정리할 참고 데이터만 건넨다.
  */
 function feedbackForTool(host: any, figureId?: string): unknown[] | undefined {
-  if (!figureId || host?.figureId !== figureId || !Array.isArray(host?.proposalFeedback)) return undefined
-  return host.proposalFeedback.slice(-5)
+  if (!Array.isArray(host?.proposalFeedback)) return undefined
+  if (figureId && host?.figureId === figureId) {
+    return host.proposalFeedback.slice(-5)
+  }
+  if (host?.editorMode === 'new' || !host?.figureId) {
+    return host.proposalFeedback.slice(-5)
+  }
+  return undefined
 }
 
 /**
@@ -144,20 +202,36 @@ function feedbackForTool(host: any, figureId?: string): unknown[] | undefined {
  * 모델이 대상을 안 줬으면 **저작면이 지금 열고 있는 도형**이다. 사용자에게 식별자를 되묻지
  * 않는다 — 화면에 그 도형이 떠 있는데 「어느 도형입니까」를 묻는 것은 문맥을 안 읽은 것이다.
  */
-function figureIdFrom(args: any, host: any): string {
+function figureIdFrom(args: any, host: any): string | undefined {
   const said = typeof args?.figureId === 'string' ? args.figureId.trim() : ''
   if (said) return said
 
   const open = typeof host?.figureId === 'string' ? host.figureId.trim() : ''
   if (open) return open
 
-  throw new Error('figureId is required: this surface has no figure open to resolve it from')
+  return undefined
+}
+
+function missingFigureTarget() {
+  return {
+    error: 'FIGURE_TARGET_REQUIRED',
+    message: '현재 화면에는 선택된 도형이 없습니다.',
+    nextStep: 'For an existing figure, call listFigures and match the user’s requested name or type; ask which figure only if ambiguous. For a new figure, call proposeFigureCandidate with create: true and the user’s prompt. Do not invent an id or repeat this call without a target.'
+  }
+}
+
+function unsavedFigure(host: any): any | undefined {
+  if (host?.editorMode !== 'new' || !host?.draftSource) return undefined
+  const row = { id: host.figureId, type: host.draftSource.type || 'FIGURE', name: '', state: 'draft' }
+  // Reuse size, shape and vocabulary validation before trusting browser content.
+  const current = sourceForTool(row, host)
+  return { ...row, source: JSON.stringify(current.source) }
 }
 
 const FIGURE_ID_ARG = {
   figureId: {
     type: 'string',
-    description: 'The figure to read. Omit to use the one open in the editor.'
+    description: 'The figure to read. Omit when host context has an open figureId or an unsaved new editor draft. On the catalog, first resolve the user’s requested figure with listFigures.'
   }
 } as const
 
@@ -206,10 +280,11 @@ function figureToolSpecs(): ToolSpec[] {
       schema: { type: 'object', properties: { ...FIGURE_ID_ARG } },
       builder: async (args: any, ctx: any) => {
         const domainId = requireDomainId(ctx?.state)
-        const figureId = figureIdFrom(args, ctx?.state?.host)
-        const row = await figureInDomain(figureId, domainId)
+        const figureId = figureIdFrom(args, ctx?.host)
+        const row = await resolveFigureTarget(figureId, domainId, ctx?.host)
+        if (!row) return missingFigureTarget()
 
-        const current = sourceForTool(row, ctx?.state?.host)
+        const current = sourceForTool(row, ctx?.host)
         if (!current.source) {
           // 저장된 정본이 없는 상태가 실제로 있다(새로 만들고 저장 안 한 것). 지어내지 않는다.
           return { id: row.id, type: row.type, name: row.name, source: null, cost: null }
@@ -240,10 +315,11 @@ function figureToolSpecs(): ToolSpec[] {
       schema: { type: 'object', properties: { ...FIGURE_ID_ARG } },
       builder: async (args: any, ctx: any) => {
         const domainId = requireDomainId(ctx?.state)
-        const figureId = figureIdFrom(args, ctx?.state?.host)
-        const row = await figureInDomain(figureId, domainId)
+        const figureId = figureIdFrom(args, ctx?.host)
+        const row = await resolveFigureTarget(figureId, domainId, ctx?.host)
+        if (!row) return missingFigureTarget()
 
-        const current = sourceForTool(row, ctx?.state?.host)
+        const current = sourceForTool(row, ctx?.host)
         const { inspectSource } = getDeps()
         return { ...inspectSource(current.source ? JSON.stringify(current.source) : null), draft: current.draft }
       }
@@ -288,15 +364,15 @@ function figureToolSpecs(): ToolSpec[] {
           후보가 온다.
         */
         let base: unknown
-        if (!args?.create) {
-          const open = typeof args?.figureId === 'string' ? args.figureId : ctx?.state?.host?.figureId
-          if (typeof open === 'string' && open.trim()) {
-            const row = await figureInDomain(open.trim(), domainId)
-            base = sourceForTool(row, ctx?.state?.host).source ?? undefined
+        if (!args?.create && args?.figureId !== null) {
+          const open = typeof args?.figureId === 'string' ? args.figureId : ctx?.host?.figureId
+          const row = await resolveFigureTarget(open ? open.trim() : undefined, domainId, ctx?.host)
+          if (row) {
+            base = sourceForTool(row, ctx?.host).source ?? undefined
           }
         }
 
-        const palette = Array.isArray(ctx?.state?.host?.palette) ? ctx.state.host.palette : []
+        const palette = Array.isArray(ctx?.host?.palette) ? ctx.host.palette : []
         if (palette.length === 0) {
           /*
             팔레트를 모르면 모델이 색을 지어내고, 그 후보는 그리는 시점에 `resolveToken` 이 던진다 —
@@ -306,14 +382,24 @@ function figureToolSpecs(): ToolSpec[] {
         }
 
         const { proposeFigure } = getDeps()
-        const feedback = feedbackForTool(ctx?.state?.host, base ? figureIdFrom(args, ctx?.state?.host) : undefined)
+        const feedback = feedbackForTool(ctx?.host, base ? figureIdFrom(args, ctx?.host) : undefined)
+
+        let hostImage: { data: Buffer; mediaType: any } | undefined
+        const rawImage = ctx?.host?.image || (Array.isArray(ctx?.host?.attachments) ? ctx.host.attachments.find((a: any) => a?.mediaType?.startsWith('image/')) : undefined)
+        if (rawImage?.data) {
+          const rawData = typeof rawImage.data === 'string' ? rawImage.data.replace(/^data:image\/[a-zA-Z]+;base64,/, '') : rawImage.data
+          const buf = typeof rawData === 'string' ? Buffer.from(rawData, 'base64') : Buffer.from(rawData)
+          hostImage = { data: buf, mediaType: rawImage.mediaType || 'image/png' }
+        }
+
         const result = await proposeFigure({
           prompt,
           palette,
           base,
           type: typeof args?.type === 'string' ? args.type : undefined,
           feedback,
-          refine: !!feedback?.length
+          refine: !!feedback?.length,
+          image: hostImage
         })
 
         /*
@@ -330,7 +416,19 @@ function figureToolSpecs(): ToolSpec[] {
           reason:
             `등급 ${result.score.grade} · 삼각형 ${result.cost.triangles} · ` +
             `한 도면 ${result.cost.at.drawCalls}회 · ${result.attempts}번에 만들었다`,
-          args: { prompt, figureId: base ? figureIdFrom(args, ctx?.state?.host) : undefined },
+          choices: [
+            { id: 'save', action: 'save', label: '초안 저장', icon: 'save', primary: true },
+            { id: 'repropose', action: 'prompt', label: '다른 콘셉트 재제안', icon: 'refresh', prompt: '현재 제안과 완전히 다른 스타일과 콘셉트로 새로 제안해줘' },
+            { id: 'slim', action: 'prompt', label: '비율 슬림 조정', icon: 'aspect_ratio', prompt: '전체적인 가로세로 비율을 좀 더 슬림하고 날렵하게 조정해줘' },
+            { id: 'palette', action: 'prompt', label: '색상 테마 변경', icon: 'palette', prompt: '다른 팔레트 색상 테마로 부품 색상을 변경해줘' },
+            { id: 'detail', action: 'prompt', label: '디테일 보강', icon: 'extension', prompt: '주요 디테일과 센서 부품을 좀 더 풍부하게 보강해줘' }
+          ],
+          args: {
+            baseSource: base ? JSON.stringify(base) : undefined,
+            prompt, figureId: base ? figureIdFrom(args, ctx?.host) : undefined,
+            draftType: base && ctx?.host?.editorMode === 'new' && !figureIdFrom(args, ctx?.host)
+              ? (base as any).type : undefined
+          },
           source: result.source,
           score: result.score,
           cost: result.cost,
@@ -356,10 +454,14 @@ export function registerFigureToolCategories(): void {
     description: '저작 중인 Figure 의 정본·비용·발행 판정 조회와 후보 제안(저장 아님).',
     guidance: [
       '- **도형을 만들거나 고쳐 달라는 요청에는 반드시 `proposeFigureCandidate` 를 호출한다.**',
+      '  후보는 도구 호출 결과에서만 생긴다.',
       '  호출 없이 "후보를 만들었습니다" · "적용 버튼을 누르세요" 라고 쓰는 것은 **거짓이다** — 화면에 후보 칸이 생기지 않으므로 사용자는 존재하지 않는 버튼을 찾는다.',
-      '  후보 화면(「지금 / 후보」 두 칸)은 **도구 호출 결과에서만** 만들어진다. 문장으로는 아무 일도 일어나지 않는다.',
+      '  도크에서는 후보 카드의 버튼을 눌러야 모델러 비교 화면이 열린다. 아직 열지 않은 비교 화면이 이미 보인다고 말하지 않는다.',
+      '  요청한 부품만 바뀌었다고 추측하지 않는다. 후보와 원본을 비교한 근거가 없으면 변경 범위는 비교 화면에서 확인하도록 안내한다.',
       '- 정본을 직접 저장하지 않는다. 후보만 내고, 받을지 버릴지는 저작자가 화면에서 정한다.',
+      '- 좌표와 크기는 X/Y/Z 축 이름으로 정확하게 표시한다. 확인되지 않은 가로·세로·높이 의미를 임의로 붙이지 않는다.',
       '- **고치라는 요청이면 먼저 `getFigureSource` 를 부른다.** 읽지 않고 고치면 짐작을 상대로 후보를 내게 되고, 부품 이름이 갈려 이미 배치된 보드의 바인딩이 끊긴다.',
+      '- 목록 화면에는 열린 도형이 없다. 기존 도형을 조회·수정할 때는 `listFigures`에서 사용자가 말한 이름·타입의 id를 찾는다. 대상이 모호할 때만 어느 도형인지 묻는다. 새 제작 요청에는 기존 정본을 조회하지 않고 `proposeFigureCandidate`를 create: true로 호출한다.',
       '- 타입 이름을 지어내지 않는다 — `listFigures` 로 확인한다. **타입 이름은 저장되는 식별자**이고 보드가 그 이름으로 도형을 찾는다.',
       '- 발행이 안 되는 이유를 물으면 `inspectFigure` 를 부른다. 판정 없이 "발행할 수 있습니다"라고 말하지 않는다.',
       '  판정의 각 항목은 **왜 문제인지와 무엇을 바꾸면 되는지**를 함께 담고 있다. 그대로 전한다 — 요약하면 저작자가 어느 부품의 어느 축을 고칠지 다시 찾아야 한다.',