From e89fcb668796d64314ae85e89463ba4bf632dd0e Mon Sep 17 00:00:00 2001 From: Prathamesh Musale Date: Mon, 7 Oct 2024 12:42:47 +0530 Subject: [PATCH] Add a command to request an application undeployment --- .../webapp/request_webapp_deployment.py | 6 +- .../webapp/request_webapp_undeployment.py | 106 ++++++++++++++++++ stack_orchestrator/main.py | 4 +- 3 files changed, 114 insertions(+), 2 deletions(-) create mode 100644 stack_orchestrator/deploy/webapp/request_webapp_undeployment.py diff --git a/stack_orchestrator/deploy/webapp/request_webapp_deployment.py b/stack_orchestrator/deploy/webapp/request_webapp_deployment.py index 412f937f..097dcc0e 100644 --- a/stack_orchestrator/deploy/webapp/request_webapp_deployment.py +++ b/stack_orchestrator/deploy/webapp/request_webapp_deployment.py @@ -3,7 +3,6 @@ # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. -import base64 # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of @@ -17,6 +16,7 @@ import shutil import sys import tempfile from datetime import datetime +import base64 import gnupg import click @@ -86,6 +86,10 @@ def command( # noqa: C901 print("Cannot specify both --auction-id and --deployer", file=sys.stderr) sys.exit(2) + if not auction_id and not deployer: + print("Must specify either --auction-id or --deployer", file=sys.stderr) + sys.exit(2) + if auction_id and (make_payment or use_payment): print("Cannot specify --auction-id with --make-payment or --use-payment", file=sys.stderr) sys.exit(2) diff --git a/stack_orchestrator/deploy/webapp/request_webapp_undeployment.py b/stack_orchestrator/deploy/webapp/request_webapp_undeployment.py new file mode 100644 index 00000000..80cee3ce --- /dev/null +++ b/stack_orchestrator/deploy/webapp/request_webapp_undeployment.py @@ -0,0 +1,106 @@ +# Copyright ©2023 Vulcanize +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. + +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. + +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see . + +import sys + +import click +import yaml + +from stack_orchestrator.deploy.webapp.util import (LaconicRegistryClient) + + +def fatal(msg: str): + print(msg, file=sys.stderr) + sys.exit(1) + + +@click.command() +@click.option( + "--laconic-config", help="Provide a config file for laconicd", required=True +) +@click.option( + "--deployer", + help="The LRN of the deployer to process this request.", + required=True +) +@click.option( + "--deployment", + help="Deployment record (ApplicationDeploymentRecord) id of the deployment to remove.", + required=True, +) +@click.option( + "--make-payment", + help="The payment to make (in alnt). The value should be a number or 'auto' to use the deployer's minimum required payment.", +) +@click.option( + "--use-payment", help="The TX id of an existing, unused payment", default=None +) +@click.option( + "--dry-run", + help="Don't publish anything, just report what would be done.", + is_flag=True, +) +@click.pass_context +def command( + ctx, + laconic_config, + deployer, + deployment, + make_payment, + use_payment, + dry_run, +): + if make_payment and use_payment: + fatal("Cannot use --make-payment and --use-payment at the same time.") + + laconic = LaconicRegistryClient(laconic_config) + + deployer_record = laconic.get_record(deployer) + if not deployer_record: + fatal(f"Unable to locate deployer: {deployer}") + + undeployment_request = { + "record": { + "type": "ApplicationDeploymentRemovalRequest", + "version": "1.0.0", + "deployer": deployer, + "deployment": deployment, + } + } + + if make_payment: + amount = 0 + if dry_run: + undeployment_request["record"]["payment"] = "DRY_RUN" + elif "auto" == make_payment: + if "minimumPayment" in deployer_record.attributes: + amount = int( + deployer_record.attributes.minimumPayment.replace("alnt", "") + ) + else: + amount = make_payment + if amount: + receipt = laconic.send_tokens( + deployer_record.attributes.paymentAddress, amount + ) + undeployment_request["record"]["payment"] = receipt.tx.hash + print("Payment TX:", receipt.tx.hash) + elif use_payment: + undeployment_request["record"]["payment"] = use_payment + + if dry_run: + print(yaml.dump(undeployment_request)) + return + + laconic.publish(undeployment_request) diff --git a/stack_orchestrator/main.py b/stack_orchestrator/main.py index b8ca430d..a50c7c9b 100644 --- a/stack_orchestrator/main.py +++ b/stack_orchestrator/main.py @@ -28,7 +28,8 @@ from stack_orchestrator.deploy.webapp import (run_webapp, publish_webapp_deployer, publish_deployment_auction, handle_deployment_auction, - request_webapp_deployment) + request_webapp_deployment, + request_webapp_undeployment) from stack_orchestrator.deploy import deploy from stack_orchestrator import version from stack_orchestrator.deploy import deployment @@ -69,6 +70,7 @@ cli.add_command(publish_webapp_deployer.command, "publish-deployer-to-registry") cli.add_command(publish_deployment_auction.command, "publish-deployment-auction") cli.add_command(handle_deployment_auction.command, "handle-deployment-auction") cli.add_command(request_webapp_deployment.command, "request-webapp-deployment") +cli.add_command(request_webapp_undeployment.command, "request-webapp-undeployment") cli.add_command(deploy.command, "deploy") # deploy is an alias for deploy-system cli.add_command(deploy.command, "deploy-system") cli.add_command(deployment.command, "deployment")