package main

import (
	"fmt"
	"gopkg.in/yaml.v2"
	"log"
	"os"
	"os/exec"
)

type ObjectSpec struct {
	Kind string `yaml:"kind"`
	Name string `yaml:"name"`
}

func main() {
	if isDelAllCmd() {
		execDelAllCmd()
	} else if isApplyCmd() {
		execDelAllCmd()
		execOriginCmd([]string{"object", "create", "-f", os.Args[len(os.Args)-1]})
	} else {
		execOriginCmd(os.Args[1:])
	}
}

func isApplyCmd() bool {
	//egctl+ object apply -f file
	return len(os.Args) == 5 && os.Args[1] == "object" && os.Args[2] == "apply" && os.Args[3] == "-f" && os.Args[4] != ""
}

func isDelAllCmd() bool {
	//egctl+ object delete --all
	return len(os.Args) == 4 && os.Args[1] == "object" && os.Args[2] == "delete" && os.Args[3] == "--all"
}

func execOriginCmd(args []string) {
	cmd := exec.Command("egctl", args...)
	output, err := cmd.Output()
	if err != nil {
		log.Fatalf("execute cmd %s %s err %s", cmd.Path, cmd.Args, err)
	}
	fmt.Print(string(output))
}

func execDelAllCmd() {
	cmd := exec.Command("egctl", "object", "list")
	output, err := cmd.Output()
	if err != nil {
		log.Fatalf("execute cmd output: %s", err)
	}
	var objSpecs []*ObjectSpec
	err = yaml.Unmarshal(output, &objSpecs)
	if err != nil {
		log.Fatalf("unmarshal object specs failure: %s", err)
	}
	var cmds []*exec.Cmd
	for _, spec := range objSpecs {
		cmds = append(cmds, exec.Command("egctl", "object", "delete", spec.Name))
	}
	for _, cmd := range cmds {
		_, err := cmd.Output()
		if err != nil {
			log.Fatalf("execute cmd %s %s err %s", cmd.Path, cmd.Args, err)
		}
	}
}