Weapon_OneKeyToMax

GameServer/Server/CallGS/Handlers/Weapon/Weapon_OneKeyToMax.cs

@@ -0,0 +1,183 @@
+using MikuSB.Data;
+using MikuSB.Database;
+using MikuSB.Database.Inventory;
+using MikuSB.Proto;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+
+namespace MikuSB.GameServer.Server.CallGS.Handlers.Weapon;
+
+[CallGSApi("Weapon_OneKeyToMax")]
+public class Weapon_OneKeyToMax : ICallGSHandler
+{
+    private const uint MaxBreak = 6;
+
+    public async Task Handle(Connection connection, string param, ushort seqNo)
+    {
+        var player = connection.Player!;
+        var req = JsonSerializer.Deserialize<OneKeyToMaxParam>(param);
+        if (req == null || req.Id <= 0 || req.TbBreakUpgradeMat == null || req.TbBreakUpgradeMat.Count == 0)
+        {
+            await CallGSRouter.SendScript(connection, "Weapon_OneKeyToMax", "{\"sErr\":\"error.BadParam\"}");
+            return;
+        }
+
+        var weapon = player.InventoryManager.GetWeaponItem((uint)req.Id);
+        if (weapon == null)
+        {
+            await CallGSRouter.SendScript(connection, "Weapon_OneKeyToMax", "{\"sErr\":\"error.BadParam\"}");
+            return;
+        }
+
+        var config = WeaponUpgradeConfig.Load();
+        if (!config.TryGetWeaponTemplate(weapon.TemplateId, out var targetTemplate))
+        {
+            await CallGSRouter.SendScript(connection, "Weapon_OneKeyToMax", "{\"sErr\":\"error.BadParam\"}");
+            return;
+        }
+
+        var inventory = player.InventoryManager.InventoryData;
+        var equippedWeaponIds = player.CharacterManager.CharacterData.Characters
+            .Select(x => x.WeaponUniqueId)
+            .Where(x => x != 0)
+            .ToHashSet();
+
+        // Validate all materials upfront before making any changes
+        foreach (var stage in req.TbBreakUpgradeMat)
+        {
+            if (stage == null || stage.Count < 3) continue;
+            var matList = stage[1].Deserialize<List<List<int>>>();
+            if (matList == null) continue;
+
+            foreach (var entry in matList)
+            {
+                if (entry == null || entry.Count < 2) continue;
+                var itemId = (uint)Math.Max(0, entry[0]);
+                var count = (uint)Math.Max(0, entry[1]);
+                if (itemId == 0 || count == 0) continue;
+
+                if (itemId == weapon.UniqueId)
+                {
+                    await CallGSRouter.SendScript(connection, "Weapon_OneKeyToMax", "{\"sErr\":\"tip.material_not_enough\"}");
+                    return;
+                }
+
+                var material = FindInventoryItem(inventory, itemId);
+                if (material == null || material.ItemCount < count)
+                {
+                    await CallGSRouter.SendScript(connection, "Weapon_OneKeyToMax", "{\"sErr\":\"tip.material_not_enough\"}");
+                    return;
+                }
+
+                if (material is GameWeaponInfo materialWeapon &&
+                    (materialWeapon.EquipAvatarId != 0 || equippedWeaponIds.Contains(materialWeapon.UniqueId)))
+                {
+                    await CallGSRouter.SendScript(connection, "Weapon_OneKeyToMax", "{\"sErr\":\"tip.material_not_enough\"}");
+                    return;
+                }
+            }
+        }
+
+        var syncItems = new List<Item>();
+        var weaponLevel = weapon.Level == 0 ? 1u : weapon.Level;
+        weapon.Level = weaponLevel;
+
+        // Process each break stage
+        foreach (var stage in req.TbBreakUpgradeMat)
+        {
+            if (stage == null || stage.Count < 3) continue;
+
+            var matList = stage[1].Deserialize<List<List<int>>>();
+            var doBreak = stage[2].GetInt32() == 1;
+
+            if (matList != null && matList.Count > 0)
+            {
+                // Aggregate materials, skip duplicates
+                var materialsUsed = new Dictionary<uint, uint>();
+                foreach (var entry in matList)
+                {
+                    if (entry == null || entry.Count < 2) continue;
+                    var itemId = (uint)Math.Max(0, entry[0]);
+                    var count = (uint)Math.Max(0, entry[1]);
+                    if (itemId == 0 || count == 0) continue;
+                    materialsUsed[itemId] = materialsUsed.GetValueOrDefault(itemId) + count;
+                }
+
+                ulong totalExp = 0;
+                foreach (var (itemId, count) in materialsUsed)
+                {
+                    var material = FindInventoryItem(inventory, itemId)!;
+                    if (config.TryGetMaterialGain(material, out var gainExp))
+                        totalExp += gainExp * count;
+                }
+
+                // Consume materials
+                foreach (var (itemId, count) in materialsUsed)
+                {
+                    var material = FindInventoryItem(inventory, itemId)!;
+                    material.ItemCount -= count;
+                    if (material.ItemCount == 0)
+                    {
+                        RemoveInventoryItem(inventory, itemId);
+                        var proto = material.ToProto();
+                        proto.Count = 0;
+                        syncItems.Add(proto);
+                    }
+                    else
+                    {
+                        syncItems.Add(material.ToProto());
+                    }
+                }
+
+                // Apply exp to weapon
+                if (totalExp > 0)
+                {
+                    var maxLevel = config.GetWeaponMaxLevel(targetTemplate.BreakLimitId, weapon.Break);
+                    var (newLevel, newExp) = config.ApplyWeaponExp(weapon.Level, weapon.Exp, totalExp, targetTemplate.Color, maxLevel);
+                    weapon.Level = newLevel;
+                    weapon.Exp = newExp;
+                }
+            }
+
+            // Perform break
+            if (doBreak && weapon.Break < MaxBreak)
+                weapon.Break++;
+        }
+
+        syncItems.Add(weapon.ToProto());
+        DatabaseHelper.SaveDatabaseType(inventory);
+
+        var finalMaxLevel = config.GetWeaponMaxLevel(targetTemplate.BreakLimitId, weapon.Break);
+        var bMaxUnlock = finalMaxLevel > 0 && weapon.Level >= finalMaxLevel;
+
+        var sync = new NtfSyncPlayer();
+        sync.Items.AddRange(syncItems);
+
+        await CallGSRouter.SendScript(connection, "Weapon_OneKeyToMax",
+            $"{{\"bMaxUnLock\":{(bMaxUnlock ? "true" : "false")}}}", sync);
+    }
+
+    private static BaseGameItemInfo? FindInventoryItem(InventoryData inventory, uint itemId)
+    {
+        if (inventory.Weapons.TryGetValue(itemId, out var weapon)) return weapon;
+        if (inventory.Skins.TryGetValue(itemId, out var skin)) return skin;
+        if (inventory.Items.TryGetValue(itemId, out var item)) return item;
+        return null;
+    }
+
+    private static void RemoveInventoryItem(InventoryData inventory, uint itemId)
+    {
+        inventory.Weapons.Remove(itemId);
+        inventory.Skins.Remove(itemId);
+        inventory.Items.Remove(itemId);
+    }
+}
+
+internal sealed class OneKeyToMaxParam
+{
+    [JsonPropertyName("Id")]
+    public int Id { get; set; }
+
+    [JsonPropertyName("tbBreakUpgradeMat")]
+    public List<List<System.Text.Json.JsonElement>> TbBreakUpgradeMat { get; set; } = [];
+}

