mirror of
https://github.com/kubevela/kubevela.git
synced 2026-05-20 16:23:24 +00:00
119 lines
3.7 KiB
Go
119 lines
3.7 KiB
Go
/*
|
|
Copyright 2021 The KubeVela Authors.
|
|
|
|
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.
|
|
*/
|
|
|
|
package process
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/kubevela/workflow/pkg/cue/process"
|
|
|
|
"github.com/oam-dev/kubevela/apis/core.oam.dev/common"
|
|
"github.com/oam-dev/kubevela/apis/types"
|
|
"github.com/oam-dev/kubevela/pkg/oam/util"
|
|
)
|
|
|
|
// ContextData is the core data of process context
|
|
type ContextData struct {
|
|
Namespace string
|
|
Cluster string
|
|
AppName string
|
|
CompName string
|
|
StepName string
|
|
CompRevision string
|
|
AppRevisionName string
|
|
WorkflowName string
|
|
PublishVersion string
|
|
ReplicaKey string
|
|
|
|
Ctx context.Context
|
|
BaseHooks []process.BaseHook
|
|
AuxiliaryHooks []process.AuxiliaryHook
|
|
Components []common.ApplicationComponent
|
|
|
|
AppLabels map[string]string
|
|
AppAnnotations map[string]string
|
|
|
|
ClusterVersion types.ClusterVersion
|
|
Output interface{}
|
|
}
|
|
|
|
// policyAdditionalContextKeyString is the string key for policy additionalContext in Go context
|
|
// We use a string key to avoid type mismatches across packages
|
|
const policyAdditionalContextKeyString = "kubevela.oam.dev/policy-additional-context"
|
|
|
|
// NewContext creates a new process context
|
|
func NewContext(data ContextData) process.Context {
|
|
// Extract policy additionalContext from Go context if it exists
|
|
// This allows Application-scoped policies to inject data into component/trait rendering
|
|
var customData map[string]interface{}
|
|
if data.Ctx != nil {
|
|
if val := data.Ctx.Value(policyAdditionalContextKeyString); val != nil {
|
|
if contextMap, ok := val.(map[string]interface{}); ok {
|
|
// Wrap additionalContext under "custom" key so it's accessible as context.custom
|
|
customData = map[string]interface{}{
|
|
"custom": contextMap,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
ctx := process.NewContext(process.ContextData{
|
|
Namespace: data.Namespace,
|
|
Name: data.CompName,
|
|
StepName: data.StepName,
|
|
WorkflowName: data.WorkflowName,
|
|
PublishVersion: data.PublishVersion,
|
|
Ctx: data.Ctx,
|
|
BaseHooks: data.BaseHooks,
|
|
AuxiliaryHooks: data.AuxiliaryHooks,
|
|
CustomData: customData,
|
|
})
|
|
ctx.PushData(ContextAppName, data.AppName)
|
|
ctx.PushData(ContextAppRevision, data.AppRevisionName)
|
|
ctx.PushData(ContextCompRevisionName, data.CompRevision)
|
|
ctx.PushData(ContextComponents, data.Components)
|
|
ctx.PushData(ContextAppLabels, data.AppLabels)
|
|
ctx.PushData(ContextAppAnnotations, data.AppAnnotations)
|
|
ctx.PushData(ContextReplicaKey, data.ReplicaKey)
|
|
revNum, _ := util.ExtractRevisionNum(data.AppRevisionName, "-")
|
|
ctx.PushData(ContextAppRevisionNum, revNum)
|
|
ctx.PushData(ContextCluster, data.Cluster)
|
|
ctx.PushData(ContextClusterVersion, parseClusterVersion(data.ClusterVersion))
|
|
if data.Output != nil {
|
|
ctx.PushData(OutputFieldName, data.Output)
|
|
}
|
|
return ctx
|
|
}
|
|
|
|
func parseClusterVersion(cv types.ClusterVersion) map[string]interface{} {
|
|
// no minor found, use control plane cluster version instead.
|
|
if cv.Minor == "" {
|
|
cv = types.ControlPlaneClusterVersion
|
|
}
|
|
minorS := strings.TrimSpace(cv.Minor)
|
|
minorS = strings.TrimRight(minorS, ".+-/?!")
|
|
minor, _ := strconv.ParseInt(minorS, 10, 64)
|
|
return map[string]interface{}{
|
|
"major": cv.Major,
|
|
"gitVersion": cv.GitVersion,
|
|
"platform": cv.Platform,
|
|
"minor": minor,
|
|
}
|
|
}
|