main.go

package main

import (
	"fmt"
	"os"
	"scan2html/internal/common"
	"scan2html/internal/logger"
	"scan2html/internal/report"
	"scan2html/internal/trivy"
)

var (
	version = "dev"
)

func main() {
	logger.Logger.Info("Scan2html started")

	if common.IsHelp() {
		helpMessage()
	}

	pluginFlags, trivyCommand := common.RetrievePluginFlagsAndCommand(os.Args)

	var exitCode = 0
	if _, exists := pluginFlags["generate"]; exists {
		if err := report.CombineReports(pluginFlags); err != nil {
			fmt.Printf("Error: %v\n", err)
			os.Exit(1)
		}
	} else {
		exitCode, _ = trivy.GenerateJsonReport(trivyCommand)
	}

	err := report.GenerateHtmlReport(pluginFlags, version)
	if err != nil {
		logger.Logger.Fatalf("Error generating HTML report: %v", err)
	}

	os.Exit(exitCode)
}

func helpMessage() {
	common.PrintHelp(version)
	os.Exit(0)
}