From 80d07edc669316d3f5583fb3a45679ac80477be7 Mon Sep 17 00:00:00 2001
From: Janne Savolainen <janne.savolainen@live.fi>
Date: Fri, 11 Feb 2022 10:44:26 +0200
Subject: [PATCH] Switch to using injectable to setup route for rollbacking
 release

Signed-off-by: Janne Savolainen <janne.savolainen@live.fi>
---
 src/main/router.ts                            |  1 -
 .../rollback-release-route.injectable.ts      | 37 +++++++++++++++++++
 2 files changed, 37 insertions(+), 1 deletion(-)
 create mode 100644 src/main/routes/helm/releases/rollback-release-route.injectable.ts

diff --git a/src/main/router.ts b/src/main/router.ts
index 00306d64e6..681842284b 100644
--- a/src/main/router.ts
+++ b/src/main/router.ts
@@ -183,7 +183,6 @@ export class Router {
     this.router.add({ method: "delete", path: `${apiPrefix}/pods/port-forward/{namespace}/{resourceType}/{resourceName}` }, PortForwardRoute.routeCurrentPortForwardStop);
 
     // Helm API
-    this.router.add({ method: `put`, path: `${apiPrefix}/v2/releases/{namespace}/{release}/rollback` }, HelmApiRoute.rollbackRelease);
     this.router.add({ method: "get", path: `${apiPrefix}/v2/releases/{namespace?}` }, HelmApiRoute.listReleases);
     this.router.add({ method: "get", path: `${apiPrefix}/v2/releases/{namespace}/{release}` }, HelmApiRoute.getRelease);
     this.router.add({ method: "get", path: `${apiPrefix}/v2/releases/{namespace}/{release}/values` }, HelmApiRoute.getReleaseValues);
diff --git a/src/main/routes/helm/releases/rollback-release-route.injectable.ts b/src/main/routes/helm/releases/rollback-release-route.injectable.ts
new file mode 100644
index 0000000000..bff42c3f9b
--- /dev/null
+++ b/src/main/routes/helm/releases/rollback-release-route.injectable.ts
@@ -0,0 +1,37 @@
+/**
+ * Copyright (c) OpenLens Authors. All rights reserved.
+ * Licensed under MIT License. See LICENSE in root directory for more information.
+ */
+import { apiPrefix } from "../../../../common/vars";
+import type { LensApiRequest } from "../../../router";
+import { helmService } from "../../../helm/helm-service";
+import { respondText } from "../../../utils/http-responses";
+import { routeInjectionToken } from "../../../router/router.injectable";
+import { getInjectable } from "@ogre-tools/injectable";
+import logger from "../../../logger";
+
+const rollbackReleaseRouteInjectable = getInjectable({
+  id: "rollback-release-route",
+
+  instantiate: () => ({
+    method: "put",
+    path: `${apiPrefix}/v2/releases/{namespace}/{release}/rollback`,
+
+    handler: async (request: LensApiRequest) => {
+      const { cluster, params, payload, response } = request;
+
+      try {
+        await helmService.rollback(cluster, params.release, params.namespace, payload.revision);
+
+        response.end();
+      } catch (error) {
+        logger.debug(error);
+        respondText(response, error?.toString() || "Error rolling back chart", 422);
+      }
+    },
+  }),
+
+  injectionToken: routeInjectionToken,
+});
+
+export default rollbackReleaseRouteInjectable;