mirror of
https://github.com/BigfootACA/arch-image-builder.git
synced 2024-11-11 07:57:53 +08:00
96 lines
2.2 KiB
Python
96 lines
2.2 KiB
Python
import os
|
|
from logging import getLogger
|
|
from builder.build import filesystem
|
|
from builder.lib.context import ArchBuilderContext
|
|
from builder.lib.config import ArchBuilderConfigError
|
|
from builder.lib.utils import open_config
|
|
log = getLogger(__name__)
|
|
|
|
|
|
def reset_locale(ctx: ArchBuilderContext):
|
|
"""
|
|
Remove old locale settings
|
|
"""
|
|
root = ctx.get_rootfs()
|
|
archive = os.path.join(root, "usr/lib/locale/locale-archive")
|
|
if os.path.exists(archive): os.remove(archive)
|
|
|
|
|
|
def enable_all(ctx: ArchBuilderContext):
|
|
"""
|
|
Add all enabled locale for build locale-archive
|
|
"""
|
|
root = ctx.get_rootfs()
|
|
|
|
# default none
|
|
locales = ctx.get("locale.enable", [])
|
|
log.info("setup enabled locale")
|
|
|
|
# create locale.gen
|
|
file = os.path.join(root, "etc/locale.gen")
|
|
with open_config(file) as f:
|
|
for line in locales:
|
|
log.debug(f"adding locale {line}")
|
|
f.write(line)
|
|
f.write(os.linesep)
|
|
if len(locales) == 0:
|
|
f.write("# No any locales enabled\n")
|
|
|
|
# run locale-gen
|
|
filesystem.chroot_run(ctx, "locale-gen")
|
|
|
|
|
|
def set_default(ctx: ArchBuilderContext):
|
|
"""
|
|
Setup default locale
|
|
"""
|
|
root = ctx.get_rootfs()
|
|
|
|
# default to C
|
|
default = ctx.get("locale.default", "C")
|
|
log.info(f"default locale: {default}")
|
|
|
|
# default locale config (see man:locale.conf(5))
|
|
conf = os.path.join(root, "etc/locale.conf")
|
|
with open_config(conf) as f:
|
|
f.write(f"LANG={default}\n")
|
|
|
|
|
|
def set_timezone(ctx: ArchBuilderContext):
|
|
"""
|
|
Setup tzdata timezone info
|
|
"""
|
|
root = ctx.get_rootfs()
|
|
|
|
# default to UTC
|
|
timezone = ctx.get("timezone", "UTC")
|
|
log.info(f"timezone: {timezone}")
|
|
|
|
# tzdata install path
|
|
dst = os.path.join("/usr/share/zoneinfo", timezone)
|
|
real = os.path.join(root, dst[1:])
|
|
if not os.path.exists(real): raise ArchBuilderConfigError(
|
|
f"timezone {timezone} not found"
|
|
)
|
|
|
|
# localtime symbolic link (see man:tzset(3))
|
|
lnk = os.path.join(root, "etc/localtime")
|
|
if os.path.exists(lnk): os.remove(lnk)
|
|
os.symlink(dst, lnk)
|
|
|
|
# timezone file
|
|
conf = os.path.join(root, "etc/timezone")
|
|
with open(conf, "w") as f:
|
|
f.write(timezone)
|
|
f.write(os.linesep)
|
|
|
|
|
|
def proc_locale(ctx: ArchBuilderContext):
|
|
"""
|
|
Setup user locale settings
|
|
"""
|
|
reset_locale(ctx)
|
|
enable_all(ctx)
|
|
set_default(ctx)
|
|
set_timezone(ctx)
|