gibson-cli 0.8.9__py3-none-any.whl → 0.8.10__py3-none-any.whl

gibson/api/ProjectApi.py

@@ -1,3 +1,5 @@
+from typing import List
+
 from gibson.api.BaseApi import BaseApi
 from gibson.core.Configuration import Configuration
 
@@ -20,8 +22,13 @@ class ProjectApi(BaseApi):
     def lookup(self, uuid: str):
         return self.get(f"{uuid}")
 
-    def deploy(self, uuid: str):
-        return self.post(f"{uuid}/deploy")
+    def deploy(self, uuid: str, environments: List[str] | None = None):
+        return self.post(
+            f"{uuid}/deploy", {"environments": environments} if environments else None
+        )
+
+    def diff(self, uuid: str):
+        return self.get(f"{uuid}/diff")
 
     def mcp(self, uuid: str):
         return self.get(f"{uuid}/mcp")

gibson/command/Deploy.py

@@ -0,0 +1,52 @@
+from rich.prompt import Confirm
+
+from gibson.api.ProjectApi import ProjectApi
+from gibson.command.BaseCommand import BaseCommand
+from gibson.core.Select import MultiSelect
+from gibson.core.Spinner import Spinner
+
+
+class Deploy(BaseCommand):
+    def execute(self):
+        self.configuration.require_login()
+        project_id = self.configuration.require_project_id()
+        project = ProjectApi(self.configuration).lookup(project_id)
+        databases = [database["environment"] for database in project["databases"]]
+
+        selected = MultiSelect(
+            title="Select the database(s) to deploy",
+            options=databases,
+        ).prompt()
+
+        if not selected:
+            self.conversation.type("No database selected\n")
+            exit(1)
+
+        for database in selected:
+            diff = ProjectApi(self.configuration).diff(project_id)
+            if diff:
+                self.conversation.type(f"\nDiff for {database} database:\n\n")
+                self.conversation.type(diff)
+                self.conversation.newline()
+
+                if Confirm.ask(
+                    f"After reviewing the diff, are you sure you want to deploy the {database} database?"
+                ):
+                    self.conversation.newline()
+                    with Spinner(
+                        start_text=f"Deploying {database} database...",
+                        success_text=f"Deployed {database} database",
+                        fail_text=f"Deployment failed for {database} database",
+                    ):
+                        ProjectApi(self.configuration).deploy(
+                            project_id, environments=[database]
+                        )
+
+                else:
+                    self.conversation.type(
+                        f"\nSkipping deployment for {database} database\n",
+                    )
+            else:
+                self.conversation.type(
+                    f"\nNo changes to deploy for {database} database\n",
+                )

gibson/command/Help.py

@@ -47,6 +47,11 @@ class Help(BaseCommand):
                 "subcommands": ["last", "stored"],
                 "memory": "based on user selection",
             },
