config-xonsh/rc.xsh

$VI_MODE = True
$AUTO_PUSHD = True
$XONSH_AUTOPAIR = True
# tab selection: do not execute cmd when pressing enter
$COMPLETIONS_CONFIRM = True

xontrib load vox z

def _last_exit_status():
    try:
        exit_status = __xonsh_history__.rtns[-1]
        return exit_status if exit_status != 0 else None
    except IndexError:
        return None
$PROMPT_FIELDS['last_exit_status'] = _last_exit_status

$SHLVL = int($SHLVL) + 1 if 'SHLVL' in ${...} else 1
$XONSH_STDERR_PREFIX = '{RED}'
$XONSH_STDERR_POSTFIX = '{NO_COLOR}'
$DYNAMIC_CWD_WIDTH = '30%'
$DYNAMIC_CWD_ELISION_CHAR = '…'
$PROMPT = ''.join([
    '{RED}{last_exit_status:[{}] }',
    '{BOLD_GREEN}{user}@{hostname} ',
    '{YELLOW}{cwd} ',
    '{{BLUE}}{} '.format('{prompt_end}' * $SHLVL),
    '{NO_COLOR}',
])
$RIGHT_PROMPT = '{gitstatus}{env_name: {}}'

import datetime as dt
import io
import json
import os
import re
import select
import shutil
import subprocess
import sys
import threading

# default locale
# will be used for all non-explicitly set LC_* variables
$LANG = 'en_US.UTF-8'
# fallback locales
# GNU gettext gives preference to LANGUAGE over LC_ALL and LANG
# for the purpose of message handling
# https://www.gnu.org/software/gettext/manual/html_node/The-LANGUAGE-variable.html
# cave: if this list contains 'de(_.*)?' at any (sic!) position
#       vim 7.4.1689 will switch to german
$LANGUAGE = ':'.join(['en_US', 'en'])
$LC_COLLATE = 'C.UTF-8'
# char classification, case conversion & other char attrs
$LC_CTYPE = 'de_AT.UTF-8'
# $ locale currency_symbol
$LC_MONETARY = 'de_AT.UTF-8'
# $ locale -k LC_NUMERIC | head -n 3
# decimal_point="."
# thousands_sep=""
# grouping=-1
$LC_NUMERIC = 'C.UTF-8'
# A4
$LC_PAPER = 'de_AT.UTF-8'
USER_BIN_PATH = os.path.join($HOME, '.local', 'bin')
if os.path.isdir(USER_BIN_PATH):
    $PATH.insert(0, USER_BIN_PATH)
$PAGER = 'less'
$EDITOR = 'vim'
# required by pinentry-tty when using gpg command:
$GPG_TTY = $(tty)
if shutil.which('gpgconf'):
    # required by scute
    $GPG_AGENT_INFO = $(gpgconf --list-dir agent-socket).rstrip() + ':0:1'
    if not 'SSH_CLIENT' in ${...}:
        # in gnupg 2.1.13 the location of agents socket changed
        $SSH_AUTH_SOCK = $(gpgconf --list-dir agent-ssh-socket).rstrip()
# wrapper for termite required when launching termite from ranger:
$TERMCMD = os.path.join(os.path.dirname(__file__), 'ranger-termite-termcmd')
# https://docs.docker.com/engine/security/trust/content_trust/
$DOCKER_CONTENT_TRUST = 1

class DockerImage:

    def __init__(self, image):
        attrs, = json.loads(subprocess.check_output(['sudo', 'docker', 'image', 'inspect', image])
                            .decode(sys.stdout.encoding))
        self._id = attrs['Id']
        self._tags = attrs['RepoTags']

    def __repr__(self):
        return '{}(id={!r}, tags={!r})'.format(type(self).__name__, self._id, self._tags)

    @classmethod
    def pull(cls, image):
        out = io.BytesIO()
        with StdoutTee(out) as tee:
            subprocess.run(['sudo', 'docker', 'image', 'pull', image], stdout=tee)
        repo_digest, = re.search(rb'^Digest: (sha\S+:\S+)$', out.getvalue(), re.MULTILINE).groups()
        return cls('{}@{}'.format(image, repo_digest.decode()))

