"""Bridge where the final showdown with the AI occurs."""

import random

from pygame.colordict import THECOLORS
from pygame.color import Color
from pygame.rect import Rect
from albow.music import change_playlist, get_music, PlayList
from albow.resource import get_image

from gamelib.cursor import CursorSprite
from gamelib.state import Scene, Item, Thing, Result
from gamelib.sound import get_current_playlist
from gamelib.constants import DEBUG

from gamelib.scenes.game_constants import PLAYER_ID
from gamelib.scenes.scene_widgets import (Door, InteractText, InteractNoImage,
                                          InteractRectUnion, InteractImage,
                                          InteractAnimated, GenericDescThing,
                                          BaseCamera, make_jim_dialog)


class Bridge(Scene):

    FOLDER = "bridge"
    BACKGROUND = 'bridge.png'

    OFFSET = (0, -50)

    MUSIC = [
            'beep1.ogg',
            'beep2.ogg',
            'beep880.ogg',
            'beep660.ogg',
            'beep3.ogg',
            'silent.ogg',
            'creaking.ogg',
            'silent.ogg',
            ]

    INITIAL_DATA = {
        'ai status' : 'online', # online, looping, dead
        'ai panel'  : 'closed', # closed, open, broken
        }

    def __init__(self, state):
        super(Bridge, self).__init__(state)
        self.background_playlist = None
        self.add_item(Superconductor('superconductor'))
        self.add_item(Stethoscope('stethoscope'))
        self.add_thing(ToMap())
        self.add_thing(MonitorCamera())
        self.add_thing(MassageChair())
        self.add_thing(MassageChairBase())
        self.add_thing(StethoscopeThing())
        self.add_thing(BridgeComputer())
        self.add_thing(LeftLights())
        self.add_thing(RightLights())
        self.add_thing(JimPanel())
        self.add_thing(StarField())
        self.add_thing(GenericDescThing('bridge.wires', 1,
            "The brightly coloured wires contrast with the drab walls.",
            ((46, 4, 711, 143),)))
        self.add_thing(GenericDescThing('bridge.note', 2,
            "\"Dammit JIM, I'm a doctor, not an engineer!\"",
            (
                (491, 494, 194, 105),
                (422, 533, 71, 66),
                )))
        self.doctor = GenericDescThing('bridge.skel', 3,
                "A skeleton hangs improbably from the wires.",
                (
                    (632, 148, 40, 29),
                    (683, 176, 30, 101),
                    (652, 274, 45, 96),
                    (639, 180, 11, 95),
                    ))
        self.add_thing(self.doctor)


    def enter(self):
        pieces = [get_music(x, prefix='sounds') for x in self.MUSIC]
        self.background_playlist = PlayList(pieces, random=True, repeat=True)
        change_playlist(self.background_playlist)

    def leave(self):
        change_playlist(None)


class ToMap(Door):

    SCENE = "bridge"

    INTERACTS = {
        "door": InteractNoImage(707, 344, 84, 245),
        }

    INITIAL = "door"


class BridgeComputer(Thing):
    """The bridge computer. Gives status updates"""

    NAME = "bridge.comp"

    INTERACTS = {
        'screen' : InteractNoImage(338, 296, 123, 74),
    }

    INITIAL = 'screen'

    def interact_without(self):
        return Result(detail_view='bridge_comp_detail')

    def interact_with_titanium_leg(self):
        return Result("You can't break the duraplastic screen.")

    def interact_with_machete(self):
        return Result("Scratching the screen won't help you.")

    def get_description(self):
        return "The main bridge computer screen."


class MassageChairBase(Thing):
    "The captain's massage chair, contains superconductor"

    NAME = 'bridge.massagechair_base'

    INTERACTS = {
        'chair': InteractNoImage(127, 518, 69, 64),
    }

    INITIAL = 'chair'

    INITIAL_DATA = {
            'contains_superconductor': True,
    }

    def interact_without(self):
        return Result(detail_view='chair_detail')

    def get_description(self):
        if self.get_data('contains_superconductor'):
            return "A top of the line Massage-o-Matic Captain's Executive Command Chair. " \
                   "It's massaging a skeleton."
        return "The chair won't work any more, it has no power."


