main.go

package main

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

	"github.com/s-index/go-cve-search/nvd"
	"github.com/urfave/cli/v2"
)

var (
	version = "0.1.0"
)

func main() {
	app := &cli.App{
		Name:  "go-cve-search",
		Usage: "search CVE",
		Flags: []cli.Flag{
			&cli.BoolFlag{
				Name:    "version",
				Aliases: []string{"v"},
				Usage:   "Show version",
			},
			&cli.BoolFlag{
				Name:    "pretty",
				Aliases: []string{"p"},
				Usage:   "Pretty Print Json",
			},
		},
		Action: func(c *cli.Context) error {
			if c.Bool("version") {
				fmt.Println("go-cve-search v" + version)
				return nil
			}
			nvd, err := nvd.FetchNVD(c.Args().Get(0))
			if err != nil {
				return err
			}
			if c.Bool("pretty") {
				prettyNvd, _ := json.MarshalIndent(nvd, "", "   ")
				fmt.Println(string(prettyNvd))
				return nil
			}
			fmt.Println(nvd)
			return nil
		},
	}
	err := app.Run(os.Args)
	if err != nil {
		log.Fatal(err)
	}
}