Skip to content

Commit 117261b

Browse files
committed
Provide opaque token by image check
1 parent 514b754 commit 117261b

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

components/renku_data_services/notebooks/core_sessions.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -531,10 +531,8 @@ async def __requires_image_pull_secret(nb_config: NotebooksConfig, image: str, i
531531
return False
532532

533533

534-
def __format_image_pull_secret(secret_name: str, access_token: str, registry_domain: str) -> ExtraSecret:
535-
registry_secret = {
536-
"auths": {registry_domain: {"auth": base64.b64encode(f"oauth2:{access_token}".encode()).decode()}}
537-
}
534+
def __format_image_pull_secret(secret_name: str, token: str, registry_domain: str) -> ExtraSecret:
535+
registry_secret = {"auths": {registry_domain: {"auth": token}}}
538536
registry_secret = json.dumps(registry_secret)
539537
registry_secret = base64.b64encode(registry_secret.encode()).decode()
540538
return ExtraSecret(
@@ -561,7 +559,7 @@ async def __get_gitlab_image_pull_secret_v2(
561559

562560
return __format_image_pull_secret(
563561
secret_name=secret_name,
564-
access_token=image_check_result.token,
562+
token=image_check_result.token,
565563
registry_domain=image_check_result.image_provider.registry_url,
566564
)
567565

components/renku_data_services/notebooks/image_check.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414

1515
from __future__ import annotations
1616

17+
import base64
1718
from dataclasses import dataclass, field
1819
from typing import final
1920

@@ -22,7 +23,7 @@
2223
from renku_data_services.app_config import logging
2324
from renku_data_services.base_models.core import APIUser
2425
from renku_data_services.connected_services.db import ConnectedServicesRepository
25-
from renku_data_services.connected_services.models import ImageProvider, OAuth2Client, OAuth2Connection
26+
from renku_data_services.connected_services.models import ImageProvider, OAuth2Client, OAuth2Connection, ProviderKind
2627
from renku_data_services.errors import errors
2728
from renku_data_services.notebooks.api.classes.image import Image, ImageRepoDockerAPI
2829

@@ -111,10 +112,13 @@ async def check_image(image: Image, user: APIUser, connected_services: Connected
111112
logger.info(f"Error getting connected account: {e}")
112113
unauth_error = e
113114

115+
token = f"{reg_api.username}:{reg_api.oauth2_token}"
116+
token = base64.b64encode(token.encode()).decode()
117+
114118
return CheckResult(
115119
accessible=result == 200,
116120
response_code=result,
117121
image_provider=image_provider,
118-
token=reg_api.oauth2_token,
122+
token=token,
119123
error=unauth_error,
120124
)

0 commit comments

Comments
 (0)