From c79453645d885dadcb46d55fe03da26149174bea Mon Sep 17 00:00:00 2001
From: MichaelClifford <mcliffor@redhat.com>
Date: Thu, 8 Dec 2022 11:55:47 -0500
Subject: [PATCH] add inital ray commands

---
 src/codeflare_sdk/cluster/cluster.py | 31 ++++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/src/codeflare_sdk/cluster/cluster.py b/src/codeflare_sdk/cluster/cluster.py
index 2eb8b5f10..aac52da40 100644
--- a/src/codeflare_sdk/cluster/cluster.py
+++ b/src/codeflare_sdk/cluster/cluster.py
@@ -22,6 +22,7 @@
 from typing import List, Optional, Tuple
 
 import openshift as oc
+from ray.job_submission import JobSubmissionClient
 
 from ..utils import pretty_print
 from ..utils.generate_yaml import generate_appwrapper
@@ -140,9 +141,11 @@ def cluster_dashboard_uri(self, namespace: str = "default") -> str:
         try:
             with oc.project(namespace):
                 route = oc.invoke(
-                    "get", ["route", "-o", "jsonpath='{$.items[0].spec.host}'"]
+                    "get", ["route", "-o", "jsonpath='{$.items[*].spec.host}'"]
                 )
-                route = route.out().strip().strip("'")
+                route = route.out().split(" ")
+                route = [x for x in route if f"ray-dashboard-{self.config.name}" in x]
+                route = route[0].strip().strip("'")
             return f"http://{route}"
         except:
             return "Dashboard route not available yet. Did you run cluster.up()?"
@@ -200,6 +203,30 @@ def is_ready(self, print_to_console: bool = True):
                 pretty_print.print_clusters([cluster])
         return status, ready
 
+    def list_jobs(self) -> List:
+        """
+        This method accesses the head ray node in your cluster and lists the running jobs.
+        """
+        dashboard_route = self.cluster_dashboard_uri(namespace=self.config.namespace)
+        client = JobSubmissionClient(dashboard_route)
+        return client.list_jobs()
+
+    def job_status(self, job_id: str) -> str:
+        """
+        This method accesses the head ray node in your cluster and returns the job status for the provided job id.
+        """
+        dashboard_route = self.cluster_dashboard_uri(namespace=self.config.namespace)
+        client = JobSubmissionClient(dashboard_route)
+        return client.get_job_status(job_id)
+
+    def job_logs(self, job_id: str) -> str:
+        """
+        This method accesses the head ray node in your cluster and returns the logs for the provided job id.
+        """
+        dashboard_route = self.cluster_dashboard_uri(namespace=self.config.namespace)
+        client = JobSubmissionClient(dashboard_route)
+        return client.get_job_logs(job_id)
+
 
 def get_current_namespace() -> str:
     """