invoke/luanti-remote/tasks.py

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