#!/usr/bin/env python

# Raspi-sump, a sump pump monitoring system.
# Al Audet
# https://www.linuxnorth.org/raspi-sump/
#
# All configuration changes should be done in raspisump.conf
# MIT License -- https://www.linuxnorth.org/raspi-sump/license.html

"""This script dumps the raspi-sump environment to
/home/__user__/raspi-sump/support/support_date_time.txt for troublshooting"""

import os
import time
import subprocess
from datetime import datetime
from raspisump import config_values


def get_os_version():
    with open("/etc/os-release", "r") as file:
        for line in file:
            if line.startswith("VERSION="):
                return line.strip().split("=")[1].strip('"')


def get_raspi_sump_version(actual_user):
    with open(f"/home/{actual_user}/raspi-sump/VERSION", "r") as file:
        return file.read().strip()


def get_python_version():
    return subprocess.check_output(["python3", "--version"], text=True).strip()


def run_command(command):
    try:
        # return subprocess.check_output(command, text=True).strip()
        return subprocess.check_output(
            command, text=True, stderr=subprocess.PIPE
        ).strip()
    except subprocess.CalledProcessError as e:
        return f"Error executing command: {e.cmd}. Return code: {e.returncode}. Output: {e.output}"
    except Exception as e:
        return f"An unexpected error occurred: {str(e)}"


def write_to_file(file_path, content):
    with open(file_path, "w") as file:
        file.write(content)


