# Copyright (C) 2024 AOSC Developers # # Author: Yuanhang Sun <leavelet@aosc.io> # # This file is part of cloud-init. See LICENSE file for license information. import logging from cloudinit import distros, helpers, subp, util from cloudinit.distros import PackageList from cloudinit.distros.parsers.hostname import HostnameConf from cloudinit.distros.parsers.sys_conf import SysConf from cloudinit.settings import PER_INSTANCE LOG = logging.getLogger(__name__) class Distro(distros.Distro): systemd_locale_conf_fn = "/etc/locale.conf" init_cmd = ["systemctl"] network_conf_dir = "/etc/sysconfig/network" resolve_conf_fn = "/etc/systemd/resolved.conf" tz_local_fn = "/etc/localtime" dhclient_lease_directory = "/var/lib/NetworkManager" dhclient_lease_file_regex = r"dhclient-[\w-]+\.lease" renderer_configs = { "sysconfig": { "control": "etc/sysconfig/network", "iface_templates": "%(base)s/network-scripts/ifcfg-%(name)s", "route_templates": { "ipv4": "%(base)s/network-scripts/route-%(name)s", "ipv6": "%(base)s/network-scripts/route6-%(name)s", }, } } prefer_fqdn = False def __init__(self, name, cfg, paths): distros.Distro.__init__(self, name, cfg, paths) self._runner = helpers.Runners(paths) self.osfamily = "aosc" self.default_locale = "en_US.UTF-8" cfg["ssh_svcname"] = "sshd" def apply_locale(self, locale, out_fn=None): if not out_fn: out_fn = self.systemd_locale_conf_fn locale_cfg = { "LANG": locale, } update_locale_conf(out_fn, locale_cfg) def _write_hostname(self, hostname, filename): if filename.endswith("/previous-hostname"): conf = HostnameConf("") conf.set_hostname(hostname) util.write_file(filename, str(conf), 0o644) create_hostname_file = util.get_cfg_option_bool( self._cfg, "create_hostname_file", True ) if create_hostname_file: subp.subp(["hostnamectl", "set-hostname", str(hostname)]) else: subp.subp( [ "hostnamectl", "set-hostname", "--transient", str(hostname), ] ) LOG.info("create_hostname_file is False; hostname set transiently") def _read_hostname(self, filename, default=None): if filename.endswith("/previous-hostname"): return util.load_text_file(filename).strip() (out, _err) = subp.subp(["hostname"]) out = out.strip() if len(out): return out else: return default def _read_system_hostname(self): sys_hostname = self._read_hostname(self.hostname_conf_fn) return (self.hostname_conf_fn, sys_hostname) def set_timezone(self, tz): tz_file = self._find_tz_file(tz) util.del_file(self.tz_local_fn) util.sym_link(tz_file, self.tz_local_fn) def package_command(self, command, args=None, pkgs=None): if pkgs is None: pkgs = [] cmd = ["oma"] if command: cmd.append(command) cmd.append("-y") cmd.extend(pkgs) subp.subp(cmd, capture=False) def install_packages(self, pkglist: PackageList): self.package_command("install", pkgs=pkglist) def update_package_sources(self, *, force=False): self._runner.run( "update-sources", self.package_command, "refresh", freq=PER_INSTANCE, ) def read_locale_conf(sys_path): exists = False try: contents = util.load_text_file(sys_path).splitlines() exists = True except IOError: contents = [] return (exists, SysConf(contents)) def update_locale_conf(sys_path, locale_cfg): if not locale_cfg: return (exists, contents) = read_locale_conf(sys_path) updated_am = 0 for k, v in locale_cfg.items(): if v is None: continue v = str(v) if not v: continue contents[k] = v updated_am += 1 if updated_am: lines = [ str(contents), ] if not exists: lines.insert(0, util.make_header()) util.write_file(sys_path, "\n".join(lines) + "\n", 0o644)