feat: add config defaults, Firefox cookies, and Sweetistics whoami

src/lib/cookies.ts

@@ -5,7 +5,7 @@
 
 import { execSync } from 'node:child_process';
 import { createDecipheriv, pbkdf2Sync } from 'node:crypto';
-import { copyFileSync, existsSync, mkdtempSync, rmSync } from 'node:fs';
+import { copyFileSync, existsSync, mkdtempSync, readdirSync, rmSync } from 'node:fs';
 import { tmpdir } from 'node:os';
 import { join } from 'node:path';
 
@@ -34,6 +34,26 @@ function getChromeCookiesPath(profile?: string): string {
   return join(home, 'Library', 'Application Support', 'Google', 'Chrome', profileDir, 'Cookies');
 }
 
+function getFirefoxCookiesPath(profile?: string): string | null {
+  const home = process.env.HOME || '';
+  const profilesRoot = join(home, 'Library', 'Application Support', 'Firefox', 'Profiles');
+  if (!existsSync(profilesRoot)) return null;
+
+  if (profile) {
+    const candidate = join(profilesRoot, profile, 'cookies.sqlite');
+    return existsSync(candidate) ? candidate : null;
+  }
+
+  // Pick the default-release profile if present, otherwise first profile dir containing cookies.sqlite
+  const entries = readdirSync(profilesRoot, { withFileTypes: true });
+  const defaultRelease = entries.find((entry) => entry.isDirectory() && entry.name.endsWith('.default-release'));
+  const targetDir = defaultRelease?.name ?? entries.find((e) => e.isDirectory())?.name;
+  if (!targetDir) return null;
+
+  const candidate = join(profilesRoot, targetDir, 'cookies.sqlite');
+  return existsSync(candidate) ? candidate : null;
+}
+
 /**
  * Decrypt Chrome cookie value using macOS keychain
  * Chrome encrypts cookies with a key stored in the keychain
@@ -178,6 +198,74 @@ export async function extractCookiesFromChrome(profile?: string): Promise<Cookie
   return { cookies, warnings };
 }
 
+/**
+ * Extract Twitter cookies from Firefox browser using sqlite3 CLI
+ * @param profile - Firefox profile directory name (optional, auto-detected)
+ */
+export async function extractCookiesFromFirefox(profile?: string): Promise<CookieExtractionResult> {
+  const warnings: string[] = [];
+  const cookies: TwitterCookies = {
+    authToken: null,
+    ct0: null,
+    source: null,
+  };
+
+  const cookiesPath = getFirefoxCookiesPath(profile);
+
+  if (!cookiesPath) {
+    warnings.push('Firefox cookies database not found.');
+    return { cookies, warnings };
+  }
+
+  let tempDir: string | null = null;
+
+  try {
+    tempDir = mkdtempSync(join(tmpdir(), 'twitter-cli-'));
+    const tempDbPath = join(tempDir, 'cookies.sqlite');
+    copyFileSync(cookiesPath, tempDbPath);
+
+    const query = `SELECT name, value FROM moz_cookies WHERE host IN ('.x.com', '.twitter.com', 'x.com', 'twitter.com') AND name IN ('auth_token', 'ct0');`;
+
+    const result = execSync(`sqlite3 -separator '|' "${tempDbPath}" "${query}"`, {
+      encoding: 'utf8',
+      maxBuffer: 1024 * 1024,
+    }).trim();
+
+    if (result) {
+      for (const line of result.split('\n')) {
+        const [name, value] = line.split('|');
+        if (!name || !value) continue;
+        if (name === 'auth_token' && !cookies.authToken) {
+          cookies.authToken = value;
+        } else if (name === 'ct0' && !cookies.ct0) {
+          cookies.ct0 = value;
+        }
+      }
+    }
+
+    if (cookies.authToken || cookies.ct0) {
+      cookies.source = profile ? `Firefox profile "${profile}"` : 'Firefox default profile';
+    }
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    warnings.push(`Failed to read Firefox cookies: ${message}`);
+  } finally {
+    if (tempDir) {
+      try {
+        rmSync(tempDir, { recursive: true, force: true });
+      } catch {
+        // ignore cleanup errors
+      }
+    }
+  }
+
+  if (!cookies.authToken && !cookies.ct0) {
+    warnings.push('No Twitter cookies found in Firefox. Make sure you are logged into x.com in Firefox.');
+  }
+
+  return { cookies, warnings };
+}
+
 /**
  * Resolve Twitter credentials from multiple sources
  * Priority: CLI args > environment variables > Chrome cookies
@@ -186,6 +274,7 @@ export async function resolveCredentials(options: {
   authToken?: string;
   ct0?: string;
   chromeProfile?: string;
+  firefoxProfile?: string;
 }): Promise<CookieExtractionResult> {
   const warnings: string[] = [];
   const cookies: TwitterCookies = {
@@ -245,12 +334,27 @@ export async function resolveCredentials(options: {
     }
   }
 
+  // 4. Firefox cookies (fallback if still missing)
+  if (!cookies.authToken || !cookies.ct0) {
+    const firefoxResult = await extractCookiesFromFirefox(options.firefoxProfile);
+    warnings.push(...firefoxResult.warnings);
+
+    if (!cookies.authToken && firefoxResult.cookies.authToken) {
+      cookies.authToken = firefoxResult.cookies.authToken;
+      cookies.source = firefoxResult.cookies.source;
+    }
+    if (!cookies.ct0 && firefoxResult.cookies.ct0) {
+      cookies.ct0 = firefoxResult.cookies.ct0;
+      if (!cookies.source) cookies.source = firefoxResult.cookies.source;
+    }
+  }
+
   // Validation
   if (!cookies.authToken) {
-    warnings.push('Missing auth_token - provide via --auth-token, AUTH_TOKEN env var, or login to x.com in Chrome');
+    warnings.push('Missing auth_token - provide via --auth-token, AUTH_TOKEN env var, or login to x.com in Chrome/Firefox');
   }
   if (!cookies.ct0) {
-    warnings.push('Missing ct0 - provide via --ct0, CT0 env var, or login to x.com in Chrome');
+    warnings.push('Missing ct0 - provide via --ct0, CT0 env var, or login to x.com in Chrome/Firefox');
   }
 
   return { cookies, warnings };