if __name__ == "__main__":
    # Determine the actual username
    actual_user = os.getlogin()

    # Paths
    home_folder = f"/home/{actual_user}"
    raspi_sump_folder = f"{home_folder}/raspi-sump"
    support_folder = f"{raspi_sump_folder}/support"
    support_file_path = f"{support_folder}/support-{time.strftime('%Y%m%d_%H%M%S')}.txt"
    logfile = f"{raspi_sump_folder}/csv/waterlevel-{time.strftime('%Y%m%d')}.csv"
    chart_folder = f"{raspi_sump_folder}/charts"
    systemd_folder = f"{home_folder}/.config/systemd/user"
    venv_folder = f"/opt/raspi-sump"

    # Create support folder if it doesn't exist
    os.makedirs(support_folder, exist_ok=True)

    # Get the required information
    current_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    os_version = get_os_version()
    raspi_sump_version = get_raspi_sump_version(actual_user)
    python_version = get_python_version()

    # Config values
    configs = config_values.configuration()

    # Run commands with exception handling
    try:
        raspi_sump_status = run_command(["systemctl", "--user", "status", "raspisump"])
    except Exception as e:
        raspi_sump_status = f"Error getting status: {str(e)}"

    try:
        rsumpwebchart_timer_status = run_command(
            ["systemctl", "--user", "status", "rsumpwebchart.timer"]
        )
    except Exception as e:
        rsumpwebchart_timer_status = f"Error getting status: {str(e)}"

    try:
        logs_error_log = run_command(["cat", f"{raspi_sump_folder}/logs/error_log"])
    except Exception as e:
        logs_error_log = f"Error reading logs/error_log: {str(e)}"

    try:
        logs_info_log = run_command(["cat", f"{raspi_sump_folder}/logs/info_log"])
    except Exception as e:
        logs_info_log = f"Error reading logs/info_log: {str(e)}"

    try:
        logs_heartbeat_log = run_command(
            ["tail", "-n", "10", f"{raspi_sump_folder}/logs/heartbeat_log"]
        )
    except Exception as e:
        logs_heartbeat_log = f"Error reading logs/heartbeat_log: {str(e)}"

    try:
        logs_alert_log = run_command(
            ["tail", "-n", "10", f"{raspi_sump_folder}/logs/alert_log"]
        )
    except Exception as e:
        logs_alert_log = f"Error reading logs/alert_log: {str(e)}"

    try:
        crontab = run_command(["crontab", "-l"])
    except Exception as e:
        crontab = f"Error reading crontab: {str(e)}"

    try:
        waterlevel_csv = run_command(["cat", f"{logfile}"])
    except Exception as e:
        waterlevel_csv = f"Error reading waterlevel_csv {str(e)}"

    try:
        journal = run_command(
            [
                "journalctl",
                "--user",
                "-u",
                "raspisump.service",
                "-u",
                "rsumpwebchart.service",
                "-u",
                "rsumwebchart.timer",
                "-b",
            ]
        )
    except Exception as e:
        journal = f"Error reading journal: {str(e)}"

    try:
        charts = run_command(["ls", "-alR", f"{chart_folder}"])
    except Exception as e:
        charts = f"Error reading journal: {str(e)}"

    try:
        rsump_service = run_command(["cat", f"{systemd_folder}/raspisump.service"])
    except Exception as e:
        rsump_service = f"Error reading raspisump.service: {str(e)}"

    try:
        rsumpwebchart_service = run_command(
            ["cat", f"{systemd_folder}/rsumpwebchart.service"]
        )
    except Exception as e:
        rsumpwebchart_service = f"Error reading rsumpwebchart.service: {str(e)}"

    try:
        rsumpwebchart_timer = run_command(
            ["cat", f"{systemd_folder}/rsumpwebchart.timer"]
        )
    except Exception as e:
        rsumpwebchart_timer = f"Error reading rsumpwebchart.timer: {str(e)}"

    try:
        list_systemd_files = run_command(["ls", "-alR", f"{systemd_folder}"])
    except Exception as e:
        list_systemd_files = f"Error displaying systemd folder: {str(e)}"

    try:
        list_venv_files = run_command(["ls", "-alR", f"{venv_folder}"])
    except Exception as e:
        list_venv_files = f"Error displaying {venv_folder} folder: {str(e)}"

    # Prepare the content for the support.txt file
    content = f"""\
Date file generated: {current_date}
os version: {os_version}
raspi-sump version: {raspi_sump_version}
python version on system: {python_version}

Systemctl status for raspisump:
{raspi_sump_status}

Systemctl status for rsumpwebchart.timer:
{rsumpwebchart_timer_status}

Raspi-Sump Variables:
critical_water_level: {configs["critical_water_level"]}
pit_depth: {configs["pit_depth"]}
reading_interval: {configs["reading_interval"]}
temperature: {configs["temperature"]}
trig_pin: {configs["trig_pin"]}
echo_pin: {configs["echo_pin"]}
unit: {configs["unit"]}
alert_type: {configs["alert_type"]}
smtp_tls: {configs["smtp_tls"]}
smtp_ssl: {configs["smtp_ssl"]}
alert_when: {configs["alert_when"]}
alert_interval: {configs["alert_interval"]}
heartbeat: {configs["heartbeat"]}
heartbeat_interval: {configs["heartbeat_interval"]}
line_color: {configs["line_color"]}

User Crontab:
{crontab}

Logs/error_log content:
{logs_error_log}

Logs/info_log content:
{logs_info_log}

Last 10 lines of logs/heartbeat.log:
{logs_heartbeat_log}

Last 10 lines of logs/alert.log:
{logs_alert_log}

waterlevel csv:
{waterlevel_csv}

journalctl --user -u raspisump.service -u rsumpwebchart.service -u rsumwebchart.timer -b:
{journal}

Charts directory:
{charts}

Systemd files:
** Raspi-sump service file:
{rsump_service}
*************************************

** Raspi-sump webchart service file:
{rsumpwebchart_service}
*************************************

** Raspi-sump webchart timer file:
{rsumpwebchart_timer}
*************************************

List of files in systemd folder:
{list_systemd_files}

List of files in venv folder:
{list_venv_files}


"""

    # Write the content to the support.txt file
    write_to_file(support_file_path, content)

    print(
        f"Support information has been saved to /home/{actual_user}/raspi-sump/support/support_date_time.txt."
    )
