feat(image): add stage import endpoints

Author: JSYoo5B

internal/api/image/data.go

@@ -30,6 +30,52 @@ func (h Handler) uploadImageData(w http.ResponseWriter, r *http.Request) {
 	w.WriteHeader(http.StatusNoContent)
 }
 
+func (h Handler) stageImageData(w http.ResponseWriter, r *http.Request) {
+	data, err := io.ReadAll(r.Body)
+	if err != nil {
+		respond.Error(w, http.StatusBadRequest, "invalid image data")
+		return
+	}
+
+	err = h.service.StageData(chi.URLParam(r, "image_id"), data)
+	if errors.Is(err, appimage.ErrImageNotFound) {
+		respond.Error(w, http.StatusNotFound, "image not found")
+		return
+	}
+	if err != nil {
+		respond.Error(w, http.StatusInternalServerError, "image data stage failed")
+		return
+	}
+
+	w.WriteHeader(http.StatusNoContent)
+}
+
+func (h Handler) importImageData(w http.ResponseWriter, r *http.Request) {
+	err := h.service.ImportData(chi.URLParam(r, "image_id"))
+	if errors.Is(err, appimage.ErrImageNotFound) {
+		respond.Error(w, http.StatusNotFound, "image not found")
+		return
+	}
+	if err != nil {
+		respond.Error(w, http.StatusInternalServerError, "image import failed")
+		return
+	}
+
+	w.WriteHeader(http.StatusAccepted)
+}
+
+func (h Handler) getImportInfo(w http.ResponseWriter, r *http.Request) {
+	respond.JSON(w, http.StatusOK, importInfoResponse{
+		ImportMethods: importMethodsDocument{
+			Description: "Import methods available.",
+			Type:        "array",
+			Value: []string{
+				"glance-direct",
+			},
+		},
+	})
+}
+
 func (h Handler) downloadImageData(w http.ResponseWriter, r *http.Request) {
 	data, err := h.service.DownloadData(chi.URLParam(r, "image_id"))
 	if errors.Is(err, appimage.ErrImageNotFound) {

internal/api/image/import_dto.go

@@ -0,0 +1,11 @@
+package image
+
+type importInfoResponse struct {
+	ImportMethods importMethodsDocument `json:"import-methods"`
+}
+
+type importMethodsDocument struct {
+	Description string   `json:"description"`
+	Type        string   `json:"type"`
+	Value       []string `json:"value"`
+}

internal/api/image/import_test.go

@@ -0,0 +1,78 @@
+package image_test
+
+import (
+	"bytes"
+	"context"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/JSYoo5B/SandStack/internal/api/image"
+	"github.com/JSYoo5B/SandStack/internal/testhelper"
+	"github.com/gophercloud/gophercloud/v2/openstack/image/v2/imagedata"
+	"github.com/gophercloud/gophercloud/v2/openstack/image/v2/imageimport"
+	"github.com/gophercloud/gophercloud/v2/openstack/image/v2/images"
+	"github.com/stretchr/testify/suite"
+)
+
+type ImportSuite struct {
+	suite.Suite
+	server *httptest.Server
+}
+
+func TestImportSuite(t *testing.T) {
+	suite.Run(t, new(ImportSuite))
+}
+
+func (s *ImportSuite) SetupTest() {
+	s.server = httptest.NewServer(
+		image.NewRouter(testhelper.DefaultConfig()),
+	)
+}
+
+func (s *ImportSuite) TearDownTest() {
+	s.server.Close()
+}
+
+func (s *ImportSuite) TestGetImportInfo() {
+	client := testhelper.ServiceClient(s.server.URL)
+
+	info, err := imageimport.Get(context.Background(), client).Extract()
+	s.Require().NoError(err)
+
+	s.Assert().Contains(info.ImportMethods.Value, "glance-direct")
+}
+
+func (s *ImportSuite) TestStageAndImportImageData() {
+	client := testhelper.ServiceClient(s.server.URL)
+
+	created, err := images.Create(
+		context.Background(),
+		client,
+		images.CreateOpts{Name: "ubuntu"},
+	).Extract()
+	s.Require().NoError(err)
+
+	err = imagedata.Stage(
+		context.Background(),
+		client,
+		created.ID,
+		bytes.NewBufferString("image-bytes"),
+	).ExtractErr()
+	s.Require().NoError(err)
+
+	staged, err := images.Get(context.Background(), client, created.ID).Extract()
+	s.Require().NoError(err)
+	s.Assert().Equal(images.ImageStatus("uploading"), staged.Status)
+
+	err = imageimport.Create(
+		context.Background(),
+		client,
+		created.ID,
+		imageimport.CreateOpts{Name: imageimport.GlanceDirectMethod},
+	).ExtractErr()
+	s.Require().NoError(err)
+
+	imported, err := images.Get(context.Background(), client, created.ID).Extract()
+	s.Require().NoError(err)
+	s.Assert().Equal(images.ImageStatusActive, imported.Status)
+}

internal/api/image/router.go

@@ -60,6 +60,9 @@ func (h Handler) Router() http.Handler {
 	router.Delete("/images/{image_id}", h.deleteImage)
 	router.Put("/images/{image_id}/file", h.uploadImageData)
 	router.Get("/images/{image_id}/file", h.downloadImageData)
+	router.Put("/images/{image_id}/stage", h.stageImageData)
+	router.Post("/images/{image_id}/import", h.importImageData)
+	router.Get("/info/import", h.getImportInfo)
 	router.Get("/images/{image_id}/members", h.listMembers)
 	router.Post("/images/{image_id}/members", h.createMember)
 	router.Get("/images/{image_id}/members/{member_id}", h.getMember)

internal/app/image/data.go

@@ -21,6 +21,39 @@ func (s *Service) UploadData(id string, data []byte) error {
 	return nil
 }
 
+func (s *Service) StageData(id string, data []byte) error {
+	image, err := s.repository.Get(id)
+	if err != nil {
+		return err
+	}
+
+	now := s.clock.Now().UTC()
+	image.Status = "uploading"
+	image.SizeBytes = int64(len(data))
+	image.UpdatedAt = now.Format(time.RFC3339)
+
+	if _, err := s.repository.Update(image); err != nil {
+		return err
+	}
+	s.dataRepository.Put(id, data)
+
+	return nil
+}
+
+func (s *Service) ImportData(id string) error {
+	image, err := s.repository.Get(id)
+	if err != nil {
+		return err
+	}
+
+	now := s.clock.Now().UTC()
+	image.Status = "active"
+	image.UpdatedAt = now.Format(time.RFC3339)
+
+	_, err = s.repository.Update(image)
+	return err
+}
+
 func (s *Service) DownloadData(id string) ([]byte, error) {
 	if _, err := s.repository.Get(id); err != nil {
 		return nil, err