class StdoutTee:

    def __init__(self, sink):
        self._sink = sink

    def __enter__(self):
        self._read_fd, self._write_fd = os.pipe()
        self._thread = threading.Thread(target=self._loop)
        self._thread.start()
        return self

    def _loop(self):
        while True:
            try:
                data = os_read_non_blocking(self._read_fd)
            except OSError:  # fd closed
                return
            if data:
                self._sink.write(data)
                sys.stdout.buffer.write(data)
                sys.stdout.flush()

    def fileno(self):
        return self._write_fd

    def __exit__(self, exc_type, exc_value, traceback):
        os.close(self._read_fd)
        os.close(self._write_fd)
        self._thread.join()

def dpkg_listfiles(pkg_name):
    assert isinstance(pkg_name, str)
    paths = $(dpkg --listfiles @(pkg_name)).split('\n')[:-1]
    assert len(paths) > 0, 'pkg {!r} not installed'.format(pkg_name)
    return paths

def dpkg_search(path_search_pattern):
    assert isinstance(path_search_pattern, str)
    return re.findall(
        '^(\S+): (.*)$\n',
        $(dpkg --search @(path_search_pattern)),
        flags=re.MULTILINE,
    )

def dpkg_welse(cmd):
    pkg_name, cmd_path = dpkg_which(cmd)
    return dpkg_listfiles(pkg_name)

def dpkg_which(cmd):
    cmd_path = shutil.which(cmd)
    assert cmd_path, 'cmd {!r} not found'.format(cmd)
    matches = dpkg_search(cmd_path)
    assert len(matches) != 0, '{!r} not installed via dpkg'.format(cmd_path)
    assert len(matches) == 1
    return matches[0]

def docker_build(dockerfile):
    p = subprocess.Popen(['sudo', 'docker', 'build', '-'],
                         stdin=subprocess.PIPE, stdout=subprocess.PIPE)
    p.stdin.write(dockerfile.encode())
    p.stdin.close()
    image_id_regex = re.compile(rb'^Successfully built (\S+)\n$')
    image_id = None
    for line in p.stdout:
        sys.stdout.write(line.decode(sys.stdout.encoding))
        image_id_match = image_id_regex.search(line)
        if image_id_match:
            image_id, = image_id_match.groups()
    assert not image_id is None, 'could not determine image id'
    return image_id.decode(sys.stdout.encoding)

def docker_run(image_id=None, dockerfile=None, caps=[]):
    assert image_id is None or dockerfile is None, \
        'either pass kwarg image_id or dockerfile'
    if not image_id:
        image_id = docker_build(dockerfile)
    params = ['sudo', 'docker', 'run',
              '--rm=true', '--interactive=true', '--tty=true',
              '--cap-drop=all', '--security-opt=no-new-privileges']
    params.extend(['--cap-add={}'.format(c) for c in caps])
    params.append(image_id)
    import shlex
    print(' '.join([shlex.quote(p) for p in params]))
    subprocess.run(params, check=True)

def locate(*patterns, match_all=True, ignore_case=True):
    params = []
    if match_all:
        params.insert(0, '--all')
    if ignore_case:
        params.insert(0, '--ignore-case')
    return $(locate @(params) -- @(patterns)).split('\n')[:-1]

def os_read_non_blocking(fd, buffer_size_bytes=8*1024, timeout_seconds=0.1):
    if fd in select.select([fd], [], [], timeout_seconds)[0]:
        return os.read(fd, buffer_size_bytes)
    else:
        return None

def timestamp_now_utc():
    return dt.datetime.utcnow().replace(tzinfo=dt.timezone.utc)

def timestamp_now_local():
    # if called without tz argument astimezone() assumes
    # the system local timezone for the target timezone
    return timestamp_now_utc().astimezone()

def timestamp_iso_local():
    # if called without tz argument astimezone() assumes
    # the system local timezone for the target timezone
    return timestamp_now_local().strftime('%Y%m%dT%H%M%S%z')

aliases['d'] = ['sudo', 'docker']
aliases['d-r'] = lambda args: docker_run(**{
    'dockerfile' if '\n' in args[0] else 'image_id': args[0],
})
aliases['dpkg-welse'] = lambda args: '\n'.join(dpkg_welse(args[0]))
aliases['dpkg-which'] = lambda args: '\t'.join(dpkg_which(args[0]))
aliases['g'] = ['git']
aliases['ll'] = ['ls', '-l', '--all', '--indicator-style=slash',
                 '--human-readable', '--time-style=long-iso', '--color=auto']

if shutil.which('startx') and $(tty).rstrip() == '/dev/tty1':
    startx

# vim: filetype=python