viruagent-cli 0.6.0 → 0.6.1

package/src/providers/tistory/index.js

@@ -15,7 +15,7 @@ const { enrichContentWithUploadedImages, resolveMandatoryThumbnail } = require('
 const { createWithProviderSession } = require('./session');
 const { createAskForAuthentication } = require('./auth');
 
-const createTistoryProvider = ({ sessionPath }) => {
+const createTistoryProvider = ({ sessionPath, account }) => {
   const tistoryApi = createTistoryApiClient({ sessionPath });
 
   const pending2faResult = (mode = 'kakao') => ({
@@ -33,7 +33,7 @@ const createTistoryProvider = ({ sessionPath }) => {
     pending2faResult,
   });
 
-  const withProviderSession = createWithProviderSession(askForAuthentication);
+  const withProviderSession = createWithProviderSession(askForAuthentication, account);
 
   return {
     id: 'tistory',
@@ -49,7 +49,7 @@ const createTistoryProvider = ({ sessionPath }) => {
             blogName,
             blogUrl: `https://${blogName}.tistory.com`,
             sessionPath,
-            metadata: getProviderMeta('tistory') || {},
+            metadata: getProviderMeta('tistory', account) || {},
           };
         } catch (error) {
           return {
@@ -57,7 +57,7 @@ const createTistoryProvider = ({ sessionPath }) => {
             loggedIn: false,
             sessionPath,
             error: error.message,
-            metadata: getProviderMeta('tistory') || {},
+            metadata: getProviderMeta('tistory', account) || {},
           };
         }
       });
@@ -89,7 +89,7 @@ const createTistoryProvider = ({ sessionPath }) => {
         blogName: result.blogName,
         blogUrl: result.blogUrl,
         sessionPath: result.sessionPath,
-      });
+      }, account);
       return result;
     },
 
@@ -529,7 +529,7 @@ const createTistoryProvider = ({ sessionPath }) => {
     },
 
     async logout() {
-      clearProviderMeta('tistory');
+      clearProviderMeta('tistory', account);
       return {
         provider: 'tistory',
         loggedOut: true,

package/src/providers/tistory/session.js

@@ -3,19 +3,27 @@ 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');
-  });
+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 +41,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 +63,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 +91,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 +103,7 @@ const createWithProviderSession = (askForAuthentication) => async (fn) => {
         loggedIn: false,
         lastError: buildLoginErrorMessage(reloginError),
         lastValidatedAt: new Date().toISOString(),
-      });
+      }, account);
       throw reloginError;
     }
   }

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);
 };