Feature 10997: add oci flag as new property in helm-chart repos

[osm/osmclient.git] / osmclient / cli_commands / repo.py

diff --git a/osmclient/cli_commands/repo.py b/osmclient/cli_commands/repo.py
index 2e5b860..495bd7f 100755 (executable)
--- a/osmclient/cli_commands/repo.py
+++ b/osmclient/cli_commands/repo.py
@@ -29,9 +29,9 @@ logger = logging.getLogger("osmclient")
 @click.argument("uri")
 @click.option(
     "--type",
-    type=click.Choice(["helm-chart", "juju-bundle", "osm"]),
+    type=click.Choice(["osm", "helm-chart", "juju-bundle"]),
     default="osm",
-    help="type of repo (helm-chart for Helm Charts, juju-bundle for Juju Bundles, osm for OSM Repositories)",
+    help="type of repo (osm for OSM repositories, helm-chart for Helm Charts, juju-bundle for Juju Bundles)",
 )
 @click.option("--description", default=None, help="human readable description")
 @click.option(
@@ -42,6 +42,11 @@ logger = logging.getLogger("osmclient")
     default=None,
     help="OSM repository: The password of the OSM repository",
 )
+@click.option(
+    "--oci",
+    is_flag=True,
+    help="enable OCI (only for helm-chart repos, default: false, automatically set to true for oci:// URI)",
+)
 @click.pass_context
 def repo_add(ctx, **kwargs):
     """adds a repo to OSM
@@ -52,6 +57,8 @@ def repo_add(ctx, **kwargs):
     kwargs = {k: v for k, v in kwargs.items() if v is not None}
     repo = kwargs
     repo["url"] = repo.pop("uri")
+    if repo["url"].startswith("oci://"):
+        repo["oci"] = True
     if repo["type"] in ["helm-chart", "juju-bundle"]:
         ctx.obj.repo.create(repo["name"], repo)
     else:
@@ -63,8 +70,13 @@ def repo_add(ctx, **kwargs):
 @click.option("--newname", help="New name for the repo")
 @click.option("--uri", help="URI of the repo")
 @click.option("--description", help="human readable description")
+@click.option(
+    "--oci",
+    is_flag=True,
+    help="enable OCI (only for helm-chart repos, default: false, automatically set to true for oci:// URI)",
+)
 @click.pass_context
-def repo_update(ctx, name, newname, uri, description):
+def repo_update(ctx, name, newname, uri, description, oci):
     """updates a repo in OSM
 
     NAME: name of the repo
@@ -74,9 +86,13 @@ def repo_update(ctx, name, newname, uri, description):
     if newname:
         repo["name"] = newname
     if uri:
-        repo["uri"] = uri
+        repo["url"] = uri
+        if uri.startswith("oci://"):
+            repo["oci"] = True
     if description:
         repo["description"] = description
+    if oci:
+        repo["oci"] = oci
     try:
         ctx.obj.repo.update(name, repo)
     except NotFound: