viruagent 1.0.0

package/src/cli-post.js

@@ -0,0 +1,93 @@
+#!/usr/bin/env node
+
+const path = require('path');
+require('dotenv').config({ path: path.join(__dirname, '..', '.env') });
+
+const { generatePost, loadConfig } = require('./lib/ai');
+const { initBlog, publishPost, saveDraft, getCategories, VISIBILITY } = require('./lib/tistory');
+
+const parseArgs = (argv) => {
+  const args = {};
+  for (let i = 0; i < argv.length; i++) {
+    const arg = argv[i];
+    if (arg === '--dry-run') { args.dryRun = true; continue; }
+    if (arg === '--draft') { args.draft = true; continue; }
+    if (arg === '--list-categories') { args.listCategories = true; continue; }
+    if (arg.startsWith('--') && i + 1 < argv.length) {
+      args[arg.slice(2)] = argv[++i];
+    }
+  }
+  return args;
+};
+
+const visibilityMap = {
+  public: VISIBILITY.PUBLIC,
+  private: VISIBILITY.PRIVATE,
+  protected: VISIBILITY.PROTECTED,
+};
+
+const output = (obj) => {
+  console.log(JSON.stringify(obj));
+};
+
+const main = async () => {
+  const args = parseArgs(process.argv.slice(2));
+
+  await initBlog();
+
+  // 카테고리 목록 조회
+  if (args.listCategories) {
+    const cats = await getCategories();
+    output({ success: true, categories: Object.entries(cats).map(([name, id]) => ({ name, id })) });
+    return;
+  }
+
+  // topic 필수
+  if (!args.topic) {
+    output({ success: false, error: '--topic은 필수입니다.' });
+    process.exit(1);
+  }
+
+  const config = loadConfig();
+
+  // 글 생성
+  const post = await generatePost(args.topic, {
+    model: args.model || config.defaultModel,
+    tone: args.tone || config.defaultTone,
+  });
+
+  // dry-run: 생성만
+  if (args.dryRun) {
+    output({ success: true, title: post.title, tags: post.tags, preview: post.content.slice(0, 200) });
+    return;
+  }
+
+  const visibility = visibilityMap[args.visibility] ?? VISIBILITY.PUBLIC;
+  const category = Number(args.category) || 0;
+
+  // 임시저장
+  if (args.draft) {
+    const result = await saveDraft({ title: post.title, content: post.content });
+    output({ success: true, mode: 'draft', title: post.title, tags: post.tags, sequence: result.draft?.sequence });
+    return;
+  }
+
+  // 발행
+  const result = await publishPost({
+    title: post.title,
+    content: post.content,
+    visibility,
+    category,
+    tag: post.tags,
+    thumbnail: post.thumbnailKage || null,
+  });
+
+  const url = result.entryUrl || null;
+
+  output({ success: true, mode: 'publish', title: post.title, tags: post.tags, url });
+};
+
+main().catch((e) => {
+  output({ success: false, error: e.message });
+  process.exit(1);
+});

package/src/lib/ai.js