class MassageChair(Thing):
    "The captain's massage chair, contains superconductor"

    NAME = 'bridge.massagechair'

    INTERACTS = {
        'chair': InteractRectUnion((
            (148, 418, 77, 68),
            (69, 363, 80, 156),
            (104, 514, 18, 40),
            (147, 487, 106, 32),
            (220, 518, 83, 49),
            (196, 502, 75, 34),
            (207, 534, 69, 51),
        )),
    }

    INITIAL = 'chair'

    def get_description(self):
        return self.state.current_scene.things['bridge.massagechair_base'] \
                   .get_description()

    def is_interactive(self):
        return False


class Stethoscope(Item):
    "Used for cracking safes. Found on the doctor on the chair"

    INVENTORY_IMAGE = 'stethoscope.png'
    CURSOR = CursorSprite('stethoscope.png')


class StethoscopeThing(Thing):
    "Stehoscope on the doctor"

    NAME ='bridge.stethoscope'

    INTERACTS = {
        'stethoscope': InteractImage(650, 178, 'hanging_stethoscope.png'),
    }

    INITIAL = 'stethoscope'

    def get_description(self):
        return "A stethoscope hangs from the neck of the skeleton."

    def interact_without(self):
        self.state.add_inventory_item('stethoscope')
        self.scene.remove_thing(self)
        # Fill in the doctor's rect
        self.scene.doctor.rect.append(self.rect)
        return Result("You pick up the stethoscope and verify that the doctor's "
                      "heart has stopped. Probably a while ago.")


class TapedSuperconductor(Item):
    "Used for connecting high-powered parts of the ship up"

    INVENTORY_IMAGE = 'superconductor_taped.png'
    CURSOR = CursorSprite('superconductor_taped_cursor.png')


class Superconductor(Item):
    "Used for connecting high-powered parts of the ship up"

    INVENTORY_IMAGE = 'superconductor_fixed.png'
    CURSOR = CursorSprite('superconductor_fixed.png')

    def interact_with_duct_tape(self, item, state):
        taped_superconductor = TapedSuperconductor('taped_superconductor')
        state.add_item(taped_superconductor)
        state.replace_inventory_item(self.name, taped_superconductor.name)
        return Result("You rip off a piece of duct tape and stick it on the superconductor. "
                      "It almost sticks to itself, but you successfully avoid disaster.")


class SuperconductorThing(Thing):
    "Superconductor from the massage chair."

    NAME ='bridge.superconductor'

    INTERACTS = {
        'superconductor': InteractImage(158, 138, 'superconductor.png'),
    }

    INITIAL = 'superconductor'

    def interact_without(self):
        self.state.add_inventory_item('superconductor')
        self.state.current_scene.things['bridge.massagechair_base'] \
                          .set_data('contains_superconductor', False)
        self.scene.remove_thing(self)
        return (Result("The superconductor module unclips easily."),
                make_jim_dialog(("Prisoner %s. That chair you've destroyed was "
                                 "property of the ship's captain. "
                                 "You will surely be punished."
                                ) % PLAYER_ID, self.state))

class StarField(Thing):

    NAME = 'bridge.stars'

    INTERACTS = {
            #'stars' : InteractImage(190, 145, 'stars_3.png'),
            'stars' : InteractAnimated(190, 145,
                ['stars_%d.png' % (i+1) for i in range(3) + range(1,0,-1)], 30)
            }

    INITIAL = 'stars'

    def is_interactive(self):
        return False


class BlinkingLights(Thing):

    def __init__(self):
        super(BlinkingLights, self).__init__()
        self.description = None

    def is_interactive(self):
        return False

    def leave(self):
        self.description = random.choice([
            "The lights flash in interesting patterns.",
            "The flashing lights don't mean anything to you.",
            "The console lights flash and flicker.",
            ])

    def get_description(self):
        if not self.description:
            self.leave()
        return self.description

class LeftLights(BlinkingLights):

    NAME ='bridge.lights.1'

    INTERACTS = {
        "lights": InteractAnimated(176, 337, ["bridge_lights_1_1.png", "bridge_lights_1_2.png", "bridge_lights_1_3.png", "bridge_lights_1_2.png"], 5)
    }

    INITIAL = 'lights'

