cvesearch.go

package main

import (
	"flag"
	"log"
	"os"
	"regexp"

	"github.com/wdahlenburg/CVESearch/utils"
	"github.com/wdahlenburg/CVESearch/utils/search"
)

func main() {
	var (
		cve         string
		verbose     bool
		description bool
	)
	flag.StringVar(&cve, "cve", "", "CVE to query")
	flag.BoolVar(&verbose, "v", false, "Enable verbose mode")
	flag.BoolVar(&description, "description", false, "List CVE description")
	flag.Parse()

	validateCVE(cve)

	if description {
		utils.PrintDescription(cve)
	}

	apiKeys := utils.ApiKeys{
		GitHub: os.Getenv("GITHUB_KEY"),
		GitLab: os.Getenv("GITLAB_KEY"),
	}

	search.New().Start(cve, apiKeys, verbose)
}

func validateCVE(cve string) {
	re := regexp.MustCompile(`^CVE-\d{4}-\d+$`)
	if !re.Match([]byte(cve)) {
		log.Fatal("Error. CVE must be in the format of CVE-20XX-XXXX")
	}
}