feat: support artifacts search

cncf-tags/wg-artifacts#34 An experiment to make this possible Signed-off-by: Ramkumar Chinchani <rchincha@cisco.com>
project-zot · Jan 29, 2024 · cb5b60e · cb5b60e
1 parent 9def35f
commit cb5b60e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/pkg/api/constants/extensions.go b/pkg/api/constants/extensions.go
@@ -16,6 +16,14 @@ const (
 	ExtSearchPrefix  = ExtPrefix + ExtSearch
 	FullSearchPrefix = RoutePrefix + ExtSearchPrefix
 
+	// CNCF artifacts search extension.
+	BaseArtifactsPrefix = "/_artifacts"
+	ExtArtifactsPrefix  = BaseArtifactsPrefix + "/ext"
+
+	ExtArtifactsSearch        = "/search"
+	ExtArtifactsSearchPrefix  = ExtArtifactsPrefix + ExtArtifactsSearch
+	FullSearchArtifactsPrefix = RoutePrefix + ExtArtifactsSearchPrefix
+
 	// mgmt extension.
 	Mgmt     = "/mgmt"
 	ExtMgmt  = ExtPrefix + Mgmt

diff --git a/pkg/extensions/extension_search.go b/pkg/extensions/extension_search.go
@@ -102,4 +102,11 @@ func SetupSearchRoutes(conf *config.Config, router *mux.Router, storeController
 		Handler(gqlHandler.NewDefaultServer(gql_generated.NewExecutableSchema(resConfig)))
 
 	log.Info().Msg("finished setting up search routes")
+
+	extRouter = router.PathPrefix(constants.ExtArtifactsSearchPrefix).Subrouter()
+	extRouter.Use(zcommon.CORSHeadersMiddleware(conf.HTTP.AllowOrigin))
+	extRouter.Use(zcommon.ACHeadersMiddleware(conf, allowedMethods...))
+	extRouter.Use(zcommon.AddExtensionSecurityHeaders())
+	extRouter.Methods(allowedMethods...).
+		Handler(gqlHandler.NewDefaultServer(gql_generated.NewExecutableSchema(resConfig)))
 }