# Copyright 2015 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging
from subiquitycore.model import ModelPolicy


log = logging.getLogger('subiquitycore.models.installprogress')


class InstallProgressModel(ModelPolicy):
    """ Model representing install progress
    """
    # FIXME: Decide what to do here if ESC is pressed, it's probably in
    # a state of no return so may be better to just exit with error.

    signals = [
        ("Run curtin",
         'installprogress:curtin-install',
         'curtin_install'),
        ("Run curtin postinstall",
         'installprogress:curtin-postinstall',
         'curtin_postinstall'),
        ("Indicate curtin config written",
         'installprogress:wrote-install',
         'curtin_wrote_install'),
        ("Indicate curtin postconfig written",
         'installprogress:wrote-postinstall',
         'curtin_wrote_postinstall'),
        ("Install progress final view",
         'menu:installprogress:main',
         'show_progress'),
        ("Reboot curtin",
         "installprogress:curtin-reboot",
         "reboot")
    ]

    installprogress_menu = []

    def get_signals(self):
        return self.signals

    def get_menu(self):
        return self.installprogress_menu

    def get_signal_by_name(self, selection):
        for x, y, z in self.get_menu():
            if x == selection:
                return y