Merge pull request #6 from rich-iannone/feat-db-sql-mode

feat: Add DB/SQL mode

Author: rich-iannone

sweet/cli.py

@@ -10,7 +10,10 @@
 @click.command()
 @click.version_option()
 @click.option("--file", "-f", type=click.Path(exists=True), help="Load data file on startup")
-def main(file: str | None):
+@click.option(
+    "--db", type=str, help="Connect to remote database (e.g., mysql://user:pass@host:port/db)"
+)
+def main(file: str | None, db: str | None):
     """Sweet - Interactive data engineering CLI utility."""
     try:
         # Check if data is being piped from stdin
@@ -51,6 +54,9 @@ def main(file: str | None):
                     Path(file).unlink()
                 except OSError:
                     pass
+        elif db:
+            click.echo(f"Starting Sweet with database: {db}")
+            run_app(startup_db=db)
         else:
             click.echo("Starting Sweet...")
             run_app()

sweet/ui/app.py

@@ -48,10 +48,11 @@ class SweetApp(App):
         Binding("f1", "show_command_reference", "Command Reference", show=True),
     ]
 
-    def __init__(self, startup_file: str | None = None, **kwargs):
-        """Initialize the app with optional startup file."""
+    def __init__(self, startup_file: str | None = None, startup_db: str | None = None, **kwargs):
+        """Initialize the app with optional startup file or database connection."""
         super().__init__(**kwargs)
         self.startup_file = startup_file
+        self.startup_db = startup_db
         self.command_mode = False
         self.current_filename = None
         self._update_title()
@@ -84,6 +85,11 @@ def on_mount(self) -> None:
             self._data_grid.load_file(self.startup_file)
             # Set the current filename and update title
             self.set_current_filename(self.startup_file)
+        elif self.startup_db:
+            # If a database connection string was provided, connect to it
+            self._data_grid.connect_to_database(self.startup_db)
+            # Set the connection info and update title
+            self.set_current_filename(f"Database: {self.startup_db}")
 
     def _update_title(self) -> None:
         """Update the window title with current filename."""
@@ -399,7 +405,7 @@ def action_quit(self) -> None:
         self.exit()
 
 
-def run_app(startup_file: str | None = None) -> None:
+def run_app(startup_file: str | None = None, startup_db: str | None = None) -> None:
     """Run the Sweet application."""
-    app = SweetApp(startup_file=startup_file)
+    app = SweetApp(startup_file=startup_file, startup_db=startup_db)
     app.run()

sweet/ui/sweet.tcss

@@ -693,3 +693,8 @@ ToolsPanel #llm-transform-content {
 ToolsPanel .panel-section {
     height: 1fr;
 }
+
+/* Green border for approval-ready SQL code */
+.approval-ready {
+    border: solid green;
+}