feat: APIs to add repositories to favorites and store repo pull count

Signed-off-by: jay-dee7 <[email protected]>

registry/v2/extensions/analytics.go

@@ -0,0 +1,73 @@
+package extensions
+
+import (
+	"net/http"
+	"time"
+
+	"github.com/containerish/OpenRegistry/store/v1/types"
+	"github.com/google/uuid"
+	"github.com/labstack/echo/v4"
+)
+
+type FavoriteRepositoryRequest struct {
+	RepositoryID uuid.UUID `json:"repository_id" query:"repository_id"`
+	UserID       uuid.UUID `json:"user_id" query:"user_id"`
+}
+
+func (ext *extension) AddRepositoryToFavorites(ctx echo.Context) error {
+	ctx.Set(types.HandlerStartTime, time.Now())
+
+	var body FavoriteRepositoryRequest
+	if err := ctx.Bind(&body); err != nil {
+		echoErr := ctx.JSON(http.StatusBadRequest, echo.Map{
+			"error": err.Error(),
+		})
+		ext.logger.Log(ctx, err).Send()
+		return echoErr
+	}
+	defer ctx.Request().Body.Close()
+
+	err := ext.store.AddRepositoryToFavorites(ctx.Request().Context(), body.RepositoryID, body.UserID)
+	if err != nil {
+		echoErr := ctx.JSON(http.StatusBadRequest, echo.Map{
+			"error": err.Error(),
+		})
+		ext.logger.Log(ctx, err).Send()
+		return echoErr
+	}
+
+	echoErr := ctx.JSON(http.StatusOK, echo.Map{
+		"message": "repository added to favorites",
+	})
+	ext.logger.Log(ctx, nil).Send()
+	return echoErr
+}
+
+func (ext *extension) RemoveRepositoryFromFavorites(ctx echo.Context) error {
+	ctx.Set(types.HandlerStartTime, time.Now())
+
+	var body FavoriteRepositoryRequest
+	if err := ctx.Bind(&body); err != nil {
+		echoErr := ctx.JSON(http.StatusBadRequest, echo.Map{
+			"error": err.Error(),
+		})
+		ext.logger.Log(ctx, err).Send()
+		return echoErr
+	}
+	defer ctx.Request().Body.Close()
+
+	err := ext.store.RemoveRepositoryFromFavorites(ctx.Request().Context(), body.RepositoryID, body.UserID)
+	if err != nil {
+		echoErr := ctx.JSON(http.StatusBadRequest, echo.Map{
+			"error": err.Error(),
+		})
+		ext.logger.Log(ctx, err).Send()
+		return echoErr
+	}
+
+	echoErr := ctx.JSON(http.StatusOK, echo.Map{
+		"message": "repository removed from favorites",
+	})
+	ext.logger.Log(ctx, nil).Send()
+	return echoErr
+}

registry/v2/extensions/catalog_detail.go

