subiquity/subiquitycore/ui/views/iscsi.py

# 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/>.

from urwid import Text, Columns, Pile, ListBox
from subiquitycore.view import ViewPolicy
from subiquitycore.ui.buttons import menu_btn
from subiquitycore.ui.interactive import (StringEditor, YesNo,
                                          PasswordEditor)
from subiquitycore.ui.utils import Color, Padding
import logging

log = logging.getLogger('subiquitycore.iscsi')


class IscsiDiskView(ViewPolicy):
    def __init__(self, model, signal):
        self.model = model
        self.signal = signal
        self.iscsi_host = StringEditor(caption="")
        self.connect_anon = YesNo()
        self.connect_username = StringEditor(caption="")
        self.connect_password = PasswordEditor(caption="")
        self.server_auth = YesNo()
        self.server_username = StringEditor(caption="")
        self.server_password = PasswordEditor(caption="")
        body = [
            Padding.center_50(self._build_model_inputs()),
            Padding.line_break(""),
            Padding.center_50(self._build_menu()),
            Padding.line_break(""),
            Padding.center_75(self._build_volume_mount_selector())
        ]
        super().__init__(ListBox(body))

    def _build_model_inputs(self):
        items = [
            Columns(
                [
                    ("weight", 0.2, Text("iSCSI Server Host", align="right")),
                    ("weight", 0.3,
                     Color.string_input(self.iscsi_host,
                                        focus_map="string_input focus"))
                ],
                dividechars=4
            ),
            Columns(
                [
                    ("weight", 0.2, Text("Connect anonymously",
                                         align="right")),
                    ("weight", 0.3,
                     Color.string_input(Pile(self.connect_anon.group),
                                        focus_map="string_input focus"))
                ],
                dividechars=4
            ),
            Columns(
                [
                    ("weight", 0.2, Text("Connect as user", align="right")),
                    ("weight", 0.3,
                     Color.string_input(self.connect_username,
                                        focus_map="string_input focus"))
                ],
                dividechars=4
            ),
            Columns(
                [
                    ("weight", 0.2, Text("Password", align="right")),
                    ("weight", 0.3,
                     Color.string_input(self.connect_password,
                                        focus_map="string_input focus"))
                ],
                dividechars=4
            ),
            Columns(
                [
                    ("weight", 0.2, Text("Require server auth",
                                         align="right")),
                    ("weight", 0.3,
                     Color.string_input(Pile(self.server_auth.group),
                                        focus_map="string_input focus"))
                ],
                dividechars=4
            ),
            Columns(
                [
                    ("weight", 0.2, Text("Server identity", align="right")),
                    ("weight", 0.3,
                     Color.string_input(self.server_username,
                                        focus_map="string_input focus"))
                ],
                dividechars=4
            ),
            Columns(
                [
                    ("weight", 0.2, Text("Server password", align="right")),
                    ("weight", 0.3,
                     Color.string_input(self.server_password,
                                        focus_map="string_input focus"))
                ],
                dividechars=4
            )
        ]
        return Pile(items)

    def _build_menu(self):
        items = []
        for label, _, _ in self.model.get_menu():
            items.append(
                Columns(
                    [
                        ("weight", 0.2, Text("")),
                        ("weight", 0.3,
                         Color.menu_button(
                             menu_btn(label=label,
                                      on_press=self.confirm),
                             focus_map="menu_button focus"))
                    ]))
        return Pile(items)

    def _build_volume_mount_selector(self):
        items = [Text("SELECT VOLUME TO MOUNT")]
        # TODO: List found volumes
        return Pile(items)

    def confirm(self, result):
        self.signal.emit_signal(
            self.model.get_signal_by_name(result.label))