Feat: finish object storage

handlers/api/getImages.go

@@ -29,8 +29,8 @@ func (self *Handlers) GetImages(
 ) error {
 	ctx := req.Context()
 
-	rawReqImages := strings.Split(req.Param("images"), ",")
-	rawReqAliases := strings.Split(req.Param("aliases"), ",")
+	rawReqImages := strings.Split(req.URL.Query().Get("images"), ",")
+	rawReqAliases := strings.Split(req.URL.Query().Get("aliases"), ",")
 
 	if (len(rawReqImages) == 0 && len(rawReqAliases) == 0) ||
 		(len(rawReqImages) > 0 && len(rawReqAliases) > 0) {

handlers/api/handlers.go

@@ -4,8 +4,15 @@ import "gitea.konchin.com/go2025/backend/interfaces"
 
 type Handlers struct {
 	db interfaces.Database
+	s3 interfaces.ObjectStorage
 }
 
-func NewHandlers(db interfaces.Database) *Handlers {
-	return &Handlers{db: db}
+func NewHandlers(
+	db interfaces.Database,
+	s3 interfaces.ObjectStorage,
+) *Handlers {
+	return &Handlers{
+		db: db,
+		s3: s3,
+	}
 }

handlers/api/postImage.go

@@ -3,8 +3,11 @@ package api
 import (
 	"net/http"
 	"strings"
+	"time"
 
 	"gitea.konchin.com/go2025/backend/middlewares"
+	"gitea.konchin.com/go2025/backend/models"
+	"gitea.konchin.com/go2025/backend/types"
 	"gitea.konchin.com/go2025/backend/utils"
 	"github.com/uptrace/bunrouter"
 )
@@ -21,6 +24,16 @@ import (
 func (self *Handlers) PostImage(
 	w http.ResponseWriter, req bunrouter.Request,
 ) error {
+	ctx := req.Context()
+
+	claim, ok := ctx.Value(types.AccessToken("")).(models.AccessTokenClaim)
+	if !ok {
+		return middlewares.HTTPError{
+			StatusCode: http.StatusUnauthorized,
+			Message:    "missing access token",
+		}
+	}
+
 	typeHeader := strings.Split(req.Header.Get("Content-Type"), "/")
 	if len(typeHeader) != 2 || typeHeader[0] != "image" {
 		return middlewares.HTTPError{
@@ -29,6 +42,28 @@ func (self *Handlers) PostImage(
 		}
 	}
 
-	// TODO
+	image := models.Image{
+		Extension: typeHeader[1],
+		Uploader:  claim.UserId,
+		UploadTS:  time.Now(),
+	}
+	err := self.db.InsertImage(ctx, &image)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to insert image to db",
+			OriginError: err,
+		}
+	}
+
+	err = self.s3.PutImage(ctx,
+		image.Filename(), req.Body, req.ContentLength)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to put image to s3",
+			OriginError: err,
+		}
+	}
 	return utils.Success(w)
 }

handlers/img/get.go

@@ -0,0 +1,35 @@
+package img
+
+import (
+	"io"
+	"net/http"
+
+	"gitea.konchin.com/go2025/backend/middlewares"
+	"github.com/uptrace/bunrouter"
+)
+
+func (self *Handlers) Get(
+	w http.ResponseWriter,
+	req bunrouter.Request,
+) error {
+	ctx := req.Context()
+
+	filename := req.Param("filename")
+
+	r, err := self.s3.GetImage(ctx, filename)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to get image",
+			OriginError: err,
+		}
+	}
+	if _, err := io.Copy(w, r); err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to copy buffer",
+			OriginError: err,
+		}
+	}
+	return nil
+}

handlers/img/handlers.go

@@ -0,0 +1,11 @@
+package img
+
+import "gitea.konchin.com/go2025/backend/interfaces"
+
+type Handlers struct {
+	s3 interfaces.ObjectStorage
+}
+
+func NewHandlers(s3 interfaces.ObjectStorage) *Handlers {
+	return &Handlers{s3: s3}
+}