Implement_Advanced_Stock_Referral_Logic

src/controllers/userController.js

@@ -71,9 +71,13 @@ exports.handleReferral = async (ctx, referrerId) => {
             // We can fetch referrer user.
             const referrerUser = await User.findOne({ id: referrerId });
             if (referrerUser) {
+                // Increment Count
+                referrerUser.referralCount = (referrerUser.referralCount || 0) + 1;
+                await referrerUser.save();
+
                 const locales = require('../locales');
                 const rLang = locales[referrerUser.language] || locales.uz;
-                await ctx.telegram.sendMessage(referrerId, `${rLang.invite_referral} ${ctx.from.first_name}!`);
+                await ctx.telegram.sendMessage(referrerId, `${rLang.invite_referral} ${ctx.from.first_name}!\n📈 Jami takliflar: ${referrerUser.referralCount}`);
             }
         } catch (e) {
             // Referrer might have blocked bot
@@ -86,7 +90,11 @@ exports.inviteFriends = async (ctx) => {
     const botUsername = ctx.botInfo.username;
     const inviteLink = `https://t.me/${botUsername}?start=${ctx.from.id}`;
 
-    const text = `${ctx.i18n.invite_text}\n\n🔗 ${inviteLink}\n\n👇`;
+    // Get Stats
+    const user = await User.findOne({ id: ctx.from.id });
+    const count = user ? (user.referralCount || 0) : 0;
+
+    const text = `${ctx.i18n.invite_text}\n\n📈 **Sizning natijangiz:** ${count} ta do'st taklif qilingan.\n\n🔗 ${inviteLink}\n\n👇`;
 
     const forwardText = `${ctx.i18n.invite_forward}${inviteLink}`;
 
@@ -593,11 +601,53 @@ exports.showOrderDetails = async (ctx, orderId) => {
 
         text += `\n💰 **Jami: ${order.total} so'm**`;
 
-        const keyboard = Markup.inlineKeyboard([[Markup.button.callback("🔙 Orqaga", "my_orders_list")]]);
+        const buttons = [];
+        // Allow cancellation only if status is 'new'
+        if (order.status === 'new') {
+            buttons.push([Markup.button.callback("❌ Bekor qilish", `user_cancel_order_${order.id}`)]);
+        }
+        buttons.push([Markup.button.callback("🔙 Orqaga", "my_orders_list")]);
+
+        ctx.editMessageText(text, { parse_mode: 'HTML', ...Markup.inlineKeyboard(buttons) })
+            .catch(() => ctx.replyWithHTML(text, Markup.inlineKeyboard(buttons)));
+    } catch (err) {
+        console.error(err);
+        ctx.reply("Xatolik");
+    }
+};
+
+// User Cancel Order (Restock)
+exports.cancelUserOrder = async (ctx, orderId) => {
+    try {
+        const id = parseInt(orderId);
+        const order = await Order.findOne({ id: id, userId: ctx.from.id });
+
+        if (!order) return ctx.answerCbQuery("Buyurtma topilmadi");
+        if (order.status !== 'new') return ctx.answerCbQuery("Ushbu buyurtmani bekor qilib bo'lmaydi");
+
+        // Restock Logic
+        const Product = require('../models/Product');
+        for (const item of order.items) {
+            await Product.updateOne({ id: item.id }, { $inc: { quantity: item.count } });
+        }
+
+        await Order.updateOne({ id: id }, { status: 'canceled' });
+
+        ctx.answerCbQuery("Buyurtma bekor qilindi ❌");
 
-        ctx.editMessageText(text, { parse_mode: 'HTML', ...keyboard })
-            .catch(() => ctx.replyWithHTML(text, keyboard));
+        // Notify Admin about cancellation (Optional but good logic)
+        const config = require('../config');
+        for (const adminId of config.ADMIN_IDS) {
+            ctx.telegram.sendMessage(adminId, `⚠️ <b>Buyurtma bekor qilindi!</b>\n#${id} bekor qilindi (Foydalanuvchi tomonidan).`, { parse_mode: 'HTML' }).catch(e => { });
+        }
+
+        // Refresh View
+        exports.showOrderDetails(ctx, id);
     } catch (err) {
         console.error(err);
+        ctx.reply("Xatolik");
+    }
+};
+console.error(err);
     }
 };

src/main.js

@@ -152,6 +152,9 @@ bot.action(/order_deliver_(.+)/, (ctx) => adminController.deliverOrder(ctx, ctx.
 bot.action(/gen_invoice_(.+)/, (ctx) => adminController.showInvoice(ctx, ctx.match[1]));
 bot.action(/delete_prod_(.+)/, (ctx) => adminController.deleteProduct(ctx, ctx.match[1]));
 
+// User Actions
+bot.action(/user_cancel_order_(.+)/, (ctx) => userController.cancelUserOrder(ctx, ctx.match[1]));
+
 // User Commands
 bot.start(async (ctx) => {
     try {

src/models/User.js

@@ -12,6 +12,7 @@ const userSchema = new mongoose.Schema({
         longitude: Number
     }],
     referredBy: { type: String, default: null },
+    referralCount: { type: Number, default: 0 },
     isBlocked: { type: Boolean, default: false },
     createdAt: { type: Date, default: Date.now }
 });