banners/init.lua

local MP = core.get_modpath("banners") .. "/"
dofile(MP .. "smartfs.lua")

banners = {}

banners.masks = {
    "bend_left", "bend_left_outline",
    "bend_right", "bend_right_outline",
    "central_circle", "central_circle_outline",
    "chevron_bottom", "chevron_bottom_outline",
    "chevron_top", "chevron_top_outline",
    "cross", "cross_outline",
    "nordic_cross", "nordic_cross_outline",
    "saltire", "saltire_outline",
    "mask_background", "david_star",
    "fleur_de_lis", "fullpentagram", "fullpentagram_outline",
    "greek_cross", "greek_cross_outline", "greek_cross_halfoutline",
    "triskel", "triskel_outline",
    "iron_cross", "iron_cross_outline",
    "hilal", "quadrisection_left",
    "quadrisection_right", "outline",
    "thirdfess_bottom", "thirdfess_middle", "thirdfess_top",
    "thirdpale_right", "thirdpale_middle", "thirdpale_left",
    "halfpale_right", "halfpale_left",
    "halffess_top", "halffess_down",
    "soviet_hammer_sickle", "per_bend_sinister_high",
    "per_bend_sinister_low", "per_bend_high", "per_bend_low",
    "canton", "star5_canton", "stripes_horiz_6", "stripes_horiz_4",
    "star_chevron", "checkered_8_4", "checkered_16_8"
}

-- The amount of transformations needs to be capped somewhere
-- to avoid crashing server.
-- It doesn't make sense to have any one mask multiple times
-- but users normally don't follow strict logic when being
-- creative, so we give them plenty of space to work with.
-- A better approach would be to strip duplicates and to
-- drop all previous masks if a full background is dropped
-- on top of other masks.
banners.max_transformations = #banners.masks * #banners.masks

banners.colors = {
    "black", "cyan", "green", "white",
    "blue", "darkblue", "red", "yellow",
    "grey", "orange", "pink", "violet",
    "brown", "darkbrown"
}

banners.base_transform = {
    texture = "bg_white.png",
    mask = "mask_background.png"
}

banners.creation_form_func = function(state)
    -- helper functions
    state.update_player_inv = function(self, transform_string)
        local player = core.get_player_by_name(self.player)
        local newbanner = player:get_wielded_item()
        newbanner:get_meta():set_string("", transform_string)
        player:set_wielded_item(newbanner)
    end
    state.update_preview = function(self, transform_string)
        self:get("banner_preview"):setImage(transform_string)
        self:get("color_indicator"):setImage(self.current_color)
    end
    state.update_all = function(self)
        local transform_string = self.banner:get_transform_string()
        self:update_preview(transform_string)
        self:update_player_inv(transform_string)
    end
    -- initialize with empty banner
    state.banner = banners.Banner:new(nil)
    state.banner:push_transform(banners.base_transform)
    state.current_color = "bg_white.png"
    state:size(20, 10)
    state:image(3, 0.4, 4, 2, "banner_preview", nil)
    state:image(2.4, 0.8, 0.7, 0.7, "color_indicator", state.current_color)
    state:update_all()
    -- color indicator
    -- undo button
    state:button(0.5, 0.3, 2, 1, "undo", "Undo"):click(function(_, state2)
        if #state2.banner.transforms > 1 then
            state2.banner:pop_transform()
            state2:update_all()
        end
    end)
    -- delete button
    state:button(0.5, 1.3, 2, 1, "delete", "Delete"):click(function(_, state2)
        state2.banner.transforms = { banners.base_transform }
        state2:update_all()
    end)
    -- add banners colors
    local x = 7
    local y = .3
    for _, color in ipairs(banners.colors) do
        local b = state:button(x, y, 1, 1, color, "")
        b:setImage("bg_" .. color .. ".png")
        b:click(function(self, state2)
            state2.current_color = "bg_" .. self.name .. ".png"
            state2:get("color_indicator"):setImage(state2.current_color)
            -- todo: update masks or something
        end)
        x = x + 1
        if x > 19 then
            y = y + 1
            x = 7
        end
    end
    -- add banners buttons
    x = 1
    y = 3
    for _, mask in ipairs(banners.masks) do
        local b = state:button(x, y, 2, 1, mask, "")
        b:setImage(mask .. ".png")
        b:click(function(self, state2)
            state2.banner:push_transform({
                texture = state2.current_color,
                mask = self.name .. ".png"
            })
            state2:update_all()
        end)
        x = x + 2
        if x > 17.5 then
            y = y + 1
            x = 1
        end
    end
    return true
