Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/openai_quota_reset_credits.go

@@ -0,0 +1,141 @@
+package service
+
+import (
+	"bytes"
+	"encoding/json"
+	"strconv"
+	"strings"
+)
+
+type openAIRateLimitResetCreditDetailPayload struct {
+	ExpiresAt      string `json:"expires_at,omitempty"`
+	ExpiresAtCamel string `json:"expiresAt,omitempty"`
+	ResetType      string `json:"reset_type,omitempty"`
+	ResetTypeCamel string `json:"resetType,omitempty"`
+	Status         string `json:"status,omitempty"`
+}
+
+type openAIRateLimitResetCreditDetailsPayload struct {
+	AvailableCount        json.RawMessage `json:"available_count,omitempty"`
+	AvailableCountCamel   json.RawMessage `json:"availableCount,omitempty"`
+	Credits               json.RawMessage `json:"credits,omitempty"`
+	RateLimitResetCredits json.RawMessage `json:"rate_limit_reset_credits,omitempty"`
+	Items                 json.RawMessage `json:"items,omitempty"`
+	Data                  json.RawMessage `json:"data,omitempty"`
+}
+
+type openAIRateLimitResetCreditDetails struct {
+	AvailableCount       *int
+	AvailableCreditCount int
+	CreditListPresent    bool
+	Credits              []OpenAIRateLimitResetCreditDetail
+}
+
+func parseOpenAIRateLimitResetCreditDetails(body []byte) (openAIRateLimitResetCreditDetails, error) {
+	trimmed := bytes.TrimSpace(body)
+	if len(trimmed) == 0 {
+		return openAIRateLimitResetCreditDetails{}, nil
+	}
+
+	var rawCredits []*openAIRateLimitResetCreditDetailPayload
+	var availableCount *int
+	var creditListPresent bool
+	if trimmed[0] == '[' {
+		if err := json.Unmarshal(trimmed, &rawCredits); err != nil {
+			return openAIRateLimitResetCreditDetails{}, err
+		}
+		creditListPresent = true
+	} else {
+		var payload openAIRateLimitResetCreditDetailsPayload
+		if err := json.Unmarshal(trimmed, &payload); err != nil {
+			return openAIRateLimitResetCreditDetails{}, err
+		}
+		availableCount = parseOpenAIResetCreditAvailableCount(payload.AvailableCount, payload.AvailableCountCamel)
+		var err error
+		rawCredits, creditListPresent, err = firstPresentResetCreditPayload(
+			payload.Credits,
+			payload.RateLimitResetCredits,
+			payload.Items,
+			payload.Data,
+		)
+		if err != nil {
+			return openAIRateLimitResetCreditDetails{AvailableCount: availableCount}, err
+		}
+	}
+
+	credits := make([]OpenAIRateLimitResetCreditDetail, 0, len(rawCredits))
+	availableCreditCount := 0
+	for _, raw := range rawCredits {
+		if raw == nil {
+			continue
+		}
+		resetType := strings.TrimSpace(raw.ResetType)
+		if resetType == "" {
+			resetType = strings.TrimSpace(raw.ResetTypeCamel)
+		}
+		if resetType != "" && !strings.EqualFold(resetType, "codex_rate_limits") {
+			continue
+		}
+		if status := strings.TrimSpace(raw.Status); status != "" && !strings.EqualFold(status, "available") {
+			continue
+		}
+		availableCreditCount++
+		expiresAt := strings.TrimSpace(raw.ExpiresAt)
+		if expiresAt == "" {
+			expiresAt = strings.TrimSpace(raw.ExpiresAtCamel)
+		}
+		if expiresAt == "" {
+			continue
+		}
+		credits = append(credits, OpenAIRateLimitResetCreditDetail{ExpiresAt: expiresAt})
+	}
+	return openAIRateLimitResetCreditDetails{
+		AvailableCount:       availableCount,
+		AvailableCreditCount: availableCreditCount,
+		CreditListPresent:    creditListPresent,
+		Credits:              credits,
+	}, nil
+}
+
+func parseOpenAIResetCreditAvailableCount(values ...json.RawMessage) *int {
+	for _, value := range values {
+		trimmed := bytes.TrimSpace(value)
+		if len(trimmed) == 0 || bytes.Equal(trimmed, []byte("null")) {
+			continue
+		}
+
+		var count int
+		if trimmed[0] == '"' {
+			var text string
+			if err := json.Unmarshal(trimmed, &text); err != nil {
+				continue
+			}
+			parsed, err := strconv.Atoi(strings.TrimSpace(text))
+			if err != nil {
+				continue
+			}
+			count = parsed
+		} else if err := json.Unmarshal(trimmed, &count); err != nil {
+			continue
+		}
+		if count >= 0 {
+			return &count
+		}
+	}
+	return nil
+}
+
+func firstPresentResetCreditPayload(values ...json.RawMessage) ([]*openAIRateLimitResetCreditDetailPayload, bool, error) {
+	for _, value := range values {
+		trimmed := bytes.TrimSpace(value)
+		if len(trimmed) == 0 || bytes.Equal(trimmed, []byte("null")) {
+			continue
+		}
+		var credits []*openAIRateLimitResetCreditDetailPayload
+		if err := json.Unmarshal(trimmed, &credits); err != nil {
+			return nil, false, err
+		}
+		return credits, true, nil
+	}
+	return nil, false, nil
+}