-- SPDX-License-Identifier: GPL-3.0-or-later

local extension = Package:new("hegemony_cards", Package.CardPack)
extension.extensionName = "hegemony"

local H = require "packages/hegemony/util"
local U = require "packages/utility/utility"

Fk:loadTranslationTable{
  ["hegemony_cards"] = "国战标准版",
}

extension.game_modes_whitelist = { 'nos_heg_mode', 'new_heg_mode' }
extension.game_modes_blacklist = {"aaa_role_mode", "m_1v1_mode", "m_1v2_mode", "m_2v2_mode", "zombie_mode", "chaos_mode"}


extension:addCards{
  Fk:cloneCard("slash", Card.Spade, 5),
  Fk:cloneCard("slash", Card.Spade, 7),
  Fk:cloneCard("slash", Card.Spade, 8),
  Fk:cloneCard("slash", Card.Spade, 8),
  Fk:cloneCard("slash", Card.Spade, 9),
  Fk:cloneCard("slash", Card.Spade, 10),
  Fk:cloneCard("slash", Card.Spade, 11),
  Fk:cloneCard("slash", Card.Heart, 10),
  Fk:cloneCard("slash", Card.Heart, 12),
  Fk:cloneCard("slash", Card.Club, 2),
  Fk:cloneCard("slash", Card.Club, 3),
  Fk:cloneCard("slash", Card.Club, 4),
  Fk:cloneCard("slash", Card.Club, 5),
  Fk:cloneCard("slash", Card.Club, 8),
  Fk:cloneCard("slash", Card.Club, 9),
  Fk:cloneCard("slash", Card.Club, 10),
  Fk:cloneCard("slash", Card.Club, 11),
  Fk:cloneCard("slash", Card.Club, 12),
  Fk:cloneCard("slash", Card.Diamond, 10),
  Fk:cloneCard("slash", Card.Diamond, 11),
  Fk:cloneCard("slash", Card.Diamond, 12),

  Fk:cloneCard("thunder__slash", Card.Spade, 6),
  Fk:cloneCard("thunder__slash", Card.Spade, 7),
  Fk:cloneCard("thunder__slash", Card.Club, 6),
  Fk:cloneCard("thunder__slash", Card.Club, 7),
  Fk:cloneCard("thunder__slash", Card.Club, 8),

  Fk:cloneCard("fire__slash", Card.Heart, 4),
  Fk:cloneCard("fire__slash", Card.Diamond, 4),
  Fk:cloneCard("fire__slash", Card.Diamond, 5),

  Fk:cloneCard("jink", Card.Heart, 2),
  Fk:cloneCard("jink", Card.Heart, 11),
  Fk:cloneCard("jink", Card.Heart, 13),
  Fk:cloneCard("jink", Card.Diamond, 2),
  Fk:cloneCard("jink", Card.Diamond, 3),
  Fk:cloneCard("jink", Card.Diamond, 6),
  Fk:cloneCard("jink", Card.Diamond, 7),
  Fk:cloneCard("jink", Card.Diamond, 7),
  Fk:cloneCard("jink", Card.Diamond, 8),
  Fk:cloneCard("jink", Card.Diamond, 8),
  Fk:cloneCard("jink", Card.Diamond, 9),
  Fk:cloneCard("jink", Card.Diamond, 10),
  Fk:cloneCard("jink", Card.Diamond, 11),
  Fk:cloneCard("jink", Card.Diamond, 13),

  Fk:cloneCard("peach", Card.Heart, 4),
  Fk:cloneCard("peach", Card.Heart, 6),
  Fk:cloneCard("peach", Card.Heart, 7),
  Fk:cloneCard("peach", Card.Heart, 8),
  Fk:cloneCard("peach", Card.Heart, 9),
  Fk:cloneCard("peach", Card.Heart, 10),
  Fk:cloneCard("peach", Card.Heart, 12),
  Fk:cloneCard("peach", Card.Diamond, 2),

  Fk:cloneCard("analeptic", Card.Spade, 9),
  Fk:cloneCard("analeptic", Card.Club, 9),
  Fk:cloneCard("analeptic", Card.Diamond, 9),
}

