gst-env.py: Output a setting for the prompt with --only-environment

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/3708>
This commit is contained in:
Nirbheek Chauhan 2023-01-11 02:47:50 +05:30 committed by GStreamer Marge Bot
parent 3d13a0d252
commit c4d8403782

View file

@ -572,6 +572,7 @@ if __name__ == "__main__":
if not args: if not args:
if os.name != 'nt': if os.name != 'nt':
args = [os.environ.get("SHELL", os.path.realpath("/bin/sh"))] args = [os.environ.get("SHELL", os.path.realpath("/bin/sh"))]
prompt_export = f'export PS1="[{gst_version}] $PS1"'
if args[0].endswith('bash') and not str_to_bool(os.environ.get("GST_BUILD_DISABLE_PS1_OVERRIDE", r"FALSE")): if args[0].endswith('bash') and not str_to_bool(os.environ.get("GST_BUILD_DISABLE_PS1_OVERRIDE", r"FALSE")):
# Let the GC remove the tmp file # Let the GC remove the tmp file
tmprc = tempfile.NamedTemporaryFile(mode='w') tmprc = tempfile.NamedTemporaryFile(mode='w')
@ -579,7 +580,7 @@ if __name__ == "__main__":
if os.path.exists(bashrc): if os.path.exists(bashrc):
with open(bashrc, 'r') as src: with open(bashrc, 'r') as src:
shutil.copyfileobj(src, tmprc) shutil.copyfileobj(src, tmprc)
tmprc.write('\nexport PS1="[%s] $PS1"' % gst_version) tmprc.write('\n' + prompt_export)
tmprc.flush() tmprc.flush()
if is_bash_completion_available(options): if is_bash_completion_available(options):
bash_completions_files = [] bash_completions_files = []
@ -592,6 +593,7 @@ if __name__ == "__main__":
args.append("--rcfile") args.append("--rcfile")
args.append(tmprc.name) args.append(tmprc.name)
elif args[0].endswith('fish'): elif args[0].endswith('fish'):
prompt_export = None # FIXME
# Ignore SIGINT while using fish as the shell to make it behave # Ignore SIGINT while using fish as the shell to make it behave
# like other shells such as bash and zsh. # like other shells such as bash and zsh.
# See: https://gitlab.freedesktop.org/gstreamer/gst-build/issues/18 # See: https://gitlab.freedesktop.org/gstreamer/gst-build/issues/18
@ -604,6 +606,7 @@ if __name__ == "__main__":
end'''.format(gst_version) end'''.format(gst_version)
args.append(prompt_cmd) args.append(prompt_cmd)
elif args[0].endswith('zsh'): elif args[0].endswith('zsh'):
prompt_export = f'export PROMPT="[{gst_version}] $PROMPT"'
tmpdir = tempfile.TemporaryDirectory() tmpdir = tempfile.TemporaryDirectory()
# Let the GC remove the tmp file # Let the GC remove the tmp file
tmprc = open(os.path.join(tmpdir.name, '.zshrc'), 'w') tmprc = open(os.path.join(tmpdir.name, '.zshrc'), 'w')
@ -611,7 +614,7 @@ if __name__ == "__main__":
if os.path.exists(zshrc): if os.path.exists(zshrc):
with open(zshrc, 'r') as src: with open(zshrc, 'r') as src:
shutil.copyfileobj(src, tmprc) shutil.copyfileobj(src, tmprc)
tmprc.write('\nexport PROMPT="[{}] $PROMPT"'.format(gst_version)) tmprc.write('\n' + prompt_export)
tmprc.flush() tmprc.flush()
env['ZDOTDIR'] = tmpdir.name env['ZDOTDIR'] = tmpdir.name
try: try:
@ -619,6 +622,8 @@ if __name__ == "__main__":
for name, value in env.items(): for name, value in env.items():
print('{}={}'.format(name, shlex.quote(value))) print('{}={}'.format(name, shlex.quote(value)))
print('export {}'.format(name)) print('export {}'.format(name))
if prompt_export:
print(prompt_export)
else: else:
if os.environ.get("CI_PROJECT_NAME"): if os.environ.get("CI_PROJECT_NAME"):
print("Ignoring SIGINT when running on the CI," print("Ignoring SIGINT when running on the CI,"