#!/usr/local/bin/python3


import time
import os
import subprocess
import shlex
import re
from tqdm import tqdm

mode = "all"  # download "all" files or only "latest"
autoreboot = True  # reboot between each download, to avoid memory issues

print("\n\nRunning gc.collect()...")
mem_cmd = './tools/wasptool --verbose --eval \'wasp.gc.collect()\''
subprocess.check_output(shlex.split(mem_cmd))

print("\n\nListing remote files...")
ls_cmd = './tools/wasptool --verbose --eval \'from shell import ls ; ls(\"/flash/logs/sleep/\")\''
out = subprocess.check_output(shlex.split(ls_cmd)).decode()
files = re.findall(r"\d*\.csv", out)
print(f"Found files {', '.join(files)}")

reset_cmd = './tools/wasptool --verbose --reset'


if mode == "latest":
    to_dl = files[-1]
elif mode == "all":
    to_dl = files
else:
    raise Exception("Wrong value for 'mode'")

print("\n\n")
for fi in tqdm(to_dl):
    if os.path.exists(f"./logs/sleep/{fi}"):
        tqdm.write(f"Skipping file {fi}: already exists")
    else:
        tqdm.write(f"Downloading file '{fi}'")
        pull_cmd = f'./tools/wasptool --verbose --pull logs/sleep/{fi}'
        try:
            out = subprocess.check_output(shlex.split(pull_cmd))
            if b"Watch reported error" in out:
                raise Exception("Watch reported error")
            tqdm.write(f"Succesfully downloaded to './logs/sleep/{fi}'")
        except Exception as e:
            tqdm.write(f"Error happened while downloading {fi}, deleting local incomplete file")
            os.system(f"rm ./logs/sleep/{fi}")
        if mode == "all" and autoreboot:
            tqdm.write("Restarting watch.")
            out = subprocess.check_output(shlex.split(reset_cmd))
            time.sleep(10)

    print("\n\n")