extension:addCards{
  Fk:cloneCard("dismantlement", Card.Spade, 3),
  Fk:cloneCard("dismantlement", Card.Spade, 4),
  Fk:cloneCard("dismantlement", Card.Heart, 12),
  Fk:cloneCard("snatch", Card.Diamond, 3),
  Fk:cloneCard("snatch", Card.Spade, 3),
  Fk:cloneCard("snatch", Card.Spade, 4),
  Fk:cloneCard("duel", Card.Spade, 1),
  Fk:cloneCard("duel", Card.Club, 1),
  Fk:cloneCard("collateral", Card.Club, 12),
  Fk:cloneCard("ex_nihilo", Card.Heart, 7),
  Fk:cloneCard("ex_nihilo", Card.Heart, 8),
  Fk:cloneCard("nullification", Card.Spade, 11),
  H.hegNullification:clone(Card.Club, 13),
  H.hegNullification:clone(Card.Diamond, 12),
  Fk:cloneCard("savage_assault", Card.Spade, 13),
  Fk:cloneCard("savage_assault", Card.Club, 7),
  Fk:cloneCard("archery_attack", Card.Heart, 1),
  Fk:cloneCard("god_salvation", Card.Heart, 1),
  Fk:cloneCard("amazing_grace", Card.Heart, 3),
  Fk:cloneCard("lightning", Card.Spade, 1),
  Fk:cloneCard("indulgence", Card.Heart, 6),
  Fk:cloneCard("indulgence", Card.Club, 6),
  Fk:cloneCard("fire_attack", Card.Heart, 2),
  Fk:cloneCard("fire_attack", Card.Heart, 3),
  Fk:cloneCard("iron_chain", Card.Spade, 12),
  Fk:cloneCard("iron_chain", Card.Club, 12),
  Fk:cloneCard("iron_chain", Card.Club, 13),
  Fk:cloneCard("supply_shortage", Card.Spade, 10),
  Fk:cloneCard("supply_shortage", Card.Club, 10),
}

local befriendAttackingSkill = fk.CreateActiveSkill{
  name = "befriend_attacking_skill",
  prompt = "#befriend_attacking_skill",
  target_num = 1,
  mod_target_filter = function(self, to_select, selected, user)
    local target = Fk:currentRoom():getPlayerById(to_select)
    local player = Fk:currentRoom():getPlayerById(user)
    return player ~= target and H.compareKingdomWith(target, player, true)
  end,
  target_filter = Util.TargetFilter,
  can_use = function(self, player, card)
    return player.kingdom ~= "unknown" and not player:prohibitUse(card) 
  end,
  on_effect = function(self, room, effect)
    local player = room:getPlayerById(effect.from)
    local target = room:getPlayerById(effect.to)
    if target.dead then return end
    target:drawCards(1, "befriend_attacking")
    if player.dead then return end
    player:drawCards(3, "befriend_attacking")
  end
}
local befriendAttacking = fk.CreateTrickCard{
  name = "befriend_attacking",
  skill = befriendAttackingSkill,
}
extension:addCards({
  befriendAttacking:clone(Card.Heart, 9),
})
Fk:loadTranslationTable{
  ["befriend_attacking"] = "远交近攻",
  ["befriend_attacking_skill"] = "远交近攻",
  [":befriend_attacking"] = "锦囊牌<br/><b>时机</b>：出牌阶段<br/><b>目标</b>：有明置武将且势力与你不同的一名角色<br/><b>效果</b>：目标角色"..
  "摸一张牌，然后你摸三张牌。",
  ["#befriend_attacking_skill"] = "选择势力与你不同的一名角色，其摸一张牌，你摸三张牌",
}

