# coding=utf8
from __future__ import unicode_literals, absolute_import, division, print_function
"""
This is the SpiceBot Logs system.

This Class stores logs in an easy to access manner
"""

import threading
import os
import sys
import spicemanip

from .Config import config as botconfig


# TODO timestamps

class BotLogs():
    """This Is a contained source of log information"""
    def __init__(self):
        self.setup_logs()
        self.lock = threading.Lock()
        self.dict = {
                    "list": {"Sopel_systemd": [], "Sopel_stdio": []},
                    "queue": []
                    }

    def setup_logs(self):
        botconfig.core.logs_stdio = os.path.join(botconfig.core.logdir, 'stdio.log')
        # botconfig.core.logs_stdio = os.path.os.path.join(botconfig.core.logdir, botconfig.basename + '.stdio.log')
        botconfig.core.logs_exceptions = os.path.join(botconfig.core.logdir, 'exceptions.log')
        # botconfig.core.logs_exceptions = os.path.os.path.join(botconfig.core.logdir, botconfig.basename + '.exceptions.log')
        botconfig.core.logs_raw = os.path.join(botconfig.core.logdir, 'raw.log')
        # botconfig.core.logs_raw = os.path.os.path.join(botconfig.core.logdir, botconfig.basename + '.raw.log')

    def botstderr(self, logmessage):
        sys.stderr.write(logmessage)

    def log(self, logtype, logentry, stdio=False):

        self.lock.acquire()

        logtitle = "[" + logtype + "]"
        logmessage = logtitle + "    " + logentry

        if botconfig.logging_channel:
            self.dict["queue"].append(logmessage)

        if stdio:
            self.botstderr(logmessage + "\n")

        if logtype not in list(self.dict["list"].keys()):
            self.dict["list"][logtype] = []
        self.dict["list"][logtype].append(logentry)

        self.lock.release()

    def get_logs(self, logtype):
        if logtype == "Sopel_systemd":
            logindex = self.systemd_logs_fetch()
        elif logtype == "Sopel_stdio":
            logindex = self.stdio_logs_fetch()
        else:
            logindex = self.dict["list"][logtype]
        return logindex

    def stdio_logs_fetch(self):

        stdio_ignore = []
        for logtype in list(self.dict["list"].keys()):
            stdio_ignore.append("[" + logtype + "]")

        try:
            log_file_lines = []
            log_file = open(botconfig.logs_stdio, 'r')
            lines = log_file.readlines()
            for line in lines:
                log_file_lines.append(line)
            log_file.close()

            currentline, linenumindex = 0, []
            for line in log_file_lines:
                if line.startswith("Welcome to Sopel. Loading modules..."):
                    linenumindex.append(currentline)
                currentline += 1
            last_start = max(linenumindex)
            filelines = log_file_lines[last_start:]
        except Exception as e:
            debuglines = e
            filelines = []

        debuglines = []
        loadedmodules = []
        for line in filelines:
            if line.startswith("Loaded:"):
                loadedmodules.append(str(line).split("Loaded:")[-1])
            else:
                if not line.startswith(tuple(stdio_ignore)) and not line.isspace():
                    debuglines.append(str(line))
        loadedmodules = "Loaded: " + spicemanip.main(loadedmodules, 'andlist')
        debuglines.insert(1, loadedmodules)

        return debuglines

    def systemd_logs_fetch(self):
        servicepid = self.get_running_pid()
        debuglines = []
        for line in os.popen(str("sudo journalctl _PID=" + str(servicepid))).read().split('\n'):
            line = str(line).split(str(os.uname()[1] + " "))[-1]
            lineparts = str(line).split(": ")
            del lineparts[0]
            line = spicemanip.main(lineparts, 0)
            if not line.isspace():
                debuglines.append(str(line))
        return debuglines

    def get_running_pid(self):
        try:
            filename = "/run/sopel/sopel-" + str(botconfig.nick) + ".pid"
            with open(filename, 'r') as pid_file:
                pidnum = int(pid_file.read())
        except Exception as e:
            pidnum = e
            pidnum = str(os.popen("systemctl show " + str(botconfig.nick) + " --property=MainPID").read()).split("=")[-1]
        return pidnum


logs = BotLogs()