end

banners.creation_form = smartfs.create("banners:banner_creation",
    banners.creation_form_func)


-- banner definition
banners.Banner = {}

function banners.Banner:new(banner)
    banner = banner or { transforms = {} }
    setmetatable(banner, self)
    self.__index = self
    return banner
end
function banners.Banner.push_transform(self, transform)
    table.insert(self.transforms, transform)
    if #self.transforms > banners.max_transformations then
        table.remove(self.transforms, 1)
    end
end
function banners.Banner.pop_transform(self)
    table.remove(self.transforms)
end
function banners.Banner.get_transform_string(self)
    local final = {}
    for _, transform in ipairs(self.transforms) do
        table.insert(final, "(" .. transform.texture
            .. "^[mask:" .. transform.mask .. "^[makealpha:0,0,0)")
    end
    local ret = table.concat(final, "^")
    return ret
end

-- helper function for determining the flag's direction
-- (pos, pointed_thing)
banners.determine_flag_direction = function(_, pointed_thing)
    local above = pointed_thing.above
    local under = pointed_thing.under
    local dir = {
        x = under.x - above.x,
        y = under.y - above.y,
        z = under.z - above.z
    }
    return core.dir_to_wallmounted(dir)
end

-- (itemstack, player, pointed_thing)
banners.banner_on_use = function(_, player)
    if player.is_player then
        banners.creation_form:show(player:get_player_name())
    end
end

banners.banner_on_dig = function(pos, node, player)
    if not player or core.is_protected(pos, player:get_player_name()) then
		return
	end
    local meta = core.get_meta(pos)
    local inventory = player:get_inventory()
    inventory:add_item("main", {
        name = node.name,
        count = 1,
        wear = 0,
        metadata = meta:get_string("banner")
    })
    core.remove_node(pos)
end

-- (pos, node, player)
banners.banner_on_destruct = function(pos)
    local objects = core.get_objects_inside_radius(pos, 0.5)
    for _, v in ipairs(objects) do
        local e = v:get_luaentity()
        if e and e.name == "banners:banner_ent" then
            v:remove()
        end
    end
end

-- (pos, player, itemstack, pointed_thing)
banners.banner_after_place = function(pos, _, itemstack, pointed_thing)
    core.get_node(pos).param2 = banners.determine_flag_direction(pos, pointed_thing)
    core.get_meta(pos):set_string("banner", itemstack:get_meta():get_string(""))
    core.add_entity(pos, "banners:banner_ent")
end

-- banner entity
local set_banner_texture = function(obj, texture)
    obj:set_properties({ textures = { "banner_uv_text.png^" .. texture } })
end


banners.banner_on_activate = function(self)
    local pos = self.object:get_pos()
    local banner = core.get_meta(pos):get_string("banner")
    local banner_face = core.get_node(pos).param2
    local yaw = 0.
    if banner_face == 2 then
        yaw = 0.
    elseif banner_face == 0 then
        yaw = 3.141592653589793 -- pi, 180 degrees
    elseif banner_face == 1 then
        yaw = 1.5707963267948966 -- pi / 2
    elseif banner_face == 3 then
        yaw = 4.71238898038469 -- 3 * pi / 2
    end
    self.object:set_yaw(yaw)
    if banner then
        set_banner_texture(self.object, banner)
    end
end

core.register_entity("banners:banner_ent", {
    initial_properties = {
        collisionbox = { 0, 0, 0, 0, 0, 0 },
        visual = "mesh",
        textures = { "banner_uv_text" },
        mesh = "banner_pole.x",
    },
    on_activate = banners.banner_on_activate,
})

if core.get_modpath("factions") then
    dofile(MP .. "factions.lua")
end

dofile(MP .. "items.lua")
dofile(MP .. "nodes.lua")
dofile(MP .. "crafts.lua")