local knownBothSkill = fk.CreateActiveSkill{
  name = "known_both_skill",
  prompt = "#known_both_skill",
  can_use = Util.CanUse,
  target_num = 1,
  mod_target_filter = function(self, to_select, selected, user)
    local target = Fk:currentRoom():getPlayerById(to_select)
    return Fk:currentRoom():getPlayerById(user) ~= target and (not target:isKongcheng() or target.general == "anjiang" or target.deputyGeneral == "anjiang")
  end,
  target_filter = Util.TargetFilter,
  on_effect = function(self, room, effect)
    local player = room:getPlayerById(effect.from)
    local target = room:getPlayerById(effect.to)
    if target.dead or player.dead then return end
    local all_choices = {"known_both_main", "known_both_deputy", "known_both_hand"}
    local choices = table.clone(all_choices)
    if target:isKongcheng() then
      table.remove(choices)
    end
    if target.general ~= "anjiang" then
      table.remove(choices, 1)
    end
    if target.deputyGeneral ~= "anjiang" then
      table.removeOne(choices, "known_both_deputy")
    end
    if #choices == 0 then return end
    local choice = room:askForChoice(player, choices, self.name, "#known_both-choice::"..target.id, false, all_choices)
    if choice == "known_both_hand" then
      U.viewCards(player, target:getCardIds(Player.Hand), self.name, "#known_both-hand::"..target.id)
      room:sendLog{
        type = "#know_hand",
        from = player.id,
        toast = true,
      }
    else
      local general = choice == "known_both_main" and {target:getMark("__heg_general"), target.deputyGeneral, tostring(target.seat)} or {target.general, target:getMark("__heg_deputy"), tostring(target.seat)}
      room:askForCustomDialog(player, self.name, "packages/hegemony/qml/KnownBothBox.qml", general)
      room:sendLog{
        type = "#know_general",
        from = player.id,
        toast = true,
      }
    end
  end,
}
local knownBoth = fk.CreateTrickCard{
  name = "known_both",
  skill = knownBothSkill,
  special_skills = {"recast"},
}
extension:addCards{
  knownBoth:clone(Card.Club, 3),
  knownBoth:clone(Card.Club, 4),
}
Fk:loadTranslationTable{
  ["known_both"] = "知己知彼",
  ["known_both_skill"] = "知己知彼",
  [":known_both"] = "锦囊牌<br/><b>时机</b>：出牌阶段<br/><b>目标</b>：一名其他角色<br/><b>效果</b>：观看目标角色一张暗置的武将牌或其手牌。",
  ["#known_both-choice"] = "知己知彼：选择对 %dest 执行的一项",
  ["known_both_main"] = "观看主将",
  ["known_both_deputy"] = "观看副将",
  ["known_both_hand"] = "观看手牌",
  ["#KnownBothGeneral"] = "观看 %1 武将",
  ["#known_both-hand"] = "知己知彼：观看%dest的手牌",
  ["#known_both_skill"] = "选择一名其他角色，观看其一张暗置的武将牌或其手牌",
  ["#know_hand"] = "%from 观看了手牌",
  ["#know_general"] = "%from 观看了武将",
}

local awaitExhaustedSkill = fk.CreateActiveSkill{
  name = "await_exhausted_skill",
  prompt = "#await_exhausted_skill",
  mod_target_filter = function(self, to_select, selected, user)
    local target = Fk:currentRoom():getPlayerById(to_select)
    local player = Fk:currentRoom():getPlayerById(user)
    return H.compareKingdomWith(target, player, false)
  end,
  can_use = function(self, player, card)
    if player:prohibitUse(card) then return end
    for _, p in ipairs(Fk:currentRoom().alive_players) do
      if not player:isProhibited(p, card) and self:modTargetFilter(p.id, {}, player.id, card, true) then
        return true
      end
    end
  end,
  on_use = function(self, room, use)
    if not use.tos or #TargetGroup:getRealTargets(use.tos) == 0 then
      local player = room:getPlayerById(use.from)
      if player.kingdom == "unknown" then
        use.tos = {{use.from}}
      else
        use.tos = {}
        for _, p in ipairs(room:getAlivePlayers()) do
          if not player:isProhibited(p, use.card) and H.compareKingdomWith(p, player) then --权宜
            TargetGroup:pushTargets(use.tos, p.id)
          end
        end
      end
    end
  end,
  on_effect = function(self, room, effect)
    local target = room:getPlayerById(effect.to)
    if target.dead then return end
    target:drawCards(2, "await_exhausted")
    if target.dead then return end
      room:askForDiscard(target, 2, 2, true, self.name, false)
  end,
}
local awaitExhausted = fk.CreateTrickCard{
  name = "await_exhausted",
  skill = awaitExhaustedSkill,
  multiple_targets = true,
}
extension:addCards({
  awaitExhausted:clone(Card.Heart, 11),
  awaitExhausted:clone(Card.Diamond, 4),
})
Fk:loadTranslationTable{
  ["await_exhausted"] = "以逸待劳",
  ["await_exhausted_skill"] = "以逸待劳",
  [":await_exhausted"] = "锦囊牌<br/><b>时机</b>：出牌阶段<br/><b>目标</b>：你和与你势力相同的角色<br/><b>效果</b>：每名目标角色各摸两张牌，"..
  "然后弃置两张牌。",
  ["#await_exhausted_skill"] = "你和与你势力相同的角色各摸两张牌，然后弃置两张牌",
}

extension:addCards{
  Fk:cloneCard("crossbow", Card.Diamond, 1),
  Fk:cloneCard("qinggang_sword", Card.Spade, 6),
  Fk:cloneCard("ice_sword", Card.Spade, 2),
  Fk:cloneCard("double_swords", Card.Spade, 2),
}

