from otree.api import *

doc = """
Sequential game (asymmetric)
"""


class C(BaseConstants):
    NAME_IN_URL = 'sequential'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1
    MAIN_TEMPLATE = __name__ + '/Decide.html'
    CHAT_TEMPLATE = __name__ + '/chat.html'


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    mixer = models.StringField(
        choices=['Pineapple juice', 'Orange juice', 'Cola', 'Milk'],
        label="Choose a mixer",
        widget=widgets.RadioSelect,
    )
    liqueur = models.StringField(
        choices=['Blue curacao', 'Triple sec', 'Amaretto', 'Kahlua'],
        label="Choose a liqueur",
        widget=widgets.RadioSelect,
    )
    spirit = models.StringField(
        choices=['Vodka', 'Rum', 'Gin', 'Tequila'],
        label="Choose a spirit",
        widget=widgets.RadioSelect,
    )


def chat_channel(player):
    group = player.group

    return 'Group {}'.format(group.id_in_subsession)


class Player(BasePlayer):
    pass


class Message(ExtraModel):
    group = models.Link(Group)
    sender = models.Link(Player)
    text = models.StringField()


def to_dict(msg: Message):
    return dict(sender=msg.sender.id_in_group, text=msg.text)


# PAGES
class P1(Page):
    form_model = 'group'
    form_fields = ['mixer']
    template_name = C.MAIN_TEMPLATE

    @staticmethod
    def is_displayed(player: Player):
        return player.id_in_group == 1

    def vars_for_template(player):
        return dict(
            channel=chat_channel(player)
        )


class WaitPage1(WaitPage):
    pass


class P2(Page):
    form_model = 'group'
    form_fields = ['liqueur']
    template_name = C.MAIN_TEMPLATE

    @staticmethod
    def is_displayed(player: Player):
        return player.id_in_group == 2

    def vars_for_template(player):
        return dict(
            channel=chat_channel(player)
        )


class WaitPage2(WaitPage):
    pass


class P3(Page):
    form_model = 'group'
    form_fields = ['spirit']
    template_name = C.MAIN_TEMPLATE

    @staticmethod
    def is_displayed(player: Player):
        return player.id_in_group == 3

    def vars_for_template(player):
        return dict(
            channel=chat_channel(player)
        )


class WaitPage3(WaitPage):
    pass


class MyPage(Page):
    # template_name = C.CHAT_TEMPLATE
    @staticmethod
    def js_vars(player: Player):
        return dict(my_id=player.id_in_group)

    @staticmethod
    def live_method(player: Player, data):
        my_id = player.id_in_group
        group = player.group

        if 'text' in data:
            text = data['text']
            msg = Message.create(group=group, sender=player, text=text)
            return {0: [to_dict(msg)]}
        return {my_id: [to_dict(msg) for msg in Message.filter(group=group)]}

    def vars_for_template(player: Player):
        sender = player.id_in_group
        return dict(
            sender = (sender==1),
            receiver = (sender==2)
        )


class Results(Page):
    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(players_with_contributions=group.get_players())


# page_sequence = [P1, WaitPage1, P2, WaitPage2, P3, WaitPage3, Results]
page_sequence = [P1, WaitPage1, P2, WaitPage2, MyPage, Results]