import os
import dotenv
import logging
import yaml
import sys
import atexit
import logging.config

import config.exit_codes as ec
from visuals.dash_app import app


def set_up_logging():
    try:
        with open("config/logging_config.yaml", "r") as f:
            log_config = yaml.safe_load(f)
        logging.config.dictConfig(log_config)
    except yaml.YAMLError as e:
        print(f"Error parsing logging configuration file: {e}")
        log_config = {}  # Initialize log_config to an empty dictionary
        # Fallback to a basic configuration
        logging.basicConfig(level=logging.INFO)
    queue_handler = logging.getHandlerByName("queue_handler")
    if queue_handler is not None:
        queue_handler.listener.start()
        atexit.register(queue_handler.listener.stop)


def load_config():
    try:
        with open("config/config.yaml", "r") as file:
            config = yaml.safe_load(file)
        return config
    except FileNotFoundError:
        logging.error("config.yaml file not found")
        sys.exit(ec.MISSING_CONFIG_FILE)
    except yaml.YAMLError:
        # Attempt to print a more informative error message
        try:
            parsed_yaml = yaml.safe_load_all(file)
            for key, value in parsed_yaml:
                print(f"Error parsing key '{key}': {value}")
        except Exception as e:
            logging.error(f"Error loading config.yaml: {e}")
        sys.exit(ec.CORRUPTED_CONFIG_FILE)
    except Exception as e:
        logging.error(f"some other problem {e}")


logger = logging.getLogger("data_pipeline_for_ynab")
os.makedirs("logs", exist_ok=True)
set_up_logging()

# Load environment variables
dotenv.load_dotenv()

API_TOKEN = os.getenv("API_TOKEN")
BUDGET_ID = os.getenv("BUDGET_ID")

if not API_TOKEN or not BUDGET_ID:
    logging.error("API_TOKEN or BUDGET_ID is not set in .env file")
    sys.exit(ec.MISSING_ENV_VARS)

if __name__ == "__main__":
    config = load_config()
    config["API_TOKEN"] = API_TOKEN
    config["BUDGET_ID"] = BUDGET_ID
    try:
        # pipeline_main(config)

        # Check if the data was successfully created
        data_exists = os.path.exists("data/processed") and os.listdir("data/processed")
        if data_exists:
            app.run(debug=True)
        else:
            logging.error(
                "Data pipeline did not produce any data. Dash app will not run."
            )
            sys.exit(ec.NO_DATA_PRODUCED)
    except SystemExit as e:
        exit_code = e.code
        if exit_code == ec.SUCCESS:
            logging.info("Program exited successfully")
        else:
            logging.error(f"Program exited with code {exit_code}")
        raise