@@ -0,0 +1,192 @@
+const OpenAI = require('openai');
+const fs = require('fs');
+const path = require('path');
+require('dotenv').config({ path: path.join(__dirname, '..', '..', '.env') });
+
+if (!process.env.OPENAI_API_KEY) {
+  console.error('\x1b[31m✗ OPENAI_API_KEY가 설정되지 않았습니다.\x1b[0m');
+  console.error('\x1b[33m  1. 프로젝트 루트에 .env 파일을 생성하세요');
+  console.error('  2. OPENAI_API_KEY=sk-... 형식으로 키를 입력하세요');
+  console.error('  3. https://platform.openai.com/api-keys 에서 키를 발급받을 수 있습니다\x1b[0m');
+  process.exit(1);
+}
+
+const client = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
+
+const { replaceImagePlaceholders } = require('./unsplash');
+const { createLogger } = require('./logger');
+const aiLog = createLogger('ai');
+
+const handleApiError = (e) => {
+  if (e?.status === 401 || e?.code === 'invalid_api_key') {
+    throw new Error('OpenAI API 키가 유효하지 않습니다. .env 파일의 OPENAI_API_KEY를 확인하세요.');
+  }
+  if (e?.status === 429) {
+    throw new Error('API 요청 한도를 초과했습니다. 잠시 후 다시 시도하거나 요금제를 확인하세요.');
+  }
+  if (e?.status === 403) {
+    throw new Error('API 키에 해당 모델 접근 권한이 없습니다. OpenAI 대시보드에서 권한을 확인하세요.');
+  }
+  throw e;
+};
+
+const CONFIG_PATH = path.join(__dirname, '..', '..', 'config', 'prompt-config.json');
+const SYSTEM_PROMPT_PATH = path.join(__dirname, '..', '..', 'config', 'system-prompt.md');
+
+const loadConfig = () => {
+  const config = JSON.parse(fs.readFileSync(CONFIG_PATH, 'utf-8'));
+  config.systemPrompt = fs.readFileSync(SYSTEM_PROMPT_PATH, 'utf-8');
+  return config;
+};
+
+const MODELS = [
+  // GPT-4o
+  'gpt-4o-mini', 'gpt-4o',
+  // GPT-4.1
+  'gpt-4.1-nano', 'gpt-4.1-mini', 'gpt-4.1',
+  // GPT-5
+  'gpt-5-nano', 'gpt-5-mini', 'gpt-5',
+  // GPT-5.1
+  'gpt-5.1', 'gpt-5.1-codex-mini', 'gpt-5.1-codex',
+  // GPT-5.2
+  'gpt-5.2', 'gpt-5.2-pro', 'gpt-5.2-codex',
+  // Reasoning
+  'o3-mini', 'o3', 'o3-pro', 'o4-mini',
+];
+
+// reasoning 모델은 temperature를 지원하지 않음
+const REASONING_MODELS = /^(o[1-9]|o\d+-)/;
+const isReasoningModel = (model) => REASONING_MODELS.test(model);
+
+/**
+ * 블로그 글 초안 생성
+ * @param {string} topic - 주제
+ * @param {Object} [options]
+ * @param {string} [options.tone] - 말투
+ * @param {number} [options.length] - 대략적인 글자 수
+ * @param {string} [options.model] - 모델명
+ * @returns {Promise<{title: string, content: string, tags: string}>}
+ */
+const generatePost = async (topic, options = {}) => {
+  const config = loadConfig();
+  const {
+    tone = config.defaultTone,
+    length = config.defaultLength,
+    model = config.defaultModel,
+  } = options;
+
+  let res;
+  try {
+    res = await client.chat.completions.create({
+      model,
+      messages: [
+        { role: 'system', content: config.systemPrompt },
+        {
+          role: 'user',
+          content: `주제: ${topic}
+말투: ${tone}
+분량: 약 ${length}자
+
+작성 요구사항:
+- 제목은 검색 키워드를 포함하면서 클릭을 유도하는 형태로 작성 (숫자 활용 권장)
+- 주제에 가장 적합한 글 유형과 구조를 자율적으로 선택하여 작성
+- 본문 중 적절한 위치에 <!-- IMAGE: 영문키워드 --> 플레이스홀더를 3개 내외 삽입
+- 태그는 검색 유입에 효과적인 키워드 5~7개
+
+다음 JSON 형식으로 응답하세요:
+{"title": "글 제목", "content": "<p>HTML 본문...</p>", "tags": "태그1,태그2,태그3,태그4,태그5"}`,
+        },
+      ],
+      response_format: { type: 'json_object' },
+      ...(!isReasoningModel(model) && { temperature: 0.7 }),
+    });
+  } catch (e) {
+    handleApiError(e);
+  }
+
+  const result = JSON.parse(res.choices[0].message.content);
+  aiLog.info('GPT 응답 수신', { title: result.title, contentLength: result.content?.length });
+
+  // 티스토리 업로드 함수가 있으면 전달 (initBlog 완료 상태에서만 동작)
+  let uploadFn;
+  try {
+    const { uploadImage } = require('./tistory');
+    uploadFn = uploadImage;
+  } catch (e) {
+    aiLog.warn('tistory uploadImage 로드 실패', { error: e.message });
+  }
+  const imageResult = await replaceImagePlaceholders(result.content, { uploadFn });
+  result.content = imageResult.html;
+  result.thumbnailUrl = imageResult.thumbnailUrl;
+  result.thumbnailKage = imageResult.thumbnailKage;
+  aiLog.info('이미지 처리 완료', { thumbnailKage: result.thumbnailKage });
+
+  return result;
+};
+
+/**
+ * 글 수정
+ * @param {string} content - 현재 HTML 본문
+ * @param {string} instruction - 수정 지시
+ * @returns {Promise<{title: string, content: string, tags: string}>}
+ */
+const revisePost = async (content, instruction, model) => {
+  const config = loadConfig();
+  model = model || config.defaultModel;
+
+  let res;
+  try {
+    res = await client.chat.completions.create({
+      model,
+      messages: [
+        { role: 'system', content: config.systemPrompt },
+        {
+          role: 'user',
+          content: `다음 글을 수정해주세요.
+
+수정 지시: ${instruction}
+
+현재 본문:
+${content}
+
+다음 JSON 형식으로 응답하세요:
+{"title": "수정된 제목", "content": "<p>수정된 HTML 본문...</p>", "tags": "태그1,태그2,태그3,태그4,태그5"}`,
+        },
+      ],
+      response_format: { type: 'json_object' },
+      ...(!isReasoningModel(model) && { temperature: 0.7 }),
+    });
+  } catch (e) {
+    handleApiError(e);
+  }
+
+  return JSON.parse(res.choices[0].message.content);
+};
+
+/**
+ * 자유 대화
+ * @param {Array<{role: string, content: string}>} messages
+ * @returns {Promise<string>}
+ */
+const chat = async (messages, model) => {
+  const config = loadConfig();
+  model = model || config.defaultModel;
+
+  let res;
+  try {
+    res = await client.chat.completions.create({
+      model,
+      messages: [
+        { role: 'system', content: '당신은 블로그 글쓰기를 돕는 AI 어시스턴트입니다. 주제 논의, 아이디어 브레인스토밍, 글 구조 제안 등을 도와줍니다. 한국어로 대화하세요.' },
+        ...messages,
+      ],
+      ...(!isReasoningModel(model) && { temperature: 0.7 }),
+    });
+  } catch (e) {
+    handleApiError(e);
+  }
+
+  return res.choices[0].message.content;
+};
+
+module.exports = { generatePost, revisePost, chat, MODELS, loadConfig };

