extract.js

const fs = require("fs");
const path = require("path");
const Database = require("better-sqlite3");

const dbFile = process.argv[2];
const outFile = process.argv[3];

if (!dbFile || !outFile) {
  console.error("❌ Invalid arguments passed to extract.js");
  console.error("  dbFile:", dbFile);
  console.error("  outFile:", outFile);
  console.error("❌ Usage: node extract.js path/to/chat.db path/to/output.csv");
  process.exit(1);
}

const resolvedDbFile = path.resolve(dbFile);
const resolvedOutFile = path.resolve(outFile);

console.log("🧪 Running extract.js");
console.log("📂 dbFile:", resolvedDbFile);
console.log("📤 outFile:", resolvedOutFile);

if (!fs.existsSync(resolvedDbFile)) {
  console.error("❌ Missing or invalid chat.db file.");
  process.exit(1);
}

try {
  const db = new Database(resolvedDbFile);
  const tables = db.prepare("SELECT name FROM sqlite_master WHERE type='table';").all();
  console.log("🧬 Tables in DB:", tables.map(t => t.name).join(", "));

  const stmt = db.prepare(`
    SELECT
      datetime(message.date / 1000000000 + strftime('%s','2001-01-01'), 'unixepoch', 'localtime') AS message_date,
      handle.id AS sender_or_recipient,
      message.is_from_me,
      message.text,
      chat.rowid AS chat_id
    FROM
      message
    LEFT JOIN
      handle ON message.handle_id = handle.rowid
    LEFT JOIN
      chat_message_join ON message.rowid = chat_message_join.message_id
    LEFT JOIN
      chat ON chat_message_join.chat_id = chat.rowid
    ORDER BY
      message.date ASC;
  `);

  const rows = stmt.all();
  console.log("📊 Rows returned:", rows.length);
  if (rows.length > 0) {
    console.log("🔍 First row:", rows[0]);
  }

  const csv = [
    "message_date,sender_or_recipient,is_from_me,text,chat_id",
    ...rows.map((row, index) => {
      const message_date = row.message_date || "";
      const sender = row.sender_or_recipient || "Unknown";
      const is_from_me = row.is_from_me != null ? row.is_from_me : "";
      const text = typeof row.text === "string" ? row.text : JSON.stringify(row.text || "");
      const chat_id = row.chat_id || "";

      try {
        return `"${message_date}","${sender}","${is_from_me}","${text.replace(/\n/g, ' ').replace(/"/g, '""')}","${chat_id}"`;
      } catch (err) {
        console.error(`❌ Failed to process row ${index}:`, row);
        throw err;
      }
    })
  ].join("\n");

  console.log("📁 Attempting to write CSV to:", resolvedOutFile);
  const resolvedDir = path.dirname(resolvedOutFile);
  if (!fs.existsSync(resolvedDir)) {
    console.warn("📁 Output directory does not exist. Creating:", resolvedDir);
    fs.mkdirSync(resolvedDir, { recursive: true });
  }

  // Debug log for the resolved output file path
  console.log("🧠 Writing to this exact path:", resolvedOutFile);
  try {
    fs.writeFileSync(resolvedOutFile, csv, "utf-8");
    console.log(`✅ CSV successfully written to: ${resolvedOutFile}`);
  } catch (writeErr) {
    console.error("❌ WRITE FAILURE:");
    console.error("💣 Failed Path:", resolvedOutFile);
    console.error("🛑 Resolved Directory Exists:", fs.existsSync(resolvedDir));
    console.error("📂 Directory Path:", resolvedDir);
    console.error("💥 Error Message:", writeErr.stack || writeErr.message || writeErr);
    process.exit(1);
  }
  const preview = csv.split('\n').slice(0, 5).join('\n');
  console.log("📝 Preview of CSV output:\n", preview);
  console.log(`✅ Export complete: ${resolvedOutFile}`);
  console.log("✅ CSV generation logic executed without error.");
} catch (err) {
  console.error("❌ UNCAUGHT ERROR:");
  console.error(err.stack || err.message || err);
  process.exit(1);
}
// Allow direct execution for dev testing
if (require.main === module) {
  const [,, dbPath, outputPath] = process.argv;
  if (!dbPath || !outputPath) {
    console.error("❌ Usage: node extract.js <path/to/chat.db> <path/to/output.csv>");
    process.exit(1);
  }

  // You can test this line in dev directly:
  console.log(`✅ Dev Test - Running with DB: ${dbPath} to OUTPUT: ${outputPath}`);
}