feature: default paths in config, updated/simplified readme

Author: somogyijanos

.gitignore

@@ -1,3 +1,4 @@
 .venv
 **/__pycache__
-output
+output
+**/.DS_Store

README.md

@@ -8,8 +8,8 @@ Also see [this](https://forum.cursor.com/t/guide-5-steps-exporting-chats-prompts
 
 ## Features
 
-- **Discover Chats**: Discover all `state.vscdb` files in a directory and print a few lines of dialogue so one can identify which is the workspace (chat) one is searching for. It's also possible to filter by text.
-- **Export Chats**: Export chats data for a certain workspace to Markdown files or print it to the command line.
+- **Discover Chats**: Discover all chats from all workspaces and print a few lines of dialogue so one can identify which is the workspace (or chat) one is searching for. It's also possible to filter by text.
+- **Export Chats**: Export chats for the most recent (or a specific) workspace to Markdown files or print it to the command line.
 
 ## Installation
 
@@ -26,49 +26,44 @@ Also see [this](https://forum.cursor.com/t/guide-5-steps-exporting-chats-prompts
 
 ## Usage
 
-Find, where the `state.vscdb` is located in your computer. The table below may help:
+First, find, where the `state.vscdb` files are located on your computer. Confirm that corresponding to your system, the right path is set in the [config.yml](./config.yml) file. Update it if not set correctly.
 
-| OS               | Path                                                      |
-|------------------|-----------------------------------------------------------|
-| Windows          | `%APPDATA%\Cursor\User\workspaceStorage`                  |
-| macOS            | `/Users/YOUR_USERNAME/Library/Application Support/Cursor/User/workspaceStorage` |
-| Linux            | `/home/YOUR_USERNAME/.config/Cursor/User/workspaceStorage` |
+Both the `discover` and `export` commands will work with this path by default, but you can also provide a custom path any time.
 
-### Discover Chats of all Workspaces
-```sh
-python3 chat.py discover --search-text "matplotlib"
-```
+---
 
-### Export all Chats of a specific Workspace
+### Discover Chats
 ```sh
-python3 chat.py export --output-dir output "/Users/myuser/Library/Application Support/Cursor/User/workspaceStorage/b989572f2e2186b48b808da2da437416/state.vscdb"
-```
+# Help on usage
+./chat.py discover --help
 
-If no directory is provided, the tool will automatically use the default Cursor workspace storage directory for your OS.
-
-### Export the latest Tab from a specific Workspace
-```sh
-python3 chat.py export --output-dir output --latest-tab "/Users/myuser/Library/Application Support/Cursor/User/workspaceStorage/b989572f2e2186b48b808da2da437416/state.vscdb"
-```
+# Discover all chats from all workspaces
+./chat.py discover
 
-### Auto-detect and Export Workspace Chats
+# Apply text filter
+./chat.py discover --search-text "matplotlib"
 
-Examples:
+# Discover all chats from all workspaces at a custom path
+./chat.py discover "/path/to/workspaces"
+```
 
+---
 
-#### Export all tabs from the most recent workspace
+### Export Chats
+See `./chat.py export --help` for general help. Examples:
 ```sh
-python3 chat.py export --output-dir output
-```
+# Help on usage
+./chat.py export --help
 
-If no database path is provided, the tool will automatically use the most recent workspace.
+# Print all chats of the most recent workspace to the command line
+./chat.py export
 
-#### Export the latest tab from the most recent workspace
+# Export all chats of the most recent workspace as Markdown
+./chat.py export --output-dir "/path/to/output"
 
-You can use the `--latest-tab` flag to export only the most recent tab:
+# Export only the latest chat of the most recent workspace
+./chat.py export --latest-tab --output-dir "/path/to/output"
 
-```sh
-python3 chat.py export --output-dir output --latest-tab
+# Export all chats of a specifc workspace
+./chat.py export --output-dir "/path/to/output" "/path/to/workspaces/workspace-dir/state.vscdb"
 ```
-
-These commands will automatically detect the appropriate directory for your OS when needed, find the latest workspace if no path is provided, and export the chats to the specified output directory according to the chosen options.

chat.py

@@ -9,6 +9,7 @@
 from rich.markdown import Markdown
 from loguru import logger
 import json
+import yaml
 import platform
 from pathlib import Path
 
@@ -91,21 +92,35 @@ def export(
         raise typer.Exit(code=1)
 
 def get_cursor_workspace_path() -> Path:
+    config_path = Path("config.yml")
+    logger.debug(f"Looking for configuration file at: {config_path}")
+    
+    if not config_path.exists():
+        error_message = f"Configuration file not found: {config_path}"
+        logger.error(error_message)
+        raise FileNotFoundError(error_message)
+
+    with open(config_path, 'r') as f:
+        config = yaml.safe_load(f)
+    logger.debug("Configuration file loaded successfully")
+
     system = platform.system()
-    home = Path.home()
+    logger.debug(f"Detected operating system: {system}")
 
-    if system == "Windows":
-        base_path = Path(os.environ.get("APPDATA")) / "Cursor" / "User" / "workspaceStorage"
-    elif system == "Darwin":  # macOS
-        base_path = home / "Library" / "Application Support" / "Cursor" / "User" / "workspaceStorage"
-    elif system == "Linux":
-        base_path = home / ".config" / "Cursor" / "User" / "workspaceStorage"
-    else:
-        raise ValueError(f"Unsupported operating system: {system}")
+    if system not in config["default_vscdb_dir_paths"]:
+        error_message = f"Unsupported operating system: {system}"
+        logger.error(error_message)
+        raise ValueError(error_message)
+
+    base_path = Path(os.path.expandvars(config["default_vscdb_dir_paths"][system])).expanduser()
+    logger.debug(f"Resolved base path: {base_path}")
 
     if not base_path.exists():
-        raise FileNotFoundError(f"Cursor workspace storage directory not found: {base_path}")
+        error_message = f"Cursor workspace storage directory not found: {base_path}"
+        logger.error(error_message)
+        raise FileNotFoundError(error_message)
 
+    logger.info(f"Cursor workspace storage directory found: {base_path}")
     return base_path
 
 def get_latest_workspace_db_path() -> str:

config.yml

@@ -1 +1,5 @@
+default_vscdb_dir_paths:
+  Windows: "%APPDATA%/Cursor/User/workspaceStorage"
+  Darwin: "~/Library/Application Support/Cursor/User/workspaceStorage"
+  Linux: "~/.config/Cursor/User/workspaceStorage"
 aichat_query: "SELECT value FROM ItemTable WHERE [key] IN ('workbench.panel.aichat.view.aichat.chatdata');"