'use strict'; const Member = require('../../models/Member'); const ShopItem = require('../../models/Economy'); const Helpers = require('../../utils/helpers'); const config = require('../../config/config'); const logger = require('../../utils/logger'); class EconomySystem { static async getBalance(userId, guildId) { const member = await Member.getOrCreate(userId, guildId); return { wallet: member.wallet, bank: member.bank, total: member.wallet + member.bank }; } static async addWallet(userId, guildId, amount) { const member = await Member.getOrCreate(userId, guildId); member.wallet = Math.max(0, member.wallet + amount); await member.save(); return member; } static async addBank(userId, guildId, amount) { const member = await Member.getOrCreate(userId, guildId); member.bank = Math.max(0, Math.min(member.bank + amount, member.maxBank)); await member.save(); return member; } static async transfer(fromId, toId, guildId, amount) { const from = await Member.getOrCreate(fromId, guildId); if (from.wallet < amount) return { success: false, reason: 'Insufficient funds' }; const to = await Member.getOrCreate(toId, guildId); from.wallet -= amount; to.wallet += amount; await from.save(); await to.save(); return { success: true, from, to }; } static async deposit(userId, guildId, amount) { const member = await Member.getOrCreate(userId, guildId); if (member.wallet < amount) return { success: false, reason: 'Insufficient wallet balance' }; const space = member.maxBank - member.bank; const deposited = Math.min(amount, space); member.wallet -= deposited; member.bank += deposited; await member.save(); return { success: true, deposited, member }; } static async withdraw(userId, guildId, amount) { const member = await Member.getOrCreate(userId, guildId); if (member.bank < amount) return { success: false, reason: 'Insufficient bank balance' }; member.bank -= amount; member.wallet += amount; await member.save(); return { success: true, withdrawn: amount, member }; } static async daily(userId, guildId) { const member = await Member.getOrCreate(userId, guildId); const now = new Date(); const cooldown = 24 * 60 * 60 * 1000; if (member.lastDaily && now - member.lastDaily < cooldown) { const remaining = cooldown - (now - member.lastDaily); return { success: false, remaining }; } const amount = config.economy.dailyAmount; member.wallet += amount; member.lastDaily = now; await member.save(); return { success: true, amount, member }; } static async weekly(userId, guildId) { const member = await Member.getOrCreate(userId, guildId); const now = new Date(); const cooldown = 7 * 24 * 60 * 60 * 1000; if (member.lastWeekly && now - member.lastWeekly < cooldown) { const remaining = cooldown - (now - member.lastWeekly); return { success: false, remaining }; } const amount = config.economy.weeklyAmount; member.wallet += amount; member.lastWeekly = now; await member.save(); return { success: true, amount, member }; } static async work(userId, guildId) { const member = await Member.getOrCreate(userId, guildId); const now = new Date(); const cooldown = 60 * 60 * 1000; if (member.lastWork && now - member.lastWork < cooldown) { const remaining = cooldown - (now - member.lastWork); return { success: false, remaining }; } const amount = Helpers.randomInt(config.economy.workMin, config.economy.workMax); member.wallet += amount; member.lastWork = now; await member.save(); return { success: true, amount, member }; } static async crime(userId, guildId) { const member = await Member.getOrCreate(userId, guildId); const now = new Date(); const cooldown = 2 * 60 * 60 * 1000; if (member.lastCrime && now - member.lastCrime < cooldown) { const remaining = cooldown - (now - member.lastCrime); return { success: false, remaining }; } const failed = Math.random() < config.economy.crimeFail; const amount = Helpers.randomInt(config.economy.crimeMin, config.economy.crimeMax); if (failed) { const fine = Math.floor(amount * 0.5); member.wallet = Math.max(0, member.wallet - fine); member.lastCrime = now; await member.save(); return { success: true, won: false, amount: fine, member }; } member.wallet += amount; member.lastCrime = now; await member.save(); return { success: true, won: true, amount, member }; } static async rob(robberId, targetId, guildId) { const robber = await Member.getOrCreate(robberId, guildId); const target = await Member.getOrCreate(targetId, guildId); const now = new Date(); const cooldown = 3 * 60 * 60 * 1000; if (robber.lastRob && now - robber.lastRob < cooldown) { const remaining = cooldown - (now - robber.lastRob); return { success: false, remaining }; } if (target.wallet < 100) { return { success: false, reason: 'Target has less than 100 coins in their wallet.' }; } const failed = Math.random() < config.economy.robFail; robber.lastRob = now; if (failed) { const fine = Helpers.randomInt(100, 500); robber.wallet = Math.max(0, robber.wallet - fine); await robber.save(); return { success: true, won: false, amount: fine }; } const percentage = Helpers.randomInt( Math.floor(config.economy.robMin * 100), Math.floor(config.economy.robMax * 100) ) / 100; const stolen = Math.floor(target.wallet * percentage); target.wallet -= stolen; robber.wallet += stolen; await robber.save(); await target.save(); return { success: true, won: true, amount: stolen }; } static async getLeaderboard(guildId, type = 'total', limit = 10) { const sortField = type === 'wallet' ? 'wallet' : type === 'bank' ? 'bank' : { $add: ['$wallet', '$bank'] }; return Member.find({ guildId }).sort({ wallet: -1 }).limit(limit); } } module.exports = EconomySystem;