package/src/lib/logger.js

@@ -0,0 +1,27 @@
+const fs = require('fs');
+const path = require('path');
+
+const LOGS_DIR = path.join(__dirname, '..', '..', 'logs');
+
+const getLogFile = () => {
+  const date = new Date().toISOString().slice(0, 10);
+  return path.join(LOGS_DIR, `${date}.log`);
+};
+
+const formatTime = () => new Date().toISOString().slice(11, 23);
+
+const write = (level, module, message, data) => {
+  const line = `[${formatTime()}] [${level}] [${module}] ${message}${data !== undefined ? ' | ' + JSON.stringify(data) : ''}\n`;
+  try {
+    fs.appendFileSync(getLogFile(), line);
+  } catch {}
+};
+
+const createLogger = (module) => ({
+  info: (msg, data) => write('INFO', module, msg, data),
+  warn: (msg, data) => write('WARN', module, msg, data),
+  error: (msg, data) => write('ERROR', module, msg, data),
+  debug: (msg, data) => write('DEBUG', module, msg, data),
+});
+
+module.exports = { createLogger };

package/src/lib/login.js

@@ -0,0 +1,25 @@
+const { chromium } = require('playwright');
+const readline = require('readline');
+const path = require('path');
+
+(async () => {
+  const browser = await chromium.launch({ headless: false });
+  const context = await browser.newContext();
+  const page = await context.newPage();
+
+  await page.goto('https://www.tistory.com/auth/login');
+
+  console.log('\n========================================');
+  console.log('브라우저에서 로그인을 완료하세요.');
+  console.log('로그인 후 여기서 Enter를 누르면 세션이 저장됩니다.');
+  console.log('========================================\n');
+
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  await new Promise((resolve) => rl.question('', resolve));
+  rl.close();
+
+  await context.storageState({ path: path.join(__dirname, '..', '..', 'data', 'session.json') });
+  console.log('세션이 session.json에 저장되었습니다!');
+
+  await browser.close();
+})();

package/src/lib/tistory.js

