# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import gtk
import os
from autotest_lib.client.cros import factory
from autotest_lib.client.cros.factory import gooftools
from autotest_lib.client.cros.factory import task
from autotest_lib.client.cros.factory import ui


class SelectHwidTask(task.FactoryTask):

    SELECTION_PER_PAGE = 10
    HWID_AUTODETECT = None

    def __init__(self, data):
        self.data = data

    def build_hwid_list(self):
        with os.popen("crossystem hwid 2>/dev/null", "r") as p:
            current_hwid = p.read().strip()

        (stdout, _, result) = gooftools.run("hwid_tool list_hwids",
                                            ignore_status=True)
        known_list = stdout.splitlines()
        if (not known_list) or (result != 0):
            factory.log('Warning: No valid HWID database in system.')
            known_list = []

        # Build a list with elements in (display_text, hwid_value).
        # The first element is "current value".
        hwids = [('<Current Value> %s' % current_hwid, current_hwid)]
        hwids += [(hwid, hwid) for hwid in known_list]
        return hwids

    def window_key_press(self, widget, event):
        # Process page navigation
        KEY_PREV = [65361, 65362, ord('h'), ord('k')]  # Left, Up
        KEY_NEXT = [65363, 65364, ord('l'), ord('j')]  # Right, Down
        if event.keyval in KEY_PREV:
            if self.page_index > 0:
                self.page_index -= 1
            self.render_page()
            return True
        if event.keyval in KEY_NEXT:
            if self.page_index < self.pages - 1:
                self.page_index += 1
            self.render_page()
            return True

        char = chr(event.keyval) if event.keyval in range(32, 127) else None
        factory.log('key_press %s(%s)' % (event.keyval, char))
        try:
            select = int(char)
        except ValueError:
            factory.log('Need a number.')
            return True

        select = select + self.page_index * self.SELECTION_PER_PAGE
        if select < 0 or select >= len(self.hwid_list):
            factory.log('Invalid selection: %d' % select)
            return True

        data = self.hwid_list[select]
        hwid = data[1]
        factory.log('Selected: %s' % ', '.join(data).replace('\n', ' '))

        # TODO(tammo) Use hwid_tool or probe to quick probe if selected HWID
        # matches current system, by checking non-firmware components.

        self.data['hwid'] = hwid
        self.stop()
        return True

    def render_page(self):
        msg = 'Choose a HWID:\n\n'
        start = self.page_index * self.SELECTION_PER_PAGE
        end = start + self.SELECTION_PER_PAGE
        for index, data in enumerate(self.hwid_list[start:end]):
            msg += '%s) %s\n\n' % (index, data[0])
        if self.pages > 1:
            msg += '[Page %d / %d, navigate with arrow keys]' % (
                    self.page_index + 1, self.pages)
        self.label.set_text(msg)

    def start(self):
        self.page_index = 0
        self.pages = 0

        self.hwid_list = self.build_hwid_list()
        self.pages = len(self.hwid_list) / self.SELECTION_PER_PAGE
        if len(self.hwid_list) % self.SELECTION_PER_PAGE:
            self.pages += 1

        self.label = ui.make_label('')
        self.add_widget(self.label)
        self.render_page()
        self.connect_window('key-press-event', self.window_key_press)