class RightLights(BlinkingLights):

    NAME ='bridge.lights.2'

    INTERACTS = {
        "lights": InteractAnimated(559, 332, ["bridge_lights_2_1.png", "bridge_lights_2_2.png", "bridge_lights_2_3.png", "bridge_lights_2_2.png"], 5)
    }

    INITIAL = 'lights'

class JimPanel(Thing):
    "The panel to JIM's internals'"

    NAME = "jim_panel"

    INTERACTS = {
            'closed' : InteractNoImage(506, 430, 137, 47),
            'open'   : InteractImage(500, 427, 'jim_panel_open.png'),
            'broken' : InteractImage(488, 412, 'jim_panel_destroyed.png'),
            }

    INITIAL = 'closed'

    def get_description(self):
        if self.scene.get_data('ai panel') == 'closed':
            return "The sign reads 'Warning: Authorized Techinicians Only'."

    def interact_without(self):
        if self.scene.get_data('ai status') == 'online':
            return self.interact_default()
        elif self.scene.get_data('ai panel') == 'closed':
            return Result("You are unable to open the panel with your bare hands.")
        elif self.scene.get_data('ai panel') == 'open':
            self.scene.set_data('ai panel', 'broken')
            self.scene.set_data('ai status', 'dead')
            self.set_interact('broken')
            return Result("You unplug various important-looking wires.")


    def interact_with_machete(self, item):
        if self.scene.get_data('ai status') == 'online':
            return self.interact_default()
        elif self.scene.get_data('ai panel') == 'closed':
            self.scene.set_data('ai panel', 'open')
            self.set_interact('open')
            return Result("Using the machete, you lever the panel off.")
        elif self.scene.get_data('ai panel') == 'open':
            self.scene.set_data('ai panel', 'broken')
            self.scene.set_data('ai status', 'dead')
            self.set_interact('broken')
            return Result("You smash various delicate components with the machete.")

    def interact_default(self):
        if self.scene.get_data('ai status') == 'online':
            return (Result('You feel a shock from the panel.'),
                    make_jim_dialog("Prisoner %s. Please step away from the panel. "
                        "You are not an authorized technician." % PLAYER_ID, self.state))

class ChairDetail(Scene):

    FOLDER = 'bridge'
    BACKGROUND = 'chair_detail.png'
    NAME = 'chair_detail'

    SIZE = (300, 300)

    def __init__(self, state):
        super(ChairDetail, self).__init__(state)
        self.add_thing(SuperconductorThing())


# classes related the computer detail


class LogTab(Thing):
    """Tab for log screen"""

    NAME = 'bridge_comp.screen'

    INTERACTS = {
            'log tab' : InteractNoImage(100, 53, 94, 37)
            }
    INITIAL = 'log tab'
    COMPUTER = 'bridge_comp_detail'

    def is_interactive(self):
        return self.state.detail_views[self.COMPUTER].get_data('tab') != 'log'

    def interact_without(self):
        self.state.detail_views[self.COMPUTER].set_data('tab', 'log')
        self.state.detail_views[self.COMPUTER].set_background()
        return Result(soundfile='beep550.ogg')


class AlertTab(Thing):
    """Tab for alert screen"""

    NAME = 'bridge_comp.alert_tab'

    INTERACTS = {
            'alert tab' : InteractNoImage(12, 53, 88, 37)
            }
    INITIAL = 'alert tab'
    COMPUTER = 'bridge_comp_detail'

    def is_interactive(self):
        return self.state.detail_views[self.COMPUTER].get_data('tab') != 'alert'

    def interact_without(self):
        self.state.detail_views[self.COMPUTER].set_data('tab', 'alert')
        self.state.detail_views[self.COMPUTER].set_background()
        return Result(soundfile='beep550.ogg')

class NavTab(Thing):
    """Tab for the Navigation screen"""

    NAME = 'bridge_comp.nav_tab'

    INTERACTS = {
            'nav tab' : InteractNoImage(197, 53, 126, 37),
            }
    INITIAL = 'nav tab'
    COMPUTER = 'bridge_comp_detail'

    def is_interactive(self):
        return self.state.detail_views[self.COMPUTER].get_data('tab') != 'nav'

    def interact_without(self):
        self.state.detail_views[self.COMPUTER].set_data('tab', 'nav')
        self.state.detail_views[self.COMPUTER].set_background()
        return Result(soundfile='beep550.ogg')

