feat(staticroutes): adds v2 static routes

This commit introduces support for V2 static routes

riocli/apply/resolver.py

@@ -139,7 +139,7 @@ def _list_functors(self, kind):
             'secret': self.client.list_secrets,
             "project": self.v2client.list_projects,
             "package": self.client.get_all_packages,
-            "staticroute": self.client.get_all_static_routes,
+            "staticroute": self.v2client.list_static_routes,
             "build": self.client.list_builds,
             "deployment": functools.partial(self.client.get_all_deployments,
                                             phases=[DeploymentPhaseConstants.SUCCEEDED,

riocli/jsonschema/schemas/static_route-schema.yaml

@@ -14,6 +14,10 @@ definitions:
         const: StaticRoute
       metadata:
         "$ref": "#/definitions/metadata"
+      spec:
+        "$ref": "#/definitions/staticRouteSpec"
+      status:
+        "$ref": "#/definitions/staticRouteStatus"
     required:
       - apiVersion
       - kind
@@ -47,3 +51,17 @@ definitions:
   uuid:
     type: string
     pattern: "^[a-fA-F0-9]{8}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{12}$"
+  staticRouteSpec:
+    type: object
+    properties:
+      url:
+        type: string
+  staticRouteStatus:
+    type: object
+    properties:
+      status:
+        type: string
+      packageID:
+        type: string
+      deploymentID:
+        type: string

riocli/static_route/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2021 Rapyuta Robotics
+# Copyright 2023 Rapyuta Robotics
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

riocli/static_route/create.py

@@ -14,30 +14,32 @@
 import click
 from click_help_colors import HelpColorsCommand
 
-from riocli.config import new_client
+from riocli.config import new_v2_client
 from riocli.constants import Colors, Symbols
 from riocli.utils.spinner import with_spinner
 
-
 @click.command(
     'create',
     cls=HelpColorsCommand,
     help_headers_color=Colors.YELLOW,
     help_options_color=Colors.GREEN,
 )
-@click.argument('prefix', type=str)
+@click.argument('name', type=str)
 @with_spinner(text="Creating static route...")
-def create_static_route(prefix: str, spinner=None) -> None:
+def create_static_route(name: str, spinner=None) -> None:
     """
     Creates a new static route
     """
     try:
-        client = new_client()
-        route = client.create_static_route(prefix)
+        client = new_v2_client(with_project=True)
+        payload = {
+            "metadata": {"name": name}
+        }
+        route = client.create_static_route(payload)
         spinner.text = click.style(
-            'Static Route created successfully for URL {}'.format(route.urlString), fg=Colors.GREEN)
+            'Static Route created successfully for URL {}'.format(route.spec.url), fg=Colors.GREEN)
         spinner.green.ok(Symbols.SUCCESS)
     except Exception as e:
         spinner.text = click.style('Failed to create static route: {}'.format(e), fg=Colors.RED)
         spinner.red.fail(Symbols.ERROR)
-        raise SystemExit(1)
+        raise SystemExit(1) from e

riocli/static_route/delete.py

@@ -14,12 +14,11 @@
 import click
 from click_help_colors import HelpColorsCommand
 
-from riocli.config import new_client
-from riocli.constants import Colors, Symbols
 from riocli.static_route.util import name_to_guid
+from riocli.config import new_v2_client
+from riocli.constants import Colors, Symbols
 from riocli.utils.spinner import with_spinner
 
-
 @click.command(
     'delete',
     cls=HelpColorsCommand,
@@ -46,12 +45,12 @@ def delete_static_route(
                     static_route, static_route_guid), abort=True)
 
     try:
-        client = new_client()
-        client.delete_static_route(static_route_guid)
+        client = new_v2_client()
+        client.delete_static_route(static_route)
         spinner.text = click.style(
             'Static Route deleted successfully ', fg=Colors.GREEN)
         spinner.green.ok(Symbols.SUCCESS)
     except Exception as e:
         spinner.text = click.style('Failed to delete static route: {}'.format(e), fg=Colors.RED)
         spinner.red.fail(Symbols.ERROR)
-        raise SystemExit(1)
+        raise SystemExit(1) from e

riocli/static_route/inspect.py

@@ -13,9 +13,9 @@
 # limitations under the License.
 import click
 from click_help_colors import HelpColorsCommand
-from rapyuta_io.clients.static_route import StaticRoute
+from munch import unmunchify
 
-from riocli.config import new_client
+from riocli.config import new_v2_client
 from riocli.constants import Colors
 from riocli.static_route.util import name_to_guid
 from riocli.utils import inspect_with_format
@@ -41,24 +41,11 @@ def inspect_static_route(
     Inspect a static route
     """
     try:
-        client = new_client()
-        route = client.get_static_route(static_route_guid)
-        data = make_static_route_inspectable(route)
-        inspect_with_format(data, format_type)
+        client = new_v2_client()
+        route = client.get_static_route(static_route)
+        inspect_with_format(unmunchify(route), format_type)
     except Exception as e:
         click.secho(str(e), fg=Colors.RED)
-        raise SystemExit(1)
+        raise SystemExit(1) from e
 
 
-def make_static_route_inspectable(static_route_data: StaticRoute) -> dict:
-    return {
-        'created_at': static_route_data.CreatedAt,
-        'updated_at': static_route_data.UpdatedAt,
-        'deleted_at': static_route_data.DeletedAt,
-        'guid': static_route_data.guid,
-        'url_prefix': static_route_data.urlPrefix,
-        'url': static_route_data.urlString,
-        'creator': static_route_data.creator,
-        'project': static_route_data.projectGUID,
-        'metadata': static_route_data.metadata.__dict__,
-    }

riocli/static_route/list.py

@@ -17,11 +17,10 @@
 from click_help_colors import HelpColorsCommand
 from rapyuta_io.clients.static_route import StaticRoute
 
-from riocli.config import new_client
+from riocli.config import new_v2_client
 from riocli.constants import Colors
 from riocli.utils import tabulate_data
 
-
 @click.command(
     'list',
     cls=HelpColorsCommand,
@@ -33,25 +32,24 @@ def list_static_routes() -> None:
     List the static routes in the selected project
     """
     try:
-        client = new_client()
-        routes = client.get_all_static_routes()
+        client = new_v2_client(with_project=True)
+        routes = client.list_static_routes()
         _display_routes_list(routes)
     except Exception as e:
         click.secho(str(e), fg=Colors.RED)
-        raise SystemExit(1)
-
+        raise SystemExit(1) from e
 
 def _display_routes_list(routes: List[StaticRoute]) -> None:
     headers = ['Route ID', 'Name', 'URL', 'Creator', 'CreatedAt']
 
     data = []
     for route in routes:
         data.append([
-            route.guid,
-            route.urlPrefix,
-            route.urlString,
-            route.creator,
-            route.CreatedAt,
+            route.metadata.guid,
+            route.metadata.name,
+            route.spec.url,
+            route.metadata.creatorGUID,
+            route.metadata.createdAt,
         ])
 
     tabulate_data(data, headers)

riocli/static_route/model.py

@@ -12,10 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import typing
+from munch import unmunchify
 
 from rapyuta_io import Client
-from rapyuta_io.clients.static_route import StaticRoute as v1StaticRoute
 
+from riocli.config import new_v2_client
 from riocli.jsonschema.validate import load_schema
 from riocli.model import Model
 
@@ -29,21 +30,28 @@ def find_object(self, client: Client) -> bool:
             'kind': 'staticroute',
             'nameOrGUID': self.metadata.name
         })
