#!/usr/bin/env python3
"""Run the vanilla 26.3 dedicated server headless on a world folder, optionally
send console commands (`sleep SECONDS` pauses between them; `login NAME
SECONDS` joins with tools/vanilla_login.py in the background), then stop it
cleanly and report problems from its log.

  vanilla_world_check.py --jar server.jar --dir RUN_DIR [--world NAME]
                         [--seed N] [--port P] [--command CMD ...]
                         [--wait-after SECONDS]

RUN_DIR receives eula.txt, server.properties and logs/. An existing
RUN_DIR/NAME world is opened as is (this is how Horizon saves are checked); a
missing one is generated by vanilla (this is how reference worlds are made).
Exit status 1 when the log contains errors, upgrade/datafix work or chunk
load failures.
"""
import argparse
import os
import re
import subprocess
import sys
import threading
import time

try:
    from . import lab
except ImportError:
    import lab

SUSPICIOUS = re.compile(
    r'(ERROR|login failed|Exception|Couldn\'t load|Failed to|Skipping|Invalid|Unknown|corrupt|'
    r'Recreating|upgrad|DataFix|out of bounds|missing|Mismatch)', re.IGNORECASE)
# Benign lines the server prints on every start.
BENIGN = re.compile(r'(Unpacking |OFFLINE/INSECURE|online-mode|authenticate usernames|hackers|'
                    r'Failed to load properties|Failed to load eula)')


def main():
    p = argparse.ArgumentParser()
    p.add_argument('--jar', required=True)
    p.add_argument('--dir', required=True)
    p.add_argument('--world', default='world')
    p.add_argument('--seed', default='12345')
    p.add_argument('--port', type=int, default=0, help='default: a free port')
    p.add_argument('--command', action='append', default=[])
    p.add_argument('--wait-after', type=float, default=1.0)
    p.add_argument('--timeout', type=float, default=240)
    args = p.parse_args()
    port = args.port or lab.free_port()
    os.makedirs(args.dir, exist_ok=True)
    with open(os.path.join(args.dir, 'eula.txt'), 'w') as f:
        f.write('eula=true\n')
    props = {
        'server-port': port, 'online-mode': 'false', 'level-seed': args.seed,
        'level-name': args.world, 'spawn-protection': '0', 'view-distance': '4',
        'simulation-distance': '4', 'sync-chunk-writes': 'true', 'enable-query': 'false',
        'enable-rcon': 'false', 'management-server-enabled': 'false',
        'pause-when-empty-seconds': '-1', 'white-list': 'false', 'enforce-whitelist': 'false',
    }
    with open(os.path.join(args.dir, 'server.properties'), 'w') as f:
        f.writelines(f'{k}={v}\n' for k, v in props.items())
    proc = subprocess.Popen(
        ['java', '-Xmx2G', '-jar', os.path.abspath(args.jar), '--nogui'], cwd=args.dir,
        stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
    lines = []
    done = threading.Event()

    def pump():
        for line in proc.stdout:
            lines.append(line.rstrip('\n'))
            if 'Done (' in line:
                done.set()
        done.set()

    threading.Thread(target=pump, daemon=True).start()
    if not done.wait(args.timeout) or proc.poll() is not None:
        proc.kill()
        print('\n'.join(lines))
        print('server did not start', file=sys.stderr)
        return 1
    for command in args.command:
        if command.startswith('sleep '):
            time.sleep(float(command.split()[1]))
            continue
        if command.startswith('login '):
            sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
            import vanilla_login
            _, name, seconds = command.split()

            def run(name=name, seconds=float(seconds)):
                try:
                    vanilla_login.join('127.0.0.1', port, name, seconds)
                except Exception as e:  # reported in the log check below
                    lines.append(f'login failed: {e}')

            threading.Thread(target=run, daemon=True).start()
            continue
        proc.stdin.write(command + '\n')
        proc.stdin.flush()
        time.sleep(0.5)
    time.sleep(args.wait_after)
    proc.stdin.write('stop\n')
    proc.stdin.flush()
    try:
        proc.wait(args.timeout)
    except subprocess.TimeoutExpired:
        proc.kill()
    time.sleep(0.2)
    problems = [l for l in lines if SUSPICIOUS.search(l) and not BENIGN.search(l)]
    with open(os.path.join(args.dir, 'check.log'), 'w') as f:
        f.write('\n'.join(lines) + '\n')
    for l in problems:
        print(l)
    print(f'{len(lines)} log lines, {len(problems)} suspicious')
    return 1 if problems else 0


if __name__ == '__main__':
    sys.exit(main())