viruagent-cli 0.6.0 → 0.6.2

package/src/providers/tistory/session.js

@@ -3,19 +3,67 @@ const path = require('path');
 const { saveProviderMeta } = require('../../storage/sessionStore');
 const { sleep, readCredentialsFromEnv, parseSessionError, buildLoginErrorMessage } = require('./utils');
 const { clickKakaoAccountContinue } = require('./browserHelpers');
+const { extractAllCookies, filterCookies, cookiesToSessionFormat } = require('../chromeManager');
 
-const isLoggedInByCookies = async (context) => {
-  const cookies = await context.cookies('https://www.tistory.com');
-  return cookies.some((cookie) => {
-    const name = cookie.name.toLowerCase();
-    return name.includes('tistory') || name.includes('access') || name.includes('login');
-  });
+// ── Rate Limit (일일 발행 제한) ──
+const DAILY_LIMIT = { publish: 15 };
+
+const readSessionFile = (sessionPath) => {
+  if (!fs.existsSync(sessionPath)) return null;
+  try { return JSON.parse(fs.readFileSync(sessionPath, 'utf-8')); } catch { return null; }
+};
+const writeSessionFile = (sessionPath, data) => {
+  fs.mkdirSync(path.dirname(sessionPath), { recursive: true });
+  fs.writeFileSync(sessionPath, JSON.stringify(data, null, 2), 'utf-8');
+};
+
+const checkAndIncrementRateLimit = (sessionPath, type) => {
+  const raw = readSessionFile(sessionPath) || {};
+  if (!raw.rateLimits) raw.rateLimits = {};
+  const c = raw.rateLimits[type] || { daily: 0, dayStart: Date.now() };
+
+  const now = Date.now();
+  if (now - c.dayStart > 86400000) { c.daily = 0; c.dayStart = now; }
+
+  const dailyMax = DAILY_LIMIT[type];
+  if (dailyMax && c.daily >= dailyMax) {
+    throw new Error(`daily_limit: ${type} exceeded daily limit of ${dailyMax}. Try again tomorrow.`);
+  }
+
+  c.daily++;
+  raw.rateLimits[type] = { ...c, savedAt: new Date().toISOString() };
+  writeSessionFile(sessionPath, raw);
+};
+
+const getRateLimitStatus = (sessionPath) => {
+  const raw = readSessionFile(sessionPath) || {};
+  const result = {};
+  for (const [type, max] of Object.entries(DAILY_LIMIT)) {
+    const c = raw.rateLimits?.[type] || { daily: 0 };
+    result[type] = { daily: `${c.daily}/${max}` };
+  }
+  return result;
+};
+
+const isLoggedInByCookies = async (context, page) => {
+  try {
+    // Use CDP to get all cookies including httpOnly (TSSESSION)
+    const all = await extractAllCookies(context, page);
+    return all.some((c) => c.domain.includes('tistory') && c.name === 'TSSESSION');
+  } catch {
+    // Fallback to context.cookies if CDP fails
+    const cookies = await context.cookies('https://www.tistory.com');
+    return cookies.some((cookie) => {
+      const name = cookie.name.toLowerCase();
+      return name.includes('tistory') || name.includes('access') || name.includes('login');
+    });
+  }
 };
 
 const waitForLoginFinish = async (page, context, timeoutMs = 45000) => {
   const deadline = Date.now() + timeoutMs;
   while (Date.now() < deadline) {
-    if (await isLoggedInByCookies(context)) {
+    if (await isLoggedInByCookies(context, page)) {
       return true;
     }
 
@@ -33,16 +81,11 @@ const waitForLoginFinish = async (page, context, timeoutMs = 45000) => {
   return false;
 };
 
-const persistTistorySession = async (context, targetSessionPath) => {
-  const cookies = await context.cookies('https://www.tistory.com');
-  const sanitized = cookies.map((cookie) => ({
-    ...cookie,
-    expires: Number(cookie.expires || -1),
-    size: undefined,
-    partitionKey: undefined,
-    sourcePort: undefined,
-    sourceScheme: undefined,
-  }));
+const persistTistorySession = async (context, page, targetSessionPath) => {
+  // Use CDP Network.getAllCookies to capture httpOnly cookies (e.g. TSSESSION)
+  const all = await extractAllCookies(context, page);
+  const tistoryCookies = filterCookies(all, ['tistory.com']);
+  const sanitized = cookiesToSessionFormat(tistoryCookies);
 
   const payload = {
     cookies: sanitized,
@@ -60,16 +103,13 @@ const persistTistorySession = async (context, targetSessionPath) => {
  * withProviderSession factory.
  * Receives askForAuthentication via dependency injection to avoid scope issues.
  */
-const createWithProviderSession = (askForAuthentication) => async (fn) => {
+const createWithProviderSession = (askForAuthentication, account) => async (fn) => {
   const credentials = readCredentialsFromEnv();
   const hasCredentials = Boolean(credentials.username && credentials.password);
 
   try {
     const result = await fn();
-    saveProviderMeta('tistory', {
-      loggedIn: true,
-      lastValidatedAt: new Date().toISOString(),
-    });
+    saveProviderMeta('tistory', { loggedIn: true, lastValidatedAt: new Date().toISOString() }, account);
     return result;
   } catch (error) {
     if (!parseSessionError(error) || !hasCredentials) {
@@ -91,7 +131,7 @@ const createWithProviderSession = (askForAuthentication) => async (fn) => {
         sessionPath: loginResult.sessionPath,
         lastRefreshedAt: new Date().toISOString(),
         lastError: null,
-      });
+      }, account);
 
       if (!loginResult.loggedIn) {
         throw new Error(loginResult.message || 'Login status could not be confirmed after session refresh.');
@@ -103,7 +143,7 @@ const createWithProviderSession = (askForAuthentication) => async (fn) => {
         loggedIn: false,
         lastError: buildLoginErrorMessage(reloginError),
         lastValidatedAt: new Date().toISOString(),
-      });
+      }, account);
       throw reloginError;
     }
   }
@@ -114,4 +154,6 @@ module.exports = {
   waitForLoginFinish,
   persistTistorySession,
   createWithProviderSession,
+  checkAndIncrementRateLimit,
+  getRateLimitStatus,
 };

package/src/runner.js

@@ -86,9 +86,10 @@ const runCommand = async (command, opts = {}) => {
   }
 
   const providerName = opts.provider || 'tistory';
+  const account = opts.account || undefined;
   let provider;
   try {
-    provider = manager.getProvider(providerName);
+    provider = manager.getProvider(providerName, account);
   } catch {
     throw createError(
       'PROVIDER_NOT_FOUND',

package/src/services/providerManager.js

@@ -15,23 +15,25 @@ const providers = ['tistory', 'naver', 'insta'];
 const createProviderManager = () => {
   const cache = new Map();
 
-  const getProvider = (provider = 'tistory') => {
+  const getProvider = (provider = 'tistory', account) => {
     const normalized = String(provider || 'tistory').toLowerCase();
     if (!providerFactory[normalized]) {
       throw new Error(`Unsupported provider: ${provider}. Available options: ${providers.join(', ')}`);
     }
 
-    if (!cache.has(normalized)) {
-      const sessionPath = getSessionPath(normalized);
+    const cacheKey = account ? `${normalized}:${String(account).toLowerCase()}` : normalized;
+    if (!cache.has(cacheKey)) {
+      const sessionPath = getSessionPath(normalized, account);
       const options = {
         provider: normalized,
         sessionPath,
+        account: account || undefined,
       };
       const providerInstance = providerFactory[normalized](options);
-      cache.set(normalized, providerInstance);
+      cache.set(cacheKey, providerInstance);
     }
 
-    return cache.get(normalized);
+    return cache.get(cacheKey);
   };
 
   const providerNames = { tistory: 'Tistory', naver: 'Naver Blog', insta: 'Instagram' };

package/src/storage/sessionStore.js

@@ -27,9 +27,11 @@ const writeJson = (target, data) => {
   fs.writeFileSync(target, JSON.stringify(data, null, 2), 'utf-8');
 };
 
-const getSessionPath = (provider) => {
+const getSessionPath = (provider, account) => {
   ensureDir(SESSION_DIR);
-  return path.join(SESSION_DIR, `${normalizeProvider(provider)}-session.json`);
+  const base = normalizeProvider(provider);
+  const suffix = account ? `-${String(account).toLowerCase().replace(/[^a-z0-9._-]/g, '_')}` : '';
+  return path.join(SESSION_DIR, `${base}${suffix}-session.json`);
 };
 
 const getProvidersMeta = () => {
@@ -37,25 +39,32 @@ const getProvidersMeta = () => {
   return readJson(META_FILE);
 };
 
-const saveProviderMeta = (provider, patch) => {
+const metaKey = (provider, account) => {
+  const base = normalizeProvider(provider);
+  return account ? `${base}:${String(account).toLowerCase()}` : base;
+};
+
+const saveProviderMeta = (provider, patch, account) => {
   const meta = getProvidersMeta();
-  meta[normalizeProvider(provider)] = {
-    ...(meta[normalizeProvider(provider)] || {}),
+  const key = metaKey(provider, account);
+  meta[key] = {
+    ...(meta[key] || {}),
     ...patch,
     provider: normalizeProvider(provider),
+    ...(account ? { account: String(account).toLowerCase() } : {}),
     updatedAt: new Date().toISOString(),
   };
   writeJson(META_FILE, meta);
 };
 
-const getProviderMeta = (provider) => {
+const getProviderMeta = (provider, account) => {
   const meta = getProvidersMeta();
-  return meta[normalizeProvider(provider)] || null;
+  return meta[metaKey(provider, account)] || null;
 };
 
-const clearProviderMeta = (provider) => {
+const clearProviderMeta = (provider, account) => {
   const meta = getProvidersMeta();
-  delete meta[normalizeProvider(provider)];
+  delete meta[metaKey(provider, account)];
   writeJson(META_FILE, meta);
 };