Initial open-source release (MIT): image2api AI gateway

Full Go backend + Vue 3 frontend, OpenAI-compatible API, multi-provider account pools, billing/admin, Docker one-command deploy with auto HTTPS. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-25 22:59:04 +08:00
commit 606caaf047
142 changed files with 33648 additions and 0 deletions
@@ -0,0 +1,338 @@
+package handler
+
+import (
+	"errors"
+	"net/http"
+
+	"backend/internal/service"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type ProviderAdminHandler struct {
+	tokens  *service.TokenService
+	refresh *service.RefreshProfileService
+}
+
+func NewProviderAdminHandler(tokens *service.TokenService, refresh *service.RefreshProfileService) *ProviderAdminHandler {
+	return &ProviderAdminHandler{
+		tokens:  tokens,
+		refresh: refresh,
+	}
+}
+
+func (h *ProviderAdminHandler) TokensList(c *gin.Context) {
+	data, err := h.tokens.List(c.Request.Context())
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"detail": "failed to load tokens"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"data": data})
+}
+
+func (h *ProviderAdminHandler) TokensCreate(c *gin.Context) {
+	var body struct {
+		Pool  string `json:"pool"`
+		Value string `json:"value"`
+		ID    string `json:"id"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	item, err := h.tokens.Add(c.Request.Context(), body.Pool, body.Value, body.ID)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "id": item.ID})
+}
+
+func (h *ProviderAdminHandler) ImportChatGPTToken(c *gin.Context) {
+	var body struct {
+		AccessToken string `json:"access_token"`
+		Value       string `json:"value"`
+		Name        string `json:"name"`
+		ID          string `json:"id"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	token := body.AccessToken
+	if token == "" {
+		token = body.Value
+	}
+	name := body.Name
+	if name == "" {
+		name = body.ID
+	}
+	item, err := h.tokens.ImportChatGPTToken(c.Request.Context(), token, name)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "id": item.ID, "status": item.Status, "pending": item.Status == "pending"})
+}
+
+func (h *ProviderAdminHandler) ImportRunwayToken(c *gin.Context) {
+	var body struct {
+		AccessToken string `json:"access_token"`
+		Value       string `json:"value"`
+		Name        string `json:"name"`
+		ID          string `json:"id"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	token := body.AccessToken
+	if token == "" {
+		token = body.Value
+	}
+	name := body.Name
+	if name == "" {
+		name = body.ID
+	}
+	item, err := h.tokens.ImportRunwayToken(c.Request.Context(), token, name)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "id": item.ID, "status": item.Status, "pending": item.Status == "pending"})
+}
+
+func (h *ProviderAdminHandler) ImportKreaCookie(c *gin.Context) {
+	var body struct {
+		Cookie string `json:"cookie"`
+		Value  string `json:"value"`
+		Name   string `json:"name"`
+		ID     string `json:"id"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	cookie := body.Cookie
+	if cookie == "" {
+		cookie = body.Value
+	}
+	name := body.Name
+	if name == "" {
+		name = body.ID
+	}
+	item, err := h.tokens.ImportKreaCookie(c.Request.Context(), cookie, name)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "id": item.ID, "status": item.Status, "pending": item.Status == "pending"})
+}
+
+func (h *ProviderAdminHandler) ImportImagineToken(c *gin.Context) {
+	var body struct {
+		Cookie string `json:"cookie"`
+		Value  string `json:"value"`
+		Name   string `json:"name"`
+		ID     string `json:"id"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	cred := body.Cookie
+	if cred == "" {
+		cred = body.Value
+	}
+	name := body.Name
+	if name == "" {
+		name = body.ID
+	}
+	item, err := h.tokens.ImportImagineToken(c.Request.Context(), cred, name)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "id": item.ID, "status": item.Status, "pending": item.Status == "pending"})
+}
+
+func (h *ProviderAdminHandler) ImportLeonardoCookie(c *gin.Context) {
+	var body struct {
+		Cookie string `json:"cookie"`
+		Value  string `json:"value"`
+		Name   string `json:"name"`
+		ID     string `json:"id"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	cookie := body.Cookie
+	if cookie == "" {
+		cookie = body.Value
+	}
+	name := body.Name
+	if name == "" {
+		name = body.ID
+	}
+	item, err := h.tokens.ImportLeonardoCookie(c.Request.Context(), cookie, name)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "id": item.ID, "status": item.Status, "pending": item.Status == "pending"})
+}
+
+func (h *ProviderAdminHandler) ImportAdobeCookie(c *gin.Context) {
+	var body struct {
+		Cookie string `json:"cookie"`
+		Value  string `json:"value"`
+		Name   string `json:"name"`
+		ID     string `json:"id"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	cookie := body.Cookie
+	if cookie == "" {
+		cookie = body.Value
+	}
+	name := body.Name
+	if name == "" {
+		name = body.ID
+	}
+	item, profile, err := h.tokens.ImportAdobeCookie(c.Request.Context(), cookie, name)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"ok":         true,
+		"profile_id": profile.ID,
+		"id":         item.ID,
+		"status":     item.Status,
+		"pending":    item.Status == "pending",
+	})
+}
+
+func (h *ProviderAdminHandler) TokenUpdate(c *gin.Context) {
+	var body map[string]any
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	item, err := h.tokens.Update(c.Request.Context(), c.Param("pool"), c.Param("id"), body)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "data": item})
+}
+
+func (h *ProviderAdminHandler) TokenDelete(c *gin.Context) {
+	if err := h.tokens.Delete(c.Request.Context(), c.Param("pool"), c.Param("id")); err != nil {
+		if errors.Is(err, service.ErrNotFound) {
+			c.JSON(http.StatusNotFound, gin.H{"detail": "token not found"})
+			return
+		}
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true})
+}
+
+// TokenDeleteBulk removes multiple accounts in one call (account multi-select).
+func (h *ProviderAdminHandler) TokenDeleteBulk(c *gin.Context) {
+	var body struct {
+		IDs []string `json:"ids"`
+	}
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	if len(body.IDs) == 0 {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "未选择任何账号"})
+		return
+	}
+	n, err := h.tokens.DeleteBulk(c.Request.Context(), body.IDs)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "deleted": n})
+}
+
+func (h *ProviderAdminHandler) AccountsList(c *gin.Context) {
+	data, err := h.tokens.Accounts(c.Request.Context())
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"detail": "failed to load accounts"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"data": data})
+}
+
+func (h *ProviderAdminHandler) AccountQuota(c *gin.Context) {
+	data, err := h.tokens.Quota(c.Request.Context(), c.Param("pool"), c.Param("id"))
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			c.JSON(http.StatusNotFound, gin.H{"detail": "account not found"})
+			return
+		}
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, data)
+}
+
+func (h *ProviderAdminHandler) AccountEmail(c *gin.Context) {
+	data, err := h.tokens.Email(c.Request.Context(), c.Param("pool"), c.Param("id"))
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			c.JSON(http.StatusNotFound, gin.H{"detail": "account not found"})
+			return
+		}
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, data)
+}
+
+func (h *ProviderAdminHandler) RefreshProfiles(c *gin.Context) {
+	items, err := h.refresh.List(c.Request.Context())
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"detail": "failed to load refresh profiles"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"data": items})
+}
+
+func (h *ProviderAdminHandler) RefreshNow(c *gin.Context) {
+	if err := h.refresh.RefreshNow(c.Request.Context(), c.Param("profile_id")); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true})
+}
+
+func (h *ProviderAdminHandler) RefreshUpdate(c *gin.Context) {
+	var body map[string]any
+	if err := c.ShouldBindJSON(&body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
+		return
+	}
+	item, err := h.refresh.Update(c.Request.Context(), c.Param("profile_id"), body)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true, "data": item})
+}
+
+func (h *ProviderAdminHandler) RefreshDelete(c *gin.Context) {
+	if err := h.refresh.Delete(c.Request.Context(), c.Param("profile_id")); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"ok": true})
+}