invoke/luanti-remote/tasks.py

   1 from invoke import task
   2 import os
   3 from datetime import datetime as dt
   4
   5 IP=os.environ.get('LUANTI_IP')
   6 USER=os.environ.get('LUANTI_USER')
   7 HOME=os.environ.get('LUANTI_HOME')
   8 LOCAL_HOME=os.environ.get('HOME')
   9 TODAY = dt.now().strftime('%Y.%m.%d')
  10
  11 WORLD='world'  # World folder
  12
  13 SSH=f'ssh {USER}@{IP}'
  14 SCP=f'scp {USER}@{IP}'
  15 BACKUP_FILE=f'world.{TODAY}.tar'
  16 LOCAL_BACKUPS=f'{LOCAL_HOME}/backup/luanti'
  17
  18 LUANTI=f'{HOME}/luanti'
  19 MODS=f'{LUANTI}/mods'
  20 SERVE=f'screen -d -m {LUANTI}/bin/luantiserver --gameid minetest --worldname world'
  21
  22 def browse(c, url):
  23     c.run(f"librewolf {url}")
  24
  25 @task
  26 def list_mods(c):
  27     c.run(f'{SSH} -C ls {MODS}')
  28
  29 @task
  30 def lookup_mod(c, mod):
  31     browse(c, f'https://content.luanti.org/packages/?q={mod}')
  32
  33 @task
  34 def backup_mod(c, mod):
  35     c.run(f'mkdir -p {LOCAL_BACKUPS}/mods')
  36     tarfile = f'/tmp/{mod}.tar'
  37     c.run(f'{SSH} -C rm -f {tarfile} 2> /dev/null')
  38     c.run(f'{SSH} -C tar cf {tarfile} {MODS}/{mod}')
  39     c.run(f'{SCP}:{tarfile} {LOCAL_BACKUPS}/mods')
  40     c.run(f'ls -l {LOCAL_BACKUPS}/mods')
  41
  42 @task
  43 def unpack_mod(c, mod):
  44     tarfile = f'{LOCAL_BACKUPS}/mods/{mod}.tar'
  45     unpack = f'{LOCAL_BACKUPS}/mods/unpack'
  46     c.run(f'mkdir -p {unpack}')
  47     c.run(f'cp {tarfile} {unpack}/')
  48     c.run(f'cd {unpack}; tar xf {unpack}/{mod}.tar')
  49     c.run(f'ls {unpack}')
  50
  51
  52 @task
  53 def compress(c):
  54     c.run(f'{SSH} -C tar cf {HOME}/{BACKUP_FILE} --directory={HOME} {WORLD}')
  55
  56 @task(compress)
  57 def backup(c):
  58     stopped = input("Is the server stopped?")
  59     if(stopped != 'y'):
  60         print("so go stop the server...")
  61         exit()
  62     print(f'Backup {HOME}/{BACKUP_FILE} to {LOCAL_BACKUPS}')
  63     c.run(f'mkdir -p {LOCAL_BACKUPS}')
  64     c.run(f'{SCP}:{BACKUP_FILE} {LOCAL_BACKUPS}')
  65     print(f'Backedup up {BACKUP_FILE} to {LOCAL_BACKUPS}')
  66     c.run(f'ls {LOCAL_BACKUPS} | grep world')
  67     c.run(f'tar tvf {LOCAL_BACKUPS}/{BACKUP_FILE}')
  68
  69
  70 @task
  71 def cli_hint(c):
  72     browse(c, 'https://console.kamatera.com/cli')
  73
  74 @task
  75 def cheat_sheets(c):
  76     browse(c, 'https://edward.delaporte.us/blog/luanti/')
  77
  78 @task
  79 def verify_backup(c):
  80     TAR_FILE=f'{LOCAL_BACKUPS}/{BACKUP_FILE}'
  81     DB_FILE='map.sqlite'
  82     print(f'Extracting {DB_FILE} from {TAR_FILE}')
  83     c.run(f'cd /tmp;tar --extract --file={LOCAL_BACKUPS}/{BACKUP_FILE} {WORLD}/{DB_FILE}')
  84     result = c.run(f'echo "PRAGMA quick_check;" | sqlite3 /tmp/{DB_FILE}')
  85     # print(f'Check on {DB_FILE}: {result}')
  86     # print(f'Exit code: {result.exited}')
  87
  88     if(not 'ok' in result.stdout):
  89         print("=========================================")
  90         print("database is corrupt, restore from backup!")
  91         print("=========================================")
  92
  93
  94 @task(verify_backup)
  95 def serve(c):
  96     print(f"Attempting to connect to {IP}")
  97     result = c.run(f'ssh {USER}@{IP} -C {SERVE}')
  98     print(f'Server should now be running in a remote screen session.')
  99     print(result)