package models

import (
	"strings"

	w "cloud.o-forge.io/core/oc-lib/models/workflow"
)

type WorkflowsDependancies struct {
	FirstWfTasks   map[string][]string
	RelatedWfTasks map[string][]string
	LastWfTasks    map[string][]string
}

func NewWorkflowDependancies() *WorkflowsDependancies {
	return &WorkflowsDependancies{
		FirstWfTasks:   map[string][]string{},
		RelatedWfTasks: map[string][]string{},
		LastWfTasks:    map[string][]string{},
	}
}

func (w *WorkflowsDependancies) BindFirstTasks(depsFunc func(v string) []w.Deps, dag *Dag) {
	for wfID, firstTasks := range w.FirstWfTasks {
		deps := depsFunc(wfID)
		if task := dag.GetTask(wfID); task != nil && len(deps) > 0 {
			task.Dependencies = append(task.Dependencies, firstTasks...)
		}
	}
}

func (w *WorkflowsDependancies) BindRelatedTasks(dag *Dag) {
	for wfID, relatedWfTasks := range w.RelatedWfTasks {
		for _, dep := range relatedWfTasks {
			if task := dag.GetTask(dep); task != nil {
				index := -1
				if i, deps := task.GetDeps(wfID); deps != "" {
					index = i
				}
				if index != -1 {
					task.Dependencies = append(task.Dependencies[:index], task.Dependencies[index+1:]...)
				}
				if w.LastWfTasks[wfID] != nil {
					task.Dependencies = append(task.Dependencies, w.LastWfTasks[wfID]...)
				}
			}
		}
	}
}

type Workflow struct {
	ApiVersion string `yaml:"apiVersion"`
	Kind       string `yaml:"kind"`
	Metadata   struct {
		Name string `yaml:"name"`
	} `yaml:"metadata"`
	Spec Spec `yaml:"spec,omitempty"`
}

func (b *Workflow) GetDag() *Dag {
	for _, t := range b.Spec.Templates {
		if t.Name == "dag" {
			return t.Dag
		}
	}
	b.Spec.Templates = append(b.Spec.Templates, Template{Name: "dag", Dag: &Dag{}})
	return b.Spec.Templates[len(b.Spec.Templates)-1].Dag
}

type Spec struct {
	ServiceAccountName string                `yaml:"serviceAccountName"`
	Entrypoint         string                `yaml:"entrypoint"`
	Arguments          []Parameter           `yaml:"arguments,omitempty"`
	Volumes            []VolumeClaimTemplate `yaml:"volumeClaimTemplates,omitempty"`
	Templates          []Template            `yaml:"templates"`
	Timeout            int                   `yaml:"activeDeadlineSeconds,omitempty"`
}

func GetArgoName(raw_name string, component_id string) (formatedName string) {
	formatedName = strings.ReplaceAll(raw_name, " ", "-")
	formatedName += "-" + component_id
	formatedName = strings.ToLower(formatedName)
	return
}

func TransformDepsToArgo(deps []w.Deps) []string {
	argoDeps := []string{}
	for _, dep := range deps {
		argoDeps = append(argoDeps, GetArgoName(dep.Source, dep.Dest))
	}
	return argoDeps
}