@@ -18,6 +18,8 @@ type Extenion interface {
 	ChangeContainerImageVisibility(ctx echo.Context) error
 	PublicCatalog(ctx echo.Context) error
 	GetUserCatalog(ctx echo.Context) error
+	AddRepositoryToFavorites(ctx echo.Context) error
+	RemoveRepositoryFromFavorites(ctx echo.Context) error
 }
 
 type extension struct {

registry/v2/registry.go

@@ -226,7 +226,8 @@ func (r *registry) PullManifest(ctx echo.Context) error {
 
 	manifest, err := r.store.GetManifestByReference(ctx.Request().Context(), namespace, ref)
 	if err != nil {
-		errMsg := common.RegistryErrorResponse(RegistryErrorCodeManifestUnknown, err.Error(), echo.Map{
+		errMsg := common.RegistryErrorResponse(RegistryErrorCodeManifestUnknown, "manifest not found", echo.Map{
+			"error":     err.Error(),
 			"namespace": namespace,
 			"ref":       ref,
 		})
@@ -235,6 +236,14 @@ func (r *registry) PullManifest(ctx echo.Context) error {
 		return echoErr
 	}
 
+	defer func() {
+		err = r.store.IncrementRepositoryPullCounter(ctx.Request().Context(), manifest.RepositoryID)
+		// silently fail
+		if err != nil {
+			r.logger.DebugWithContext(ctx).Err(err).Send()
+		}
+	}()
+
 	trimmedMf := manifest.ToOCISubject()
 	ctx.Response().Header().Set("Docker-Content-Digest", manifest.Digest)
 	ctx.Response().Header().Set("Content-Type", manifest.MediaType)

router/middlewares.go

@@ -25,7 +25,7 @@ func registryNamespaceValidator(logger telemetry.Logger) echo.MiddlewareFunc {
 			}
 
 			namespace := ctx.Param("username") + "/" + ctx.Param("imagename")
-			if !nsRegex.MatchString(namespace) {
+			if namespace != "/" && !nsRegex.MatchString(namespace) {
 				registryErr := common.RegistryErrorResponse(
 					registry.RegistryErrorCodeNameInvalid,
 					"invalid user namespace",

router/registry.go

@@ -104,4 +104,6 @@ func RegisterExtensionsRoutes(
 	group.Add(http.MethodGet, UserCatalog, ext.GetUserCatalog, middlewares...)
 	group.Add(http.MethodPost, ChangeRepositoryVisibility, ext.ChangeContainerImageVisibility, middlewares...)
 	group.Add(http.MethodPost, CreateRepository, reg.CreateRepository, middlewares...)
+	group.Add(http.MethodPost, RepositoryFavorites, ext.AddRepositoryToFavorites, middlewares...)
+	group.Add(http.MethodDelete, RepositoryFavorites, ext.RemoveRepositoryFromFavorites, middlewares...)
 }

router/route_names.go

@@ -74,4 +74,5 @@ const (
 
 	ChangeRepositoryVisibility = Ext + "/repository/visibility"
 	CreateRepository           = Ext + "/repository/create"
+	RepositoryFavorites        = Ext + "/repository/favorites"
 )

router/router.go

@@ -38,9 +38,7 @@ func Register(
 	usersStore users_store.UserStore,
 	automationStore automation.BuildAutomationStore,
 ) *echo.Echo {
-	e := echo.New()
-
-	setDefaultEchoOptions(e, cfg.WebAppConfig, healthCheckApi)
+	e := setDefaultEchoOptions(cfg.WebAppConfig, healthCheckApi)
 
 	baseAPIRouter := e.Group("/api")
 	githubRouter := e.Group("/github")
@@ -63,6 +61,7 @@ func Register(
 	RegisterExtensionsRoutes(ociRouter, registryApi, extensionsApi)
 	RegisterWebauthnRoutes(webauthnRouter, webauthnApi)
 	RegisterOrgModeRoutes(orgModeRouter, orgModeApi)
+
 	if cfg.Integrations.GetGithubConfig() != nil && cfg.Integrations.GetGithubConfig().Enabled {
 		RegisterGitHubRoutes(
 			githubRouter,
@@ -96,7 +95,8 @@ func Register(
 	return e
 }
 
-func setDefaultEchoOptions(e *echo.Echo, webConfig config.WebAppConfig, healthCheck http.HandlerFunc) {
+func setDefaultEchoOptions(webConfig config.WebAppConfig, healthCheck http.HandlerFunc) *echo.Echo {
+	e := echo.New()
 	e.HideBanner = true
 
 	e.Use(middleware.Recover())
@@ -123,4 +123,6 @@ func setDefaultEchoOptions(e *echo.Echo, webConfig config.WebAppConfig, healthCh
 	p.Use(e)
 
 	e.Add(http.MethodGet, "/health", echo.WrapHandler(healthCheck))
+
+	return e
 }