// Package adminhandler wires up the authenticated admin console: a JSON write // API mounted under a base path (default /ping) plus the embedded SPA. package adminhandler import ( "encoding/json" "errors" "io/fs" "net/http" "strings" "time" "github.com/go-chi/chi/v5" "github.com/go-chi/chi/v5/middleware" "github.com/baicai2026-baicai/goods/api/internal/adminstore" "github.com/baicai2026-baicai/goods/api/internal/auth" "github.com/baicai2026-baicai/goods/api/internal/gtin" ) // Handler holds the admin dependencies. type Handler struct { store *adminstore.Store authn *auth.Authenticator basePath string spa fs.FS submitLimit *rateLimiter } // New constructs an admin Handler. basePath is e.g. "/ping" (no trailing slash). func New(store *adminstore.Store, authn *auth.Authenticator, basePath string, spa fs.FS) *Handler { basePath = "/" + strings.Trim(basePath, "/") return &Handler{ store: store, authn: authn, basePath: basePath, spa: spa, submitLimit: newRateLimiter(5, 10*time.Minute), } } // Router builds the HTTP handler. func (h *Handler) Router() http.Handler { r := chi.NewRouter() r.Use(middleware.RequestID) r.Use(middleware.RealIP) r.Use(middleware.Recoverer) r.Route(h.basePath, func(r chi.Router) { r.Get("/healthz", func(w http.ResponseWriter, _ *http.Request) { writeJSON(w, http.StatusOK, map[string]string{"status": "ok"}) }) r.Post("/api/login", h.Login) r.Group(func(r chi.Router) { r.Use(h.authn.Middleware) r.Get("/api/me", h.Me) r.Get("/api/products", h.ListProducts) r.Get("/api/products/{id}", h.GetProduct) r.Put("/api/products/{id}", h.UpdateProduct) r.Get("/api/products/{id}/audit", h.ListAudit) r.Post("/api/products/{id}/images", h.AddImage) r.Delete("/api/products/{id}/images/{imageID}", h.DeleteImage) r.Post("/api/products/{id}/msrp", h.AddMSRP) r.Delete("/api/products/{id}/msrp/{msrpID}", h.DeleteMSRP) r.Post("/api/products/{id}/barcodes", h.AddBarcode) r.Delete("/api/products/{id}/barcodes/{barcodeID}", h.DeleteBarcode) r.Post("/api/products/{id}/barcodes/{barcodeID}/primary", h.SetPrimaryBarcode) r.Get("/api/brands", h.ListBrands) r.Get("/api/categories", h.ListCategories) r.Get("/api/submissions", h.ListSubmissions) r.Get("/api/submissions/{id}", h.GetSubmission) r.Post("/api/submissions/{id}/approve", h.ApproveSubmission) r.Post("/api/submissions/{id}/reject", h.RejectSubmission) }) r.Handle("/*", http.HandlerFunc(h.serveSPA)) }) // Public, unauthenticated contribution endpoint (proxied at /api/public/*). // Submissions enter a moderation queue and never touch products until an // admin approves them. r.Post("/api/public/submissions", h.CreateSubmission) return r } func (h *Handler) serveSPA(w http.ResponseWriter, r *http.Request) { rel := strings.TrimPrefix(r.URL.Path, h.basePath) rel = strings.TrimPrefix(rel, "/") if rel == "" { rel = "index.html" } if f, err := h.spa.Open(rel); err == nil { f.Close() http.StripPrefix(h.basePath+"/", http.FileServer(http.FS(h.spa))).ServeHTTP(w, r) return } // SPA fallback: serve index.html for client-side routes. index, err := h.spa.Open("index.html") if err != nil { http.NotFound(w, r) return } defer index.Close() data, _ := fs.ReadFile(h.spa, "index.html") w.Header().Set("Content-Type", "text/html; charset=utf-8") _, _ = w.Write(data) } // ---------- auth ---------- // Login authenticates and returns a bearer token. func (h *Handler) Login(w http.ResponseWriter, r *http.Request) { var body struct { Username string `json:"username"` Password string `json:"password"` } if err := json.NewDecoder(r.Body).Decode(&body); err != nil { writeError(w, http.StatusBadRequest, "bad_request", "invalid body") return } token, err := h.authn.Login(body.Username, body.Password) if err != nil { writeError(w, http.StatusUnauthorized, "unauthorized", "用户名或密码错误") return } writeJSON(w, http.StatusOK, map[string]string{"token": token, "username": body.Username}) } // Me returns the current authenticated user. func (h *Handler) Me(w http.ResponseWriter, r *http.Request) { writeJSON(w, http.StatusOK, map[string]string{"username": auth.UserFrom(r.Context())}) } // ---------- products ---------- // ListProducts returns a paginated product list. func (h *Handler) ListProducts(w http.ResponseWriter, r *http.Request) { q := r.URL.Query().Get("q") page, size := pageParams(r) items, total, err := h.store.ListProducts(r.Context(), q, size, (page-1)*size) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]any{ "items": items, "page": page, "size": size, "total": total, "completeness_fields": adminstore.CompletenessFields, }) } // GetProduct returns full editable detail. func (h *Handler) GetProduct(w http.ResponseWriter, r *http.Request) { d, err := h.store.GetProduct(r.Context(), chi.URLParam(r, "id")) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, d) } // UpdateProduct applies an edit. func (h *Handler) UpdateProduct(w http.ResponseWriter, r *http.Request) { var in adminstore.ProductInput if err := json.NewDecoder(r.Body).Decode(&in); err != nil { writeError(w, http.StatusBadRequest, "bad_request", "invalid body") return } if strings.TrimSpace(in.Name) == "" { writeError(w, http.StatusBadRequest, "bad_request", "名称不能为空") return } d, err := h.store.UpdateProduct(r.Context(), chi.URLParam(r, "id"), auth.UserFrom(r.Context()), in) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, d) } // ListAudit returns audit history for a product. func (h *Handler) ListAudit(w http.ResponseWriter, r *http.Request) { items, err := h.store.ListAudit(r.Context(), chi.URLParam(r, "id"), 100) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]any{"items": items}) } // AddImage adds an image URL. func (h *Handler) AddImage(w http.ResponseWriter, r *http.Request) { var body struct { URL string `json:"url"` Kind string `json:"kind"` } if err := json.NewDecoder(r.Body).Decode(&body); err != nil || strings.TrimSpace(body.URL) == "" { writeError(w, http.StatusBadRequest, "bad_request", "图片 URL 不能为空") return } im, err := h.store.AddImage(r.Context(), chi.URLParam(r, "id"), auth.UserFrom(r.Context()), body.URL, body.Kind) if h.handleErr(w, err) { return } writeJSON(w, http.StatusCreated, im) } // DeleteImage removes an image. func (h *Handler) DeleteImage(w http.ResponseWriter, r *http.Request) { err := h.store.DeleteImage(r.Context(), chi.URLParam(r, "id"), chi.URLParam(r, "imageID"), auth.UserFrom(r.Context())) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]string{"status": "deleted"}) } // AddMSRP adds a suggested-retail-price snapshot. func (h *Handler) AddMSRP(w http.ResponseWriter, r *http.Request) { var in adminstore.MSRPInput if err := json.NewDecoder(r.Body).Decode(&in); err != nil { writeError(w, http.StatusBadRequest, "bad_request", "invalid body") return } m, err := h.store.AddMSRP(r.Context(), chi.URLParam(r, "id"), auth.UserFrom(r.Context()), in) if h.handleErr(w, err) { return } writeJSON(w, http.StatusCreated, m) } // DeleteMSRP removes an MSRP snapshot. func (h *Handler) DeleteMSRP(w http.ResponseWriter, r *http.Request) { err := h.store.DeleteMSRP(r.Context(), chi.URLParam(r, "id"), chi.URLParam(r, "msrpID"), auth.UserFrom(r.Context())) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]string{"status": "deleted"}) } // ---------- barcodes ---------- // AddBarcode validates and attaches a barcode to a product. A code already // owned by another product yields 409 with the conflicting product so the // operator can de-duplicate; an invalid GTIN yields 400. func (h *Handler) AddBarcode(w http.ResponseWriter, r *http.Request) { var in adminstore.BarcodeInput if err := json.NewDecoder(r.Body).Decode(&in); err != nil { writeError(w, http.StatusBadRequest, "bad_request", "invalid body") return } b, err := h.store.AddBarcode(r.Context(), chi.URLParam(r, "id"), auth.UserFrom(r.Context()), in) if h.handleBarcodeErr(w, err) { return } writeJSON(w, http.StatusCreated, b) } // DeleteBarcode removes a barcode; a primary one is replaced automatically. func (h *Handler) DeleteBarcode(w http.ResponseWriter, r *http.Request) { err := h.store.DeleteBarcode(r.Context(), chi.URLParam(r, "id"), chi.URLParam(r, "barcodeID"), auth.UserFrom(r.Context())) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]string{"status": "deleted"}) } // SetPrimaryBarcode marks one barcode primary and mirrors it to product.gtin. func (h *Handler) SetPrimaryBarcode(w http.ResponseWriter, r *http.Request) { b, err := h.store.SetPrimaryBarcode(r.Context(), chi.URLParam(r, "id"), chi.URLParam(r, "barcodeID"), auth.UserFrom(r.Context())) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, b) } // handleBarcodeErr maps barcode-specific errors (GTIN validation, ownership // conflict) to client-facing statuses, falling back to handleErr otherwise. func (h *Handler) handleBarcodeErr(w http.ResponseWriter, err error) bool { if err == nil { return false } var conflict *adminstore.ConflictError if errors.As(err, &conflict) { writeJSON(w, http.StatusConflict, map[string]any{ "error": map[string]string{"code": "barcode_conflict", "message": err.Error()}, "conflict": map[string]string{ "gtin": conflict.GTIN, "product_id": conflict.ProductID, "product_name": conflict.ProductName, }, }) return true } if errors.Is(err, gtin.ErrEmpty) || errors.Is(err, gtin.ErrFormat) || errors.Is(err, gtin.ErrCheck) || errors.Is(err, gtin.ErrRestricted) { writeError(w, http.StatusBadRequest, "invalid_gtin", err.Error()) return true } return h.handleErr(w, err) } // ListBrands returns brand options. func (h *Handler) ListBrands(w http.ResponseWriter, r *http.Request) { items, err := h.store.ListBrands(r.Context()) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]any{"items": items}) } // ListCategories returns category options. func (h *Handler) ListCategories(w http.ResponseWriter, r *http.Request) { items, err := h.store.ListCategories(r.Context()) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]any{"items": items}) } // ---------- submissions ---------- // CreateSubmission accepts an anonymous public contribution into the queue. func (h *Handler) CreateSubmission(w http.ResponseWriter, r *http.Request) { if !h.submitLimit.allow(realIP(r)) { writeError(w, http.StatusTooManyRequests, "rate_limited", "提交过于频繁,请稍后再试") return } var in adminstore.SubmissionInput if err := json.NewDecoder(http.MaxBytesReader(w, r.Body, 1<<20)).Decode(&in); err != nil { writeError(w, http.StatusBadRequest, "bad_request", "invalid body") return } if strings.TrimSpace(in.Name) == "" { writeError(w, http.StatusBadRequest, "bad_request", "商品名称不能为空") return } id, err := h.store.CreateSubmission(r.Context(), in, realIP(r)) if err != nil { writeError(w, http.StatusInternalServerError, "internal_error", err.Error()) return } writeJSON(w, http.StatusCreated, map[string]string{"id": id, "status": "pending"}) } // ListSubmissions returns the moderation queue (admin). func (h *Handler) ListSubmissions(w http.ResponseWriter, r *http.Request) { status := r.URL.Query().Get("status") page, size := pageParams(r) items, total, err := h.store.ListSubmissions(r.Context(), status, size, (page-1)*size) if h.handleErr(w, err) { return } pending, _ := h.store.PendingSubmissionCount(r.Context()) writeJSON(w, http.StatusOK, map[string]any{ "items": items, "page": page, "size": size, "total": total, "pending": pending, }) } // GetSubmission returns full submission detail (admin). func (h *Handler) GetSubmission(w http.ResponseWriter, r *http.Request) { d, err := h.store.GetSubmission(r.Context(), chi.URLParam(r, "id")) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, d) } // ApproveSubmission applies a contribution to the product store (admin). func (h *Handler) ApproveSubmission(w http.ResponseWriter, r *http.Request) { d, err := h.store.ApproveSubmission(r.Context(), chi.URLParam(r, "id"), auth.UserFrom(r.Context())) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, d) } // RejectSubmission rejects a contribution with a reviewer note (admin). func (h *Handler) RejectSubmission(w http.ResponseWriter, r *http.Request) { var body struct { Note string `json:"note"` } _ = json.NewDecoder(r.Body).Decode(&body) err := h.store.RejectSubmission(r.Context(), chi.URLParam(r, "id"), auth.UserFrom(r.Context()), body.Note) if h.handleErr(w, err) { return } writeJSON(w, http.StatusOK, map[string]string{"status": "rejected"}) } // ---------- helpers ---------- func realIP(r *http.Request) string { if ip := r.Header.Get("X-Forwarded-For"); ip != "" { if i := strings.IndexByte(ip, ','); i >= 0 { return strings.TrimSpace(ip[:i]) } return strings.TrimSpace(ip) } if ip := r.Header.Get("X-Real-IP"); ip != "" { return ip } return r.RemoteAddr } func (h *Handler) handleErr(w http.ResponseWriter, err error) bool { if err == nil { return false } if errors.Is(err, adminstore.ErrNotFound) { writeError(w, http.StatusNotFound, "not_found", "资源不存在") return true } if errors.Is(err, adminstore.ErrConflict) { writeError(w, http.StatusConflict, "conflict", "该投稿已被处理") return true } writeError(w, http.StatusInternalServerError, "internal_error", err.Error()) return true } func pageParams(r *http.Request) (page, size int) { page = atoiDefault(r.URL.Query().Get("page"), 1) if page < 1 { page = 1 } size = atoiDefault(r.URL.Query().Get("size"), 20) if size < 1 { size = 20 } if size > 100 { size = 100 } return page, size } func atoiDefault(s string, fallback int) int { if s == "" { return fallback } n := 0 for _, c := range s { if c < '0' || c > '9' { return fallback } n = n*10 + int(c-'0') } return n } func writeJSON(w http.ResponseWriter, status int, body any) { w.Header().Set("Content-Type", "application/json; charset=utf-8") w.WriteHeader(status) _ = json.NewEncoder(w).Encode(body) } func writeError(w http.ResponseWriter, status int, code, message string) { writeJSON(w, status, map[string]any{"error": map[string]string{"code": code, "message": message}}) }