class DestNavPageLine(Thing):
    """The destination navigation lines."""

    INITIAL = 'line'
    COMPUTER = 'bridge_comp_detail'

    def __init__(self, number, rect, ai_blocked):
        super(DestNavPageLine, self).__init__()
        self.name = 'bridge_comp.nav_line%s' % number
        if DEBUG:
            self._interact_hilight_color = Color(THECOLORS.keys()[number])
        r = Rect(rect)
        self.interacts = {}
        self.interacts['line'] = InteractNoImage(r.x, r.y, r.w, r.h)
        # Whether JIM blocks this
        self.ai_blocked = ai_blocked
        self.set_interact('line')

    def is_interactive(self):
        return self.state.detail_views[self.COMPUTER].get_data('tab') == 'nav'

    def interact_without(self):
        if self.state.scenes['bridge'].get_data('ai status') == 'online':
            return make_jim_dialog("You are not authorized to change the destination.", self.state)
        if not self.ai_blocked:
            return Result("There's no good reason to choose to go to the penal colony.")
        if self.state.scenes['bridge'].get_data('ai status') == 'looping':
            return Result("You could change the destination, but when JIM recovers, it'll just get reset.")
        if self.state.scenes['bridge'].get_data('ai status') == 'dead':
            return Result("You change the destination.", soundfile="beep550.ogg", end_game=True)

class CompUpButton(Thing):
    """Up button on log screen"""

    NAME = 'bridge_comp.up_button'

    INTERACTS = {
            'up' : InteractNoImage(594, 82, 30, 58)
            }
    INITIAL = 'up'
    COMPUTER = 'bridge_comp_detail'

    def is_interactive(self):
        tab = self.state.detail_views[self.COMPUTER].get_data('tab')
        page = self.state.detail_views[self.COMPUTER].get_data('log page')
        return tab == 'log' and page > 0

    def interact_without(self):
        page = self.state.detail_views[self.COMPUTER].get_data('log page')
        self.state.detail_views[self.COMPUTER].set_data('log page', page-1)
        self.state.detail_views[self.COMPUTER].set_background()
        return Result(soundfile='beep550.ogg')


class CompDownButton(Thing):
    """Down button on log screen"""

    NAME = 'bridge_comp.down_button'

    INTERACTS = {
            'down' : InteractNoImage(594, 293, 30, 58)
            }
    INITIAL = 'down'
    COMPUTER = 'bridge_comp_detail'

    def is_interactive(self):
        tab = self.state.detail_views[self.COMPUTER].get_data('tab')
        page = self.state.detail_views[self.COMPUTER].get_data('log page')
        max_page = self.state.detail_views[self.COMPUTER].get_data('max page')
        return tab == 'log' and (page + 1) < max_page

    def interact_without(self):
        page = self.state.detail_views[self.COMPUTER].get_data('log page')
        self.state.detail_views[self.COMPUTER].set_data('log page', page+1)
        self.state.detail_views[self.COMPUTER].set_background()
        return Result(soundfile='beep550.ogg')


class MonitorCamera(BaseCamera):
    "A camera on the bridge"

    NAME = "bridge.camera"

    INTERACTS = {
        'online': InteractImage(33, 192, 'camera_small.png'),
        'dead': InteractImage(33, 192, 'camera_small_gray.png'),
        'looping': InteractAnimated(33, 192, ('camera_small.png',
                                              'camera_small_gray.png'),
                                    15),
    }