@@ -0,0 +1,243 @@
+const fs = require('fs');
+const path = require('path');
+const { createLogger } = require('./logger');
+const log = createLogger('tistory');
+
+let blogName = null;
+let blogInfo = null;
+
+const loadCookies = () => {
+  const session = JSON.parse(fs.readFileSync(path.join(__dirname, '..', '..', 'data', 'session.json'), 'utf-8'));
+  return session.cookies
+    .filter(c => c.domain.includes('tistory'))
+    .map(c => `${c.name}=${c.value}`)
+    .join('; ');
+};
+
+const getBase = () => {
+  if (!blogName) throw new Error('블로그 이름이 초기화되지 않았습니다. initBlog()를 먼저 호출하세요.');
+  return `https://${blogName}.tistory.com/manage`;
+};
+
+const getHeaders = () => ({
+  'Cookie': loadCookies(),
+  'Content-Type': 'application/json;charset=UTF-8',
+  'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36',
+  'Referer': `${getBase()}/newpost`,
+  'X-Requested-With': 'XMLHttpRequest',
+});
+
+/**
+ * 로그인 세션에서 블로그 정보를 자동 감지
+ * @returns {Promise<string>} 블로그 이름
+ */
+const initBlog = async () => {
+  if (blogName) return blogName;
+
+  const res = await fetch('https://www.tistory.com/legacy/member/blog/api/myBlogs', {
+    headers: {
+      'Cookie': loadCookies(),
+      'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36',
+    },
+  });
+
+  if (!res.ok) throw new Error(`블로그 정보 조회 실패: ${res.status}`);
+
+  const contentType = res.headers.get('content-type') || '';
+  if (!contentType.includes('application/json')) {
+    throw new Error('세션이 만료되었습니다. /login으로 다시 로그인하세요.');
+  }
+
+  const json = await res.json();
+
+  const defaultBlog = json.data?.find(b => b.defaultBlog) || json.data?.[0];
+  if (!defaultBlog) throw new Error('블로그를 찾을 수 없습니다.');
+
+  blogName = defaultBlog.name;
+  blogInfo = defaultBlog;
+  return blogName;
+};
+
+const getBlogName = () => blogName;
+const getBlogInfo = () => blogInfo;
+
+// visibility: 0=비공개, 15=보호, 20=공개
+const VISIBILITY = { PRIVATE: 0, PROTECTED: 15, PUBLIC: 20 };
+
+/**
+ * 글 발행
+ * @param {Object} options
+ * @param {string} options.title - 제목
+ * @param {string} options.content - HTML 본문
+ * @param {number} [options.visibility=20] - 0: 비공개, 15: 보호, 20: 공개
+ * @param {number} [options.category=0] - 카테고리 ID
+ * @param {string} [options.tag=''] - 태그 (쉼표 구분)
+ * @param {string} [options.thumbnail=null] - 썸네일 kage@ 경로
+ * @returns {Promise<Object>}
+ */
+const publishPost = async ({ title, content, visibility = VISIBILITY.PUBLIC, category = 0, tag = '', thumbnail = null }) => {
+  const body = {
+    id: '0',
+    title,
+    content,
+    visibility,
+    category,
+    tag,
+    published: 1,
+    type: 'post',
+    uselessMarginForEntry: 1,
+    cclCommercial: 0,
+    cclDerive: 0,
+    attachments: [],
+    recaptchaValue: '',
+    draftSequence: null,
+  };
+  if (thumbnail) body.thumbnail = thumbnail;
+
+  const res = await fetch(`${getBase()}/post.json`, {
+    method: 'POST',
+    headers: getHeaders(),
+    body: JSON.stringify(body),
+  });
+
+  if (!res.ok) throw new Error(`발행 실패: ${res.status}`);
+  const data = await res.json();
+  return data;
+};
+
+/**
+ * 임시저장
+ */
+const saveDraft = async ({ title, content }) => {
+  const res = await fetch(`${getBase()}/drafts`, {
+    method: 'POST',
+    headers: getHeaders(),
+    body: JSON.stringify({ title, content }),
+  });
+
+  if (!res.ok) throw new Error(`임시저장 실패: ${res.status}`);
+  const data = await res.json();
+  console.log(`임시저장 완료 (sequence: ${data.draft.sequence})`);
+  return data;
+};
+
+/**
+ * 글 목록 조회
+ */
+const getPosts = async () => {
+  const res = await fetch(`${getBase()}/posts.json`, { headers: getHeaders() });
+  if (!res.ok) throw new Error(`목록 조회 실패: ${res.status}`);
+  return res.json();
+};
+
+/**
+ * 카테고리 목록을 글쓰기 페이지에서 동적으로 가져옴
+ * @returns {Promise<Record<string, number>>} { "카테고리명": id }
+ */
+const getCategories = async () => {
+  const res = await fetch(`${getBase()}/newpost`, { headers: getHeaders() });
+  if (!res.ok) throw new Error(`카테고리 조회 실패: ${res.status}`);
+  const html = await res.text();
+
+  // window.Config의 blog.categories를 vm으로 안전하게 추출
+  const vm = require('vm');
+  const configMatch = html.match(/window\.Config\s*=\s*(\{[\s\S]*?\})\s*\n/);
+  if (!configMatch) throw new Error('카테고리 파싱 실패');
+
+  const sandbox = {};
+  vm.runInNewContext(`var result = ${configMatch[1]}`, sandbox);
+  const catList = sandbox.result.blog.categories;
+  const categories = {};
+
+  const flatten = (list) => {
+    for (const cat of list) {
+      categories[cat.label] = cat.id;
+      if (cat.children?.length) flatten(cat.children);
+    }
+  };
+  flatten(catList);
+
+  return categories;
+};
+
+/**
+ * 이미지 업로드
+ * @param {Buffer} imageBuffer - 이미지 데이터
+ * @param {string} [filename='image.jpg'] - 파일명
+ * @returns {Promise<string>} 티스토리 이미지 치환자 문자열
+ */
+const uploadImage = async (imageBuffer, filename = 'image.jpg') => {
+  if (!blogName) await initBlog();
+
+  const formData = new FormData();
+  const blob = new Blob([imageBuffer], { type: 'image/jpeg' });
+  formData.append('file', blob, filename);
+
+  const cookies = loadCookies();
+  const res = await fetch(`${getBase()}/post/attach.json`, {
+    method: 'POST',
+    headers: {
+      'Cookie': cookies,
+      'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36',
+      'Referer': `${getBase()}/newpost`,
+      'Accept': 'application/json, text/plain, */*',
+    },
+    body: formData,
+  });
+
+  if (!res.ok) {
+    const text = await res.text().catch(() => '');
+    log.error('이미지 업로드 실패', { status: res.status, body: text.substring(0, 500) });
+    throw new Error(`이미지 업로드 실패: ${res.status}`);
+  }
+  const data = await res.json();
+  log.info('이미지 업로드 성공', { url: data.url, key: data.key, filename: data.filename });
+
+  if (!data.url) throw new Error('이미지 업로드 응답에 URL이 없습니다.');
+  return data;
+};
+
+module.exports = { initBlog, getBlogName, getBlogInfo, publishPost, saveDraft, getPosts, getCategories, uploadImage, VISIBILITY };
+
+// CLI 모드
+if (require.main === module) {
+  const args = process.argv.slice(2);
+  const cmd = args[0];
+
+  initBlog().then(() => {
+
+  if (cmd === 'publish') {
+    publishPost({
+      title: args[1] || '테스트 글',
+      content: args[2] || '<p>자동 작성된 글입니다.</p>',
+      visibility: Number(args[3]) || 0,
+      category: Number(args[4]) || 0,
+      tag: args[5] || '',
+    });
+  } else if (cmd === 'draft') {
+    saveDraft({
+      title: args[1] || '임시저장 테스트',
+      content: args[2] || '<p>임시저장 내용</p>',
+    });
+  } else if (cmd === 'list') {
+    getPosts().then(d => {
+      console.log(`총 ${d.totalCount}개 글`);
+      d.items?.forEach(p => console.log(`  [${p.id}] ${p.title} (${p.visibility})`));
+    });
+  } else if (cmd === 'categories') {
+    getCategories().then(cats => {
+      console.log('카테고리 목록:');
+      Object.entries(cats).forEach(([name, id]) => console.log(`  ${id} → ${name}`));
+    });
+  } else {
+    console.log(`사용법:
+  node tistory.js publish "제목" "<p>내용</p>" [visibility] [categoryId] [tags]
+  node tistory.js draft "제목" "<p>내용</p>"
+  node tistory.js list
+  node tistory.js categories
+
+visibility: 0=비공개, 3=공개`);
+  }
+
+  }).catch(e => console.error(`초기화 실패: ${e.message}`));
+}