-
         if not static_route:
             return False
 
         return static_route
 
-    def create_object(self, client: Client, **kwargs) -> v1StaticRoute:
-        static_route = client.create_static_route(self.metadata.name)
-        return static_route
+    def create_object(self, client: Client, **kwargs) -> typing.Any:
+        client = new_v2_client()
+
+        # convert to a dict and remove the ResolverCache
+        # field since it's not JSON serializable
+        static_route = unmunchify(self)
+        static_route.pop("rc", None)
+        r = client.create_static_route(static_route)
+        return unmunchify(r)
 
     def update_object(self, client: Client, obj: typing.Any) -> None:
         pass
 
     def delete_object(self, client: Client, obj: typing.Any):
-        client.delete_static_route(obj.guid)
+        client = new_v2_client()
+        import pdb;pdb.set_trace()
+        client.delete_static_route(obj.name)
 
     @classmethod
     def pre_process(cls, client: Client, d: typing.Dict) -> None:

riocli/static_route/open.py

@@ -14,7 +14,7 @@
 import click
 from click_help_colors import HelpColorsCommand
 
-from riocli.config import new_client
+from riocli.config import new_v2_client
 from riocli.constants import Colors
 from riocli.static_route.util import name_to_guid
 
@@ -32,9 +32,9 @@ def open_static_route(static_route, static_route_guid) -> None:
     Opens the static route in the default browser
     """
     try:
-        client = new_client()
-        route = client.get_static_route(static_route_guid)
-        click.launch(url='https://{}'.format(route.urlString), wait=False)
+        client = new_v2_client()
+        route = client.get_static_route(static_route)
+        click.launch(url='https://{}'.format(route.spec.url), wait=False)
     except Exception as e:
         click.secho(str(e), fg=Colors.RED)
         raise SystemExit(1)

riocli/static_route/util.py

@@ -16,23 +16,23 @@
 
 from rapyuta_io import Client
 
-from riocli.config import new_client
+from riocli.config import new_v2_client
 
 
 def name_to_guid(f: typing.Callable) -> typing.Callable:
     @functools.wraps(f)
     def decorated(**kwargs: typing.Any):
-        client = new_client()
+        client = new_v2_client()
         name = kwargs.pop('static_route')
         guid = None
 
+        name = name.rsplit(".", 2)[0]
         if name.startswith('staticroute-'):
             guid = name
             name = None
 
         if name is None:
             name = get_static_route_name(client, guid)
-
         if guid is None:
             guid = find_static_route_guid(client, name)
 
@@ -44,18 +44,20 @@ def decorated(**kwargs: typing.Any):
 
 
 def get_static_route_name(client: Client, guid: str) -> str:
-    static_route = client.get_static_route(guid)
-    return static_route.urlPrefix.split("-")[0]
-
 
-def find_static_route_guid(client: Client, name: str) -> str:
-    routes = client.get_all_static_routes()
+    routes = client.list_static_routes()
     for route in routes:
-        if route.urlPrefix == name or route.urlString == name:
-            return route.guid
+        if route.metadata.guid == guid:
+            return route.metadata.name
 
     raise StaticRouteNotFound()
 
+def find_static_route_guid(client: Client, name: str) -> str:
+    static_route = client.get_static_route(name)
+    if not static_route:
+        raise StaticRouteNotFound()
+    return static_route.metadata.guid
+
 
 class StaticRouteNotFound(Exception):
     def __init__(self):

riocli/utils/__init__.py

@@ -161,7 +161,7 @@ def is_pip_installation() -> bool:
     return 'python' in sys.executable
 
 
-def check_for_updates(current_version: str) -> tuple[bool, str]:
+def check_for_updates(current_version: str) -> typing.Tuple[bool, str]:
     try:
         package_info = requests.get(
             'https://pypi.org/pypi/rapyuta-io-cli/json').json()