zulip/static/js/poll_widget.js

const render_widgets_poll_widget = require("../templates/widgets/poll_widget.hbs");
const render_widgets_poll_widget_results = require("../templates/widgets/poll_widget_results.hbs");

exports.poll_data_holder = function (is_my_poll, question, options) {
    // This object just holds data for a poll, although it
    // works closely with the widget's concept of how data
    // should be represented for rendering, plus how the
    // server sends us data.
    const self = {};

    const me = people.my_current_user_id();
    let poll_question = question;
    const key_to_option = new Map();
    let my_idx = 1;

    let input_mode = is_my_poll; // for now

    self.set_question = function (new_question) {
        input_mode = false;
        poll_question = new_question;
    };

    self.get_question = function () {
        return poll_question;
    };

    self.set_input_mode = function () {
        input_mode = true;
    };

    self.clear_input_mode = function () {
        input_mode = false;
    };

    self.get_input_mode = function () {
        return input_mode;
    };

    if (question) {
        self.set_question(question);
    }

    self.get_widget_data = function () {
        const options = [];

        for (const [key, obj] of key_to_option) {
            const voters = Array.from(obj.votes.keys());
            const current_user_vote = voters.includes(me);

            options.push({
                option: obj.option,
                names: people.safe_full_names(voters),
                count: voters.length,
                key,
                current_user_vote,
            });
        }

        const widget_data = {
            options,
            question: poll_question,
        };

        return widget_data;
    };

    self.handle = {
        new_option: {
            outbound(option) {
                const event = {
                    type: "new_option",
                    idx: my_idx,
                    option,
                };

                my_idx += 1;

                return event;
            },

            inbound(sender_id, data) {
                const idx = data.idx;
                const key = sender_id + "," + idx;
                const option = data.option;
                const votes = new Map();

                key_to_option.set(key, {
                    option,
                    user_id: sender_id,
                    votes,
                });

                if (my_idx <= idx) {
                    my_idx = idx + 1;
                }
            },
        },

        question: {
            outbound(question) {
                const event = {
                    type: "question",
                    question,
                };
                if (is_my_poll) {
                    return event;
                }
                return;
            },

            inbound(sender_id, data) {
                self.set_question(data.question);
            },
        },

        vote: {
            outbound(key) {
                let vote = 1;

                // toggle
                if (key_to_option.get(key).votes.get(me)) {
                    vote = -1;
                }

                const event = {
                    type: "vote",
                    key,
                    vote,
                };

                return event;
            },

            inbound(sender_id, data) {
                const key = data.key;
                const vote = data.vote;
                const option = key_to_option.get(key);

                if (option === undefined) {
                    blueslip.warn("unknown key for poll: " + key);
                    return;
                }

                const votes = option.votes;

                if (vote === 1) {
                    votes.set(sender_id, 1);
                } else {
                    votes.delete(sender_id);
                }
            },
        },
    };

    self.handle_event = function (sender_id, data) {
        const type = data.type;
        if (self.handle[type]) {
            self.handle[type].inbound(sender_id, data);
        }
    };

    // function to check whether option already exists
    self.is_option_present = function (data, latest_option) {
        return data.some((el) => el.option === latest_option);
    };

    // function to add all options added along with the /poll command
    for (const [i, option] of options.entries()) {
        self.handle.new_option.inbound("canned", {
            idx: i,
            option,
        });
    }

    return self;
};

