package cmd

import (
	"encoding/json"
	"fmt"
	"os"

	"github.com/anchore/imgbom/imgbom"
	"github.com/anchore/imgbom/imgbom/presenter"
	"github.com/anchore/imgbom/internal"
	"github.com/anchore/imgbom/internal/log"
	"github.com/anchore/stereoscope"
	"github.com/spf13/cobra"
)

var rootCmd = &cobra.Command{
	Use:   fmt.Sprintf("%s [IMAGE]", internal.ApplicationName),
	Short: "A container image BOM tool", // TODO: add copy
	Long: internal.Tprintf(`\
Supports the following image sources:
    {{.appName}} yourrepo/yourimage:tag             defaults to using images from a docker daemon
    {{.appName}} docker://yourrepo/yourimage:tag    explicitly use the docker daemon
    {{.appName}} tar://path/to/yourimage.tar        use a tarball from disk
`, map[string]interface{}{
		"appName": internal.ApplicationName,
	}),
	Args: cobra.MaximumNArgs(1),
	Run:  doRunCmd,
}

func init() {
	setCliOptions()

	cobra.OnInitialize(loadAppConfig)
	cobra.OnInitialize(setupLoggingFromAppConfig)
}

func Execute() {
	if err := rootCmd.Execute(); err != nil {
		log.Errorf("could not start application: %w", err)
		os.Exit(1)
	}
}

func doRunCmd(cmd *cobra.Command, args []string) {
	appCfgStr, err := json.MarshalIndent(&appConfig, "  ", "  ")
	if err != nil {
		log.Debugf("could not display application config: %+v", err)
	} else {
		log.Debugf("application config:\n%+v", string(appCfgStr))
	}

	userImageStr := args[0]
	log.Infof("fetching image %s...", userImageStr)
	img, err := stereoscope.GetImage(userImageStr)
	if err != nil {
		log.Errorf("could not fetch image '%s': %w", userImageStr, err)
		os.Exit(1)
	}
	defer stereoscope.Cleanup()

	log.Info("cataloging image...")
	catalog, err := imgbom.CatalogImage(img, appConfig.ScopeOpt)
	if err != nil {
		log.Errorf("could not catalog image: %w", err)
		os.Exit(1)
	}

	log.Info("done!")
	err = presenter.GetPresenter(appConfig.PresenterOpt).Present(os.Stdout, img, catalog)
	if err != nil {
		log.Errorf("could not format catalog results: %w", err)
		os.Exit(1)
	}
}