locales and console settings are done

aif/system/console.py

@@ -0,0 +1,97 @@
+import configparser
+import io
+import os
+import pathlib
+import re
+
+
+_font_re = re.compile(r'(\.(psfu?|fnt))?(\.gz)?$', re.IGNORECASE)
+_kbd_re = re.compile(r'(\.map)?(\.gz)?$')
+
+
+class Font(object):
+    def __init__(self, font_xml):
+        self.xml = font_xml
+        self.settings = {}
+        if self.xml:
+            chk = {'FONT': self.xml.find('font'),
+                   'FONT_MAP': self.xml.find('map'),
+                   'FONT_UNIMAP': self.xml.find('unicodeMap')}
+            for setting, xml in chk.items():
+                if xml:
+                    self.settings[setting] = xml.text.strip()
+
+    def verify(self, chroot_base = '/'):
+        if 'FONT' not in self.settings.keys():
+            return(None)
+        fontdir = pathlib.Path(chroot_base).joinpath('usr', 'share', 'kbd', 'consolefonts')
+        fontnames = [_font_re.sub('', p.stem) for p in fontdir.iterdir() if not p.stem.startswith(('README.',
+                                                                                                   'partialfonts',
+                                                                                                   'ERRORS'))]
+        if self.settings['FONT'] not in fontnames:
+            raise ValueError('console font is not installed on target system')
+        return(True)
+
+
+class Keyboard(object):
+    def __init__(self, keyboard_xml):
+        self.xml = keyboard_xml
+        self.settings = {}
+        if self.xml:
+            chk = {'KEYMAP': self.xml.find('map'),
+                   'KEYMAP_TOGGLE': self.xml.find('toggle')}
+            for setting, xml in chk.items():
+                if xml:
+                    self.settings[setting] = xml.text.strip()
+
+    def verify(self, chroot_base):
+        kbdnames = []
+        for i in ('KEYMAP', 'KEYMAP_TOGGLE'):
+            if i in self.settings.keys():
+                kbdnames.append(self.settings[i])
+        if not kbdnames:
+            return(None)
+        keymapdir = os.path.join(chroot_base, 'usr', 'share', 'kbd', 'keymaps')
+        kbdmaps = []
+        for root, dirs, files in os.walk(keymapdir, topdown = True):
+            if root.endswith('/include'):
+                dirs[:] = []
+                files[:] = []
+                continue
+            for f in files:
+                if f.endswith('.inc'):
+                    continue
+                kbdmaps.append(_kbd_re.sub('', f))
+        for k in kbdnames:
+            if k not in kbdmaps:
+                raise ValueError('keyboard map "{0}" is not installed on target system'.format(k))
+        return(True)
+
+
+class Console(object):
+    def __init__(self, console_xml):
+        self.xml = console_xml
+        self._cfg = configparser.ConfigParser()
+        self._cfg.optionxform(str)
+        self.keyboard = Keyboard(self.xml.find('keyboard'))
+        self.font = Font(self.xml.find('text'))
+        self._cfg['BASE'] = {}
+        for i in (self.keyboard, self.font):
+            self._cfg['BASE'].update(i.settings)
+
+    def writeConf(self, chroot_base):
+        for x in (self.font, self.keyboard):
+            x.verify(chroot_base)
+        cfg = os.path.join(chroot_base, 'etc', 'vconsole.conf')
+        # We have to strip out the section from the ini.
+        cfgbuf = io.StringIO()
+        self._cfg.write(cfgbuf, space_around_delimiters = False)
+        cfgbuf.seek(0, 0)
+        with open(cfg, 'w') as fh:
+            for line in cfgbuf.readlines():
+                if line.startswith('[BASE]') or line.strip() == '':
+                    continue
+                fh.write(line)
+        os.chmod(cfg, 0o0644)
+        os.chown(cfg, 0, 0)
+        return()