subiquity/console_conf/ui/views/login.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/>.

""" Login

Login provides user with language selection

"""
import logging

from urwid import Text

from subiquitycore.ui.buttons import done_btn
from subiquitycore.ui.container import ListBox, Pile
from subiquitycore.ui.utils import Padding, button_pile
from subiquitycore.view import BaseView

log = logging.getLogger("console_conf.ui.views.login")


class LoginView(BaseView):
    title = "Ubuntu Core"

    def __init__(self, opts, model, controller, netdevs):
        self.opts = opts
        self.model = model
        self.controller = controller
        self.netdevs = netdevs
        self.items = []

        super().__init__(
            Pile(
                [
                    ("pack", Text("")),
                    Padding.center_79(ListBox(self._build_model_inputs())),
                    (
                        "pack",
                        Pile(
                            [
                                ("pack", Text("")),
                                button_pile(self._build_buttons()),
                                ("pack", Text("")),
                            ]
                        ),
                    ),
                ]
            )
        )

    def _build_buttons(self):
        return [
            done_btn("Done", on_press=self.done),
        ]

    def _build_model_inputs(self):
        user = self.model.user
        ips = []
        for dev in self.netdevs:
            for addr in dev.actual_global_ip_addresses:
                ips.append(addr)

        if not user:
            sl = []
            sl.append(Text("no owner"))
            return sl

        local_tpl = "This device is registered to {realname}."

        remote_tpl = (
            "\n\nRemote access was enabled via authentication with SSO user"
            " <{username}>.\nPublic SSH keys were added to the device "
            "for remote access.\n\n{realname} can connect remotely to this "
            "device via SSH:"
        )

        sl = []
        login_info = {
            "realname": user.realname,
            "username": user.username,
        }
        login_text = local_tpl.format(**login_info)
        login_text += remote_tpl.format(**login_info)

        sl += [Text(login_text), Padding.line_break("")]
        for ip in ips:
            ssh_iface = "    ssh %s@%s" % (user.username, ip)
            sl.append(Text(ssh_iface))

        return sl

    def confirm(self, result):
        self.done()

    def done(self, button):
        self.controller.login_done()