210 lines
9.5 KiB
Python
Executable file
210 lines
9.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import os, sys, yaml, time, getpass, shutil, fileinput, subprocess
|
|
|
|
def get_containers():
|
|
container_list = subprocess.run(['sudo', '-u', user, 'podman', 'ps', '-a', '--no-trunc', '--size', '--sort=created', '--format',
|
|
'{{.Names}}'], stdout=subprocess.PIPE).stdout.decode('utf-8').strip().split('\n')
|
|
|
|
try:
|
|
with open(os.path.expanduser('~/.config/blend/config.yaml')) as config_file:
|
|
data = yaml.safe_load(config_file)
|
|
order = data['container_order'].copy()
|
|
order.reverse()
|
|
container_list.reverse()
|
|
for i in container_list:
|
|
if i.strip() not in order:
|
|
order.insert(0, i)
|
|
for i, o in enumerate(order):
|
|
if o not in container_list:
|
|
del order[i]
|
|
return order
|
|
except:
|
|
return container_list
|
|
|
|
def list_use_container_bin():
|
|
try:
|
|
with open(os.path.expanduser('~/.config/blend/config.yaml')) as config_file:
|
|
data = yaml.safe_load(config_file)
|
|
return data['use_container_bins']
|
|
except:
|
|
return []
|
|
|
|
def check_if_present(attr, desktop_str):
|
|
for l in desktop_str:
|
|
if l.startswith(attr + '='):
|
|
return True
|
|
return False
|
|
|
|
def which(bin):
|
|
results = []
|
|
for dir in os.environ.get('PATH').split(':'):
|
|
if os.path.isdir(dir):
|
|
for i in os.listdir(dir):
|
|
if os.path.basename(bin) == i:
|
|
results.append(os.path.join(dir, i))
|
|
if results == []:
|
|
return None
|
|
return results
|
|
|
|
def create_container_binaries():
|
|
_binaries = []
|
|
remove_binaries = []
|
|
|
|
for c in _list:
|
|
c = c.strip()
|
|
for i in con_get_output(c, '''find /usr/bin -type f -printf "%P\n" 2>/dev/null;
|
|
find /usr/local/bin -type f -printf "%P\n" 2>/dev/null;
|
|
find /usr/sbin -type f -printf "%P\n" 2>/dev/null;''').split('\n'):
|
|
i = i.strip()
|
|
os.makedirs(os.path.expanduser(f'~/.local/bin/blend_{c}'), exist_ok=True)
|
|
i_present = False
|
|
orig_which_out = which(os.path.basename(i))
|
|
which_out = None
|
|
if orig_which_out != None:
|
|
which_out = orig_which_out.copy()
|
|
try:
|
|
which_out.remove(os.path.expanduser(f'~/.local/bin/blend_bin/{os.path.basename(i)}'))
|
|
except ValueError:
|
|
pass
|
|
if which_out == []:
|
|
which_out = None
|
|
if which_out != None and os.path.basename(i) not in _exceptions:
|
|
i_present = True
|
|
|
|
if os.path.basename(i) != 'host-spawn' and i != '' and not i_present:
|
|
with open(os.path.expanduser(f'~/.local/bin/blend_{c}/{os.path.basename(i)}.tmp'), 'w') as f:
|
|
f.write('#!/bin/sh\n')
|
|
f.write(f'# blend container: {i}\n')
|
|
if os.path.basename(i) in _exceptions:
|
|
f.write(f'# EXCEPTION\n')
|
|
f.write(f'BLEND_ALLOW_ROOT= BLEND_NO_CHECK= blend enter -cn {c} -- {i} "$@"\n')
|
|
# XXX: make this bit fully atomic
|
|
os.chmod(os.path.expanduser(f'~/.local/bin/blend_{c}/{os.path.basename(i)}.tmp'), 0o775)
|
|
subprocess.call(['mv', os.path.expanduser(f'~/.local/bin/blend_{c}/{os.path.basename(i)}.tmp'),
|
|
os.path.expanduser(f'~/.local/bin/blend_{c}/{os.path.basename(i)}')])
|
|
_binaries.append((c, os.path.basename(os.path.basename(i))))
|
|
|
|
os.makedirs(os.path.expanduser(f'~/.local/bin/blend_bin'), exist_ok=True)
|
|
|
|
for c, i in _binaries:
|
|
try:
|
|
os.symlink(os.path.expanduser(f'~/.local/bin/blend_{c}/{i}'), os.path.expanduser(f'~/.local/bin/blend_bin/{i}'))
|
|
except FileExistsError:
|
|
if not subprocess.call(['grep', '-q', f'^# container: {c}$', os.path.expanduser(f'~/.local/bin/blend_bin/{i}')]):
|
|
os.remove(os.path.expanduser(f'~/.local/bin/blend_bin/{i}'))
|
|
os.symlink(os.path.expanduser(f'~/.local/bin/blend_{c}/{i}'), os.path.expanduser(f'~/.local/bin/blend_bin/{i}'))
|
|
|
|
for i in remove_binaries:
|
|
try:
|
|
os.remove(i)
|
|
except:
|
|
pass
|
|
|
|
for b in os.listdir(os.path.expanduser(f'~/.local/bin/blend_bin')):
|
|
if [_b for _b in _binaries if _b[1] == b] == []:
|
|
os.remove(os.path.join(os.path.expanduser(f'~/.local/bin/blend_bin'), b))
|
|
|
|
def create_container_applications():
|
|
_apps = []
|
|
|
|
os.makedirs(os.path.expanduser(f'~/.local/share/applications'), exist_ok=True)
|
|
|
|
for c in _list:
|
|
c = c.strip()
|
|
for i in con_get_output(c, 'find /usr/share/applications -type f 2>/dev/null; find /usr/local/share/applications -type f 2>/dev/null').split('\n'):
|
|
orig_path = i.strip()
|
|
i = os.path.basename(orig_path)
|
|
i_present = (os.path.isfile(f'/usr/share/applications/{i}') or os.path.isfile(f'/usr/local/share/applications/{i}')
|
|
or os.path.isfile(os.path.expanduser(f'~/.local/share/applications/{i}')))
|
|
if i != '' and not i_present:
|
|
with open(os.path.expanduser(f'~/.local/share/applications/blend;{i}'), 'w') as f:
|
|
_ = con_get_output(c, f"sudo sed -i '/^DBusActivatable=/d' {orig_path}")
|
|
_ = con_get_output(c, f"sudo sed -i '/^TryExec=/d' {orig_path}")
|
|
contents = con_get_output(c, f'cat {orig_path}')
|
|
f.write(contents)
|
|
for line in fileinput.input(os.path.expanduser(f'~/.local/share/applications/blend;{i}'), inplace=True):
|
|
if line.strip().startswith('Exec='):
|
|
line = f'Exec=env BLEND_NO_CHECK= blend enter -cn {c} -- {line[5:]}\n'
|
|
elif line.strip().startswith('Icon='):
|
|
if '/' in line:
|
|
line = line.strip()
|
|
_ = con_get_output(c, f"mkdir -p ~/.local/share/blend/icons/file/\"{c}_{i}\"; cp {line[5:]} ~/.local/share/blend/icons/file/\"{c}_{i}\"")
|
|
line = f'Icon={os.path.expanduser("~/.local/share/blend/icons/file/" + c + "_" + i + "/" + os.path.basename(line[5:]))}\n'
|
|
else:
|
|
line = line.strip()
|
|
icons = con_get_output(c, f'''find /usr/share/icons /usr/share/pixmaps /var/lib/flatpak/exports/share/icons \\
|
|
-type f -iname "*{line[5:]}*" 2> /dev/null | sort''').split('\r\n')
|
|
_ = con_get_output(c, f"mkdir -p ~/.local/share/blend/icons/\"{c}_{i}\"; cp {icons[0]} ~/.local/share/blend/icons/\"{c}_{i}\"")
|
|
line = f'Icon={os.path.expanduser("~/.local/share/blend/icons/" + c + "_" + i + "/" + os.path.basename(icons[0]))}\n'
|
|
sys.stdout.write(line)
|
|
os.chmod(os.path.expanduser(f'~/.local/share/applications/blend;{i}'), 0o775)
|
|
_apps.append((c, i))
|
|
del _
|
|
|
|
for a in os.listdir(os.path.expanduser(f'~/.local/share/applications')):
|
|
if a.startswith('blend;'):
|
|
a = a.removeprefix('blend;')
|
|
if [_a for _a in _apps if _a[1] == a] == []:
|
|
os.remove(os.path.expanduser(f'~/.local/share/applications/blend;{a}'))
|
|
|
|
def create_container_sessions(type='xsessions'):
|
|
session_dir = f'/usr/share/{type}'
|
|
|
|
os.makedirs('/usr/share/xsessions', exist_ok=True)
|
|
|
|
for session in os.listdir(session_dir):
|
|
if session.startswith(os.path.join(session_dir, 'blend-')):
|
|
os.remove(os.path.join(session_dir, session))
|
|
|
|
for c in _list:
|
|
c = c.strip()
|
|
for i in con_get_output(c, f'find {session_dir} -type f 2>/dev/null').split('\n'):
|
|
orig_path = i.strip()
|
|
i = os.path.basename(orig_path)
|
|
if i != '':
|
|
with open(os.path.expanduser(f'{session_dir}/blend-{c};{i}'), 'w') as f:
|
|
contents = con_get_output(c, f'cat {orig_path}')
|
|
f.write(contents)
|
|
for line in fileinput.input(os.path.expanduser(f'/{session_dir}/blend-{c};{i}'), inplace=True):
|
|
if line.strip().startswith('Name'):
|
|
name = line.split('=')[1]
|
|
line = f'Name=Container {c}: {name}'
|
|
elif line.strip().startswith('Exec='):
|
|
line = f'Exec=blend enter -cn {c} -- {line[5:]}'
|
|
elif line.strip().startswith('TryExec='):
|
|
continue
|
|
|
|
sys.stdout.write(line)
|
|
os.chmod(os.path.expanduser(f'{session_dir}/blend-{c};{i}'), 0o775)
|
|
|
|
con_get_output = lambda name, cmd: subprocess.run(['sudo', '-u', user, 'podman', 'exec', '--user', getpass.getuser(), '-it', name, 'bash', '-c', cmd],
|
|
stdout=subprocess.PIPE, stderr=subprocess.DEVNULL).stdout.decode('UTF-8').strip()
|
|
|
|
user = getpass.getuser()
|
|
|
|
try:
|
|
user = sys.argv[2]
|
|
except:
|
|
pass
|
|
|
|
try:
|
|
if sys.argv[1] == 'sessions':
|
|
_list = get_containers()
|
|
create_container_sessions(type='xsessions')
|
|
create_container_sessions(type='wayland-sessions')
|
|
exit(0)
|
|
except IndexError:
|
|
pass
|
|
|
|
for c in get_containers():
|
|
c = c.strip()
|
|
subprocess.call(['podman', 'start', c])
|
|
|
|
while True:
|
|
_list = get_containers()
|
|
_exceptions = list_use_container_bin()
|
|
|
|
create_container_binaries()
|
|
create_container_applications()
|
|
time.sleep(6)
|