add upload api

This commit is contained in:
delete 2026-06-11 17:44:06 +03:00
parent a90dd9bc50
commit 77da1ffd94
4 changed files with 249 additions and 0 deletions

View file

@ -12,6 +12,7 @@ import (
"update_server/internal/apikeys"
"update_server/internal/db"
"update_server/internal/releases"
)
var errInvalidReleaseID = errors.New("invalid release id")
@ -144,6 +145,71 @@ func (h *handler) apiReleaseDownload(w http.ResponseWriter, r *http.Request) {
http.ServeContent(w, r, release.Filename, release.UpdatedAt, file)
}
func (h *handler) apiUploadRelease(w http.ResponseWriter, r *http.Request) {
state, ok := apikeys.FromContext(r.Context())
if !ok {
writeAPIKeyAuthError(w, http.StatusUnauthorized, "api key authentication required")
return
}
project, err := h.accessibleProjectBySlug(r, state.APIKey)
if err != nil {
h.writeClientResourceError(w, err, "project lookup failed")
return
}
r.Body = http.MaxBytesReader(w, r.Body, maxUploadRequestLimit(h.config.MaxUploadBytes))
if err := r.ParseMultipartForm(16 << 20); err != nil {
if isMaxBytesError(err) {
writeJSON(w, http.StatusRequestEntityTooLarge, map[string]any{"error": fmt.Sprintf("upload exceeds the configured %d MB limit", maxUploadMegabytes(h.config.MaxUploadBytes))})
return
}
writeJSON(w, http.StatusBadRequest, map[string]any{"error": "upload form could not be read"})
return
}
defer func() {
if r.MultipartForm != nil {
_ = r.MultipartForm.RemoveAll()
}
}()
file, header, err := r.FormFile("artifact")
if err != nil {
writeJSON(w, http.StatusBadRequest, map[string]any{"error": "artifact file is required"})
return
}
defer file.Close()
result, err := h.releases.Upload(r.Context(), releases.UploadParams{
ProjectID: project.ID,
Version: r.FormValue("version"),
Build: r.FormValue("build"),
ReleaseNotes: r.FormValue("release_notes"),
OriginalFilename: header.Filename,
DeclaredType: header.Header.Get("Content-Type"),
Reader: file,
})
if err != nil {
switch {
case errors.Is(err, db.ErrNotFound):
writeJSON(w, http.StatusNotFound, map[string]any{"error": "resource not found"})
case errors.Is(err, db.ErrConflict):
writeJSON(w, http.StatusConflict, map[string]any{"error": "release version and build already exist for this project"})
default:
writeJSON(w, http.StatusBadRequest, map[string]any{"error": err.Error()})
}
return
}
releasePayload := apiReleasePayload(*result.Release)
w.Header().Set("Location", releasePayload.MetadataURL)
writeJSON(w, http.StatusCreated, map[string]any{
"project": apiProjectPayload(*result.Project),
"release": releasePayload,
})
}
func (h *handler) accessibleProjectBySlug(r *http.Request, apiKey db.APIKey) (*db.Project, error) {
projectSlug := strings.TrimSpace(chi.URLParam(r, "projectSlug"))
if projectSlug == "" {

View file

@ -1,10 +1,17 @@
package httpserver_test
import (
"bytes"
"context"
"crypto/sha256"
"encoding/hex"
"encoding/json"
"mime/multipart"
"net/http"
"net/http/httptest"
"net/url"
"os"
"path/filepath"
"strings"
"testing"
"time"
@ -192,6 +199,134 @@ func TestClientAPIListsAuthorizedProjectsReturnsLatestMetadataAndStreamsDownload
}
}
func TestClientAPIUploadsReleaseWithUploadPermission(t *testing.T) {
t.Parallel()
router, cfg, store := newTestRouterWithStore(t)
sessionCookie := loginAsAdmin(t, router)
desktopProjectID := extractResourceID(t, submitForm(t, router, http.MethodPost, "/admin/projects", url.Values{
"name": {"Desktop App"},
"slug": {"desktop-app"},
}, http.StatusSeeOther, sessionCookie), "/admin/projects/")
extractResourceID(t, submitForm(t, router, http.MethodPost, "/admin/projects", url.Values{
"name": {"Mobile App"},
"slug": {"mobile-app"},
}, http.StatusSeeOther, sessionCookie), "/admin/projects/")
desktopProjectIDInt := mustParseInt64(t, desktopProjectID)
apiKeyService := apikeys.NewService(store)
uploadKey, err := apiKeyService.Create(t.Context(), apikeys.CreateParams{
Name: "Release Uploader",
ScopeMode: db.ScopeModeProjectAllowList,
CanUpload: true,
ProjectIDs: []int64{desktopProjectIDInt},
})
if err != nil {
t.Fatalf("create upload api key: %v", err)
}
downloadOnlyKey, err := apiKeyService.Create(t.Context(), apikeys.CreateParams{
Name: "Download Only",
ScopeMode: db.ScopeModeAllProjects,
CanDownload: true,
})
if err != nil {
t.Fatalf("create download-only api key: %v", err)
}
artifactBody := []byte("api-release-payload-1.2.3")
uploadRecorder := performAPIMultipartRequest(t, router, "/api/v1/projects/desktop-app/releases", map[string]string{
"version": "1.2.3",
"build": "api-build-7",
"release_notes": "Uploaded through bearer API.",
}, "artifact", "..\\Desktop API 1.2.3.ZIP", artifactBody, "Bearer "+uploadKey.RawKey)
if uploadRecorder.Code != http.StatusCreated {
t.Fatalf("expected API upload to return 201, got %d with body %s", uploadRecorder.Code, uploadRecorder.Body.String())
}
var uploadPayload struct {
Project struct {
Slug string `json:"slug"`
} `json:"project"`
Release struct {
ID int64 `json:"id"`
Version string `json:"version"`
Build string `json:"build"`
Filename string `json:"filename"`
ChecksumSHA256 string `json:"checksum_sha256"`
MetadataURL string `json:"metadata_url"`
DownloadURL string `json:"download_url"`
} `json:"release"`
}
decodeJSONBody(t, uploadRecorder, &uploadPayload)
if uploadPayload.Project.Slug != "desktop-app" {
t.Fatalf("expected desktop-app project payload, got %+v", uploadPayload.Project)
}
if uploadPayload.Release.Version != "1.2.3" || uploadPayload.Release.Build != "api-build-7" {
t.Fatalf("unexpected uploaded release payload %+v", uploadPayload.Release)
}
if uploadPayload.Release.Filename != "desktop-api-1.2.3.zip" {
t.Fatalf("expected sanitized filename, got %q", uploadPayload.Release.Filename)
}
if uploadRecorder.Header().Get("Location") != uploadPayload.Release.MetadataURL {
t.Fatalf("expected Location to point at metadata URL, got %q", uploadRecorder.Header().Get("Location"))
}
expectedChecksum := sha256.Sum256(artifactBody)
if uploadPayload.Release.ChecksumSHA256 != hex.EncodeToString(expectedChecksum[:]) {
t.Fatalf("expected checksum %s, got %s", hex.EncodeToString(expectedChecksum[:]), uploadPayload.Release.ChecksumSHA256)
}
releases, err := store.Releases.ListByProjectID(context.Background(), desktopProjectIDInt)
if err != nil {
t.Fatalf("load releases: %v", err)
}
if len(releases) != 1 {
t.Fatalf("expected one release, got %d", len(releases))
}
release := releases[0].Release
if release.UploadedByUserID != nil {
t.Fatalf("expected API-uploaded release to have no admin uploader, got %d", *release.UploadedByUserID)
}
if release.StoragePath != "desktop-app/1.2.3/api-build-7/desktop-api-1.2.3.zip" {
t.Fatalf("unexpected storage path %q", release.StoragePath)
}
storedArtifact, err := os.ReadFile(filepath.Join(cfg.ArtifactsDir, filepath.FromSlash(release.StoragePath)))
if err != nil {
t.Fatalf("read stored artifact: %v", err)
}
if !bytes.Equal(storedArtifact, artifactBody) {
t.Fatal("stored artifact body did not match uploaded payload")
}
duplicateRecorder := performAPIMultipartRequest(t, router, "/api/v1/projects/desktop-app/releases", map[string]string{
"version": "1.2.3",
"build": "api-build-7",
}, "artifact", "desktop-api-alt.zip", []byte("duplicate"), "Bearer "+uploadKey.RawKey)
if duplicateRecorder.Code != http.StatusConflict {
t.Fatalf("expected duplicate version/build to return 409, got %d with body %s", duplicateRecorder.Code, duplicateRecorder.Body.String())
}
forbiddenRecorder := performAPIMultipartRequest(t, router, "/api/v1/projects/desktop-app/releases", map[string]string{
"version": "2.0.0",
}, "artifact", "desktop-api-2.0.0.zip", []byte("forbidden"), "Bearer "+downloadOnlyKey.RawKey)
if forbiddenRecorder.Code != http.StatusForbidden {
t.Fatalf("expected key without upload permission to return 403, got %d with body %s", forbiddenRecorder.Code, forbiddenRecorder.Body.String())
}
blockedProjectRecorder := performAPIMultipartRequest(t, router, "/api/v1/projects/mobile-app/releases", map[string]string{
"version": "2.0.0",
}, "artifact", "mobile-api-2.0.0.apk", []byte("blocked"), "Bearer "+uploadKey.RawKey)
if blockedProjectRecorder.Code != http.StatusNotFound {
t.Fatalf("expected out-of-scope project upload to return 404, got %d with body %s", blockedProjectRecorder.Code, blockedProjectRecorder.Body.String())
}
}
func TestClientAPIRejectsMissingPermissionDisabledAndExpiredKeys(t *testing.T) {
t.Parallel()
@ -281,6 +416,40 @@ func performAPIRequest(t *testing.T, handler http.Handler, method, target, autho
return recorder
}
func performAPIMultipartRequest(t *testing.T, handler http.Handler, target string, fields map[string]string, fileField, filename string, body []byte, authorization string) *httptest.ResponseRecorder {
t.Helper()
var requestBody bytes.Buffer
writer := multipart.NewWriter(&requestBody)
for key, value := range fields {
if err := writer.WriteField(key, value); err != nil {
t.Fatalf("write multipart field %s: %v", key, err)
}
}
part, err := writer.CreateFormFile(fileField, filename)
if err != nil {
t.Fatalf("create multipart file part: %v", err)
}
if _, err := part.Write(body); err != nil {
t.Fatalf("write multipart file body: %v", err)
}
if err := writer.Close(); err != nil {
t.Fatalf("close multipart writer: %v", err)
}
req := httptest.NewRequest(http.MethodPost, target, &requestBody)
req.RemoteAddr = "127.0.0.1:12345"
req.Header.Set("Content-Type", writer.FormDataContentType())
if authorization != "" {
req.Header.Set("Authorization", authorization)
}
recorder := httptest.NewRecorder()
handler.ServeHTTP(recorder, req)
return recorder
}
func decodeJSONBody(t *testing.T, recorder *httptest.ResponseRecorder, target any) {
t.Helper()

View file

@ -124,6 +124,11 @@ func (h *handler) apiIndex(w http.ResponseWriter, r *http.Request) {
"path": "/api/v1/projects/{projectSlug}/releases/latest",
"description": "Get the latest active release metadata for an accessible project.",
},
{
"method": http.MethodPost,
"path": "/api/v1/projects/{projectSlug}/releases",
"description": "Upload a release artifact for an accessible project.",
},
{
"method": http.MethodGet,
"path": "/api/v1/releases/{releaseID}",

View file

@ -106,6 +106,15 @@ func NewRouter(cfg config.Config, logger *slog.Logger, renderer *Renderer, store
r.Get("/releases/{releaseID}", h.apiReleaseMetadata)
r.Get("/releases/{releaseID}/download", h.apiReleaseDownload)
})
r.Group(func(r chi.Router) {
r.Use(protectedAPIResponseHeaders)
r.Use(h.clientAPIRateLimit)
r.Use(h.requireAPIKey)
r.Use(h.requireAPIKeyPermission(apikeys.PermissionUpload))
r.Post("/projects/{projectSlug}/releases", h.apiUploadRelease)
})
})
})