import jingrow
from jingrow.utils import get_datetime

from jcloude.agent import Agent
from jcloude.api.server import all as get_all_servers


@jingrow.whitelist()
def get_servers(server_filter):
	servers = get_all_servers(server_filter=server_filter)

	for server in servers:
		security_updates_count = jingrow.db.count("Security Update", {"server": server.name})
		server["security_updates_status"] = "Up to date"

		if security_updates_count != 0:
			server[
				"security_updates_status"
			] = f"{security_updates_count} security update(s) available"

	return servers


@jingrow.whitelist()
def fetch_security_updates(
	filters=None, order_by=None, limit_start=None, limit_page_length=None
):
	return jingrow.get_all(
		"Security Update",
		filters=filters,
		fields=["name", "package", "version", "priority", "priority_level", "datetime"],
		order_by=order_by or "priority_level asc",
		start=limit_start,
		limit=limit_page_length,
	)


@jingrow.whitelist()
def get_security_update_details(update_id):
	return jingrow.get_pg("Security Update", update_id).as_dict()


@jingrow.whitelist()
def fetch_ssh_sessions(server, start=0, limit=10):
	return jingrow.get_all(
		"SSH Session",
		filters={"server": server},
		fields=["name", "user", "datetime"],
		order_by="datetime desc",
		start=start,
		limit=limit,
	)


@jingrow.whitelist()
def fetch_ssh_session_logs(server):
	logs_to_display = []
	ssh_logs = Agent(server=server).get("security/ssh_session_logs")

	for log in ssh_logs.get("logs", []):
		if not log["name"].endswith(".timing"):
			log["created_at"] = get_datetime(log["created"]).strftime("%Y-%m-%d %H-%M")

			splited_log = log["name"].split(".")
			log["user"] = splited_log[1]
			log["session_id"] = splited_log[2]

			logs_to_display.append(log)

	return logs_to_display


@jingrow.whitelist()
def fetch_ssh_session_activity(server, filename):
	content = Agent(server=server).get(f"security/retrieve_ssh_session_log/{filename}")
	splited_filename = filename.split(".")
	session_user = splited_filename[1]
	session_id = splited_filename[2]

	return {
		"session_user": session_user,
		"session_id": session_id,
		"content": content.get("log_details", "Not Found"),
	}