From f4b5e7d034c8155a223fdf51d099ac983a8207d9 Mon Sep 17 00:00:00 2001
From: Yang Chun Ung <qooraven@gmail.com>
Date: Wed, 20 Mar 2024 20:02:33 +0900
Subject: [PATCH] Fix stat conversion debuff check

---
 .Lib9c.Tests/Model/Skill/BuffFactoryTest.cs | 62 +++++++++++++++++++++
 Lib9c/Model/Buff/StatBuff.cs                |  4 +-
 2 files changed, 64 insertions(+), 2 deletions(-)

diff --git a/.Lib9c.Tests/Model/Skill/BuffFactoryTest.cs b/.Lib9c.Tests/Model/Skill/BuffFactoryTest.cs
index f5fad4e550..6b03467d54 100644
--- a/.Lib9c.Tests/Model/Skill/BuffFactoryTest.cs
+++ b/.Lib9c.Tests/Model/Skill/BuffFactoryTest.cs
@@ -149,5 +149,67 @@ public void Thorns()
             Assert.NotNull(buff3.CustomField);
             Assert.True(buff3.CustomField.Value.BuffValue > buff2.CustomField.Value.BuffValue);
         }
+
+        [Theory]
+        [InlineData(204003, false)]
+        [InlineData(206002, true)]
+        public void IsDebuff(int buffId, bool hasCustom)
+        {
+            var player = new Player(
+                level: 1,
+                _tableSheets.CharacterSheet,
+                _tableSheets.CharacterLevelSheet,
+                _tableSheets.EquipmentItemSetEffectSheet);
+            var skillId = _tableSheets.SkillBuffSheet.Values.First(r => r.BuffIds.Contains(buffId)).SkillId;
+            var skillRow = _tableSheets.SkillSheet[skillId];
+            int power = hasCustom ? 0 : 100;
+            int statPower = hasCustom ? 250 : 0;
+            StatType referencedStat = hasCustom ? StatType.HP : StatType.NONE;
+            var skill = SkillFactory.Get(skillRow, power, 100, statPower, referencedStat);
+            var buffs = BuffFactory.GetBuffs(
+                player.Stats,
+                skill,
+                _tableSheets.SkillBuffSheet,
+                _tableSheets.StatBuffSheet,
+                _tableSheets.SkillActionBuffSheet,
+                _tableSheets.ActionBuffSheet,
+                hasCustom
+            );
+            var buff = Assert.IsType<StatBuff>(buffs.Single(r => r.BuffInfo.Id == buffId));
+            Assert.Equal(buff.CustomField is not null, hasCustom);
+            Assert.False(buff.IsBuff());
+            Assert.True(buff.IsDebuff());
+        }
+
+        [Theory]
+        [InlineData(102001, false)]
+        [InlineData(102003, true)]
+        public void IsBuff(int buffId, bool hasCustom)
+        {
+            var player = new Player(
+                level: 1,
+                _tableSheets.CharacterSheet,
+                _tableSheets.CharacterLevelSheet,
+                _tableSheets.EquipmentItemSetEffectSheet);
+            var skillId = _tableSheets.SkillBuffSheet.Values.First(r => r.BuffIds.Contains(buffId)).SkillId;
+            var skillRow = _tableSheets.SkillSheet[skillId];
+            int power = hasCustom ? 0 : 100;
+            int statPower = hasCustom ? 250 : 0;
+            StatType referencedStat = hasCustom ? StatType.ATK : StatType.NONE;
+            var skill = SkillFactory.Get(skillRow, power, 100, statPower, referencedStat);
+            var buffs = BuffFactory.GetBuffs(
+                player.Stats,
+                skill,
+                _tableSheets.SkillBuffSheet,
+                _tableSheets.StatBuffSheet,
+                _tableSheets.SkillActionBuffSheet,
+                _tableSheets.ActionBuffSheet,
+                hasCustom
+            );
+            var buff = Assert.IsType<StatBuff>(buffs.Single(r => r.BuffInfo.Id == buffId));
+            Assert.NotNull(buff.CustomField);
+            Assert.True(buff.IsBuff());
+            Assert.False(buff.IsDebuff());
+        }
     }
 }
diff --git a/Lib9c/Model/Buff/StatBuff.cs b/Lib9c/Model/Buff/StatBuff.cs
index 348adc9967..71b3ebe573 100644
--- a/Lib9c/Model/Buff/StatBuff.cs
+++ b/Lib9c/Model/Buff/StatBuff.cs
@@ -44,12 +44,12 @@ public StatModifier GetModifier()
 
         public override bool IsBuff()
         {
-            return RowData.Value >= 0;
+            return !IsDebuff();
         }
 
         public override bool IsDebuff()
         {
-            return RowData.Value < 0;
+            return RowData.Value < 0 || CustomField?.BuffValue < 0;
         }
 
         public override object Clone()