GameServer/Server/CallGS/Handlers/Weapon/Weapon_Upgrade.cs

@@ -258,9 +258,9 @@ internal sealed class WeaponUpgradeConfig
         var weaponTemplates = GameData.WeaponData.Values.ToDictionary(
             x => GameResourceTemplateId.FromGdpl(x.Genre, x.Detail, x.Particular, x.Level),
             x => new MaterialTemplate(x.Color, x.ProvideExp, x.ConsumeGold, x.RecycleID, x.BreakLimitID));
-        var suppliesTemplates = GameData.SuppliesData.Values.ToDictionary(
+        var suppliesTemplates = GameData.AllSuppliesData
-            x => GameResourceTemplateId.FromGdpl(x.Genre, x.Detail, x.Particular, x.Level),
+            .GroupBy(x => GameResourceTemplateId.FromGdpl(x.Genre, x.Detail, x.Particular, x.Level))
-            x => new MaterialTemplate(x.Color, x.ProvideExp, x.ConsumeGold, 0, 0));
+            .ToDictionary(g => g.Key, g => new MaterialTemplate(g.First().Color, g.First().ProvideExp, g.First().ConsumeGold, 0, 0));
 
         return new WeaponUpgradeConfig(normalExp, ssrExp, breakLimits, recycleById, weaponTemplates, suppliesTemplates);
     }
@@ -271,6 +271,24 @@ internal sealed class WeaponUpgradeConfig
     public bool TryGetSuppliesTemplate(ulong templateId, out MaterialTemplate template) =>
         _suppliesTemplates.TryGetValue(templateId, out template!);
 
+    public bool TryGetMaterialGain(BaseGameItemInfo item, out ulong exp)
+    {
+        exp = 0;
+        if (TryGetWeaponTemplate(item.TemplateId, out var weaponTemplate))
+        {
+            exp = weaponTemplate.ProvideExp;
+            if (item is GameWeaponInfo weapon && weapon.Level > 1)
+                exp += GetWeaponRecycleExp(weaponTemplate, weapon.Level);
+            return true;
+        }
+        if (TryGetSuppliesTemplate(item.TemplateId, out var suppliesTemplate))
+        {
+            exp = suppliesTemplate.ProvideExp;
+            return true;
+        }
+        return false;
+    }
+
     public ulong GetWeaponRecycleExp(MaterialTemplate template, uint level)
     {
         if (template.RecycleId <= 0 || !_recycleById.TryGetValue(template.RecycleId, out var recycle))