From d6986f756c324f0f2833af0b4d30cbbaa31b5f73 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bla=C5=BE=20Hrastnik?= <blaz@mxxn.io>
Date: Fri, 12 Jul 2024 12:49:02 +0900
Subject: [PATCH] scripts: keystone: Add utility to delete a workflow spec

---
 core/scripts/keystone/main.go                 |  1 +
 .../keystone/src/04_delete_ocr3_jobs_cmd.go   |  5 ++
 .../keystone/src/07_delete_workflows_cmd.go   | 61 +++++++++++++++++++
 3 files changed, 67 insertions(+)
 create mode 100644 core/scripts/keystone/src/07_delete_workflows_cmd.go

diff --git a/core/scripts/keystone/main.go b/core/scripts/keystone/main.go
index 26639c94aa9..3486830ca32 100644
--- a/core/scripts/keystone/main.go
+++ b/core/scripts/keystone/main.go
@@ -21,6 +21,7 @@ func main() {
 		src.NewDeleteJobsCommand(),
 		src.NewDeployAndInitializeCapabilitiesRegistryCommand(),
 		src.NewDeployWorkflowsCommand(),
+		src.NewDeleteWorkflowsCommand(),
 	}
 
 	commandsList := func(commands []command) string {
diff --git a/core/scripts/keystone/src/04_delete_ocr3_jobs_cmd.go b/core/scripts/keystone/src/04_delete_ocr3_jobs_cmd.go
index 2ebed000ed4..65c494d5766 100644
--- a/core/scripts/keystone/src/04_delete_ocr3_jobs_cmd.go
+++ b/core/scripts/keystone/src/04_delete_ocr3_jobs_cmd.go
@@ -22,11 +22,16 @@ type BootSpec struct {
 	ContractID string
 }
 
+type WorkflowSpec struct {
+	WorkflowID string
+}
+
 type JobSpec struct {
 	Id                           string
 	Name                         string
 	BootstrapSpec                BootSpec
 	OffChainReporting2OracleSpec OCRSpec
+	WorkflowSpec                 WorkflowSpec
 }
 
 func NewDeleteJobsCommand() *deleteJobs {
diff --git a/core/scripts/keystone/src/07_delete_workflows_cmd.go b/core/scripts/keystone/src/07_delete_workflows_cmd.go
new file mode 100644
index 00000000000..62e4047a066
--- /dev/null
+++ b/core/scripts/keystone/src/07_delete_workflows_cmd.go
@@ -0,0 +1,61 @@
+package src
+
+import (
+	"bytes"
+	"encoding/json"
+	"flag"
+	"fmt"
+
+	"github.com/urfave/cli"
+
+	helpers "github.com/smartcontractkit/chainlink/core/scripts/common"
+)
+
+type deleteWorkflows struct {
+}
+
+func NewDeleteWorkflowsCommand() *deleteWorkflows {
+	return &deleteWorkflows{}
+}
+
+func (g *deleteWorkflows) Name() string {
+	return "delete-workflows"
+}
+
+func (g *deleteWorkflows) Run(args []string) {
+	nodes := downloadNodeAPICredentials(".cache/NodeList.txt")
+
+	for _, node := range nodes {
+		output := &bytes.Buffer{}
+		client, app := newApp(node, output)
+
+		fmt.Println("Logging in:", node.url)
+		loginFs := flag.NewFlagSet("test", flag.ContinueOnError)
+		loginFs.Bool("bypass-version-check", true, "")
+		loginCtx := cli.NewContext(app, loginFs, nil)
+		err := client.RemoteLogin(loginCtx)
+		helpers.PanicErr(err)
+		output.Reset()
+
+		fileFs := flag.NewFlagSet("test", flag.ExitOnError)
+		err = client.ListJobs(cli.NewContext(app, fileFs, nil))
+		helpers.PanicErr(err)
+
+		var parsed []JobSpec
+		err = json.Unmarshal(output.Bytes(), &parsed)
+		helpers.PanicErr(err)
+
+		for _, jobSpec := range parsed {
+			if jobSpec.WorkflowSpec.WorkflowID != "" {
+				fmt.Println("Deleting workflow job ID:", jobSpec.Id, "name:", jobSpec.Name)
+				set := flag.NewFlagSet("test", flag.ExitOnError)
+				err = set.Parse([]string{jobSpec.Id})
+				helpers.PanicErr(err)
+				err = client.DeleteJob(cli.NewContext(app, set, nil))
+				helpers.PanicErr(err)
+			}
+		}
+
+		output.Reset()
+	}
+}