telegram-bot-workers/src/n8n-service.ts

import { Env, IntentAnalysis, N8nResponse } from './types';

// n8n으로 처리할 기능 목록
const N8N_CAPABILITIES = [
  'weather',      // 날씨
  'search',       // 검색
  'image',        // 이미지 생성
  'translate',    // 번역
  'schedule',     // 일정
  'reminder',     // 알림
  'news',         // 뉴스
  'calculate',    // 계산
  'summarize_url', // URL 요약
];

// AI가 의도를 분석하여 n8n 호출 여부 결정
export async function analyzeIntent(
  ai: Ai,
  userMessage: string
): Promise<IntentAnalysis> {
  const prompt = `사용자 메시지를 분석하여 어떤 처리가 필요한지 JSON으로 응답하세요.

## 외부 기능이 필요한 경우 (action: "n8n")
- 날씨 정보: type = "weather"
- 웹 검색: type = "search"
- 이미지 생성: type = "image"
- 번역: type = "translate"
- 일정/캘린더: type = "schedule"
- 알림 설정: type = "reminder"
- 뉴스: type = "news"
- 복잡한 계산: type = "calculate"
- URL/링크 요약: type = "summarize_url"

## 일반 대화인 경우 (action: "chat")
- 인사, 잡담, 질문, 조언 요청 등

## 응답 형식 (JSON만 출력)
{"action": "n8n", "type": "weather", "confidence": 0.9}
또는
{"action": "chat", "confidence": 0.95}

## 사용자 메시지
${userMessage}

JSON:`;

  try {
    const response = await ai.run('@cf/meta/llama-3.1-8b-instruct', {
      messages: [{ role: 'user', content: prompt }],
      max_tokens: 100,
    });

    const text = response.response || '';

    // JSON 추출
    const jsonMatch = text.match(/\{[^}]+\}/);
    if (jsonMatch) {
      const parsed = JSON.parse(jsonMatch[0]);
      return {
        action: parsed.action || 'chat',
        type: parsed.type,
        confidence: parsed.confidence || 0.5,
      };
    }
  } catch (error) {
    console.error('Intent analysis error:', error);
  }

  // 기본값: 일반 대화
  return { action: 'chat', confidence: 0.5 };
}

// n8n Webhook 호출
export async function callN8n(
  env: Env,
  type: string,
  userMessage: string,
  userId: number,
  chatId: string,
  userProfile?: string | null
): Promise<N8nResponse> {
  if (!env.N8N_WEBHOOK_URL) {
    return { error: 'n8n이 설정되지 않았습니다.' };
  }

  const webhookUrl = `${env.N8N_WEBHOOK_URL}/webhook/telegram-bot`;

  try {
    const response = await fetch(webhookUrl, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({
        type,
        message: userMessage,
        user_id: userId,
        chat_id: chatId,
        profile: userProfile,
        timestamp: new Date().toISOString(),
      }),
    });

    if (!response.ok) {
      console.error('n8n error:', response.status, await response.text());
      return { error: `n8n 호출 실패 (${response.status})` };
    }

    const data = await response.json() as N8nResponse;
    return data;
  } catch (error) {
    console.error('n8n fetch error:', error);
    return { error: 'n8n 연결 실패' };
  }
}

// 스마트 라우팅: AI 판단 → n8n 또는 로컬 AI
export async function smartRoute(
  env: Env,
  userMessage: string,
  userId: number,
  chatId: string,
  userProfile?: string | null
): Promise<{ useN8n: boolean; n8nType?: string; n8nResponse?: N8nResponse }> {
  // n8n URL이 없으면 항상 로컬 AI 사용
  if (!env.N8N_WEBHOOK_URL) {
    return { useN8n: false };
  }

  // 의도 분석
  const intent = await analyzeIntent(env.AI, userMessage);

  // n8n 호출이 필요하고 신뢰도가 높은 경우
  if (intent.action === 'n8n' && intent.confidence >= 0.7 && intent.type) {
    const n8nResponse = await callN8n(
      env,
      intent.type,
      userMessage,
      userId,
      chatId,
      userProfile
    );

    // n8n 응답이 성공적인 경우
    if (n8nResponse.reply && !n8nResponse.error) {
      return {
        useN8n: true,
        n8nType: intent.type,
        n8nResponse,
      };
    }
  }

  return { useN8n: false };
}