<?php

if ($text == "💰 افزودن موجودی" && $user->logged_in == 1 && isAdmin($user->login_username)) {

    setStep($from_id, "add_balance:select_type");
    $responseText =
        "💰 *مدیریت موجودی کاربران*\n\n" .
        "لطفاً نوع افزایش موجودی را انتخاب نمایید:\n\n" .
        "➕ افزودن سود: افزایش سود قابل برداشت\n" .
        "➕ افزودن پایه: افزایش سرمایه پایه";

    sendMessage($from_id, $responseText, $setBalanceKeyboard);
    die;
}

if ($text == "➕ افزودن سود" && $user->step == "add_balance:select_type" && isAdmin($user->login_username)) {

    setStep($from_id, "add_profit:enter_username");

    $responseText =
        "➕ *افزودن سود*\n\n" .
        "لطفاً *شماره تلفن کاربر* را وارد نمایید.\n\n" .
        "📱 این سود به حساب کاربر اضافه شده و قابل برداشت خواهد بود.";

    sendMessage($from_id, $responseText, $backToAdminKeyboard);
    die;
}

if ($text == "➕ افزودن پایه" && $user->step == "add_balance:select_type" && isAdmin($user->login_username)) {

    setStep($from_id, "add_base:enter_username");

    $responseText =
        "➕ *افزودن مبلغ پایه سرمایه*\n\n" .
        "لطفاً *شماره تلفن کاربر* را وارد نمایید.\n\n" .
        "💰 این مبلغ به سرمایه پایه کاربر اضافه خواهد شد.";

    sendMessage($from_id, $responseText, $backToAdminKeyboard);
    die;
}

if ($user->step == "add_profit:enter_username" && isAdmin($user->login_username)) {

    $username = trim($text);

    $stmt = $pdo->prepare("SELECT `id` FROM `accounts` WHERE `username` = ?");
    $stmt->execute([$username]);
    if (!$stmt->fetch()) {
        sendMessage($from_id, "❌ *کاربر یافت نشد*\n\nکاربری با این شماره تلفن در سیستم یافت نشد.\n\n📱 لطفاً شماره تلفن صحیح را وارد نمایید.");
        die;
    }

    setStep($from_id, "add_profit:enter_amount:$username");

    $responseText =
        "💰 *مبلغ سود*\n\n" .
        "لطفاً مبلغ سود را به تومان وارد نمایید.\n\n" .
        "📝 *مثال:* 500000";

    sendMessage($from_id, $responseText);
    die;
}

if (strpos($user->step, "add_profit:enter_amount:") === 0 && isAdmin($user->login_username)) {

    $username = str_replace("add_profit:enter_amount:", "", $user->step);
    $amount   = intval($text);
    
    // اعتبارسنجی مبلغ
    if ($amount <= 0) {
        sendMessage($from_id, "❌ *مبلغ نامعتبر است*\n\nلطفاً یک عدد معتبر و بزرگتر از صفر وارد نمایید.");
        die;
    }

    // بررسی وجود کاربر
    $stmt = $pdo->prepare("SELECT `id` FROM `accounts` WHERE `username` = ?");
    $stmt->execute([$username]);
    if (!$stmt->fetch()) {
        sendMessage($from_id, "❌ *کاربر یافت نشد*\n\nکاربری با این شماره تلفن در سیستم یافت نشد.");
        die;
    }

    $trackingCode = uniqid("P");

    $pdo->prepare("UPDATE `accounts` SET `current_profit` = `current_profit` + ? WHERE `username` = ?")->execute([$amount, $username]);
    $pdo->prepare("INSERT INTO `profit_transactions` (`tracking_code`,`username`,`amount`) VALUES (?,?,?)")->execute([$trackingCode, $username, $amount]);

    $stmt = $pdo->prepare("SELECT `telegram_id` FROM `users` WHERE `login_username` = ?");
    $stmt->execute([$username]);
    while ( $u = $stmt->fetch() ) {
        sendMessage($u->telegram_id, "🎉 *سود جدید واریز شد*\n\n💰 مبلغ: *" . number_format($amount) . "* تومان\n📌 کد: `$trackingCode`\n\n✅ این مبلغ به سود قابل برداشت شما اضافه شد.");
    }

    setStep($from_id, "home");
    sendMessage($from_id, "✅ *سود با موفقیت ثبت شد*\n\nمبلغ سود به حساب کاربر اضافه شد و به وی اطلاع داده شد.", $adminMainKeyboard);
    die;
}

if ($user->step == "add_base:enter_username" && isAdmin($user->login_username)) {

    $username = trim($text);

    $stmt = $pdo->prepare("SELECT `id` FROM `accounts` WHERE `username` = ?");
    $stmt->execute([$username]);
    if (!$stmt->fetch()) {
        sendMessage($from_id, "❌ *کاربر یافت نشد*\n\nکاربری با این شماره تلفن در سیستم یافت نشد.\n\n📱 لطفاً شماره تلفن صحیح را وارد نمایید.");
        die;
    }

    setStep($from_id, "add_base:enter_amount:$username");

    $responseText =
        "💰 *مبلغ پایه سرمایه*\n\n" .
        "لطفاً مبلغ پایه سرمایه را به تومان وارد نمایید.\n\n" .
        "📝 *مثال:* 1000000";

    sendMessage($from_id, $responseText);
    die;
}

if (strpos($user->step, "add_base:enter_amount:") === 0 && isAdmin($user->login_username)) {

    $username = str_replace("add_base:enter_amount:", "", $user->step);
    $amount   = intval($text);
    
    // اعتبارسنجی مبلغ
    if ($amount <= 0) {
        sendMessage($from_id, "❌ *مبلغ نامعتبر است*\n\nلطفاً یک عدد معتبر و بزرگتر از صفر وارد نمایید.");
        die;
    }

    // بررسی وجود کاربر
    $stmt = $pdo->prepare("SELECT `id` FROM `accounts` WHERE `username` = ?");
    $stmt->execute([$username]);
    if (!$stmt->fetch()) {
        sendMessage($from_id, "❌ *کاربر یافت نشد*\n\nکاربری با این شماره تلفن در سیستم یافت نشد.");
        die;
    }

    $trackingCode = uniqid("B");

    $pdo->prepare("UPDATE `accounts` SET `base_investment` = `base_investment` + ? WHERE `username` = ?")->execute([$amount, $username]);
    $pdo->prepare("INSERT INTO `base_investment_transactions` (`tracking_code`,`username`,`amount`) VALUES (?,?,?)")->execute([$trackingCode, $username, $amount]);

    $stmt = $pdo->prepare("SELECT `telegram_id` FROM `users` WHERE `login_username` = ?");
    $stmt->execute([$username]);
    while ( $u = $stmt->fetch() ) {
        sendMessage($u->telegram_id, "💼 *افزایش سرمایه انجام شد*\n\n💰 مبلغ: *" . number_format($amount) . "* تومان\n📌 کد: `$trackingCode`\n\n✅ این مبلغ به سرمایه پایه شما اضافه شد.");
    }

    setStep($from_id, "home");
    sendMessage($from_id, "✅ *مبلغ پایه با موفقیت ثبت شد*\n\nمبلغ پایه سرمایه به حساب کاربر اضافه شد و به وی اطلاع داده شد.", $adminMainKeyboard);
    die;
}