//go:build llm_script

package main

import (
	"fmt"
	"regexp"
	"strings"
)

const defaultPPIOPricingURL = "https://resource.ppio.com/pricing?type=enterprise"

var ppioBlockPattern = regexp.MustCompile(`(?s)([a-z0-9._/-]+)\n([\d,]+)\n(.*?)在线体验`)
var ppioPricePattern = regexp.MustCompile(`(?s)￥\s*([\d.]+)\s*/\s*Mt`)

func parsePPIOPricingCatalog(raw string) ([]officialPricingRecord, error) {
	matches := ppioBlockPattern.FindAllStringSubmatch(raw, -1)
	records := make([]officialPricingRecord, 0, len(matches))
	for _, match := range matches {
		modelLine := strings.TrimSpace(match[1])
		contextLength := parseContextLengthCommon(match[2])
		section := match[3]
		if strings.Contains(section, "阶梯计费") {
			continue
		}
		priceMatches := ppioPricePattern.FindAllStringSubmatch(section, -1)
		if len(priceMatches) < 2 {
			continue
		}
		inputPrice := mustParseSubscriptionPrice(priceMatches[len(priceMatches)-2][1])
		outputPrice := mustParseSubscriptionPrice(priceMatches[len(priceMatches)-1][1])
		providerName := providerFromModelPath(modelLine)
		providerNameCn, providerCountry, providerWebsite := providerMetadata(providerName)
		record := officialPricingRecord{
			ModelID:         normalizeExternalID("ppio", modelLine),
			ModelName:       modelLine,
			ProviderName:    providerName,
			ProviderNameCn:  providerNameCn,
			ProviderCountry: providerCountry,
			ProviderWebsite: providerWebsite,
			OperatorName:    "PPIO Model API",
			OperatorNameCn:  "PPIO 模型 API",
			OperatorCountry: "CN",
			OperatorWebsite: "https://ppinfra.com",
			OperatorType:    "relay",
			Region:          "CN",
			Currency:        "CNY",
			InputPrice:      inputPrice,
			OutputPrice:     outputPrice,
			ContextLength:   contextLength,
			SourceURL:       defaultPPIOPricingURL,
			ModelSourceURL:  defaultPPIOPricingURL,
			DateConfidence:  "unknown",
			DateSourceKind:  "official_product_page",
			Modality:        detectModality(modelLine),
		}
		record.IsFree = record.InputPrice == 0 && record.OutputPrice == 0
		records = append(records, record)
	}
	if len(records) == 0 {
		return nil, fmt.Errorf("unexpected ppio pricing content")
	}
	return records, nil
}