Implement jmp client lease release

Author: NickCao

jumpstarter/cli/client.py

@@ -41,6 +41,27 @@ def lease_list(name):
         print(lease)
 
 
+@lease.command("release")
+@click.argument("name", type=str, default="")
+@click.option("-l", "--lease", "lease", type=str, default="")
+@click.option("--all", "all_leases", is_flag=True)
+def lease_release(name, lease, all_leases):
+    if name:
+        config = ClientConfigV1Alpha1.load(name)
+    else:
+        config = UserConfigV1Alpha1.load_or_create().config.current_client
+    if not config:
+        raise ValueError("no client specified")
+
+    if all_leases:
+        for lease in config.list_leases():
+            config.release_lease(lease)
+    else:
+        if not lease:
+            raise ValueError("no lease specified")
+        config.release_lease(lease)
+
+
 @click.command("create", short_help="Create a client configuration.")
 @click.argument("name")
 @click.option(

jumpstarter/config/client.py

@@ -66,10 +66,18 @@ def list_leases(self):
         with start_blocking_portal() as portal:
             return portal.call(self.list_leases_async)
 
+    def release_lease(self, name):
+        with start_blocking_portal() as portal:
+            portal.call(self.release_lease_async, name)
+
     async def list_leases_async(self):
         stub = jumpstarter_pb2_grpc.ControllerServiceStub(await self.channel())
         return (await stub.ListLeases(jumpstarter_pb2.ListLeasesRequest())).names
 
+    async def release_lease_async(self, name):
+        stub = jumpstarter_pb2_grpc.ControllerServiceStub(await self.channel())
+        await stub.ReleaseLease(jumpstarter_pb2.ReleaseLeaseRequest(name=name))
+
     @asynccontextmanager
     async def lease_async(self, metadata_filter: MetadataFilter, portal: BlockingPortal):
         async with LeaseRequest(