mirror of
https://github.com/systemd/systemd.git
synced 2026-06-30 19:57:29 +00:00
9e1e2bcedc
We need promptvars, otherwise the prompt strings won't undergo parameter
expansion and we'd print them literally:
$ shopt -u promptvars
$ echo foo
$(__systemd_osc_context_ps0)foo
Resolves: #40620
(cherry picked from commit 7d1b6b7b96)
89 lines
3.8 KiB
Bash
89 lines
3.8 KiB
Bash
# SPDX-License-Identifier: LGPL-2.1-or-later
|
|
# shellcheck shell=bash
|
|
# shellcheck disable=SC2016
|
|
# shellcheck disable=SC1003
|
|
|
|
# This file is part of systemd.
|
|
#
|
|
# systemd is free software; you can redistribute it and/or modify it
|
|
# under the terms of the GNU Lesser General Public License as published by
|
|
# the Free Software Foundation; either version 2.1 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This implements the UAPI.15 "OSC 3008: Hierarchical Context Signalling"
|
|
# specification for the shell prompt. For details see:
|
|
# https://uapi-group.org/specifications/specs/osc_context/
|
|
|
|
# Not bash?
|
|
[ -n "${BASH_VERSION:-}" ] || return 0
|
|
|
|
# If we're on a "dumb" terminal, do not install the prompt.
|
|
# Treat missing $TERM same as "dumb".
|
|
[ "${TERM:-dumb}" = "dumb" ] && return 0
|
|
|
|
# We need promptvars, otherwise the prompt strings won't undergo parameter expansion
|
|
# and we'd print them literally
|
|
shopt -q promptvars || return 0
|
|
|
|
__systemd_osc_context_escape() {
|
|
# Escape according to the OSC 3008 spec. Since this requires shelling out
|
|
# to 'sed' we'll only do it where it's strictly necessary, and skip it when
|
|
# processing strings we are pretty sure we won't need it for, such as
|
|
# uuids, id128, hostnames, usernames, since they all come with syntax
|
|
# requirements that exclude \ and ; anyway. This hence primarily is about
|
|
# escaping the current working directory.
|
|
echo "$1" | sed -e 's/\\/\\x5c/g' -e 's/;/\\x3b/g' -e 's/[[:cntrl:]]/⍰/g'
|
|
}
|
|
|
|
__systemd_osc_context_common() {
|
|
if [ -f /etc/machine-id ]; then
|
|
printf ";machineid=%.36s" "$(</etc/machine-id)"
|
|
fi
|
|
printf ";user=%.255s;hostname=%.255s;bootid=%.36s;pid=%.20d" "$USER" "$HOSTNAME" "$(</proc/sys/kernel/random/boot_id)" "$$"
|
|
}
|
|
|
|
__systemd_osc_context_precmdline() {
|
|
local systemd_exitstatus="$?" systemd_signal
|
|
|
|
# Close previous command
|
|
if [ -n "${systemd_osc_context_cmd_id:-}" ]; then
|
|
if [ "$systemd_exitstatus" -gt 128 ] && systemd_signal=$(kill -l "$systemd_exitstatus" 2>&-); then
|
|
printf "\033]3008;end=%.64s;exit=failure;status=%d;signal=SIG%s\033\\" "$systemd_osc_context_cmd_id" "$systemd_exitstatus" "$systemd_signal"
|
|
elif [ "$systemd_exitstatus" -ne 0 ]; then
|
|
printf "\033]3008;end=%.64s;exit=failure;status=%d\033\\" "$systemd_osc_context_cmd_id" $((systemd_exitstatus))
|
|
else
|
|
printf "\033]3008;end=%.64s;exit=success\033\\" "$systemd_osc_context_cmd_id"
|
|
fi
|
|
fi
|
|
|
|
# Prepare a context ID for this shell if we have none
|
|
if [ -z "${systemd_osc_context_shell_id:-}" ]; then
|
|
read -r systemd_osc_context_shell_id </proc/sys/kernel/random/uuid
|
|
fi
|
|
|
|
# Create or update the shell session
|
|
printf "\033]3008;start=%.64s%s;type=shell;cwd=%.255s\033\\" "$systemd_osc_context_shell_id" "$(__systemd_osc_context_common)" "$(__systemd_osc_context_escape "$PWD")"
|
|
|
|
# Prepare cmd id for next command
|
|
read -r systemd_osc_context_cmd_id </proc/sys/kernel/random/uuid
|
|
}
|
|
|
|
__systemd_osc_context_ps0() {
|
|
# Skip if PROMPT_COMMAND= is cleared manually or by other profiles.
|
|
[ -n "${systemd_osc_context_cmd_id:-}" ] || return
|
|
|
|
printf "\033]3008;start=%.64s%s;type=command;cwd=%.255s\033\\" "$systemd_osc_context_cmd_id" "$(__systemd_osc_context_common)" "$(__systemd_osc_context_escape "$PWD")"
|
|
}
|
|
|
|
if [ -n "${BASH_VERSION:-}" ]; then
|
|
# Legacy bashrc will assign PROMPT_COMMAND=, which is equivalent to assigning
|
|
# index 0 in the array. Leave an empty spot to handle this gracefully.
|
|
[ -n "$(declare -p PROMPT_COMMAND 2>/dev/null)" ] || PROMPT_COMMAND+=('')
|
|
|
|
# Whenever a new prompt is shown, close the previous command, and prepare new command
|
|
PROMPT_COMMAND+=(__systemd_osc_context_precmdline)
|
|
|
|
# PS0 is shown right after a prompt completed, but before the command is executed
|
|
PS0='$(__systemd_osc_context_ps0)'"${PS0:-}"
|
|
fi
|