2021-08-31 10:07:20 +00:00
|
|
|
# Copyright 2020 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 subiquity.common.errorreport import ErrorReportKind
|
2021-09-23 08:52:31 +00:00
|
|
|
from subiquity.common.types import ApplicationState
|
|
|
|
from subiquity.server.controller import SubiquityController
|
|
|
|
from subiquitycore.context import with_context
|
|
|
|
from subiquitycore.utils import run_command
|
2021-10-04 10:55:25 +00:00
|
|
|
from system_setup.common.wsl_conf import wsl_config_update
|
2021-09-30 16:19:05 +00:00
|
|
|
from system_setup.common.wsl_utils import get_userandgroups
|
2021-08-31 10:07:20 +00:00
|
|
|
|
2021-09-27 12:36:25 +00:00
|
|
|
log = logging.getLogger("system_setup.server.controllers.configure")
|
2021-08-31 10:07:20 +00:00
|
|
|
|
|
|
|
|
|
|
|
class ConfigureController(SubiquityController):
|
|
|
|
|
|
|
|
def __init__(self, app):
|
|
|
|
super().__init__(app)
|
2021-09-20 06:37:55 +00:00
|
|
|
self.app = app
|
2021-08-31 10:07:20 +00:00
|
|
|
self.model = app.base_model
|
|
|
|
|
|
|
|
def start(self):
|
|
|
|
self.install_task = self.app.aio_loop.create_task(self.configure())
|
|
|
|
|
|
|
|
@with_context(
|
|
|
|
description="final system configuration", level="INFO",
|
|
|
|
childlevel="DEBUG")
|
|
|
|
async def configure(self, *, context):
|
2021-09-01 08:48:44 +00:00
|
|
|
context.set('is-install-context', True)
|
2021-08-31 10:07:20 +00:00
|
|
|
try:
|
|
|
|
|
|
|
|
self.app.update_state(ApplicationState.WAITING)
|
|
|
|
|
2021-09-01 08:48:44 +00:00
|
|
|
await self.model.wait_install()
|
|
|
|
|
2021-08-31 10:07:20 +00:00
|
|
|
self.app.update_state(ApplicationState.NEEDS_CONFIRMATION)
|
|
|
|
|
|
|
|
self.app.update_state(ApplicationState.RUNNING)
|
|
|
|
|
2021-09-01 08:48:44 +00:00
|
|
|
await self.model.wait_postinstall()
|
|
|
|
|
2021-08-31 10:07:20 +00:00
|
|
|
self.app.update_state(ApplicationState.POST_WAIT)
|
|
|
|
|
|
|
|
# TODO WSL:
|
|
|
|
# 1. Use self.model to get all data to commit
|
2021-08-31 11:26:49 +00:00
|
|
|
# 2. Write directly (without wsl utilities) to wsl.conf and other
|
|
|
|
# fstab files
|
2021-08-31 13:56:08 +00:00
|
|
|
# 3. If not in reconfigure mode: create User, otherwise just write
|
|
|
|
# wsl.conf files.
|
2021-08-31 10:07:20 +00:00
|
|
|
# This must not use subprocesses.
|
|
|
|
# If dry-run: write in .subiquity
|
|
|
|
|
|
|
|
self.app.update_state(ApplicationState.POST_RUNNING)
|
|
|
|
|
2021-09-23 08:52:31 +00:00
|
|
|
dryrun = self.app.opts.dry_run
|
2021-09-14 08:16:06 +00:00
|
|
|
variant = self.app.variant
|
2021-10-06 14:10:56 +00:00
|
|
|
root_dir = self.model.root
|
2021-09-28 12:48:30 +00:00
|
|
|
if variant == "wsl_setup":
|
2021-10-04 07:23:02 +00:00
|
|
|
wsl_id = self.model.identity.user
|
2021-09-23 08:52:31 +00:00
|
|
|
if dryrun:
|
|
|
|
log.debug("mimicking creating user %s",
|
2021-10-04 07:23:02 +00:00
|
|
|
wsl_id.username)
|
2021-09-23 08:52:31 +00:00
|
|
|
else:
|
2021-10-04 07:23:02 +00:00
|
|
|
create_user_act = \
|
|
|
|
run_command(["/usr/sbin/useradd", "-m", "-s",
|
|
|
|
"/bin/bash", "-p",
|
|
|
|
wsl_id.password,
|
|
|
|
wsl_id.username])
|
|
|
|
if create_user_act.returncode != 0:
|
2021-10-05 01:38:10 +00:00
|
|
|
raise Exception("Failed to create user %s: %s"
|
|
|
|
% (wsl_id.username,
|
|
|
|
create_user_act.stderr))
|
2021-10-04 07:23:02 +00:00
|
|
|
log.debug("created user %s", wsl_id.username)
|
|
|
|
assign_grp_act = \
|
|
|
|
run_command(["/usr/sbin/usermod", "-a",
|
|
|
|
"-c", wsl_id.realname,
|
|
|
|
"-G", get_userandgroups(),
|
|
|
|
wsl_id.username])
|
|
|
|
if assign_grp_act.returncode != 0:
|
2021-10-05 01:38:10 +00:00
|
|
|
raise Exception(("Failed to assign group"
|
|
|
|
" to user %s: %s")
|
|
|
|
% (wsl_id.username,
|
|
|
|
assign_grp_act.stderr))
|
2021-09-28 12:48:30 +00:00
|
|
|
else:
|
2021-10-04 10:55:25 +00:00
|
|
|
wsl_config_update(self.model.wslconfadvanced.wslconfadvanced,
|
2021-10-06 14:10:56 +00:00
|
|
|
root_dir)
|
2021-09-30 16:00:32 +00:00
|
|
|
|
2021-10-06 14:10:56 +00:00
|
|
|
wsl_config_update(self.model.wslconfbase.wslconfbase, root_dir)
|
2021-09-28 12:48:30 +00:00
|
|
|
|
2021-08-31 10:07:20 +00:00
|
|
|
self.app.update_state(ApplicationState.DONE)
|
|
|
|
except Exception:
|
|
|
|
kw = {}
|
|
|
|
self.app.make_apport_report(
|
|
|
|
ErrorReportKind.INSTALL_FAIL, "configuration failed", **kw)
|
|
|
|
raise
|
|
|
|
|
|
|
|
def stop_uu(self):
|
|
|
|
# This is a no-op to allow Shutdown controller to depend on this one
|
2021-08-31 11:26:49 +00:00
|
|
|
pass
|