argo/approve_former/flow.go

package approve_former

import "strings"

const flowTemplate = `apiVersion: argoproj.io/v1alpha1
kind: WorkflowTemplate
metadata:
  name: [[ .name ]]
spec:
  entrypoint: approve-flow
  templates:
    - name: approve-flow
      dag:
        tasks:
          [[- if .childNodesContentLines ]]
          [[- range .childNodesContentLines ]]
          [[ . ]]
          [[- end ]]
          [[- else ]]
          - name: node
            template: approval
          [[- end ]]
          
    - name: approval
      suspend: {}
      inputs:
          parameters:
            - name: approve
              default: '通过'
              enum:
                  - '通过'
                  - '拒绝'
      outputs:
          parameters:
            - name: approve
              valueFrom:
                  supplied: {}
`

type Flow struct {
	Name       string
	ChildNodes []Node
}

func NewFlow(name string, childNodes ...Node) *Flow {
	return &Flow{
		Name:       name,
		ChildNodes: childNodes,
	}
}

func (flow *Flow) GetName() string {
	return flow.Name
}

func (flow *Flow) GetChildNodes() []Node {
	return flow.ChildNodes
}

func (flow *Flow) Render(_ ...Node) (string, error) {
	childNodesContents := make([]string, 0)
	for i, childNode := range flow.ChildNodes {
		var childNodeContent string

		if i == 0 {
			content, err := childNode.Render()
			if err != nil {
				return "", err
			}

			childNodeContent = content
		} else {
			upperNodes := make([]Node, 0)
			upperChildNodes := flow.ChildNodes[i-1].GetChildNodes()
			if len(upperChildNodes) == 0 {
				upperNodes = append(upperNodes, flow.ChildNodes[i-1])
			} else {
				upperNodes = append(upperNodes, upperChildNodes...)
			}

			content, err := childNode.Render(upperNodes...)
			if err != nil {
				return "", err
			}

			childNodeContent = content
		}

		childNodesContents = append(childNodesContents, strings.Split(childNodeContent, "\n")...)
	}

	return render("flow", flowTemplate, map[string]any{
		"name":                   flow.Name,
		"childNodesContentLines": childNodesContents,
	})
}