local sixSwordsSkill = fk.CreateAttackRangeSkill{
  name = "#six_swords_skill",
  attached_equip = "six_swords",
  frequency = Skill.Compulsory,
  correct_func = function (self, from, to)
    if from.kingdom ~= "unknown" then
      if table.find(Fk:currentRoom().alive_players, function(p)
        return from ~= p and H.compareKingdomWith(from, p) and
          p:getEquipment(Card.SubtypeWeapon) and Fk:getCardById(p:getEquipment(Card.SubtypeWeapon)).name == "six_swords" end) then
        return 1
      end
    end
    return 0
  end,
}
Fk:addSkill(sixSwordsSkill)
local sixSwords = fk.CreateWeapon{
  name = "six_swords",
  suit = Card.Diamond,
  number = 2,
  attack_range = 2,
  equip_skill = sixSwordsSkill,
}
extension:addCard(sixSwords)
Fk:loadTranslationTable{
  ["six_swords"] = "吴六剑",
  [":six_swords"] = "装备牌·武器<br/><b>攻击范围</b>：２ <br/><b>武器技能</b>：锁定技，与你势力相同的其他角色攻击范围+1。",
}

extension:addCards{
  Fk:cloneCard("spear", Card.Spade, 12),
  Fk:cloneCard("axe", Card.Diamond, 5),
}

local tribladeSkill = fk.CreateTriggerSkill{
  name = "#triblade_skill",
  attached_equip = "triblade",
  events = {fk.Damage},
  can_trigger = function(self, event, target, player, data)
    return target == player and player:hasSkill(self) and data.card and data.card.trueName == "slash" and not data.to.dead and not data.chain and
      not player:isKongcheng() and table.find(player.room.alive_players, function(p) return data.to:distanceTo(p) == 1 and p ~= player end)
  end,
  on_cost = function(self, event, target, player, data)
    local room = player.room
    local targets = table.map(table.filter(room.alive_players, function(p)
      return data.to:distanceTo(p) == 1 and p ~= player end), Util.IdMapper)
    if #targets == 0 then return false end
    local to, card = room:askForChooseCardAndPlayers(player, targets, 1, 1, ".|.|.|hand", "#triblade-invoke::"..data.to.id, self.name, true)
    if #to > 0 then
      self.cost_data = {to[1], card}
      return true
    end
  end,
  on_use = function(self, event, target, player, data)
    local room = player.room
    local to = self.cost_data[1]
    room:notifySkillInvoked(player, self.name, "offensive")
    room:doIndicate(player.id, {to})
    room:throwCard(self.cost_data[2], self.name, player, player)
    room:damage{
      from = player,
      to = room:getPlayerById(to),
      damage = 1,
      skillName = self.name,
    }
  end
}
Fk:addSkill(tribladeSkill)
local triblade = fk.CreateWeapon{
  name = "triblade",
  suit = Card.Diamond,
  number = 12,
  attack_range = 3,
  equip_skill = tribladeSkill,
}
extension:addCard(triblade)
Fk:loadTranslationTable{
  ["triblade"] = "三尖两刃刀",
  ["#triblade_skill"] = "三尖两刃刀",
  [":triblade"] = "装备牌·武器<br/><b>攻击范围</b>：３ <br/><b>武器技能</b>：当你使用【杀】对目标角色造成伤害后，你可以弃置一张手牌，"..
  "对其距离1的一名其他角色造成1点伤害。",
  ["#triblade-invoke"] = "三尖两刃刀：你可以弃置一张手牌，对 %dest 距离1的一名其他角色造成1点伤害",
}

extension:addCards{
  Fk:cloneCard("fan", Card.Diamond, 1),
  Fk:cloneCard("kylin_bow", Card.Heart, 5),

  Fk:cloneCard("eight_diagram", Card.Spade, 2),
  Fk:cloneCard("nioh_shield", Card.Club, 2),
  Fk:cloneCard("silver_lion", Card.Club, 1),
  Fk:cloneCard("vine", Card.Club, 2),

  Fk:cloneCard("dilu", Card.Club, 5),
  Fk:cloneCard("jueying", Card.Spade, 5),
  Fk:cloneCard("zhuahuangfeidian", Card.Heart, 13),
  Fk:cloneCard("chitu", Card.Heart, 5),
  Fk:cloneCard("dayuan", Card.Spade, 13),
  Fk:cloneCard("zixing", Card.Diamond, 13),
}

return extension