From c44b45426274024d9b29683a0753706fcb9e7632 Mon Sep 17 00:00:00 2001
From: Damon <damondouglas@users.noreply.github.com>
Date: Tue, 2 Apr 2024 13:21:42 -0700
Subject: [PATCH] Add cancel button to Prism UI (#30811)

* Add cancel button to Prism UI

* Modify TODO fmt
---
 .../prism/internal/web/assets/job-action.js   | 97 +++++++++++++++++++
 .../prism/internal/web/assets/style.css       |  8 +-
 .../prism/internal/web/jobdetails.html        |  6 ++
 3 files changed, 110 insertions(+), 1 deletion(-)
 create mode 100644 sdks/go/pkg/beam/runners/prism/internal/web/assets/job-action.js

diff --git a/sdks/go/pkg/beam/runners/prism/internal/web/assets/job-action.js b/sdks/go/pkg/beam/runners/prism/internal/web/assets/job-action.js
new file mode 100644
index 000000000000..999fd22bbf88
--- /dev/null
+++ b/sdks/go/pkg/beam/runners/prism/internal/web/assets/job-action.js
@@ -0,0 +1,97 @@
+/**
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ */
+
+/** Element class for job action container. */
+const JOB_ACTION = '.job-action'
+
+/** Element class for cancel button. */
+const CANCEL = '.cancel'
+
+/** Element class assigned to RUNNING Job state. */
+const RUNNING = 'RUNNING'
+
+/**
+ * Client for Job Management.
+ *
+ * Invokes backend for various Job Management tasks.
+ */
+const jobManager = {
+
+    /**
+     * Cancel a Job.
+     * Invokes backend handler to request a Job cancellation state.
+     * @param jobId
+     * TODO(https://github.com/apache/beam/issues/29669) Send request to backend service.
+     */
+    cancel: function(jobId) {
+        console.debug(`cancel button for Job: ${jobId} clicked`)
+    }
+}
+
+/**
+ * Encapsulates UI state such as enabling/disabling elements and querying various elements and their properties.
+ */
+const uiStateProvider = {
+
+    /**
+     * Queries the DOM for the Job Action container.
+     * Logs an error if not found.
+     * @returns {Element}
+     */
+    get jobAction() {
+        let element = document.querySelector(JOB_ACTION)
+        if (element === null) {
+            console.error(`no element found at ${JOB_ACTION}`)
+        }
+        return element
+    },
+
+    /**
+     * Queries Job Action container DOM for the cancel button.
+     * Logs an error if not found.
+     * @returns {Element}
+     */
+    get cancelButton() {
+        let element = this.jobAction.querySelector(CANCEL)
+        if (element === null) {
+            console.error(`no element found at ${CANCEL} within ${this.jobAction}`)
+        }
+        return element
+    },
+
+    /**
+     * Initializes the uiStateManager.
+     * Called from the window's load event.
+     */
+    init() {
+        this.cancelButton.disabled = this.isStateRunning === false
+    },
+
+    /**
+     * Queries whether the Job Action container contains the RUNNING class name.
+     * @returns {boolean}
+     */
+    get isStateRunning() {
+        return this.jobAction.classList.contains(RUNNING)
+    }
+}
+
+/**
+ * Attaches an event listener to the window for 'load' events.
+ */
+window.addEventListener("load", function(){
+    console.debug(JOB_ACTION, uiStateProvider.jobAction)
+    console.debug(CANCEL, uiStateProvider.cancelButton)
+    uiStateProvider.init()
+})
diff --git a/sdks/go/pkg/beam/runners/prism/internal/web/assets/style.css b/sdks/go/pkg/beam/runners/prism/internal/web/assets/style.css
index d252dc020e63..48cc7acecb30 100644
--- a/sdks/go/pkg/beam/runners/prism/internal/web/assets/style.css
+++ b/sdks/go/pkg/beam/runners/prism/internal/web/assets/style.css
@@ -205,4 +205,10 @@ footer {
         display: inline-block;
         margin-bottom: 10px;
     }
-}
\ No newline at end of file
+}
+
+/* Make Job action related buttons more visible. */
+.job-action button {
+    border-radius: 5px;
+    padding: 5px;
+}
diff --git a/sdks/go/pkg/beam/runners/prism/internal/web/jobdetails.html b/sdks/go/pkg/beam/runners/prism/internal/web/jobdetails.html
index cfe82c15147c..ff87f677e2da 100644
--- a/sdks/go/pkg/beam/runners/prism/internal/web/jobdetails.html
+++ b/sdks/go/pkg/beam/runners/prism/internal/web/jobdetails.html
@@ -18,6 +18,7 @@
     <meta http-equiv="X-UA-Compatible" content="ie=edge" />
     <title>{{.JobID}} - {{ .JobName }} - Job Details - Beam Prism</title>
     <link rel="stylesheet" href="/assets/style.css" />
+    <script src="/assets/job-action.js"></script>
     {{/*
     <meta http-equiv="refresh" content="10"> refresh page every 10 seconds */}}
 </head>
@@ -27,6 +28,11 @@
         <header>
             <a class="logo" href="/">Job Details - Beam Prism</a>
             <div>{{.JobID}} - {{ .JobName }}</div>
+            <div class="job-action {{.State}}">
+                <button class="cancel"
+                        onclick="if (jobManager !== null) { jobManager.cancel('{{.JobID}}') }"
+                >Cancel</button>
+            </div>
             <div>{{.State}}</div>
         </header>
         <section class="container">