wip: oci image management

internal/registry/handlers/blob.go

@@ -0,0 +1,65 @@
+package handlers
+
+import (
+	"net/http"
+	"strconv"
+	"strings"
+
+	"gitea.loveuer.com/loveuer/cluster/internal/registry/storage"
+
+	"github.com/gin-gonic/gin"
+)
+
+// GetBlob 获取 blob
+func GetBlob(store storage.Storage) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		repo := strings.TrimPrefix(c.Param("name"), "/")
+		digest := c.Param("digest")
+
+		reader, size, err := store.GetBlob(repo, digest)
+		if err != nil {
+			c.JSON(http.StatusNotFound, gin.H{
+				"errors": []gin.H{
+					{
+						"code":    "BLOB_UNKNOWN",
+						"message": err.Error(),
+					},
+				},
+			})
+			return
+		}
+		defer reader.Close()
+
+		c.Header("Content-Type", "application/octet-stream")
+		c.Header("Content-Length", strconv.FormatInt(size, 10))
+		c.Header("Docker-Content-Digest", digest)
+		c.DataFromReader(http.StatusOK, size, "application/octet-stream", reader, nil)
+	}
+}
+
+// HeadBlob 检查 blob 是否存在
+func HeadBlob(store storage.Storage) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		repo := c.GetString("repo_name")
+		if repo == "" {
+			repo = strings.TrimPrefix(c.Param("name"), "/")
+		}
+		digest := c.Param("digest")
+
+		exists, err := store.BlobExists(repo, digest)
+		if err != nil || !exists {
+			c.Status(http.StatusNotFound)
+			return
+		}
+
+		size, err := store.GetBlobSize(repo, digest)
+		if err != nil {
+			c.Status(http.StatusNotFound)
+			return
+		}
+
+		c.Header("Content-Length", strconv.FormatInt(size, 10))
+		c.Header("Docker-Content-Digest", digest)
+		c.Status(http.StatusOK)
+	}
+}