mirror of
https://github.com/kovidgoyal/kitty.git
synced 2026-02-01 11:34:59 +01:00
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
# License: GPLv3 Copyright: 2020, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
from typing import Any
|
|
|
|
from .base import ArgsType, Boss, PayloadGetType, PayloadType, RCOptions, RemoteCommand, ResponseType, Window
|
|
|
|
|
|
class Env(RemoteCommand):
|
|
|
|
protocol_spec = __doc__ = '''
|
|
env+/dict.str: Dictionary of environment variables to values. When a env var ends with = it is removed from the environment.
|
|
'''
|
|
|
|
short_desc = 'Change environment variables seen by future children'
|
|
desc = (
|
|
'Change the environment variables that will be seen in newly launched windows.'
|
|
' Similar to the :opt:`env` option in :file:`kitty.conf`, but affects running kitty instances.'
|
|
' If no = is present, the variable is removed from the environment.'
|
|
)
|
|
args = RemoteCommand.Args(spec='env_var1=val env_var2=val ...', minimum_count=1, json_field='env')
|
|
|
|
def message_to_kitty(self, global_opts: RCOptions, opts: Any, args: ArgsType) -> PayloadType:
|
|
if len(args) < 1:
|
|
self.fatal('Must specify at least one env var to set')
|
|
env = {}
|
|
for x in args:
|
|
if '=' in x:
|
|
key, val = x.split('=', 1)
|
|
env[key] = val
|
|
else:
|
|
env[x + '='] = ''
|
|
return {'env': env}
|
|
|
|
def response_from_kitty(self, boss: Boss, window: Window | None, payload_get: PayloadGetType) -> ResponseType:
|
|
from kitty.child import default_env, set_default_env
|
|
from kitty.utils import expandvars
|
|
new_env = payload_get('env') or {}
|
|
env = default_env().copy()
|
|
for k, v in new_env.items():
|
|
if k.endswith('='):
|
|
env.pop(k[:-1], None)
|
|
else:
|
|
env[k] = expandvars(v or '', env)
|
|
set_default_env(env)
|
|
return None
|
|
|
|
|
|
env = Env()
|