tup.include("util/strict.lua")
tup.include("util/lua-ext.lua")
tup.include("util/tup-ext.lua")

-- Generate uniform size minisprites

function padsprite(input, output, w, h)
    tup.foreach_rule(
        input,
        "^ padsprite %f^ tools/padsprite.sh %f %o " .. w .. " " .. h,
        output
    )
end

for canon in iter{"canonical", "noncanonical"} do 
    for dir in iter{"asymmetrical", "custom", "misc", "pokemon"} do
        -- TODO fix this, tup hates git's quirk of not creating directories when
        -- there aren't any files
        if canon == "noncanonical" and dir ~= "pokemon" then
            goto continue
        end
            
        for file in iglob{"src/" .. canon .. "/minisprites/gen6/" .. dir .. "/*"} do
            padsprite(file,
                      "build/gen6-minisprites-padded/" .. canon .. "/" .. dir .. "/" .. tup.file(file),
                      40, 30)
        end
        ::continue::
    end
end

-- PS spritesheet

tup.rule(
    {"build/gen6-minisprites-padded/canonical/pokemon/*", "build/gen6-minisprites-padded/noncanonical/pokemon/*"},
    "node tools/sprites/ps.js build/gen6-minisprites-padded/canonical/pokemon/ build/gen6-minisprites-padded/noncanonical/pokemon/ --output-image build/ps/pokemonicons-sheet.png --output-metadata build/ps/pokemonicons.json",
    {"build/ps/pokemonicons-sheet.png", "build/ps/pokemonicons.json"}
)

-- Smogdex social images

function fbsprite(input, output)
    tup.foreach_rule(
        input,
        "^ fbsprite %f^ tools/fbsprite.sh %f %o",
        output
    )
end

function twittersprite(input, output)
    tup.foreach_rule(
        input,
        "^ twittersprite %f^ tools/twittersprite.sh %f %o",
        output
    )
end

local files =
    {"src/canonical/models/front/*", "src/noncanonical/models/front/*",
     -- TODO: only pick noncanonical that aren't already in models/
     "src/noncanonical/sprites/gen5/front/*"}

fbsprite(files, "build/smogon/fbsprites/xy/%B.png")
twittersprite(files, "build/smogon/twittersprites/xy/%B.png")

-- Trainers

function padtrainer(input, output)
    return "convert " .. input .. " -background transparent -gravity center -extent 80x80 " .. output
end

-- TODO: move some of these to util, when we figure out the precise abstractions desired

function compresspng(filename, opts)
    local cmds = {}
    if opts.optipng then
        cmds += "optipng -q " .. opts.optipng .. " " .. filename
    end
    if opts.advpng then
        cmds += "advpng -q " .. opts.advpng .. " " .. filename
    end
    return cmds
end

function makecmd(cmds)
    return table.concat(flatten(cmds), " && ")
end

tup.foreach_rule(
    {"src/canonical/trainers/*"},
    "^ pad trainer %f^ " .. makecmd{padtrainer("%f", "%o"), compresspng("%f",  {})},
    {"build/padded-trainers/canonical/%b"}
)