import json
from desktop_env.envs.desktop_env import DesktopEnv

import logging
import os
import sys
import datetime

#  Logger Configs {{{ # 
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

datetime_str: str = datetime.datetime.now().strftime("%Y%m%d@%H%M%S")

file_handler = logging.FileHandler(os.path.join("logs", "normal-{:}.log".format(datetime_str)))
debug_handler = logging.FileHandler(os.path.join("logs", "debug-{:}.log".format(datetime_str)))
stdout_handler = logging.StreamHandler(sys.stdout)
sdebug_handler = logging.FileHandler(os.path.join("logs", "sdebug-{:}.log".format(datetime_str)))

file_handler.setLevel(logging.INFO)
debug_handler.setLevel(logging.DEBUG)
stdout_handler.setLevel(logging.INFO)
sdebug_handler.setLevel(logging.DEBUG)

formatter = logging.Formatter(fmt="\x1b[1;33m[%(asctime)s \x1b[31m%(levelname)s \x1b[32m%(module)s/%(lineno)d-%(processName)s\x1b[1;33m] \x1b[0m%(message)s")
file_handler.setFormatter(formatter)
debug_handler.setFormatter(formatter)
stdout_handler.setFormatter(formatter)
sdebug_handler.setFormatter(formatter)

stdout_handler.addFilter(logging.Filter("desktopenv"))
sdebug_handler.addFilter(logging.Filter("desktopenv"))

logger.addHandler(file_handler)
logger.addHandler(debug_handler)
logger.addHandler(stdout_handler)
logger.addHandler(sdebug_handler)
#  }}} Logger Configs # 

logger = logging.getLogger("desktopenv.main")

def human_agent():
    """
    Runs the Gym environment with human input.
    """

    with open("evaluation_examples/examples/thunderbird/e1e75309-3ddb-4d09-92ec-de869c928143.json.nosetup", "r") as f:
        example = json.load(f)
    example["snapshot"] = "Snapshot 18"

    env = DesktopEnv( path_to_vm="../../../../大文件/镜像/Ubuntu-1218/Ubuntu/Ubuntu.vmx"
                    , action_space="computer_13"
                    , task_config=example
                    )
    # reset the environment to certain snapshot
    observation = env.reset()
    done = False

    trajectory = [
        {
            "action_type": "MOVE_TO",
            "parameters": {
                "x": 754,
                "y": 1057
            }
        },
        {"action_type": "CLICK", "parameters": {"button": "right", "num_clicks": 1}}
    ]

    for i in range(len(trajectory)):
        # action = get_human_action()

        # action = {
        #     "action_type": 0,
        #     "click_type": 3,
        # }
        logger.info(trajectory[i])

        observation, reward, done, info = env.step(trajectory[i], pause=5)
        logger.info("Observation: %s", observation)
        logger.info("Reward: %.2f", reward)
        logger.info("Info: %s", info)

        logger.info("================================\n")

        if done:
            logger.info("The episode is done.")
            break

    result = env.evaluate()
    logger.info("Result: %.2f", result)

    #input("PAUSING")

    #env.close()
    logger.info("Environment closed.")


if __name__ == "__main__":
    human_agent()