Add Weapon Mod 1.1

-- Add Weapon Mod 1.1
-- A Payday 2 LUA script to add specific weapon mods to inventory
-- Changelog:
--   1.1 - support for DLC weapon mods
-- Credits:
--   * notbulletproof
--   * acepilot1023978: original script
--   * B1313: code review
PD2 = class()

function PD2:getWeaponModPairs()
    return pairs(tweak_data.blackmarket.weapon_mods)
end

function PD2:getWeaponModName(mod_id)
    return self:localize(self:getWeaponModData(mod_id).name_id)
end

function PD2:getWeaponModDlc(mod_id)
    return self:getWeaponModData(mod_id).dlc or "normal"
end

function PD2:getWeaponModData(mod_id)
    return tweak_data.blackmarket.weapon_mods[mod_id]
end

function PD2:isLocalized(name_id)
    return name_id and managers.localization:exists(name_id)
end

function PD2:localize(name_id)
    if (self:isLocalized(name_id)) then
        return managers.localization:text(name_id)
    end

    return name_id
end

function PD2:getWeaponModAmount(mod_id)
    local dlc = self:getWeaponModDlc(mod_id)

    return managers.blackmarket:get_item_amount(dlc, "weapon_mods", mod_id, true)
end

function PD2:addWeaponMod(mod_id)
    local dlc = self:getWeaponModDlc(mod_id)

    self:println("Adding " .. self:getWeaponModName(mod_id) .. " to inventory" .. ", id=" .. mod_id .. ", dlc=" .. dlc)

    managers.blackmarket:add_to_inventory(dlc, "weapon_mods", mod_id, true)

    self:println("Added " .. self:getWeaponModName(mod_id) .. " to inventory, current stock " .. self:getWeaponModAmount(mod_id))
end

function PD2:println(text)
    io.stdout:write(text .. "\n")
end

function PD2:showDialog(title, options)
    show_sorted_dialog(title, "", options, nil, 20, 1)
end

WeaponModDialog = class()

function WeaponModDialog:init(pd2)
    self.pd2 = pd2
end

function WeaponModDialog:showIfNotInGame()
    if not inGame() then
        self:show()
    end
end

function WeaponModDialog:show()
    self.pd2:showDialog("Add Weapon Mods", self:getDialogOptions())
end

function WeaponModDialog:getDialogOptions()
    local dialog_options = {}
    local pd2 = self.pd2

    local function addWeaponModCallback(mod_id)
        pd2:addWeaponMod(mod_id)
    end

    local function sortByName(a, b)
        return a.text < b.text
    end

    for mod_id, data in pd2.getWeaponModPairs() do
        local name_id = data.name_id

        if (pd2:isLocalized(name_id)) then
            local qty = pd2:getWeaponModAmount(mod_id)
            local name = pd2:localize(name_id) .. " / " .. mod_id .. " (" .. qty .. ")"
            table.insert(dialog_options, {
                text = name,
                callback = addWeaponModCallback,
                data = mod_id })
        end
    end

    table.sort(dialog_options, sortByName)

    return dialog_options
end

local pd2 = PD2:new()
local weaponModDialog = WeaponModDialog:new(pd2)
weaponModDialog:showIfNotInGame()