--overwrites musket smoke

local eventful = require 'plugins.eventful'

eventful.onProjItemCheckMovement.musket = function(projectile)
    if projectile.distance_flown == 1 then
        setFireRate(projectile)
        bulletSmoke(projectile, 5)
    elseif projectile.distance_flown == 2 then
        bulletSmoke(projectile, 10)
    elseif projectile.distance_flown == 3 then
        bulletSmoke(projectile, 25)
    end
end

function bulletSmoke(projectile, amount)
    if projectile.item:getSubtype() ~= -1 and dfhack.items.getSubtypeDef(projectile.item:getType(),projectile.item:getSubtype()).id == "ITEM_AMMO_BULLETS" then
        dfhack.maps.spawnFlow(projectile.cur_pos, 5, -1, -1, amount)
    end
end

function setFireRate(projectile)
    if projectile.item:getSubtype() ~= -1 and dfhack.items.getSubtypeDef(projectile.item:getType(),projectile.item:getSubtype()).id == "ITEM_AMMO_BULLETS" then
        projectile.firer.counters.think_counter = 1
    end
end