add login and out of oc

Author: MichaelClifford

src/codeflare_sdk/cluster/auth.py

@@ -0,0 +1,59 @@
+# Copyright 2022 IBM, Red Hat
+#
+# 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.
+
+import openshift as oc
+
+
+class Authentication:
+    
+    def login(self):
+        pass
+
+    def logout(self):
+        response = oc.invoke("logout")
+        return response.out()
+
+
+class TokenAuthentication(Authentication):
+
+    def __init__(
+        self, 
+        token: str= None,
+        server: str = None,):
+
+        self.token = token
+        self.server = server
+            
+
+    def login(self):
+        token = self.token
+        server = self.server
+        response = oc.invoke("login",[f'--token={token}', f'--server={server}:6443'])
+        return response.out()
+
+
+class PasswordUserAuthentication(Authentication):
+   
+    def __init__(
+        self, 
+        username: str= None,
+        password: str = None,):
+
+        self.username = username
+        self.password = password
+        
+    def login(self):
+        response = oc.login(self.username, 
+                            self.password)
+        return response.out()

src/codeflare_sdk/cluster/cluster.py

@@ -50,6 +50,7 @@ def create_app_wrapper(self):
 
     # creates a new cluster with the provided or default spec    
     def up(self):
+        self.config.auth.login()
         namespace = self.config.namespace
         with oc.project(namespace):
             oc.invoke("apply", ["-f", self.app_wrapper_yaml])
@@ -58,6 +59,7 @@ def down(self):
         namespace = self.config.namespace
         with oc.project(namespace):
             oc.invoke("delete", ["AppWrapper", self.app_wrapper_name])
+        self.config.auth.logout()
 
     def status(self, print_to_console=True):
         cluster = _ray_cluster_status(self.config.name, self.config.namespace)

src/codeflare_sdk/cluster/config.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 from dataclasses import dataclass, field
+from .auth import Authentication
 import pathlib
 
 dir = pathlib.Path(__file__).parent.parent.resolve()
@@ -34,3 +35,5 @@ class ClusterConfiguration:
     instascale: bool = False
     envs: dict = field(default_factory=dict)
     image: str = "ghcr.io/ibm-ai-foundation/base:ray1.13.0-py38-gpu-pytorch1.12.0cu116-20220826-202124"
+    auth: Authentication = Authentication()
+