from jinja2 import Environment, FileSystemLoader
import subprocess
from datetime import datetime, timedelta
import os
import fnmatch

amount_of_logs = len(fnmatch.filter(os.listdir("./logs"), '*.html'))

now = datetime.today()
next_report = now + timedelta(hours=3)

def run_command(command, based = False):
    result = subprocess.run(command, capture_output=True, input=based)
    result.check_returncode()
    return result.stdout.decode('UTF-8').strip()
    
last_user_added = subprocess.run(["sudo", "journalctl","_COMM=useradd","-r","-n", "1" , "--output-fields=MESSAGE"], capture_output=True)
last_user_added_name = run_command(['grep', '-Po', "(?<=name)\W*\K[^ ]*"], based=last_user_added.stdout)

users_created_today = run_command(["sudo", "journalctl", "-S","today","_COMM=useradd","-r","-n","1","--output-fields=MESSAGE"])


since_last_boot = run_command(["uptime","-s"])
slb_date = datetime.strptime(since_last_boot, "%Y-%m-%d %H:%M:%S")
time_since_last_boot = now - slb_date

list_package_installs = subprocess.run(["grep", 'install', "/var/log/dpkg.log"], capture_output=True).stdout.decode('UTF-8')
list_package_upgrade = subprocess.run(["grep", 'upgrade', "/var/log/dpkg.log"], capture_output=True).stdout.decode('UTF-8')
list_package_remove = subprocess.run(["grep", 'remove', "/var/log/dpkg.log"], capture_output=True).stdout.decode('UTF-8')

device_info_grab = subprocess.run(["grep", 'Model' , "/proc/cpuinfo"], check=True, capture_output=True)
device_info = run_command(['awk', '-F:', '{ print $2}'], based=device_info_grab.stdout)

ip_address_show = subprocess.run(["ip", "addr", "show", "end0"], check=True, capture_output=True)
ip_address = run_command(['awk', '$1 == "inet" {gsub(/\/.*$/, "", $2); print $2}' ], based=ip_address_show.stdout)
kitchen_services = run_command(["sudo", "journalctl", "-S", "today", "-u", "kitchen-stove.service", "-u", "kitchen-bin.service", "-u", "kitchen-fridge.service", "-r", "-n"])

# loading the environment
env = Environment(loader=FileSystemLoader("templates"))

# # loading the template
template = env.get_template("template.jinja")

# rendering the template and storing the resultant text in variable output
output = template.render(
    now = now,
    next_report = next_report,
    amount_of_logs = amount_of_logs,
    last_user_added=last_user_added.stdout.decode('UTF-8'),
    last_user_added_name=last_user_added_name,
    users_created_today=users_created_today,
    since_last_boot=since_last_boot,
    list_package_installs=list_package_installs,
    list_package_upgrade=list_package_upgrade,
    list_package_remove=list_package_remove,
    device_info=device_info,
    days_since_last_boot = time_since_last_boot.days,
    kitchen_services = kitchen_services,
    ip_address = ip_address,
    list_groups = run_command(["getent","group"]),
    list_active_services = run_command(["sudo", "service", "--status-all"]),
    debian_version = run_command(["cat", "/etc/debian_version"]),
    kernel_version = run_command(["uname","-a"]),
    hostname = run_command(["hostname","-i"])
)

log_file_name = "logs/log_" + now.strftime("%Y-%m-%d_%H:%M:%S") + ".html"

with open("index.html", "w") as f:
    print(output, file=f)

with open(log_file_name, "w") as f:
    print(output, file=f)