导入 AKSRT 主题开发源码

inc/classes/class-aksrt-revenue.php

@@ -0,0 +1,118 @@
+<?php
+/**
+ * AKSRT Revenue — 收益台账
+ *
+ * 管理用户收益记录、结算和统计。
+ *
+ * @package AKSRT
+ */
+
+defined('ABSPATH') || exit;
+
+class AKSRT_Revenue {
+
+    private static $instance = null;
+
+    public static function get_instance(): self {
+        if (null === self::$instance) self::$instance = new self();
+        return self::$instance;
+    }
+
+    private function table(): string {
+        return AKSRT_DB_Migrator::table('revenue');
+    }
+
+    /**
+     * 记录收益
+     */
+    public function add(int $user_id, float $amount, string $order_number = '', string $type = 'income'): int {
+        global $wpdb;
+
+        if ($user_id <= 0 || $amount <= 0) return 0;
+
+        $wpdb->insert($this->table(), [
+            'user_id'      => $user_id,
+            'order_number' => sanitize_text_field($order_number),
+            'amount'       => $amount,
+            'type'         => sanitize_key($type),
+            'status'       => 'pending',
+            'created_at'   => current_time('mysql'),
+        ], ['%d', '%s', '%f', '%s', '%s', '%s']);
+
+        return (int) $wpdb->insert_id;
+    }
+
+    /**
+     * 结算收益
+     */
+    public function settle(int $revenue_id): bool {
+        global $wpdb;
+
+        return $wpdb->update(
+            $this->table(),
+            ['status' => 'settled', 'settled_at' => current_time('mysql')],
+            ['id' => $revenue_id, 'status' => 'pending'],
+            ['%s', '%s'],
+            ['%d', '%s']
+        ) !== false;
+    }
+
+    /**
+     * 获取用户收益列表
+     */
+    public function get_user_revenues(int $user_id, array $args = []): array {
+        global $wpdb;
+
+        $where = ['user_id = %d'];
+        $params = [$user_id];
+
+        $status = sanitize_key((string) ($args['status'] ?? ''));
+        if ($status !== '') {
+            $where[] = 'status = %s';
+            $params[] = $status;
+        }
+
+        $limit  = max(1, min(100, (int) ($args['limit'] ?? 20)));
+        $offset = max(0, (int) ($args['offset'] ?? 0));
+
+        $sql = "SELECT * FROM {$this->table()} WHERE " . implode(' AND ', $where) . " ORDER BY created_at DESC LIMIT %d OFFSET %d";
+        $params[] = $limit;
+        $params[] = $offset;
+
+        return (array) $wpdb->get_results($wpdb->prepare($sql, $params), ARRAY_A);
+    }
+
+    /**
+     * 获取收益统计
+     */
+    public function get_stats(int $user_id = 0): array {
+        global $wpdb;
+
+        $where = '';
+        $params = [];
+        if ($user_id > 0) {
+            $where = 'WHERE user_id = %d';
+            $params[] = $user_id;
+        }
+
+        $sql = "SELECT
+                    COALESCE(SUM(amount), 0) as total,
+                    COALESCE(SUM(CASE WHEN status = 'pending' THEN amount ELSE 0 END), 0) as pending,
+                    COALESCE(SUM(CASE WHEN status = 'settled' THEN amount ELSE 0 END), 0) as settled,
+                    COUNT(*) as count
+                FROM {$this->table()} {$where}";
+
+        if (!empty($params)) {
+            $row = $wpdb->get_row($wpdb->prepare($sql, $params), ARRAY_A);
+        } else {
+            $row = $wpdb->get_row($sql, ARRAY_A);
+        }
+
+        return [
+            'total'   => (float) ($row['total'] ?? 0),
+            'pending' => (float) ($row['pending'] ?? 0),
+            'settled' => (float) ($row['settled'] ?? 0),
+            'count'   => (int) ($row['count'] ?? 0),
+        ];
+    }
+}