<?php
/**
 * 删除分销测试用户及相关数据
 *
 * 默认清理用户 ID：900001 ~ 900004
 * （与 commission_setup_test_users.php 一致）
 *
 * 用法（项目根目录）：
 *   php addons/shopro/tests/commission_cleanup_test_users.php --dry-run   # 预览将删除的数据
 *   php addons/shopro/tests/commission_cleanup_test_users.php --confirm   # 实际删除
 */
if (PHP_SAPI !== 'cli') {
    http_response_code(403);
    exit("请通过命令行运行\n");
}

define('APP_PATH', __DIR__ . '/../../../application/');
define('ROOT_PATH', __DIR__ . '/../../../');

if (!is_file(APP_PATH . 'admin/command/Install/install.lock')) {
    fwrite(STDERR, "项目未安装。\n");
    exit(1);
}

require ROOT_PATH . 'thinkphp/base.php';
\think\App::initCommon();

use think\Db;

function getTestUserIds()
{
    return [900001, 900002, 900003, 900004];
}

function getTestUserMobiles()
{
    return ['19900000001', '19900000002', '19900000003', '19900000004'];
}

$argv = $GLOBALS['argv'] ?? [];
$dryRun = in_array('--dry-run', $argv, true);
$confirm = in_array('--confirm', $argv, true);

if (!$dryRun && !$confirm) {
    echo "此脚本将删除测试用户 900001~900004 及其订单、佣金、地址等数据。\n\n";
    echo "预览：php addons/shopro/tests/commission_cleanup_test_users.php --dry-run\n";
    echo "执行：php addons/shopro/tests/commission_cleanup_test_users.php --confirm\n\n";
    echo "注意：不会回滚分销配置、等级档位、分销商品开关，仅删测试账号相关数据。\n";
    exit(0);
}

$userIds = getTestUserIds();
$mobiles = getTestUserMobiles();

echo "=== 分销测试数据清理 ===\n";
echo $dryRun ? "[预览模式，不会实际删除]\n\n" : "[执行删除]\n\n";

// 校验测试用户
$foundUsers = Db::name('user')->where('id', 'in', $userIds)->field('id,nickname,mobile')->select();
if (!$foundUsers) {
    echo "未找到测试用户 (900001~900004)，无需清理。\n";
    exit(0);
}

echo "将处理的测试用户：\n";
foreach ($foundUsers as $row) {
    echo "  - {$row['id']} {$row['nickname']} ({$row['mobile']})\n";
}
echo "\n";

$stats = [];

try {
    if ($dryRun) {
        previewCleanup($userIds, $stats);
    } else {
        Db::startTrans();
        runCleanup($userIds, $stats);
        Db::commit();
    }
} catch (\Throwable $e) {
    if (!$dryRun) {
        Db::rollback();
    }
    fwrite(STDERR, "失败: " . $e->getMessage() . "\n");
    exit(1);
}

echo "\n========== 汇总 ==========\n";
foreach ($stats as $label => $count) {
    echo sprintf("  %-28s %d\n", $label, $count);
}
echo $dryRun ? "\n确认无误后请加 --confirm 执行删除。\n" : "\n清理完成。\n";

function previewCleanup(array $userIds, array &$stats)
{
    collectRelatedIds($userIds, $orderIds, $orderItemIds, $commissionOrderIds);

    countWhere($stats, '订单', 'shopro_order', ['id' => ['in', $orderIds ?: [0]]]);
    countWhere($stats, '订单商品', 'shopro_order_item', ['order_id' => ['in', $orderIds ?: [0]]]);
    countWhere($stats, '发票', 'shopro_order_invoice', ['order_id' => ['in', $orderIds ?: [0]]]);
    countWhere($stats, '分销订单', 'shopro_commission_order', function ($q) use ($userIds, $orderIds) {
        $q->where(function ($sub) use ($userIds, $orderIds) {
            $sub->where('buyer_id', 'in', $userIds)->whereOr('agent_id', 'in', $userIds);
            if ($orderIds) {
                $sub->whereOr('order_id', 'in', $orderIds);
            }
        });
    });
    countWhere($stats, '佣金记录', 'shopro_commission_reward', function ($q) use ($userIds, $orderIds) {
        $q->where(function ($sub) use ($userIds, $orderIds) {
            $sub->where('agent_id', 'in', $userIds)->whereOr('buyer_id', 'in', $userIds);
            if ($orderIds) {
                $sub->whereOr('order_id', 'in', $orderIds);
            }
        });
    });
    countWhere($stats, '分销动态', 'shopro_commission_log', ['agent_id' => ['in', $userIds]]);
    countWhere($stats, '分享记录', 'shopro_share', function ($q) use ($userIds) {
        $q->where('user_id', 'in', $userIds)->whereOr('share_id', 'in', $userIds);
    });
    countWhere($stats, '钱包流水', 'shopro_user_wallet_log', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '余额日志', 'user_money_log', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '积分日志', 'user_score_log', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '用户优惠券', 'shopro_user_coupons', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '商品评价', 'shopro_goods_comment', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '收货地址', 'shopro_user_address', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '购物车', 'shopro_cart', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '第三方授权', 'shopro_user_oauth', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '分销商', 'shopro_commission_agent', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '用户Token', 'user_token', ['user_id' => ['in', $userIds]]);
    countWhere($stats, '用户账号', 'user', ['id' => ['in', $userIds]]);

    $orphanParent = Db::name('user')->where('parent_user_id', 'in', $userIds)->count();
    $stats['解除上级引用(将置0)'] = $orphanParent;
}

