61 lines
1.5 KiB
Python
Executable file
61 lines
1.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import argparse
|
|
|
|
from pathlib import Path
|
|
from subprocess import check_call
|
|
|
|
# TODO rename to grasp_server.py
|
|
from server import setup_parser
|
|
|
|
SYSTEMD_CONFIG = """
|
|
[Unit]
|
|
Description=Grasp extension server conterpart
|
|
|
|
[Install]
|
|
WantedBy=default.target
|
|
|
|
[Service]
|
|
ExecStart={server} {extra_args}
|
|
Type=simple
|
|
Restart=always
|
|
"""
|
|
|
|
|
|
def systemd(*args):
|
|
check_call([
|
|
'systemctl', '--user', *args,
|
|
])
|
|
|
|
def setup(args):
|
|
unit_name = args.unit_name
|
|
out = Path(f'~/.config/systemd/user/{unit_name}').expanduser()
|
|
print(f"Writing systemd config to {out}:")
|
|
|
|
server_bin = Path(__file__).parent.joinpath('server.py').absolute()
|
|
extra_args = f'--port {args.port} --path {args.path}'
|
|
|
|
with out.open('w') as fo:
|
|
fo.write(SYSTEMD_CONFIG.format(
|
|
server=server_bin,
|
|
extra_args=extra_args,
|
|
))
|
|
|
|
try:
|
|
systemd('daemon-reload')
|
|
systemd('enable', unit_name)
|
|
systemd('start' , unit_name)
|
|
systemd('status', unit_name)
|
|
except Exception as e:
|
|
print(f"Something has gone wrong... you might want to use 'journalctl --user -u {unit_name}' to debug")
|
|
raise e
|
|
|
|
def main():
|
|
p = argparse.ArgumentParser('grasp server setup', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
|
p.add_argument('--unit-name', type=str, default='grasp.service', help='Systemd unit name')
|
|
setup_parser(p)
|
|
args = p.parse_args()
|
|
setup(args)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|