from invoke import task
import os
from datetime import datetime as dt

IP=os.environ.get('LUANTI_IP')
USER=os.environ.get('LUANTI_USER')
HOME=os.environ.get('LUANTI_HOME')
LOCAL_HOME=os.environ.get('HOME')
TODAY = dt.now().strftime('%Y.%m.%d')

WORLD='world'
SERVE=f'screen -d -m {HOME}/luanti/bin/luantiserver --gameid minetest --worldname world'

SSH=f'ssh {USER}@{IP}'
SCP=f'scp {USER}@{IP}'
BACKUP_FILE=f'world.{TODAY}.tar'
LOCAL_BACKUPS=f'{LOCAL_HOME}/backup/luanti'


def browse(c, url):
    c.run(f"librefox {url}")


@task
def compress(c):
    c.run(f'{SSH} -C tar cf {HOME}/{BACKUP_FILE} --directory={HOME} {WORLD}')

@task(compress)
def backup(c):
    stopped = input("Is the server stopped?")
    if(stopped != 'y'):
        print("so go stop the server...")
        exit()
    print(f'Backup {HOME}/{BACKUP_FILE} to {LOCAL_BACKUPS}')
    c.run(f'mkdir -p {LOCAL_BACKUPS}')
    c.run(f'{SCP}:{BACKUP_FILE} {LOCAL_BACKUPS}')
    print(f'Backedup up {BACKUP_FILE} to {LOCAL_BACKUPS}')
    c.run(f'ls {LOCAL_BACKUPS} | grep world')
    c.run(f'tar tvf {LOCAL_BACKUPS}/{BACKUP_FILE}')


@task
def cli_hint(c):
    browse(c, 'https://console.kamatera.com/cli')

@task
def cheat_sheets(c):
    browse(c, 'https://edward.delaporte.us/blog/luanti/')

@task
def verify_backup(c):
    TAR_FILE=f'{LOCAL_BACKUPS}/{BACKUP_FILE}'
    DB_FILE='map.sqlite'
    print(f'Extracting {DB_FILE} from {TAR_FILE}')
    c.run(f'tar --extract --file={LOCAL_BACKUPS}/{BACKUP_FILE} {WORLD}/{DB_FILE}')
    result = c.run(f'echo "PRAGMA quick_check;" | sqlite3 {DB_FILE}')
    # print(f'Check on {DB_FILE}: {result}')
    # print(f'Exit code: {result.exited}')

    if(not 'ok' in result.stdout):
        print("=========================================")
        print("database is corrupt, restore from backup!")
        print("=========================================")

@task(verify_backup)
def serve(c):
    print(f"Attempting to connect to {IP}")
    result = c.run(f'ssh {USER}@{IP} -C {SERVE}')
    print(f'Server should now be running in a remote screen session.')
    print(result)