adb6c49301
135 文件 19,211 处自动修复：空格→Tab 缩进、括号间距、 函数声明空格、前置自增、尾逗号等纯格式化改动。 零逻辑变更，php -l + token 级对比验证通过。 新增 phpcs.xml.dist / phpstan.neon.dist 项目配置。 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
489 lines
11 KiB
PHP
489 lines
11 KiB
PHP
<?php
|
||
/**
|
||
* Budget Checker - 预算检查器
|
||
*
|
||
* 检查当前费用是否超出预算限额
|
||
*
|
||
* @package WPMind\Modules\CostControl
|
||
* @since 1.0.0
|
||
*/
|
||
|
||
declare(strict_types=1);
|
||
|
||
namespace WPMind\Modules\CostControl;
|
||
|
||
class BudgetChecker {
|
||
|
||
/**
|
||
* 检查结果常量
|
||
*/
|
||
public const STATUS_OK = 'ok';
|
||
public const STATUS_WARNING = 'warning';
|
||
public const STATUS_EXCEEDED = 'exceeded';
|
||
|
||
/**
|
||
* 单例实例
|
||
*/
|
||
private static ?BudgetChecker $instance = null;
|
||
|
||
/**
|
||
* 告警状态缓存
|
||
*/
|
||
private const ALERT_TRANSIENT_KEY = 'wpmind_budget_alerts_sent';
|
||
|
||
/**
|
||
* 获取单例实例
|
||
*/
|
||
public static function instance(): BudgetChecker {
|
||
if ( null === self::$instance ) {
|
||
self::$instance = new self();
|
||
}
|
||
return self::$instance;
|
||
}
|
||
|
||
/**
|
||
* 私有构造函数
|
||
*/
|
||
private function __construct() {}
|
||
|
||
/**
|
||
* 检查全局预算状态
|
||
*
|
||
* @return array ['status' => string, 'details' => array]
|
||
*/
|
||
public function check_global_budget(): array {
|
||
$manager = BudgetManager::instance();
|
||
|
||
if ( ! $manager->is_enabled() || ! $manager->has_any_limits() ) {
|
||
return [
|
||
'status' => self::STATUS_OK,
|
||
'details' => [],
|
||
];
|
||
}
|
||
|
||
$global = $manager->get_global_budget();
|
||
$threshold = $manager->get_alert_threshold();
|
||
|
||
// 获取当前用量
|
||
$today = UsageTracker::get_today_stats();
|
||
$month = UsageTracker::get_month_stats();
|
||
|
||
$details = [];
|
||
$status = self::STATUS_OK;
|
||
|
||
// 检查每日 USD 限额
|
||
if ( ( $global['daily_limit_usd'] ?? 0 ) > 0 ) {
|
||
$result = $this->check_limit(
|
||
$today['cost_usd'] ?? 0,
|
||
$global['daily_limit_usd'],
|
||
$threshold,
|
||
'daily_usd'
|
||
);
|
||
$details['daily_usd'] = $result;
|
||
$status = $this->merge_status( $status, $result['status'] );
|
||
}
|
||
|
||
// 检查每日 CNY 限额
|
||
if ( ( $global['daily_limit_cny'] ?? 0 ) > 0 ) {
|
||
$result = $this->check_limit(
|
||
$today['cost_cny'] ?? 0,
|
||
$global['daily_limit_cny'],
|
||
$threshold,
|
||
'daily_cny'
|
||
);
|
||
$details['daily_cny'] = $result;
|
||
$status = $this->merge_status( $status, $result['status'] );
|
||
}
|
||
|
||
// 检查每月 USD 限额
|
||
if ( ( $global['monthly_limit_usd'] ?? 0 ) > 0 ) {
|
||
$result = $this->check_limit(
|
||
$month['cost_usd'] ?? 0,
|
||
$global['monthly_limit_usd'],
|
||
$threshold,
|
||
'monthly_usd'
|
||
);
|
||
$details['monthly_usd'] = $result;
|
||
$status = $this->merge_status( $status, $result['status'] );
|
||
}
|
||
|
||
// 检查每月 CNY 限额
|
||
if ( ( $global['monthly_limit_cny'] ?? 0 ) > 0 ) {
|
||
$result = $this->check_limit(
|
||
$month['cost_cny'] ?? 0,
|
||
$global['monthly_limit_cny'],
|
||
$threshold,
|
||
'monthly_cny'
|
||
);
|
||
$details['monthly_cny'] = $result;
|
||
$status = $this->merge_status( $status, $result['status'] );
|
||
}
|
||
|
||
return [
|
||
'status' => $status,
|
||
'details' => $details,
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 检查服务商预算状态
|
||
*
|
||
* @param string $provider 服务商 ID
|
||
* @return array ['status' => string, 'details' => array]
|
||
*/
|
||
public function check_provider_budget( string $provider ): array {
|
||
$manager = BudgetManager::instance();
|
||
|
||
if ( ! $manager->is_enabled() ) {
|
||
return [
|
||
'status' => self::STATUS_OK,
|
||
'details' => [],
|
||
];
|
||
}
|
||
|
||
$providerBudget = $manager->get_provider_budget( $provider );
|
||
if ( ! $providerBudget ) {
|
||
return [
|
||
'status' => self::STATUS_OK,
|
||
'details' => [],
|
||
];
|
||
}
|
||
|
||
$threshold = $manager->get_alert_threshold();
|
||
|
||
// 获取服务商用量
|
||
$stats = UsageTracker::get_stats();
|
||
$providerStats = $stats['providers'][ $provider ] ?? null;
|
||
|
||
if ( ! $providerStats ) {
|
||
return [
|
||
'status' => self::STATUS_OK,
|
||
'details' => [],
|
||
];
|
||
}
|
||
|
||
$details = [];
|
||
$status = self::STATUS_OK;
|
||
|
||
// 获取服务商货币类型
|
||
$currency = UsageTracker::get_currency( $provider );
|
||
|
||
// 检查每日限额
|
||
if ( ! empty( $providerBudget['daily_limit'] ) && $providerBudget['daily_limit'] > 0 ) {
|
||
$todayCost = $this->get_provider_today_cost( $provider );
|
||
$result = $this->check_limit(
|
||
$todayCost,
|
||
$providerBudget['daily_limit'],
|
||
$threshold,
|
||
"provider_{$provider}_daily"
|
||
);
|
||
$details['daily'] = $result;
|
||
$status = $this->merge_status( $status, $result['status'] );
|
||
}
|
||
|
||
// 检查每月限额
|
||
if ( ! empty( $providerBudget['monthly_limit'] ) && $providerBudget['monthly_limit'] > 0 ) {
|
||
$monthCost = $this->get_provider_month_cost( $provider );
|
||
$result = $this->check_limit(
|
||
$monthCost,
|
||
$providerBudget['monthly_limit'],
|
||
$threshold,
|
||
"provider_{$provider}_monthly"
|
||
);
|
||
$details['monthly'] = $result;
|
||
$status = $this->merge_status( $status, $result['status'] );
|
||
}
|
||
|
||
return [
|
||
'status' => $status,
|
||
'details' => $details,
|
||
'currency' => $currency,
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 检查是否允许发起请求
|
||
*
|
||
* @param string|null $provider 服务商 ID（可选）
|
||
* @return array ['allowed' => bool, 'reason' => string|null, 'status' => array]
|
||
*/
|
||
public function can_make_request( ?string $provider = null ): array {
|
||
$manager = BudgetManager::instance();
|
||
|
||
if ( ! $manager->is_enabled() ) {
|
||
return [
|
||
'allowed' => true,
|
||
'reason' => null,
|
||
'status' => [],
|
||
];
|
||
}
|
||
|
||
$mode = $manager->get_enforcement_mode();
|
||
|
||
// 仅告警模式始终允许请求
|
||
if ( $mode === BudgetManager::MODE_ALERT ) {
|
||
return [
|
||
'allowed' => true,
|
||
'reason' => null,
|
||
'status' => $this->check_global_budget(),
|
||
];
|
||
}
|
||
|
||
// 检查全局预算
|
||
$globalCheck = $this->check_global_budget();
|
||
if ( $globalCheck['status'] === self::STATUS_EXCEEDED ) {
|
||
return [
|
||
'allowed' => false,
|
||
'reason' => __( '已超出全局预算限额', 'wpmind' ),
|
||
'status' => $globalCheck,
|
||
];
|
||
}
|
||
|
||
// 检查服务商预算
|
||
if ( $provider ) {
|
||
$providerCheck = $this->check_provider_budget( $provider );
|
||
if ( $providerCheck['status'] === self::STATUS_EXCEEDED ) {
|
||
return [
|
||
'allowed' => false,
|
||
'reason' => sprintf( __( '已超出 %s 的预算限额', 'wpmind' ), $provider ),
|
||
'status' => $providerCheck,
|
||
];
|
||
}
|
||
}
|
||
|
||
return [
|
||
'allowed' => true,
|
||
'reason' => null,
|
||
'status' => $globalCheck,
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 获取所有服务商的预算状态
|
||
*
|
||
* @return array
|
||
*/
|
||
public function get_all_provider_status(): array {
|
||
$manager = BudgetManager::instance();
|
||
$settings = $manager->get_settings();
|
||
$result = [];
|
||
|
||
foreach ( ( $settings['providers'] ?? [] ) as $provider => $limits ) {
|
||
$result[ $provider ] = $this->check_provider_budget( $provider );
|
||
}
|
||
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 获取预算摘要（用于 UI 显示）
|
||
*
|
||
* @return array
|
||
*/
|
||
public function get_summary(): array {
|
||
$manager = BudgetManager::instance();
|
||
|
||
if ( ! $manager->is_enabled() ) {
|
||
return [
|
||
'enabled' => false,
|
||
'global' => null,
|
||
'providers' => [],
|
||
];
|
||
}
|
||
|
||
$global = $this->check_global_budget();
|
||
$providers = $this->get_all_provider_status();
|
||
|
||
// 计算使用百分比
|
||
$globalBudget = $manager->get_global_budget();
|
||
$today = UsageTracker::get_today_stats();
|
||
$month = UsageTracker::get_month_stats();
|
||
|
||
$percentages = [];
|
||
|
||
if ( ( $globalBudget['daily_limit_usd'] ?? 0 ) > 0 ) {
|
||
$percentages['daily_usd'] = min(
|
||
100,
|
||
round(
|
||
( ( $today['cost_usd'] ?? 0 ) / $globalBudget['daily_limit_usd'] ) * 100,
|
||
1
|
||
)
|
||
);
|
||
}
|
||
|
||
if ( ( $globalBudget['daily_limit_cny'] ?? 0 ) > 0 ) {
|
||
$percentages['daily_cny'] = min(
|
||
100,
|
||
round(
|
||
( ( $today['cost_cny'] ?? 0 ) / $globalBudget['daily_limit_cny'] ) * 100,
|
||
1
|
||
)
|
||
);
|
||
}
|
||
|
||
if ( ( $globalBudget['monthly_limit_usd'] ?? 0 ) > 0 ) {
|
||
$percentages['monthly_usd'] = min(
|
||
100,
|
||
round(
|
||
( ( $month['cost_usd'] ?? 0 ) / $globalBudget['monthly_limit_usd'] ) * 100,
|
||
1
|
||
)
|
||
);
|
||
}
|
||
|
||
if ( ( $globalBudget['monthly_limit_cny'] ?? 0 ) > 0 ) {
|
||
$percentages['monthly_cny'] = min(
|
||
100,
|
||
round(
|
||
( ( $month['cost_cny'] ?? 0 ) / $globalBudget['monthly_limit_cny'] ) * 100,
|
||
1
|
||
)
|
||
);
|
||
}
|
||
|
||
return [
|
||
'enabled' => true,
|
||
'global' => $global,
|
||
'providers' => $providers,
|
||
'percentages' => $percentages,
|
||
'threshold' => $manager->get_alert_threshold(),
|
||
'mode' => $manager->get_enforcement_mode(),
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 检查单个限额
|
||
*/
|
||
private function check_limit( float $current, float $limit, int $threshold, string $key ): array {
|
||
if ( $limit <= 0 ) {
|
||
return [
|
||
'status' => self::STATUS_OK,
|
||
'current' => $current,
|
||
'limit' => $limit,
|
||
'percentage' => 0,
|
||
];
|
||
}
|
||
|
||
$percentage = ( $current / $limit ) * 100;
|
||
|
||
if ( $percentage >= 100 ) {
|
||
$status = self::STATUS_EXCEEDED;
|
||
} elseif ( $percentage >= $threshold ) {
|
||
$status = self::STATUS_WARNING;
|
||
} else {
|
||
$status = self::STATUS_OK;
|
||
}
|
||
|
||
return [
|
||
'status' => $status,
|
||
'current' => $current,
|
||
'limit' => $limit,
|
||
'percentage' => round( $percentage, 1 ),
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 合并状态（取最严重的）
|
||
*/
|
||
private function merge_status( string $current, string $new ): string {
|
||
$priority = [
|
||
self::STATUS_OK => 0,
|
||
self::STATUS_WARNING => 1,
|
||
self::STATUS_EXCEEDED => 2,
|
||
];
|
||
|
||
return ( $priority[ $new ] ?? 0 ) > ( $priority[ $current ] ?? 0 ) ? $new : $current;
|
||
}
|
||
|
||
/**
|
||
* 获取服务商今日费用
|
||
*/
|
||
private function get_provider_today_cost( string $provider ): float {
|
||
$history = UsageTracker::get_history( 1000 );
|
||
$today = wp_date( 'Y-m-d' );
|
||
$todayStart = strtotime( $today );
|
||
$cost = 0;
|
||
|
||
foreach ( $history as $record ) {
|
||
if ( ( $record['provider'] ?? '' ) !== $provider ) {
|
||
continue;
|
||
}
|
||
if ( ( $record['timestamp'] ?? 0 ) >= $todayStart ) {
|
||
$cost += $record['cost'] ?? 0;
|
||
}
|
||
}
|
||
|
||
return $cost;
|
||
}
|
||
|
||
/**
|
||
* 获取服务商本月费用
|
||
*/
|
||
private function get_provider_month_cost( string $provider ): float {
|
||
$stats = UsageTracker::get_stats();
|
||
return $stats['providers'][ $provider ]['total_cost'] ?? 0;
|
||
}
|
||
|
||
/**
|
||
* 检查是否需要发送告警
|
||
*/
|
||
public function should_send_alert( string $alertKey ): bool {
|
||
$sent = get_transient( self::ALERT_TRANSIENT_KEY );
|
||
if ( ! is_array( $sent ) ) {
|
||
$sent = [];
|
||
}
|
||
|
||
$today = wp_date( 'Y-m-d' );
|
||
$key = "{$alertKey}_{$today}";
|
||
|
||
return ! isset( $sent[ $key ] );
|
||
}
|
||
|
||
/**
|
||
* 标记告警已发送
|
||
*/
|
||
public function mark_alert_sent( string $alertKey ): void {
|
||
$sent = get_transient( self::ALERT_TRANSIENT_KEY );
|
||
if ( ! is_array( $sent ) ) {
|
||
$sent = [];
|
||
}
|
||
|
||
$today = wp_date( 'Y-m-d' );
|
||
$key = "{$alertKey}_{$today}";
|
||
$sent[ $key ] = time();
|
||
|
||
// 清理过期的告警记录
|
||
$yesterday = wp_date( 'Y-m-d', strtotime( '-1 day' ) );
|
||
foreach ( $sent as $k => $v ) {
|
||
if ( ! str_contains( $k, $today ) && ! str_contains( $k, $yesterday ) ) {
|
||
unset( $sent[ $k ] );
|
||
}
|
||
}
|
||
|
||
set_transient( self::ALERT_TRANSIENT_KEY, $sent, DAY_IN_SECONDS * 2 );
|
||
}
|
||
|
||
/**
|
||
* 获取状态的显示标签
|
||
*/
|
||
public static function get_status_label( string $status ): string {
|
||
$labels = [
|
||
self::STATUS_OK => __( '正常', 'wpmind' ),
|
||
self::STATUS_WARNING => __( '接近限额', 'wpmind' ),
|
||
self::STATUS_EXCEEDED => __( '已超限', 'wpmind' ),
|
||
];
|
||
return $labels[ $status ] ?? $status;
|
||
}
|
||
|
||
/**
|
||
* 获取状态的 CSS 类
|
||
*/
|
||
public static function get_status_class( string $status ): string {
|
||
$classes = [
|
||
self::STATUS_OK => 'wpmind-budget-ok',
|
||
self::STATUS_WARNING => 'wpmind-budget-warning',
|
||
self::STATUS_EXCEEDED => 'wpmind-budget-exceeded',
|
||
];
|
||
return $classes[ $status ] ?? '';
|
||
}
|
||
}
|
