Initial commit: Telegram bot with Cloudflare Workers

- OpenAI GPT-4o-mini with Function Calling
- Cloudflare D1 for user profiles and message buffer
- Sliding window (3 summaries max) for infinite context
- Tools: weather, search, time, calculator
- Workers AI fallback support
- Webhook security with rate limiting

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
kappa
2026-01-14 13:00:44 +09:00
commit 1e71e035e7
15 changed files with 2272 additions and 0 deletions

26
.gitignore vendored Normal file
View File

@@ -0,0 +1,26 @@
# Dependencies
node_modules/
.npm
# Build output
dist/
.wrangler/
# Environment & Secrets
.env
.env.*
.dev.vars
# IDE
.idea/
.vscode/
*.swp
*.swo
.DS_Store
# Logs
*.log
npm-debug.log*
# Test coverage
coverage/

327
README.md Normal file
View File

@@ -0,0 +1,327 @@
# Cloudflare Workers 텔레그램 봇
> Cloudflare Workers + D1 + OpenAI를 활용한 사용자 프로필 기반 텔레그램 봇
## 목차
1. [개요](#개요)
2. [아키텍처](#아키텍처)
3. [Function Calling](#function-calling)
4. [프로젝트 구조](#프로젝트-구조)
5. [배포 가이드](#배포-가이드)
6. [보안 설정](#보안-설정)
7. [봇 명령어](#봇-명령어)
---
## 개요
### 주요 기능
- **OpenAI GPT-4o-mini**: 고품질 AI 응답 및 Function Calling 지원
- **사용자 프로필**: 대화에서 사용자의 관심사, 목표, 맥락을 추출하여 프로필 구축
- **Function Calling**: 날씨, 검색, 시간, 계산 등 AI가 자동으로 도구 호출
- **무한 컨텍스트**: 슬라이딩 윈도우(3개)로 프로필 유지, 무제한 대화 기억
- **개인화 응답**: 프로필 기반으로 맞춤형 AI 응답 제공
- **폴백 지원**: OpenAI 미설정 시 Workers AI(Llama)로 자동 전환
### 기술 스택
| 서비스 | 용도 |
|--------|------|
| **Workers** | 서버리스 런타임 |
| **D1** | SQLite 데이터베이스 |
| **OpenAI** | GPT-4o-mini + Function Calling |
| **Workers AI** | 폴백용 (Llama 3.1 8B) |
---
## 아키텍처
### 메시지 처리 흐름
```
[사용자 메시지]
┌──────────────────┐
│ Cloudflare │
│ Worker │
└──────────────────┘
┌──────────────────┐
│ OpenAI API │ ← GPT-4o-mini
│ (Function Call) │ 도구 호출 자동 판단
└──────────────────┘
┌───┴───┬───────┬───────┐
▼ ▼ ▼ ▼
[날씨] [검색] [시간] [계산] → 외부 API
│ │ │ │
└───┬───┴───────┴───────┘
┌──────────────────┐
│ 최종 응답 생성 │
└──────────────────┘
[D1 저장] → [Telegram 응답]
```
### 사용자 프로필 시스템
```
[사용자 메시지]
┌──────────────────┐
│ message_buffer │ ← 최대 19개 (20개 되면 프로필 업데이트)
└──────────────────┘
│ 20개 도달
┌──────────────────┐
│ 프로필 분석 │ ← 사용자 발언만 추출하여 분석
│ (OpenAI) │ 봇 응답은 무시
└──────────────────┘
┌──────────────────┐
│ summaries │ ← 최근 3개만 유지 (슬라이딩 윈도우)
│ [v1] [v2] [v3] │
└──────────────────┘
```
### 프로필 분석 내용
| 추출 정보 | 설명 |
|-----------|------|
| **관심사** | 사용자가 자주 언급하는 주제 |
| **목표** | 해결하려는 문제, 달성하려는 것 |
| **맥락** | 직업, 상황, 배경 정보 |
| **선호도** | 좋아하는 것, 싫어하는 것 |
| **질문 패턴** | 무엇에 대해 알고 싶어하는지 |
---
## Function Calling
OpenAI Function Calling을 통해 AI가 자동으로 필요한 도구를 호출합니다.
### 지원 기능
| 기능 | 예시 질문 | API |
|------|-----------|-----|
| **날씨** | "서울 날씨", "도쿄 날씨 알려줘" | wttr.in |
| **검색** | "파이썬이 뭐야", "클라우드플레어란" | DuckDuckGo |
| **시간** | "지금 몇 시야", "뉴욕 시간" | 내장 |
| **계산** | "123 * 456", "100의 20%" | 내장 |
### 동작 방식
```
사용자: "서울 날씨 어때?"
OpenAI: "get_weather 함수를 호출해야겠다"
Worker: wttr.in API 호출 → 날씨 데이터 수신
OpenAI: 날씨 데이터를 자연어로 응답 생성
응답: "🌤 서울 날씨\n온도: 5°C\n습도: 45%..."
```
---
## 프로젝트 구조
```
telegram-bot-workers/
├── src/
│ ├── index.ts # 메인 Worker
│ ├── types.ts # 타입 정의
│ ├── security.ts # Webhook 보안 검증
│ ├── telegram.ts # Telegram API 유틸
│ ├── summary-service.ts # 프로필 분석 서비스
│ ├── openai-service.ts # OpenAI + Function Calling
│ ├── n8n-service.ts # n8n 연동 (선택)
│ └── commands.ts # 봇 명령어 핸들러
├── schema.sql # D1 스키마
├── wrangler.toml # Wrangler 설정
├── n8n-workflow-example.json # n8n 워크플로우 예시
├── package.json
├── tsconfig.json
└── README.md
```
---
## 배포 가이드
### 1. 프로젝트 설정
```bash
cd telegram-bot-workers
npm install
```
### 2. D1 데이터베이스
현재 설정:
```toml
[[d1_databases]]
binding = "DB"
database_name = "telegram-conversations"
database_id = "c285bb5b-888b-405d-b36f-475ae5aed20e"
```
스키마:
- `users` - 사용자 정보
- `message_buffer` - 메시지 임시 저장
- `summaries` - 프로필 저장
### 3. Secrets 설정
```bash
# Bot Token (BotFather에서 발급)
wrangler secret put BOT_TOKEN
# Webhook Secret
wrangler secret put WEBHOOK_SECRET
# OpenAI API Key (필수)
wrangler secret put OPENAI_API_KEY
```
### Vault 연동 (선택)
API 키는 HashiCorp Vault에서 중앙 관리됩니다.
```bash
# Vault에서 OpenAI API 키 조회
vault kv get secret/openai
# 저장된 정보
# - api_key: OpenAI API 키
# - email: kappa.inouter@gmail.com (계정 관리용)
# Vault에서 키 가져와서 Worker에 설정
OPENAI_KEY=$(vault kv get -field=api_key secret/openai)
echo $OPENAI_KEY | wrangler secret put OPENAI_API_KEY
```
> **참고**: Vault 서버: `https://vault.anvil.it.com`
### 4. 배포
```bash
wrangler deploy
```
### 5. Webhook 설정
```bash
curl https://telegram-summary-bot.kappa-d8e.workers.dev/setup-webhook
curl https://telegram-summary-bot.kappa-d8e.workers.dev/webhook-info
```
---
## 보안 설정
### Webhook 보안 검증
| 검증 항목 | 설명 |
|-----------|------|
| **Secret Token** | `X-Telegram-Bot-Api-Secret-Token` 헤더 검증 |
| **Timing-safe 비교** | 타이밍 공격 방지 |
| **Timestamp** | 60초 이내 메시지만 처리 |
| **Rate Limiting** | 30req/분/사용자 |
### API 키 관리
| 키 | 저장 위치 | 비고 |
|----|-----------|------|
| `BOT_TOKEN` | Wrangler Secret | BotFather 발급 |
| `WEBHOOK_SECRET` | Wrangler Secret | 자동 생성 |
| `OPENAI_API_KEY` | Wrangler Secret + Vault | kappa.inouter@gmail.com |
**Vault 경로**: `secret/openai` @ `vault.anvil.it.com`
---
## 봇 명령어
| 명령어 | 설명 |
|--------|------|
| `/start` | 봇 시작, 기능 소개 |
| `/help` | 도움말 |
| `/context` | 현재 컨텍스트 상태 (버퍼 수, 프로필 버전) |
| `/profile` | 내 프로필 보기 |
| `/stats` | 대화 통계 |
| `/reset` | 모든 대화 기록 삭제 |
---
## 설정값
`wrangler.toml`:
```toml
name = "telegram-summary-bot"
main = "src/index.ts"
compatibility_date = "2024-01-01"
[ai]
binding = "AI"
[vars]
SUMMARY_THRESHOLD = "20"
MAX_SUMMARIES_PER_USER = "3"
N8N_WEBHOOK_URL = "https://n8n.anvil.it.com"
[[d1_databases]]
binding = "DB"
database_name = "telegram-conversations"
database_id = "c285bb5b-888b-405d-b36f-475ae5aed20e"
```
---
## 비용 예측
### 월간 예상 비용
| 사용량 | D1 | OpenAI | Workers | 총 |
|--------|-----|--------|---------|-----|
| 1만 메시지 | $0 | ~$0.20 | $0 | **~$0.20** |
| 10만 메시지 | $0 | ~$2.00 | $0 | **~$2.00** |
| 100만 메시지 | $0 | ~$20.00 | ~$0.30 | **~$20.30** |
> GPT-4o-mini: 입력 $0.15/1M, 출력 $0.60/1M 토큰
---
## API 엔드포인트
| 경로 | 메서드 | 설명 |
|------|--------|------|
| `/` | GET | 서비스 정보 |
| `/health` | GET | 헬스 체크 |
| `/webhook-info` | GET | Webhook 상태 |
| `/setup-webhook` | GET | Webhook 설정 |
| `/webhook` | POST | Telegram Webhook |
---
## 참고
- [OpenAI API](https://platform.openai.com/docs)
- [Cloudflare D1](https://developers.cloudflare.com/d1/)
- [Cloudflare Workers](https://developers.cloudflare.com/workers/)
- [Telegram Bot API](https://core.telegram.org/bots/api)

445
n8n-workflow-example.json Normal file
View File

@@ -0,0 +1,445 @@
{
"name": "Telegram Bot Handler",
"nodes": [
{
"parameters": {
"httpMethod": "POST",
"path": "telegram-bot",
"responseMode": "responseNode",
"options": {}
},
"id": "webhook-1",
"name": "Webhook",
"type": "n8n-nodes-base.webhook",
"typeVersion": 2,
"position": [250, 300],
"webhookId": "telegram-bot"
},
{
"parameters": {
"rules": {
"rules": [
{
"outputKey": "weather",
"conditions": {
"options": { "version": 2 },
"combinator": "and",
"conditions": [
{
"leftValue": "={{ $json.type }}",
"rightValue": "weather",
"operator": { "type": "string", "operation": "equals" }
}
]
}
},
{
"outputKey": "search",
"conditions": {
"options": { "version": 2 },
"combinator": "and",
"conditions": [
{
"leftValue": "={{ $json.type }}",
"rightValue": "search",
"operator": { "type": "string", "operation": "equals" }
}
]
}
},
{
"outputKey": "translate",
"conditions": {
"options": { "version": 2 },
"combinator": "and",
"conditions": [
{
"leftValue": "={{ $json.type }}",
"rightValue": "translate",
"operator": { "type": "string", "operation": "equals" }
}
]
}
},
{
"outputKey": "image",
"conditions": {
"options": { "version": 2 },
"combinator": "and",
"conditions": [
{
"leftValue": "={{ $json.type }}",
"rightValue": "image",
"operator": { "type": "string", "operation": "equals" }
}
]
}
},
{
"outputKey": "summarize_url",
"conditions": {
"options": { "version": 2 },
"combinator": "and",
"conditions": [
{
"leftValue": "={{ $json.type }}",
"rightValue": "summarize_url",
"operator": { "type": "string", "operation": "equals" }
}
]
}
},
{
"outputKey": "default",
"conditions": {
"options": { "version": 2 },
"combinator": "and",
"conditions": [
{
"leftValue": "={{ $json.type }}",
"rightValue": "",
"operator": { "type": "string", "operation": "notEmpty" }
}
]
}
}
]
},
"options": {}
},
"id": "switch-1",
"name": "Route by Type",
"type": "n8n-nodes-base.switch",
"typeVersion": 3,
"position": [480, 300]
},
{
"parameters": {
"method": "GET",
"url": "=https://api.openweathermap.org/data/2.5/weather?q={{ $json.message.match(/[가-힣a-zA-Z]+/)?.[0] || 'Seoul' }}&appid={{ $env.OPENWEATHER_API_KEY }}&units=metric&lang=kr",
"options": {}
},
"id": "weather-api",
"name": "Weather API",
"type": "n8n-nodes-base.httpRequest",
"typeVersion": 4.2,
"position": [720, 100]
},
{
"parameters": {
"assignments": {
"assignments": [
{
"id": "reply",
"name": "reply",
"value": "={{ '🌤 ' + $json.name + ' 날씨\\n온도: ' + $json.main.temp + '°C\\n체감: ' + $json.main.feels_like + '°C\\n습도: ' + $json.main.humidity + '%\\n' + $json.weather[0].description }}",
"type": "string"
}
]
},
"options": {}
},
"id": "weather-format",
"name": "Format Weather",
"type": "n8n-nodes-base.set",
"typeVersion": 3.4,
"position": [940, 100]
},
{
"parameters": {
"model": "gpt-4o-mini",
"messages": {
"values": [
{
"content": "=다음 질문에 대해 웹 검색 결과를 바탕으로 답변해주세요: {{ $('Webhook').item.json.message }}"
}
]
},
"options": {}
},
"id": "search-ai",
"name": "AI Search",
"type": "@n8n/n8n-nodes-langchain.openAi",
"typeVersion": 1.4,
"position": [720, 220]
},
{
"parameters": {
"assignments": {
"assignments": [
{
"id": "reply",
"name": "reply",
"value": "={{ $json.message.content }}",
"type": "string"
}
]
},
"options": {}
},
"id": "search-format",
"name": "Format Search",
"type": "n8n-nodes-base.set",
"typeVersion": 3.4,
"position": [940, 220]
},
{
"parameters": {
"method": "POST",
"url": "https://api-free.deepl.com/v2/translate",
"sendBody": true,
"bodyParameters": {
"parameters": [
{
"name": "text",
"value": "={{ $('Webhook').item.json.message }}"
},
{
"name": "target_lang",
"value": "EN"
}
]
},
"options": {
"headers": {
"header": [
{
"name": "Authorization",
"value": "=DeepL-Auth-Key {{ $env.DEEPL_API_KEY }}"
}
]
}
}
},
"id": "translate-api",
"name": "Translate API",
"type": "n8n-nodes-base.httpRequest",
"typeVersion": 4.2,
"position": [720, 340]
},
{
"parameters": {
"assignments": {
"assignments": [
{
"id": "reply",
"name": "reply",
"value": "=🌐 번역 결과:\\n{{ $json.translations[0].text }}",
"type": "string"
}
]
},
"options": {}
},
"id": "translate-format",
"name": "Format Translate",
"type": "n8n-nodes-base.set",
"typeVersion": 3.4,
"position": [940, 340]
},
{
"parameters": {
"method": "POST",
"url": "https://api.openai.com/v1/images/generations",
"sendHeaders": true,
"headerParameters": {
"parameters": [
{
"name": "Authorization",
"value": "=Bearer {{ $env.OPENAI_API_KEY }}"
}
]
},
"sendBody": true,
"specifyBody": "json",
"jsonBody": "={\n \"model\": \"dall-e-3\",\n \"prompt\": \"{{ $('Webhook').item.json.message }}\",\n \"n\": 1,\n \"size\": \"1024x1024\"\n}",
"options": {}
},
"id": "image-api",
"name": "Image Generation",
"type": "n8n-nodes-base.httpRequest",
"typeVersion": 4.2,
"position": [720, 460]
},
{
"parameters": {
"assignments": {
"assignments": [
{
"id": "reply",
"name": "reply",
"value": "=🎨 이미지 생성 완료!\\n{{ $json.data[0].url }}",
"type": "string"
}
]
},
"options": {}
},
"id": "image-format",
"name": "Format Image",
"type": "n8n-nodes-base.set",
"typeVersion": 3.4,
"position": [940, 460]
},
{
"parameters": {
"method": "GET",
"url": "={{ $('Webhook').item.json.message.match(/https?:\\/\\/[^\\s]+/)?.[0] }}",
"options": {}
},
"id": "fetch-url",
"name": "Fetch URL",
"type": "n8n-nodes-base.httpRequest",
"typeVersion": 4.2,
"position": [720, 580]
},
{
"parameters": {
"model": "gpt-4o-mini",
"messages": {
"values": [
{
"content": "=다음 웹페이지 내용을 한국어로 300자 이내로 요약해주세요:\\n\\n{{ $json.data.substring(0, 5000) }}"
}
]
},
"options": {}
},
"id": "summarize-ai",
"name": "Summarize URL",
"type": "@n8n/n8n-nodes-langchain.openAi",
"typeVersion": 1.4,
"position": [940, 580]
},
{
"parameters": {
"assignments": {
"assignments": [
{
"id": "reply",
"name": "reply",
"value": "=📄 URL 요약:\\n{{ $json.message.content }}",
"type": "string"
}
]
},
"options": {}
},
"id": "summarize-format",
"name": "Format Summary",
"type": "n8n-nodes-base.set",
"typeVersion": 3.4,
"position": [1160, 580]
},
{
"parameters": {
"assignments": {
"assignments": [
{
"id": "reply",
"name": "reply",
"value": "=❓ 지원하지 않는 기능입니다: {{ $('Webhook').item.json.type }}",
"type": "string"
}
]
},
"options": {}
},
"id": "default-response",
"name": "Default Response",
"type": "n8n-nodes-base.set",
"typeVersion": 3.4,
"position": [720, 700]
},
{
"parameters": {
"options": {}
},
"id": "respond-1",
"name": "Respond to Webhook",
"type": "n8n-nodes-base.respondToWebhook",
"typeVersion": 1.1,
"position": [1380, 300]
}
],
"connections": {
"Webhook": {
"main": [
[{ "node": "Route by Type", "type": "main", "index": 0 }]
]
},
"Route by Type": {
"main": [
[{ "node": "Weather API", "type": "main", "index": 0 }],
[{ "node": "AI Search", "type": "main", "index": 0 }],
[{ "node": "Translate API", "type": "main", "index": 0 }],
[{ "node": "Image Generation", "type": "main", "index": 0 }],
[{ "node": "Fetch URL", "type": "main", "index": 0 }],
[{ "node": "Default Response", "type": "main", "index": 0 }]
]
},
"Weather API": {
"main": [
[{ "node": "Format Weather", "type": "main", "index": 0 }]
]
},
"Format Weather": {
"main": [
[{ "node": "Respond to Webhook", "type": "main", "index": 0 }]
]
},
"AI Search": {
"main": [
[{ "node": "Format Search", "type": "main", "index": 0 }]
]
},
"Format Search": {
"main": [
[{ "node": "Respond to Webhook", "type": "main", "index": 0 }]
]
},
"Translate API": {
"main": [
[{ "node": "Format Translate", "type": "main", "index": 0 }]
]
},
"Format Translate": {
"main": [
[{ "node": "Respond to Webhook", "type": "main", "index": 0 }]
]
},
"Image Generation": {
"main": [
[{ "node": "Format Image", "type": "main", "index": 0 }]
]
},
"Format Image": {
"main": [
[{ "node": "Respond to Webhook", "type": "main", "index": 0 }]
]
},
"Fetch URL": {
"main": [
[{ "node": "Summarize URL", "type": "main", "index": 0 }]
]
},
"Summarize URL": {
"main": [
[{ "node": "Format Summary", "type": "main", "index": 0 }]
]
},
"Format Summary": {
"main": [
[{ "node": "Respond to Webhook", "type": "main", "index": 0 }]
]
},
"Default Response": {
"main": [
[{ "node": "Respond to Webhook", "type": "main", "index": 0 }]
]
}
},
"settings": {
"executionOrder": "v1"
}
}

27
package.json Normal file
View File

@@ -0,0 +1,27 @@
{
"name": "telegram-summary-bot",
"version": "1.0.0",
"description": "Telegram bot with rolling summary using Cloudflare Workers + D1",
"main": "src/index.ts",
"scripts": {
"dev": "wrangler dev",
"deploy": "wrangler deploy",
"db:create": "wrangler d1 create telegram-summary-db",
"db:init": "wrangler d1 execute telegram-summary-db --file=schema.sql",
"db:init:local": "wrangler d1 execute telegram-summary-db --local --file=schema.sql",
"tail": "wrangler tail"
},
"devDependencies": {
"@cloudflare/workers-types": "^4.20241127.0",
"typescript": "^5.3.3",
"wrangler": "^3.93.0"
},
"keywords": [
"telegram",
"bot",
"cloudflare",
"workers",
"d1",
"ai"
]
}

42
schema.sql Normal file
View File

@@ -0,0 +1,42 @@
-- Telegram Bot Rolling Summary Schema
-- D1 Database for Cloudflare Workers
-- 사용자 테이블
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
telegram_id TEXT UNIQUE NOT NULL,
username TEXT,
first_name TEXT,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
);
-- 메시지 버퍼 (요약 전 임시 저장)
CREATE TABLE IF NOT EXISTS message_buffer (
id INTEGER PRIMARY KEY AUTOINCREMENT,
user_id INTEGER NOT NULL,
chat_id TEXT NOT NULL,
role TEXT NOT NULL CHECK(role IN ('user', 'bot')),
message TEXT NOT NULL,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
FOREIGN KEY (user_id) REFERENCES users(id)
);
-- 요약 저장 테이블 (슬라이딩 윈도우: 최대 3개만 유지)
CREATE TABLE IF NOT EXISTS summaries (
id INTEGER PRIMARY KEY AUTOINCREMENT,
user_id INTEGER NOT NULL,
chat_id TEXT NOT NULL,
generation INTEGER NOT NULL DEFAULT 1,
summary TEXT NOT NULL,
message_count INTEGER NOT NULL,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
FOREIGN KEY (user_id) REFERENCES users(id)
);
-- 인덱스
CREATE INDEX IF NOT EXISTS idx_buffer_user ON message_buffer(user_id);
CREATE INDEX IF NOT EXISTS idx_buffer_chat ON message_buffer(user_id, chat_id);
CREATE INDEX IF NOT EXISTS idx_summary_user ON summaries(user_id, chat_id);
CREATE INDEX IF NOT EXISTS idx_summary_latest ON summaries(user_id, chat_id, generation DESC);
CREATE INDEX IF NOT EXISTS idx_users_telegram ON users(telegram_id);

124
src/commands.ts Normal file
View File

@@ -0,0 +1,124 @@
import { Env } from './types';
import { getConversationContext, getLatestSummary } from './summary-service';
export async function handleCommand(
env: Env,
userId: number,
chatId: string,
command: string,
_args: string
): Promise<string> {
const config = {
threshold: parseInt(env.SUMMARY_THRESHOLD || '20', 10),
maxSummaries: parseInt(env.MAX_SUMMARIES_PER_USER || '3', 10),
};
switch (command) {
case '/start':
return `👋 안녕하세요! AI 어시스턴트입니다.
<b>특징:</b>
• 대화를 통해 당신을 이해합니다
${config.threshold}개 메시지마다 프로필 업데이트
• 무한한 대화 기억 가능
<b>명령어:</b>
/help - 도움말
/context - 현재 컨텍스트 확인
/profile - 내 프로필 보기
/stats - 대화 통계
/reset - 대화 초기화`;
case '/help':
return `📖 <b>도움말</b>
<b>기본 명령어:</b>
/start - 봇 시작
/context - 현재 컨텍스트 상태
/profile - 내 프로필 보기
/stats - 대화 통계
/reset - 모든 대화 기록 삭제
<b>사용자 프로필 시스템:</b>
• 메시지 ${config.threshold}개마다 프로필 업데이트
• 사용자 발언 위주로 관심사/목표/맥락 분석
• 의미 없는 대화(인사, 확인 등)는 제외
일반 메시지를 보내면 AI가 응답합니다.`;
case '/context': {
const ctx = await getConversationContext(env.DB, userId, chatId);
const remaining = config.threshold - ctx.recentMessages.length;
return `📊 <b>현재 컨텍스트</b>
분석된 메시지: ${ctx.previousSummary?.message_count || 0}
버퍼 메시지: ${ctx.recentMessages.length}
프로필 버전: ${ctx.previousSummary?.generation || 0}
총 메시지: ${ctx.totalMessages}
💡 ${remaining > 0 ? `${remaining}개 메시지 후 프로필 업데이트` : '업데이트 대기 중'}`;
}
case '/profile':
case '/summary': {
const summary = await getLatestSummary(env.DB, userId, chatId);
if (!summary) {
return '📭 아직 프로필이 없습니다.\n대화를 더 나눠보세요!';
}
const createdAt = new Date(summary.created_at).toLocaleString('ko-KR', {
timeZone: 'Asia/Seoul',
});
return `👤 <b>내 프로필</b> (v${summary.generation})
${summary.summary}
<i>분석된 메시지: ${summary.message_count}개</i>
<i>업데이트: ${createdAt}</i>`;
}
case '/stats': {
const ctx = await getConversationContext(env.DB, userId, chatId);
const profileCount = await env.DB
.prepare('SELECT COUNT(*) as cnt FROM summaries WHERE user_id = ?')
.bind(userId)
.first<{ cnt: number }>();
return `📈 <b>대화 통계</b>
총 메시지: ${ctx.totalMessages}
프로필 버전: ${ctx.previousSummary?.generation || 0}
저장된 프로필: ${profileCount?.cnt || 0}
버퍼 대기: ${ctx.recentMessages.length}`;
}
case '/reset': {
await env.DB.batch([
env.DB.prepare('DELETE FROM message_buffer WHERE user_id = ?').bind(userId),
env.DB.prepare('DELETE FROM summaries WHERE user_id = ?').bind(userId),
]);
return '🗑️ 모든 대화 기록과 요약이 초기화되었습니다.';
}
case '/debug': {
// 디버그용 명령어 (개발 시 유용)
const ctx = await getConversationContext(env.DB, userId, chatId);
const recentMsgs = ctx.recentMessages.slice(-5).map((m, i) =>
`${i + 1}. [${m.role}] ${m.message.substring(0, 30)}...`
).join('\n');
return `🔧 <b>디버그 정보</b>
User ID: ${userId}
Chat ID: ${chatId}
Buffer Count: ${ctx.recentMessages.length}
Summary Gen: ${ctx.previousSummary?.generation || 0}
<b>최근 버퍼 (5개):</b>
${recentMsgs || '(비어있음)'}`;
}
default:
return '❓ 알 수 없는 명령어입니다.\n/help를 입력해보세요.';
}
}

201
src/index.ts Normal file
View File

@@ -0,0 +1,201 @@
import { Env, TelegramUpdate } from './types';
import { validateWebhookRequest, checkRateLimit } from './security';
import { sendMessage, setWebhook, getWebhookInfo, sendChatAction } from './telegram';
import {
addToBuffer,
processAndSummarize,
generateAIResponse,
} from './summary-service';
import { handleCommand } from './commands';
// 사용자 조회/생성
async function getOrCreateUser(
db: D1Database,
telegramId: string,
firstName: string,
username?: string
): Promise<number> {
const existing = await db
.prepare('SELECT id FROM users WHERE telegram_id = ?')
.bind(telegramId)
.first<{ id: number }>();
if (existing) {
// 마지막 활동 시간 업데이트
await db
.prepare('UPDATE users SET updated_at = CURRENT_TIMESTAMP WHERE id = ?')
.bind(existing.id)
.run();
return existing.id;
}
// 새 사용자 생성
const result = await db
.prepare('INSERT INTO users (telegram_id, first_name, username) VALUES (?, ?, ?)')
.bind(telegramId, firstName, username || null)
.run();
return result.meta.last_row_id as number;
}
// 메시지 처리
async function handleMessage(
env: Env,
update: TelegramUpdate
): Promise<void> {
if (!update.message?.text) return;
const { message } = update;
const chatId = message.chat.id;
const chatIdStr = chatId.toString();
const text = message.text;
const telegramUserId = message.from.id.toString();
// Rate Limiting 체크
if (!checkRateLimit(telegramUserId)) {
await sendMessage(
env.BOT_TOKEN,
chatId,
'⚠️ 너무 많은 요청입니다. 잠시 후 다시 시도해주세요.'
);
return;
}
// 사용자 처리
const userId = await getOrCreateUser(
env.DB,
telegramUserId,
message.from.first_name,
message.from.username
);
let responseText: string;
// 명령어 처리
if (text.startsWith('/')) {
const [command, ...argParts] = text.split(' ');
const args = argParts.join(' ');
responseText = await handleCommand(env, userId, chatIdStr, command, args);
} else {
// 타이핑 표시
await sendChatAction(env.BOT_TOKEN, chatId, 'typing');
// 1. 사용자 메시지 버퍼에 추가
await addToBuffer(env.DB, userId, chatIdStr, 'user', text);
try {
// 2. AI 응답 생성
responseText = await generateAIResponse(env, userId, chatIdStr, text);
// 3. 봇 응답 버퍼에 추가
await addToBuffer(env.DB, userId, chatIdStr, 'bot', responseText);
// 4. 임계값 도달시 프로필 업데이트
const { summarized } = await processAndSummarize(env, userId, chatIdStr);
if (summarized) {
responseText += '\n\n<i>👤 프로필이 업데이트되었습니다.</i>';
}
} catch (error) {
console.error('AI Response error:', error);
responseText = `⚠️ AI 응답 생성 중 오류가 발생했습니다.\n\n<code>${String(error)}</code>`;
}
}
await sendMessage(env.BOT_TOKEN, chatId, responseText);
}
export default {
// HTTP 요청 핸들러
async fetch(request: Request, env: Env): Promise<Response> {
const url = new URL(request.url);
// Webhook 설정 엔드포인트
if (url.pathname === '/setup-webhook') {
if (!env.BOT_TOKEN) {
return Response.json({ error: 'BOT_TOKEN not configured' }, { status: 500 });
}
if (!env.WEBHOOK_SECRET) {
return Response.json({ error: 'WEBHOOK_SECRET not configured' }, { status: 500 });
}
const webhookUrl = `${url.origin}/webhook`;
const result = await setWebhook(env.BOT_TOKEN, webhookUrl, env.WEBHOOK_SECRET);
return Response.json(result);
}
// Webhook 정보 조회
if (url.pathname === '/webhook-info') {
if (!env.BOT_TOKEN) {
return Response.json({ error: 'BOT_TOKEN not configured' }, { status: 500 });
}
const result = await getWebhookInfo(env.BOT_TOKEN);
return Response.json(result);
}
// 헬스 체크
if (url.pathname === '/health') {
try {
const userCount = await env.DB
.prepare('SELECT COUNT(*) as cnt FROM users')
.first<{ cnt: number }>();
const summaryCount = await env.DB
.prepare('SELECT COUNT(*) as cnt FROM summaries')
.first<{ cnt: number }>();
return Response.json({
status: 'ok',
timestamp: new Date().toISOString(),
stats: {
users: userCount?.cnt || 0,
summaries: summaryCount?.cnt || 0,
},
});
} catch (error) {
return Response.json({
status: 'error',
error: String(error),
}, { status: 500 });
}
}
// Telegram Webhook 처리
if (url.pathname === '/webhook') {
// 보안 검증
const validation = await validateWebhookRequest(request, env);
if (!validation.valid) {
console.error('Webhook validation failed:', validation.error);
return new Response(validation.error, { status: 401 });
}
try {
await handleMessage(env, validation.update!);
return new Response('OK');
} catch (error) {
console.error('Message handling error:', error);
return new Response('Error', { status: 500 });
}
}
// 루트 경로
if (url.pathname === '/') {
return new Response(`
Telegram Rolling Summary Bot
Endpoints:
GET /health - Health check
GET /webhook-info - Webhook status
GET /setup-webhook - Configure webhook
POST /webhook - Telegram webhook (authenticated)
Documentation: https://github.com/your-repo
`.trim(), {
headers: { 'Content-Type': 'text/plain' },
});
}
return new Response('Not Found', { status: 404 });
},
};

155
src/n8n-service.ts Normal file
View File

@@ -0,0 +1,155 @@
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 };
}

272
src/openai-service.ts Normal file
View File

@@ -0,0 +1,272 @@
import { Env } from './types';
interface OpenAIMessage {
role: 'system' | 'user' | 'assistant' | 'tool';
content: string | null;
tool_calls?: ToolCall[];
tool_call_id?: string;
}
interface ToolCall {
id: string;
type: 'function';
function: {
name: string;
arguments: string;
};
}
interface OpenAIResponse {
choices: {
message: OpenAIMessage;
finish_reason: string;
}[];
}
// 사용 가능한 도구 정의
const tools = [
{
type: 'function',
function: {
name: 'get_weather',
description: '특정 도시의 현재 날씨 정보를 가져옵니다',
parameters: {
type: 'object',
properties: {
city: {
type: 'string',
description: '도시 이름 (예: Seoul, Tokyo, New York)',
},
},
required: ['city'],
},
},
},
{
type: 'function',
function: {
name: 'search_web',
description: '웹에서 정보를 검색합니다',
parameters: {
type: 'object',
properties: {
query: {
type: 'string',
description: '검색 쿼리',
},
},
required: ['query'],
},
},
},
{
type: 'function',
function: {
name: 'get_current_time',
description: '현재 시간을 가져옵니다',
parameters: {
type: 'object',
properties: {
timezone: {
type: 'string',
description: '타임존 (예: Asia/Seoul, UTC)',
},
},
required: [],
},
},
},
{
type: 'function',
function: {
name: 'calculate',
description: '수학 계산을 수행합니다',
parameters: {
type: 'object',
properties: {
expression: {
type: 'string',
description: '계산할 수식 (예: 2+2, 100*5)',
},
},
required: ['expression'],
},
},
},
];
// 도구 실행
async function executeTool(name: string, args: Record<string, string>): Promise<string> {
switch (name) {
case 'get_weather': {
const city = args.city || 'Seoul';
try {
const response = await fetch(
`https://wttr.in/${encodeURIComponent(city)}?format=j1`
);
const data = await response.json() as any;
const current = data.current_condition[0];
return `🌤 ${city} 날씨
온도: ${current.temp_C}°C (체감 ${current.FeelsLikeC}°C)
상태: ${current.weatherDesc[0].value}
습도: ${current.humidity}%
풍속: ${current.windspeedKmph} km/h`;
} catch (error) {
return `날씨 정보를 가져올 수 없습니다: ${city}`;
}
}
case 'search_web': {
// 간단한 DuckDuckGo Instant Answer API
const query = args.query;
try {
const response = await fetch(
`https://api.duckduckgo.com/?q=${encodeURIComponent(query)}&format=json&no_html=1`
);
const data = await response.json() as any;
if (data.Abstract) {
return `🔍 검색 결과: ${query}\n\n${data.Abstract}\n\n출처: ${data.AbstractSource}`;
} else if (data.RelatedTopics?.length > 0) {
const topics = data.RelatedTopics.slice(0, 3)
.filter((t: any) => t.Text)
.map((t: any) => `${t.Text}`)
.join('\n');
return `🔍 관련 정보: ${query}\n\n${topics}`;
}
return `"${query}"에 대한 즉시 답변을 찾을 수 없습니다. 더 구체적인 질문을 해주세요.`;
} catch (error) {
return `검색 중 오류가 발생했습니다.`;
}
}
case 'get_current_time': {
const timezone = args.timezone || 'Asia/Seoul';
try {
const now = new Date();
const formatted = now.toLocaleString('ko-KR', { timeZone: timezone });
return `🕐 현재 시간 (${timezone}): ${formatted}`;
} catch (error) {
return `시간 정보를 가져올 수 없습니다.`;
}
}
case 'calculate': {
const expression = args.expression;
try {
// 안전한 수식 계산 (기본 연산만)
const sanitized = expression.replace(/[^0-9+\-*/().% ]/g, '');
const result = Function('"use strict"; return (' + sanitized + ')')();
return `🔢 계산 결과: ${expression} = ${result}`;
} catch (error) {
return `계산할 수 없는 수식입니다: ${expression}`;
}
}
default:
return `알 수 없는 도구: ${name}`;
}
}
// OpenAI API 호출
async function callOpenAI(
apiKey: string,
messages: OpenAIMessage[],
useTools: boolean = true
): Promise<OpenAIResponse> {
const response = await fetch('https://api.openai.com/v1/chat/completions', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
'Authorization': `Bearer ${apiKey}`,
},
body: JSON.stringify({
model: 'gpt-4o-mini',
messages,
tools: useTools ? tools : undefined,
tool_choice: useTools ? 'auto' : undefined,
max_tokens: 1000,
}),
});
if (!response.ok) {
const error = await response.text();
throw new Error(`OpenAI API error: ${response.status} - ${error}`);
}
return response.json();
}
// 메인 응답 생성 함수
export async function generateOpenAIResponse(
env: Env,
userMessage: string,
systemPrompt: string,
recentContext: { role: 'user' | 'assistant'; content: string }[]
): Promise<string> {
if (!env.OPENAI_API_KEY) {
throw new Error('OPENAI_API_KEY not configured');
}
const messages: OpenAIMessage[] = [
{ role: 'system', content: systemPrompt },
...recentContext.map((m) => ({
role: m.role as 'user' | 'assistant',
content: m.content,
})),
{ role: 'user', content: userMessage },
];
// 첫 번째 호출
let response = await callOpenAI(env.OPENAI_API_KEY, messages);
let assistantMessage = response.choices[0].message;
// Function Calling 처리 (최대 3회 반복)
let iterations = 0;
while (assistantMessage.tool_calls && iterations < 3) {
iterations++;
// 도구 호출 결과 수집
const toolResults: OpenAIMessage[] = [];
for (const toolCall of assistantMessage.tool_calls) {
const args = JSON.parse(toolCall.function.arguments);
const result = await executeTool(toolCall.function.name, args);
toolResults.push({
role: 'tool',
tool_call_id: toolCall.id,
content: result,
});
}
// 대화에 추가
messages.push({
role: 'assistant',
content: assistantMessage.content,
tool_calls: assistantMessage.tool_calls,
});
messages.push(...toolResults);
// 다시 호출
response = await callOpenAI(env.OPENAI_API_KEY, messages, false);
assistantMessage = response.choices[0].message;
}
return assistantMessage.content || '응답을 생성할 수 없습니다.';
}
// 프로필 생성용 (도구 없이)
export async function generateProfileWithOpenAI(
env: Env,
prompt: string
): Promise<string> {
if (!env.OPENAI_API_KEY) {
throw new Error('OPENAI_API_KEY not configured');
}
const response = await callOpenAI(
env.OPENAI_API_KEY,
[{ role: 'user', content: prompt }],
false
);
return response.choices[0].message.content || '프로필 생성 실패';
}

159
src/security.ts Normal file
View File

@@ -0,0 +1,159 @@
import { Env, TelegramUpdate } from './types';
// Telegram 서버 IP 대역 (2024년 기준)
// https://core.telegram.org/bots/webhooks#the-short-version
const TELEGRAM_IP_RANGES = [
'149.154.160.0/20',
'91.108.4.0/22',
];
// CIDR 범위 체크 유틸
function ipInCIDR(ip: string, cidr: string): boolean {
const [range, bits] = cidr.split('/');
const mask = ~(2 ** (32 - parseInt(bits)) - 1);
const ipNum = ip.split('.').reduce((acc, octet) => (acc << 8) + parseInt(octet), 0);
const rangeNum = range.split('.').reduce((acc, octet) => (acc << 8) + parseInt(octet), 0);
return (ipNum & mask) === (rangeNum & mask);
}
// IP 화이트리스트 검증
function isValidTelegramIP(ip: string): boolean {
return TELEGRAM_IP_RANGES.some(range => ipInCIDR(ip, range));
}
// Webhook Secret Token 검증 (Timing-safe comparison)
function isValidSecretToken(request: Request, expectedSecret: string): boolean {
const secretHeader = request.headers.get('X-Telegram-Bot-Api-Secret-Token');
if (!secretHeader || !expectedSecret) {
return false;
}
// Timing-safe comparison
if (secretHeader.length !== expectedSecret.length) {
return false;
}
let result = 0;
for (let i = 0; i < secretHeader.length; i++) {
result |= secretHeader.charCodeAt(i) ^ expectedSecret.charCodeAt(i);
}
return result === 0;
}
// 요청 본문 검증
function isValidRequestBody(body: unknown): body is TelegramUpdate {
return (
body !== null &&
typeof body === 'object' &&
'update_id' in body &&
typeof (body as TelegramUpdate).update_id === 'number'
);
}
// 타임스탬프 검증 (리플레이 공격 방지)
function isRecentUpdate(message: TelegramUpdate['message']): boolean {
if (!message?.date) return true; // 메시지가 없으면 통과
const messageTime = message.date * 1000; // Unix timestamp to ms
const now = Date.now();
const maxAge = 60 * 1000; // 60초
return now - messageTime < maxAge;
}
export interface SecurityCheckResult {
valid: boolean;
error?: string;
update?: TelegramUpdate;
}
// 통합 보안 검증
export async function validateWebhookRequest(
request: Request,
env: Env
): Promise<SecurityCheckResult> {
// 1. HTTP 메서드 검증
if (request.method !== 'POST') {
return { valid: false, error: 'Method not allowed' };
}
// 2. Content-Type 검증
const contentType = request.headers.get('Content-Type');
if (!contentType?.includes('application/json')) {
return { valid: false, error: 'Invalid content type' };
}
// 3. Secret Token 검증 (필수)
if (env.WEBHOOK_SECRET) {
if (!isValidSecretToken(request, env.WEBHOOK_SECRET)) {
console.error('Invalid webhook secret token');
return { valid: false, error: 'Invalid secret token' };
}
} else {
console.warn('WEBHOOK_SECRET not configured - skipping token validation');
}
// 4. IP 화이트리스트 검증 (선택적 - CF에서는 CF-Connecting-IP 사용)
const clientIP = request.headers.get('CF-Connecting-IP');
if (clientIP && !isValidTelegramIP(clientIP)) {
// 경고만 로그 (Cloudflare 프록시 환경에서는 정확하지 않을 수 있음)
console.warn(`Request from non-Telegram IP: ${clientIP}`);
}
// 5. 요청 본문 파싱 및 검증
let body: unknown;
try {
body = await request.json();
} catch {
return { valid: false, error: 'Invalid JSON body' };
}
if (!isValidRequestBody(body)) {
return { valid: false, error: 'Invalid request body structure' };
}
// 6. 타임스탬프 검증 (리플레이 공격 방지)
if (!isRecentUpdate(body.message)) {
return { valid: false, error: 'Message too old' };
}
return { valid: true, update: body };
}
// Rate Limiting
const rateLimitMap = new Map<string, { count: number; resetAt: number }>();
export function checkRateLimit(
userId: string,
maxRequests: number = 30,
windowMs: number = 60000
): boolean {
const now = Date.now();
const userLimit = rateLimitMap.get(userId);
if (!userLimit || now > userLimit.resetAt) {
rateLimitMap.set(userId, { count: 1, resetAt: now + windowMs });
return true;
}
if (userLimit.count >= maxRequests) {
return false;
}
userLimit.count++;
return true;
}
// Rate limit 정리 (메모리 관리)
export function cleanupRateLimits(): void {
const now = Date.now();
for (const [key, value] of rateLimitMap.entries()) {
if (now > value.resetAt) {
rateLimitMap.delete(key);
}
}
}

277
src/summary-service.ts Normal file
View File

@@ -0,0 +1,277 @@
import { Env, BufferedMessage, Summary, ConversationContext } from './types';
// 설정값 가져오기
const getConfig = (env: Env) => ({
summaryThreshold: parseInt(env.SUMMARY_THRESHOLD || '20', 10),
maxSummaries: parseInt(env.MAX_SUMMARIES_PER_USER || '3', 10),
});
// 버퍼에 메시지 추가
export async function addToBuffer(
db: D1Database,
userId: number,
chatId: string,
role: 'user' | 'bot',
message: string
): Promise<number> {
await db
.prepare(`
INSERT INTO message_buffer (user_id, chat_id, role, message)
VALUES (?, ?, ?, ?)
`)
.bind(userId, chatId, role, message)
.run();
const count = await db
.prepare('SELECT COUNT(*) as cnt FROM message_buffer WHERE user_id = ? AND chat_id = ?')
.bind(userId, chatId)
.first<{ cnt: number }>();
return count?.cnt || 0;
}
// 버퍼 메시지 조회
export async function getBufferedMessages(
db: D1Database,
userId: number,
chatId: string
): Promise<BufferedMessage[]> {
const { results } = await db
.prepare(`
SELECT id, role, message, created_at
FROM message_buffer
WHERE user_id = ? AND chat_id = ?
ORDER BY created_at ASC
`)
.bind(userId, chatId)
.all();
return (results || []) as BufferedMessage[];
}
// 최신 요약 조회
export async function getLatestSummary(
db: D1Database,
userId: number,
chatId: string
): Promise<Summary | null> {
const summary = await db
.prepare(`
SELECT id, generation, summary, message_count, created_at
FROM summaries
WHERE user_id = ? AND chat_id = ?
ORDER BY generation DESC
LIMIT 1
`)
.bind(userId, chatId)
.first<Summary>();
return summary || null;
}
// 전체 컨텍스트 조회
export async function getConversationContext(
db: D1Database,
userId: number,
chatId: string
): Promise<ConversationContext> {
const [previousSummary, recentMessages] = await Promise.all([
getLatestSummary(db, userId, chatId),
getBufferedMessages(db, userId, chatId),
]);
const totalMessages = (previousSummary?.message_count || 0) + recentMessages.length;
return {
previousSummary,
recentMessages,
totalMessages,
};
}
// AI 요약 생성
async function generateSummary(
env: Env,
previousSummary: string | null,
messages: BufferedMessage[]
): Promise<string> {
// 사용자 메시지만 추출
const userMessages = messages
.filter((m) => m.role === 'user')
.map((m) => `- ${m.message}`)
.join('\n');
// 사용자 메시지 수
const userMsgCount = messages.filter((m) => m.role === 'user').length;
let prompt: string;
if (previousSummary) {
prompt = `당신은 사용자 프로필 분석 전문가입니다.
기존 사용자 프로필과 새로운 대화를 통합하여 사용자에 대한 이해를 업데이트하세요.
## 기존 사용자 프로필
${previousSummary}
## 새로운 사용자 발언 (${userMsgCount}개)
${userMessages}
## 요구사항
1. **사용자 중심**: 봇 응답은 무시하고 사용자가 말한 내용만 분석
2. **의미 있는 정보 추출**:
- 사용자의 관심사, 취미, 선호도
- 질문한 주제들 (무엇에 대해 알고 싶어하는지)
- 요청사항, 목표, 해결하려는 문제
- 개인적 맥락 (직업, 상황, 배경 등)
- 감정 상태나 태도 변화
3. **무의미한 내용 제외**: 인사말, 단순 확인, 감사 표현 등은 생략
4. **간결하게**: 300-400자 이내
5. **한국어로 작성**
업데이트된 사용자 프로필:`;
} else {
prompt = `당신은 사용자 프로필 분석 전문가입니다.
대화 내용에서 사용자에 대한 정보를 추출하여 프로필을 작성하세요.
## 사용자 발언 (${userMsgCount}개)
${userMessages}
## 요구사항
1. **사용자 중심**: 봇 응답은 무시하고 사용자가 말한 내용만 분석
2. **의미 있는 정보 추출**:
- 사용자의 관심사, 취미, 선호도
- 질문한 주제들 (무엇에 대해 알고 싶어하는지)
- 요청사항, 목표, 해결하려는 문제
- 개인적 맥락 (직업, 상황, 배경 등)
3. **무의미한 내용 제외**: 인사말, 단순 확인, 감사 표현 등은 생략
4. **간결하게**: 200-300자 이내
5. **한국어로 작성**
사용자 프로필:`;
}
// OpenAI 사용 (설정된 경우)
if (env.OPENAI_API_KEY) {
const { generateProfileWithOpenAI } = await import('./openai-service');
return generateProfileWithOpenAI(env, prompt);
}
// 폴백: Workers AI
const response = await env.AI.run('@cf/meta/llama-3.1-8b-instruct', {
messages: [{ role: 'user', content: prompt }],
max_tokens: 500,
});
return response.response || '프로필 생성 실패';
}
// 오래된 요약 정리
async function cleanupOldSummaries(
db: D1Database,
userId: number,
chatId: string,
maxSummaries: number
): Promise<void> {
await db
.prepare(`
DELETE FROM summaries
WHERE user_id = ? AND chat_id = ?
AND id NOT IN (
SELECT id FROM summaries
WHERE user_id = ? AND chat_id = ?
ORDER BY generation DESC
LIMIT ?
)
`)
.bind(userId, chatId, userId, chatId, maxSummaries)
.run();
}
// 요약 실행 및 저장
export async function processAndSummarize(
env: Env,
userId: number,
chatId: string
): Promise<{ summarized: boolean; summary?: string }> {
const config = getConfig(env);
const messages = await getBufferedMessages(env.DB, userId, chatId);
if (messages.length < config.summaryThreshold) {
return { summarized: false };
}
const previousSummary = await getLatestSummary(env.DB, userId, chatId);
// AI 요약 생성
const newSummary = await generateSummary(
env,
previousSummary?.summary || null,
messages
);
const newGeneration = (previousSummary?.generation || 0) + 1;
const newMessageCount = (previousSummary?.message_count || 0) + messages.length;
// 트랜잭션 실행
await env.DB.batch([
env.DB
.prepare(`
INSERT INTO summaries (user_id, chat_id, generation, summary, message_count)
VALUES (?, ?, ?, ?, ?)
`)
.bind(userId, chatId, newGeneration, newSummary, newMessageCount),
env.DB
.prepare('DELETE FROM message_buffer WHERE user_id = ? AND chat_id = ?')
.bind(userId, chatId),
]);
// 오래된 요약 정리
await cleanupOldSummaries(env.DB, userId, chatId, config.maxSummaries);
return { summarized: true, summary: newSummary };
}
// AI 응답 생성 (컨텍스트 포함)
export async function generateAIResponse(
env: Env,
userId: number,
chatId: string,
userMessage: string
): Promise<string> {
const context = await getConversationContext(env.DB, userId, chatId);
const systemPrompt = `당신은 친절하고 유능한 AI 어시스턴트입니다.
${context.previousSummary ? `
## 사용자 프로필
${context.previousSummary.summary}
위 프로필을 바탕으로 사용자의 관심사와 맥락을 이해하고 개인화된 응답을 제공하세요.
` : ''}
- 날씨, 시간, 계산, 검색 등의 요청은 제공된 도구를 사용하세요.
- 응답은 간결하고 도움이 되도록 한국어로 작성하세요.`;
const recentContext = context.recentMessages.slice(-10).map((m) => ({
role: m.role === 'user' ? 'user' as const : 'assistant' as const,
content: m.message,
}));
// OpenAI 사용 (설정된 경우)
if (env.OPENAI_API_KEY) {
const { generateOpenAIResponse } = await import('./openai-service');
return generateOpenAIResponse(env, userMessage, systemPrompt, recentContext);
}
// 폴백: Workers AI
const response = await env.AI.run('@cf/meta/llama-3.1-8b-instruct', {
messages: [
{ role: 'system', content: systemPrompt },
...recentContext,
{ role: 'user', content: userMessage },
],
max_tokens: 500,
});
return response.response || '응답을 생성할 수 없습니다.';
}

107
src/telegram.ts Normal file
View File

@@ -0,0 +1,107 @@
// Telegram API 메시지 전송
export async function sendMessage(
token: string,
chatId: number,
text: string,
options?: {
parse_mode?: 'HTML' | 'Markdown' | 'MarkdownV2';
reply_to_message_id?: number;
disable_notification?: boolean;
}
): Promise<boolean> {
try {
const response = await fetch(
`https://api.telegram.org/bot${token}/sendMessage`,
{
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({
chat_id: chatId,
text,
parse_mode: options?.parse_mode || 'HTML',
reply_to_message_id: options?.reply_to_message_id,
disable_notification: options?.disable_notification,
}),
}
);
if (!response.ok) {
const error = await response.text();
console.error('Telegram API error:', error);
return false;
}
return true;
} catch (error) {
console.error('Failed to send message:', error);
return false;
}
}
// Webhook 설정 (Secret Token 포함)
export async function setWebhook(
token: string,
webhookUrl: string,
secretToken: string
): Promise<{ ok: boolean; description?: string }> {
const response = await fetch(
`https://api.telegram.org/bot${token}/setWebhook`,
{
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({
url: webhookUrl,
secret_token: secretToken,
allowed_updates: ['message'],
drop_pending_updates: true,
}),
}
);
return response.json();
}
// Webhook 정보 조회
export async function getWebhookInfo(
token: string
): Promise<unknown> {
const response = await fetch(
`https://api.telegram.org/bot${token}/getWebhookInfo`
);
return response.json();
}
// Webhook 삭제
export async function deleteWebhook(
token: string
): Promise<{ ok: boolean }> {
const response = await fetch(
`https://api.telegram.org/bot${token}/deleteWebhook`,
{ method: 'POST' }
);
return response.json();
}
// 타이핑 액션 전송
export async function sendChatAction(
token: string,
chatId: number,
action: 'typing' | 'upload_photo' | 'upload_document' = 'typing'
): Promise<boolean> {
try {
const response = await fetch(
`https://api.telegram.org/bot${token}/sendChatAction`,
{
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({
chat_id: chatId,
action,
}),
}
);
return response.ok;
} catch {
return false;
}
}

68
src/types.ts Normal file
View File

@@ -0,0 +1,68 @@
export interface Env {
DB: D1Database;
AI: Ai;
BOT_TOKEN: string;
WEBHOOK_SECRET: string;
SUMMARY_THRESHOLD?: string;
MAX_SUMMARIES_PER_USER?: string;
N8N_WEBHOOK_URL?: string;
OPENAI_API_KEY?: string;
}
export interface IntentAnalysis {
action: 'chat' | 'n8n';
type?: string;
confidence: number;
}
export interface N8nResponse {
reply?: string;
error?: string;
}
export interface TelegramUpdate {
update_id: number;
message?: TelegramMessage;
}
export interface TelegramMessage {
message_id: number;
from: TelegramUser;
chat: TelegramChat;
date: number;
text?: string;
}
export interface TelegramUser {
id: number;
is_bot: boolean;
first_name: string;
last_name?: string;
username?: string;
}
export interface TelegramChat {
id: number;
type: string;
}
export interface BufferedMessage {
id: number;
role: 'user' | 'bot';
message: string;
created_at: string;
}
export interface Summary {
id: number;
generation: number;
summary: string;
message_count: number;
created_at: string;
}
export interface ConversationContext {
previousSummary: Summary | null;
recentMessages: BufferedMessage[];
totalMessages: number;
}

22
tsconfig.json Normal file
View File

@@ -0,0 +1,22 @@
{
"compilerOptions": {
"target": "ES2021",
"module": "ESNext",
"moduleResolution": "Bundler",
"lib": ["ES2021"],
"types": ["@cloudflare/workers-types"],
"strict": true,
"noEmit": true,
"skipLibCheck": true,
"allowSyntheticDefaultImports": true,
"forceConsistentCasingInFileNames": true,
"resolveJsonModule": true,
"isolatedModules": true,
"noUnusedLocals": true,
"noUnusedParameters": true,
"noImplicitReturns": true,
"noFallthroughCasesInSwitch": true
},
"include": ["src/**/*"],
"exclude": ["node_modules"]
}

20
wrangler.toml Normal file
View File

@@ -0,0 +1,20 @@
name = "telegram-summary-bot"
main = "src/index.ts"
compatibility_date = "2024-01-01"
[ai]
binding = "AI"
[vars]
SUMMARY_THRESHOLD = "20"
MAX_SUMMARIES_PER_USER = "3"
N8N_WEBHOOK_URL = "https://n8n.anvil.it.com"
[[d1_databases]]
binding = "DB"
database_name = "telegram-conversations"
database_id = "c285bb5b-888b-405d-b36f-475ae5aed20e"
# Secrets (wrangler secret put 으로 설정):
# - BOT_TOKEN: Telegram Bot Token
# - WEBHOOK_SECRET: Webhook 검증용 시크릿