jx/cmd/cli/main.go

// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.

package main

import (
	"context"
	"io"
  "os"
  "flag"
  "log"
	"log/slog"
_	"errors"
	"fmt"
_ "gopkg.in/yaml.v3"
  "decl/internal/resource"
  "decl/internal/source"
)

const (
	FormatYaml = "yaml"
	FormatJson = "json"
)

var GlobalOformat *string
var GlobalQuiet *bool

var ImportMerge *bool
var ImportResource *string


var ctx context.Context = context.Background()

type RunCommand func(cmd *flag.FlagSet, output io.Writer) error

type SubCommand struct {
	Name string
	Run RunCommand
}

var jxSubCommands = []SubCommand {
	{
		Name: "diff",
		Run: DiffSubCommand,
	},
	{
		Name: "apply",
		Run: ApplySubCommand,
	},
	{
		Name: "import",
		Run: ImportSubCommand,
	},
}

func LoggerConfig() {
	var programLevel = new(slog.LevelVar)
	logger := slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: programLevel}))
	slog.SetDefault(logger)
	if debugLogging,ok := os.LookupEnv("DECL_DEBUG"); ok && debugLogging != "" {
		programLevel.Set(slog.LevelDebug)
	} else {
		programLevel.Set(slog.LevelError)
	}
}

func LoadSourceURI(uri string) []*resource.Document {
	slog.Info("loading ", "uri", uri)
	if uri != "" {
		ds, err := source.SourceTypes.New(uri)
		if err != nil {
			log.Fatal(err)
		}
		extractDocuments, extractErr := ds.ExtractResources(nil)
		if extractErr != nil {
			log.Fatal(extractErr)
		}
		slog.Info("extract documents", "documents", extractDocuments)
		return extractDocuments
	}
	return []*resource.Document{ resource.NewDocument() }
}

func ImportSubCommand(cmd *flag.FlagSet, output io.Writer) (err error) {
	ImportResource = cmd.String("resource", "", "(uri) Add a resource to the document.")
	ImportMerge = cmd.Bool("merge", false, "Merge resources into a single document.")
	cmd.Parse(os.Args[2:])
	var encoder resource.Encoder
	merged := resource.NewDocument()
	documents := make([]*resource.Document, 0, 100)
	for _,source := range cmd.Args() {
		loaded :=  LoadSourceURI(source)
		if loaded != nil {
			documents = append(documents, loaded...)
		}
	}
	
	switch *GlobalOformat {
	case FormatYaml:
		encoder = resource.NewYAMLEncoder(output)
	case FormatJson:
		encoder = resource.NewJSONEncoder(output)
	}

	if len(documents) == 0 {
		documents = append(documents, resource.NewDocument())
	}

	for _,d := range documents {
		if d != nil {

			if *ImportResource != "" {
				slog.Info("ImportResource", "resource", ImportResource)
				if addResourceErr := d.AddResource(*ImportResource); addResourceErr != nil {
					log.Fatal(addResourceErr)
				}
			}

			if *GlobalQuiet {
				for _, dr := range d.Resources() {
					output.Write([]byte(dr.Resource().URI()))
				}
			} else {
				if *ImportMerge {
					merged.ResourceDecls = append(merged.ResourceDecls, d.ResourceDecls...)
					slog.Info("merging", "doc",  merged.ResourceDecls, "src", d.ResourceDecls)
				} else {
					if documentGenerateErr := encoder.Encode(d); documentGenerateErr != nil {
						return documentGenerateErr
					}
				}
			}
		}
	}
	if *ImportMerge {
		if documentGenerateErr := encoder.Encode(merged); documentGenerateErr != nil {
			return documentGenerateErr
		}
	}
	return err
}

func ApplySubCommand(cmd *flag.FlagSet, output io.Writer) (err error) {
	cmd.Parse(os.Args[2:])
	var encoder resource.Encoder
	documents := make([]*resource.Document, 0, 100)
	for _,source := range cmd.Args() {
		loaded :=  LoadSourceURI(source)
		if loaded != nil {
			documents = append(documents, loaded...)
		}
	}

	slog.Info("main.Apply()", "documents", documents)
	for _,d := range documents {
		slog.Info("main.Appl()", "doc", d)
		if e := d.Apply(); e != nil {
			return e
		}

		switch *GlobalOformat {
		case FormatYaml:
			encoder = resource.NewYAMLEncoder(output)
		case FormatJson:
			encoder = resource.NewJSONEncoder(output)
		}
		if *GlobalQuiet {
			for _, dr := range d.Resources() {
				output.Write([]byte(dr.Resource().URI()))
			}
		} else {
			if documentGenerateErr := encoder.Encode(d); documentGenerateErr != nil {
				return documentGenerateErr
			}
		}
	}
	return err
}

func DiffSubCommand(cmd *flag.FlagSet, output io.Writer) (err error) {
	cmd.Parse(os.Args[2:])
	leftSource := cmd.Arg(0)
	rightSource := cmd.Arg(1)
	leftDocuments := make([]*resource.Document, 0, 100)
	rightDocuments := make([]*resource.Document, 0, 100)

	slog.Info("jx diff subcommand", "left", leftSource, "right", rightSource, "flagset", cmd)
	leftDocuments = append(leftDocuments, LoadSourceURI(leftSource)...)

	if rightSource == "" {
		slog.Info("jx diff clone", "docs", leftDocuments)
		for i, doc := range leftDocuments {
			if doc != nil {
				rightDocuments = append(rightDocuments, doc.Clone())
				for _,resourceDeclaration := range leftDocuments[i].Resources() {
					if _, e := resourceDeclaration.Resource().Read(ctx); e != nil {
						return e
					}
				}
			}
		}
	} else {
		rightDocuments = append(rightDocuments, LoadSourceURI(rightSource)...)
	}

	index := 0
	for {
		if index >= len(rightDocuments) && index >= len(leftDocuments) {
			break
		}
		if index >= len(rightDocuments) {
			if _,e := leftDocuments[index].Diff(resource.NewDocument(), output); e != nil {
				return e
			}
			index++
			continue
		}
		if index >= len(leftDocuments) {
			if _,e := resource.NewDocument().Diff(rightDocuments[index], output); e != nil {
				return e
			}
			index++
			continue
		}
		if _,e := leftDocuments[index].Diff(rightDocuments[index], output); e != nil {
			return e
		}
		index++
	}
	return err
}

func main() {

	LoggerConfig()

	if len(os.Args) < 2 {
		fmt.Println("expected subcommands: diff, apply, import")
		os.Exit(1)
	}

	for _,subCmd := range jxSubCommands {
		cmdFlagSet := flag.NewFlagSet(subCmd.Name, flag.ExitOnError)
		GlobalOformat = cmdFlagSet.String("oformat", "yaml", "Output serialization format")
		GlobalQuiet = cmdFlagSet.Bool("quiet", false, "Generate terse output.")

		switch subCmd.Name {
		case "diff":
			cmdFlagSet.Usage = func() {
				fmt.Println("jx diff source [source2]")
				cmdFlagSet.PrintDefaults()
			}
		case "apply":
			cmdFlagSet.Usage = func() {
				fmt.Println("jx diff source [source2]")
				cmdFlagSet.PrintDefaults()
			}
		case "import":
			cmdFlagSet.Usage = func() {
				fmt.Println("jx import source [source2]")
				cmdFlagSet.PrintDefaults()
			}
		}
		slog.Info("command", "command", subCmd)
		if os.Args[1] == subCmd.Name {
			subCmd.Run(cmdFlagSet, os.Stdout)
			return
		}
	}

}