export-cursor-chat.py

#!/usr/bin/env python3
import sqlite3
import json
import datetime
from pathlib import Path
import sys


def get_chat_data_from_db(db_path) -> list[dict]:
    """Extract chat data from the SQLite database"""
    conn = None
    try:
        # Connect to the SQLite database
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()

        cursor.execute("""SELECT value FROM cursorDiskKV""")

        results = cursor.fetchall()

        return results

    except sqlite3.Error as e:
        print(f"SQLite error occurred while processing {db_path}: {e}")
        return []

    finally:
        if conn:
            conn.close()


def ts_to_str(ts, readable=False):
    ts = ts / 1000  # Convert milliseconds to seconds
    if readable:
        timestamp = datetime.datetime.fromtimestamp(ts).strftime("%Y-%m-%d %H:%M:%S")
    else:
        timestamp = datetime.datetime.fromtimestamp(ts).strftime("%Y%m%d-%H%M%S")
    return timestamp


def handle_chat_data(chat_data, output_dir):
    for chat in chat_data:
        chat = json.loads(chat)
        name = chat.get("name", "")
        ts = ts_to_str(chat["createdAt"])
        md_filename = f"{output_dir}/{ts}.md"
        json_filename = f"{output_dir}/{ts}.json"
        with open(md_filename, "w", encoding="utf-8") as f:
            f.write(f"# {name}\n")
            for block in chat["conversation"]:
                f.write("\n\n---\n")

                if "context" in block:
                    for fileSelection in block["context"].get("fileSelections", []):
                        if "fsPath" in fileSelection["uri"]:
                            f.write(f"{fileSelection['uri']['fsPath']}\n")
                    for selection in block["context"].get("selections", []):
                        if "fsPath" in selection["uri"]:
                            f.write(f"{selection['uri']['fsPath']}\n")
                        f.write(f"\n{selection['text']}\n")

                f.write(block["text"])

        with open(json_filename, "w", encoding="utf-8") as f:
            json.dump(chat, f, indent=2, ensure_ascii=False)

        print(f"Processed {ts} {name}")


def main():
    db_file_path = Path.home() / "Library/Application Support/Cursor/User/globalStorage/state.vscdb"

    if len(sys.argv) > 1:
        output_dir = Path(sys.argv[1])
    else:
        output_dir = Path("/tmp/cursor")

    if not output_dir.exists():
        output_dir.mkdir(parents=True, exist_ok=True)

    chats = get_chat_data_from_db(db_file_path)
    for chat in chats:
        handle_chat_data(chat, output_dir)


if __name__ == "__main__":
    print("Usage: python export-cursor-chat.py [output_directory]")
    print("  If no output directory is specified, /tmp/cursor will be used")
    main()

Traceback (most recent call last):
  File "/Applications/export-cursor-chat.py", line 89, in <module>
    main()
  File "/Applications/export-cursor-chat.py", line 83, in main
    handle_chat_data(chat, output_dir)
  File "/Applications/export-cursor-chat.py", line 44, in handle_chat_data
    name = chat.get("name", "")
AttributeError: 'list' object has no attribute 'get'