chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,238 @@
|
||||
package workflows
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"slices"
|
||||
|
||||
portainer "github.com/portainer/portainer/api"
|
||||
"github.com/portainer/portainer/api/dataservices"
|
||||
gittypes "github.com/portainer/portainer/api/git/types"
|
||||
"github.com/portainer/portainer/api/gitops/sources"
|
||||
"github.com/portainer/portainer/api/internal/endpointutils"
|
||||
"github.com/portainer/portainer/api/set"
|
||||
"github.com/portainer/portainer/api/slicesx"
|
||||
)
|
||||
|
||||
// BuildGroupEndpoints builds a map between EdgeGroup id and its endpoints
|
||||
func BuildGroupEndpoints(tx dataservices.DataStoreTx, groups []portainer.EdgeGroup) (map[portainer.EdgeGroupID][]portainer.EndpointID, error) {
|
||||
m := make(map[portainer.EdgeGroupID][]portainer.EndpointID, len(groups))
|
||||
for _, g := range groups {
|
||||
if g.Dynamic {
|
||||
ids, err := endpointutils.GetEndpointsByTags(tx, g.TagIDs, g.PartialMatch)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to resolve endpoints for dynamic edge group: %w", err)
|
||||
}
|
||||
m[g.ID] = ids
|
||||
} else {
|
||||
m[g.ID] = g.EndpointIDs.ToSlice()
|
||||
}
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
|
||||
// MapStackToWorkflow converts a stack to a Workflow. gitConfig is passed separately
|
||||
// because EE embeds a different GitConfig type that shadows the CE field.
|
||||
// source and artifact are the pre-computed git phase statuses from the caller.
|
||||
func MapStackToWorkflow(s portainer.Stack, sourceID portainer.SourceID, gitConfig *gittypes.RepoConfig, source, artifact WorkflowPhaseStatus) Workflow {
|
||||
return Workflow{
|
||||
ID: s.WorkflowID,
|
||||
Name: s.Name,
|
||||
Type: TypeStack,
|
||||
Platform: platformFromStackType(s.Type),
|
||||
Status: WorkflowStatusObject{
|
||||
Source: source,
|
||||
Artifact: artifact,
|
||||
Target: deriveStackTargetState(s),
|
||||
},
|
||||
SourceID: sourceID,
|
||||
GitConfig: gitConfig,
|
||||
AutoUpdate: s.AutoUpdate,
|
||||
Target: Target{
|
||||
EndpointID: s.EndpointID,
|
||||
Namespace: s.Namespace,
|
||||
},
|
||||
CreationDate: s.CreationDate,
|
||||
LastSyncDate: StackLastSyncDate(s),
|
||||
}
|
||||
}
|
||||
|
||||
// MapEdgeStackToWorkflow converts an edge stack to a Workflow. gitConfig is passed separately
|
||||
// because EE embeds a different GitConfig type that shadows the CE field.
|
||||
// source and artifact are the pre-computed git phase statuses from the caller.
|
||||
func MapEdgeStackToWorkflow(wfID portainer.WorkflowID, es portainer.EdgeStack, sourceID portainer.SourceID, gitConfig *gittypes.RepoConfig, statuses []portainer.EdgeStackStatusForEnv, groupEndpoints map[portainer.EdgeGroupID][]portainer.EndpointID, source, artifact WorkflowPhaseStatus) Workflow {
|
||||
platform := DeploymentPlatformDockerStandalone
|
||||
if es.DeploymentType == portainer.EdgeStackDeploymentKubernetes {
|
||||
platform = DeploymentPlatformKubernetes
|
||||
}
|
||||
return Workflow{
|
||||
ID: wfID,
|
||||
Name: es.Name,
|
||||
Type: TypeEdgeStack,
|
||||
Platform: platform,
|
||||
Status: WorkflowStatusObject{
|
||||
Source: source,
|
||||
Artifact: artifact,
|
||||
Target: deriveEdgeStackTargetState(statuses),
|
||||
},
|
||||
SourceID: sourceID,
|
||||
GitConfig: gitConfig,
|
||||
Target: Target{
|
||||
EdgeGroupIDs: es.EdgeGroups,
|
||||
GroupStatus: edgeStackTargetStatuses(es.EdgeGroups, statuses, groupEndpoints),
|
||||
ResolvedEndpointIDs: resolveEdgeGroupEndpoints(es.EdgeGroups, groupEndpoints),
|
||||
},
|
||||
CreationDate: es.CreationDate,
|
||||
LastSyncDate: edgeStackLastSyncDate(statuses),
|
||||
}
|
||||
}
|
||||
|
||||
// MapStackToArtifactDetail converts a stack to an ArtifactDetail. source and artifact are the
|
||||
// pre-computed git phase statuses from the caller; files are the artifact's resolved file refs.
|
||||
func MapStackToArtifactDetail(stack portainer.Stack, files []portainer.ArtifactFile, source, artifact WorkflowPhaseStatus) ArtifactDetail {
|
||||
return ArtifactDetail{
|
||||
ID: int(stack.ID),
|
||||
Type: TypeStack,
|
||||
Name: stack.Name,
|
||||
Platform: platformFromStackType(stack.Type),
|
||||
Status: WorkflowStatusObject{
|
||||
Source: source,
|
||||
Artifact: artifact,
|
||||
Target: deriveStackTargetState(stack),
|
||||
},
|
||||
AutoUpdate: stack.AutoUpdate,
|
||||
Target: Target{
|
||||
EndpointID: stack.EndpointID,
|
||||
Namespace: stack.Namespace,
|
||||
},
|
||||
Files: mapFilesToFileDetails(files),
|
||||
CreationDate: stack.CreationDate,
|
||||
LastSyncDate: StackLastSyncDate(stack),
|
||||
}
|
||||
}
|
||||
|
||||
// MapEdgeStackToArtifactDetail converts an edge stack to an ArtifactDetail. source and artifact are
|
||||
// the pre-computed git phase statuses from the caller; files are the artifact's resolved file refs.
|
||||
func MapEdgeStackToArtifactDetail(es portainer.EdgeStack, files []portainer.ArtifactFile, statuses []portainer.EdgeStackStatusForEnv, groupEndpoints map[portainer.EdgeGroupID][]portainer.EndpointID, source, artifact WorkflowPhaseStatus) ArtifactDetail {
|
||||
platform := DeploymentPlatformDockerStandalone
|
||||
if es.DeploymentType == portainer.EdgeStackDeploymentKubernetes {
|
||||
platform = DeploymentPlatformKubernetes
|
||||
}
|
||||
return ArtifactDetail{
|
||||
ID: int(es.ID),
|
||||
Type: TypeEdgeStack,
|
||||
Name: es.Name,
|
||||
Platform: platform,
|
||||
Status: WorkflowStatusObject{
|
||||
Source: source,
|
||||
Artifact: artifact,
|
||||
Target: deriveEdgeStackTargetState(statuses),
|
||||
},
|
||||
Target: Target{
|
||||
EdgeGroupIDs: es.EdgeGroups,
|
||||
GroupStatus: edgeStackTargetStatuses(es.EdgeGroups, statuses, groupEndpoints),
|
||||
ResolvedEndpointIDs: resolveEdgeGroupEndpoints(es.EdgeGroups, groupEndpoints),
|
||||
},
|
||||
Files: mapFilesToFileDetails(files),
|
||||
CreationDate: es.CreationDate,
|
||||
LastSyncDate: edgeStackLastSyncDate(statuses),
|
||||
}
|
||||
}
|
||||
|
||||
func StackLastSyncDate(s portainer.Stack) int64 {
|
||||
for _, ds := range slices.Backward(s.DeploymentStatus) {
|
||||
if ds.Status == portainer.StackStatusActive {
|
||||
return ds.Time
|
||||
}
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func edgeStackLastSyncDate(statuses []portainer.EdgeStackStatusForEnv) int64 {
|
||||
var oldest int64
|
||||
for _, epStatus := range statuses {
|
||||
last := endpointLastSyncDate(epStatus)
|
||||
if last == 0 {
|
||||
return 0
|
||||
}
|
||||
if oldest == 0 || last < oldest {
|
||||
oldest = last
|
||||
}
|
||||
}
|
||||
return oldest
|
||||
}
|
||||
|
||||
func endpointLastSyncDate(epStatus portainer.EdgeStackStatusForEnv) int64 {
|
||||
for _, s := range slices.Backward(epStatus.Status) {
|
||||
if isEdgeStackHealthyStatus(s.Type) {
|
||||
return s.Time
|
||||
}
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func platformFromStackType(t portainer.StackType) DeploymentPlatform {
|
||||
switch t {
|
||||
case portainer.KubernetesStack:
|
||||
return DeploymentPlatformKubernetes
|
||||
case portainer.DockerSwarmStack:
|
||||
return DeploymentPlatformDockerSwarm
|
||||
default:
|
||||
return DeploymentPlatformDockerStandalone
|
||||
}
|
||||
}
|
||||
|
||||
func isEdgeStackHealthyStatus(t portainer.EdgeStackStatusType) bool {
|
||||
switch t {
|
||||
case portainer.EdgeStackStatusRunning,
|
||||
portainer.EdgeStackStatusRolledBack,
|
||||
portainer.EdgeStackStatusCompleted,
|
||||
portainer.EdgeStackStatusRemoved,
|
||||
portainer.EdgeStackStatusRemoteUpdateSuccess:
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func resolveEdgeGroupEndpoints(groups []portainer.EdgeGroupID, groupEndpoints map[portainer.EdgeGroupID][]portainer.EndpointID) []portainer.EndpointID {
|
||||
seen := set.Set[portainer.EndpointID]{}
|
||||
for _, gid := range groups {
|
||||
for _, epID := range groupEndpoints[gid] {
|
||||
seen.Add(epID)
|
||||
}
|
||||
}
|
||||
return seen.Keys()
|
||||
}
|
||||
|
||||
func edgeStackTargetStatuses(
|
||||
groups []portainer.EdgeGroupID,
|
||||
statuses []portainer.EdgeStackStatusForEnv,
|
||||
groupEndpoints map[portainer.EdgeGroupID][]portainer.EndpointID,
|
||||
) map[portainer.EdgeGroupID]Status {
|
||||
epMap := make(map[portainer.EndpointID]Status, len(statuses))
|
||||
for _, s := range statuses {
|
||||
ws, _ := endpointWorkflowStatus(s)
|
||||
epMap[s.EndpointID] = ws
|
||||
}
|
||||
|
||||
result := make(map[portainer.EdgeGroupID]Status, len(groups))
|
||||
for _, gid := range groups {
|
||||
gStatus := StatusUnknown
|
||||
for _, epID := range groupEndpoints[gid] {
|
||||
if ws := epMap[epID]; statusPriority(ws) > statusPriority(gStatus) {
|
||||
gStatus = ws
|
||||
}
|
||||
}
|
||||
result[gid] = gStatus
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func mapFilesToFileDetails(files []portainer.ArtifactFile) []ArtifactFileDetail {
|
||||
return slicesx.Map(files, func(file portainer.ArtifactFile) ArtifactFileDetail {
|
||||
return ArtifactFileDetail{
|
||||
ArtifactFile: file,
|
||||
RefStatus: sources.StatusString(file.RefStatus),
|
||||
PathStatus: sources.StatusString(file.PathStatus),
|
||||
}
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user