Skip to content

Commit 909fa48

Browse files
committed
[wsman-mk2] Remove secret from initializer
1 parent 662589c commit 909fa48

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

components/ws-manager-mk2/service/manager.go

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -110,11 +110,6 @@ func (wsm *WorkspaceManagerServer) StartWorkspace(ctx context.Context, req *wsma
110110
return nil, status.Errorf(codes.InvalidArgument, "unsupported workspace type: %v", req.Type)
111111
}
112112

113-
initializer, err := proto.Marshal(req.Spec.Initializer)
114-
if err != nil {
115-
return nil, status.Errorf(codes.InvalidArgument, "cannot serialise content initializer: %v", err)
116-
}
117-
118113
var git *workspacev1.GitSpec
119114
if req.Spec.Git != nil {
120115
git = &workspacev1.GitSpec{
@@ -198,7 +193,11 @@ func (wsm *WorkspaceManagerServer) StartWorkspace(ctx context.Context, req *wsma
198193
userEnvVars, envData := extractWorkspaceUserEnv(envSecretName, req.Spec.Envvars, req.Spec.SysEnvvars)
199194
sysEnvVars := extractWorkspaceSysEnv(req.Spec.SysEnvvars)
200195

201-
tokenData, _ := extractWorkspaceTokenData(req.Spec)
196+
tokenData := extractWorkspaceTokenData(req.Spec)
197+
initializer, err := proto.Marshal(req.Spec.Initializer)
198+
if err != nil {
199+
return nil, status.Errorf(codes.InvalidArgument, "cannot serialise content initializer: %v", err)
200+
}
202201

203202
ws := workspacev1.Workspace{
204203
TypeMeta: metav1.TypeMeta{
@@ -850,13 +849,12 @@ func extractWorkspaceSysEnv(sysEnvs []*wsmanapi.EnvironmentVariable) []corev1.En
850849
return envs
851850
}
852851

853-
func extractWorkspaceTokenData(spec *wsmanapi.StartWorkspaceSpec) (secrets map[string]string, secretsLen int) {
854-
secrets = make(map[string]string)
855-
for k, v := range csapi.GatherSecretsFromInitializer(spec.Initializer) {
852+
func extractWorkspaceTokenData(spec *wsmanapi.StartWorkspaceSpec) map[string]string {
853+
secrets := make(map[string]string)
854+
for k, v := range csapi.ExtractAndReplaceSecretsFromInitializer(spec.Initializer) {
856855
secrets[k] = v
857-
secretsLen += len(v)
858856
}
859-
return secrets, secretsLen
857+
return secrets
860858
}
861859

862860
func extractWorkspaceStatus(ws *workspacev1.Workspace) *wsmanapi.WorkspaceStatus {

0 commit comments

Comments
 (0)