85 lines
2.1 KiB
Python
85 lines
2.1 KiB
Python
import jingrow
|
|
from frappe.utils import get_datetime
|
|
|
|
from press.agent import Agent
|
|
from press.api.server import all as get_all_servers
|
|
|
|
|
|
@frappe.whitelist()
|
|
def get_servers(server_filter):
|
|
servers = get_all_servers(server_filter=server_filter)
|
|
|
|
for server in servers:
|
|
security_updates_count = frappe.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
|
|
|
|
|
|
@frappe.whitelist()
|
|
def fetch_security_updates(
|
|
filters=None, order_by=None, limit_start=None, limit_page_length=None
|
|
):
|
|
return frappe.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,
|
|
)
|
|
|
|
|
|
@frappe.whitelist()
|
|
def get_security_update_details(update_id):
|
|
return frappe.get_pg("Security Update", update_id).as_dict()
|
|
|
|
|
|
@frappe.whitelist()
|
|
def fetch_ssh_sessions(server, start=0, limit=10):
|
|
return frappe.get_all(
|
|
"SSH Session",
|
|
filters={"server": server},
|
|
fields=["name", "user", "datetime"],
|
|
order_by="datetime desc",
|
|
start=start,
|
|
limit=limit,
|
|
)
|
|
|
|
|
|
@frappe.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
|
|
|
|
|
|
@frappe.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"),
|
|
}
|