Skip to content

Commit 83be5f6

Browse files
Remove creation of OAuth resources/logic and add annotation
1 parent 47381fb commit 83be5f6

File tree

5 files changed

+18
-317
lines changed

5 files changed

+18
-317
lines changed

src/codeflare_sdk/cluster/cluster.py

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,7 @@
3333
)
3434
from ..utils.kube_api_helpers import _kube_api_error_handling
3535
from ..utils.generate_yaml import is_openshift_cluster
36-
from ..utils.openshift_oauth import (
37-
create_openshift_oauth_objects,
38-
delete_openshift_oauth_objects,
39-
)
36+
4037
from .config import ClusterConfiguration
4138
from .model import (
4239
AppWrapper,
@@ -226,10 +223,6 @@ def up(self):
226223
the MCAD queue.
227224
"""
228225
namespace = self.config.namespace
229-
if self.config.openshift_oauth:
230-
create_openshift_oauth_objects(
231-
cluster_name=self.config.name, namespace=namespace
232-
)
233226

234227
try:
235228
config_check()
@@ -271,11 +264,6 @@ def down(self):
271264
except Exception as e: # pragma: no cover
272265
return _kube_api_error_handling(e)
273266

274-
if self.config.openshift_oauth:
275-
delete_openshift_oauth_objects(
276-
cluster_name=self.config.name, namespace=namespace
277-
)
278-
279267
def status(
280268
self, print_to_console: bool = True
281269
) -> Tuple[CodeFlareClusterStatus, bool]:

src/codeflare_sdk/templates/base-template.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,8 @@ spec:
4040
apiVersion: ray.io/v1
4141
kind: RayCluster
4242
metadata:
43+
annotations:
44+
codeflare.dev/oauth: 'False'
4345
labels:
4446
workload.codeflare.dev/appwrapper: "aw-kuberay"
4547
controller-tools.k8s.io: "1.0"
@@ -169,6 +171,12 @@ spec:
169171
- mountPath: /etc/ssl/certs/odh-ca-bundle.crt
170172
name: odh-ca-cert
171173
subPath: odh-ca-bundle.crt
174+
env:
175+
- name: COOKIE_SECRET
176+
valueFrom:
177+
secretKeyRef:
178+
name: jobtest-oauth-config
179+
key: cookie_secret
172180
initContainers:
173181
- command:
174182
- sh

src/codeflare_sdk/utils/generate_yaml.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,12 +219,17 @@ def update_rayclient_ingress(
219219
spec["rules"][0]["host"] = f"rayclient-{cluster_name}-{namespace}.{ingress_domain}"
220220

221221

222-
def update_names(yaml, item, appwrapper_name, cluster_name, namespace):
222+
def update_names(yaml, item, appwrapper_name, cluster_name, namespace, openshift_oauth):
223223
metadata = yaml.get("metadata")
224224
metadata["name"] = appwrapper_name
225225
metadata["namespace"] = namespace
226226
lower_meta = item.get("generictemplate", {}).get("metadata")
227227
lower_meta["labels"]["workload.codeflare.dev/appwrapper"] = appwrapper_name
228+
lower_meta["annotations"]["codeflare.dev/oauth"] = f"{openshift_oauth}"
229+
lower_spec = item.get("generictemplate", {}).get("spec")
230+
lower_spec["headGroupSpec"]["template"]["spec"]["containers"][0]["env"][-1][
231+
"valueFrom"
232+
]["secretKeyRef"]["name"] = f"{cluster_name}-oauth-config"
228233
lower_meta["name"] = cluster_name
229234
lower_meta["namespace"] = namespace
230235

@@ -672,7 +677,9 @@ def generate_appwrapper(
672677
item = resources["resources"].get("GenericItems")[0]
673678
ingress_item = resources["resources"].get("GenericItems")[1]
674679
route_item = resources["resources"].get("GenericItems")[2]
675-
update_names(user_yaml, item, appwrapper_name, cluster_name, namespace)
680+
update_names(
681+
user_yaml, item, appwrapper_name, cluster_name, namespace, openshift_oauth
682+
)
676683
update_labels(user_yaml, instascale, instance_types)
677684
update_priority(user_yaml, item, dispatch_priority, priority_val)
678685
update_custompodresources(

src/codeflare_sdk/utils/openshift_oauth.py

Lines changed: 0 additions & 199 deletions
This file was deleted.

0 commit comments

Comments
 (0)