function runCleanup(array $userIds, array &$stats)
{
    collectRelatedIds($userIds, $orderIds, $orderItemIds, $commissionOrderIds);

    // 解除其他用户对测试账号的上级引用
    $stats['解除上级引用'] = Db::name('user')->where('parent_user_id', 'in', $userIds)->update(['parent_user_id' => 0]);

    // 佣金
    $stats['佣金记录'] = deleteWhere('shopro_commission_reward', function ($q) use ($userIds, $orderIds) {
        $q->where(function ($sub) use ($userIds, $orderIds) {
            $sub->where('agent_id', 'in', $userIds)->whereOr('buyer_id', 'in', $userIds);
            if ($orderIds) {
                $sub->whereOr('order_id', 'in', $orderIds);
            }
        });
    });

    $stats['分销订单'] = deleteWhere('shopro_commission_order', function ($q) use ($userIds, $orderIds) {
        $q->where(function ($sub) use ($userIds, $orderIds) {
            $sub->where('buyer_id', 'in', $userIds)->whereOr('agent_id', 'in', $userIds);
            if ($orderIds) {
                $sub->whereOr('order_id', 'in', $orderIds);
            }
        });
    });

    $stats['分销动态'] = deleteWhere('shopro_commission_log', ['agent_id' => ['in', $userIds]]);

    // 订单附属
    if ($orderIds) {
        $stats['发票'] = safeDelete('shopro_order_invoice', 'order_id', $orderIds);
        $stats['售后日志'] = safeDelete('shopro_order_aftersale_log', 'order_id', $orderIds);
        $stats['售后单'] = safeDelete('shopro_order_aftersale', 'order_id', $orderIds);
        $stats['物流日志'] = safeDelete('shopro_order_express_log', 'order_id', $orderIds);
        $stats['物流单'] = safeDelete('shopro_order_express', 'order_id', $orderIds);
        $stats['订单操作'] = safeDelete('shopro_order_action', 'order_id', $orderIds);
        $stats['订单商品'] = safeDelete('shopro_order_item', 'order_id', $orderIds);
        $stats['订单'] = safeDelete('shopro_order', 'id', $orderIds);
    } else {
        $stats['订单'] = 0;
    }

    $stats['分享记录'] = deleteWhere('shopro_share', function ($q) use ($userIds) {
        $q->where('user_id', 'in', $userIds)->whereOr('share_id', 'in', $userIds);
    });
    $stats['钱包流水'] = deleteWhere('shopro_user_wallet_log', ['user_id' => ['in', $userIds]]);
    $stats['余额日志'] = deleteWhere('user_money_log', ['user_id' => ['in', $userIds]]);
    $stats['积分日志'] = deleteWhere('user_score_log', ['user_id' => ['in', $userIds]]);
    $stats['用户优惠券'] = deleteWhere('shopro_user_coupons', ['user_id' => ['in', $userIds]]);
    $stats['商品评价'] = deleteWhere('shopro_goods_comment', ['user_id' => ['in', $userIds]]);
    $stats['收货地址'] = deleteWhere('shopro_user_address', ['user_id' => ['in', $userIds]]);
    $stats['购物车'] = deleteWhere('shopro_cart', ['user_id' => ['in', $userIds]]);
    $stats['第三方授权'] = deleteWhere('shopro_user_oauth', ['user_id' => ['in', $userIds]]);
    $stats['用户Token'] = deleteWhere('user_token', ['user_id' => ['in', $userIds]]);

    $stats['分销商'] = deleteWhere('shopro_commission_agent', ['user_id' => ['in', $userIds]]);
    $stats['用户账号'] = deleteWhere('user', ['id' => ['in', $userIds]]);
}

function collectRelatedIds(array $userIds, &$orderIds, &$orderItemIds, &$commissionOrderIds)
{
    $orderIds = Db::name('shopro_order')->where('user_id', 'in', $userIds)->column('id');
    $orderIds = array_values(array_unique(array_map('intval', $orderIds ?: [])));

    $orderItemIds = [];
    if ($orderIds) {
        $orderItemIds = Db::name('shopro_order_item')->where('order_id', 'in', $orderIds)->column('id');
    }

    $commissionOrderIds = Db::name('shopro_commission_order')
        ->where(function ($q) use ($userIds, $orderIds) {
            $q->where('buyer_id', 'in', $userIds)->whereOr('agent_id', 'in', $userIds);
            if ($orderIds) {
                $q->whereOr('order_id', 'in', $orderIds);
            }
        })
        ->column('id');
}

function tableExists($table)
{
    static $cache = [];
    if (!isset($cache[$table])) {
        $cache[$table] = !empty(Db::query("SHOW TABLES LIKE '" . config('database.prefix') . $table . "'"));
    }
    return $cache[$table];
}

function safeDelete($table, $field, array $ids)
{
    if (!$ids || !tableExists($table)) {
        return 0;
    }
    return Db::name($table)->where($field, 'in', $ids)->delete();
}

function deleteWhere($table, $where)
{
    if (!tableExists($table)) {
        return 0;
    }
    $query = Db::name($table);
    if (is_callable($where)) {
        $where($query);
    } else {
        $query->where($where);
    }
    return $query->delete();
}

function countWhere(array &$stats, $label, $table, $where)
{
    if (!tableExists($table)) {
        $stats[$label] = 0;
        return;
    }
    $query = Db::name($table);
    if (is_callable($where)) {
        $where($query);
    } else {
        $query->where($where);
    }
    $stats[$label] = $query->count();
}