Files
openfreemap/init-server.py
Zsolt Ero 5831068231 work
2023-12-07 22:50:36 +01:00

87 lines
2.1 KiB
Python
Executable File

#!/usr/bin/env python3
import click
from dotenv import dotenv_values
from fabric import Config, Connection
from ssh_lib.config import scripts
from ssh_lib.kernel import set_cpu_governor, setup_kernel_settings
from ssh_lib.nginx import certbot, nginx
from ssh_lib.pkg_base import pkg_base, pkg_clean, pkg_upgrade
from ssh_lib.planetiler import PLANETILER_DIR, install_planetiler
from ssh_lib.utils import add_user, put, setup_time
def prepare_shared(c):
add_user(c, 'ofm')
pkg_upgrade(c)
pkg_clean(c)
pkg_base(c)
setup_time(c)
setup_kernel_settings(c)
set_cpu_governor(c)
def prepare_tile_creator(c):
install_planetiler(c)
put(
c,
scripts / 'tile_creator' / 'run_planet.sh',
PLANETILER_DIR,
permissions='755',
owner='ofm',
)
def prepare_http_host(c):
nginx(c)
certbot(c)
@click.command()
@click.argument('hostname')
@click.option('--port', type=int, help='SSH port')
@click.option('--user', help='SSH user')
@click.option('--tile-creator', is_flag=True, help='Install tile-creator task')
@click.option('--http-host', is_flag=True, help='Install http-host task')
def main(hostname, user, port, tile_creator, http_host):
if not click.confirm(f'Run script on {hostname}?'):
return
if not tile_creator and not http_host:
tile_creator = click.confirm('Would you like to install tile-creator task?')
http_host = click.confirm('Would you like to install http-host task?')
if not tile_creator and not http_host:
return
ssh_passwd = dotenv_values('.env').get('SSH_PASSWD')
if ssh_passwd:
c = Connection(
host=hostname,
user=user,
port=port,
connect_kwargs={'password': ssh_passwd},
config=Config(overrides={'sudo': {'password': ssh_passwd}}),
)
else:
c = Connection(
host=hostname,
user=user,
port=port,
)
# prepare_shared(c)
if tile_creator:
prepare_tile_creator(c)
if http_host:
prepare_http_host(c)
if __name__ == '__main__':
main()