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'
  12 SERVE=f'screen -d -m {HOME}/luanti/bin/luantiserver --gameid minetest --worldname world'
  13
  14 SSH=f'ssh {USER}@{IP}'
  15 SCP=f'scp {USER}@{IP}'
  16 BACKUP_FILE=f'world.{TODAY}.tar'
  17 LOCAL_BACKUPS=f'{LOCAL_HOME}/backup/luanti'
  18
  19
  20 def browse(c, url):
  21     c.run(f"librefox {url}")
  22
  23
  24 @task
  25 def compress(c):
  26     c.run(f'{SSH} -C tar cf {HOME}/{BACKUP_FILE} --directory={HOME} {WORLD}')
  27
  28 @task(compress)
  29 def backup(c):
  30     stopped = input("Is the server stopped?")
  31     if(stopped != 'y'):
  32         print("so go stop the server...")
  33         exit()
  34     print(f'Backup {HOME}/{BACKUP_FILE} to {LOCAL_BACKUPS}')
  35     c.run(f'mkdir -p {LOCAL_BACKUPS}')
  36     c.run(f'{SCP}:{BACKUP_FILE} {LOCAL_BACKUPS}')
  37     print(f'Backedup up {BACKUP_FILE} to {LOCAL_BACKUPS}')
  38     c.run(f'ls {LOCAL_BACKUPS} | grep world')
  39     c.run(f'tar tvf {LOCAL_BACKUPS}/{BACKUP_FILE}')
  40
  41
  42 @task
  43 def cli_hint(c):
  44     browse(c, 'https://console.kamatera.com/cli')
  45
  46 @task
  47 def cheat_sheets(c):
  48     browse(c, 'https://edward.delaporte.us/blog/luanti/')
  49
  50 @task
  51 def verify_backup(c):
  52     TAR_FILE=f'{LOCAL_BACKUPS}/{BACKUP_FILE}'
  53     DB_FILE='map.sqlite'
  54     print(f'Extracting {DB_FILE} from {TAR_FILE}')
  55     c.run(f'cd /tmp;tar --extract --file={LOCAL_BACKUPS}/{BACKUP_FILE} {WORLD}/{DB_FILE}')
  56     result = c.run(f'echo "PRAGMA quick_check;" | sqlite3 /tmp/{DB_FILE}')
  57     # print(f'Check on {DB_FILE}: {result}')
  58     # print(f'Exit code: {result.exited}')
  59
  60     if(not 'ok' in result.stdout):
  61         print("=========================================")
  62         print("database is corrupt, restore from backup!")
  63         print("=========================================")
  64
  65 @task(verify_backup)
  66 def serve(c):
  67     print(f"Attempting to connect to {IP}")
  68     result = c.run(f'ssh {USER}@{IP} -C {SERVE}')
  69     print(f'Server should now be running in a remote screen session.')
  70     print(result)