exports.activate = function (opts) {
    const elem = opts.elem;
    const callback = opts.callback;

    let question = "";
    let options = [];
    if (opts.extra_data) {
        question = opts.extra_data.question || "";
        options = opts.extra_data.options || [];
    }

    const is_my_poll = people.is_my_user_id(opts.message.sender_id);
    const poll_data = exports.poll_data_holder(is_my_poll, question, options);

    function update_edit_controls() {
        const has_question = elem.find("input.poll-question").val().trim() !== "";
        elem.find("button.poll-question-check").toggle(has_question);
    }

    function render_question() {
        const question = poll_data.get_question();
        const input_mode = poll_data.get_input_mode();
        const can_edit = is_my_poll && !input_mode;
        const has_question = question.trim() !== "";
        const can_vote = has_question;
        const waiting = !is_my_poll && !has_question;
        const author_help = is_my_poll && !has_question;

        elem.find(".poll-question-header").toggle(!input_mode);
        elem.find(".poll-question-header").text(question);
        elem.find(".poll-edit-question").toggle(can_edit);
        update_edit_controls();

        elem.find(".poll-question-bar").toggle(input_mode);
        elem.find(".poll-option-bar").toggle(can_vote);

        elem.find(".poll-please-wait").toggle(waiting);

        elem.find(".poll-author-help").toggle(author_help);
    }

    function start_editing() {
        poll_data.set_input_mode();

        const question = poll_data.get_question();
        elem.find("input.poll-question").val(question);
        render_question();
        elem.find("input.poll-question").trigger("focus");
    }

    function abort_edit() {
        poll_data.clear_input_mode();
        render_question();
    }

    function submit_question() {
        const poll_question_input = elem.find("input.poll-question");
        let new_question = poll_question_input.val().trim();
        const old_question = poll_data.get_question();

        // We should disable the button for blank questions,
        // so this is just defensive code.
        if (new_question.trim() === "") {
            new_question = old_question;
        }

        // Optimistically set the question locally.
        poll_data.set_question(new_question);
        render_question();

        // If there were no actual edits, we can exit now.
        if (new_question === old_question) {
            return;
        }

        // Broadcast the new question to our peers.
        const data = poll_data.handle.question.outbound(new_question);
        callback(data);
    }

    function submit_option() {
        const poll_option_input = elem.find("input.poll-option");
        const option = poll_option_input.val().trim();
        const options = poll_data.get_widget_data().options;

        if (poll_data.is_option_present(options, option)) {
            return;
        }

        if (option === "") {
            return;
        }

        poll_option_input.val("").trigger("focus");

        const data = poll_data.handle.new_option.outbound(option);
        callback(data);
    }

    function submit_vote(key) {
        const data = poll_data.handle.vote.outbound(key);
        callback(data);
    }

    function build_widget() {
        const html = render_widgets_poll_widget();
        elem.html(html);

        elem.find("input.poll-question").on("keyup", (e) => {
            e.stopPropagation();
            update_edit_controls();
        });

        elem.find("input.poll-question").on("keydown", (e) => {
            e.stopPropagation();

            if (e.keyCode === 13) {
                submit_question();
                return;
            }

            if (e.keyCode === 27) {
                abort_edit();
                return;
            }
        });

        elem.find(".poll-edit-question").on("click", (e) => {
            e.stopPropagation();
            start_editing();
        });

        elem.find("button.poll-question-check").on("click", (e) => {
            e.stopPropagation();
            submit_question();
        });

        elem.find("button.poll-question-remove").on("click", (e) => {
            e.stopPropagation();
            abort_edit();
        });

        elem.find("button.poll-option").on("click", (e) => {
            e.stopPropagation();
            submit_option();
        });

        elem.find("input.poll-option").on("keydown", (e) => {
            e.stopPropagation();

            if (e.keyCode === 13) {
                submit_option();
                return;
            }

            if (e.keyCode === 27) {
                $("input.poll-option").val("");
                return;
            }
        });
    }

    function render_results() {
        const widget_data = poll_data.get_widget_data();

        const html = render_widgets_poll_widget_results(widget_data);
        elem.find("ul.poll-widget").html(html);

        elem.find("button.poll-vote")
            .off("click")
            .on("click", (e) => {
                e.stopPropagation();
                const key = $(e.target).attr("data-key");
                submit_vote(key);
            });
    }

    elem.handle_events = function (events) {
        for (const event of events) {
            poll_data.handle_event(event.sender_id, event.data);
        }

        render_question();
        render_results();
    };

    build_widget();
    render_question();
    render_results();
};

window.poll_widget = exports;