build: add build tags to create customizable binaries

pkg/extensions/extension.go

@@ -0,0 +1,61 @@
+// +build extended
+
+package extensions
+
+import (
+	"github.com/anuvu/zot/pkg/extensions/search"
+	"github.com/anuvu/zot/pkg/storage"
+	"github.com/gorilla/mux"
+
+	"time"
+
+	gqlHandler "github.com/99designs/gqlgen/graphql/handler"
+	cveinfo "github.com/anuvu/zot/pkg/extensions/search/cve"
+
+	"github.com/anuvu/zot/pkg/log"
+)
+
+// DownloadTrivyDB ...
+func DownloadTrivyDB(dbDir string, log log.Logger, updateInterval time.Duration) error {
+	for {
+		log.Info().Msg("Updating the CVE database")
+
+		err := cveinfo.UpdateCVEDb(dbDir, log)
+		if err != nil {
+			return err
+		}
+
+		log.Info().Str("Db update completed, next update scheduled after", updateInterval.String()).Msg("")
+
+		time.Sleep(updateInterval)
+	}
+}
+
+func EnableExtension(extension *ExtensionConfig, log log.Logger, rootDir string) {
+	if extension != nil && extension.Search != nil &&
+		extension.Search.CVE != nil {
+		defaultUpdateInterval, _ := time.ParseDuration("2h")
+
+		if extension.Search.CVE.UpdateInterval < defaultUpdateInterval {
+			extension.Search.CVE.UpdateInterval = defaultUpdateInterval
+
+			log.Warn().Msg("CVE update interval set to too-short interval <= 1, changing update duration to 2 hours and continuing.") // nolint: lll
+		}
+
+		go func() {
+			err := DownloadTrivyDB(rootDir, log,
+				extension.Search.CVE.UpdateInterval)
+			if err != nil {
+				panic(err)
+			}
+		}()
+	} else {
+		log.Info().Msg("Cve config not provided, skipping cve update")
+	}
+}
+
+func SetupRoutes(router *mux.Router, rootDir string, imgStore *storage.ImageStore, log log.Logger) {
+	resConfig := search.GetResolverConfig(rootDir, log, imgStore)
+	router.PathPrefix("/query").Methods("GET", "POST").
+		Handler(gqlHandler.NewDefaultServer(search.NewExecutableSchema(resConfig)))
+}