+            "deploy": {
+                "description": "deploy the project database(s) with the current schema",
+                "subcommands": None,
+                "memory": None,
+            },
             "dev": {
                 "description": Text.assemble(
                     "gibson will automatically write code for you",
@@ -111,6 +116,11 @@ class Help(BaseCommand):
                 "subcommands": None,
                 "memory": "last > stored",
             },
+            "studio": {
+                "description": "connect to your database and launch the SQL studio",
+                "subcommands": None,
+                "memory": None,
+            },
             "tree": {
                 "description": "illustrate the project layout in a tree view",
                 "subcommands": None,

gibson/command/Studio.py

@@ -0,0 +1,58 @@
+from urllib.parse import urlparse
+
+from harlequin.app import Harlequin
+from harlequin.plugins import load_adapter_plugins
+
+from gibson.api.ProjectApi import ProjectApi
+from gibson.command.BaseCommand import BaseCommand
+from gibson.core.Select import Select
+from gibson.core.Spinner import DisappearingSpinner
+
+
+class Studio(BaseCommand):
+    def execute(self):
+        self.configuration.require_login()
+        project_id = self.configuration.require_project_id()
+
+        project = ProjectApi(self.configuration).lookup(project_id)
+
+        choices = [database["environment"] for database in project["databases"]]
+
+        selected = Select(
+            title="Select a database",
+            options=choices,
+        ).prompt()
+
+        if not selected:
+            self.conversation.type("No database selected\n")
+            exit(1)
+
+        database = next(
+            (db for db in project["databases"] if db["environment"] == selected), None
+        )
+
+        with DisappearingSpinner(
+            start_text=f"Connecting to {database['environment']} database...",
+            success_text=f"Connected to {database['environment']} database",
+            fail_text=f"Failed to connect to {database['environment']} database",
+        ):
+            plugins = load_adapter_plugins()
+            adapter = plugins[database["datastore_type"]]
+
+            if database["datastore_type"] == "mysql":
+                connection = urlparse(database["connection_string"])
+                adapter = adapter(
+                    conn_str=None,
+                    host=connection.hostname,
+                    port=connection.port,
+                    user=connection.username,
+                    password=connection.password,
+                    database=connection.path.lstrip("/"),
+                )
+            else:
+                raise ValueError(
+                    f"Unsupported database type: {database['datastore_type']}"
+                )
+
+            app = Harlequin(adapter=adapter)
+            app.run()

gibson/core/CommandRouter.py

@@ -5,6 +5,7 @@ from gibson.command.Build import Build
 from gibson.command.code.Code import Code
 from gibson.command.Conf import Conf
 from gibson.command.Count import Count
+from gibson.command.Deploy import Deploy
 from gibson.command.Dev import Dev
 from gibson.command.Forget import Forget
 from gibson.command.Help import Help
@@ -19,6 +20,7 @@ from gibson.command.Remove import Remove
 from gibson.command.rename.Rename import Rename
 from gibson.command.rewrite.Rewrite import Rewrite
 from gibson.command.Show import Show
+from gibson.command.Studio import Studio
 from gibson.command.Tree import Tree
 from gibson.command.Version import Version
 from gibson.core.Configuration import Configuration
@@ -52,6 +54,8 @@ class CommandRouter:
             command = Conf(self.configuration)
         elif sys.argv[1] == "count":
             command = Count(self.configuration)
+        elif sys.argv[1] == "deploy":
+            command = Deploy(self.configuration)
         elif sys.argv[1] == "dev":
             command = Dev(self.configuration)
         elif sys.argv[1] == "forget":
@@ -78,6 +82,8 @@ class CommandRouter:
             command = Rewrite(self.configuration, with_header=True)
         elif sys.argv[1] == "show":
             command = Show(self.configuration)
+        elif sys.argv[1] == "studio":
+            command = Studio(self.configuration)
         elif sys.argv[1] == "tree":
             command = Tree(self.configuration)
         elif sys.argv[1] in ["q"]:

gibson/core/Configuration.py

@@ -237,6 +237,10 @@ class Configuration:
         return self.settings[self.project.name]
 
     def get_project_id(self):
+        project_id = os.getenv("GIBSONAI_PROJECT_ID")
+        if project_id:
+            return project_id
+
         self.require_project()
         return self.project.id
 

gibson/core/Select.py

@@ -0,0 +1,270 @@
+from typing import Any, List, Optional
+
+import click
+from rich.align import Align
+from rich.console import Group
+from rich.live import Live
+from rich.panel import Panel
+from rich.text import Text
+
+
+class BaseSelect:
+    """Base class for select components with shared functionality."""
+
+    def __init__(
+        self,
+        options: List[str] = [],
+        title: str = "",
+        color: str = "bold magenta",
+        align: str = "left",
+        selection: str = "→",
+        highlight: str = "green",
+    ):
+        self.options = options
+        self.index = 0
+        self.title = title
+        self.color = color
+        self.align = align
+        self.selection = selection
+        self.highlight = highlight
+
+    def _get_click(self) -> Optional[str]:
+        match click.getchar():
+            case "\r":
+                return "enter"
+            case "\x1b[B" | "s" | "S" | "àP":
+                return "down"
+            case "\x1b[A" | "w" | "W" | "àH":
+                return "up"
+            case "\x1b[D" | "a" | "A" | "àK":
+                return "left"
+            case "\x1b[C" | "d" | "D" | "àM":
+                return "right"
+            case "\x1b":
+                return "exit"
+            case _:
+                return None
+
+    def _set_index(self, key: str) -> None:
+        if key == "down":
+            self.index += 1
+        elif key == "up":
+            self.index -= 1
+
+        if self.index > len(self.options) - 1:
+            self.index = 0
+        elif self.index < 0:
+            self.index = len(self.options) - 1
+
+    def _clear(self) -> None:
+        for _ in range(len(self.options) + 5):
+            print("\x1b[A\x1b[K", end="")
+
+    @property
+    def _usage_info(self) -> Text:
+        """Return usage information text for the select menu."""
+        return Text("Use ↑/↓ to navigate, ENTER to submit", "dim")
+
+    @property
+    def _layout(self) -> Group:
+        """Generate the display group for the menu.
+        This method should be overridden by subclasses.
+        """
+        raise NotImplementedError("Subclasses must implement _layout")
+
+    @property
+    def _panel_width(self) -> int:
+        """Return the width of the panel."""
+        # Calculate minimum width needed for the usage text
+        usage_width = len(self._usage_info.plain) + 2
+        # Get the max width of options
+        options_width = (
+            max((len(option) for option in self.options), default=0) + 5
+        )  # +5 for marker and spacing
+        # Use the larger of the two widths, plus some padding
+        return max(usage_width, options_width) + 4
+
+    def prompt(self) -> Any:
+        """Display the menu and handle user input.
+        This method should be overridden by subclasses.
+        """
+        raise NotImplementedError("Subclasses must implement prompt")
+
+
+class Select(BaseSelect):
+    def __init__(
+        self,
+        options: List[str] = [],
+        start_index: int = 0,
+        title: str = "",
+        color: str = "bold magenta",
+        align: str = "left",
+        selection: str = "→",
+        highlight: str = "green",
+    ):
+        super().__init__(
+            options=options,
+            title=title,
+            color=color,
+            align=align,
+            selection=selection,
+            highlight=highlight,
+        )
+        self.index = start_index
+
+    @property
+    def _layout(self) -> Group:
+        menu = Text(justify="left")
+
+        selected = Text(self.selection + " ", self.highlight)
+        not_selected = Text(" " * (len(self.selection) + 1))
+
+        for idx, option in enumerate(self.options):
+            if idx == self.index:
+                menu.append(
+                    Text.assemble(selected, Text(option + "\n", self.highlight))
+                )
+            else:
+                menu.append(Text.assemble(not_selected, option + "\n"))
+
+        menu.rstrip()
+
+        menu = Panel(menu, padding=1, width=self._panel_width)
+        menu.title = Text(self.title, self.color)
+        menu.subtitle = self._usage_info
+
+        return Group(Align(menu, self.align))
+
+    def prompt(self) -> str:
+        with Live(self._layout, auto_refresh=False, screen=False) as live:
+            live.update(self._layout, refresh=True)
+            while True:
+                try:
+                    key = self._get_click()
+                    if key == "enter":
+                        break
+                    elif key == "exit":
+                        exit()
+
+                    self._set_index(key)
+                    live.update(self._layout, refresh=True)
+                except (KeyboardInterrupt, EOFError):
+                    exit()
+
+        self._clear()
+
+        return self.options[self.index]
+
+
+class MultiSelect(BaseSelect):
+    def __init__(
+        self,
+        options: List[str] = [],
+        start_indices: Optional[List[int]] = None,
+        title: str = "",
+        color: str = "bold magenta",
+        align: str = "left",
+        selection: str = "→",
+        highlight: str = "green",
+        selected_marker: str = "●",
+        unselected_marker: str = "○",
+    ):
+        super().__init__(
+            options=options,
+            title=title,
+            color=color,
+            align=align,
+            selection=selection,
+            highlight=highlight,
+        )
+        self.selected_indices = start_indices or []
+        self.selected_marker = selected_marker
+        self.unselected_marker = unselected_marker
+
+    def _get_click(self) -> Optional[str]:
+        # Override to add space key handling
+        match click.getchar():
+            case "\r":
+                return "enter"
+            case "\x1b[B" | "s" | "S" | "àP":
+                return "down"
+            case "\x1b[A" | "w" | "W" | "àH":
+                return "up"
+            case "\x1b[D" | "a" | "A" | "àK":
+                return "left"
+            case "\x1b[C" | "d" | "D" | "àM":
+                return "right"
+            case " ":
+                return "space"
+            case "\x1b":
+                return "exit"
+            case _:
+                return None
+
+    def _toggle_selection(self) -> None:
+        if self.index in self.selected_indices:
+            self.selected_indices.remove(self.index)
+        else:
+            self.selected_indices.append(self.index)
+
+    @property
+    def _usage_info(self) -> Text:
+        """Return usage information text for the multi-select menu."""
+        return Text("Use ↑/↓ to navigate, SPACE to toggle, ENTER to submit", "dim")
+
+    @property
+    def _layout(self) -> Group:
+        menu = Text(justify="left")
+
+        cursor_selected = Text(self.selection + " ", self.highlight)
+        cursor_not_selected = Text(" " * (len(self.selection) + 1))
+
+        for idx, option in enumerate(self.options):
+            # Determine if this option is selected
+            marker = (
+                self.selected_marker
+                if idx in self.selected_indices
+                else self.unselected_marker
+            )
+
+            # Determine if cursor is on this item
+            if idx == self.index:
+                menu.append(
+                    Text.assemble(
+                        cursor_selected, Text(f"{marker} {option}\n", self.highlight)
+                    )
+                )
+            else:
+                menu.append(Text.assemble(cursor_not_selected, f"{marker} {option}\n"))
+
+        menu.rstrip()
+
+        menu = Panel(menu, padding=1, width=self._panel_width)
+        menu.title = Text(self.title, self.color)
+        menu.subtitle = self._usage_info
+
+        return Group(Align(menu, self.align))
+
+    def prompt(self) -> List[str]:
+        with Live(self._layout, auto_refresh=False, screen=False) as live:
+            live.update(self._layout, refresh=True)
+            while True:
+                try:
+                    key = self._get_click()
+                    if key == "enter":
+                        break
+                    elif key == "exit":
+                        exit()
+                    elif key == "space":
+                        self._toggle_selection()
+                        live.update(self._layout, refresh=True)
+                    else:
+                        self._set_index(key)
+                        live.update(self._layout, refresh=True)
+                except (KeyboardInterrupt, EOFError):
+                    exit()
+
+        self._clear()
+
+        # Return the selected options
+        return [self.options[i] for i in self.selected_indices]

{gibson_cli-0.8.9.dist-info → gibson_cli-0.8.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gibson-cli
-Version: 0.8.9
+Version: 0.8.10
 Summary: Gibson Command Line Interface
 Author-email: GibsonAI <noc@gibsonai.com>
 Project-URL: Homepage, https://gibsonai.com/
@@ -27,6 +27,8 @@ Requires-Dist: faker-sqlalchemy==0.10.2208140
 Requires-Dist: fastapi==0.115.12
 Requires-Dist: fastapi-cli==0.0.2
 Requires-Dist: h11==0.14.0
+Requires-Dist: harlequin==2.1.2
+Requires-Dist: harlequin-mysql==1.1.0
 Requires-Dist: httpcore==1.0.5
 Requires-Dist: httptools==0.6.4
 Requires-Dist: httpx==0.27.0
@@ -60,7 +62,7 @@ Requires-Dist: sniffio==1.3.1
 Requires-Dist: sqlalchemy==1.4.41
 Requires-Dist: sse-starlette==2.2.1
 Requires-Dist: starlette==0.46.1
-Requires-Dist: textual==0.83.0
+Requires-Dist: textual==0.89.1
 Requires-Dist: tomli==2.0.1
 Requires-Dist: typer==0.12.3
 Requires-Dist: typing-extensions==4.13.0

{gibson_cli-0.8.9.dist-info → gibson_cli-0.8.10.dist-info}/RECORD

@@ -4,20 +4,22 @@ bin/release.sh,sha256=6pnY_GctbwF-3sZB67b0nzUlTsxlhY1hqJQoqRutXuw,137
 gibson/api/BaseApi.py,sha256=Wza4yuLHhFjkPxt2dhtUtM_r9b5IGd72V80VDl7v3Gw,3510
 gibson/api/Cli.py,sha256=Qcm5NIQ4x1Wn6KfkrAzwvZeWyt-cKF_xD7_lTWL4Lbw,8071
 gibson/api/DataApi.py,sha256=JZN6q7bO2O87xBCsU9w96NgINOJJZx3d7jOiHoVH1Wo,542
-gibson/api/ProjectApi.py,sha256=1_gRAAgaqblV0I0D-R-wwgSQArQ4fSCxTDfTO3zDQzY,1212
+gibson/api/ProjectApi.py,sha256=IY3njspfanpaAIGnkrraQvqFmgQde7kzdg0ZNA2N1Vs,1428
 gibson/bin/gibson.py,sha256=ybtdrqLfmTb-gG2U7kWSHWSZDvc864omYR9vres38ZU,458
 gibson/command/BaseCommand.py,sha256=0LdxshtNj8sySMSaDFIPooDbdZx0mNMyhCuyqcF4JsU,769
 gibson/command/Build.py,sha256=nFwYhO5lpq7cM6oSH6nBA2ubW8wW7aJFsXIwjE1F2EY,4382
 gibson/command/Conf.py,sha256=yuAGL6M8MUURG4hW3MAW043c-h_ALw3FHWbyCOR8YTQ,2375
 gibson/command/Count.py,sha256=QOagwCwDxUHYPivFntr-RWUmlKWKDHgDIGZBfju6VpY,1040
+gibson/command/Deploy.py,sha256=-SqM3w7feTJmYNrZ70EKzuwlNBSZgaB_d_MQaR8zRwc,2006
 gibson/command/Dev.py,sha256=zbZjsgAKAHlNUflEr_Lj2QmRBXNbIGIHrPJ3t5DcFC8,4213
 gibson/command/Forget.py,sha256=rpUmnHISvQlMgsFwa0p2nndaygy9UvDph3mlOmwlugo,1016
-gibson/command/Help.py,sha256=Gih3JVZ3SLXrdZPInZccb-NtlzvSUTbTwW2NztKnrbo,6456
+gibson/command/Help.py,sha256=DVce2KCeNTjhxJ3v6nHub9vVJ-Vi0FNFFcA9dU0KToQ,6846
 gibson/command/Merge.py,sha256=R5ybMC1tUR5_T8YyUfXutzFa_V9j1_flv0s7KTJRq0M,1061
 gibson/command/Modify.py,sha256=XgTM6EodL7sVhWrnWuTCDx2PNJw3xo-aKI80HCmSpV8,1171
 gibson/command/Question.py,sha256=g8SwopbzeG14WWP0bc-fXIDVqOOicMzjC9YXoGd-NxY,3830
 gibson/command/Remove.py,sha256=Ar8-vSNwmCaBupCLY_rcvyU_kWIILU_qVX5njV-tZVw,2478
 gibson/command/Show.py,sha256=qkkprY2JhA4qOOhYOwAECDnFZwTdqcsKsG4cwB_b-84,1409
+gibson/command/Studio.py,sha256=ZOcXGzrkGGFYtsrxP0qTeComKfUcpTAZWJ9Njd7JfR4,2019
 gibson/command/Tree.py,sha256=BeJ_13xrrRCK5FP2rQHWpDKrshVzte-_D1pNG1GXPIw,3056
 gibson/command/Version.py,sha256=jxkRdbQiyTdto18RpbL-5vudcbvLLX9kcl8vmkt7USw,1187
 gibson/command/auth/Auth.py,sha256=DAvnKq3Ks77QJwuGJCWA9Iv3c0Qq5pHFIpEA-gy6CxM,1086
@@ -66,14 +68,15 @@ gibson/conf/dev/Schema.py,sha256=kOSlX1jEyVb82xd8TO8jEAimLcaefIFJr6d2JYvyTqg,74
 gibson/conf/tests/test_conf_Dependencies.py,sha256=-MAS3SzB1euHFwFhd0dpyIHYNMYgxEaN2LTeZ1pqc6w,363
 gibson/conf/tests/test_conf_Platform.py,sha256=Zc53IsZmV-hT9VRrZEPNrsuehSdWnJXWKGMmOhEqWHo,138
 gibson/core/Colors.py,sha256=sllEmJAb2AAUH0e-ZLP1_C8pfz5U_w0fo5kubSH5g1o,3426
-gibson/core/CommandRouter.py,sha256=V7awLSPjOx2GQqJNoHQSFsiA2uFkrxtpysyTDbyPloA,3442
+gibson/core/CommandRouter.py,sha256=WiTQJ2HwIoWOcoT0yL_7wXClOWqPSdWP5eoOd1f1tWw,3698
 gibson/core/Completions.py,sha256=a26WRh40UpnTT5HGTPT8TCcL8h80HvvZiTJXZofDjx8,1207
-gibson/core/Configuration.py,sha256=Cf1V5H17qibDXKcFt6UoyDczeQR33h8HTuKDk51Cae4,16581
+gibson/core/Configuration.py,sha256=eBGBoCe4rrBxlAUg0-UhKwrQSnyxFoey5jC_FeaGlCI,16689
 gibson/core/Conversation.py,sha256=KF7YPXijhhz6HOkife__ycHox4WeRKNHIpv3juDPhq0,10237
 gibson/core/Diff.py,sha256=onUJ5_0_S1vKAY_oFgX4vmwQo4byrnXLV4w7QSNA8fY,1071
 gibson/core/Env.py,sha256=08dZRHzzR0ahrbM4S0bXC7V1xhYQkT8Zefs00qUHf0U,498
 gibson/core/Memory.py,sha256=3ItGef4RCfBplbjxhNyid8eiPVKHmW-DKAMFeYujqN0,4063
 gibson/core/PythonPath.py,sha256=0Pj9VUmN-lbv_jGJjjI9UreU4JCdHPdRHSpxQDBrKQw,1577
+gibson/core/Select.py,sha256=8Ax8-rpK3BNrSAlSSSKpv1_uqQuwfy_4kNz7IJoZkjc,8510
 gibson/core/Spinner.py,sha256=_BO26dOa3h0ZCNaPgMvwOh7K6DUiDv8LgP7QMR3EvNw,1225
 gibson/core/TimeKeeper.py,sha256=dSeIgGOQJOi0ULlFGAigroGTBfAZXrvP9a1Op_jIsZ0,300
 gibson/core/utils.py,sha256=KTnPvA3sUYnLFTZG7Tke5YEdls8Da0rNbeaOm8hapiU,408
@@ -129,8 +132,8 @@ gibson/tests/test_Env.py,sha256=DPWmP0-aEelducq9bAwv7rKoY2NjWXUeCrzfJDQkn2M,369
 gibson/tests/test_Memory.py,sha256=YP7owToABAk_-s7fD5UG0HTc4lamDjdA39JUlLnk3Fg,2574
 gibson/tests/test_utils.py,sha256=r_y-EG05YTCNtL8MWiAK1KmPsmeoMgypKsQC_lVgOtM,559
 venv/bin/activate_this.py,sha256=E1T7r3559tBsyqFpdcQW0HbY7gDvNiIv5Pc6HQ4bpoA,2383
-gibson_cli-0.8.9.dist-info/METADATA,sha256=M3ndr7LtsbjmX__89dHsi8f3DVddNDKHxuiSRiSss4o,14574
-gibson_cli-0.8.9.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-gibson_cli-0.8.9.dist-info/entry_points.txt,sha256=j5VUvq3AzL21xPvVC24zMoXFt-I5lUWulr66nL3OAPM,50
-gibson_cli-0.8.9.dist-info/top_level.txt,sha256=fSV3vegbdbSDwiB6n5z3FCeYwkIonzFrx4ek3F_OSdI,16
-gibson_cli-0.8.9.dist-info/RECORD,,
+gibson_cli-0.8.10.dist-info/METADATA,sha256=5Rz6qup5c7MX09nxiEi3xKUgEnChRIsVHqy-pLU45WM,14645
+gibson_cli-0.8.10.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+gibson_cli-0.8.10.dist-info/entry_points.txt,sha256=j5VUvq3AzL21xPvVC24zMoXFt-I5lUWulr66nL3OAPM,50
+gibson_cli-0.8.10.dist-info/top_level.txt,sha256=fSV3vegbdbSDwiB6n5z3FCeYwkIonzFrx4ek3F_OSdI,16
+gibson_cli-0.8.10.dist-info/RECORD,,