class BridgeCompDetail(Scene):

    FOLDER = 'bridge'
    NAME = 'bridge_comp_detail'

    SIZE = (640, 400)

    ALERT_BASE = 'comp_alert_base.png'
    ALERTS = {
            'ai looping' : 'comp_alert_ai_looping.png',
            'ai offline' : 'comp_alert_ai_offline.png',
            'engine offline' : 'comp_alert_engine_offline.png',
            'life support' : 'comp_alert_life_support.png',
            'life support partial' : 'comp_alert_life_support_partial.png',
            }

    # Point to start drawing changeable alerts
    ALERT_OFFSET = (16, 140)
    ALERT_SPACING = 4

    LOGS = ['comp_log_start.png', 'comp_log_1.png',
            'comp_log_end.png']

    NAVIGATION = {
            'engine offline' : 'bridge_nav_engine.png',
            'life support' : 'bridge_nav_life_support.png',
            'final' : 'bridge_nav_dest.png'
            }

    BACKGROUND = ALERT_BASE


    INITIAL_DATA = {
            'tab' : 'alert',
            'log page' : 0,
            'max page' : len(LOGS),
    }

    def __init__(self, state):
        super(BridgeCompDetail, self).__init__(state)

        self.add_thing(LogTab())
        self.add_thing(AlertTab())
        self.add_thing(NavTab())
        #self.add_thing(CompUpButton())
        #self.add_thing(CompDownButton())
        self._scene_playlist = None
        self._alert = get_image(self.FOLDER, self.ALERT_BASE)
        self._alert_messages = {}
        self._nav_messages = {}
        for key, name in self.ALERTS.iteritems():
            self._alert_messages[key] = get_image(self.FOLDER, name)
        for key, name in self.NAVIGATION.iteritems():
            self._nav_messages[key] = get_image(self.FOLDER, name)
        self._nav_lines = []
        self._nav_lines.append(DestNavPageLine(1, (14, 99, 595, 30), False))
        self._nav_lines.append(DestNavPageLine(2, (14, 135, 595, 30), True))
        self._nav_lines.append(DestNavPageLine(3, (14, 167, 595, 30), True))
        self._nav_lines.append(DestNavPageLine(4, (14, 203, 595, 30), True))
        self._nav_lines.append(DestNavPageLine(5, (14, 239, 595, 30), True))
        self._logs = [get_image(self.FOLDER, x) for x in self.LOGS]

    def enter(self):
        self._scene_playlist = get_current_playlist()
        change_playlist(None)
        self.set_background()

    def leave(self):
        change_playlist(self._scene_playlist)

    def set_background(self):
        if self.get_data('tab') == 'alert':
            self._clear_navigation()
            self._background = self._alert
        elif self.get_data('tab') == 'log':
            self._clear_navigation()
            self._background = self._logs[self.get_data('log page')]
        elif self.get_data('tab') == 'nav':
            self._background = self._get_nav_page()

    def _clear_navigation(self):
        "Remove navigation things if necessary"
        for thing in self._nav_lines:
            if thing.name in self.things.keys():
                # Much fiddling to do the right thing when we reinsert it
                del self.things[thing.name]
                thing.scene = None

    def _get_nav_page(self):
        if not self.state.scenes['engine'].get_data('engine online'):
            return self._nav_messages['engine offline']
        elif not self.state.scenes['mess'].get_data('life support status') == 'fixed':
            return self._nav_messages['life support']
        else:
            for thing in self._nav_lines:
                if thing.name not in self.things:
                    self.add_thing(thing)
            return self._nav_messages['final']

    def _draw_alerts(self, surface):
        xpos, ypos = self.ALERT_OFFSET
        if self.state.scenes['bridge'].get_data('ai status') == 'looping':
            surface.blit(self._alert_messages['ai looping'], (xpos, ypos))
            ypos += self._alert_messages['ai looping'].get_size()[1] + self.ALERT_SPACING
        if self.state.scenes['bridge'].get_data('ai status') == 'dead':
            surface.blit(self._alert_messages['ai offline'], (xpos, ypos))
            ypos += self._alert_messages['ai offline'].get_size()[1] + self.ALERT_SPACING
        if not self.state.scenes['engine'].get_data('engine online'):
            surface.blit(self._alert_messages['engine offline'], (xpos, ypos))
            ypos += self._alert_messages['engine offline'].get_size()[1] + self.ALERT_SPACING
        if self.state.scenes['mess'].get_data('life support status') == 'broken':
            surface.blit(self._alert_messages['life support'], (xpos, ypos))
            ypos += self._alert_messages['life support'].get_size()[1] + self.ALERT_SPACING
        if self.state.scenes['mess'].get_data('life support status') == 'replaced':
            surface.blit(self._alert_messages['life support partial'], (xpos, ypos))
            ypos += self._alert_messages['life support partial'].get_size()[1] + self.ALERT_SPACING

    def draw_things(self, surface):
        if self.get_data('tab') == 'alert':
            self._draw_alerts(surface)
        super(BridgeCompDetail, self).draw_things(surface)


SCENES = [Bridge]
DETAIL_VIEWS = [ChairDetail, BridgeCompDetail]