1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-12 10:04:40 +02:00
RedMew/poll.lua

440 lines
12 KiB
Lua
Raw Normal View History

2018-06-17 15:51:06 +02:00
local Gui = require 'utils.gui'
local Global = require 'utils.global'
local Event = require 'utils.event'
2018-06-18 20:38:10 +02:00
local UserGroup = require 'user_groups'
2018-06-17 15:51:06 +02:00
2018-06-19 18:44:56 +02:00
local normal_color = {r = 1, g = 1, b = 1}
local focus_color = {r = 1, g = 0.55, b = 0.1}
2018-06-17 15:51:06 +02:00
local polls = {}
local no_notify_players = {}
2018-06-18 20:38:10 +02:00
local player_poll_data = {}
local player_poll_index = {}
2018-06-17 15:51:06 +02:00
Global.register(
2018-06-18 20:38:10 +02:00
{
polls = polls,
no_notify_players = no_notify_players,
player_poll_data = player_poll_data,
player_poll_index = player_poll_index
},
2018-06-17 15:51:06 +02:00
function(tbl)
polls = tbl.polls
no_notify_players = tbl.no_notify_players
2018-06-18 20:38:10 +02:00
player_poll_data = tbl.player_poll_data
player_poll_index = tbl.player_poll_index
2018-06-17 15:51:06 +02:00
end
)
local main_button_name = Gui.uid_name()
local main_frame_name = Gui.uid_name()
local create_poll_button_name = Gui.uid_name()
local notify_checkbox_name = Gui.uid_name()
2018-06-18 20:38:10 +02:00
local poll_view_back_name = Gui.uid_name()
local poll_view_forward_name = Gui.uid_name()
local poll_view_vote_name = Gui.uid_name()
2018-06-17 15:51:06 +02:00
local create_poll_frame_name = Gui.uid_name()
local create_poll_label_name = Gui.uid_name()
local create_poll_textfield_name = Gui.uid_name()
local create_poll_close_name = Gui.uid_name()
2018-06-18 20:38:10 +02:00
local create_poll_clear_name = Gui.uid_name()
2018-06-17 15:51:06 +02:00
local create_poll_confirm_name = Gui.uid_name()
local function player_joined(event)
local player = game.players[event.player_index]
if not player or not player.valid then
return
end
if player.gui.top[main_button_name] ~= nil then
return
end
player.gui.top.add {type = 'sprite-button', name = main_button_name, sprite = 'item/programmable-speaker'}
end
2018-06-19 17:33:33 +02:00
local function redraw_poll_viewer_content(data)
local poll_viewer_content = data.poll_viewer_content
local poll_index = data.poll_index
data.vote_buttons = nil
Gui.remove_data_recursivly(poll_viewer_content)
poll_viewer_content.clear()
2018-06-18 20:38:10 +02:00
local poll = polls[poll_index]
if not poll then
return
end
2018-06-19 18:44:56 +02:00
local question_label = poll_viewer_content.add {type = 'label', caption = poll.question}
question_label.style.height = 32
question_label.style.font_color = focus_color
question_label.style.font = 'default-listbox'
2018-06-18 20:38:10 +02:00
2018-06-19 17:33:33 +02:00
local grid = poll_viewer_content.add {type = 'table', column_count = 2}
2018-06-18 20:38:10 +02:00
2018-06-19 17:33:33 +02:00
local vote_buttons = {}
2018-06-18 20:38:10 +02:00
for i, a in ipairs(poll.answers) do
local vote_button =
grid.add({type = 'flow'}).add {type = 'button', name = poll_view_vote_name, caption = a.voted_count}
2018-06-19 18:44:56 +02:00
vote_button.style.height = 24
vote_button.style.font = 'default-small'
vote_button.style.top_padding = 0
vote_button.style.bottom_padding = 0
2018-06-19 17:33:33 +02:00
Gui.set_data(vote_button, {vote_index = i, data = data})
vote_buttons[i] = vote_button
2018-06-19 18:44:56 +02:00
local label = grid.add {type = 'label', caption = a.text}
label.style.height = 24
2018-06-18 20:38:10 +02:00
end
2018-06-19 17:33:33 +02:00
data.vote_buttons = vote_buttons
2018-06-18 20:38:10 +02:00
end
2018-06-19 17:33:33 +02:00
local function update_poll_viewer(data)
local back_button = data.back_button
local forward_button = data.forward_button
local poll_index_label = data.poll_index_label
local poll_index = data.poll_index
poll_index = math.clamp(poll_index, 0, #polls)
2018-06-18 20:38:10 +02:00
if poll_index == 0 then
2018-06-19 17:33:33 +02:00
poll_index_label.caption = 'No Polls'
2018-06-18 20:38:10 +02:00
else
2018-06-19 17:33:33 +02:00
poll_index_label.caption = 'Poll ' .. poll_index .. ' / ' .. #polls
2018-06-18 20:38:10 +02:00
end
2018-06-19 17:33:33 +02:00
back_button.enabled = poll_index > 1
forward_button.enabled = poll_index < #polls
redraw_poll_viewer_content(data)
end
local function draw_main_frame(left, player)
local frame = left.add {type = 'frame', name = main_frame_name, caption = 'Polls', direction = 'vertical'}
2018-06-18 20:38:10 +02:00
local poll_viewer_top_flow = frame.add {type = 'flow', direction = 'horizontal'}
2018-06-19 17:33:33 +02:00
2018-06-18 20:38:10 +02:00
local back_button = poll_viewer_top_flow.add {type = 'button', name = poll_view_back_name, caption = '<'}
local forward_button = poll_viewer_top_flow.add {type = 'button', name = poll_view_forward_name, caption = '>'}
2018-06-19 17:33:33 +02:00
local poll_index_label = poll_viewer_top_flow.add {type = 'label'}
local poll_viewer_content = frame.add {type = 'scroll-pane'}
2018-06-19 18:44:56 +02:00
poll_viewer_content.style.maximal_height = 400
poll_viewer_content.style.width = 300
2018-06-19 17:33:33 +02:00
local poll_index = player_poll_index[player.index] or #polls
2018-06-18 20:38:10 +02:00
local data = {
back_button = back_button,
forward_button = forward_button,
poll_index_label = poll_index_label,
2018-06-19 17:33:33 +02:00
poll_viewer_content = poll_viewer_content,
poll_index = poll_index
2018-06-18 20:38:10 +02:00
}
Gui.set_data(frame, data)
Gui.set_data(back_button, data)
Gui.set_data(forward_button, data)
2018-06-19 17:33:33 +02:00
update_poll_viewer(data)
2018-06-17 15:51:06 +02:00
local bottom_flow = frame.add {type = 'flow', direction = 'horizontal'}
local left_flow = bottom_flow.add {type = 'flow'}
left_flow.style.align = 'left'
left_flow.style.horizontally_stretchable = true
left_flow.add {type = 'button', name = main_button_name, caption = 'Close'}
local right_flow = bottom_flow.add {type = 'flow'}
right_flow.style.align = 'right'
2018-06-18 20:38:10 +02:00
if player.admin or UserGroup.is_regular(player.name) then
right_flow.add {type = 'button', name = create_poll_button_name, caption = 'Create Poll'}
end
2018-06-17 15:51:06 +02:00
right_flow.add {
type = 'checkbox',
name = notify_checkbox_name,
caption = 'Show Polls',
state = not no_notify_players[player.index],
tooltip = 'Notify me when new polls are created'
}
end
local function toggle(event)
local left = event.player.gui.left
local main_frame = left[main_frame_name]
if main_frame then
Gui.remove_data_recursivly(main_frame)
main_frame.destroy()
local create_poll_frame = left[create_poll_frame_name]
if create_poll_frame and create_poll_frame.valid then
Gui.remove_data_recursivly(create_poll_frame)
create_poll_frame.destroy()
end
else
draw_main_frame(left, event.player)
end
end
2018-06-18 20:38:10 +02:00
local function add_answer_field(data)
local grid = data.grid
local answers = data.answers
local count = #answers + 1
local label =
grid.add({type = 'flow'}).add {
type = 'label',
name = create_poll_label_name,
caption = 'Answer #' .. count .. ':'
}
local textfield = grid.add({type = 'flow'}).add {type = 'textfield', name = create_poll_textfield_name}
2018-06-19 18:44:56 +02:00
textfield.style.width = 200
2018-06-18 20:38:10 +02:00
Gui.set_data(label, textfield)
answers[count] = textfield
Gui.set_data(textfield, data)
end
2018-06-17 15:51:06 +02:00
local function draw_create_poll_frame(event)
local left = event.player.gui.left
2018-06-18 20:38:10 +02:00
local frame = left[create_poll_frame_name]
if frame and frame.valid then
Gui.remove_data_recursivly(frame)
frame.destroy()
else
frame = left.add {type = 'frame', name = create_poll_frame_name, caption = 'New Poll', direction = 'vertical'}
2018-06-17 15:51:06 +02:00
2018-06-18 20:38:10 +02:00
local scroll_pane = frame.add {type = 'scroll-pane', direction = 'vertical', vertical_scroll_policy = 'always'}
2018-06-19 18:44:56 +02:00
scroll_pane.style.maximal_height = 400
scroll_pane.style.maximal_width = 300
2018-06-17 15:51:06 +02:00
2018-06-18 20:38:10 +02:00
local grid = scroll_pane.add {type = 'table', column_count = 2}
2018-06-19 18:44:56 +02:00
--grid.style.horizontally_stretchable = true
2018-06-17 15:51:06 +02:00
2018-06-18 20:38:10 +02:00
local question_label =
grid.add({type = 'flow'}).add {type = 'label', name = create_poll_label_name, caption = 'Question:'}
local question_textfield = grid.add({type = 'flow'}).add {type = 'textfield'}
2018-06-19 18:44:56 +02:00
question_textfield.style.width = 200
2018-06-17 15:51:06 +02:00
2018-06-18 20:38:10 +02:00
Gui.set_data(question_label, question_textfield)
2018-06-17 15:51:06 +02:00
2018-06-18 20:38:10 +02:00
local data = {
frame = frame,
grid = grid,
question = question_textfield,
answers = {}
}
2018-06-17 15:51:06 +02:00
2018-06-19 18:44:56 +02:00
for _ = 1, 30 do
2018-06-18 20:38:10 +02:00
add_answer_field(data)
end
2018-06-17 15:51:06 +02:00
2018-06-18 20:38:10 +02:00
local bottom_flow = frame.add {type = 'flow', direction = 'horizontal'}
local left_flow = bottom_flow.add {type = 'flow'}
left_flow.style.align = 'left'
left_flow.style.horizontally_stretchable = true
local close_button = left_flow.add {type = 'button', name = create_poll_close_name, caption = 'Close'}
Gui.set_data(close_button, frame)
2018-06-17 15:51:06 +02:00
2018-06-18 20:38:10 +02:00
local clear_button = left_flow.add {type = 'button', name = create_poll_clear_name, caption = 'Clear'}
Gui.set_data(clear_button, data)
local right_flow = bottom_flow.add {type = 'flow'}
right_flow.style.align = 'right'
local confirm_button =
right_flow.add {type = 'button', name = create_poll_confirm_name, caption = 'Create Poll'}
Gui.set_data(confirm_button, data)
end
end
local function show_new_poll(poll_index)
for _, p in ipairs(game.connected_players) do
if p.valid then
local frame = p.gui.left[main_frame_name]
if frame and frame.valid then
local data = Gui.get_data(frame)
2018-06-19 17:33:33 +02:00
data.poll_index = poll_index
2018-06-18 20:38:10 +02:00
2018-06-19 17:33:33 +02:00
update_poll_viewer(data)
2018-06-18 20:38:10 +02:00
end
end
end
end
local function create_poll(event)
local data = Gui.get_data(event.element)
local frame = data.frame
local question = data.question.text
if not question:find('%S') then
event.player.print('Sorry, the poll needs a question.')
return
end
local answers = {}
for _, a in ipairs(data.answers) do
local s = a.text
if s:find('%S') then
table.insert(answers, {text = s, voted_count = 0})
end
end
if #answers < 1 then
event.player.print('Sorry, the poll needs at least one answer.')
return
end
local tick = game.tick
local duration = 3 * 60 * 60
local poll_data = {
question = question,
answers = answers,
voters = {},
start_tick = tick,
end_tick = tick + duration,
created_by = event.player,
edited_by = {}
}
table.insert(polls, poll_data)
show_new_poll(#polls)
Gui.remove_data_recursivly(frame)
frame.destroy()
2018-06-17 15:51:06 +02:00
end
2018-06-19 17:33:33 +02:00
local function update_votes(poll_index)
for _, p in ipairs(game.connected_players) do
local frame = p.gui.left[main_frame_name]
if frame and frame.valid then
local data = Gui.get_data(frame)
if data.poll_index == poll_index then
redraw_poll_viewer_content(data)
end
end
end
end
local function vote(event)
local player_index = event.player_index
local voted_button = event.element
local button_data = Gui.get_data(voted_button)
local vote_index = button_data.vote_index
local data = button_data.data
local poll_index = data.poll_index
local poll = polls[poll_index]
local voters = poll.voters
local previous_vote = voters[player_index]
if previous_vote and previous_vote == vote_index then
return
end
local answers = poll.answers
if previous_vote then
local answer_data = answers[previous_vote]
answer_data.voted_count = answer_data.voted_count - 1
end
local answer_data = answers[vote_index]
answer_data.voted_count = answer_data.voted_count + 1
voters[player_index] = vote_index
update_votes(poll_index)
end
2018-06-17 15:51:06 +02:00
Event.add(defines.events.on_player_joined_game, player_joined)
Gui.on_click(main_button_name, toggle)
Gui.on_click(create_poll_button_name, draw_create_poll_frame)
2018-06-18 20:38:10 +02:00
Gui.on_click(
create_poll_label_name,
function(event)
local textfield = Gui.get_data(event.element)
textfield.focus()
end
)
2018-06-17 15:51:06 +02:00
Gui.on_click(
create_poll_close_name,
function(event)
local element = event.element
local frame = Gui.get_data(element)
Gui.remove_data_recursivly(frame)
frame.destroy()
end
)
2018-06-18 20:38:10 +02:00
Gui.on_click(
create_poll_clear_name,
function(event)
local data = Gui.get_data(event.element)
data.question.text = ''
for _, answer in ipairs(data.answers) do
answer.text = ''
end
end
)
Gui.on_click(create_poll_confirm_name, create_poll)
2018-06-17 15:51:06 +02:00
Gui.on_click(
notify_checkbox_name,
function(event)
local player_index = event.player_index
local checkbox = event.element
local new_state
if checkbox.state then
new_state = nil
else
new_state = true
end
no_notify_players[player_index] = new_state
end
)
2018-04-06 21:58:50 +02:00
2018-06-19 17:33:33 +02:00
Gui.on_click(
poll_view_back_name,
function(event)
local data = Gui.get_data(event.element)
data.poll_index = data.poll_index - 1
update_poll_viewer(data)
end
)
2017-06-13 13:16:07 +02:00
2018-06-19 17:33:33 +02:00
Gui.on_click(
poll_view_forward_name,
function(event)
local data = Gui.get_data(event.element)
data.poll_index = data.poll_index + 1
update_poll_viewer(data)
end
)
2017-06-13 13:16:07 +02:00
2018-06-19 17:33:33 +02:00
Gui.on_click(poll_view_vote_name, vote)