@cryptotaxi247 / CoPilot / commits / 16451238

mkdocs sites

Taylor committed Jul 11, 2023 at 12:42 UTC 164512381e6ce5c2af2e3263ffccb23ec42b851e

7 files changed +2024 -1950

backend/site/agents/index.html

+659 -627

@@ -1346,11 +1346,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1346

1347 <details class="quote">

1348 <summary>Source code in <code>app\routes\agents.py</code></summary>

1349 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-77">77</a>

1350 -<a href="#__codelineno-0-78">78</a>

1351 -<a href="#__codelineno-0-79">79</a>

1352 -<a href="#__codelineno-0-80">80</a>

1353 -<a href="#__codelineno-0-81">81</a>

1349 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-81">81</a>

1350 <a href="#__codelineno-0-82">82</a>

1351 <a href="#__codelineno-0-83">83</a>

1352 <a href="#__codelineno-0-84">84</a>

@@ -1362,23 +1358,27 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1358 <a href="#__codelineno-0-90">90</a>

1359 <a href="#__codelineno-0-91">91</a>

1360 <a href="#__codelineno-0-92">92</a>

1365 -<a href="#__codelineno-0-93">93</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-77" name="__codelineno-0-77"></a>@bp.route("/agents/<agent_id>/delete", methods=["POST"])

1366 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a>def delete_agent(agent_id: str) -> Any:

1367 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> """

1368 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> Endpoint to delete an agent.

1369 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> Args:

1370 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> agent_id (str): The ID of the agent to be deleted.

1371 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> Returns:

1372 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> json: A JSON response indicating whether the deletion was successful.

1373 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a> """

1374 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> service = AgentService()

1375 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> result = service.delete_agent_db(agent_id=agent_id)

1376 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a>

1377 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> universal_service = UniversalService()

1378 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> agent_service = WazuhManagerAgentService(universal_service)

1379 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> agent_service.delete_agent(agent_id=agent_id)

1361 +<a href="#__codelineno-0-93">93</a>

1362 +<a href="#__codelineno-0-94">94</a>

1363 +<a href="#__codelineno-0-95">95</a>

1364 +<a href="#__codelineno-0-96">96</a>

1365 +<a href="#__codelineno-0-97">97</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-81" name="__codelineno-0-81"></a>@bp.route("/agents/<agent_id>/delete", methods=["POST"])

1366 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a>def delete_agent(agent_id: str) -> Any:

1367 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a> """

1368 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Endpoint to delete an agent.

1369 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> Args:

1370 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> agent_id (str): The ID of the agent to be deleted.

1371 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> Returns:

1372 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> json: A JSON response indicating whether the deletion was successful.

1373 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> """

1374 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> service = AgentService()

1375 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> result = service.delete_agent_db(agent_id=agent_id)

1376 <a id="__codelineno-0-92" name="__codelineno-0-92"></a>

1381 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> return result

1377 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> universal_service = UniversalService()

1378 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> agent_service = WazuhManagerAgentService(universal_service)

1379 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> agent_service.delete_agent(agent_id=agent_id)

1380 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a>

1381 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> return result

1382 </code></pre></div></td></tr></table></div>

1383 </details>

1384 </div>

@@ -1451,29 +1451,29 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1451

1452 <details class="quote">

1453 <summary>Source code in <code>app\routes\agents.py</code></summary>

1454 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-23">23</a>

1455 -<a href="#__codelineno-0-24">24</a>

1456 -<a href="#__codelineno-0-25">25</a>

1457 -<a href="#__codelineno-0-26">26</a>

1458 -<a href="#__codelineno-0-27">27</a>

1454 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-27">27</a>

1455 <a href="#__codelineno-0-28">28</a>

1456 <a href="#__codelineno-0-29">29</a>

1457 <a href="#__codelineno-0-30">30</a>

1458 <a href="#__codelineno-0-31">31</a>

1459 <a href="#__codelineno-0-32">32</a>

1460 <a href="#__codelineno-0-33">33</a>

1465 -<a href="#__codelineno-0-34">34</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-23" name="__codelineno-0-23"></a>@bp.route("/agents/<agent_id>", methods=["GET"])

1466 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a>def get_agent(agent_id: str) -> Any:

1467 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> """

1468 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> Endpoint to get the details of a specific agent.

1469 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> Args:

1470 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> agent_id (str): The ID of the agent to retrieve.

1471 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> Returns:

1472 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> json: A JSON response containing the details of the agent.

1473 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> """

1474 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> service = AgentService()

1475 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> agent = service.get_agent(agent_id=agent_id)

1476 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> return agent

1461 +<a href="#__codelineno-0-34">34</a>

1462 +<a href="#__codelineno-0-35">35</a>

1463 +<a href="#__codelineno-0-36">36</a>

1464 +<a href="#__codelineno-0-37">37</a>

1465 +<a href="#__codelineno-0-38">38</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-27" name="__codelineno-0-27"></a>@bp.route("/agents/<agent_id>", methods=["GET"])

1466 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a>def get_agent(agent_id: str) -> Any:

1467 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> """

1468 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> Endpoint to get the details of a specific agent.

1469 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> Args:

1470 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> agent_id (str): The ID of the agent to retrieve.

1471 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> Returns:

1472 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> json: A JSON response containing the details of the agent.

1473 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> """

1474 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> service = AgentService()

1475 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> agent = service.get_agent(agent_id=agent_id)

1476 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> return agent

1477 </code></pre></div></td></tr></table></div>

1478 </details>

1479 </div>

@@ -1546,11 +1546,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1546

1547 <details class="quote">

1548 <summary>Source code in <code>app\routes\agents.py</code></summary>

1549 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-96"> 96</a>

1550 -<a href="#__codelineno-0-97"> 97</a>

1551 -<a href="#__codelineno-0-98"> 98</a>

1552 -<a href="#__codelineno-0-99"> 99</a>

1553 -<a href="#__codelineno-0-100">100</a>

1549 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-100">100</a>

1550 <a href="#__codelineno-0-101">101</a>

1551 <a href="#__codelineno-0-102">102</a>

1552 <a href="#__codelineno-0-103">103</a>

@@ -1559,20 +1555,28 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1555 <a href="#__codelineno-0-106">106</a>

1556 <a href="#__codelineno-0-107">107</a>

1557 <a href="#__codelineno-0-108">108</a>

1562 -<a href="#__codelineno-0-109">109</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-96" name="__codelineno-0-96"></a>@bp.route("/agents/<agent_id>/vulnerabilities", methods=["GET"])

1563 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a>def get_agent_vulnerabilities(agent_id: str) -> Any:

1564 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> """

1565 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> Endpoint to get the vulnerabilities of a specific agent.

1566 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a> Args:

1567 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a> agent_id (str): The ID of the agent whose vulnerabilities are to be fetched.

1568 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> Returns:

1569 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a> json: A JSON response containing the vulnerabilities of the agent.

1570 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a> """

1571 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a> universal_service = UniversalService()

1572 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a> vulnerability_service = VulnerabilityService(universal_service)

1573 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a>

1574 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> agent_vulnerabilities = vulnerability_service.agent_vulnerabilities(agent_id=agent_id)

1575 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> return agent_vulnerabilities

1558 +<a href="#__codelineno-0-109">109</a>

1559 +<a href="#__codelineno-0-110">110</a>

1560 +<a href="#__codelineno-0-111">111</a>

1561 +<a href="#__codelineno-0-112">112</a>

1562 +<a href="#__codelineno-0-113">113</a>

1563 +<a href="#__codelineno-0-114">114</a>

1564 +<a href="#__codelineno-0-115">115</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-100" name="__codelineno-0-100"></a>@bp.route("/agents/<agent_id>/vulnerabilities", methods=["GET"])

1565 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a>def get_agent_vulnerabilities(agent_id: str) -> Any:

1566 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> """

1567 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> Endpoint to get the vulnerabilities of a specific agent.

1568 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Args:

1569 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a> agent_id (str): The ID of the agent whose vulnerabilities are to be fetched.

1570 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> Returns:

1571 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a> json: A JSON response containing the vulnerabilities of the agent.

1572 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a> """

1573 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a> universal_service = UniversalService()

1574 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> vulnerability_service = VulnerabilityService(universal_service)

1575 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a>

1576 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a> agent_vulnerabilities = vulnerability_service.agent_vulnerabilities(

1577 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> agent_id=agent_id,

1578 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> )

1579 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> return agent_vulnerabilities

1580 </code></pre></div></td></tr></table></div>

1581 </details>

1582 </div>

@@ -1617,25 +1621,25 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1621

1622 <details class="quote">

1623 <summary>Source code in <code>app\routes\agents.py</code></summary>

1620 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-11">11</a>

1621 -<a href="#__codelineno-0-12">12</a>

1622 -<a href="#__codelineno-0-13">13</a>

1623 -<a href="#__codelineno-0-14">14</a>

1624 -<a href="#__codelineno-0-15">15</a>

1624 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-15">15</a>

1625 <a href="#__codelineno-0-16">16</a>

1626 <a href="#__codelineno-0-17">17</a>

1627 <a href="#__codelineno-0-18">18</a>

1628 <a href="#__codelineno-0-19">19</a>

1629 -<a href="#__codelineno-0-20">20</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-11" name="__codelineno-0-11"></a>@bp.route("/agents", methods=["GET"])

1630 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a>def get_agents() -> Any:

1631 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a> """

1632 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a> Endpoint to get a list of all agents. It processes each agent and returns the results.

1633 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a> Returns:

1634 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> json: A JSON response containing the list of all available agents along with their connection verification status.

1635 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> """

1636 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> service = AgentService()

1637 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> agents = service.get_all_agents()

1638 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> return agents

1629 +<a href="#__codelineno-0-20">20</a>

1630 +<a href="#__codelineno-0-21">21</a>

1631 +<a href="#__codelineno-0-22">22</a>

1632 +<a href="#__codelineno-0-23">23</a>

1633 +<a href="#__codelineno-0-24">24</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-15" name="__codelineno-0-15"></a>@bp.route("/agents", methods=["GET"])

1634 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a>def get_agents() -> Any:

1635 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a> """

1636 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> Endpoint to get a list of all agents. It processes each agent and returns the results.

1637 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a> Returns:

1638 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> json: A JSON response containing the list of all available agents along with their connection verification status.

1639 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> """

1640 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> service = AgentService()

1641 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> agents = service.get_all_agents()

1642 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a> return agents

1643 </code></pre></div></td></tr></table></div>

1644 </details>

1645 </div>

@@ -1708,29 +1712,29 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1712

1713 <details class="quote">

1714 <summary>Source code in <code>app\routes\agents.py</code></summary>

1711 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-37">37</a>

1712 -<a href="#__codelineno-0-38">38</a>

1713 -<a href="#__codelineno-0-39">39</a>

1714 -<a href="#__codelineno-0-40">40</a>

1715 -<a href="#__codelineno-0-41">41</a>

1715 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-41">41</a>

1716 <a href="#__codelineno-0-42">42</a>

1717 <a href="#__codelineno-0-43">43</a>

1718 <a href="#__codelineno-0-44">44</a>

1719 <a href="#__codelineno-0-45">45</a>

1720 <a href="#__codelineno-0-46">46</a>

1721 <a href="#__codelineno-0-47">47</a>

1722 -<a href="#__codelineno-0-48">48</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-37" name="__codelineno-0-37"></a>@bp.route("/agents/<agent_id>/critical", methods=["POST"])

1723 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a>def mark_as_critical(agent_id: str) -> Any:

1724 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> """

1725 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> Endpoint to mark an agent as critical.

1726 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> Args:

1727 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> agent_id (str): The ID of the agent to mark as critical.

1728 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> Returns:

1729 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> json: A JSON response containing the updated agent information after being marked as critical.

1730 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> """

1731 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> service = AgentService()

1732 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> result = service.mark_agent_as_critical(agent_id=agent_id)

1733 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> return result

1722 +<a href="#__codelineno-0-48">48</a>

1723 +<a href="#__codelineno-0-49">49</a>

1724 +<a href="#__codelineno-0-50">50</a>

1725 +<a href="#__codelineno-0-51">51</a>

1726 +<a href="#__codelineno-0-52">52</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-41" name="__codelineno-0-41"></a>@bp.route("/agents/<agent_id>/critical", methods=["POST"])

1727 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a>def mark_as_critical(agent_id: str) -> Any:

1728 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> """

1729 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> Endpoint to mark an agent as critical.

1730 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> Args:

1731 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> agent_id (str): The ID of the agent to mark as critical.

1732 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> Returns:

1733 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> json: A JSON response containing the updated agent information after being marked as critical.

1734 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> """

1735 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> service = AgentService()

1736 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> result = service.mark_agent_as_critical(agent_id=agent_id)

1737 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> return result

1738 </code></pre></div></td></tr></table></div>

1739 </details>

1740 </div>

@@ -1775,25 +1779,25 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1779

1780 <details class="quote">

1781 <summary>Source code in <code>app\routes\agents.py</code></summary>

1778 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-65">65</a>

1779 -<a href="#__codelineno-0-66">66</a>

1780 -<a href="#__codelineno-0-67">67</a>

1781 -<a href="#__codelineno-0-68">68</a>

1782 -<a href="#__codelineno-0-69">69</a>

1782 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-69">69</a>

1783 <a href="#__codelineno-0-70">70</a>

1784 <a href="#__codelineno-0-71">71</a>

1785 <a href="#__codelineno-0-72">72</a>

1786 <a href="#__codelineno-0-73">73</a>

1787 -<a href="#__codelineno-0-74">74</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-65" name="__codelineno-0-65"></a>@bp.route("/agents/sync", methods=["POST"])

1788 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a>def sync_agents() -> Any:

1789 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> """

1790 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> Endpoint to synchronize all agents.

1791 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> Returns:

1792 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> json: A JSON response containing the updated information of all synchronized agents.

1793 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> """

1794 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> service = AgentSyncService()

1795 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> result = service.sync_agents()

1796 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> return jsonify(result)

1787 +<a href="#__codelineno-0-74">74</a>

1788 +<a href="#__codelineno-0-75">75</a>

1789 +<a href="#__codelineno-0-76">76</a>

1790 +<a href="#__codelineno-0-77">77</a>

1791 +<a href="#__codelineno-0-78">78</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-69" name="__codelineno-0-69"></a>@bp.route("/agents/sync", methods=["POST"])

1792 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a>def sync_agents() -> Any:

1793 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> """

1794 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> Endpoint to synchronize all agents.

1795 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> Returns:

1796 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> json: A JSON response containing the updated information of all synchronized agents.

1797 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> """

1798 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> service = AgentSyncService()

1799 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> result = service.sync_agents()

1800 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> return jsonify(result)

1801 </code></pre></div></td></tr></table></div>

1802 </details>

1803 </div>

@@ -1866,29 +1870,29 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1870

1871 <details class="quote">

1872 <summary>Source code in <code>app\routes\agents.py</code></summary>

1869 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-51">51</a>

1870 -<a href="#__codelineno-0-52">52</a>

1871 -<a href="#__codelineno-0-53">53</a>

1872 -<a href="#__codelineno-0-54">54</a>

1873 -<a href="#__codelineno-0-55">55</a>

1873 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-55">55</a>

1874 <a href="#__codelineno-0-56">56</a>

1875 <a href="#__codelineno-0-57">57</a>

1876 <a href="#__codelineno-0-58">58</a>

1877 <a href="#__codelineno-0-59">59</a>

1878 <a href="#__codelineno-0-60">60</a>

1879 <a href="#__codelineno-0-61">61</a>

1880 -<a href="#__codelineno-0-62">62</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-51" name="__codelineno-0-51"></a>@bp.route("/agents/<agent_id>/noncritical", methods=["POST"])

1881 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a>def unmark_agent_critical(agent_id: str) -> Any:

1882 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> """

1883 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> Endpoint to unmark an agent as critical.

1884 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> Args:

1885 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> agent_id (str): The ID of the agent to unmark as critical.

1886 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> Returns:

1887 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> json: A JSON response containing the updated agent information after being unmarked as critical.

1888 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> """

1889 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a> service = AgentService()

1890 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> result = service.mark_agent_as_non_critical(agent_id=agent_id)

1891 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> return result

1880 +<a href="#__codelineno-0-62">62</a>

1881 +<a href="#__codelineno-0-63">63</a>

1882 +<a href="#__codelineno-0-64">64</a>

1883 +<a href="#__codelineno-0-65">65</a>

1884 +<a href="#__codelineno-0-66">66</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-55" name="__codelineno-0-55"></a>@bp.route("/agents/<agent_id>/noncritical", methods=["POST"])

1885 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a>def unmark_agent_critical(agent_id: str) -> Any:

1886 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a> """

1887 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a> Endpoint to unmark an agent as critical.

1888 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> Args:

1889 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> agent_id (str): The ID of the agent to unmark as critical.

1890 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> Returns:

1891 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> json: A JSON response containing the updated agent information after being unmarked as critical.

1892 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> """

1893 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> service = AgentService()

1894 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> result = service.mark_agent_as_non_critical(agent_id=agent_id)

1895 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> return result

1896 </code></pre></div></td></tr></table></div>

1897 </details>

1898 </div>

@@ -1941,11 +1945,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

1945

1946 <details class="quote">

1947 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

1944 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-17"> 17</a>

1945 -<a href="#__codelineno-0-18"> 18</a>

1946 -<a href="#__codelineno-0-19"> 19</a>

1947 -<a href="#__codelineno-0-20"> 20</a>

1948 -<a href="#__codelineno-0-21"> 21</a>

1948 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-21"> 21</a>

1949 <a href="#__codelineno-0-22"> 22</a>

1950 <a href="#__codelineno-0-23"> 23</a>

1951 <a href="#__codelineno-0-24"> 24</a>

@@ -2087,153 +2087,157 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2087 <a href="#__codelineno-0-160">160</a>

2088 <a href="#__codelineno-0-161">161</a>

2089 <a href="#__codelineno-0-162">162</a>

2090 -<a href="#__codelineno-0-163">163</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-17" name="__codelineno-0-17"></a>class AgentService:

2091 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> """

2092 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> A service class that encapsulates the logic for managing agents.

2093 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> """

2094 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a>

2095 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> def get_all_agents(self) -> List[Dict[str, Union[str, bool]]]:

2096 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a> """

2097 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> Retrieves all agents from the database.

2090 +<a href="#__codelineno-0-163">163</a>

2091 +<a href="#__codelineno-0-164">164</a>

2092 +<a href="#__codelineno-0-165">165</a>

2093 +<a href="#__codelineno-0-166">166</a>

2094 +<a href="#__codelineno-0-167">167</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-21" name="__codelineno-0-21"></a>class AgentService:

2095 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> """

2096 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> A service class that encapsulates the logic for managing agents.

2097 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a> """

2098 <a id="__codelineno-0-25" name="__codelineno-0-25"></a>

2099 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> Returns:

2100 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> List[dict]: A list of dictionaries where each dictionary represents the serialized data of an agent.

2101 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> """

2102 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> agents = db.session.query(AgentMetadata).all()

2103 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> return agent_metadatas_schema.dump(agents)

2104 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a>

2105 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> def get_agent(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2106 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> """

2107 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> Retrieves a specific agent from the database using its ID.

2099 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a> def get_all_agents(self) -> List[Dict[str, Union[str, bool]]]:

2100 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> """

2101 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> Retrieves all agents from the database.

2102 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a>

2103 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> Returns:

2104 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> List[dict]: A list of dictionaries where each dictionary represents the serialized data of an agent.

2105 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> """

2106 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> agents = db.session.query(AgentMetadata).all()

2107 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> return agent_metadatas_schema.dump(agents)

2108 <a id="__codelineno-0-35" name="__codelineno-0-35"></a>

2109 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> Args:

2110 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> agent_id (str): The ID of the agent to retrieve.

2111 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a>

2112 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> Returns:

2113 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> dict: A dictionary representing the serialized data of the agent if found, otherwise a message indicating

2114 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> that the agent was not found.

2115 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> """

2116 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2117 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> if agent is None:

2118 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> return {"message": f"Agent with ID {agent_id} not found"}

2119 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> return agent_metadata_schema.dump(agent)

2120 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a>

2121 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> def mark_agent_as_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2122 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> """

2123 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> Marks a specific agent as critical.

2109 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> def get_agent(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2110 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> """

2111 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> Retrieves a specific agent from the database using its ID.

2112 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a>

2113 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> Args:

2114 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> agent_id (str): The ID of the agent to retrieve.

2115 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a>

2116 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> Returns:

2117 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> dict: A dictionary representing the serialized data of the agent if found, otherwise a message indicating

2118 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> that the agent was not found.

2119 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> """

2120 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2121 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> if agent is None:

2122 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> return {"message": f"Agent with ID {agent_id} not found"}

2123 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> return agent_metadata_schema.dump(agent)

2124 <a id="__codelineno-0-51" name="__codelineno-0-51"></a>

2125 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> Args:

2126 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> agent_id (str): The ID of the agent to mark as critical.

2127 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a>

2128 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> Returns:

2129 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2130 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> message.

2131 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> """

2132 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2133 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a>

2134 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> if agent is None:

2135 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2136 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a>

2137 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> agent.mark_as_critical()

2138 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> agent_details = agent_metadata_schema.dump(agent)

2139 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> if agent_details["critical_asset"] is False:

2140 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> return {

2141 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> "message": f"Agent {agent_id} failed to mark agent as critical",

2142 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> "success": False,

2143 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> }

2144 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> return {"message": f"Agent {agent_id} marked as critical", "success": True}

2145 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a>

2146 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> def mark_agent_as_non_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2147 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> """

2148 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> Marks a specific agent as non-critical.

2125 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> def mark_agent_as_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2126 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> """

2127 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> Marks a specific agent as critical.

2128 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a>

2129 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> Args:

2130 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a> agent_id (str): The ID of the agent to mark as critical.

2131 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a>

2132 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> Returns:

2133 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2134 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> message.

2135 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> """

2136 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2137 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a>

2138 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> if agent is None:

2139 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2140 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a>

2141 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> agent.mark_as_critical()

2142 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> agent_details = agent_metadata_schema.dump(agent)

2143 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> if agent_details["critical_asset"] is False:

2144 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> return {

2145 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> "message": f"Agent {agent_id} failed to mark agent as critical",

2146 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> "success": False,

2147 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> }

2148 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> return {"message": f"Agent {agent_id} marked as critical", "success": True}

2149 <a id="__codelineno-0-76" name="__codelineno-0-76"></a>

2150 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Args:

2151 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> agent_id (str): The ID of the agent to mark as non-critical.

2152 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a>

2153 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> Returns:

2154 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2155 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> message.

2156 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> """

2157 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2158 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a>

2159 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> if agent is None:

2160 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2161 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a>

2162 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> agent.mark_as_non_critical()

2163 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> agent_details = agent_metadata_schema.dump(agent)

2164 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> if agent_details["critical_asset"] is True:

2165 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> return {

2166 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> "message": f"Agent {agent_id} failed to mark agent as non-critical",

2167 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> "success": False,

2168 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> }

2169 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> return {"message": f"Agent {agent_id} marked as non-critical", "success": True}

2170 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a>

2171 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> def create_agent(self, agent: Dict[str, str]) -> Optional[AgentMetadata]:

2172 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> """

2173 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a> Creates a new agent in the database.

2150 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> def mark_agent_as_non_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2151 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> """

2152 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> Marks a specific agent as non-critical.

2153 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a>

2154 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> Args:

2155 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> agent_id (str): The ID of the agent to mark as non-critical.

2156 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a>

2157 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Returns:

2158 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2159 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> message.

2160 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> """

2161 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2162 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a>

2163 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> if agent is None:

2164 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2165 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a>

2166 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> agent.mark_as_non_critical()

2167 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> agent_details = agent_metadata_schema.dump(agent)

2168 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> if agent_details["critical_asset"] is True:

2169 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> return {

2170 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> "message": f"Agent {agent_id} failed to mark agent as non-critical",

2171 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> "success": False,

2172 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> }

2173 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> return {"message": f"Agent {agent_id} marked as non-critical", "success": True}

2174 <a id="__codelineno-0-101" name="__codelineno-0-101"></a>

2175 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> Args:

2176 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a> agent (dict): A dictionary containing the information of an agent.

2177 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a>

2178 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a> Returns:

2179 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a> The agent object if the agent was successfully created, None otherwise.

2180 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a> """

2181 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> try:

2182 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> agent_last_seen = datetime.strptime(

2183 -<a id="__codelineno-0-110" name="__codelineno-0-110"></a> agent["agent_last_seen"],

2184 -<a id="__codelineno-0-111" name="__codelineno-0-111"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2185 -<a id="__codelineno-0-112" name="__codelineno-0-112"></a> ) # Convert to datetime

2186 -<a id="__codelineno-0-113" name="__codelineno-0-113"></a> except ValueError:

2187 -<a id="__codelineno-0-114" name="__codelineno-0-114"></a> logger.info(

2188 -<a id="__codelineno-0-115" name="__codelineno-0-115"></a> f"Invalid format for agent_last_seen: {agent['agent_last_seen']}. Fixing...",

2189 -<a id="__codelineno-0-116" name="__codelineno-0-116"></a> )

2190 -<a id="__codelineno-0-117" name="__codelineno-0-117"></a> agent_last_seen = datetime.strptime(

2191 -<a id="__codelineno-0-118" name="__codelineno-0-118"></a> "1970-01-01T00:00:00+00:00",

2192 -<a id="__codelineno-0-119" name="__codelineno-0-119"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2193 -<a id="__codelineno-0-120" name="__codelineno-0-120"></a> ) # Use the epoch time as default

2194 -<a id="__codelineno-0-121" name="__codelineno-0-121"></a>

2195 -<a id="__codelineno-0-122" name="__codelineno-0-122"></a> agent_metadata = AgentMetadata(

2196 -<a id="__codelineno-0-123" name="__codelineno-0-123"></a> agent_id=agent["agent_id"],

2197 -<a id="__codelineno-0-124" name="__codelineno-0-124"></a> hostname=agent["agent_name"],

2198 -<a id="__codelineno-0-125" name="__codelineno-0-125"></a> ip_address=agent["agent_ip"],

2199 -<a id="__codelineno-0-126" name="__codelineno-0-126"></a> os=agent["agent_os"],

2200 -<a id="__codelineno-0-127" name="__codelineno-0-127"></a> last_seen=agent_last_seen, # Use the datetime object

2201 -<a id="__codelineno-0-128" name="__codelineno-0-128"></a> critical_asset=False,

2202 -<a id="__codelineno-0-129" name="__codelineno-0-129"></a> )

2203 -<a id="__codelineno-0-130" name="__codelineno-0-130"></a> logger.info(f"Agent metadata: {agent_metadata}")

2204 -<a id="__codelineno-0-131" name="__codelineno-0-131"></a>

2205 -<a id="__codelineno-0-132" name="__codelineno-0-132"></a> try:

2206 -<a id="__codelineno-0-133" name="__codelineno-0-133"></a> db.session.add(agent_metadata)

2207 -<a id="__codelineno-0-134" name="__codelineno-0-134"></a> db.session.commit()

2208 -<a id="__codelineno-0-135" name="__codelineno-0-135"></a> return agent_metadata

2209 -<a id="__codelineno-0-136" name="__codelineno-0-136"></a> except Exception as e:

2210 -<a id="__codelineno-0-137" name="__codelineno-0-137"></a> logger.error(f"Failed to create agent: {e}")

2211 -<a id="__codelineno-0-138" name="__codelineno-0-138"></a> return None

2212 -<a id="__codelineno-0-139" name="__codelineno-0-139"></a>

2213 -<a id="__codelineno-0-140" name="__codelineno-0-140"></a> def delete_agent_db(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2214 -<a id="__codelineno-0-141" name="__codelineno-0-141"></a> """

2215 -<a id="__codelineno-0-142" name="__codelineno-0-142"></a> Deletes a specific agent from the database using its ID.

2175 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> def create_agent(self, agent: Dict[str, str]) -> Optional[AgentMetadata]:

2176 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> """

2177 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Creates a new agent in the database.

2178 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a>

2179 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> Args:

2180 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a> agent (dict): A dictionary containing the information of an agent.

2181 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a>

2182 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a> Returns:

2183 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> The agent object if the agent was successfully created, None otherwise.

2184 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a> """

2185 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a> try:

2186 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> agent_last_seen = datetime.strptime(

2187 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> agent["agent_last_seen"],

2188 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2189 +<a id="__codelineno-0-116" name="__codelineno-0-116"></a> ) # Convert to datetime

2190 +<a id="__codelineno-0-117" name="__codelineno-0-117"></a> except ValueError:

2191 +<a id="__codelineno-0-118" name="__codelineno-0-118"></a> logger.info(

2192 +<a id="__codelineno-0-119" name="__codelineno-0-119"></a> f"Invalid format for agent_last_seen: {agent['agent_last_seen']}. Fixing...",

2193 +<a id="__codelineno-0-120" name="__codelineno-0-120"></a> )

2194 +<a id="__codelineno-0-121" name="__codelineno-0-121"></a> agent_last_seen = datetime.strptime(

2195 +<a id="__codelineno-0-122" name="__codelineno-0-122"></a> "1970-01-01T00:00:00+00:00",

2196 +<a id="__codelineno-0-123" name="__codelineno-0-123"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2197 +<a id="__codelineno-0-124" name="__codelineno-0-124"></a> ) # Use the epoch time as default

2198 +<a id="__codelineno-0-125" name="__codelineno-0-125"></a>

2199 +<a id="__codelineno-0-126" name="__codelineno-0-126"></a> agent_metadata = AgentMetadata(

2200 +<a id="__codelineno-0-127" name="__codelineno-0-127"></a> agent_id=agent["agent_id"],

2201 +<a id="__codelineno-0-128" name="__codelineno-0-128"></a> hostname=agent["agent_name"],

2202 +<a id="__codelineno-0-129" name="__codelineno-0-129"></a> ip_address=agent["agent_ip"],

2203 +<a id="__codelineno-0-130" name="__codelineno-0-130"></a> os=agent["agent_os"],

2204 +<a id="__codelineno-0-131" name="__codelineno-0-131"></a> last_seen=agent_last_seen, # Use the datetime object

2205 +<a id="__codelineno-0-132" name="__codelineno-0-132"></a> critical_asset=False,

2206 +<a id="__codelineno-0-133" name="__codelineno-0-133"></a> )

2207 +<a id="__codelineno-0-134" name="__codelineno-0-134"></a> logger.info(f"Agent metadata: {agent_metadata}")

2208 +<a id="__codelineno-0-135" name="__codelineno-0-135"></a>

2209 +<a id="__codelineno-0-136" name="__codelineno-0-136"></a> try:

2210 +<a id="__codelineno-0-137" name="__codelineno-0-137"></a> db.session.add(agent_metadata)

2211 +<a id="__codelineno-0-138" name="__codelineno-0-138"></a> db.session.commit()

2212 +<a id="__codelineno-0-139" name="__codelineno-0-139"></a> return agent_metadata

2213 +<a id="__codelineno-0-140" name="__codelineno-0-140"></a> except Exception as e:

2214 +<a id="__codelineno-0-141" name="__codelineno-0-141"></a> logger.error(f"Failed to create agent: {e}")

2215 +<a id="__codelineno-0-142" name="__codelineno-0-142"></a> return None

2216 <a id="__codelineno-0-143" name="__codelineno-0-143"></a>

2217 -<a id="__codelineno-0-144" name="__codelineno-0-144"></a> Args:

2218 -<a id="__codelineno-0-145" name="__codelineno-0-145"></a> agent_id (str): The ID of the agent to delete.

2219 -<a id="__codelineno-0-146" name="__codelineno-0-146"></a>

2220 -<a id="__codelineno-0-147" name="__codelineno-0-147"></a> Returns:

2221 -<a id="__codelineno-0-148" name="__codelineno-0-148"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2222 -<a id="__codelineno-0-149" name="__codelineno-0-149"></a> message.

2223 -<a id="__codelineno-0-150" name="__codelineno-0-150"></a> """

2224 -<a id="__codelineno-0-151" name="__codelineno-0-151"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2225 -<a id="__codelineno-0-152" name="__codelineno-0-152"></a> if agent is None:

2226 -<a id="__codelineno-0-153" name="__codelineno-0-153"></a> return {"message": f"Agent with ID {agent_id} not found", "success": False}

2227 -<a id="__codelineno-0-154" name="__codelineno-0-154"></a> try:

2228 -<a id="__codelineno-0-155" name="__codelineno-0-155"></a> db.session.delete(agent)

2229 -<a id="__codelineno-0-156" name="__codelineno-0-156"></a> db.session.commit()

2230 -<a id="__codelineno-0-157" name="__codelineno-0-157"></a> return {"message": f"Agent with ID {agent_id} deleted", "success": True}

2231 -<a id="__codelineno-0-158" name="__codelineno-0-158"></a> except Exception as e:

2232 -<a id="__codelineno-0-159" name="__codelineno-0-159"></a> logger.error(f"Failed to delete agent: {e}")

2233 -<a id="__codelineno-0-160" name="__codelineno-0-160"></a> return {

2234 -<a id="__codelineno-0-161" name="__codelineno-0-161"></a> "message": f"Failed to delete agent with ID {agent_id}",

2235 -<a id="__codelineno-0-162" name="__codelineno-0-162"></a> "success": False,

2236 -<a id="__codelineno-0-163" name="__codelineno-0-163"></a> }

2217 +<a id="__codelineno-0-144" name="__codelineno-0-144"></a> def delete_agent_db(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2218 +<a id="__codelineno-0-145" name="__codelineno-0-145"></a> """

2219 +<a id="__codelineno-0-146" name="__codelineno-0-146"></a> Deletes a specific agent from the database using its ID.

2220 +<a id="__codelineno-0-147" name="__codelineno-0-147"></a>

2221 +<a id="__codelineno-0-148" name="__codelineno-0-148"></a> Args:

2222 +<a id="__codelineno-0-149" name="__codelineno-0-149"></a> agent_id (str): The ID of the agent to delete.

2223 +<a id="__codelineno-0-150" name="__codelineno-0-150"></a>

2224 +<a id="__codelineno-0-151" name="__codelineno-0-151"></a> Returns:

2225 +<a id="__codelineno-0-152" name="__codelineno-0-152"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2226 +<a id="__codelineno-0-153" name="__codelineno-0-153"></a> message.

2227 +<a id="__codelineno-0-154" name="__codelineno-0-154"></a> """

2228 +<a id="__codelineno-0-155" name="__codelineno-0-155"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2229 +<a id="__codelineno-0-156" name="__codelineno-0-156"></a> if agent is None:

2230 +<a id="__codelineno-0-157" name="__codelineno-0-157"></a> return {"message": f"Agent with ID {agent_id} not found", "success": False}

2231 +<a id="__codelineno-0-158" name="__codelineno-0-158"></a> try:

2232 +<a id="__codelineno-0-159" name="__codelineno-0-159"></a> db.session.delete(agent)

2233 +<a id="__codelineno-0-160" name="__codelineno-0-160"></a> db.session.commit()

2234 +<a id="__codelineno-0-161" name="__codelineno-0-161"></a> return {"message": f"Agent with ID {agent_id} deleted", "success": True}

2235 +<a id="__codelineno-0-162" name="__codelineno-0-162"></a> except Exception as e:

2236 +<a id="__codelineno-0-163" name="__codelineno-0-163"></a> logger.error(f"Failed to delete agent: {e}")

2237 +<a id="__codelineno-0-164" name="__codelineno-0-164"></a> return {

2238 +<a id="__codelineno-0-165" name="__codelineno-0-165"></a> "message": f"Failed to delete agent with ID {agent_id}",

2239 +<a id="__codelineno-0-166" name="__codelineno-0-166"></a> "success": False,

2240 +<a id="__codelineno-0-167" name="__codelineno-0-167"></a> }

2241 </code></pre></div></td></tr></table></div>

2242 </details>

2243

@@ -2315,11 +2319,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2319

2320 <details class="quote">

2321 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

2318 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-98"> 98</a>

2319 -<a href="#__codelineno-0-99"> 99</a>

2320 -<a href="#__codelineno-0-100">100</a>

2321 -<a href="#__codelineno-0-101">101</a>

2322 -<a href="#__codelineno-0-102">102</a>

2322 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-102">102</a>

2323 <a href="#__codelineno-0-103">103</a>

2324 <a href="#__codelineno-0-104">104</a>

2325 <a href="#__codelineno-0-105">105</a>

@@ -2355,47 +2355,51 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2355 <a href="#__codelineno-0-135">135</a>

2356 <a href="#__codelineno-0-136">136</a>

2357 <a href="#__codelineno-0-137">137</a>

2358 -<a href="#__codelineno-0-138">138</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-98" name="__codelineno-0-98"></a>def create_agent(self, agent: Dict[str, str]) -> Optional[AgentMetadata]:

2359 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> """

2360 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a> Creates a new agent in the database.

2361 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a>

2362 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> Args:

2363 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a> agent (dict): A dictionary containing the information of an agent.

2364 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a>

2365 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a> Returns:

2366 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a> The agent object if the agent was successfully created, None otherwise.

2367 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a> """

2368 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> try:

2369 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> agent_last_seen = datetime.strptime(

2370 -<a id="__codelineno-0-110" name="__codelineno-0-110"></a> agent["agent_last_seen"],

2371 -<a id="__codelineno-0-111" name="__codelineno-0-111"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2372 -<a id="__codelineno-0-112" name="__codelineno-0-112"></a> ) # Convert to datetime

2373 -<a id="__codelineno-0-113" name="__codelineno-0-113"></a> except ValueError:

2374 -<a id="__codelineno-0-114" name="__codelineno-0-114"></a> logger.info(

2375 -<a id="__codelineno-0-115" name="__codelineno-0-115"></a> f"Invalid format for agent_last_seen: {agent['agent_last_seen']}. Fixing...",

2376 -<a id="__codelineno-0-116" name="__codelineno-0-116"></a> )

2377 -<a id="__codelineno-0-117" name="__codelineno-0-117"></a> agent_last_seen = datetime.strptime(

2378 -<a id="__codelineno-0-118" name="__codelineno-0-118"></a> "1970-01-01T00:00:00+00:00",

2379 -<a id="__codelineno-0-119" name="__codelineno-0-119"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2380 -<a id="__codelineno-0-120" name="__codelineno-0-120"></a> ) # Use the epoch time as default

2381 -<a id="__codelineno-0-121" name="__codelineno-0-121"></a>

2382 -<a id="__codelineno-0-122" name="__codelineno-0-122"></a> agent_metadata = AgentMetadata(

2383 -<a id="__codelineno-0-123" name="__codelineno-0-123"></a> agent_id=agent["agent_id"],

2384 -<a id="__codelineno-0-124" name="__codelineno-0-124"></a> hostname=agent["agent_name"],

2385 -<a id="__codelineno-0-125" name="__codelineno-0-125"></a> ip_address=agent["agent_ip"],

2386 -<a id="__codelineno-0-126" name="__codelineno-0-126"></a> os=agent["agent_os"],

2387 -<a id="__codelineno-0-127" name="__codelineno-0-127"></a> last_seen=agent_last_seen, # Use the datetime object

2388 -<a id="__codelineno-0-128" name="__codelineno-0-128"></a> critical_asset=False,

2389 -<a id="__codelineno-0-129" name="__codelineno-0-129"></a> )

2390 -<a id="__codelineno-0-130" name="__codelineno-0-130"></a> logger.info(f"Agent metadata: {agent_metadata}")

2391 -<a id="__codelineno-0-131" name="__codelineno-0-131"></a>

2392 -<a id="__codelineno-0-132" name="__codelineno-0-132"></a> try:

2393 -<a id="__codelineno-0-133" name="__codelineno-0-133"></a> db.session.add(agent_metadata)

2394 -<a id="__codelineno-0-134" name="__codelineno-0-134"></a> db.session.commit()

2395 -<a id="__codelineno-0-135" name="__codelineno-0-135"></a> return agent_metadata

2396 -<a id="__codelineno-0-136" name="__codelineno-0-136"></a> except Exception as e:

2397 -<a id="__codelineno-0-137" name="__codelineno-0-137"></a> logger.error(f"Failed to create agent: {e}")

2398 -<a id="__codelineno-0-138" name="__codelineno-0-138"></a> return None

2358 +<a href="#__codelineno-0-138">138</a>

2359 +<a href="#__codelineno-0-139">139</a>

2360 +<a href="#__codelineno-0-140">140</a>

2361 +<a href="#__codelineno-0-141">141</a>

2362 +<a href="#__codelineno-0-142">142</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-102" name="__codelineno-0-102"></a>def create_agent(self, agent: Dict[str, str]) -> Optional[AgentMetadata]:

2363 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> """

2364 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Creates a new agent in the database.

2365 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a>

2366 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> Args:

2367 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a> agent (dict): A dictionary containing the information of an agent.

2368 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a>

2369 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a> Returns:

2370 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> The agent object if the agent was successfully created, None otherwise.

2371 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a> """

2372 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a> try:

2373 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> agent_last_seen = datetime.strptime(

2374 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> agent["agent_last_seen"],

2375 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2376 +<a id="__codelineno-0-116" name="__codelineno-0-116"></a> ) # Convert to datetime

2377 +<a id="__codelineno-0-117" name="__codelineno-0-117"></a> except ValueError:

2378 +<a id="__codelineno-0-118" name="__codelineno-0-118"></a> logger.info(

2379 +<a id="__codelineno-0-119" name="__codelineno-0-119"></a> f"Invalid format for agent_last_seen: {agent['agent_last_seen']}. Fixing...",

2380 +<a id="__codelineno-0-120" name="__codelineno-0-120"></a> )

2381 +<a id="__codelineno-0-121" name="__codelineno-0-121"></a> agent_last_seen = datetime.strptime(

2382 +<a id="__codelineno-0-122" name="__codelineno-0-122"></a> "1970-01-01T00:00:00+00:00",

2383 +<a id="__codelineno-0-123" name="__codelineno-0-123"></a> "%Y-%m-%dT%H:%M:%S+00:00",

2384 +<a id="__codelineno-0-124" name="__codelineno-0-124"></a> ) # Use the epoch time as default

2385 +<a id="__codelineno-0-125" name="__codelineno-0-125"></a>

2386 +<a id="__codelineno-0-126" name="__codelineno-0-126"></a> agent_metadata = AgentMetadata(

2387 +<a id="__codelineno-0-127" name="__codelineno-0-127"></a> agent_id=agent["agent_id"],

2388 +<a id="__codelineno-0-128" name="__codelineno-0-128"></a> hostname=agent["agent_name"],

2389 +<a id="__codelineno-0-129" name="__codelineno-0-129"></a> ip_address=agent["agent_ip"],

2390 +<a id="__codelineno-0-130" name="__codelineno-0-130"></a> os=agent["agent_os"],

2391 +<a id="__codelineno-0-131" name="__codelineno-0-131"></a> last_seen=agent_last_seen, # Use the datetime object

2392 +<a id="__codelineno-0-132" name="__codelineno-0-132"></a> critical_asset=False,

2393 +<a id="__codelineno-0-133" name="__codelineno-0-133"></a> )

2394 +<a id="__codelineno-0-134" name="__codelineno-0-134"></a> logger.info(f"Agent metadata: {agent_metadata}")

2395 +<a id="__codelineno-0-135" name="__codelineno-0-135"></a>

2396 +<a id="__codelineno-0-136" name="__codelineno-0-136"></a> try:

2397 +<a id="__codelineno-0-137" name="__codelineno-0-137"></a> db.session.add(agent_metadata)

2398 +<a id="__codelineno-0-138" name="__codelineno-0-138"></a> db.session.commit()

2399 +<a id="__codelineno-0-139" name="__codelineno-0-139"></a> return agent_metadata

2400 +<a id="__codelineno-0-140" name="__codelineno-0-140"></a> except Exception as e:

2401 +<a id="__codelineno-0-141" name="__codelineno-0-141"></a> logger.error(f"Failed to create agent: {e}")

2402 +<a id="__codelineno-0-142" name="__codelineno-0-142"></a> return None

2403 </code></pre></div></td></tr></table></div>

2404 </details>

2405 </div>

@@ -2478,11 +2482,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2482

2483 <details class="quote">

2484 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

2481 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-140">140</a>

2482 -<a href="#__codelineno-0-141">141</a>

2483 -<a href="#__codelineno-0-142">142</a>

2484 -<a href="#__codelineno-0-143">143</a>

2485 -<a href="#__codelineno-0-144">144</a>

2485 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-144">144</a>

2486 <a href="#__codelineno-0-145">145</a>

2487 <a href="#__codelineno-0-146">146</a>

2488 <a href="#__codelineno-0-147">147</a>

@@ -2501,30 +2501,34 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2501 <a href="#__codelineno-0-160">160</a>

2502 <a href="#__codelineno-0-161">161</a>

2503 <a href="#__codelineno-0-162">162</a>

2504 -<a href="#__codelineno-0-163">163</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-140" name="__codelineno-0-140"></a>def delete_agent_db(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2505 -<a id="__codelineno-0-141" name="__codelineno-0-141"></a> """

2506 -<a id="__codelineno-0-142" name="__codelineno-0-142"></a> Deletes a specific agent from the database using its ID.

2507 -<a id="__codelineno-0-143" name="__codelineno-0-143"></a>

2508 -<a id="__codelineno-0-144" name="__codelineno-0-144"></a> Args:

2509 -<a id="__codelineno-0-145" name="__codelineno-0-145"></a> agent_id (str): The ID of the agent to delete.

2510 -<a id="__codelineno-0-146" name="__codelineno-0-146"></a>

2511 -<a id="__codelineno-0-147" name="__codelineno-0-147"></a> Returns:

2512 -<a id="__codelineno-0-148" name="__codelineno-0-148"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2513 -<a id="__codelineno-0-149" name="__codelineno-0-149"></a> message.

2514 -<a id="__codelineno-0-150" name="__codelineno-0-150"></a> """

2515 -<a id="__codelineno-0-151" name="__codelineno-0-151"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2516 -<a id="__codelineno-0-152" name="__codelineno-0-152"></a> if agent is None:

2517 -<a id="__codelineno-0-153" name="__codelineno-0-153"></a> return {"message": f"Agent with ID {agent_id} not found", "success": False}

2518 -<a id="__codelineno-0-154" name="__codelineno-0-154"></a> try:

2519 -<a id="__codelineno-0-155" name="__codelineno-0-155"></a> db.session.delete(agent)

2520 -<a id="__codelineno-0-156" name="__codelineno-0-156"></a> db.session.commit()

2521 -<a id="__codelineno-0-157" name="__codelineno-0-157"></a> return {"message": f"Agent with ID {agent_id} deleted", "success": True}

2522 -<a id="__codelineno-0-158" name="__codelineno-0-158"></a> except Exception as e:

2523 -<a id="__codelineno-0-159" name="__codelineno-0-159"></a> logger.error(f"Failed to delete agent: {e}")

2524 -<a id="__codelineno-0-160" name="__codelineno-0-160"></a> return {

2525 -<a id="__codelineno-0-161" name="__codelineno-0-161"></a> "message": f"Failed to delete agent with ID {agent_id}",

2526 -<a id="__codelineno-0-162" name="__codelineno-0-162"></a> "success": False,

2527 -<a id="__codelineno-0-163" name="__codelineno-0-163"></a> }

2504 +<a href="#__codelineno-0-163">163</a>

2505 +<a href="#__codelineno-0-164">164</a>

2506 +<a href="#__codelineno-0-165">165</a>

2507 +<a href="#__codelineno-0-166">166</a>

2508 +<a href="#__codelineno-0-167">167</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-144" name="__codelineno-0-144"></a>def delete_agent_db(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2509 +<a id="__codelineno-0-145" name="__codelineno-0-145"></a> """

2510 +<a id="__codelineno-0-146" name="__codelineno-0-146"></a> Deletes a specific agent from the database using its ID.

2511 +<a id="__codelineno-0-147" name="__codelineno-0-147"></a>

2512 +<a id="__codelineno-0-148" name="__codelineno-0-148"></a> Args:

2513 +<a id="__codelineno-0-149" name="__codelineno-0-149"></a> agent_id (str): The ID of the agent to delete.

2514 +<a id="__codelineno-0-150" name="__codelineno-0-150"></a>

2515 +<a id="__codelineno-0-151" name="__codelineno-0-151"></a> Returns:

2516 +<a id="__codelineno-0-152" name="__codelineno-0-152"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2517 +<a id="__codelineno-0-153" name="__codelineno-0-153"></a> message.

2518 +<a id="__codelineno-0-154" name="__codelineno-0-154"></a> """

2519 +<a id="__codelineno-0-155" name="__codelineno-0-155"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2520 +<a id="__codelineno-0-156" name="__codelineno-0-156"></a> if agent is None:

2521 +<a id="__codelineno-0-157" name="__codelineno-0-157"></a> return {"message": f"Agent with ID {agent_id} not found", "success": False}

2522 +<a id="__codelineno-0-158" name="__codelineno-0-158"></a> try:

2523 +<a id="__codelineno-0-159" name="__codelineno-0-159"></a> db.session.delete(agent)

2524 +<a id="__codelineno-0-160" name="__codelineno-0-160"></a> db.session.commit()

2525 +<a id="__codelineno-0-161" name="__codelineno-0-161"></a> return {"message": f"Agent with ID {agent_id} deleted", "success": True}

2526 +<a id="__codelineno-0-162" name="__codelineno-0-162"></a> except Exception as e:

2527 +<a id="__codelineno-0-163" name="__codelineno-0-163"></a> logger.error(f"Failed to delete agent: {e}")

2528 +<a id="__codelineno-0-164" name="__codelineno-0-164"></a> return {

2529 +<a id="__codelineno-0-165" name="__codelineno-0-165"></a> "message": f"Failed to delete agent with ID {agent_id}",

2530 +<a id="__codelineno-0-166" name="__codelineno-0-166"></a> "success": False,

2531 +<a id="__codelineno-0-167" name="__codelineno-0-167"></a> }

2532 </code></pre></div></td></tr></table></div>

2533 </details>

2534 </div>

@@ -2607,11 +2611,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2611

2612 <details class="quote">

2613 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

2610 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-32">32</a>

2611 -<a href="#__codelineno-0-33">33</a>

2612 -<a href="#__codelineno-0-34">34</a>

2613 -<a href="#__codelineno-0-35">35</a>

2614 -<a href="#__codelineno-0-36">36</a>

2614 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-36">36</a>

2615 <a href="#__codelineno-0-37">37</a>

2616 <a href="#__codelineno-0-38">38</a>

2617 <a href="#__codelineno-0-39">39</a>

@@ -2621,21 +2621,25 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2621 <a href="#__codelineno-0-43">43</a>

2622 <a href="#__codelineno-0-44">44</a>

2623 <a href="#__codelineno-0-45">45</a>

2624 -<a href="#__codelineno-0-46">46</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-32" name="__codelineno-0-32"></a>def get_agent(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2625 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> """

2626 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> Retrieves a specific agent from the database using its ID.

2627 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a>

2628 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> Args:

2629 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> agent_id (str): The ID of the agent to retrieve.

2630 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a>

2631 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> Returns:

2632 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> dict: A dictionary representing the serialized data of the agent if found, otherwise a message indicating

2633 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> that the agent was not found.

2634 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> """

2635 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2636 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> if agent is None:

2637 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> return {"message": f"Agent with ID {agent_id} not found"}

2638 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> return agent_metadata_schema.dump(agent)

2624 +<a href="#__codelineno-0-46">46</a>

2625 +<a href="#__codelineno-0-47">47</a>

2626 +<a href="#__codelineno-0-48">48</a>

2627 +<a href="#__codelineno-0-49">49</a>

2628 +<a href="#__codelineno-0-50">50</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-36" name="__codelineno-0-36"></a>def get_agent(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2629 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> """

2630 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> Retrieves a specific agent from the database using its ID.

2631 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a>

2632 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> Args:

2633 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> agent_id (str): The ID of the agent to retrieve.

2634 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a>

2635 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> Returns:

2636 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> dict: A dictionary representing the serialized data of the agent if found, otherwise a message indicating

2637 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> that the agent was not found.

2638 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> """

2639 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2640 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> if agent is None:

2641 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> return {"message": f"Agent with ID {agent_id} not found"}

2642 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> return agent_metadata_schema.dump(agent)

2643 </code></pre></div></td></tr></table></div>

2644 </details>

2645 </div>

@@ -2680,23 +2684,23 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2684

2685 <details class="quote">

2686 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

2683 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-22">22</a>

2684 -<a href="#__codelineno-0-23">23</a>

2685 -<a href="#__codelineno-0-24">24</a>

2686 -<a href="#__codelineno-0-25">25</a>

2687 -<a href="#__codelineno-0-26">26</a>

2687 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-26">26</a>

2688 <a href="#__codelineno-0-27">27</a>

2689 <a href="#__codelineno-0-28">28</a>

2690 <a href="#__codelineno-0-29">29</a>

2691 -<a href="#__codelineno-0-30">30</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-22" name="__codelineno-0-22"></a>def get_all_agents(self) -> List[Dict[str, Union[str, bool]]]:

2692 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a> """

2693 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> Retrieves all agents from the database.

2694 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a>

2695 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> Returns:

2696 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> List[dict]: A list of dictionaries where each dictionary represents the serialized data of an agent.

2697 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> """

2698 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> agents = db.session.query(AgentMetadata).all()

2699 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> return agent_metadatas_schema.dump(agents)

2691 +<a href="#__codelineno-0-30">30</a>

2692 +<a href="#__codelineno-0-31">31</a>

2693 +<a href="#__codelineno-0-32">32</a>

2694 +<a href="#__codelineno-0-33">33</a>

2695 +<a href="#__codelineno-0-34">34</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-26" name="__codelineno-0-26"></a>def get_all_agents(self) -> List[Dict[str, Union[str, bool]]]:

2696 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> """

2697 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> Retrieves all agents from the database.

2698 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a>

2699 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> Returns:

2700 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> List[dict]: A list of dictionaries where each dictionary represents the serialized data of an agent.

2701 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> """

2702 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> agents = db.session.query(AgentMetadata).all()

2703 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> return agent_metadatas_schema.dump(agents)

2704 </code></pre></div></td></tr></table></div>

2705 </details>

2706 </div>

@@ -2779,11 +2783,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2783

2784 <details class="quote">

2785 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

2782 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-48">48</a>

2783 -<a href="#__codelineno-0-49">49</a>

2784 -<a href="#__codelineno-0-50">50</a>

2785 -<a href="#__codelineno-0-51">51</a>

2786 -<a href="#__codelineno-0-52">52</a>

2786 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-52">52</a>

2787 <a href="#__codelineno-0-53">53</a>

2788 <a href="#__codelineno-0-54">54</a>

2789 <a href="#__codelineno-0-55">55</a>

@@ -2802,30 +2802,34 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2802 <a href="#__codelineno-0-68">68</a>

2803 <a href="#__codelineno-0-69">69</a>

2804 <a href="#__codelineno-0-70">70</a>

2805 -<a href="#__codelineno-0-71">71</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-48" name="__codelineno-0-48"></a>def mark_agent_as_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2806 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> """

2807 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> Marks a specific agent as critical.

2808 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a>

2809 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> Args:

2810 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> agent_id (str): The ID of the agent to mark as critical.

2811 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a>

2812 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> Returns:

2813 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2814 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> message.

2815 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> """

2816 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2817 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a>

2818 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> if agent is None:

2819 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2820 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a>

2821 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> agent.mark_as_critical()

2822 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> agent_details = agent_metadata_schema.dump(agent)

2823 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> if agent_details["critical_asset"] is False:

2824 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> return {

2825 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> "message": f"Agent {agent_id} failed to mark agent as critical",

2826 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> "success": False,

2827 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> }

2828 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> return {"message": f"Agent {agent_id} marked as critical", "success": True}

2805 +<a href="#__codelineno-0-71">71</a>

2806 +<a href="#__codelineno-0-72">72</a>

2807 +<a href="#__codelineno-0-73">73</a>

2808 +<a href="#__codelineno-0-74">74</a>

2809 +<a href="#__codelineno-0-75">75</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-52" name="__codelineno-0-52"></a>def mark_agent_as_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2810 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> """

2811 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> Marks a specific agent as critical.

2812 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a>

2813 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> Args:

2814 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a> agent_id (str): The ID of the agent to mark as critical.

2815 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a>

2816 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> Returns:

2817 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2818 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> message.

2819 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> """

2820 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2821 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a>

2822 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> if agent is None:

2823 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2824 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a>

2825 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> agent.mark_as_critical()

2826 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> agent_details = agent_metadata_schema.dump(agent)

2827 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> if agent_details["critical_asset"] is False:

2828 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> return {

2829 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> "message": f"Agent {agent_id} failed to mark agent as critical",

2830 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> "success": False,

2831 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> }

2832 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> return {"message": f"Agent {agent_id} marked as critical", "success": True}

2833 </code></pre></div></td></tr></table></div>

2834 </details>

2835 </div>

@@ -2908,53 +2912,53 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2912

2913 <details class="quote">

2914 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

2911 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-73">73</a>

2912 -<a href="#__codelineno-0-74">74</a>

2913 -<a href="#__codelineno-0-75">75</a>

2914 -<a href="#__codelineno-0-76">76</a>

2915 -<a href="#__codelineno-0-77">77</a>

2916 -<a href="#__codelineno-0-78">78</a>

2917 -<a href="#__codelineno-0-79">79</a>

2918 -<a href="#__codelineno-0-80">80</a>

2919 -<a href="#__codelineno-0-81">81</a>

2920 -<a href="#__codelineno-0-82">82</a>

2921 -<a href="#__codelineno-0-83">83</a>

2922 -<a href="#__codelineno-0-84">84</a>

2923 -<a href="#__codelineno-0-85">85</a>

2924 -<a href="#__codelineno-0-86">86</a>

2925 -<a href="#__codelineno-0-87">87</a>

2926 -<a href="#__codelineno-0-88">88</a>

2927 -<a href="#__codelineno-0-89">89</a>

2928 -<a href="#__codelineno-0-90">90</a>

2929 -<a href="#__codelineno-0-91">91</a>

2930 -<a href="#__codelineno-0-92">92</a>

2931 -<a href="#__codelineno-0-93">93</a>

2932 -<a href="#__codelineno-0-94">94</a>

2933 -<a href="#__codelineno-0-95">95</a>

2934 -<a href="#__codelineno-0-96">96</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-73" name="__codelineno-0-73"></a>def mark_agent_as_non_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2935 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> """

2936 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> Marks a specific agent as non-critical.

2937 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a>

2938 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Args:

2939 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> agent_id (str): The ID of the agent to mark as non-critical.

2940 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a>

2941 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> Returns:

2942 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2943 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> message.

2944 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> """

2945 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2946 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a>

2947 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> if agent is None:

2948 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2949 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a>

2950 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> agent.mark_as_non_critical()

2951 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> agent_details = agent_metadata_schema.dump(agent)

2952 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> if agent_details["critical_asset"] is True:

2953 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> return {

2954 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> "message": f"Agent {agent_id} failed to mark agent as non-critical",

2955 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> "success": False,

2956 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> }

2957 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> return {"message": f"Agent {agent_id} marked as non-critical", "success": True}

2915 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-77"> 77</a>

2916 +<a href="#__codelineno-0-78"> 78</a>

2917 +<a href="#__codelineno-0-79"> 79</a>

2918 +<a href="#__codelineno-0-80"> 80</a>

2919 +<a href="#__codelineno-0-81"> 81</a>

2920 +<a href="#__codelineno-0-82"> 82</a>

2921 +<a href="#__codelineno-0-83"> 83</a>

2922 +<a href="#__codelineno-0-84"> 84</a>

2923 +<a href="#__codelineno-0-85"> 85</a>

2924 +<a href="#__codelineno-0-86"> 86</a>

2925 +<a href="#__codelineno-0-87"> 87</a>

2926 +<a href="#__codelineno-0-88"> 88</a>

2927 +<a href="#__codelineno-0-89"> 89</a>

2928 +<a href="#__codelineno-0-90"> 90</a>

2929 +<a href="#__codelineno-0-91"> 91</a>

2930 +<a href="#__codelineno-0-92"> 92</a>

2931 +<a href="#__codelineno-0-93"> 93</a>

2932 +<a href="#__codelineno-0-94"> 94</a>

2933 +<a href="#__codelineno-0-95"> 95</a>

2934 +<a href="#__codelineno-0-96"> 96</a>

2935 +<a href="#__codelineno-0-97"> 97</a>

2936 +<a href="#__codelineno-0-98"> 98</a>

2937 +<a href="#__codelineno-0-99"> 99</a>

2938 +<a href="#__codelineno-0-100">100</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-77" name="__codelineno-0-77"></a>def mark_agent_as_non_critical(self, agent_id: str) -> Dict[str, Union[str, bool]]:

2939 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> """

2940 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> Marks a specific agent as non-critical.

2941 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a>

2942 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> Args:

2943 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> agent_id (str): The ID of the agent to mark as non-critical.

2944 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a>

2945 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Returns:

2946 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> dict: A dictionary representing a success message if the operation was successful, otherwise an error

2947 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> message.

2948 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> """

2949 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> agent = db.session.query(AgentMetadata).filter_by(agent_id=agent_id).first()

2950 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a>

2951 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> if agent is None:

2952 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> return {"message": f"Agent {agent_id} not found", "success": False}

2953 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a>

2954 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> agent.mark_as_non_critical()

2955 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> agent_details = agent_metadata_schema.dump(agent)

2956 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> if agent_details["critical_asset"] is True:

2957 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> return {

2958 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> "message": f"Agent {agent_id} failed to mark agent as non-critical",

2959 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> "success": False,

2960 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> }

2961 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> return {"message": f"Agent {agent_id} marked as non-critical", "success": True}

2962 </code></pre></div></td></tr></table></div>

2963 </details>

2964 </div>

@@ -2986,11 +2990,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

2990

2991 <details class="quote">

2992 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

2989 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-166">166</a>

2990 -<a href="#__codelineno-0-167">167</a>

2991 -<a href="#__codelineno-0-168">168</a>

2992 -<a href="#__codelineno-0-169">169</a>

2993 -<a href="#__codelineno-0-170">170</a>

2993 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-170">170</a>

2994 <a href="#__codelineno-0-171">171</a>

2995 <a href="#__codelineno-0-172">172</a>

2996 <a href="#__codelineno-0-173">173</a>

@@ -3088,109 +3088,127 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

3088 <a href="#__codelineno-0-265">265</a>

3089 <a href="#__codelineno-0-266">266</a>

3090 <a href="#__codelineno-0-267">267</a>

3091 -<a href="#__codelineno-0-268">268</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-166" name="__codelineno-0-166"></a>class AgentSyncService:

3092 -<a id="__codelineno-0-167" name="__codelineno-0-167"></a> def __init__(self):

3093 -<a id="__codelineno-0-168" name="__codelineno-0-168"></a> self.agent_service = AgentService()

3094 -<a id="__codelineno-0-169" name="__codelineno-0-169"></a>

3095 -<a id="__codelineno-0-170" name="__codelineno-0-170"></a> def collect_wazuh_details(self, connector_name: str) -> Tuple[Optional[str], Optional[str], Optional[str]]:

3096 -<a id="__codelineno-0-171" name="__codelineno-0-171"></a> """

3097 -<a id="__codelineno-0-172" name="__codelineno-0-172"></a> Collects the information of all Wazuh API credentials using the WazuhIndexerConnector class details.

3091 +<a href="#__codelineno-0-268">268</a>

3092 +<a href="#__codelineno-0-269">269</a>

3093 +<a href="#__codelineno-0-270">270</a>

3094 +<a href="#__codelineno-0-271">271</a>

3095 +<a href="#__codelineno-0-272">272</a>

3096 +<a href="#__codelineno-0-273">273</a>

3097 +<a href="#__codelineno-0-274">274</a>

3098 +<a href="#__codelineno-0-275">275</a>

3099 +<a href="#__codelineno-0-276">276</a>

3100 +<a href="#__codelineno-0-277">277</a>

3101 +<a href="#__codelineno-0-278">278</a>

3102 +<a href="#__codelineno-0-279">279</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-170" name="__codelineno-0-170"></a>class AgentSyncService:

3103 +<a id="__codelineno-0-171" name="__codelineno-0-171"></a> def __init__(self):

3104 +<a id="__codelineno-0-172" name="__codelineno-0-172"></a> self.agent_service = AgentService()

3105 <a id="__codelineno-0-173" name="__codelineno-0-173"></a>

3099 -<a id="__codelineno-0-174" name="__codelineno-0-174"></a> Returns:

3100 -<a id="__codelineno-0-175" name="__codelineno-0-175"></a> tuple: A tuple containing the connection URL, username, and password.

3101 -<a id="__codelineno-0-176" name="__codelineno-0-176"></a> """

3102 -<a id="__codelineno-0-177" name="__codelineno-0-177"></a> connector_instance = connector_factory.create(connector_name, connector_name)

3103 -<a id="__codelineno-0-178" name="__codelineno-0-178"></a> connection_successful = connector_instance.verify_connection()

3104 -<a id="__codelineno-0-179" name="__codelineno-0-179"></a> if connection_successful:

3105 -<a id="__codelineno-0-180" name="__codelineno-0-180"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

3106 -<a id="__codelineno-0-181" name="__codelineno-0-181"></a> return (

3107 -<a id="__codelineno-0-182" name="__codelineno-0-182"></a> connection_details.get("connector_url"),

3108 -<a id="__codelineno-0-183" name="__codelineno-0-183"></a> connection_details.get("connector_username"),

3109 -<a id="__codelineno-0-184" name="__codelineno-0-184"></a> connection_details.get("connector_password"),

3110 -<a id="__codelineno-0-185" name="__codelineno-0-185"></a> )

3111 -<a id="__codelineno-0-186" name="__codelineno-0-186"></a> else:

3112 -<a id="__codelineno-0-187" name="__codelineno-0-187"></a> return None, None, None

3113 -<a id="__codelineno-0-188" name="__codelineno-0-188"></a>

3114 -<a id="__codelineno-0-189" name="__codelineno-0-189"></a> def collect_wazuh_agents(self, connection_url: str, wazuh_auth_token: str) -> Optional[List[Dict[str, str]]]:

3115 -<a id="__codelineno-0-190" name="__codelineno-0-190"></a> """

3116 -<a id="__codelineno-0-191" name="__codelineno-0-191"></a> Collects the information of all agents from the Wazuh API.

3117 -<a id="__codelineno-0-192" name="__codelineno-0-192"></a>

3118 -<a id="__codelineno-0-193" name="__codelineno-0-193"></a> Returns:

3119 -<a id="__codelineno-0-194" name="__codelineno-0-194"></a> list: A list containing the information of all Wazuh agents.

3120 -<a id="__codelineno-0-195" name="__codelineno-0-195"></a> """

3121 -<a id="__codelineno-0-196" name="__codelineno-0-196"></a> logger.info("Collecting Wazuh Agents")

3122 -<a id="__codelineno-0-197" name="__codelineno-0-197"></a> try:

3123 -<a id="__codelineno-0-198" name="__codelineno-0-198"></a> headers = {"Authorization": f"Bearer {wazuh_auth_token}"}

3124 -<a id="__codelineno-0-199" name="__codelineno-0-199"></a> limit = 1000

3125 -<a id="__codelineno-0-200" name="__codelineno-0-200"></a> agents_collected = requests.get(

3126 -<a id="__codelineno-0-201" name="__codelineno-0-201"></a> f"{connection_url}/agents?limit={limit}",

3127 -<a id="__codelineno-0-202" name="__codelineno-0-202"></a> headers=headers,

3128 -<a id="__codelineno-0-203" name="__codelineno-0-203"></a> verify=False,

3129 -<a id="__codelineno-0-204" name="__codelineno-0-204"></a> )

3130 -<a id="__codelineno-0-205" name="__codelineno-0-205"></a> if agents_collected.status_code == 200:

3131 -<a id="__codelineno-0-206" name="__codelineno-0-206"></a> wazuh_agents_list = []

3132 -<a id="__codelineno-0-207" name="__codelineno-0-207"></a> for agent in agents_collected.json()["data"]["affected_items"]:

3133 -<a id="__codelineno-0-208" name="__codelineno-0-208"></a> os_name = agent.get("os", {}).get("name", "Unknown")

3134 -<a id="__codelineno-0-209" name="__codelineno-0-209"></a> last_keep_alive = agent.get("lastKeepAlive", "Unknown")

3135 -<a id="__codelineno-0-210" name="__codelineno-0-210"></a> wazuh_agents_list.append(

3136 -<a id="__codelineno-0-211" name="__codelineno-0-211"></a> {

3137 -<a id="__codelineno-0-212" name="__codelineno-0-212"></a> "agent_id": agent["id"],

3138 -<a id="__codelineno-0-213" name="__codelineno-0-213"></a> "agent_name": agent["name"],

3139 -<a id="__codelineno-0-214" name="__codelineno-0-214"></a> "agent_ip": agent["ip"],

3140 -<a id="__codelineno-0-215" name="__codelineno-0-215"></a> "agent_os": os_name,

3141 -<a id="__codelineno-0-216" name="__codelineno-0-216"></a> "agent_last_seen": last_keep_alive,

3142 -<a id="__codelineno-0-217" name="__codelineno-0-217"></a> },

3143 -<a id="__codelineno-0-218" name="__codelineno-0-218"></a> )

3144 -<a id="__codelineno-0-219" name="__codelineno-0-219"></a> logger.info(f"Collected Wazuh Agent: {agent['name']}")

3145 -<a id="__codelineno-0-220" name="__codelineno-0-220"></a> return wazuh_agents_list

3146 -<a id="__codelineno-0-221" name="__codelineno-0-221"></a> else:

3147 -<a id="__codelineno-0-222" name="__codelineno-0-222"></a> return None

3148 -<a id="__codelineno-0-223" name="__codelineno-0-223"></a> except Exception as e:

3149 -<a id="__codelineno-0-224" name="__codelineno-0-224"></a> logger.error(f"Failed to collect Wazuh Agents: {e}")

3150 -<a id="__codelineno-0-225" name="__codelineno-0-225"></a> return None

3151 -<a id="__codelineno-0-226" name="__codelineno-0-226"></a>

3152 -<a id="__codelineno-0-227" name="__codelineno-0-227"></a> def sync_agents(self) -> Dict[str, Union[str, bool, List[Dict[str, str]]]]:

3153 -<a id="__codelineno-0-228" name="__codelineno-0-228"></a> (

3154 -<a id="__codelineno-0-229" name="__codelineno-0-229"></a> connection_url,

3155 -<a id="__codelineno-0-230" name="__codelineno-0-230"></a> connection_username,

3156 -<a id="__codelineno-0-231" name="__codelineno-0-231"></a> connection_password,

3157 -<a id="__codelineno-0-232" name="__codelineno-0-232"></a> ) = self.collect_wazuh_details("Wazuh-Manager")

3158 -<a id="__codelineno-0-233" name="__codelineno-0-233"></a> if connection_url is None:

3159 -<a id="__codelineno-0-234" name="__codelineno-0-234"></a> return {

3160 -<a id="__codelineno-0-235" name="__codelineno-0-235"></a> "message": "Failed to get Wazuh-Manager API details",

3161 -<a id="__codelineno-0-236" name="__codelineno-0-236"></a> "success": False,

3162 -<a id="__codelineno-0-237" name="__codelineno-0-237"></a> }

3163 -<a id="__codelineno-0-238" name="__codelineno-0-238"></a>

3164 -<a id="__codelineno-0-239" name="__codelineno-0-239"></a> wazuh_manager_connector = WazuhManagerConnector("Wazuh-Manager")

3165 -<a id="__codelineno-0-240" name="__codelineno-0-240"></a> wazuh_auth_token = wazuh_manager_connector.get_auth_token()

3166 -<a id="__codelineno-0-241" name="__codelineno-0-241"></a> if wazuh_auth_token is None:

3167 -<a id="__codelineno-0-242" name="__codelineno-0-242"></a> return {

3168 -<a id="__codelineno-0-243" name="__codelineno-0-243"></a> "message": "Failed to get Wazuh-Manager API Auth Token",

3169 -<a id="__codelineno-0-244" name="__codelineno-0-244"></a> "success": False,

3170 -<a id="__codelineno-0-245" name="__codelineno-0-245"></a> }

3171 -<a id="__codelineno-0-246" name="__codelineno-0-246"></a>

3172 -<a id="__codelineno-0-247" name="__codelineno-0-247"></a> wazuh_agents_list = self.collect_wazuh_agents(connection_url, wazuh_auth_token)

3173 -<a id="__codelineno-0-248" name="__codelineno-0-248"></a> if wazuh_agents_list is None:

3174 -<a id="__codelineno-0-249" name="__codelineno-0-249"></a> return {

3175 -<a id="__codelineno-0-250" name="__codelineno-0-250"></a> "message": "Failed to collect Wazuh-Manager Agents",

3176 -<a id="__codelineno-0-251" name="__codelineno-0-251"></a> "success": False,

3177 -<a id="__codelineno-0-252" name="__codelineno-0-252"></a> }

3178 -<a id="__codelineno-0-253" name="__codelineno-0-253"></a>

3179 -<a id="__codelineno-0-254" name="__codelineno-0-254"></a> logger.info(f"Collected {wazuh_agents_list} Wazuh Agents")

3180 -<a id="__codelineno-0-255" name="__codelineno-0-255"></a>

3181 -<a id="__codelineno-0-256" name="__codelineno-0-256"></a> agents_added_list = []

3182 -<a id="__codelineno-0-257" name="__codelineno-0-257"></a> for agent in wazuh_agents_list:

3183 -<a id="__codelineno-0-258" name="__codelineno-0-258"></a> agent_info = self.agent_service.get_agent(agent["agent_id"])

3184 -<a id="__codelineno-0-259" name="__codelineno-0-259"></a> logger.info(f"Agent info: {agent_info}")

3185 -<a id="__codelineno-0-260" name="__codelineno-0-260"></a> if "message" in agent_info:

3186 -<a id="__codelineno-0-261" name="__codelineno-0-261"></a> self.agent_service.create_agent(agent)

3187 -<a id="__codelineno-0-262" name="__codelineno-0-262"></a> agents_added_list.append(agent)

3188 -<a id="__codelineno-0-263" name="__codelineno-0-263"></a>

3189 -<a id="__codelineno-0-264" name="__codelineno-0-264"></a> return {

3190 -<a id="__codelineno-0-265" name="__codelineno-0-265"></a> "message": "Successfully synced agents.",

3191 -<a id="__codelineno-0-266" name="__codelineno-0-266"></a> "success": True,

3192 -<a id="__codelineno-0-267" name="__codelineno-0-267"></a> "agents_added": agents_added_list,

3193 -<a id="__codelineno-0-268" name="__codelineno-0-268"></a> }

3106 +<a id="__codelineno-0-174" name="__codelineno-0-174"></a> def collect_wazuh_details(

3107 +<a id="__codelineno-0-175" name="__codelineno-0-175"></a> self,

3108 +<a id="__codelineno-0-176" name="__codelineno-0-176"></a> connector_name: str,

3109 +<a id="__codelineno-0-177" name="__codelineno-0-177"></a> ) -> Tuple[Optional[str], Optional[str], Optional[str]]:

3110 +<a id="__codelineno-0-178" name="__codelineno-0-178"></a> """

3111 +<a id="__codelineno-0-179" name="__codelineno-0-179"></a> Collects the information of all Wazuh API credentials using the WazuhIndexerConnector class details.

3112 +<a id="__codelineno-0-180" name="__codelineno-0-180"></a>

3113 +<a id="__codelineno-0-181" name="__codelineno-0-181"></a> Returns:

3114 +<a id="__codelineno-0-182" name="__codelineno-0-182"></a> tuple: A tuple containing the connection URL, username, and password.

3115 +<a id="__codelineno-0-183" name="__codelineno-0-183"></a> """

3116 +<a id="__codelineno-0-184" name="__codelineno-0-184"></a> connector_instance = connector_factory.create(connector_name, connector_name)

3117 +<a id="__codelineno-0-185" name="__codelineno-0-185"></a> connection_successful = connector_instance.verify_connection()

3118 +<a id="__codelineno-0-186" name="__codelineno-0-186"></a> if connection_successful:

3119 +<a id="__codelineno-0-187" name="__codelineno-0-187"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

3120 +<a id="__codelineno-0-188" name="__codelineno-0-188"></a> return (

3121 +<a id="__codelineno-0-189" name="__codelineno-0-189"></a> connection_details.get("connector_url"),

3122 +<a id="__codelineno-0-190" name="__codelineno-0-190"></a> connection_details.get("connector_username"),

3123 +<a id="__codelineno-0-191" name="__codelineno-0-191"></a> connection_details.get("connector_password"),

3124 +<a id="__codelineno-0-192" name="__codelineno-0-192"></a> )

3125 +<a id="__codelineno-0-193" name="__codelineno-0-193"></a> else:

3126 +<a id="__codelineno-0-194" name="__codelineno-0-194"></a> return None, None, None

3127 +<a id="__codelineno-0-195" name="__codelineno-0-195"></a>

3128 +<a id="__codelineno-0-196" name="__codelineno-0-196"></a> def collect_wazuh_agents(

3129 +<a id="__codelineno-0-197" name="__codelineno-0-197"></a> self,

3130 +<a id="__codelineno-0-198" name="__codelineno-0-198"></a> connection_url: str,

3131 +<a id="__codelineno-0-199" name="__codelineno-0-199"></a> wazuh_auth_token: str,

3132 +<a id="__codelineno-0-200" name="__codelineno-0-200"></a> ) -> Optional[List[Dict[str, str]]]:

3133 +<a id="__codelineno-0-201" name="__codelineno-0-201"></a> """

3134 +<a id="__codelineno-0-202" name="__codelineno-0-202"></a> Collects the information of all agents from the Wazuh API.

3135 +<a id="__codelineno-0-203" name="__codelineno-0-203"></a>

3136 +<a id="__codelineno-0-204" name="__codelineno-0-204"></a> Returns:

3137 +<a id="__codelineno-0-205" name="__codelineno-0-205"></a> list: A list containing the information of all Wazuh agents.

3138 +<a id="__codelineno-0-206" name="__codelineno-0-206"></a> """

3139 +<a id="__codelineno-0-207" name="__codelineno-0-207"></a> logger.info("Collecting Wazuh Agents")

3140 +<a id="__codelineno-0-208" name="__codelineno-0-208"></a> try:

3141 +<a id="__codelineno-0-209" name="__codelineno-0-209"></a> headers = {"Authorization": f"Bearer {wazuh_auth_token}"}

3142 +<a id="__codelineno-0-210" name="__codelineno-0-210"></a> limit = 1000

3143 +<a id="__codelineno-0-211" name="__codelineno-0-211"></a> agents_collected = requests.get(

3144 +<a id="__codelineno-0-212" name="__codelineno-0-212"></a> f"{connection_url}/agents?limit={limit}",

3145 +<a id="__codelineno-0-213" name="__codelineno-0-213"></a> headers=headers,

3146 +<a id="__codelineno-0-214" name="__codelineno-0-214"></a> verify=False,

3147 +<a id="__codelineno-0-215" name="__codelineno-0-215"></a> )

3148 +<a id="__codelineno-0-216" name="__codelineno-0-216"></a> if agents_collected.status_code == 200:

3149 +<a id="__codelineno-0-217" name="__codelineno-0-217"></a> wazuh_agents_list = []

3150 +<a id="__codelineno-0-218" name="__codelineno-0-218"></a> for agent in agents_collected.json()["data"]["affected_items"]:

3151 +<a id="__codelineno-0-219" name="__codelineno-0-219"></a> os_name = agent.get("os", {}).get("name", "Unknown")

3152 +<a id="__codelineno-0-220" name="__codelineno-0-220"></a> last_keep_alive = agent.get("lastKeepAlive", "Unknown")

3153 +<a id="__codelineno-0-221" name="__codelineno-0-221"></a> wazuh_agents_list.append(

3154 +<a id="__codelineno-0-222" name="__codelineno-0-222"></a> {

3155 +<a id="__codelineno-0-223" name="__codelineno-0-223"></a> "agent_id": agent["id"],

3156 +<a id="__codelineno-0-224" name="__codelineno-0-224"></a> "agent_name": agent["name"],

3157 +<a id="__codelineno-0-225" name="__codelineno-0-225"></a> "agent_ip": agent["ip"],

3158 +<a id="__codelineno-0-226" name="__codelineno-0-226"></a> "agent_os": os_name,

3159 +<a id="__codelineno-0-227" name="__codelineno-0-227"></a> "agent_last_seen": last_keep_alive,

3160 +<a id="__codelineno-0-228" name="__codelineno-0-228"></a> },

3161 +<a id="__codelineno-0-229" name="__codelineno-0-229"></a> )

3162 +<a id="__codelineno-0-230" name="__codelineno-0-230"></a> logger.info(f"Collected Wazuh Agent: {agent['name']}")

3163 +<a id="__codelineno-0-231" name="__codelineno-0-231"></a> return wazuh_agents_list

3164 +<a id="__codelineno-0-232" name="__codelineno-0-232"></a> else:

3165 +<a id="__codelineno-0-233" name="__codelineno-0-233"></a> return None

3166 +<a id="__codelineno-0-234" name="__codelineno-0-234"></a> except Exception as e:

3167 +<a id="__codelineno-0-235" name="__codelineno-0-235"></a> logger.error(f"Failed to collect Wazuh Agents: {e}")

3168 +<a id="__codelineno-0-236" name="__codelineno-0-236"></a> return None

3169 +<a id="__codelineno-0-237" name="__codelineno-0-237"></a>

3170 +<a id="__codelineno-0-238" name="__codelineno-0-238"></a> def sync_agents(self) -> Dict[str, Union[str, bool, List[Dict[str, str]]]]:

3171 +<a id="__codelineno-0-239" name="__codelineno-0-239"></a> (

3172 +<a id="__codelineno-0-240" name="__codelineno-0-240"></a> connection_url,

3173 +<a id="__codelineno-0-241" name="__codelineno-0-241"></a> connection_username,

3174 +<a id="__codelineno-0-242" name="__codelineno-0-242"></a> connection_password,

3175 +<a id="__codelineno-0-243" name="__codelineno-0-243"></a> ) = self.collect_wazuh_details("Wazuh-Manager")

3176 +<a id="__codelineno-0-244" name="__codelineno-0-244"></a> if connection_url is None:

3177 +<a id="__codelineno-0-245" name="__codelineno-0-245"></a> return {

3178 +<a id="__codelineno-0-246" name="__codelineno-0-246"></a> "message": "Failed to get Wazuh-Manager API details",

3179 +<a id="__codelineno-0-247" name="__codelineno-0-247"></a> "success": False,

3180 +<a id="__codelineno-0-248" name="__codelineno-0-248"></a> }

3181 +<a id="__codelineno-0-249" name="__codelineno-0-249"></a>

3182 +<a id="__codelineno-0-250" name="__codelineno-0-250"></a> wazuh_manager_connector = WazuhManagerConnector("Wazuh-Manager")

3183 +<a id="__codelineno-0-251" name="__codelineno-0-251"></a> wazuh_auth_token = wazuh_manager_connector.get_auth_token()

3184 +<a id="__codelineno-0-252" name="__codelineno-0-252"></a> if wazuh_auth_token is None:

3185 +<a id="__codelineno-0-253" name="__codelineno-0-253"></a> return {

3186 +<a id="__codelineno-0-254" name="__codelineno-0-254"></a> "message": "Failed to get Wazuh-Manager API Auth Token",

3187 +<a id="__codelineno-0-255" name="__codelineno-0-255"></a> "success": False,

3188 +<a id="__codelineno-0-256" name="__codelineno-0-256"></a> }

3189 +<a id="__codelineno-0-257" name="__codelineno-0-257"></a>

3190 +<a id="__codelineno-0-258" name="__codelineno-0-258"></a> wazuh_agents_list = self.collect_wazuh_agents(connection_url, wazuh_auth_token)

3191 +<a id="__codelineno-0-259" name="__codelineno-0-259"></a> if wazuh_agents_list is None:

3192 +<a id="__codelineno-0-260" name="__codelineno-0-260"></a> return {

3193 +<a id="__codelineno-0-261" name="__codelineno-0-261"></a> "message": "Failed to collect Wazuh-Manager Agents",

3194 +<a id="__codelineno-0-262" name="__codelineno-0-262"></a> "success": False,

3195 +<a id="__codelineno-0-263" name="__codelineno-0-263"></a> }

3196 +<a id="__codelineno-0-264" name="__codelineno-0-264"></a>

3197 +<a id="__codelineno-0-265" name="__codelineno-0-265"></a> logger.info(f"Collected {wazuh_agents_list} Wazuh Agents")

3198 +<a id="__codelineno-0-266" name="__codelineno-0-266"></a>

3199 +<a id="__codelineno-0-267" name="__codelineno-0-267"></a> agents_added_list = []

3200 +<a id="__codelineno-0-268" name="__codelineno-0-268"></a> for agent in wazuh_agents_list:

3201 +<a id="__codelineno-0-269" name="__codelineno-0-269"></a> agent_info = self.agent_service.get_agent(agent["agent_id"])

3202 +<a id="__codelineno-0-270" name="__codelineno-0-270"></a> logger.info(f"Agent info: {agent_info}")

3203 +<a id="__codelineno-0-271" name="__codelineno-0-271"></a> if "message" in agent_info:

3204 +<a id="__codelineno-0-272" name="__codelineno-0-272"></a> self.agent_service.create_agent(agent)

3205 +<a id="__codelineno-0-273" name="__codelineno-0-273"></a> agents_added_list.append(agent)

3206 +<a id="__codelineno-0-274" name="__codelineno-0-274"></a>

3207 +<a id="__codelineno-0-275" name="__codelineno-0-275"></a> return {

3208 +<a id="__codelineno-0-276" name="__codelineno-0-276"></a> "message": "Successfully synced agents.",

3209 +<a id="__codelineno-0-277" name="__codelineno-0-277"></a> "success": True,

3210 +<a id="__codelineno-0-278" name="__codelineno-0-278"></a> "agents_added": agents_added_list,

3211 +<a id="__codelineno-0-279" name="__codelineno-0-279"></a> }

3212 </code></pre></div></td></tr></table></div>

3213 </details>

3214

@@ -3244,14 +3262,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

3262

3263 <details class="quote">

3264 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

3247 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-189">189</a>

3248 -<a href="#__codelineno-0-190">190</a>

3249 -<a href="#__codelineno-0-191">191</a>

3250 -<a href="#__codelineno-0-192">192</a>

3251 -<a href="#__codelineno-0-193">193</a>

3252 -<a href="#__codelineno-0-194">194</a>

3253 -<a href="#__codelineno-0-195">195</a>

3254 -<a href="#__codelineno-0-196">196</a>

3265 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-196">196</a>

3266 <a href="#__codelineno-0-197">197</a>

3267 <a href="#__codelineno-0-198">198</a>

3268 <a href="#__codelineno-0-199">199</a>

@@ -3280,43 +3291,58 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

3291 <a href="#__codelineno-0-222">222</a>

3292 <a href="#__codelineno-0-223">223</a>

3293 <a href="#__codelineno-0-224">224</a>

3283 -<a href="#__codelineno-0-225">225</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-189" name="__codelineno-0-189"></a>def collect_wazuh_agents(self, connection_url: str, wazuh_auth_token: str) -> Optional[List[Dict[str, str]]]:

3284 -<a id="__codelineno-0-190" name="__codelineno-0-190"></a> """

3285 -<a id="__codelineno-0-191" name="__codelineno-0-191"></a> Collects the information of all agents from the Wazuh API.

3286 -<a id="__codelineno-0-192" name="__codelineno-0-192"></a>

3287 -<a id="__codelineno-0-193" name="__codelineno-0-193"></a> Returns:

3288 -<a id="__codelineno-0-194" name="__codelineno-0-194"></a> list: A list containing the information of all Wazuh agents.

3289 -<a id="__codelineno-0-195" name="__codelineno-0-195"></a> """

3290 -<a id="__codelineno-0-196" name="__codelineno-0-196"></a> logger.info("Collecting Wazuh Agents")

3291 -<a id="__codelineno-0-197" name="__codelineno-0-197"></a> try:

3292 -<a id="__codelineno-0-198" name="__codelineno-0-198"></a> headers = {"Authorization": f"Bearer {wazuh_auth_token}"}

3293 -<a id="__codelineno-0-199" name="__codelineno-0-199"></a> limit = 1000

3294 -<a id="__codelineno-0-200" name="__codelineno-0-200"></a> agents_collected = requests.get(

3295 -<a id="__codelineno-0-201" name="__codelineno-0-201"></a> f"{connection_url}/agents?limit={limit}",

3296 -<a id="__codelineno-0-202" name="__codelineno-0-202"></a> headers=headers,

3297 -<a id="__codelineno-0-203" name="__codelineno-0-203"></a> verify=False,

3298 -<a id="__codelineno-0-204" name="__codelineno-0-204"></a> )

3299 -<a id="__codelineno-0-205" name="__codelineno-0-205"></a> if agents_collected.status_code == 200:

3300 -<a id="__codelineno-0-206" name="__codelineno-0-206"></a> wazuh_agents_list = []

3301 -<a id="__codelineno-0-207" name="__codelineno-0-207"></a> for agent in agents_collected.json()["data"]["affected_items"]:

3302 -<a id="__codelineno-0-208" name="__codelineno-0-208"></a> os_name = agent.get("os", {}).get("name", "Unknown")

3303 -<a id="__codelineno-0-209" name="__codelineno-0-209"></a> last_keep_alive = agent.get("lastKeepAlive", "Unknown")

3304 -<a id="__codelineno-0-210" name="__codelineno-0-210"></a> wazuh_agents_list.append(

3305 -<a id="__codelineno-0-211" name="__codelineno-0-211"></a> {

3306 -<a id="__codelineno-0-212" name="__codelineno-0-212"></a> "agent_id": agent["id"],

3307 -<a id="__codelineno-0-213" name="__codelineno-0-213"></a> "agent_name": agent["name"],

3308 -<a id="__codelineno-0-214" name="__codelineno-0-214"></a> "agent_ip": agent["ip"],

3309 -<a id="__codelineno-0-215" name="__codelineno-0-215"></a> "agent_os": os_name,

3310 -<a id="__codelineno-0-216" name="__codelineno-0-216"></a> "agent_last_seen": last_keep_alive,

3311 -<a id="__codelineno-0-217" name="__codelineno-0-217"></a> },

3312 -<a id="__codelineno-0-218" name="__codelineno-0-218"></a> )

3313 -<a id="__codelineno-0-219" name="__codelineno-0-219"></a> logger.info(f"Collected Wazuh Agent: {agent['name']}")

3314 -<a id="__codelineno-0-220" name="__codelineno-0-220"></a> return wazuh_agents_list

3315 -<a id="__codelineno-0-221" name="__codelineno-0-221"></a> else:

3316 -<a id="__codelineno-0-222" name="__codelineno-0-222"></a> return None

3317 -<a id="__codelineno-0-223" name="__codelineno-0-223"></a> except Exception as e:

3318 -<a id="__codelineno-0-224" name="__codelineno-0-224"></a> logger.error(f"Failed to collect Wazuh Agents: {e}")

3319 -<a id="__codelineno-0-225" name="__codelineno-0-225"></a> return None

3294 +<a href="#__codelineno-0-225">225</a>

3295 +<a href="#__codelineno-0-226">226</a>

3296 +<a href="#__codelineno-0-227">227</a>

3297 +<a href="#__codelineno-0-228">228</a>

3298 +<a href="#__codelineno-0-229">229</a>

3299 +<a href="#__codelineno-0-230">230</a>

3300 +<a href="#__codelineno-0-231">231</a>

3301 +<a href="#__codelineno-0-232">232</a>

3302 +<a href="#__codelineno-0-233">233</a>

3303 +<a href="#__codelineno-0-234">234</a>

3304 +<a href="#__codelineno-0-235">235</a>

3305 +<a href="#__codelineno-0-236">236</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-196" name="__codelineno-0-196"></a>def collect_wazuh_agents(

3306 +<a id="__codelineno-0-197" name="__codelineno-0-197"></a> self,

3307 +<a id="__codelineno-0-198" name="__codelineno-0-198"></a> connection_url: str,

3308 +<a id="__codelineno-0-199" name="__codelineno-0-199"></a> wazuh_auth_token: str,

3309 +<a id="__codelineno-0-200" name="__codelineno-0-200"></a>) -> Optional[List[Dict[str, str]]]:

3310 +<a id="__codelineno-0-201" name="__codelineno-0-201"></a> """

3311 +<a id="__codelineno-0-202" name="__codelineno-0-202"></a> Collects the information of all agents from the Wazuh API.

3312 +<a id="__codelineno-0-203" name="__codelineno-0-203"></a>

3313 +<a id="__codelineno-0-204" name="__codelineno-0-204"></a> Returns:

3314 +<a id="__codelineno-0-205" name="__codelineno-0-205"></a> list: A list containing the information of all Wazuh agents.

3315 +<a id="__codelineno-0-206" name="__codelineno-0-206"></a> """

3316 +<a id="__codelineno-0-207" name="__codelineno-0-207"></a> logger.info("Collecting Wazuh Agents")

3317 +<a id="__codelineno-0-208" name="__codelineno-0-208"></a> try:

3318 +<a id="__codelineno-0-209" name="__codelineno-0-209"></a> headers = {"Authorization": f"Bearer {wazuh_auth_token}"}

3319 +<a id="__codelineno-0-210" name="__codelineno-0-210"></a> limit = 1000

3320 +<a id="__codelineno-0-211" name="__codelineno-0-211"></a> agents_collected = requests.get(

3321 +<a id="__codelineno-0-212" name="__codelineno-0-212"></a> f"{connection_url}/agents?limit={limit}",

3322 +<a id="__codelineno-0-213" name="__codelineno-0-213"></a> headers=headers,

3323 +<a id="__codelineno-0-214" name="__codelineno-0-214"></a> verify=False,

3324 +<a id="__codelineno-0-215" name="__codelineno-0-215"></a> )

3325 +<a id="__codelineno-0-216" name="__codelineno-0-216"></a> if agents_collected.status_code == 200:

3326 +<a id="__codelineno-0-217" name="__codelineno-0-217"></a> wazuh_agents_list = []

3327 +<a id="__codelineno-0-218" name="__codelineno-0-218"></a> for agent in agents_collected.json()["data"]["affected_items"]:

3328 +<a id="__codelineno-0-219" name="__codelineno-0-219"></a> os_name = agent.get("os", {}).get("name", "Unknown")

3329 +<a id="__codelineno-0-220" name="__codelineno-0-220"></a> last_keep_alive = agent.get("lastKeepAlive", "Unknown")

3330 +<a id="__codelineno-0-221" name="__codelineno-0-221"></a> wazuh_agents_list.append(

3331 +<a id="__codelineno-0-222" name="__codelineno-0-222"></a> {

3332 +<a id="__codelineno-0-223" name="__codelineno-0-223"></a> "agent_id": agent["id"],

3333 +<a id="__codelineno-0-224" name="__codelineno-0-224"></a> "agent_name": agent["name"],

3334 +<a id="__codelineno-0-225" name="__codelineno-0-225"></a> "agent_ip": agent["ip"],

3335 +<a id="__codelineno-0-226" name="__codelineno-0-226"></a> "agent_os": os_name,

3336 +<a id="__codelineno-0-227" name="__codelineno-0-227"></a> "agent_last_seen": last_keep_alive,

3337 +<a id="__codelineno-0-228" name="__codelineno-0-228"></a> },

3338 +<a id="__codelineno-0-229" name="__codelineno-0-229"></a> )

3339 +<a id="__codelineno-0-230" name="__codelineno-0-230"></a> logger.info(f"Collected Wazuh Agent: {agent['name']}")

3340 +<a id="__codelineno-0-231" name="__codelineno-0-231"></a> return wazuh_agents_list

3341 +<a id="__codelineno-0-232" name="__codelineno-0-232"></a> else:

3342 +<a id="__codelineno-0-233" name="__codelineno-0-233"></a> return None

3343 +<a id="__codelineno-0-234" name="__codelineno-0-234"></a> except Exception as e:

3344 +<a id="__codelineno-0-235" name="__codelineno-0-235"></a> logger.error(f"Failed to collect Wazuh Agents: {e}")

3345 +<a id="__codelineno-0-236" name="__codelineno-0-236"></a> return None

3346 </code></pre></div></td></tr></table></div>

3347 </details>

3348 </div>

@@ -3361,11 +3387,7 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

3387

3388 <details class="quote">

3389 <summary>Source code in <code>app\services\agents\agents.py</code></summary>

3364 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-170">170</a>

3365 -<a href="#__codelineno-0-171">171</a>

3366 -<a href="#__codelineno-0-172">172</a>

3367 -<a href="#__codelineno-0-173">173</a>

3368 -<a href="#__codelineno-0-174">174</a>

3390 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-174">174</a>

3391 <a href="#__codelineno-0-175">175</a>

3392 <a href="#__codelineno-0-176">176</a>

3393 <a href="#__codelineno-0-177">177</a>

@@ -3378,24 +3400,34 @@ and boolean for critical asset. This class inherits from SQLAlchemy's Model clas

3400 <a href="#__codelineno-0-184">184</a>

3401 <a href="#__codelineno-0-185">185</a>

3402 <a href="#__codelineno-0-186">186</a>

3381 -<a href="#__codelineno-0-187">187</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-170" name="__codelineno-0-170"></a>def collect_wazuh_details(self, connector_name: str) -> Tuple[Optional[str], Optional[str], Optional[str]]:

3382 -<a id="__codelineno-0-171" name="__codelineno-0-171"></a> """

3383 -<a id="__codelineno-0-172" name="__codelineno-0-172"></a> Collects the information of all Wazuh API credentials using the WazuhIndexerConnector class details.

3384 -<a id="__codelineno-0-173" name="__codelineno-0-173"></a>

3385 -<a id="__codelineno-0-174" name="__codelineno-0-174"></a> Returns:

3386 -<a id="__codelineno-0-175" name="__codelineno-0-175"></a> tuple: A tuple containing the connection URL, username, and password.

3387 -<a id="__codelineno-0-176" name="__codelineno-0-176"></a> """

3388 -<a id="__codelineno-0-177" name="__codelineno-0-177"></a> connector_instance = connector_factory.create(connector_name, connector_name)

3389 -<a id="__codelineno-0-178" name="__codelineno-0-178"></a> connection_successful = connector_instance.verify_connection()

3390 -<a id="__codelineno-0-179" name="__codelineno-0-179"></a> if connection_successful:

3391 -<a id="__codelineno-0-180" name="__codelineno-0-180"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

3392 -<a id="__codelineno-0-181" name="__codelineno-0-181"></a> return (

3393 -<a id="__codelineno-0-182" name="__codelineno-0-182"></a> connection_details.get("connector_url"),

3394 -<a id="__codelineno-0-183" name="__codelineno-0-183"></a> connection_details.get("connector_username"),

3395 -<a id="__codelineno-0-184" name="__codelineno-0-184"></a> connection_details.get("connector_password"),

3396 -<a id="__codelineno-0-185" name="__codelineno-0-185"></a> )

3397 -<a id="__codelineno-0-186" name="__codelineno-0-186"></a> else:

3398 -<a id="__codelineno-0-187" name="__codelineno-0-187"></a> return None, None, None

3403 +<a href="#__codelineno-0-187">187</a>

3404 +<a href="#__codelineno-0-188">188</a>

3405 +<a href="#__codelineno-0-189">189</a>

3406 +<a href="#__codelineno-0-190">190</a>

3407 +<a href="#__codelineno-0-191">191</a>

3408 +<a href="#__codelineno-0-192">192</a>

3409 +<a href="#__codelineno-0-193">193</a>

3410 +<a href="#__codelineno-0-194">194</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-174" name="__codelineno-0-174"></a>def collect_wazuh_details(

3411 +<a id="__codelineno-0-175" name="__codelineno-0-175"></a> self,

3412 +<a id="__codelineno-0-176" name="__codelineno-0-176"></a> connector_name: str,

3413 +<a id="__codelineno-0-177" name="__codelineno-0-177"></a>) -> Tuple[Optional[str], Optional[str], Optional[str]]:

3414 +<a id="__codelineno-0-178" name="__codelineno-0-178"></a> """

3415 +<a id="__codelineno-0-179" name="__codelineno-0-179"></a> Collects the information of all Wazuh API credentials using the WazuhIndexerConnector class details.

3416 +<a id="__codelineno-0-180" name="__codelineno-0-180"></a>

3417 +<a id="__codelineno-0-181" name="__codelineno-0-181"></a> Returns:

3418 +<a id="__codelineno-0-182" name="__codelineno-0-182"></a> tuple: A tuple containing the connection URL, username, and password.

3419 +<a id="__codelineno-0-183" name="__codelineno-0-183"></a> """

3420 +<a id="__codelineno-0-184" name="__codelineno-0-184"></a> connector_instance = connector_factory.create(connector_name, connector_name)

3421 +<a id="__codelineno-0-185" name="__codelineno-0-185"></a> connection_successful = connector_instance.verify_connection()

3422 +<a id="__codelineno-0-186" name="__codelineno-0-186"></a> if connection_successful:

3423 +<a id="__codelineno-0-187" name="__codelineno-0-187"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

3424 +<a id="__codelineno-0-188" name="__codelineno-0-188"></a> return (

3425 +<a id="__codelineno-0-189" name="__codelineno-0-189"></a> connection_details.get("connector_url"),

3426 +<a id="__codelineno-0-190" name="__codelineno-0-190"></a> connection_details.get("connector_username"),

3427 +<a id="__codelineno-0-191" name="__codelineno-0-191"></a> connection_details.get("connector_password"),

3428 +<a id="__codelineno-0-192" name="__codelineno-0-192"></a> )

3429 +<a id="__codelineno-0-193" name="__codelineno-0-193"></a> else:

3430 +<a id="__codelineno-0-194" name="__codelineno-0-194"></a> return None, None, None

3431 </code></pre></div></td></tr></table></div>

3432 </details>

3433 </div>

backend/site/alerts/index.html

+236 -236

@@ -589,10 +589,7 @@ available.

589

590 <details class="quote">

591 <summary>Source code in <code>app\routes\alerts.py</code></summary>

592 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-6"> 6</a>

593 -<a href="#__codelineno-0-7"> 7</a>

594 -<a href="#__codelineno-0-8"> 8</a>

595 -<a href="#__codelineno-0-9"> 9</a>

592 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-9"> 9</a>

593 <a href="#__codelineno-0-10">10</a>

594 <a href="#__codelineno-0-11">11</a>

595 <a href="#__codelineno-0-12">12</a>

@@ -604,22 +601,25 @@ available.

601 <a href="#__codelineno-0-18">18</a>

602 <a href="#__codelineno-0-19">19</a>

603 <a href="#__codelineno-0-20">20</a>

607 -<a href="#__codelineno-0-21">21</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-6" name="__codelineno-0-6"></a>@bp.route("/alerts", methods=["GET"])

608 -<a id="__codelineno-0-7" name="__codelineno-0-7"></a>def get_alerts() -> jsonify:

609 -<a id="__codelineno-0-8" name="__codelineno-0-8"></a> """

610 -<a id="__codelineno-0-9" name="__codelineno-0-9"></a> Retrieves all alerts from the AlertsService.

611 -<a id="__codelineno-0-10" name="__codelineno-0-10"></a>

612 -<a id="__codelineno-0-11" name="__codelineno-0-11"></a> This endpoint retrieves all available alerts from the AlertsService. It does this by creating an instance of

613 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a> the AlertsService class and calling its `collect_alerts` method. The result is a list of all alerts currently

614 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a> available.

615 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a>

616 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a> Returns:

617 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> jsonify: A JSON response containing a list of alerts. Each item in the list is a dictionary representing an alert,

618 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> containing all its associated data.

619 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> """

620 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> service = AlertsService()

621 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> alerts = service.collect_alerts()

622 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a> return jsonify(alerts)

604 +<a href="#__codelineno-0-21">21</a>

605 +<a href="#__codelineno-0-22">22</a>

606 +<a href="#__codelineno-0-23">23</a>

607 +<a href="#__codelineno-0-24">24</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-9" name="__codelineno-0-9"></a>@bp.route("/alerts", methods=["GET"])

608 +<a id="__codelineno-0-10" name="__codelineno-0-10"></a>def get_alerts() -> jsonify:

609 +<a id="__codelineno-0-11" name="__codelineno-0-11"></a> """

610 +<a id="__codelineno-0-12" name="__codelineno-0-12"></a> Retrieves all alerts from the AlertsService.

611 +<a id="__codelineno-0-13" name="__codelineno-0-13"></a>

612 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> This endpoint retrieves all available alerts from the AlertsService. It does this by creating an instance of

613 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> the AlertsService class and calling its `collect_alerts` method. The result is a list of all alerts currently

614 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a> available.

615 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a>

616 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> Returns:

617 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a> jsonify: A JSON response containing a list of alerts. Each item in the list is a dictionary representing an alert,

618 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> containing all its associated data.

619 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> """

620 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> service = AlertsService()

621 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> alerts = service.collect_alerts()

622 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a> return jsonify(alerts)

623 </code></pre></div></td></tr></table></div>

624 </details>

625 </div>

@@ -729,8 +729,7 @@ available.

729

730 <details class="quote">

731 <summary>Source code in <code>app\services\WazuhIndexer\alerts.py</code></summary>

732 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-8"> 8</a>

733 -<a href="#__codelineno-0-9"> 9</a>

732 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-9"> 9</a>

733 <a href="#__codelineno-0-10"> 10</a>

734 <a href="#__codelineno-0-11"> 11</a>

735 <a href="#__codelineno-0-12"> 12</a>

@@ -864,142 +863,143 @@ available.

863 <a href="#__codelineno-0-140">140</a>

864 <a href="#__codelineno-0-141">141</a>

865 <a href="#__codelineno-0-142">142</a>

867 -<a href="#__codelineno-0-143">143</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-8" name="__codelineno-0-8"></a>class AlertsService:

868 -<a id="__codelineno-0-9" name="__codelineno-0-9"></a> """

869 -<a id="__codelineno-0-10" name="__codelineno-0-10"></a> A service class that encapsulates the logic for pulling alerts from the Wazuh-Indexer.

870 -<a id="__codelineno-0-11" name="__codelineno-0-11"></a>

871 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a> Attributes:

872 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a> connector_url (str): The url to the Wazuh-Indexer.

873 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a> connector_username (str): The username for the Wazuh-Indexer.

874 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a> connector_password (str): The password for the Wazuh-Indexer.

875 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> es (Elasticsearch): The Elasticsearch client.

876 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> """

877 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a>

878 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> SKIP_INDEX_NAMES: Dict[str, bool] = {

879 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> "wazuh-statistics": True,

880 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a> "wazuh-monitoring": True,

881 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> }

882 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a>

883 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> def __init__(self):

884 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> """

885 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> Initializes the service by collecting Wazuh-Indexer details and creating an Elasticsearch client.

886 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> """

887 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> (

888 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> self.connector_url,

889 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> self.connector_username,

890 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> self.connector_password,

891 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> ) = UniversalService().collect_wazuhindexer_details("Wazuh-Indexer")

892 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> self.es = Elasticsearch(

893 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> [self.connector_url],

894 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> http_auth=(self.connector_username, self.connector_password),

895 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> verify_certs=False,

896 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> timeout=15,

897 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a> max_retries=10,

898 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> retry_on_timeout=False,

899 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> )

900 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a>

901 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> def is_index_skipped(self, index_name: str) -> bool:

902 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> """

903 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> Checks whether the given index name should be skipped.

904 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a>

905 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> Args:

906 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> index_name (str): The name of the index.

907 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a>

908 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> Returns:

909 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> bool: True if the index should be skipped, False otherwise.

910 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> """

911 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> for skipped in self.SKIP_INDEX_NAMES:

912 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> if index_name.startswith(skipped):

913 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> return True

914 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> return False

915 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a>

916 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> def collect_alerts(self) -> Dict[str, object]:

917 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> """

918 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> Collects the alerts from the Wazuh-Indexer where the index name starts with "wazuh_"

919 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a> and is not in the SKIP_INDEX_NAMES list.

920 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> Returns the 10 previous alerts based on the `timestamp_utc` field.

921 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a>

922 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> Returns:

923 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> Dict[str, object]: A dictionary containing success status and alerts or an error message.

924 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> """

925 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> if not all(

926 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> [self.connector_url, self.connector_username, self.connector_password],

927 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> ):

928 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> return {

929 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "message": "Failed to collect Wazuh-Indexer details",

930 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "success": False,

931 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> }

932 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a>

933 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> indices_list = UniversalService().collect_indices()

934 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> if not indices_list["success"]:

935 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a> return {"message": "Failed to collect indices", "success": False}

936 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a>

937 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> alerts_summary = []

938 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> for index_name in indices_list["indices_list"]:

939 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> if not index_name.startswith("wazuh_") or self.is_index_skipped(index_name):

940 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> continue

941 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a>

942 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> alerts = self._collect_alerts(index_name)

943 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> if alerts["success"] and len(alerts["alerts"]) > 0:

944 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a> alerts_summary.append(

945 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> {

946 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> "index_name": index_name,

947 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> "total_alerts": len(alerts["alerts"]),

948 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> "last_10_alerts": alerts["alerts"],

949 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> },

950 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> )

951 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a>

952 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> return {

953 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> "message": "Successfully collected alerts",

954 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> "success": True,

955 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> "alerts_summary": alerts_summary,

956 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a> }

957 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a>

958 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> def _collect_alerts(self, index_name: str) -> Dict[str, object]:

959 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a> """

960 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a> Elasticsearch query to get the 10 most recent alerts where the `rule_level` is 12 or higher or the

961 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> `syslog_level` field is `ALERT` and return the results in descending order by the `timestamp_utc` field.

962 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a>

963 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Args:

964 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a> index_name (str): The name of the index to query.

965 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a>

966 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a> Returns:

967 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> Dict[str, object]: A dictionary containing success status and alerts or an error message.

968 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> """

969 -<a id="__codelineno-0-110" name="__codelineno-0-110"></a> logger.info(f"Collecting alerts from {index_name}")

970 -<a id="__codelineno-0-111" name="__codelineno-0-111"></a> query = self._build_query()

971 -<a id="__codelineno-0-112" name="__codelineno-0-112"></a> try:

972 -<a id="__codelineno-0-113" name="__codelineno-0-113"></a> alerts = self.es.search(index=index_name, body=query, size=10)

973 -<a id="__codelineno-0-114" name="__codelineno-0-114"></a> alerts_list = [alert for alert in alerts["hits"]["hits"]]

974 -<a id="__codelineno-0-115" name="__codelineno-0-115"></a> return {

975 -<a id="__codelineno-0-116" name="__codelineno-0-116"></a> "message": "Successfully collected alerts",

976 -<a id="__codelineno-0-117" name="__codelineno-0-117"></a> "success": True,

977 -<a id="__codelineno-0-118" name="__codelineno-0-118"></a> "alerts": alerts_list,

978 -<a id="__codelineno-0-119" name="__codelineno-0-119"></a> }

979 -<a id="__codelineno-0-120" name="__codelineno-0-120"></a> except Exception as e:

980 -<a id="__codelineno-0-121" name="__codelineno-0-121"></a> logger.error(f"Failed to collect alerts: {e}")

981 -<a id="__codelineno-0-122" name="__codelineno-0-122"></a> return {"message": "Failed to collect alerts", "success": False}

982 -<a id="__codelineno-0-123" name="__codelineno-0-123"></a>

983 -<a id="__codelineno-0-124" name="__codelineno-0-124"></a> @staticmethod

984 -<a id="__codelineno-0-125" name="__codelineno-0-125"></a> def _build_query() -> Dict[str, object]:

985 -<a id="__codelineno-0-126" name="__codelineno-0-126"></a> """

986 -<a id="__codelineno-0-127" name="__codelineno-0-127"></a> Builds the Elasticsearch query to get the 10 most recent alerts where the `rule_level` is 12 or higher or

987 -<a id="__codelineno-0-128" name="__codelineno-0-128"></a> the `syslog_level` field is `ALERT`.

988 -<a id="__codelineno-0-129" name="__codelineno-0-129"></a>

989 -<a id="__codelineno-0-130" name="__codelineno-0-130"></a> Returns:

990 -<a id="__codelineno-0-131" name="__codelineno-0-131"></a> Dict[str, object]: A dictionary representing the Elasticsearch query.

991 -<a id="__codelineno-0-132" name="__codelineno-0-132"></a> """

992 -<a id="__codelineno-0-133" name="__codelineno-0-133"></a> return {

993 -<a id="__codelineno-0-134" name="__codelineno-0-134"></a> "query": {

994 -<a id="__codelineno-0-135" name="__codelineno-0-135"></a> "bool": {

995 -<a id="__codelineno-0-136" name="__codelineno-0-136"></a> "should": [

996 -<a id="__codelineno-0-137" name="__codelineno-0-137"></a> {"range": {"rule_level": {"gte": 12}}},

997 -<a id="__codelineno-0-138" name="__codelineno-0-138"></a> {"match": {"syslog_level": "ALERT"}},

998 -<a id="__codelineno-0-139" name="__codelineno-0-139"></a> ],

999 -<a id="__codelineno-0-140" name="__codelineno-0-140"></a> },

1000 -<a id="__codelineno-0-141" name="__codelineno-0-141"></a> },

1001 -<a id="__codelineno-0-142" name="__codelineno-0-142"></a> "sort": [{"timestamp_utc": {"order": "desc"}}],

1002 -<a id="__codelineno-0-143" name="__codelineno-0-143"></a> }

866 +<a href="#__codelineno-0-143">143</a>

867 +<a href="#__codelineno-0-144">144</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-9" name="__codelineno-0-9"></a>class AlertsService:

868 +<a id="__codelineno-0-10" name="__codelineno-0-10"></a> """

869 +<a id="__codelineno-0-11" name="__codelineno-0-11"></a> A service class that encapsulates the logic for pulling alerts from the Wazuh-Indexer.

870 +<a id="__codelineno-0-12" name="__codelineno-0-12"></a>

871 +<a id="__codelineno-0-13" name="__codelineno-0-13"></a> Attributes:

872 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> connector_url (str): The url to the Wazuh-Indexer.

873 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> connector_username (str): The username for the Wazuh-Indexer.

874 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a> connector_password (str): The password for the Wazuh-Indexer.

875 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a> es (Elasticsearch): The Elasticsearch client.

876 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> """

877 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a>

878 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> SKIP_INDEX_NAMES: Dict[str, bool] = {

879 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> "wazuh-statistics": True,

880 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> "wazuh-monitoring": True,

881 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> }

882 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a>

883 +<a id="__codelineno-0-25" name="__codelineno-0-25"></a> def __init__(self):

884 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a> """

885 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> Initializes the service by collecting Wazuh-Indexer details and creating an Elasticsearch client.

886 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> """

887 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> (

888 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> self.connector_url,

889 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> self.connector_username,

890 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> self.connector_password,

891 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> ) = UniversalService().collect_wazuhindexer_details("Wazuh-Indexer")

892 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> self.es = Elasticsearch(

893 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> [self.connector_url],

894 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> http_auth=(self.connector_username, self.connector_password),

895 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> verify_certs=False,

896 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> timeout=15,

897 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> max_retries=10,

898 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> retry_on_timeout=False,

899 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> )

900 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a>

901 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> def is_index_skipped(self, index_name: str) -> bool:

902 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> """

903 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> Checks whether the given index name should be skipped.

904 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a>

905 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> Args:

906 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> index_name (str): The name of the index.

907 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a>

908 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> Returns:

909 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> bool: True if the index should be skipped, False otherwise.

910 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> """

911 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> for skipped in self.SKIP_INDEX_NAMES:

912 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> if index_name.startswith(skipped):

913 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a> return True

914 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> return False

915 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a>

916 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a> def collect_alerts(self) -> Dict[str, object]:

917 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> """

918 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> Collects the alerts from the Wazuh-Indexer where the index name starts with "wazuh_"

919 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> and is not in the SKIP_INDEX_NAMES list.

920 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> Returns the 10 previous alerts based on the `timestamp_utc` field.

921 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a>

922 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> Returns:

923 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> Dict[str, object]: A dictionary containing success status and alerts or an error message.

924 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> """

925 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> if not all(

926 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> [self.connector_url, self.connector_username, self.connector_password],

927 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> ):

928 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> return {

929 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "message": "Failed to collect Wazuh-Indexer details",

930 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> "success": False,

931 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> }

932 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a>

933 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> indices_list = UniversalService().collect_indices()

934 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> if not indices_list["success"]:

935 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> return {"message": "Failed to collect indices", "success": False}

936 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a>

937 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> alerts_summary = []

938 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> for index_name in indices_list["indices_list"]:

939 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> if not index_name.startswith("wazuh_") or self.is_index_skipped(index_name):

940 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> continue

941 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a>

942 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> alerts = self._collect_alerts(index_name)

943 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> if alerts["success"] and len(alerts["alerts"]) > 0:

944 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> alerts_summary.append(

945 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> {

946 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> "index_name": index_name,

947 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> "total_alerts": len(alerts["alerts"]),

948 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> "last_10_alerts": alerts["alerts"],

949 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> },

950 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> )

951 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a>

952 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> return {

953 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> "message": "Successfully collected alerts",

954 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> "success": True,

955 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> "alerts_summary": alerts_summary,

956 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> }

957 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a>

958 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> def _collect_alerts(self, index_name: str) -> Dict[str, object]:

959 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a> """

960 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> Elasticsearch query to get the 10 most recent alerts where the `rule_level` is 12 or higher or the

961 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> `syslog_level` field is `ALERT` and return the results in descending order by the `timestamp_utc` field.

962 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a>

963 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a> Args:

964 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> index_name (str): The name of the index to query.

965 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a>

966 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a> Returns:

967 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a> Dict[str, object]: A dictionary containing success status and alerts or an error message.

968 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> """

969 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a> logger.info(f"Collecting alerts from {index_name}")

970 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a> query = self._build_query()

971 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> try:

972 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> alerts = self.es.search(index=index_name, body=query, size=10)

973 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> alerts_list = [alert for alert in alerts["hits"]["hits"]]

974 +<a id="__codelineno-0-116" name="__codelineno-0-116"></a> return {

975 +<a id="__codelineno-0-117" name="__codelineno-0-117"></a> "message": "Successfully collected alerts",

976 +<a id="__codelineno-0-118" name="__codelineno-0-118"></a> "success": True,

977 +<a id="__codelineno-0-119" name="__codelineno-0-119"></a> "alerts": alerts_list,

978 +<a id="__codelineno-0-120" name="__codelineno-0-120"></a> }

979 +<a id="__codelineno-0-121" name="__codelineno-0-121"></a> except Exception as e:

980 +<a id="__codelineno-0-122" name="__codelineno-0-122"></a> logger.error(f"Failed to collect alerts: {e}")

981 +<a id="__codelineno-0-123" name="__codelineno-0-123"></a> return {"message": "Failed to collect alerts", "success": False}

982 +<a id="__codelineno-0-124" name="__codelineno-0-124"></a>

983 +<a id="__codelineno-0-125" name="__codelineno-0-125"></a> @staticmethod

984 +<a id="__codelineno-0-126" name="__codelineno-0-126"></a> def _build_query() -> Dict[str, object]:

985 +<a id="__codelineno-0-127" name="__codelineno-0-127"></a> """

986 +<a id="__codelineno-0-128" name="__codelineno-0-128"></a> Builds the Elasticsearch query to get the 10 most recent alerts where the `rule_level` is 12 or higher or

987 +<a id="__codelineno-0-129" name="__codelineno-0-129"></a> the `syslog_level` field is `ALERT`.

988 +<a id="__codelineno-0-130" name="__codelineno-0-130"></a>

989 +<a id="__codelineno-0-131" name="__codelineno-0-131"></a> Returns:

990 +<a id="__codelineno-0-132" name="__codelineno-0-132"></a> Dict[str, object]: A dictionary representing the Elasticsearch query.

991 +<a id="__codelineno-0-133" name="__codelineno-0-133"></a> """

992 +<a id="__codelineno-0-134" name="__codelineno-0-134"></a> return {

993 +<a id="__codelineno-0-135" name="__codelineno-0-135"></a> "query": {

994 +<a id="__codelineno-0-136" name="__codelineno-0-136"></a> "bool": {

995 +<a id="__codelineno-0-137" name="__codelineno-0-137"></a> "should": [

996 +<a id="__codelineno-0-138" name="__codelineno-0-138"></a> {"range": {"rule_level": {"gte": 12}}},

997 +<a id="__codelineno-0-139" name="__codelineno-0-139"></a> {"match": {"syslog_level": "ALERT"}},

998 +<a id="__codelineno-0-140" name="__codelineno-0-140"></a> ],

999 +<a id="__codelineno-0-141" name="__codelineno-0-141"></a> },

1000 +<a id="__codelineno-0-142" name="__codelineno-0-142"></a> },

1001 +<a id="__codelineno-0-143" name="__codelineno-0-143"></a> "sort": [{"timestamp_utc": {"order": "desc"}}],

1002 +<a id="__codelineno-0-144" name="__codelineno-0-144"></a> }

1003 </code></pre></div></td></tr></table></div>

1004 </details>

1005

@@ -1031,8 +1031,7 @@ available.

1031

1032 <details class="quote">

1033 <summary>Source code in <code>app\services\WazuhIndexer\alerts.py</code></summary>

1034 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-24">24</a>

1035 -<a href="#__codelineno-0-25">25</a>

1034 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-25">25</a>

1035 <a href="#__codelineno-0-26">26</a>

1036 <a href="#__codelineno-0-27">27</a>

1037 <a href="#__codelineno-0-28">28</a>

@@ -1047,23 +1046,24 @@ available.

1046 <a href="#__codelineno-0-37">37</a>

1047 <a href="#__codelineno-0-38">38</a>

1048 <a href="#__codelineno-0-39">39</a>

1050 -<a href="#__codelineno-0-40">40</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-24" name="__codelineno-0-24"></a>def __init__(self):

1051 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> """

1052 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> Initializes the service by collecting Wazuh-Indexer details and creating an Elasticsearch client.

1053 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> """

1054 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> (

1055 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> self.connector_url,

1056 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> self.connector_username,

1057 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> self.connector_password,

1058 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> ) = UniversalService().collect_wazuhindexer_details("Wazuh-Indexer")

1059 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> self.es = Elasticsearch(

1060 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> [self.connector_url],

1061 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> http_auth=(self.connector_username, self.connector_password),

1062 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> verify_certs=False,

1063 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> timeout=15,

1064 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a> max_retries=10,

1065 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> retry_on_timeout=False,

1066 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> )

1049 +<a href="#__codelineno-0-40">40</a>

1050 +<a href="#__codelineno-0-41">41</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-25" name="__codelineno-0-25"></a>def __init__(self):

1051 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a> """

1052 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> Initializes the service by collecting Wazuh-Indexer details and creating an Elasticsearch client.

1053 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> """

1054 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> (

1055 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> self.connector_url,

1056 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> self.connector_username,

1057 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> self.connector_password,

1058 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> ) = UniversalService().collect_wazuhindexer_details("Wazuh-Indexer")

1059 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> self.es = Elasticsearch(

1060 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> [self.connector_url],

1061 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> http_auth=(self.connector_username, self.connector_password),

1062 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> verify_certs=False,

1063 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> timeout=15,

1064 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> max_retries=10,

1065 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> retry_on_timeout=False,

1066 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> )

1067 </code></pre></div></td></tr></table></div>

1068 </details>

1069 </div>

@@ -1110,8 +1110,7 @@ Returns the 10 previous alerts based on the <code>timestamp_utc</code> field.</p

1110

1111 <details class="quote">

1112 <summary>Source code in <code>app\services\WazuhIndexer\alerts.py</code></summary>

1113 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-57">57</a>

1114 -<a href="#__codelineno-0-58">58</a>

1113 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-58">58</a>

1114 <a href="#__codelineno-0-59">59</a>

1115 <a href="#__codelineno-0-60">60</a>

1116 <a href="#__codelineno-0-61">61</a>

@@ -1150,47 +1149,48 @@ Returns the 10 previous alerts based on the <code>timestamp_utc</code> field.</p

1149 <a href="#__codelineno-0-94">94</a>

1150 <a href="#__codelineno-0-95">95</a>

1151 <a href="#__codelineno-0-96">96</a>

1153 -<a href="#__codelineno-0-97">97</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-57" name="__codelineno-0-57"></a>def collect_alerts(self) -> Dict[str, object]:

1154 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> """

1155 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> Collects the alerts from the Wazuh-Indexer where the index name starts with "wazuh_"

1156 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a> and is not in the SKIP_INDEX_NAMES list.

1157 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> Returns the 10 previous alerts based on the `timestamp_utc` field.

1158 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a>

1159 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> Returns:

1160 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> Dict[str, object]: A dictionary containing success status and alerts or an error message.

1161 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> """

1162 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> if not all(

1163 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> [self.connector_url, self.connector_username, self.connector_password],

1164 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> ):

1165 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> return {

1166 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "message": "Failed to collect Wazuh-Indexer details",

1167 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "success": False,

1168 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> }

1169 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a>

1170 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> indices_list = UniversalService().collect_indices()

1171 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> if not indices_list["success"]:

1172 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a> return {"message": "Failed to collect indices", "success": False}

1173 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a>

1174 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> alerts_summary = []

1175 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> for index_name in indices_list["indices_list"]:

1176 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> if not index_name.startswith("wazuh_") or self.is_index_skipped(index_name):

1177 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> continue

1178 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a>

1179 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> alerts = self._collect_alerts(index_name)

1180 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> if alerts["success"] and len(alerts["alerts"]) > 0:

1181 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a> alerts_summary.append(

1182 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> {

1183 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> "index_name": index_name,

1184 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> "total_alerts": len(alerts["alerts"]),

1185 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> "last_10_alerts": alerts["alerts"],

1186 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> },

1187 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> )

1188 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a>

1189 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> return {

1190 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> "message": "Successfully collected alerts",

1191 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> "success": True,

1192 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> "alerts_summary": alerts_summary,

1193 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a> }

1152 +<a href="#__codelineno-0-97">97</a>

1153 +<a href="#__codelineno-0-98">98</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-58" name="__codelineno-0-58"></a>def collect_alerts(self) -> Dict[str, object]:

1154 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> """

1155 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> Collects the alerts from the Wazuh-Indexer where the index name starts with "wazuh_"

1156 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> and is not in the SKIP_INDEX_NAMES list.

1157 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> Returns the 10 previous alerts based on the `timestamp_utc` field.

1158 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a>

1159 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> Returns:

1160 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> Dict[str, object]: A dictionary containing success status and alerts or an error message.

1161 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> """

1162 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> if not all(

1163 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> [self.connector_url, self.connector_username, self.connector_password],

1164 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> ):

1165 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> return {

1166 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "message": "Failed to collect Wazuh-Indexer details",

1167 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> "success": False,

1168 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> }

1169 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a>

1170 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> indices_list = UniversalService().collect_indices()

1171 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> if not indices_list["success"]:

1172 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> return {"message": "Failed to collect indices", "success": False}

1173 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a>

1174 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> alerts_summary = []

1175 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> for index_name in indices_list["indices_list"]:

1176 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> if not index_name.startswith("wazuh_") or self.is_index_skipped(index_name):

1177 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> continue

1178 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a>

1179 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> alerts = self._collect_alerts(index_name)

1180 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> if alerts["success"] and len(alerts["alerts"]) > 0:

1181 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> alerts_summary.append(

1182 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> {

1183 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> "index_name": index_name,

1184 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> "total_alerts": len(alerts["alerts"]),

1185 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> "last_10_alerts": alerts["alerts"],

1186 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> },

1187 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> )

1188 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a>

1189 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> return {

1190 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> "message": "Successfully collected alerts",

1191 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> "success": True,

1192 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> "alerts_summary": alerts_summary,

1193 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> }

1194 </code></pre></div></td></tr></table></div>

1195 </details>

1196 </div>

@@ -1263,8 +1263,7 @@ Returns the 10 previous alerts based on the <code>timestamp_utc</code> field.</p

1263

1264 <details class="quote">

1265 <summary>Source code in <code>app\services\WazuhIndexer\alerts.py</code></summary>

1266 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-42">42</a>

1267 -<a href="#__codelineno-0-43">43</a>

1266 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-43">43</a>

1267 <a href="#__codelineno-0-44">44</a>

1268 <a href="#__codelineno-0-45">45</a>

1269 <a href="#__codelineno-0-46">46</a>

@@ -1276,20 +1275,21 @@ Returns the 10 previous alerts based on the <code>timestamp_utc</code> field.</p

1275 <a href="#__codelineno-0-52">52</a>

1276 <a href="#__codelineno-0-53">53</a>

1277 <a href="#__codelineno-0-54">54</a>

1279 -<a href="#__codelineno-0-55">55</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-42" name="__codelineno-0-42"></a>def is_index_skipped(self, index_name: str) -> bool:

1280 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> """

1281 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> Checks whether the given index name should be skipped.

1282 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a>

1283 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> Args:

1284 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> index_name (str): The name of the index.

1285 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a>

1286 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> Returns:

1287 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> bool: True if the index should be skipped, False otherwise.

1288 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> """

1289 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> for skipped in self.SKIP_INDEX_NAMES:

1290 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> if index_name.startswith(skipped):

1291 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> return True

1292 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> return False

1278 +<a href="#__codelineno-0-55">55</a>

1279 +<a href="#__codelineno-0-56">56</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-43" name="__codelineno-0-43"></a>def is_index_skipped(self, index_name: str) -> bool:

1280 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> """

1281 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> Checks whether the given index name should be skipped.

1282 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a>

1283 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> Args:

1284 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> index_name (str): The name of the index.

1285 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a>

1286 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> Returns:

1287 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> bool: True if the index should be skipped, False otherwise.

1288 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> """

1289 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> for skipped in self.SKIP_INDEX_NAMES:

1290 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> if index_name.startswith(skipped):

1291 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a> return True

1292 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> return False

1293 </code></pre></div></td></tr></table></div>

1294 </details>

1295 </div>

backend/site/artifacts/index.html

+362 -362

@@ -1038,12 +1038,7 @@ It collects the artifact name and client name from the request body and returns

1038

1039 <details class="quote">

1040 <summary>Source code in <code>app\routes\velociraptor.py</code></summary>

1041 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-69"> 69</a>

1042 -<a href="#__codelineno-0-70"> 70</a>

1043 -<a href="#__codelineno-0-71"> 71</a>

1044 -<a href="#__codelineno-0-72"> 72</a>

1045 -<a href="#__codelineno-0-73"> 73</a>

1046 -<a href="#__codelineno-0-74"> 74</a>

1041 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-74"> 74</a>

1042 <a href="#__codelineno-0-75"> 75</a>

1043 <a href="#__codelineno-0-76"> 76</a>

1044 <a href="#__codelineno-0-77"> 77</a>

@@ -1071,40 +1066,45 @@ It collects the artifact name and client name from the request body and returns

1066 <a href="#__codelineno-0-99"> 99</a>

1067 <a href="#__codelineno-0-100">100</a>

1068 <a href="#__codelineno-0-101">101</a>

1074 -<a href="#__codelineno-0-102">102</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-69" name="__codelineno-0-69"></a>@bp.route("/velociraptor/artifacts/collection", methods=["POST"])

1075 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a>def collect_artifact():

1076 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> """

1077 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> Endpoint to collect an artifact.

1078 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> It collects the artifact name and client name from the request body and returns the results.

1079 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a>

1080 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> Returns:

1081 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a> json: A JSON response containing the result of the artifact collection operation.

1082 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> """

1083 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> req_data = request.get_json()

1084 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> artifact_name = req_data["artifact_name"]

1085 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> client_name = req_data["client_name"]

1086 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> service = UniversalService()

1087 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> client_id = service.get_client_id(client_name=client_name)["results"][0][

1088 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> "client_id"

1089 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> ]

1090 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a> if client_id is None:

1091 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> return (

1092 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> jsonify(

1093 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> {

1094 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> "message": f"{client_name} has not been seen in the last 30 seconds and may not be online with the "

1095 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> "Velociraptor server.",

1096 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> "success": False,

1097 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> },

1098 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> ),

1099 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> 500,

1100 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> )

1101 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a>

1102 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a> artifact_service = ArtifactsService()

1103 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> artifact_results = artifact_service.run_artifact_collection(

1104 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> client_id=client_id,

1105 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a> artifact=artifact_name,

1106 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a> )

1107 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> return artifact_results

1069 +<a href="#__codelineno-0-102">102</a>

1070 +<a href="#__codelineno-0-103">103</a>

1071 +<a href="#__codelineno-0-104">104</a>

1072 +<a href="#__codelineno-0-105">105</a>

1073 +<a href="#__codelineno-0-106">106</a>

1074 +<a href="#__codelineno-0-107">107</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-74" name="__codelineno-0-74"></a>@bp.route("/velociraptor/artifacts/collection", methods=["POST"])

1075 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a>def collect_artifact():

1076 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> """

1077 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Endpoint to collect an artifact.

1078 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> It collects the artifact name and client name from the request body and returns the results.

1079 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a>

1080 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> Returns:

1081 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> json: A JSON response containing the result of the artifact collection operation.

1082 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> """

1083 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a> req_data = request.get_json()

1084 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> artifact_name = req_data["artifact_name"]

1085 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> client_name = req_data["client_name"]

1086 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> service = UniversalService()

1087 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> client_id = service.get_client_id(client_name=client_name)["results"][0][

1088 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> "client_id"

1089 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> ]

1090 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> if client_id is None:

1091 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> return (

1092 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> jsonify(

1093 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> {

1094 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> "message": f"{client_name} has not been seen in the last 30 seconds and may not be online with the "

1095 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> "Velociraptor server.",

1096 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> "success": False,

1097 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> },

1098 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> ),

1099 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> 500,

1100 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> )

1101 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a>

1102 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> artifact_service = ArtifactsService()

1103 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> artifact_results = artifact_service.run_artifact_collection(

1104 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> client_id=client_id,

1105 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a> artifact=artifact_name,

1106 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> )

1107 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a> return artifact_results

1108 </code></pre></div></td></tr></table></div>

1109 </details>

1110 </div>

@@ -1158,8 +1158,7 @@ It processes each artifact to verify the connection and returns the results.

1158

1159 <details class="quote">

1160 <summary>Source code in <code>app\routes\velociraptor.py</code></summary>

1161 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-10">10</a>

1162 -<a href="#__codelineno-0-11">11</a>

1161 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-11">11</a>

1162 <a href="#__codelineno-0-12">12</a>

1163 <a href="#__codelineno-0-13">13</a>

1164 <a href="#__codelineno-0-14">14</a>

@@ -1170,19 +1169,20 @@ It processes each artifact to verify the connection and returns the results.

1169 <a href="#__codelineno-0-19">19</a>

1170 <a href="#__codelineno-0-20">20</a>

1171 <a href="#__codelineno-0-21">21</a>

1173 -<a href="#__codelineno-0-22">22</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-10" name="__codelineno-0-10"></a>@bp.route("/velociraptor/artifacts", methods=["GET"])

1174 -<a id="__codelineno-0-11" name="__codelineno-0-11"></a>def get_artifacts():

1175 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a> """

1176 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a> Endpoint to list all available artifacts.

1177 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a> It processes each artifact to verify the connection and returns the results.

1178 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a>

1179 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> Returns:

1180 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> json: A JSON response containing the list of all available artifacts along with their connection verification

1181 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> status.

1182 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> """

1183 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> service = ArtifactsService()

1184 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a> artifacts = service.collect_artifacts()

1185 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> return artifacts

1172 +<a href="#__codelineno-0-22">22</a>

1173 +<a href="#__codelineno-0-23">23</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-11" name="__codelineno-0-11"></a>@bp.route("/velociraptor/artifacts", methods=["GET"])

1174 +<a id="__codelineno-0-12" name="__codelineno-0-12"></a>def get_artifacts():

1175 +<a id="__codelineno-0-13" name="__codelineno-0-13"></a> """

1176 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> Endpoint to list all available artifacts.

1177 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> It processes each artifact to verify the connection and returns the results.

1178 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a>

1179 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a> Returns:

1180 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> json: A JSON response containing the list of all available artifacts along with their connection verification

1181 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a> status.

1182 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> """

1183 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> service = ArtifactsService()

1184 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> artifacts = service.collect_artifacts()

1185 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> return artifacts

1186 </code></pre></div></td></tr></table></div>

1187 </details>

1188 </div>

@@ -1237,9 +1237,7 @@ begins with <code>Linux</code>.

1237

1238 <details class="quote">

1239 <summary>Source code in <code>app\routes\velociraptor.py</code></summary>

1240 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-24">24</a>

1241 -<a href="#__codelineno-0-25">25</a>

1242 -<a href="#__codelineno-0-26">26</a>

1240 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-26">26</a>

1241 <a href="#__codelineno-0-27">27</a>

1242 <a href="#__codelineno-0-28">28</a>

1243 <a href="#__codelineno-0-29">29</a>

@@ -1250,20 +1248,22 @@ begins with <code>Linux</code>.

1248 <a href="#__codelineno-0-34">34</a>

1249 <a href="#__codelineno-0-35">35</a>

1250 <a href="#__codelineno-0-36">36</a>

1253 -<a href="#__codelineno-0-37">37</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-24" name="__codelineno-0-24"></a>@bp.route("/velociraptor/artifacts/linux", methods=["GET"])

1254 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a>def get_artifacts_linux():

1255 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> """

1256 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> Endpoint to list all available Linux artifacts.

1257 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> It processes each artifact to verify the connection and returns the results where the name

1258 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> begins with `Linux`.

1259 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a>

1260 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> Returns:

1261 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> json: A JSON response containing the list of all available Linux artifacts along with their connection verification

1262 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> status.

1263 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> """

1264 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> service = ArtifactsService()

1265 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> linux_artifacts = service.collect_artifacts_linux()

1266 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> return linux_artifacts

1251 +<a href="#__codelineno-0-37">37</a>

1252 +<a href="#__codelineno-0-38">38</a>

1253 +<a href="#__codelineno-0-39">39</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-26" name="__codelineno-0-26"></a>@bp.route("/velociraptor/artifacts/linux", methods=["GET"])

1254 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a>def get_artifacts_linux():

1255 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> """

1256 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> Endpoint to list all available Linux artifacts.

1257 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> It processes each artifact to verify the connection and returns the results where the name

1258 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> begins with `Linux`.

1259 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a>

1260 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> Returns:

1261 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> json: A JSON response containing the list of all available Linux artifacts along with their connection verification

1262 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> status.

1263 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> """

1264 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> service = ArtifactsService()

1265 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> linux_artifacts = service.collect_artifacts_linux()

1266 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> return linux_artifacts

1267 </code></pre></div></td></tr></table></div>

1268 </details>

1269 </div>

@@ -1318,11 +1318,7 @@ begins with <code>MacOS</code>.

1318

1319 <details class="quote">

1320 <summary>Source code in <code>app\routes\velociraptor.py</code></summary>

1321 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-54">54</a>

1322 -<a href="#__codelineno-0-55">55</a>

1323 -<a href="#__codelineno-0-56">56</a>

1324 -<a href="#__codelineno-0-57">57</a>

1325 -<a href="#__codelineno-0-58">58</a>

1321 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-58">58</a>

1322 <a href="#__codelineno-0-59">59</a>

1323 <a href="#__codelineno-0-60">60</a>

1324 <a href="#__codelineno-0-61">61</a>

@@ -1331,20 +1327,24 @@ begins with <code>MacOS</code>.

1327 <a href="#__codelineno-0-64">64</a>

1328 <a href="#__codelineno-0-65">65</a>

1329 <a href="#__codelineno-0-66">66</a>

1334 -<a href="#__codelineno-0-67">67</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-54" name="__codelineno-0-54"></a>@bp.route("/velociraptor/artifacts/mac", methods=["GET"])

1335 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a>def get_artifacts_mac():

1336 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> """

1337 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> Endpoint to list all available MacOS artifacts.

1338 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> It processes each artifact to verify the connection and returns the results where the name

1339 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> begins with `MacOS`.

1340 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a>

1341 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> Returns:

1342 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> json: A JSON response containing the list of all available MacOS artifacts along with their connection verification

1343 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> status.

1344 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> """

1345 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> service = ArtifactsService()

1346 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> mac_artifacts = service.collect_artifacts_macos()

1347 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> return mac_artifacts

1330 +<a href="#__codelineno-0-67">67</a>

1331 +<a href="#__codelineno-0-68">68</a>

1332 +<a href="#__codelineno-0-69">69</a>

1333 +<a href="#__codelineno-0-70">70</a>

1334 +<a href="#__codelineno-0-71">71</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-58" name="__codelineno-0-58"></a>@bp.route("/velociraptor/artifacts/mac", methods=["GET"])

1335 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a>def get_artifacts_mac():

1336 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> """

1337 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> Endpoint to list all available MacOS artifacts.

1338 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> It processes each artifact to verify the connection and returns the results where the name

1339 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> begins with `MacOS`.

1340 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a>

1341 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> Returns:

1342 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> json: A JSON response containing the list of all available MacOS artifacts along with their connection verification

1343 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> status.

1344 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> """

1345 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> service = ArtifactsService()

1346 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> mac_artifacts = service.collect_artifacts_macos()

1347 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> return mac_artifacts

1348 </code></pre></div></td></tr></table></div>

1349 </details>

1350 </div>

@@ -1399,10 +1399,7 @@ begins with <code>Windows</code>.

1399

1400 <details class="quote">

1401 <summary>Source code in <code>app\routes\velociraptor.py</code></summary>

1402 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-39">39</a>

1403 -<a href="#__codelineno-0-40">40</a>

1404 -<a href="#__codelineno-0-41">41</a>

1405 -<a href="#__codelineno-0-42">42</a>

1402 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-42">42</a>

1403 <a href="#__codelineno-0-43">43</a>

1404 <a href="#__codelineno-0-44">44</a>

1405 <a href="#__codelineno-0-45">45</a>

@@ -1412,20 +1409,23 @@ begins with <code>Windows</code>.

1409 <a href="#__codelineno-0-49">49</a>

1410 <a href="#__codelineno-0-50">50</a>

1411 <a href="#__codelineno-0-51">51</a>

1415 -<a href="#__codelineno-0-52">52</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-39" name="__codelineno-0-39"></a>@bp.route("/velociraptor/artifacts/windows", methods=["GET"])

1416 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a>def get_artifacts_windows():

1417 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> """

1418 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> Endpoint to list all available Windows artifacts.

1419 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> It processes each artifact to verify the connection and returns the results where the name

1420 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> begins with `Windows`.

1421 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a>

1422 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> Returns:

1423 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> json: A JSON response containing the list of all available Windows artifacts along with their connection verification

1424 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> status.

1425 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> """

1426 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> service = ArtifactsService()

1427 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> windows_artifacts = service.collect_artifacts_windows()

1428 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> return windows_artifacts

1412 +<a href="#__codelineno-0-52">52</a>

1413 +<a href="#__codelineno-0-53">53</a>

1414 +<a href="#__codelineno-0-54">54</a>

1415 +<a href="#__codelineno-0-55">55</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-42" name="__codelineno-0-42"></a>@bp.route("/velociraptor/artifacts/windows", methods=["GET"])

1416 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a>def get_artifacts_windows():

1417 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> """

1418 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> Endpoint to list all available Windows artifacts.

1419 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> It processes each artifact to verify the connection and returns the results where the name

1420 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> begins with `Windows`.

1421 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a>

1422 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> Returns:

1423 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> json: A JSON response containing the list of all available Windows artifacts along with their connection verification

1424 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> status.

1425 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> """

1426 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> service = ArtifactsService()

1427 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> windows_artifacts = service.collect_artifacts_windows()

1428 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a> return windows_artifacts

1429 </code></pre></div></td></tr></table></div>

1430 </details>

1431 </div>

@@ -1478,9 +1478,7 @@ begins with <code>Windows</code>.

1478

1479 <details class="quote">

1480 <summary>Source code in <code>app\services\Velociraptor\artifacts.py</code></summary>

1481 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-4"> 4</a>

1482 -<a href="#__codelineno-0-5"> 5</a>

1483 -<a href="#__codelineno-0-6"> 6</a>

1481 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-6"> 6</a>

1482 <a href="#__codelineno-0-7"> 7</a>

1483 <a href="#__codelineno-0-8"> 8</a>

1484 <a href="#__codelineno-0-9"> 9</a>

@@ -1610,139 +1608,141 @@ begins with <code>Windows</code>.

1608 <a href="#__codelineno-0-133">133</a>

1609 <a href="#__codelineno-0-134">134</a>

1610 <a href="#__codelineno-0-135">135</a>

1613 -<a href="#__codelineno-0-136">136</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-4" name="__codelineno-0-4"></a>class ArtifactsService:

1614 -<a id="__codelineno-0-5" name="__codelineno-0-5"></a> """

1615 -<a id="__codelineno-0-6" name="__codelineno-0-6"></a> A service class that encapsulates the logic for pulling artifacts from Velociraptor.

1616 -<a id="__codelineno-0-7" name="__codelineno-0-7"></a> """

1617 -<a id="__codelineno-0-8" name="__codelineno-0-8"></a>

1618 -<a id="__codelineno-0-9" name="__codelineno-0-9"></a> def __init__(self):

1619 -<a id="__codelineno-0-10" name="__codelineno-0-10"></a> self.universal_service = UniversalService()

1620 -<a id="__codelineno-0-11" name="__codelineno-0-11"></a>

1621 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a> def _create_query(self, query: str) -> str:

1622 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a> """

1623 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a> Create a query string.

1624 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a>

1625 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> Args:

1626 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> query (str): The query to be executed.

1627 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a>

1628 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> Returns:

1629 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> str: The created query string.

1630 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a> """

1631 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> return query

1632 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a>

1633 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> def _get_artifact_key(self, client_id: str, artifact: str) -> str:

1634 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> """

1635 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> Construct the artifact key.

1636 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a>

1637 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> Args:

1638 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> client_id (str): The ID of the client.

1639 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> artifact (str): The name of the artifact.

1640 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a>

1641 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> Returns:

1642 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> str: The constructed artifact key.

1643 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> """

1644 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> return f"collect_client(client_id='{client_id}', artifacts=['{artifact}'])"

1645 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a>

1646 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> def collect_artifacts(self) -> dict:

1647 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a> """

1648 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> Collect the artifacts from Velociraptor.

1649 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a>

1650 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> Returns:

1651 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1652 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> """

1653 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> query = self._create_query("SELECT name FROM artifact_definitions()")

1654 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> return self.universal_service.execute_query(query)

1655 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a>

1656 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> def collect_artifacts_prefixed(self, prefix: str) -> dict:

1657 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> """

1658 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> Collect the artifacts from Velociraptor that have a name beginning with a specific prefix.

1659 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a>

1660 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> Args:

1661 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> prefix (str): The prefix to filter the artifacts.

1662 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a>

1663 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> Returns:

1664 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1665 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> """

1666 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> artifacts_response = self.collect_artifacts()

1667 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> if not artifacts_response["success"]:

1668 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> return artifacts_response

1669 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a>

1670 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> filtered_artifacts = [

1671 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> artifact

1672 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> for artifact in artifacts_response["results"]

1673 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> if artifact["name"].startswith(prefix)

1674 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> ]

1675 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a>

1676 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> return {

1677 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> "success": True,

1678 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> "message": f"Successfully collected {prefix} artifacts",

1679 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "artifacts": filtered_artifacts,

1680 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> }

1681 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a>

1682 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> def collect_artifacts_linux(self) -> dict:

1683 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> """

1684 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> Collect the artifacts from Velociraptor that have a name beginning with `Linux`.

1685 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a>

1686 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Returns:

1687 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1688 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> """

1689 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> return self.collect_artifacts_prefixed("Linux.")

1690 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a>

1691 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> def collect_artifacts_windows(self) -> dict:

1692 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> """

1693 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Collect the artifacts from Velociraptor that have a name beginning with `Windows`.

1694 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a>

1695 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> Returns:

1696 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1697 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> """

1698 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> return self.collect_artifacts_prefixed("Windows.")

1699 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a>

1700 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> def collect_artifacts_macos(self) -> dict:

1701 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> """

1702 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> Collect the artifacts from Velociraptor that have a name beginning with `MacOS`.

1703 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a>

1704 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> Returns:

1705 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1706 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a> """

1707 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> return self.collect_artifacts_prefixed("MacOS.")

1708 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a>

1709 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a> def run_artifact_collection(self, client_id: str, artifact: str) -> dict:

1710 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a> """

1711 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> Run an artifact collection on a specific client.

1712 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a>

1713 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Args:

1714 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a> client_id (str): The ID of the client.

1715 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a> artifact (str): The name of the artifact.

1716 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a>

1717 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> Returns:

1718 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> dict: A dictionary with the success status, a message, and potentially the results.

1719 -<a id="__codelineno-0-110" name="__codelineno-0-110"></a> """

1720 -<a id="__codelineno-0-111" name="__codelineno-0-111"></a> try:

1721 -<a id="__codelineno-0-112" name="__codelineno-0-112"></a> query = self._create_query(

1722 -<a id="__codelineno-0-113" name="__codelineno-0-113"></a> f"SELECT collect_client(client_id='{client_id}', artifacts=['{artifact}']) FROM scope()",

1723 -<a id="__codelineno-0-114" name="__codelineno-0-114"></a> )

1724 -<a id="__codelineno-0-115" name="__codelineno-0-115"></a> flow = self.universal_service.execute_query(query)

1725 -<a id="__codelineno-0-116" name="__codelineno-0-116"></a> logger.info(f"Successfully ran artifact collection on {flow}")

1726 -<a id="__codelineno-0-117" name="__codelineno-0-117"></a>

1727 -<a id="__codelineno-0-118" name="__codelineno-0-118"></a> artifact_key = self._get_artifact_key(client_id, artifact)

1728 -<a id="__codelineno-0-119" name="__codelineno-0-119"></a> flow_id = flow["results"][0][artifact_key]["flow_id"]

1729 -<a id="__codelineno-0-120" name="__codelineno-0-120"></a> logger.info(f"Successfully ran artifact collection on {flow_id}")

1730 -<a id="__codelineno-0-121" name="__codelineno-0-121"></a>

1731 -<a id="__codelineno-0-122" name="__codelineno-0-122"></a> completed = self.universal_service.watch_flow_completion(flow_id)

1732 -<a id="__codelineno-0-123" name="__codelineno-0-123"></a> logger.info(f"Successfully watched flow completion on {completed}")

1733 -<a id="__codelineno-0-124" name="__codelineno-0-124"></a>

1734 -<a id="__codelineno-0-125" name="__codelineno-0-125"></a> results = self.universal_service.read_collection_results(

1735 -<a id="__codelineno-0-126" name="__codelineno-0-126"></a> client_id,

1736 -<a id="__codelineno-0-127" name="__codelineno-0-127"></a> flow_id,

1737 -<a id="__codelineno-0-128" name="__codelineno-0-128"></a> artifact,

1738 -<a id="__codelineno-0-129" name="__codelineno-0-129"></a> )

1739 -<a id="__codelineno-0-130" name="__codelineno-0-130"></a> return results

1740 -<a id="__codelineno-0-131" name="__codelineno-0-131"></a> except Exception as err:

1741 -<a id="__codelineno-0-132" name="__codelineno-0-132"></a> logger.error(f"Failed to run artifact collection: {err}")

1742 -<a id="__codelineno-0-133" name="__codelineno-0-133"></a> return {

1743 -<a id="__codelineno-0-134" name="__codelineno-0-134"></a> "message": "Failed to run artifact collection",

1744 -<a id="__codelineno-0-135" name="__codelineno-0-135"></a> "success": False,

1745 -<a id="__codelineno-0-136" name="__codelineno-0-136"></a> }

1611 +<a href="#__codelineno-0-136">136</a>

1612 +<a href="#__codelineno-0-137">137</a>

1613 +<a href="#__codelineno-0-138">138</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-6" name="__codelineno-0-6"></a>class ArtifactsService:

1614 +<a id="__codelineno-0-7" name="__codelineno-0-7"></a> """

1615 +<a id="__codelineno-0-8" name="__codelineno-0-8"></a> A service class that encapsulates the logic for pulling artifacts from Velociraptor.

1616 +<a id="__codelineno-0-9" name="__codelineno-0-9"></a> """

1617 +<a id="__codelineno-0-10" name="__codelineno-0-10"></a>

1618 +<a id="__codelineno-0-11" name="__codelineno-0-11"></a> def __init__(self):

1619 +<a id="__codelineno-0-12" name="__codelineno-0-12"></a> self.universal_service = UniversalService()

1620 +<a id="__codelineno-0-13" name="__codelineno-0-13"></a>

1621 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> def _create_query(self, query: str) -> str:

1622 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> """

1623 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a> Create a query string.

1624 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a>

1625 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> Args:

1626 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a> query (str): The query to be executed.

1627 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a>

1628 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> Returns:

1629 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> str: The created query string.

1630 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> """

1631 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a> return query

1632 +<a id="__codelineno-0-25" name="__codelineno-0-25"></a>

1633 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a> def _get_artifact_key(self, client_id: str, artifact: str) -> str:

1634 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> """

1635 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> Construct the artifact key.

1636 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a>

1637 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> Args:

1638 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> client_id (str): The ID of the client.

1639 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> artifact (str): The name of the artifact.

1640 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a>

1641 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> Returns:

1642 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> str: The constructed artifact key.

1643 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> """

1644 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> return f"collect_client(client_id='{client_id}', artifacts=['{artifact}'])"

1645 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a>

1646 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> def collect_artifacts(self) -> dict:

1647 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> """

1648 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> Collect the artifacts from Velociraptor.

1649 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a>

1650 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> Returns:

1651 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1652 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> """

1653 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> query = self._create_query("SELECT name FROM artifact_definitions()")

1654 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> return self.universal_service.execute_query(query)

1655 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a>

1656 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> def collect_artifacts_prefixed(self, prefix: str) -> dict:

1657 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> """

1658 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> Collect the artifacts from Velociraptor that have a name beginning with a specific prefix.

1659 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a>

1660 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> Args:

1661 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> prefix (str): The prefix to filter the artifacts.

1662 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a>

1663 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> Returns:

1664 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1665 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a> """

1666 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> artifacts_response = self.collect_artifacts()

1667 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> if not artifacts_response["success"]:

1668 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> return artifacts_response

1669 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a>

1670 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> filtered_artifacts = [

1671 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> artifact

1672 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> for artifact in artifacts_response["results"]

1673 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> if artifact["name"].startswith(prefix)

1674 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> ]

1675 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a>

1676 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> return {

1677 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "success": True,

1678 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "message": f"Successfully collected {prefix} artifacts",

1679 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> "artifacts": filtered_artifacts,

1680 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> }

1681 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a>

1682 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> def collect_artifacts_linux(self) -> dict:

1683 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> """

1684 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Collect the artifacts from Velociraptor that have a name beginning with `Linux`.

1685 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a>

1686 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> Returns:

1687 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1688 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> """

1689 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> return self.collect_artifacts_prefixed("Linux.")

1690 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a>

1691 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> def collect_artifacts_windows(self) -> dict:

1692 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> """

1693 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> Collect the artifacts from Velociraptor that have a name beginning with `Windows`.

1694 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a>

1695 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> Returns:

1696 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1697 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> """

1698 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> return self.collect_artifacts_prefixed("Windows.")

1699 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a>

1700 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> def collect_artifacts_macos(self) -> dict:

1701 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> """

1702 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> Collect the artifacts from Velociraptor that have a name beginning with `MacOS`.

1703 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a>

1704 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> Returns:

1705 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1706 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> """

1707 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> return self.collect_artifacts_prefixed("MacOS.")

1708 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a>

1709 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> def run_artifact_collection(self, client_id: str, artifact: str) -> dict:

1710 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> """

1711 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Run an artifact collection on a specific client.

1712 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a>

1713 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> Args:

1714 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a> client_id (str): The ID of the client.

1715 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a> artifact (str): The name of the artifact.

1716 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a>

1717 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> Returns:

1718 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a> dict: A dictionary with the success status, a message, and potentially the results.

1719 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a> """

1720 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> try:

1721 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> query = self._create_query(

1722 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> f"SELECT collect_client(client_id='{client_id}', artifacts=['{artifact}']) FROM scope()",

1723 +<a id="__codelineno-0-116" name="__codelineno-0-116"></a> )

1724 +<a id="__codelineno-0-117" name="__codelineno-0-117"></a> flow = self.universal_service.execute_query(query)

1725 +<a id="__codelineno-0-118" name="__codelineno-0-118"></a> logger.info(f"Successfully ran artifact collection on {flow}")

1726 +<a id="__codelineno-0-119" name="__codelineno-0-119"></a>

1727 +<a id="__codelineno-0-120" name="__codelineno-0-120"></a> artifact_key = self._get_artifact_key(client_id, artifact)

1728 +<a id="__codelineno-0-121" name="__codelineno-0-121"></a> flow_id = flow["results"][0][artifact_key]["flow_id"]

1729 +<a id="__codelineno-0-122" name="__codelineno-0-122"></a> logger.info(f"Successfully ran artifact collection on {flow_id}")

1730 +<a id="__codelineno-0-123" name="__codelineno-0-123"></a>

1731 +<a id="__codelineno-0-124" name="__codelineno-0-124"></a> completed = self.universal_service.watch_flow_completion(flow_id)

1732 +<a id="__codelineno-0-125" name="__codelineno-0-125"></a> logger.info(f"Successfully watched flow completion on {completed}")

1733 +<a id="__codelineno-0-126" name="__codelineno-0-126"></a>

1734 +<a id="__codelineno-0-127" name="__codelineno-0-127"></a> results = self.universal_service.read_collection_results(

1735 +<a id="__codelineno-0-128" name="__codelineno-0-128"></a> client_id,

1736 +<a id="__codelineno-0-129" name="__codelineno-0-129"></a> flow_id,

1737 +<a id="__codelineno-0-130" name="__codelineno-0-130"></a> artifact,

1738 +<a id="__codelineno-0-131" name="__codelineno-0-131"></a> )

1739 +<a id="__codelineno-0-132" name="__codelineno-0-132"></a> return results

1740 +<a id="__codelineno-0-133" name="__codelineno-0-133"></a> except Exception as err:

1741 +<a id="__codelineno-0-134" name="__codelineno-0-134"></a> logger.error(f"Failed to run artifact collection: {err}")

1742 +<a id="__codelineno-0-135" name="__codelineno-0-135"></a> return {

1743 +<a id="__codelineno-0-136" name="__codelineno-0-136"></a> "message": "Failed to run artifact collection",

1744 +<a id="__codelineno-0-137" name="__codelineno-0-137"></a> "success": False,

1745 +<a id="__codelineno-0-138" name="__codelineno-0-138"></a> }

1746 </code></pre></div></td></tr></table></div>

1747 </details>

1748

@@ -1796,23 +1796,23 @@ begins with <code>Windows</code>.

1796

1797 <details class="quote">

1798 <summary>Source code in <code>app\services\Velociraptor\artifacts.py</code></summary>

1799 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-37">37</a>

1800 -<a href="#__codelineno-0-38">38</a>

1801 -<a href="#__codelineno-0-39">39</a>

1799 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-39">39</a>

1800 <a href="#__codelineno-0-40">40</a>

1801 <a href="#__codelineno-0-41">41</a>

1802 <a href="#__codelineno-0-42">42</a>

1803 <a href="#__codelineno-0-43">43</a>

1804 <a href="#__codelineno-0-44">44</a>

1807 -<a href="#__codelineno-0-45">45</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-37" name="__codelineno-0-37"></a>def collect_artifacts(self) -> dict:

1808 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a> """

1809 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> Collect the artifacts from Velociraptor.

1810 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a>

1811 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> Returns:

1812 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1813 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> """

1814 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> query = self._create_query("SELECT name FROM artifact_definitions()")

1815 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> return self.universal_service.execute_query(query)

1805 +<a href="#__codelineno-0-45">45</a>

1806 +<a href="#__codelineno-0-46">46</a>

1807 +<a href="#__codelineno-0-47">47</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-39" name="__codelineno-0-39"></a>def collect_artifacts(self) -> dict:

1808 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> """

1809 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> Collect the artifacts from Velociraptor.

1810 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a>

1811 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> Returns:

1812 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1813 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> """

1814 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> query = self._create_query("SELECT name FROM artifact_definitions()")

1815 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> return self.universal_service.execute_query(query)

1816 </code></pre></div></td></tr></table></div>

1817 </details>

1818 </div>

@@ -1857,21 +1857,21 @@ begins with <code>Windows</code>.

1857

1858 <details class="quote">

1859 <summary>Source code in <code>app\services\Velociraptor\artifacts.py</code></summary>

1860 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-73">73</a>

1861 -<a href="#__codelineno-0-74">74</a>

1862 -<a href="#__codelineno-0-75">75</a>

1860 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-75">75</a>

1861 <a href="#__codelineno-0-76">76</a>

1862 <a href="#__codelineno-0-77">77</a>

1863 <a href="#__codelineno-0-78">78</a>

1864 <a href="#__codelineno-0-79">79</a>

1867 -<a href="#__codelineno-0-80">80</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-73" name="__codelineno-0-73"></a>def collect_artifacts_linux(self) -> dict:

1868 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> """

1869 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> Collect the artifacts from Velociraptor that have a name beginning with `Linux`.

1870 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a>

1871 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Returns:

1872 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1873 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> """

1874 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> return self.collect_artifacts_prefixed("Linux.")

1865 +<a href="#__codelineno-0-80">80</a>

1866 +<a href="#__codelineno-0-81">81</a>

1867 +<a href="#__codelineno-0-82">82</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-75" name="__codelineno-0-75"></a>def collect_artifacts_linux(self) -> dict:

1868 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> """

1869 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Collect the artifacts from Velociraptor that have a name beginning with `Linux`.

1870 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a>

1871 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> Returns:

1872 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1873 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> """

1874 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> return self.collect_artifacts_prefixed("Linux.")

1875 </code></pre></div></td></tr></table></div>

1876 </details>

1877 </div>

@@ -1916,21 +1916,21 @@ begins with <code>Windows</code>.

1916

1917 <details class="quote">

1918 <summary>Source code in <code>app\services\Velociraptor\artifacts.py</code></summary>

1919 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-91">91</a>

1920 -<a href="#__codelineno-0-92">92</a>

1921 -<a href="#__codelineno-0-93">93</a>

1922 -<a href="#__codelineno-0-94">94</a>

1923 -<a href="#__codelineno-0-95">95</a>

1924 -<a href="#__codelineno-0-96">96</a>

1925 -<a href="#__codelineno-0-97">97</a>

1926 -<a href="#__codelineno-0-98">98</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-91" name="__codelineno-0-91"></a>def collect_artifacts_macos(self) -> dict:

1927 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> """

1928 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> Collect the artifacts from Velociraptor that have a name beginning with `MacOS`.

1929 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a>

1930 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> Returns:

1931 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1932 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a> """

1933 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> return self.collect_artifacts_prefixed("MacOS.")

1919 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-93"> 93</a>

1920 +<a href="#__codelineno-0-94"> 94</a>

1921 +<a href="#__codelineno-0-95"> 95</a>

1922 +<a href="#__codelineno-0-96"> 96</a>

1923 +<a href="#__codelineno-0-97"> 97</a>

1924 +<a href="#__codelineno-0-98"> 98</a>

1925 +<a href="#__codelineno-0-99"> 99</a>

1926 +<a href="#__codelineno-0-100">100</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-93" name="__codelineno-0-93"></a>def collect_artifacts_macos(self) -> dict:

1927 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> """

1928 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> Collect the artifacts from Velociraptor that have a name beginning with `MacOS`.

1929 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a>

1930 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> Returns:

1931 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

1932 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> """

1933 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> return self.collect_artifacts_prefixed("MacOS.")

1934 </code></pre></div></td></tr></table></div>

1935 </details>

1936 </div>

@@ -2003,9 +2003,7 @@ begins with <code>Windows</code>.

2003

2004 <details class="quote">

2005 <summary>Source code in <code>app\services\Velociraptor\artifacts.py</code></summary>

2006 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-47">47</a>

2007 -<a href="#__codelineno-0-48">48</a>

2008 -<a href="#__codelineno-0-49">49</a>

2006 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-49">49</a>

2007 <a href="#__codelineno-0-50">50</a>

2008 <a href="#__codelineno-0-51">51</a>

2009 <a href="#__codelineno-0-52">52</a>

@@ -2027,31 +2025,33 @@ begins with <code>Windows</code>.

2025 <a href="#__codelineno-0-68">68</a>

2026 <a href="#__codelineno-0-69">69</a>

2027 <a href="#__codelineno-0-70">70</a>

2030 -<a href="#__codelineno-0-71">71</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-47" name="__codelineno-0-47"></a>def collect_artifacts_prefixed(self, prefix: str) -> dict:

2031 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> """

2032 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> Collect the artifacts from Velociraptor that have a name beginning with a specific prefix.

2033 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a>

2034 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> Args:

2035 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> prefix (str): The prefix to filter the artifacts.

2036 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a>

2037 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> Returns:

2038 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

2039 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> """

2040 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> artifacts_response = self.collect_artifacts()

2041 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> if not artifacts_response["success"]:

2042 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> return artifacts_response

2043 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a>

2044 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> filtered_artifacts = [

2045 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> artifact

2046 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> for artifact in artifacts_response["results"]

2047 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> if artifact["name"].startswith(prefix)

2048 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> ]

2049 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a>

2050 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> return {

2051 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> "success": True,

2052 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> "message": f"Successfully collected {prefix} artifacts",

2053 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "artifacts": filtered_artifacts,

2054 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> }

2028 +<a href="#__codelineno-0-71">71</a>

2029 +<a href="#__codelineno-0-72">72</a>

2030 +<a href="#__codelineno-0-73">73</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-49" name="__codelineno-0-49"></a>def collect_artifacts_prefixed(self, prefix: str) -> dict:

2031 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> """

2032 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> Collect the artifacts from Velociraptor that have a name beginning with a specific prefix.

2033 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a>

2034 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> Args:

2035 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> prefix (str): The prefix to filter the artifacts.

2036 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a>

2037 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> Returns:

2038 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

2039 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a> """

2040 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> artifacts_response = self.collect_artifacts()

2041 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> if not artifacts_response["success"]:

2042 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> return artifacts_response

2043 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a>

2044 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> filtered_artifacts = [

2045 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> artifact

2046 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> for artifact in artifacts_response["results"]

2047 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> if artifact["name"].startswith(prefix)

2048 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> ]

2049 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a>

2050 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> return {

2051 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "success": True,

2052 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "message": f"Successfully collected {prefix} artifacts",

2053 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> "artifacts": filtered_artifacts,

2054 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> }

2055 </code></pre></div></td></tr></table></div>

2056 </details>

2057 </div>

@@ -2096,21 +2096,21 @@ begins with <code>Windows</code>.

2096

2097 <details class="quote">

2098 <summary>Source code in <code>app\services\Velociraptor\artifacts.py</code></summary>

2099 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-82">82</a>

2100 -<a href="#__codelineno-0-83">83</a>

2101 -<a href="#__codelineno-0-84">84</a>

2099 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-84">84</a>

2100 <a href="#__codelineno-0-85">85</a>

2101 <a href="#__codelineno-0-86">86</a>

2102 <a href="#__codelineno-0-87">87</a>

2103 <a href="#__codelineno-0-88">88</a>

2106 -<a href="#__codelineno-0-89">89</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-82" name="__codelineno-0-82"></a>def collect_artifacts_windows(self) -> dict:

2107 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> """

2108 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Collect the artifacts from Velociraptor that have a name beginning with `Windows`.

2109 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a>

2110 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> Returns:

2111 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

2112 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> """

2113 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> return self.collect_artifacts_prefixed("Windows.")

2104 +<a href="#__codelineno-0-89">89</a>

2105 +<a href="#__codelineno-0-90">90</a>

2106 +<a href="#__codelineno-0-91">91</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-84" name="__codelineno-0-84"></a>def collect_artifacts_windows(self) -> dict:

2107 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> """

2108 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> Collect the artifacts from Velociraptor that have a name beginning with `Windows`.

2109 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a>

2110 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> Returns:

2111 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> dict: A dictionary with the success status, a message, and potentially the artifacts.

2112 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> """

2113 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> return self.collect_artifacts_prefixed("Windows.")

2114 </code></pre></div></td></tr></table></div>

2115 </details>

2116 </div>

@@ -2197,9 +2197,7 @@ begins with <code>Windows</code>.

2197

2198 <details class="quote">

2199 <summary>Source code in <code>app\services\Velociraptor\artifacts.py</code></summary>

2200 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-100">100</a>

2201 -<a href="#__codelineno-0-101">101</a>

2202 -<a href="#__codelineno-0-102">102</a>

2200 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-102">102</a>

2201 <a href="#__codelineno-0-103">103</a>

2202 <a href="#__codelineno-0-104">104</a>

2203 <a href="#__codelineno-0-105">105</a>

@@ -2233,43 +2231,45 @@ begins with <code>Windows</code>.

2231 <a href="#__codelineno-0-133">133</a>

2232 <a href="#__codelineno-0-134">134</a>

2233 <a href="#__codelineno-0-135">135</a>

2236 -<a href="#__codelineno-0-136">136</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-100" name="__codelineno-0-100"></a>def run_artifact_collection(self, client_id: str, artifact: str) -> dict:

2237 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a> """

2238 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> Run an artifact collection on a specific client.

2239 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a>

2240 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Args:

2241 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a> client_id (str): The ID of the client.

2242 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a> artifact (str): The name of the artifact.

2243 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a>

2244 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> Returns:

2245 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> dict: A dictionary with the success status, a message, and potentially the results.

2246 -<a id="__codelineno-0-110" name="__codelineno-0-110"></a> """

2247 -<a id="__codelineno-0-111" name="__codelineno-0-111"></a> try:

2248 -<a id="__codelineno-0-112" name="__codelineno-0-112"></a> query = self._create_query(

2249 -<a id="__codelineno-0-113" name="__codelineno-0-113"></a> f"SELECT collect_client(client_id='{client_id}', artifacts=['{artifact}']) FROM scope()",

2250 -<a id="__codelineno-0-114" name="__codelineno-0-114"></a> )

2251 -<a id="__codelineno-0-115" name="__codelineno-0-115"></a> flow = self.universal_service.execute_query(query)

2252 -<a id="__codelineno-0-116" name="__codelineno-0-116"></a> logger.info(f"Successfully ran artifact collection on {flow}")

2253 -<a id="__codelineno-0-117" name="__codelineno-0-117"></a>

2254 -<a id="__codelineno-0-118" name="__codelineno-0-118"></a> artifact_key = self._get_artifact_key(client_id, artifact)

2255 -<a id="__codelineno-0-119" name="__codelineno-0-119"></a> flow_id = flow["results"][0][artifact_key]["flow_id"]

2256 -<a id="__codelineno-0-120" name="__codelineno-0-120"></a> logger.info(f"Successfully ran artifact collection on {flow_id}")

2257 -<a id="__codelineno-0-121" name="__codelineno-0-121"></a>

2258 -<a id="__codelineno-0-122" name="__codelineno-0-122"></a> completed = self.universal_service.watch_flow_completion(flow_id)

2259 -<a id="__codelineno-0-123" name="__codelineno-0-123"></a> logger.info(f"Successfully watched flow completion on {completed}")

2260 -<a id="__codelineno-0-124" name="__codelineno-0-124"></a>

2261 -<a id="__codelineno-0-125" name="__codelineno-0-125"></a> results = self.universal_service.read_collection_results(

2262 -<a id="__codelineno-0-126" name="__codelineno-0-126"></a> client_id,

2263 -<a id="__codelineno-0-127" name="__codelineno-0-127"></a> flow_id,

2264 -<a id="__codelineno-0-128" name="__codelineno-0-128"></a> artifact,

2265 -<a id="__codelineno-0-129" name="__codelineno-0-129"></a> )

2266 -<a id="__codelineno-0-130" name="__codelineno-0-130"></a> return results

2267 -<a id="__codelineno-0-131" name="__codelineno-0-131"></a> except Exception as err:

2268 -<a id="__codelineno-0-132" name="__codelineno-0-132"></a> logger.error(f"Failed to run artifact collection: {err}")

2269 -<a id="__codelineno-0-133" name="__codelineno-0-133"></a> return {

2270 -<a id="__codelineno-0-134" name="__codelineno-0-134"></a> "message": "Failed to run artifact collection",

2271 -<a id="__codelineno-0-135" name="__codelineno-0-135"></a> "success": False,

2272 -<a id="__codelineno-0-136" name="__codelineno-0-136"></a> }

2234 +<a href="#__codelineno-0-136">136</a>

2235 +<a href="#__codelineno-0-137">137</a>

2236 +<a href="#__codelineno-0-138">138</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-102" name="__codelineno-0-102"></a>def run_artifact_collection(self, client_id: str, artifact: str) -> dict:

2237 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> """

2238 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Run an artifact collection on a specific client.

2239 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a>

2240 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> Args:

2241 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a> client_id (str): The ID of the client.

2242 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a> artifact (str): The name of the artifact.

2243 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a>

2244 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> Returns:

2245 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a> dict: A dictionary with the success status, a message, and potentially the results.

2246 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a> """

2247 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> try:

2248 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> query = self._create_query(

2249 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> f"SELECT collect_client(client_id='{client_id}', artifacts=['{artifact}']) FROM scope()",

2250 +<a id="__codelineno-0-116" name="__codelineno-0-116"></a> )

2251 +<a id="__codelineno-0-117" name="__codelineno-0-117"></a> flow = self.universal_service.execute_query(query)

2252 +<a id="__codelineno-0-118" name="__codelineno-0-118"></a> logger.info(f"Successfully ran artifact collection on {flow}")

2253 +<a id="__codelineno-0-119" name="__codelineno-0-119"></a>

2254 +<a id="__codelineno-0-120" name="__codelineno-0-120"></a> artifact_key = self._get_artifact_key(client_id, artifact)

2255 +<a id="__codelineno-0-121" name="__codelineno-0-121"></a> flow_id = flow["results"][0][artifact_key]["flow_id"]

2256 +<a id="__codelineno-0-122" name="__codelineno-0-122"></a> logger.info(f"Successfully ran artifact collection on {flow_id}")

2257 +<a id="__codelineno-0-123" name="__codelineno-0-123"></a>

2258 +<a id="__codelineno-0-124" name="__codelineno-0-124"></a> completed = self.universal_service.watch_flow_completion(flow_id)

2259 +<a id="__codelineno-0-125" name="__codelineno-0-125"></a> logger.info(f"Successfully watched flow completion on {completed}")

2260 +<a id="__codelineno-0-126" name="__codelineno-0-126"></a>

2261 +<a id="__codelineno-0-127" name="__codelineno-0-127"></a> results = self.universal_service.read_collection_results(

2262 +<a id="__codelineno-0-128" name="__codelineno-0-128"></a> client_id,

2263 +<a id="__codelineno-0-129" name="__codelineno-0-129"></a> flow_id,

2264 +<a id="__codelineno-0-130" name="__codelineno-0-130"></a> artifact,

2265 +<a id="__codelineno-0-131" name="__codelineno-0-131"></a> )

2266 +<a id="__codelineno-0-132" name="__codelineno-0-132"></a> return results

2267 +<a id="__codelineno-0-133" name="__codelineno-0-133"></a> except Exception as err:

2268 +<a id="__codelineno-0-134" name="__codelineno-0-134"></a> logger.error(f"Failed to run artifact collection: {err}")

2269 +<a id="__codelineno-0-135" name="__codelineno-0-135"></a> return {

2270 +<a id="__codelineno-0-136" name="__codelineno-0-136"></a> "message": "Failed to run artifact collection",

2271 +<a id="__codelineno-0-137" name="__codelineno-0-137"></a> "success": False,

2272 +<a id="__codelineno-0-138" name="__codelineno-0-138"></a> }

2273 </code></pre></div></td></tr></table></div>

2274 </details>

2275 </div>

backend/site/cases/index.html

+121 -117

@@ -1051,11 +1051,7 @@ This class inherits from SQLAlchemy's Model class.

1051

1052 <details class="quote">

1053 <summary>Source code in <code>app\routes\dfir_iris.py</code></summary>

1054 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-53">53</a>

1055 -<a href="#__codelineno-0-54">54</a>

1056 -<a href="#__codelineno-0-55">55</a>

1057 -<a href="#__codelineno-0-56">56</a>

1058 -<a href="#__codelineno-0-57">57</a>

1054 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-57">57</a>

1055 <a href="#__codelineno-0-58">58</a>

1056 <a href="#__codelineno-0-59">59</a>

1057 <a href="#__codelineno-0-60">60</a>

@@ -1067,23 +1063,33 @@ This class inherits from SQLAlchemy's Model class.

1063 <a href="#__codelineno-0-66">66</a>

1064 <a href="#__codelineno-0-67">67</a>

1065 <a href="#__codelineno-0-68">68</a>

1070 -<a href="#__codelineno-0-69">69</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-53" name="__codelineno-0-53"></a>@bp.route("/dfir_iris/cases/<case_id>/note", methods=["POST"])

1071 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a>def create_case_note(case_id: str):

1072 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> """

1073 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> Endpoint to create a note for a specific case in DFIR IRIS.

1074 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a>

1075 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> Args:

1076 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> case_id (str): The ID of the case to create a note for.

1077 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a>

1078 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> Returns:

1079 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> json: A JSON response containing the result of the note creation operation.

1080 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> """

1081 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> note_title = request.json["note_title"]

1082 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> note_content = request.json["note_content"]

1083 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> case_service = CasesService()

1084 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> notes_service = NotesService()

1085 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> created_note = notes_service.create_case_note(case_id=case_id, note_title=note_title, note_content=note_content)

1086 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> return created_note

1066 +<a href="#__codelineno-0-69">69</a>

1067 +<a href="#__codelineno-0-70">70</a>

1068 +<a href="#__codelineno-0-71">71</a>

1069 +<a href="#__codelineno-0-72">72</a>

1070 +<a href="#__codelineno-0-73">73</a>

1071 +<a href="#__codelineno-0-74">74</a>

1072 +<a href="#__codelineno-0-75">75</a>

1073 +<a href="#__codelineno-0-76">76</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-57" name="__codelineno-0-57"></a>@bp.route("/dfir_iris/cases/<case_id>/note", methods=["POST"])

1074 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a>def create_case_note(case_id: str):

1075 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> """

1076 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> Endpoint to create a note for a specific case in DFIR IRIS.

1077 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a>

1078 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> Args:

1079 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> case_id (str): The ID of the case to create a note for.

1080 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a>

1081 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> Returns:

1082 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> json: A JSON response containing the result of the note creation operation.

1083 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> """

1084 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> note_title = request.json["note_title"]

1085 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> note_content = request.json["note_content"]

1086 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> notes_service = NotesService()

1087 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> created_note = notes_service.create_case_note(

1088 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> cid=case_id,

1089 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> note_title=note_title,

1090 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> note_content=note_content,

1091 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> )

1092 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> return created_note

1093 </code></pre></div></td></tr></table></div>

1094 </details>

1095 </div>

@@ -1127,27 +1133,27 @@ This class inherits from SQLAlchemy's Model class.

1133

1134 <details class="quote">

1135 <summary>Source code in <code>app\routes\dfir_iris.py</code></summary>

1130 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-86">86</a>

1131 -<a href="#__codelineno-0-87">87</a>

1132 -<a href="#__codelineno-0-88">88</a>

1133 -<a href="#__codelineno-0-89">89</a>

1134 -<a href="#__codelineno-0-90">90</a>

1135 -<a href="#__codelineno-0-91">91</a>

1136 -<a href="#__codelineno-0-92">92</a>

1137 -<a href="#__codelineno-0-93">93</a>

1138 -<a href="#__codelineno-0-94">94</a>

1139 -<a href="#__codelineno-0-95">95</a>

1140 -<a href="#__codelineno-0-96">96</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-86" name="__codelineno-0-86"></a>@bp.route("/dfir_iris/alerts", methods=["GET"])

1141 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a>def get_alerts():

1142 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> """

1143 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> Endpoint to retrieve all alerts from DFIR IRIS.

1144 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a>

1145 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> Returns:

1146 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> json: A JSON response containing the list of alerts.

1147 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> """

1148 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> service = AlertsService()

1149 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> alerts = service.list_alerts()

1150 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> return alerts

1136 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-95"> 95</a>

1137 +<a href="#__codelineno-0-96"> 96</a>

1138 +<a href="#__codelineno-0-97"> 97</a>

1139 +<a href="#__codelineno-0-98"> 98</a>

1140 +<a href="#__codelineno-0-99"> 99</a>

1141 +<a href="#__codelineno-0-100">100</a>

1142 +<a href="#__codelineno-0-101">101</a>

1143 +<a href="#__codelineno-0-102">102</a>

1144 +<a href="#__codelineno-0-103">103</a>

1145 +<a href="#__codelineno-0-104">104</a>

1146 +<a href="#__codelineno-0-105">105</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-95" name="__codelineno-0-95"></a>@bp.route("/dfir_iris/alerts", methods=["GET"])

1147 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a>def get_alerts():

1148 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> """

1149 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> Endpoint to retrieve all alerts from DFIR IRIS.

1150 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a>

1151 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> Returns:

1152 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a> json: A JSON response containing the list of alerts.

1153 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> """

1154 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> service = AlertsService()

1155 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> alerts = service.list_alerts()

1156 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a> return alerts

1157 </code></pre></div></td></tr></table></div>

1158 </details>

1159 </div>

@@ -1219,10 +1225,7 @@ This class inherits from SQLAlchemy's Model class.

1225

1226 <details class="quote">

1227 <summary>Source code in <code>app\routes\dfir_iris.py</code></summary>

1222 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-22">22</a>

1223 -<a href="#__codelineno-0-23">23</a>

1224 -<a href="#__codelineno-0-24">24</a>

1225 -<a href="#__codelineno-0-25">25</a>

1228 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-25">25</a>

1229 <a href="#__codelineno-0-26">26</a>

1230 <a href="#__codelineno-0-27">27</a>

1231 <a href="#__codelineno-0-28">28</a>

@@ -1232,20 +1235,23 @@ This class inherits from SQLAlchemy's Model class.

1235 <a href="#__codelineno-0-32">32</a>

1236 <a href="#__codelineno-0-33">33</a>

1237 <a href="#__codelineno-0-34">34</a>

1235 -<a href="#__codelineno-0-35">35</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-22" name="__codelineno-0-22"></a>@bp.route("/dfir_iris/cases/<case_id>", methods=["GET"])

1236 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a>def get_case(case_id: str):

1237 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> """

1238 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> Endpoint to retrieve a specific case from DFIR IRIS.

1239 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a>

1240 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> Args:

1241 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> case_id (str): The ID of the case to retrieve.

1238 +<a href="#__codelineno-0-35">35</a>

1239 +<a href="#__codelineno-0-36">36</a>

1240 +<a href="#__codelineno-0-37">37</a>

1241 +<a href="#__codelineno-0-38">38</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-25" name="__codelineno-0-25"></a>@bp.route("/dfir_iris/cases/<case_id>", methods=["GET"])

1242 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a>def get_case(case_id: str):

1243 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> """

1244 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> Endpoint to retrieve a specific case from DFIR IRIS.

1245 <a id="__codelineno-0-29" name="__codelineno-0-29"></a>

1243 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> Returns:

1244 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> json: A JSON response containing the case data.

1245 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> """

1246 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> service = CasesService()

1247 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> case = service.get_case(case_id=case_id)

1248 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> return case

1246 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> Args:

1247 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> case_id (str): The ID of the case to retrieve.

1248 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a>

1249 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> Returns:

1250 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> json: A JSON response containing the case data.

1251 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> """

1252 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> service = CasesService()

1253 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> case = service.get_case(case_id=case_id)

1254 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> return case

1255 </code></pre></div></td></tr></table></div>

1256 </details>

1257 </div>

@@ -1317,33 +1323,33 @@ This class inherits from SQLAlchemy's Model class.

1323

1324 <details class="quote">

1325 <summary>Source code in <code>app\routes\dfir_iris.py</code></summary>

1320 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-71">71</a>

1321 -<a href="#__codelineno-0-72">72</a>

1322 -<a href="#__codelineno-0-73">73</a>

1323 -<a href="#__codelineno-0-74">74</a>

1324 -<a href="#__codelineno-0-75">75</a>

1325 -<a href="#__codelineno-0-76">76</a>

1326 -<a href="#__codelineno-0-77">77</a>

1327 -<a href="#__codelineno-0-78">78</a>

1328 -<a href="#__codelineno-0-79">79</a>

1326 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-79">79</a>

1327 <a href="#__codelineno-0-80">80</a>

1328 <a href="#__codelineno-0-81">81</a>

1329 <a href="#__codelineno-0-82">82</a>

1330 <a href="#__codelineno-0-83">83</a>

1333 -<a href="#__codelineno-0-84">84</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-71" name="__codelineno-0-71"></a>@bp.route("/dfir_iris/cases/<case_id>/assets", methods=["GET"])

1334 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a>def get_case_assets(case_id: str):

1335 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> """

1336 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> Endpoint to retrieve assets of a specific case from DFIR IRIS.

1337 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a>

1338 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a> Args:

1339 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> case_id (str): The ID of the case to retrieve assets from.

1340 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a>

1341 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> Returns:

1342 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> json: A JSON response containing the list of assets for the case.

1343 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> """

1344 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> asset_service = AssetsService()

1345 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> assets = asset_service.get_case_assets(case_id=case_id)

1346 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> return assets

1331 +<a href="#__codelineno-0-84">84</a>

1332 +<a href="#__codelineno-0-85">85</a>

1333 +<a href="#__codelineno-0-86">86</a>

1334 +<a href="#__codelineno-0-87">87</a>

1335 +<a href="#__codelineno-0-88">88</a>

1336 +<a href="#__codelineno-0-89">89</a>

1337 +<a href="#__codelineno-0-90">90</a>

1338 +<a href="#__codelineno-0-91">91</a>

1339 +<a href="#__codelineno-0-92">92</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-79" name="__codelineno-0-79"></a>@bp.route("/dfir_iris/cases/<case_id>/assets", methods=["GET"])

1340 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a>def get_case_assets(case_id: str):

1341 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> """

1342 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> Endpoint to retrieve assets of a specific case from DFIR IRIS.

1343 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a>

1344 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Args:

1345 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> case_id (str): The ID of the case to retrieve assets from.

1346 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a>

1347 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> Returns:

1348 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> json: A JSON response containing the list of assets for the case.

1349 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> """

1350 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> asset_service = AssetsService()

1351 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> assets = asset_service.get_case_assets(cid=case_id)

1352 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> return assets

1353 </code></pre></div></td></tr></table></div>

1354 </details>

1355 </div>

@@ -1415,11 +1421,7 @@ This class inherits from SQLAlchemy's Model class.

1421

1422 <details class="quote">

1423 <summary>Source code in <code>app\routes\dfir_iris.py</code></summary>

1418 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-37">37</a>

1419 -<a href="#__codelineno-0-38">38</a>

1420 -<a href="#__codelineno-0-39">39</a>

1421 -<a href="#__codelineno-0-40">40</a>

1422 -<a href="#__codelineno-0-41">41</a>

1424 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-41">41</a>

1425 <a href="#__codelineno-0-42">42</a>

1426 <a href="#__codelineno-0-43">43</a>

1427 <a href="#__codelineno-0-44">44</a>

@@ -1429,21 +1431,23 @@ This class inherits from SQLAlchemy's Model class.

1431 <a href="#__codelineno-0-48">48</a>

1432 <a href="#__codelineno-0-49">49</a>

1433 <a href="#__codelineno-0-50">50</a>

1432 -<a href="#__codelineno-0-51">51</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-37" name="__codelineno-0-37"></a>@bp.route("/dfir_iris/cases/<case_id>/notes", methods=["GET"])

1433 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a>def get_case_notes(case_id: str):

1434 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> """

1435 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> Endpoint to retrieve notes of a specific case from DFIR IRIS.

1436 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a>

1437 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> Args:

1438 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> case_id (str): The ID of the case to retrieve notes from.

1439 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a>

1440 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> Returns:

1441 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> json: A JSON response containing the list of notes for the case.

1442 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> """

1443 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> case_service = CasesService()

1444 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> notes_service = NotesService()

1445 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> notes = notes_service.get_case_notes(case_id=case_id)

1446 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> return notes

1434 +<a href="#__codelineno-0-51">51</a>

1435 +<a href="#__codelineno-0-52">52</a>

1436 +<a href="#__codelineno-0-53">53</a>

1437 +<a href="#__codelineno-0-54">54</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-41" name="__codelineno-0-41"></a>@bp.route("/dfir_iris/cases/<case_id>/notes", methods=["GET"])

1438 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a>def get_case_notes(case_id: int):

1439 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> """

1440 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> Endpoint to retrieve notes of a specific case from DFIR IRIS.

1441 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a>

1442 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> Args:

1443 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> case_id (str): The ID of the case to retrieve notes from.

1444 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a>

1445 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> Returns:

1446 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> json: A JSON response containing the list of notes for the case.

1447 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> """

1448 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> notes_service = NotesService()

1449 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> notes = notes_service.get_case_notes(search_term="%", cid=case_id)

1450 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> return notes

1451 </code></pre></div></td></tr></table></div>

1452 </details>

1453 </div>

@@ -1487,9 +1491,7 @@ This class inherits from SQLAlchemy's Model class.

1491

1492 <details class="quote">

1493 <summary>Source code in <code>app\routes\dfir_iris.py</code></summary>

1490 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-10">10</a>

1491 -<a href="#__codelineno-0-11">11</a>

1492 -<a href="#__codelineno-0-12">12</a>

1494 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-12">12</a>

1495 <a href="#__codelineno-0-13">13</a>

1496 <a href="#__codelineno-0-14">14</a>

1497 <a href="#__codelineno-0-15">15</a>

@@ -1497,17 +1499,19 @@ This class inherits from SQLAlchemy's Model class.

1499 <a href="#__codelineno-0-17">17</a>

1500 <a href="#__codelineno-0-18">18</a>

1501 <a href="#__codelineno-0-19">19</a>

1500 -<a href="#__codelineno-0-20">20</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-10" name="__codelineno-0-10"></a>@bp.route("/dfir_iris/cases", methods=["GET"])

1501 -<a id="__codelineno-0-11" name="__codelineno-0-11"></a>def get_cases():

1502 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a> """

1503 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a> Endpoint to retrieve all the cases from DFIR IRIS.

1504 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a>

1505 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a> Returns:

1506 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> json: A JSON response containing the list of cases.

1507 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> """

1508 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> service = CasesService()

1509 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> cases = service.list_cases()

1510 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> return cases

1502 +<a href="#__codelineno-0-20">20</a>

1503 +<a href="#__codelineno-0-21">21</a>

1504 +<a href="#__codelineno-0-22">22</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-12" name="__codelineno-0-12"></a>@bp.route("/dfir_iris/cases", methods=["GET"])

1505 +<a id="__codelineno-0-13" name="__codelineno-0-13"></a>def get_cases():

1506 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> """

1507 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> Endpoint to retrieve all the cases from DFIR IRIS.

1508 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a>

1509 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a> Returns:

1510 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> json: A JSON response containing the list of cases.

1511 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a> """

1512 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> service = CasesService()

1513 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> cases = service.list_cases()

1514 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> return cases

1515 </code></pre></div></td></tr></table></div>

1516 </details>

1517 </div>

backend/site/connectors/index.html

+82 -82

@@ -3361,9 +3361,7 @@ the cluster's health.

3361

3362 <details class="quote">

3363 <summary>Source code in <code>app\routes\connectors.py</code></summary>

3364 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-34">34</a>

3365 -<a href="#__codelineno-0-35">35</a>

3366 -<a href="#__codelineno-0-36">36</a>

3364 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-36">36</a>

3365 <a href="#__codelineno-0-37">37</a>

3366 <a href="#__codelineno-0-38">38</a>

3367 <a href="#__codelineno-0-39">39</a>

@@ -3380,26 +3378,28 @@ the cluster's health.

3378 <a href="#__codelineno-0-50">50</a>

3379 <a href="#__codelineno-0-51">51</a>

3380 <a href="#__codelineno-0-52">52</a>

3383 -<a href="#__codelineno-0-53">53</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-34" name="__codelineno-0-34"></a>@bp.route("/connectors/<id>", methods=["GET"])

3384 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a>def get_connector_details(id: str):

3385 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> """

3386 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> Endpoint to retrieve the details of a connector.

3387 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a>

3388 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> Args:

3389 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> id (str): The ID of the connector to retrieve.

3390 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a>

3391 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> Returns:

3392 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> json: A JSON response containing the details of the connector.

3393 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> """

3394 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> service = ConnectorService(db)

3395 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> connector = service.validate_connector_exists(int(id))

3396 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a>

3397 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> if connector["success"]:

3398 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> connector = Connectors.query.get(id)

3399 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> instantiated_connector = service.process_connector(connector.connector_name)

3400 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> return jsonify(instantiated_connector)

3401 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> else:

3402 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> return jsonify(connector), 404

3381 +<a href="#__codelineno-0-53">53</a>

3382 +<a href="#__codelineno-0-54">54</a>

3383 +<a href="#__codelineno-0-55">55</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-36" name="__codelineno-0-36"></a>@bp.route("/connectors/<id>", methods=["GET"])

3384 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a>def get_connector_details(id: str):

3385 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> """

3386 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> Endpoint to retrieve the details of a connector.

3387 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a>

3388 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> Args:

3389 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a> id (str): The ID of the connector to retrieve.

3390 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a>

3391 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> Returns:

3392 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> json: A JSON response containing the details of the connector.

3393 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> """

3394 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> service = ConnectorService(db)

3395 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> connector = service.validate_connector_exists(int(id))

3396 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a>

3397 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> if connector["success"]:

3398 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> connector = Connectors.query.get(id)

3399 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> instantiated_connector = service.process_connector(connector.connector_name)

3400 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> return jsonify(instantiated_connector)

3401 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> else:

3402 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a> return jsonify(connector), 404

3403 </code></pre></div></td></tr></table></div>

3404 </details>

3405 </div>

@@ -3443,8 +3443,7 @@ the cluster's health.

3443

3444 <details class="quote">

3445 <summary>Source code in <code>app\routes\connectors.py</code></summary>

3446 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-14">14</a>

3447 -<a href="#__codelineno-0-15">15</a>

3446 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-15">15</a>

3447 <a href="#__codelineno-0-16">16</a>

3448 <a href="#__codelineno-0-17">17</a>

3449 <a href="#__codelineno-0-18">18</a>

@@ -3461,25 +3460,26 @@ the cluster's health.

3460 <a href="#__codelineno-0-29">29</a>

3461 <a href="#__codelineno-0-30">30</a>

3462 <a href="#__codelineno-0-31">31</a>

3464 -<a href="#__codelineno-0-32">32</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-14" name="__codelineno-0-14"></a>@bp.route("/connectors", methods=["GET"])

3465 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a>def list_connectors_available():

3466 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> """

3467 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> Endpoint to retrieve all available connectors.

3468 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a>

3469 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> Returns:

3470 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> json: A JSON response containing the list of all available connectors along with their connection verification status.

3471 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a> """

3472 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> connectors_service = ConnectorService(db)

3473 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a> connectors = ConnectorsAvailable.query.all()

3474 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> result = connectors_available_schema.dump(connectors)

3475 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a>

3476 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> instantiated_connectors = [

3477 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> connectors_service.process_connector(connector["connector_name"])

3478 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> for connector in result

3479 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> if connectors_service.process_connector(connector["connector_name"])

3480 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> ]

3481 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a>

3482 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> return jsonify(instantiated_connectors)

3463 +<a href="#__codelineno-0-32">32</a>

3464 +<a href="#__codelineno-0-33">33</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-15" name="__codelineno-0-15"></a>@bp.route("/connectors", methods=["GET"])

3465 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a>def list_connectors_available():

3466 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a> """

3467 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> Endpoint to retrieve all available connectors.

3468 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a>

3469 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> Returns:

3470 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> json: A JSON response containing the list of all available connectors along with their connection verification status.

3471 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> """

3472 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> connectors_service = ConnectorService(db)

3473 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a> connectors = ConnectorsAvailable.query.all()

3474 +<a id="__codelineno-0-25" name="__codelineno-0-25"></a> result = connectors_available_schema.dump(connectors)

3475 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a>

3476 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> instantiated_connectors = [

3477 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> connectors_service.process_connector(connector["connector_name"])

3478 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> for connector in result

3479 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> if connectors_service.process_connector(connector["connector_name"])

3480 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> ]

3481 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a>

3482 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> return jsonify(instantiated_connectors)

3483 </code></pre></div></td></tr></table></div>

3484 </details>

3485 </div>

@@ -3560,10 +3560,7 @@ the cluster's health.

3560

3561 <details class="quote">

3562 <summary>Source code in <code>app\routes\connectors.py</code></summary>

3563 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-55">55</a>

3564 -<a href="#__codelineno-0-56">56</a>

3565 -<a href="#__codelineno-0-57">57</a>

3566 -<a href="#__codelineno-0-58">58</a>

3563 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-58">58</a>

3564 <a href="#__codelineno-0-59">59</a>

3565 <a href="#__codelineno-0-60">60</a>

3566 <a href="#__codelineno-0-61">61</a>

@@ -3594,41 +3591,44 @@ the cluster's health.

3591 <a href="#__codelineno-0-86">86</a>

3592 <a href="#__codelineno-0-87">87</a>

3593 <a href="#__codelineno-0-88">88</a>

3597 -<a href="#__codelineno-0-89">89</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-55" name="__codelineno-0-55"></a>@bp.route("/connectors/<id>", methods=["PUT"])

3598 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a>def update_connector_route(id: str):

3599 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> """

3600 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> Endpoint to update the details of a connector.

3601 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a>

3602 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a> Args:

3603 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> id (str): The ID of the connector to update.

3594 +<a href="#__codelineno-0-89">89</a>

3595 +<a href="#__codelineno-0-90">90</a>

3596 +<a href="#__codelineno-0-91">91</a>

3597 +<a href="#__codelineno-0-92">92</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-58" name="__codelineno-0-58"></a>@bp.route("/connectors/<id>", methods=["PUT"])

3598 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a>def update_connector_route(id: str):

3599 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> """

3600 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> Endpoint to update the details of a connector.

3601 <a id="__codelineno-0-62" name="__codelineno-0-62"></a>

3605 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> Returns:

3606 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> json: A JSON response containing the success status of the update operation and a message indicating the status.

3607 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> If the update operation was successful, it returns the connector name and the status of the connection verification.

3608 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> """

3609 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> api_key_connector = ["Shuffle", "DFIR-IRIS", "Velociraptor"]

3610 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a>

3611 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> request_data = request.get_json()

3612 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> service = ConnectorService(db)

3613 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> connector = service.validate_connector_exists(int(id))

3614 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a>

3615 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> if connector["success"]:

3616 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> if connector["connector_name"] in api_key_connector:

3617 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> data_validated = service.validate_request_data_api_key(request_data)

3618 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a> if data_validated["success"]:

3619 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> service.update_connector(int(id), request_data)

3620 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> return service.verify_connector_connection(int(id))

3621 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> else:

3622 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> return jsonify(data_validated), 400

3623 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> else:

3624 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> data_validated = service.validate_request_data(request_data)

3625 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> if data_validated["success"]:

3626 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> service.update_connector(int(id), request_data)

3627 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a> return service.verify_connector_connection(int(id))

3628 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> else:

3629 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> return jsonify(data_validated), 400

3630 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> else:

3631 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> return jsonify(connector), 404

3602 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> Args:

3603 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> id (str): The ID of the connector to update.

3604 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a>

3605 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> Returns:

3606 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> json: A JSON response containing the success status of the update operation and a message indicating the status.

3607 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> If the update operation was successful, it returns the connector name and the status of the connection verification.

3608 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> """

3609 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> api_key_connector = ["Shuffle", "DFIR-IRIS", "Velociraptor"]

3610 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a>

3611 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> request_data = request.get_json()

3612 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> service = ConnectorService(db)

3613 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> connector = service.validate_connector_exists(int(id))

3614 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a>

3615 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> if connector["success"]:

3616 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> if connector["connector_name"] in api_key_connector:

3617 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> data_validated = service.validate_request_data_api_key(request_data)

3618 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> if data_validated["success"]:

3619 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> service.update_connector(int(id), request_data)

3620 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> return service.verify_connector_connection(int(id))

3621 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> else:

3622 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a> return jsonify(data_validated), 400

3623 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> else:

3624 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> data_validated = service.validate_request_data(request_data)

3625 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> if data_validated["success"]:

3626 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> service.update_connector(int(id), request_data)

3627 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> return service.verify_connector_connection(int(id))

3628 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> else:

3629 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> return jsonify(data_validated), 400

3630 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> else:

3631 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> return jsonify(connector), 404

3632 </code></pre></div></td></tr></table></div>

3633 </details>

3634 </div>

backend/site/graylog/index.html

+564 -526

@@ -2360,7 +2360,15 @@ This class inherits from SQLAlchemy's Model class.

2360 <a href="#__codelineno-0-127">127</a>

2361 <a href="#__codelineno-0-128">128</a>

2362 <a href="#__codelineno-0-129">129</a>

2363 -<a href="#__codelineno-0-130">130</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-14" name="__codelineno-0-14"></a>class InputsService:

2363 +<a href="#__codelineno-0-130">130</a>

2364 +<a href="#__codelineno-0-131">131</a>

2365 +<a href="#__codelineno-0-132">132</a>

2366 +<a href="#__codelineno-0-133">133</a>

2367 +<a href="#__codelineno-0-134">134</a>

2368 +<a href="#__codelineno-0-135">135</a>

2369 +<a href="#__codelineno-0-136">136</a>

2370 +<a href="#__codelineno-0-137">137</a>

2371 +<a href="#__codelineno-0-138">138</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-14" name="__codelineno-0-14"></a>class InputsService:

2372 <a id="__codelineno-0-15" name="__codelineno-0-15"></a> """

2373 <a id="__codelineno-0-16" name="__codelineno-0-16"></a> A service class that encapsulates the logic for pulling index data from Graylog

2374 <a id="__codelineno-0-17" name="__codelineno-0-17"></a> """

@@ -2377,106 +2385,114 @@ This class inherits from SQLAlchemy's Model class.

2385 <a id="__codelineno-0-28" name="__codelineno-0-28"></a> self.connector_password,

2386 <a id="__codelineno-0-29" name="__codelineno-0-29"></a> ) = UniversalService().collect_graylog_details("Graylog")

2387 <a id="__codelineno-0-30" name="__codelineno-0-30"></a>

2380 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> def collect_running_inputs(self) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2381 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> """

2382 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> Collects the running inputs that are managed by Graylog.

2383 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a>

2384 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> Returns:

2385 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> dict: A dictionary containing the success status, a message, and potentially a list of running inputs.

2386 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> """

2387 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a> if (

2388 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> self.connector_url is None

2389 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> or self.connector_username is None

2390 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> or self.connector_password is None

2391 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> ):

2392 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> return {"message": "Failed to collect Graylog details", "success": False}

2393 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a>

2394 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> running_inputs = self._collect_running_inputs()

2388 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> def collect_running_inputs(

2389 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> self,

2390 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> ) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2391 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> """

2392 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> Collects the running inputs that are managed by Graylog.

2393 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a>

2394 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> Returns:

2395 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> dict: A dictionary containing the success status, a message, and potentially a list of running inputs.

2396 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> """

2397 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> if (

2398 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> self.connector_url is None

2399 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a> or self.connector_username is None

2400 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> or self.connector_password is None

2401 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> ):

2402 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> return {"message": "Failed to collect Graylog details", "success": False}

2403 <a id="__codelineno-0-46" name="__codelineno-0-46"></a>

2396 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> if running_inputs["success"]:

2397 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> return running_inputs

2398 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a>

2399 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> def _collect_running_inputs(self) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2400 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> """

2401 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> Collects the running inputs that are managed by Graylog.

2402 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a>

2403 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> Returns:

2404 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a> dict: A dictionary containing the success status, a message, and potentially a list of running inputs.

2405 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> """

2406 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> try:

2407 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> running_inputs = requests.get(

2408 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> f"{self.connector_url}/api/system/inputstates",

2409 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a> headers=self.HEADERS,

2410 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> auth=(self.connector_username, self.connector_password),

2411 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> verify=False,

2412 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> )

2413 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> inputs_list = []

2414 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> for input in running_inputs.json()["states"]:

2415 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> inputs_list.append(

2416 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> {

2417 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> "state": input["state"],

2418 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> "title": input["message_input"]["title"],

2419 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "port": input["message_input"]["attributes"]["port"],

2420 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> },

2421 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> )

2422 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> return {

2423 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> "message": "Successfully collected running inputs",

2424 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> "success": True,

2425 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a> "inputs": inputs_list,

2426 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> }

2427 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> except Exception as e:

2428 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> logger.error(f"Failed to collect running inputs: {e}")

2429 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> return {"message": "Failed to collect running inputs", "success": False}

2430 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a>

2431 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> def collect_configured_inputs(self) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2432 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> """

2433 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Collects the configured inputs that are managed by Graylog.

2404 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> running_inputs = self._collect_running_inputs()

2405 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a>

2406 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> if running_inputs["success"]:

2407 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> return running_inputs

2408 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a>

2409 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> def _collect_running_inputs(

2410 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> self,

2411 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> ) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2412 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a> """

2413 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> Collects the running inputs that are managed by Graylog.

2414 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a>

2415 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a> Returns:

2416 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> dict: A dictionary containing the success status, a message, and potentially a list of running inputs.

2417 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> """

2418 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a> try:

2419 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> running_inputs = requests.get(

2420 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> f"{self.connector_url}/api/system/inputstates",

2421 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> headers=self.HEADERS,

2422 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> auth=(self.connector_username, self.connector_password),

2423 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> verify=False,

2424 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> )

2425 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> inputs_list = []

2426 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> for input in running_inputs.json()["states"]:

2427 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> inputs_list.append(

2428 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> {

2429 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> "state": input["state"],

2430 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> "title": input["message_input"]["title"],

2431 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> "port": input["message_input"]["attributes"]["port"],

2432 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> },

2433 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> )

2434 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> return {

2435 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> "message": "Successfully collected running inputs",

2436 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> "success": True,

2437 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> "inputs": inputs_list,

2438 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> }

2439 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> except Exception as e:

2440 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a> logger.error(f"Failed to collect running inputs: {e}")

2441 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> return {"message": "Failed to collect running inputs", "success": False}

2442 <a id="__codelineno-0-85" name="__codelineno-0-85"></a>

2435 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> Returns:

2436 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> dict: A dictionary containing the success status, a message, and potentially a list of configured inputs.

2437 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> """

2438 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> if (

2439 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> self.connector_url is None

2440 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> or self.connector_username is None

2441 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> or self.connector_password is None

2442 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> ):

2443 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> return {"message": "Failed to collect Graylog details", "success": False}

2444 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a>

2445 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> configured_inputs = self._collect_configured_inputs()

2446 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a>

2447 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> if configured_inputs["success"]:

2448 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> return configured_inputs

2449 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a>

2450 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a> def _collect_configured_inputs(self) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2451 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> """

2452 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a> Collects the configured inputs that are managed by Graylog.

2453 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a>

2454 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a> Returns:

2455 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a> dict: A dictionary containing the success status, a message, and potentially a list of configured inputs.

2456 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a> """

2457 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> try:

2458 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> configured_inputs = requests.get(

2459 -<a id="__codelineno-0-110" name="__codelineno-0-110"></a> f"{self.connector_url}/api/system/inputs",

2460 -<a id="__codelineno-0-111" name="__codelineno-0-111"></a> headers=self.HEADERS,

2461 -<a id="__codelineno-0-112" name="__codelineno-0-112"></a> auth=(self.connector_username, self.connector_password),

2462 -<a id="__codelineno-0-113" name="__codelineno-0-113"></a> verify=False,

2463 -<a id="__codelineno-0-114" name="__codelineno-0-114"></a> )

2464 -<a id="__codelineno-0-115" name="__codelineno-0-115"></a> configured_inputs_list = []

2465 -<a id="__codelineno-0-116" name="__codelineno-0-116"></a> for input in configured_inputs.json()["inputs"]:

2466 -<a id="__codelineno-0-117" name="__codelineno-0-117"></a> configured_inputs_list.append(

2467 -<a id="__codelineno-0-118" name="__codelineno-0-118"></a> {

2468 -<a id="__codelineno-0-119" name="__codelineno-0-119"></a> "title": input["title"],

2469 -<a id="__codelineno-0-120" name="__codelineno-0-120"></a> "port": input["attributes"]["port"],

2470 -<a id="__codelineno-0-121" name="__codelineno-0-121"></a> },

2471 -<a id="__codelineno-0-122" name="__codelineno-0-122"></a> )

2472 -<a id="__codelineno-0-123" name="__codelineno-0-123"></a> return {

2473 -<a id="__codelineno-0-124" name="__codelineno-0-124"></a> "message": "Successfully collected configured inputs",

2474 -<a id="__codelineno-0-125" name="__codelineno-0-125"></a> "success": True,

2475 -<a id="__codelineno-0-126" name="__codelineno-0-126"></a> "configured_inputs": configured_inputs_list,

2476 -<a id="__codelineno-0-127" name="__codelineno-0-127"></a> }

2477 -<a id="__codelineno-0-128" name="__codelineno-0-128"></a> except Exception as e:

2478 -<a id="__codelineno-0-129" name="__codelineno-0-129"></a> logger.error(f"Failed to collect configured inputs: {e}")

2479 -<a id="__codelineno-0-130" name="__codelineno-0-130"></a> return {"message": "Failed to collect configured inputs", "success": False}

2443 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> def collect_configured_inputs(

2444 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> self,

2445 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a> ) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2446 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> """

2447 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> Collects the configured inputs that are managed by Graylog.

2448 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a>

2449 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> Returns:

2450 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> dict: A dictionary containing the success status, a message, and potentially a list of configured inputs.

2451 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> """

2452 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> if (

2453 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> self.connector_url is None

2454 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> or self.connector_username is None

2455 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> or self.connector_password is None

2456 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> ):

2457 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> return {"message": "Failed to collect Graylog details", "success": False}

2458 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a>

2459 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> configured_inputs = self._collect_configured_inputs()

2460 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a>

2461 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> if configured_inputs["success"]:

2462 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a> return configured_inputs

2463 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a>

2464 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a> def _collect_configured_inputs(

2465 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a> self,

2466 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a> ) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2467 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> """

2468 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a> Collects the configured inputs that are managed by Graylog.

2469 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a>

2470 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> Returns:

2471 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> dict: A dictionary containing the success status, a message, and potentially a list of configured inputs.

2472 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> """

2473 +<a id="__codelineno-0-116" name="__codelineno-0-116"></a> try:

2474 +<a id="__codelineno-0-117" name="__codelineno-0-117"></a> configured_inputs = requests.get(

2475 +<a id="__codelineno-0-118" name="__codelineno-0-118"></a> f"{self.connector_url}/api/system/inputs",

2476 +<a id="__codelineno-0-119" name="__codelineno-0-119"></a> headers=self.HEADERS,

2477 +<a id="__codelineno-0-120" name="__codelineno-0-120"></a> auth=(self.connector_username, self.connector_password),

2478 +<a id="__codelineno-0-121" name="__codelineno-0-121"></a> verify=False,

2479 +<a id="__codelineno-0-122" name="__codelineno-0-122"></a> )

2480 +<a id="__codelineno-0-123" name="__codelineno-0-123"></a> configured_inputs_list = []

2481 +<a id="__codelineno-0-124" name="__codelineno-0-124"></a> for input in configured_inputs.json()["inputs"]:

2482 +<a id="__codelineno-0-125" name="__codelineno-0-125"></a> configured_inputs_list.append(

2483 +<a id="__codelineno-0-126" name="__codelineno-0-126"></a> {

2484 +<a id="__codelineno-0-127" name="__codelineno-0-127"></a> "title": input["title"],

2485 +<a id="__codelineno-0-128" name="__codelineno-0-128"></a> "port": input["attributes"]["port"],

2486 +<a id="__codelineno-0-129" name="__codelineno-0-129"></a> },

2487 +<a id="__codelineno-0-130" name="__codelineno-0-130"></a> )

2488 +<a id="__codelineno-0-131" name="__codelineno-0-131"></a> return {

2489 +<a id="__codelineno-0-132" name="__codelineno-0-132"></a> "message": "Successfully collected configured inputs",

2490 +<a id="__codelineno-0-133" name="__codelineno-0-133"></a> "success": True,

2491 +<a id="__codelineno-0-134" name="__codelineno-0-134"></a> "configured_inputs": configured_inputs_list,

2492 +<a id="__codelineno-0-135" name="__codelineno-0-135"></a> }

2493 +<a id="__codelineno-0-136" name="__codelineno-0-136"></a> except Exception as e:

2494 +<a id="__codelineno-0-137" name="__codelineno-0-137"></a> logger.error(f"Failed to collect configured inputs: {e}")

2495 +<a id="__codelineno-0-138" name="__codelineno-0-138"></a> return {"message": "Failed to collect configured inputs", "success": False}

2496 </code></pre></div></td></tr></table></div>

2497 </details>

2498

@@ -2569,41 +2585,45 @@ This class inherits from SQLAlchemy's Model class.

2585

2586 <details class="quote">

2587 <summary>Source code in <code>app\services\Graylog\inputs.py</code></summary>

2572 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-82">82</a>

2573 -<a href="#__codelineno-0-83">83</a>

2574 -<a href="#__codelineno-0-84">84</a>

2575 -<a href="#__codelineno-0-85">85</a>

2576 -<a href="#__codelineno-0-86">86</a>

2577 -<a href="#__codelineno-0-87">87</a>

2578 -<a href="#__codelineno-0-88">88</a>

2579 -<a href="#__codelineno-0-89">89</a>

2580 -<a href="#__codelineno-0-90">90</a>

2581 -<a href="#__codelineno-0-91">91</a>

2582 -<a href="#__codelineno-0-92">92</a>

2583 -<a href="#__codelineno-0-93">93</a>

2584 -<a href="#__codelineno-0-94">94</a>

2585 -<a href="#__codelineno-0-95">95</a>

2586 -<a href="#__codelineno-0-96">96</a>

2587 -<a href="#__codelineno-0-97">97</a>

2588 -<a href="#__codelineno-0-98">98</a>

2589 -<a href="#__codelineno-0-99">99</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-82" name="__codelineno-0-82"></a>def collect_configured_inputs(self) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2590 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> """

2591 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> Collects the configured inputs that are managed by Graylog.

2592 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a>

2593 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a> Returns:

2594 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> dict: A dictionary containing the success status, a message, and potentially a list of configured inputs.

2595 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> """

2596 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> if (

2597 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> self.connector_url is None

2598 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> or self.connector_username is None

2599 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a> or self.connector_password is None

2600 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> ):

2601 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> return {"message": "Failed to collect Graylog details", "success": False}

2602 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a>

2603 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a> configured_inputs = self._collect_configured_inputs()

2604 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a>

2605 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> if configured_inputs["success"]:

2606 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> return configured_inputs

2588 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-86"> 86</a>

2589 +<a href="#__codelineno-0-87"> 87</a>

2590 +<a href="#__codelineno-0-88"> 88</a>

2591 +<a href="#__codelineno-0-89"> 89</a>

2592 +<a href="#__codelineno-0-90"> 90</a>

2593 +<a href="#__codelineno-0-91"> 91</a>

2594 +<a href="#__codelineno-0-92"> 92</a>

2595 +<a href="#__codelineno-0-93"> 93</a>

2596 +<a href="#__codelineno-0-94"> 94</a>

2597 +<a href="#__codelineno-0-95"> 95</a>

2598 +<a href="#__codelineno-0-96"> 96</a>

2599 +<a href="#__codelineno-0-97"> 97</a>

2600 +<a href="#__codelineno-0-98"> 98</a>

2601 +<a href="#__codelineno-0-99"> 99</a>

2602 +<a href="#__codelineno-0-100">100</a>

2603 +<a href="#__codelineno-0-101">101</a>

2604 +<a href="#__codelineno-0-102">102</a>

2605 +<a href="#__codelineno-0-103">103</a>

2606 +<a href="#__codelineno-0-104">104</a>

2607 +<a href="#__codelineno-0-105">105</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-86" name="__codelineno-0-86"></a>def collect_configured_inputs(

2608 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> self,

2609 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a>) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2610 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> """

2611 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> Collects the configured inputs that are managed by Graylog.

2612 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a>

2613 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> Returns:

2614 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> dict: A dictionary containing the success status, a message, and potentially a list of configured inputs.

2615 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a> """

2616 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> if (

2617 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> self.connector_url is None

2618 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> or self.connector_username is None

2619 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a> or self.connector_password is None

2620 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> ):

2621 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> return {"message": "Failed to collect Graylog details", "success": False}

2622 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a>

2623 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> configured_inputs = self._collect_configured_inputs()

2624 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a>

2625 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> if configured_inputs["success"]:

2626 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a> return configured_inputs

2627 </code></pre></div></td></tr></table></div>

2628 </details>

2629 </div>

@@ -2665,24 +2685,28 @@ This class inherits from SQLAlchemy's Model class.

2685 <a href="#__codelineno-0-45">45</a>

2686 <a href="#__codelineno-0-46">46</a>

2687 <a href="#__codelineno-0-47">47</a>

2668 -<a href="#__codelineno-0-48">48</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-31" name="__codelineno-0-31"></a>def collect_running_inputs(self) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2669 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> """

2670 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> Collects the running inputs that are managed by Graylog.

2671 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a>

2672 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> Returns:

2673 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> dict: A dictionary containing the success status, a message, and potentially a list of running inputs.

2674 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a> """

2675 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a> if (

2676 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> self.connector_url is None

2677 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> or self.connector_username is None

2678 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a> or self.connector_password is None

2679 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> ):

2680 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> return {"message": "Failed to collect Graylog details", "success": False}

2681 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a>

2682 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> running_inputs = self._collect_running_inputs()

2688 +<a href="#__codelineno-0-48">48</a>

2689 +<a href="#__codelineno-0-49">49</a>

2690 +<a href="#__codelineno-0-50">50</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-31" name="__codelineno-0-31"></a>def collect_running_inputs(

2691 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> self,

2692 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a>) -> Dict[str, Union[bool, str, List[Dict[str, Union[str, int]]]]]:

2693 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> """

2694 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> Collects the running inputs that are managed by Graylog.

2695 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a>

2696 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> Returns:

2697 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> dict: A dictionary containing the success status, a message, and potentially a list of running inputs.

2698 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> """

2699 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> if (

2700 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> self.connector_url is None

2701 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a> or self.connector_username is None

2702 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a> or self.connector_password is None

2703 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> ):

2704 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> return {"message": "Failed to collect Graylog details", "success": False}

2705 <a id="__codelineno-0-46" name="__codelineno-0-46"></a>

2684 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> if running_inputs["success"]:

2685 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> return running_inputs

2706 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a> running_inputs = self._collect_running_inputs()

2707 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a>

2708 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> if running_inputs["success"]:

2709 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> return running_inputs

2710 </code></pre></div></td></tr></table></div>

2711 </details>

2712 </div>

@@ -2742,10 +2766,7 @@ This class inherits from SQLAlchemy's Model class.

2766

2767 <details class="quote">

2768 <summary>Source code in <code>app\services\Graylog\messages.py</code></summary>

2745 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-10">10</a>

2746 -<a href="#__codelineno-0-11">11</a>

2747 -<a href="#__codelineno-0-12">12</a>

2748 -<a href="#__codelineno-0-13">13</a>

2769 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-13">13</a>

2770 <a href="#__codelineno-0-14">14</a>

2771 <a href="#__codelineno-0-15">15</a>

2772 <a href="#__codelineno-0-16">16</a>

@@ -2806,71 +2827,80 @@ This class inherits from SQLAlchemy's Model class.

2827 <a href="#__codelineno-0-71">71</a>

2828 <a href="#__codelineno-0-72">72</a>

2829 <a href="#__codelineno-0-73">73</a>

2809 -<a href="#__codelineno-0-74">74</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-10" name="__codelineno-0-10"></a>class MessagesService:

2810 -<a id="__codelineno-0-11" name="__codelineno-0-11"></a> """

2811 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a> A service class that encapsulates the logic for polling messages from Graylog.

2812 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a> """

2813 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a>

2814 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a> def __init__(self):

2815 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> (

2816 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a> self.connector_url,

2817 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> self.connector_username,

2818 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> self.connector_password,

2819 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a> ) = UniversalService().collect_graylog_details("Graylog")

2820 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a>

2821 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> def _get_messages_from_graylog(self, page_number: int) -> requests.Response:

2822 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a> """Fetches messages from Graylog for a specific page number."""

2823 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> return requests.get(

2824 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> f"{self.connector_url}/api/system/messages?page={page_number}",

2825 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> auth=(self.connector_username, self.connector_password),

2826 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> verify=False,

2827 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> )

2828 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a>

2829 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> def _handle_message_fetch_error(self, error: Exception) -> Dict[str, Union[str, bool]]:

2830 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> """Handles exceptions occurred while fetching messages."""

2831 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> logger.error(f"Failed to collect messages from Graylog: {error}")

2832 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> return {

2833 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> "message": "Failed to collect messages from Graylog",

2834 -<a id="__codelineno-0-35" name="__codelineno-0-35"></a> "success": False,

2835 -<a id="__codelineno-0-36" name="__codelineno-0-36"></a> }

2836 -<a id="__codelineno-0-37" name="__codelineno-0-37"></a>

2837 -<a id="__codelineno-0-38" name="__codelineno-0-38"></a> def collect_messages(self) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

2838 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> """

2839 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> Collects the latest 10 messages from Graylog.

2840 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a>

2841 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> Returns:

2842 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> dict: A dictionary containing the success status, a message, and potentially a list of Graylog messages.

2843 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> """

2844 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> if (

2845 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> self.connector_url is None

2846 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> or self.connector_username is None

2847 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> or self.connector_password is None

2848 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> ):

2849 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> return {"message": "Failed to collect Graylog details", "success": False}

2850 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> else:

2851 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> try:

2852 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> page_number = 1

2853 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> graylog_messages = self._get_messages_from_graylog(page_number)

2854 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a>

2855 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> if graylog_messages.status_code == 200:

2856 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> logger.info(

2857 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> f"Received {len(graylog_messages.json()['messages'])} messages from Graylog",

2858 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> )

2859 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a> return {

2860 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> "message": "Successfully retrieved messages",

2861 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> "success": True,

2862 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> "graylog_messages": graylog_messages.json()["messages"],

2863 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> }

2864 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> else:

2865 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> logger.error(

2866 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> f"Failed to collect messages from Graylog: {graylog_messages.json()}",

2867 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> )

2868 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> return {

2869 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "message": "Failed to collect messages from Graylog",

2870 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "success": False,

2871 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> }

2872 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> except Exception as e:

2873 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> return self._handle_message_fetch_error(e)

2830 +<a href="#__codelineno-0-74">74</a>

2831 +<a href="#__codelineno-0-75">75</a>

2832 +<a href="#__codelineno-0-76">76</a>

2833 +<a href="#__codelineno-0-77">77</a>

2834 +<a href="#__codelineno-0-78">78</a>

2835 +<a href="#__codelineno-0-79">79</a>

2836 +<a href="#__codelineno-0-80">80</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-13" name="__codelineno-0-13"></a>class MessagesService:

2837 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> """

2838 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> A service class that encapsulates the logic for polling messages from Graylog.

2839 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a> """

2840 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a>

2841 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> def __init__(self):

2842 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a> (

2843 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> self.connector_url,

2844 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> self.connector_username,

2845 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> self.connector_password,

2846 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> ) = UniversalService().collect_graylog_details("Graylog")

2847 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a>

2848 +<a id="__codelineno-0-25" name="__codelineno-0-25"></a> def _get_messages_from_graylog(self, page_number: int) -> requests.Response:

2849 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a> """Fetches messages from Graylog for a specific page number."""

2850 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> return requests.get(

2851 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> f"{self.connector_url}/api/system/messages?page={page_number}",

2852 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> auth=(self.connector_username, self.connector_password),

2853 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> verify=False,

2854 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> )

2855 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a>

2856 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> def _handle_message_fetch_error(

2857 +<a id="__codelineno-0-34" name="__codelineno-0-34"></a> self,

2858 +<a id="__codelineno-0-35" name="__codelineno-0-35"></a> error: Exception,

2859 +<a id="__codelineno-0-36" name="__codelineno-0-36"></a> ) -> Dict[str, Union[str, bool]]:

2860 +<a id="__codelineno-0-37" name="__codelineno-0-37"></a> """Handles exceptions occurred while fetching messages."""

2861 +<a id="__codelineno-0-38" name="__codelineno-0-38"></a> logger.error(f"Failed to collect messages from Graylog: {error}")

2862 +<a id="__codelineno-0-39" name="__codelineno-0-39"></a> return {

2863 +<a id="__codelineno-0-40" name="__codelineno-0-40"></a> "message": "Failed to collect messages from Graylog",

2864 +<a id="__codelineno-0-41" name="__codelineno-0-41"></a> "success": False,

2865 +<a id="__codelineno-0-42" name="__codelineno-0-42"></a> }

2866 +<a id="__codelineno-0-43" name="__codelineno-0-43"></a>

2867 +<a id="__codelineno-0-44" name="__codelineno-0-44"></a> def collect_messages(self) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

2868 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> """

2869 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> Collects the latest 10 messages from Graylog.

2870 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a>

2871 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> Returns:

2872 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> dict: A dictionary containing the success status, a message, and potentially a list of Graylog messages.

2873 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> """

2874 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> if (

2875 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> self.connector_url is None

2876 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> or self.connector_username is None

2877 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> or self.connector_password is None

2878 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a> ):

2879 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> return {"message": "Failed to collect Graylog details", "success": False}

2880 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a> else:

2881 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a> try:

2882 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> page_number = 1

2883 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> graylog_messages = self._get_messages_from_graylog(page_number)

2884 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a>

2885 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> if graylog_messages.status_code == 200:

2886 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> logger.info(

2887 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> f"Received {len(graylog_messages.json()['messages'])} messages from Graylog",

2888 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> )

2889 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> return {

2890 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> "message": "Successfully retrieved messages",

2891 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> "success": True,

2892 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> "graylog_messages": graylog_messages.json()["messages"],

2893 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> }

2894 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> else:

2895 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> logger.error(

2896 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> f"Failed to collect messages from Graylog: {graylog_messages.json()}",

2897 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> )

2898 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> return {

2899 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> "message": "Failed to collect messages from Graylog",

2900 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> "success": False,

2901 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> }

2902 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> except Exception as e:

2903 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> return self._handle_message_fetch_error(e)

2904 </code></pre></div></td></tr></table></div>

2905 </details>

2906

@@ -2924,13 +2954,7 @@ This class inherits from SQLAlchemy's Model class.

2954

2955 <details class="quote">

2956 <summary>Source code in <code>app\services\Graylog\messages.py</code></summary>

2927 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-38">38</a>

2928 -<a href="#__codelineno-0-39">39</a>

2929 -<a href="#__codelineno-0-40">40</a>

2930 -<a href="#__codelineno-0-41">41</a>

2931 -<a href="#__codelineno-0-42">42</a>

2932 -<a href="#__codelineno-0-43">43</a>

2933 -<a href="#__codelineno-0-44">44</a>

2957 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-44">44</a>

2958 <a href="#__codelineno-0-45">45</a>

2959 <a href="#__codelineno-0-46">46</a>

2960 <a href="#__codelineno-0-47">47</a>

@@ -2960,43 +2984,49 @@ This class inherits from SQLAlchemy's Model class.

2984 <a href="#__codelineno-0-71">71</a>

2985 <a href="#__codelineno-0-72">72</a>

2986 <a href="#__codelineno-0-73">73</a>

2963 -<a href="#__codelineno-0-74">74</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-38" name="__codelineno-0-38"></a>def collect_messages(self) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

2964 -<a id="__codelineno-0-39" name="__codelineno-0-39"></a> """

2965 -<a id="__codelineno-0-40" name="__codelineno-0-40"></a> Collects the latest 10 messages from Graylog.

2966 -<a id="__codelineno-0-41" name="__codelineno-0-41"></a>

2967 -<a id="__codelineno-0-42" name="__codelineno-0-42"></a> Returns:

2968 -<a id="__codelineno-0-43" name="__codelineno-0-43"></a> dict: A dictionary containing the success status, a message, and potentially a list of Graylog messages.

2969 -<a id="__codelineno-0-44" name="__codelineno-0-44"></a> """

2970 -<a id="__codelineno-0-45" name="__codelineno-0-45"></a> if (

2971 -<a id="__codelineno-0-46" name="__codelineno-0-46"></a> self.connector_url is None

2972 -<a id="__codelineno-0-47" name="__codelineno-0-47"></a> or self.connector_username is None

2973 -<a id="__codelineno-0-48" name="__codelineno-0-48"></a> or self.connector_password is None

2974 -<a id="__codelineno-0-49" name="__codelineno-0-49"></a> ):

2975 -<a id="__codelineno-0-50" name="__codelineno-0-50"></a> return {"message": "Failed to collect Graylog details", "success": False}

2976 -<a id="__codelineno-0-51" name="__codelineno-0-51"></a> else:

2977 -<a id="__codelineno-0-52" name="__codelineno-0-52"></a> try:

2978 -<a id="__codelineno-0-53" name="__codelineno-0-53"></a> page_number = 1

2979 -<a id="__codelineno-0-54" name="__codelineno-0-54"></a> graylog_messages = self._get_messages_from_graylog(page_number)

2980 -<a id="__codelineno-0-55" name="__codelineno-0-55"></a>

2981 -<a id="__codelineno-0-56" name="__codelineno-0-56"></a> if graylog_messages.status_code == 200:

2982 -<a id="__codelineno-0-57" name="__codelineno-0-57"></a> logger.info(

2983 -<a id="__codelineno-0-58" name="__codelineno-0-58"></a> f"Received {len(graylog_messages.json()['messages'])} messages from Graylog",

2984 -<a id="__codelineno-0-59" name="__codelineno-0-59"></a> )

2985 -<a id="__codelineno-0-60" name="__codelineno-0-60"></a> return {

2986 -<a id="__codelineno-0-61" name="__codelineno-0-61"></a> "message": "Successfully retrieved messages",

2987 -<a id="__codelineno-0-62" name="__codelineno-0-62"></a> "success": True,

2988 -<a id="__codelineno-0-63" name="__codelineno-0-63"></a> "graylog_messages": graylog_messages.json()["messages"],

2989 -<a id="__codelineno-0-64" name="__codelineno-0-64"></a> }

2990 -<a id="__codelineno-0-65" name="__codelineno-0-65"></a> else:

2991 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> logger.error(

2992 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> f"Failed to collect messages from Graylog: {graylog_messages.json()}",

2993 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> )

2994 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> return {

2995 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> "message": "Failed to collect messages from Graylog",

2996 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> "success": False,

2997 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> }

2998 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> except Exception as e:

2999 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> return self._handle_message_fetch_error(e)

2987 +<a href="#__codelineno-0-74">74</a>

2988 +<a href="#__codelineno-0-75">75</a>

2989 +<a href="#__codelineno-0-76">76</a>

2990 +<a href="#__codelineno-0-77">77</a>

2991 +<a href="#__codelineno-0-78">78</a>

2992 +<a href="#__codelineno-0-79">79</a>

2993 +<a href="#__codelineno-0-80">80</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-44" name="__codelineno-0-44"></a>def collect_messages(self) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

2994 +<a id="__codelineno-0-45" name="__codelineno-0-45"></a> """

2995 +<a id="__codelineno-0-46" name="__codelineno-0-46"></a> Collects the latest 10 messages from Graylog.

2996 +<a id="__codelineno-0-47" name="__codelineno-0-47"></a>

2997 +<a id="__codelineno-0-48" name="__codelineno-0-48"></a> Returns:

2998 +<a id="__codelineno-0-49" name="__codelineno-0-49"></a> dict: A dictionary containing the success status, a message, and potentially a list of Graylog messages.

2999 +<a id="__codelineno-0-50" name="__codelineno-0-50"></a> """

3000 +<a id="__codelineno-0-51" name="__codelineno-0-51"></a> if (

3001 +<a id="__codelineno-0-52" name="__codelineno-0-52"></a> self.connector_url is None

3002 +<a id="__codelineno-0-53" name="__codelineno-0-53"></a> or self.connector_username is None

3003 +<a id="__codelineno-0-54" name="__codelineno-0-54"></a> or self.connector_password is None

3004 +<a id="__codelineno-0-55" name="__codelineno-0-55"></a> ):

3005 +<a id="__codelineno-0-56" name="__codelineno-0-56"></a> return {"message": "Failed to collect Graylog details", "success": False}

3006 +<a id="__codelineno-0-57" name="__codelineno-0-57"></a> else:

3007 +<a id="__codelineno-0-58" name="__codelineno-0-58"></a> try:

3008 +<a id="__codelineno-0-59" name="__codelineno-0-59"></a> page_number = 1

3009 +<a id="__codelineno-0-60" name="__codelineno-0-60"></a> graylog_messages = self._get_messages_from_graylog(page_number)

3010 +<a id="__codelineno-0-61" name="__codelineno-0-61"></a>

3011 +<a id="__codelineno-0-62" name="__codelineno-0-62"></a> if graylog_messages.status_code == 200:

3012 +<a id="__codelineno-0-63" name="__codelineno-0-63"></a> logger.info(

3013 +<a id="__codelineno-0-64" name="__codelineno-0-64"></a> f"Received {len(graylog_messages.json()['messages'])} messages from Graylog",

3014 +<a id="__codelineno-0-65" name="__codelineno-0-65"></a> )

3015 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> return {

3016 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> "message": "Successfully retrieved messages",

3017 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> "success": True,

3018 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> "graylog_messages": graylog_messages.json()["messages"],

3019 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> }

3020 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> else:

3021 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> logger.error(

3022 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> f"Failed to collect messages from Graylog: {graylog_messages.json()}",

3023 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> )

3024 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> return {

3025 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> "message": "Failed to collect messages from Graylog",

3026 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> "success": False,

3027 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a> }

3028 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> except Exception as e:

3029 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> return self._handle_message_fetch_error(e)

3030 </code></pre></div></td></tr></table></div>

3031 </details>

3032 </div>

@@ -3283,7 +3313,9 @@ This class inherits from SQLAlchemy's Model class.

3313 <a href="#__codelineno-0-239">239</a>

3314 <a href="#__codelineno-0-240">240</a>

3315 <a href="#__codelineno-0-241">241</a>

3286 -<a href="#__codelineno-0-242">242</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-15" name="__codelineno-0-15"></a>class MetricsService:

3316 +<a href="#__codelineno-0-242">242</a>

3317 +<a href="#__codelineno-0-243">243</a>

3318 +<a href="#__codelineno-0-244">244</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-15" name="__codelineno-0-15"></a>class MetricsService:

3319 <a id="__codelineno-0-16" name="__codelineno-0-16"></a> """

3320 <a id="__codelineno-0-17" name="__codelineno-0-17"></a> A service class that encapsulates the logic for pulling metrics from Graylog.

3321 <a id="__codelineno-0-18" name="__codelineno-0-18"></a> """

@@ -3334,183 +3366,185 @@ This class inherits from SQLAlchemy's Model class.

3366 <a id="__codelineno-0-63" name="__codelineno-0-63"></a> return journal_size

3367 <a id="__codelineno-0-64" name="__codelineno-0-64"></a> return journal_size

3368 <a id="__codelineno-0-65" name="__codelineno-0-65"></a>

3337 -<a id="__codelineno-0-66" name="__codelineno-0-66"></a> def collect_throughput_metrics(self) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3338 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> """

3339 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> Collects the following Graylog Metrics:

3340 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> - Input Usage

3341 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> - Output Usage

3342 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> - Processor Usage

3343 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> - Input 1 Seconds Rate

3344 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> - Output 1 Seconds Rate

3345 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> - Total Input

3346 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> - Total Output

3347 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a>

3348 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Returns:

3349 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> dict: A dictionary containing the success status, a message, and the list of throughput metrics.

3350 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> """

3351 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> if (

3352 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> self.connector_url is None

3353 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> or self.connector_username is None

3354 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> or self.connector_password is None

3355 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> ):

3356 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a> return {"message": "Failed to collect Graylog details", "success": False}

3357 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a>

3358 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> throughput_usage = self._collect_metrics_throughput_usage(

3359 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> self.connector_url,

3360 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> self.connector_username,

3361 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> self.connector_password,

3362 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> )

3363 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a>

3364 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> if throughput_usage["success"] is False:

3365 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> return throughput_usage

3366 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> return throughput_usage

3367 -<a id="__codelineno-0-96" name="__codelineno-0-96"></a>

3368 -<a id="__codelineno-0-97" name="__codelineno-0-97"></a> def _collect_metrics_uncommitted_journal_size(

3369 -<a id="__codelineno-0-98" name="__codelineno-0-98"></a> self,

3370 -<a id="__codelineno-0-99" name="__codelineno-0-99"></a> connector_url: str,

3371 -<a id="__codelineno-0-100" name="__codelineno-0-100"></a> connector_username: str,

3372 -<a id="__codelineno-0-101" name="__codelineno-0-101"></a> connector_password: str,

3373 -<a id="__codelineno-0-102" name="__codelineno-0-102"></a> ) -> Dict[str, Union[str, bool, int]]:

3374 -<a id="__codelineno-0-103" name="__codelineno-0-103"></a> """

3375 -<a id="__codelineno-0-104" name="__codelineno-0-104"></a> Collects the journal size of uncommitted messages from Graylog.

3376 -<a id="__codelineno-0-105" name="__codelineno-0-105"></a>

3377 -<a id="__codelineno-0-106" name="__codelineno-0-106"></a> Args:

3378 -<a id="__codelineno-0-107" name="__codelineno-0-107"></a> connector_url (str): The URL of the Graylog connector.

3379 -<a id="__codelineno-0-108" name="__codelineno-0-108"></a> connector_username (str): The username of the Graylog connector.

3380 -<a id="__codelineno-0-109" name="__codelineno-0-109"></a> connector_password (str): The password of the Graylog connector.

3381 -<a id="__codelineno-0-110" name="__codelineno-0-110"></a>

3382 -<a id="__codelineno-0-111" name="__codelineno-0-111"></a> Returns:

3383 -<a id="__codelineno-0-112" name="__codelineno-0-112"></a> dict: A dictionary containing the success status, a message, and the size of uncommitted journal entries.

3384 -<a id="__codelineno-0-113" name="__codelineno-0-113"></a> """

3385 -<a id="__codelineno-0-114" name="__codelineno-0-114"></a> try:

3386 -<a id="__codelineno-0-115" name="__codelineno-0-115"></a> logger.info("Collecting journal size from Graylog")

3387 -<a id="__codelineno-0-116" name="__codelineno-0-116"></a> headers = {"X-Requested-By": "CoPilot"}

3388 -<a id="__codelineno-0-117" name="__codelineno-0-117"></a> # Get the Graylog Journal Size

3389 -<a id="__codelineno-0-118" name="__codelineno-0-118"></a> uncommitted_journal_size_response = requests.get(

3390 -<a id="__codelineno-0-119" name="__codelineno-0-119"></a> f"{connector_url}/api/system/journal",

3391 -<a id="__codelineno-0-120" name="__codelineno-0-120"></a> headers=headers,

3392 -<a id="__codelineno-0-121" name="__codelineno-0-121"></a> auth=(connector_username, connector_password),

3393 -<a id="__codelineno-0-122" name="__codelineno-0-122"></a> verify=False,

3394 -<a id="__codelineno-0-123" name="__codelineno-0-123"></a> )

3395 -<a id="__codelineno-0-124" name="__codelineno-0-124"></a> uncommitted_journal_size = uncommitted_journal_size_response.json()

3396 -<a id="__codelineno-0-125" name="__codelineno-0-125"></a>

3397 -<a id="__codelineno-0-126" name="__codelineno-0-126"></a> logger.info(

3398 -<a id="__codelineno-0-127" name="__codelineno-0-127"></a> f"Received {uncommitted_journal_size} uncommitted journal entries from Graylog",

3399 -<a id="__codelineno-0-128" name="__codelineno-0-128"></a> )

3400 -<a id="__codelineno-0-129" name="__codelineno-0-129"></a> return {

3401 -<a id="__codelineno-0-130" name="__codelineno-0-130"></a> "message": "Successfully retrieved journal size",

3402 -<a id="__codelineno-0-131" name="__codelineno-0-131"></a> "success": True,

3403 -<a id="__codelineno-0-132" name="__codelineno-0-132"></a> "uncommitted_journal_entries": uncommitted_journal_size.get(

3404 -<a id="__codelineno-0-133" name="__codelineno-0-133"></a> "uncommitted_journal_entries",

3405 -<a id="__codelineno-0-134" name="__codelineno-0-134"></a> 0,

3406 -<a id="__codelineno-0-135" name="__codelineno-0-135"></a> ),

3407 -<a id="__codelineno-0-136" name="__codelineno-0-136"></a> }

3408 -<a id="__codelineno-0-137" name="__codelineno-0-137"></a> except Exception as e:

3409 -<a id="__codelineno-0-138" name="__codelineno-0-138"></a> logger.error(f"Failed to collect journal size from Graylog: {e}")

3410 -<a id="__codelineno-0-139" name="__codelineno-0-139"></a> return {

3411 -<a id="__codelineno-0-140" name="__codelineno-0-140"></a> "message": "Failed to collect journal size from Graylog",

3412 -<a id="__codelineno-0-141" name="__codelineno-0-141"></a> "success": False,

3413 -<a id="__codelineno-0-142" name="__codelineno-0-142"></a> }

3414 -<a id="__codelineno-0-143" name="__codelineno-0-143"></a>

3415 -<a id="__codelineno-0-144" name="__codelineno-0-144"></a> def _collect_metrics_throughput_usage(

3416 -<a id="__codelineno-0-145" name="__codelineno-0-145"></a> self,

3417 -<a id="__codelineno-0-146" name="__codelineno-0-146"></a> connector_url: str,

3418 -<a id="__codelineno-0-147" name="__codelineno-0-147"></a> connector_username: str,

3419 -<a id="__codelineno-0-148" name="__codelineno-0-148"></a> connector_password: str,

3420 -<a id="__codelineno-0-149" name="__codelineno-0-149"></a> ) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3421 -<a id="__codelineno-0-150" name="__codelineno-0-150"></a> """

3422 -<a id="__codelineno-0-151" name="__codelineno-0-151"></a> Collects throughput usage from Graylog.

3423 -<a id="__codelineno-0-152" name="__codelineno-0-152"></a>

3424 -<a id="__codelineno-0-153" name="__codelineno-0-153"></a> Args:

3425 -<a id="__codelineno-0-154" name="__codelineno-0-154"></a> connector_url (str): The URL of the Graylog connector.

3426 -<a id="__codelineno-0-155" name="__codelineno-0-155"></a> connector_username (str): The username of the Graylog connector.

3427 -<a id="__codelineno-0-156" name="__codelineno-0-156"></a> connector_password (str): The password of the Graylog connector.

3428 -<a id="__codelineno-0-157" name="__codelineno-0-157"></a>

3429 -<a id="__codelineno-0-158" name="__codelineno-0-158"></a> Returns:

3430 -<a id="__codelineno-0-159" name="__codelineno-0-159"></a> dict: A dictionary containing the success status, a message, and the list of throughput usage metrics.

3431 -<a id="__codelineno-0-160" name="__codelineno-0-160"></a> """

3432 -<a id="__codelineno-0-161" name="__codelineno-0-161"></a> logger.info("Collecting throughput usage from Graylog")

3433 -<a id="__codelineno-0-162" name="__codelineno-0-162"></a>

3434 -<a id="__codelineno-0-163" name="__codelineno-0-163"></a> try:

3435 -<a id="__codelineno-0-164" name="__codelineno-0-164"></a> throughput_metrics = self._make_throughput_api_call(

3436 -<a id="__codelineno-0-165" name="__codelineno-0-165"></a> connector_url,

3437 -<a id="__codelineno-0-166" name="__codelineno-0-166"></a> self.HEADERS,

3438 -<a id="__codelineno-0-167" name="__codelineno-0-167"></a> connector_username,

3439 -<a id="__codelineno-0-168" name="__codelineno-0-168"></a> connector_password,

3440 -<a id="__codelineno-0-169" name="__codelineno-0-169"></a> )

3441 -<a id="__codelineno-0-170" name="__codelineno-0-170"></a> return self._parse_throughput_metrics(throughput_metrics)

3442 -<a id="__codelineno-0-171" name="__codelineno-0-171"></a> except Exception as e:

3443 -<a id="__codelineno-0-172" name="__codelineno-0-172"></a> logger.error(f"Failed to collect throughput usage from Graylog: {e}")

3444 -<a id="__codelineno-0-173" name="__codelineno-0-173"></a> return {

3445 -<a id="__codelineno-0-174" name="__codelineno-0-174"></a> "message": "Failed to collect throughput usage from Graylog",

3446 -<a id="__codelineno-0-175" name="__codelineno-0-175"></a> "success": False,

3447 -<a id="__codelineno-0-176" name="__codelineno-0-176"></a> }

3448 -<a id="__codelineno-0-177" name="__codelineno-0-177"></a>

3449 -<a id="__codelineno-0-178" name="__codelineno-0-178"></a> def _make_throughput_api_call(

3450 -<a id="__codelineno-0-179" name="__codelineno-0-179"></a> self,

3451 -<a id="__codelineno-0-180" name="__codelineno-0-180"></a> connector_url: str,

3452 -<a id="__codelineno-0-181" name="__codelineno-0-181"></a> headers: Dict[str, str],

3453 -<a id="__codelineno-0-182" name="__codelineno-0-182"></a> connector_username: str,

3454 -<a id="__codelineno-0-183" name="__codelineno-0-183"></a> connector_password: str,

3455 -<a id="__codelineno-0-184" name="__codelineno-0-184"></a> ) -> Dict[str, object]:

3456 -<a id="__codelineno-0-185" name="__codelineno-0-185"></a> """

3457 -<a id="__codelineno-0-186" name="__codelineno-0-186"></a> Makes Throughput API call to Graylog.

3458 -<a id="__codelineno-0-187" name="__codelineno-0-187"></a>

3459 -<a id="__codelineno-0-188" name="__codelineno-0-188"></a> Args:

3460 -<a id="__codelineno-0-189" name="__codelineno-0-189"></a> connector_url (str): The URL of the Graylog connector.

3461 -<a id="__codelineno-0-190" name="__codelineno-0-190"></a> headers (dict): The headers for the request.

3462 -<a id="__codelineno-0-191" name="__codelineno-0-191"></a> connector_username (str): The username of the Graylog connector.

3463 -<a id="__codelineno-0-192" name="__codelineno-0-192"></a> connector_password (str): The password of the Graylog connector.

3464 -<a id="__codelineno-0-193" name="__codelineno-0-193"></a>

3465 -<a id="__codelineno-0-194" name="__codelineno-0-194"></a> Returns:

3466 -<a id="__codelineno-0-195" name="__codelineno-0-195"></a> dict: The dictionary containing throughput metrics.

3467 -<a id="__codelineno-0-196" name="__codelineno-0-196"></a> """

3468 -<a id="__codelineno-0-197" name="__codelineno-0-197"></a> throughput = requests.get(

3469 -<a id="__codelineno-0-198" name="__codelineno-0-198"></a> f"{connector_url}/api/system/metrics",

3470 -<a id="__codelineno-0-199" name="__codelineno-0-199"></a> headers=headers,

3471 -<a id="__codelineno-0-200" name="__codelineno-0-200"></a> auth=(connector_username, connector_password),

3472 -<a id="__codelineno-0-201" name="__codelineno-0-201"></a> verify=False,

3473 -<a id="__codelineno-0-202" name="__codelineno-0-202"></a> )

3474 -<a id="__codelineno-0-203" name="__codelineno-0-203"></a> throughput_json = throughput.json()

3475 -<a id="__codelineno-0-204" name="__codelineno-0-204"></a> throughput_metrics = throughput_json["gauges"]

3476 -<a id="__codelineno-0-205" name="__codelineno-0-205"></a> input_output_throughput = throughput_json["counters"]

3477 -<a id="__codelineno-0-206" name="__codelineno-0-206"></a>

3478 -<a id="__codelineno-0-207" name="__codelineno-0-207"></a> throughput_metrics.update(input_output_throughput) # Merge the two dictionaries

3479 -<a id="__codelineno-0-208" name="__codelineno-0-208"></a> return throughput_metrics

3480 -<a id="__codelineno-0-209" name="__codelineno-0-209"></a>

3481 -<a id="__codelineno-0-210" name="__codelineno-0-210"></a> def _parse_throughput_metrics(

3482 -<a id="__codelineno-0-211" name="__codelineno-0-211"></a> self,

3483 -<a id="__codelineno-0-212" name="__codelineno-0-212"></a> throughput_metrics: Dict[str, object],

3484 -<a id="__codelineno-0-213" name="__codelineno-0-213"></a> ) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3485 -<a id="__codelineno-0-214" name="__codelineno-0-214"></a> """

3486 -<a id="__codelineno-0-215" name="__codelineno-0-215"></a> Parses throughput metrics.

3487 -<a id="__codelineno-0-216" name="__codelineno-0-216"></a>

3488 -<a id="__codelineno-0-217" name="__codelineno-0-217"></a> Args:

3489 -<a id="__codelineno-0-218" name="__codelineno-0-218"></a> throughput_metrics (dict): The dictionary containing throughput metrics.

3490 -<a id="__codelineno-0-219" name="__codelineno-0-219"></a>

3491 -<a id="__codelineno-0-220" name="__codelineno-0-220"></a> Returns:

3492 -<a id="__codelineno-0-221" name="__codelineno-0-221"></a> dict: A dictionary containing the success status, a message, and the list of parsed throughput metrics.

3493 -<a id="__codelineno-0-222" name="__codelineno-0-222"></a> """

3494 -<a id="__codelineno-0-223" name="__codelineno-0-223"></a> throughput_metrics_list = []

3495 -<a id="__codelineno-0-224" name="__codelineno-0-224"></a> results = {}

3496 -<a id="__codelineno-0-225" name="__codelineno-0-225"></a>

3497 -<a id="__codelineno-0-226" name="__codelineno-0-226"></a> for metric, data in throughput_metrics.items():

3498 -<a id="__codelineno-0-227" name="__codelineno-0-227"></a> if metric in self.METRIC_NAMES:

3499 -<a id="__codelineno-0-228" name="__codelineno-0-228"></a> value = data["value"] if "value" in data else data["count"]

3500 -<a id="__codelineno-0-229" name="__codelineno-0-229"></a> throughput_metrics_list.append({"metric": metric, "value": value})

3501 -<a id="__codelineno-0-230" name="__codelineno-0-230"></a>

3502 -<a id="__codelineno-0-231" name="__codelineno-0-231"></a> variable_name = self.METRIC_NAMES.get(metric)

3503 -<a id="__codelineno-0-232" name="__codelineno-0-232"></a> if variable_name is not None:

3504 -<a id="__codelineno-0-233" name="__codelineno-0-233"></a> results[variable_name] = value

3505 -<a id="__codelineno-0-234" name="__codelineno-0-234"></a>

3506 -<a id="__codelineno-0-235" name="__codelineno-0-235"></a> logger.info(

3507 -<a id="__codelineno-0-236" name="__codelineno-0-236"></a> f"Received throughput usage from Graylog: {throughput_metrics_list}",

3508 -<a id="__codelineno-0-237" name="__codelineno-0-237"></a> )

3509 -<a id="__codelineno-0-238" name="__codelineno-0-238"></a> return {

3510 -<a id="__codelineno-0-239" name="__codelineno-0-239"></a> "message": "Successfully retrieved throughput usage",

3511 -<a id="__codelineno-0-240" name="__codelineno-0-240"></a> "success": True,

3512 -<a id="__codelineno-0-241" name="__codelineno-0-241"></a> "throughput_metrics": throughput_metrics_list,

3513 -<a id="__codelineno-0-242" name="__codelineno-0-242"></a> }

3369 +<a id="__codelineno-0-66" name="__codelineno-0-66"></a> def collect_throughput_metrics(

3370 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> self,

3371 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a> ) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3372 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> """

3373 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> Collects the following Graylog Metrics:

3374 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> - Input Usage

3375 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> - Output Usage

3376 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> - Processor Usage

3377 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> - Input 1 Seconds Rate

3378 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> - Output 1 Seconds Rate

3379 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> - Total Input

3380 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> - Total Output

3381 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a>

3382 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> Returns:

3383 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> dict: A dictionary containing the success status, a message, and the list of throughput metrics.

3384 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> """

3385 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> if (

3386 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a> self.connector_url is None

3387 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> or self.connector_username is None

3388 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> or self.connector_password is None

3389 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> ):

3390 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> return {"message": "Failed to collect Graylog details", "success": False}

3391 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a>

3392 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> throughput_usage = self._collect_metrics_throughput_usage(

3393 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> self.connector_url,

3394 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> self.connector_username,

3395 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> self.connector_password,

3396 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> )

3397 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a>

3398 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> if throughput_usage["success"] is False:

3399 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> return throughput_usage

3400 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> return throughput_usage

3401 +<a id="__codelineno-0-98" name="__codelineno-0-98"></a>

3402 +<a id="__codelineno-0-99" name="__codelineno-0-99"></a> def _collect_metrics_uncommitted_journal_size(

3403 +<a id="__codelineno-0-100" name="__codelineno-0-100"></a> self,

3404 +<a id="__codelineno-0-101" name="__codelineno-0-101"></a> connector_url: str,

3405 +<a id="__codelineno-0-102" name="__codelineno-0-102"></a> connector_username: str,

3406 +<a id="__codelineno-0-103" name="__codelineno-0-103"></a> connector_password: str,

3407 +<a id="__codelineno-0-104" name="__codelineno-0-104"></a> ) -> Dict[str, Union[str, bool, int]]:

3408 +<a id="__codelineno-0-105" name="__codelineno-0-105"></a> """

3409 +<a id="__codelineno-0-106" name="__codelineno-0-106"></a> Collects the journal size of uncommitted messages from Graylog.

3410 +<a id="__codelineno-0-107" name="__codelineno-0-107"></a>

3411 +<a id="__codelineno-0-108" name="__codelineno-0-108"></a> Args:

3412 +<a id="__codelineno-0-109" name="__codelineno-0-109"></a> connector_url (str): The URL of the Graylog connector.

3413 +<a id="__codelineno-0-110" name="__codelineno-0-110"></a> connector_username (str): The username of the Graylog connector.

3414 +<a id="__codelineno-0-111" name="__codelineno-0-111"></a> connector_password (str): The password of the Graylog connector.

3415 +<a id="__codelineno-0-112" name="__codelineno-0-112"></a>

3416 +<a id="__codelineno-0-113" name="__codelineno-0-113"></a> Returns:

3417 +<a id="__codelineno-0-114" name="__codelineno-0-114"></a> dict: A dictionary containing the success status, a message, and the size of uncommitted journal entries.

3418 +<a id="__codelineno-0-115" name="__codelineno-0-115"></a> """

3419 +<a id="__codelineno-0-116" name="__codelineno-0-116"></a> try:

3420 +<a id="__codelineno-0-117" name="__codelineno-0-117"></a> logger.info("Collecting journal size from Graylog")

3421 +<a id="__codelineno-0-118" name="__codelineno-0-118"></a> headers = {"X-Requested-By": "CoPilot"}

3422 +<a id="__codelineno-0-119" name="__codelineno-0-119"></a> # Get the Graylog Journal Size

3423 +<a id="__codelineno-0-120" name="__codelineno-0-120"></a> uncommitted_journal_size_response = requests.get(

3424 +<a id="__codelineno-0-121" name="__codelineno-0-121"></a> f"{connector_url}/api/system/journal",

3425 +<a id="__codelineno-0-122" name="__codelineno-0-122"></a> headers=headers,

3426 +<a id="__codelineno-0-123" name="__codelineno-0-123"></a> auth=(connector_username, connector_password),

3427 +<a id="__codelineno-0-124" name="__codelineno-0-124"></a> verify=False,

3428 +<a id="__codelineno-0-125" name="__codelineno-0-125"></a> )

3429 +<a id="__codelineno-0-126" name="__codelineno-0-126"></a> uncommitted_journal_size = uncommitted_journal_size_response.json()

3430 +<a id="__codelineno-0-127" name="__codelineno-0-127"></a>

3431 +<a id="__codelineno-0-128" name="__codelineno-0-128"></a> logger.info(

3432 +<a id="__codelineno-0-129" name="__codelineno-0-129"></a> f"Received {uncommitted_journal_size} uncommitted journal entries from Graylog",

3433 +<a id="__codelineno-0-130" name="__codelineno-0-130"></a> )

3434 +<a id="__codelineno-0-131" name="__codelineno-0-131"></a> return {

3435 +<a id="__codelineno-0-132" name="__codelineno-0-132"></a> "message": "Successfully retrieved journal size",

3436 +<a id="__codelineno-0-133" name="__codelineno-0-133"></a> "success": True,

3437 +<a id="__codelineno-0-134" name="__codelineno-0-134"></a> "uncommitted_journal_entries": uncommitted_journal_size.get(

3438 +<a id="__codelineno-0-135" name="__codelineno-0-135"></a> "uncommitted_journal_entries",

3439 +<a id="__codelineno-0-136" name="__codelineno-0-136"></a> 0,

3440 +<a id="__codelineno-0-137" name="__codelineno-0-137"></a> ),

3441 +<a id="__codelineno-0-138" name="__codelineno-0-138"></a> }

3442 +<a id="__codelineno-0-139" name="__codelineno-0-139"></a> except Exception as e:

3443 +<a id="__codelineno-0-140" name="__codelineno-0-140"></a> logger.error(f"Failed to collect journal size from Graylog: {e}")

3444 +<a id="__codelineno-0-141" name="__codelineno-0-141"></a> return {

3445 +<a id="__codelineno-0-142" name="__codelineno-0-142"></a> "message": "Failed to collect journal size from Graylog",

3446 +<a id="__codelineno-0-143" name="__codelineno-0-143"></a> "success": False,

3447 +<a id="__codelineno-0-144" name="__codelineno-0-144"></a> }

3448 +<a id="__codelineno-0-145" name="__codelineno-0-145"></a>

3449 +<a id="__codelineno-0-146" name="__codelineno-0-146"></a> def _collect_metrics_throughput_usage(

3450 +<a id="__codelineno-0-147" name="__codelineno-0-147"></a> self,

3451 +<a id="__codelineno-0-148" name="__codelineno-0-148"></a> connector_url: str,

3452 +<a id="__codelineno-0-149" name="__codelineno-0-149"></a> connector_username: str,

3453 +<a id="__codelineno-0-150" name="__codelineno-0-150"></a> connector_password: str,

3454 +<a id="__codelineno-0-151" name="__codelineno-0-151"></a> ) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3455 +<a id="__codelineno-0-152" name="__codelineno-0-152"></a> """

3456 +<a id="__codelineno-0-153" name="__codelineno-0-153"></a> Collects throughput usage from Graylog.

3457 +<a id="__codelineno-0-154" name="__codelineno-0-154"></a>

3458 +<a id="__codelineno-0-155" name="__codelineno-0-155"></a> Args:

3459 +<a id="__codelineno-0-156" name="__codelineno-0-156"></a> connector_url (str): The URL of the Graylog connector.

3460 +<a id="__codelineno-0-157" name="__codelineno-0-157"></a> connector_username (str): The username of the Graylog connector.

3461 +<a id="__codelineno-0-158" name="__codelineno-0-158"></a> connector_password (str): The password of the Graylog connector.

3462 +<a id="__codelineno-0-159" name="__codelineno-0-159"></a>

3463 +<a id="__codelineno-0-160" name="__codelineno-0-160"></a> Returns:

3464 +<a id="__codelineno-0-161" name="__codelineno-0-161"></a> dict: A dictionary containing the success status, a message, and the list of throughput usage metrics.

3465 +<a id="__codelineno-0-162" name="__codelineno-0-162"></a> """

3466 +<a id="__codelineno-0-163" name="__codelineno-0-163"></a> logger.info("Collecting throughput usage from Graylog")

3467 +<a id="__codelineno-0-164" name="__codelineno-0-164"></a>

3468 +<a id="__codelineno-0-165" name="__codelineno-0-165"></a> try:

3469 +<a id="__codelineno-0-166" name="__codelineno-0-166"></a> throughput_metrics = self._make_throughput_api_call(

3470 +<a id="__codelineno-0-167" name="__codelineno-0-167"></a> connector_url,

3471 +<a id="__codelineno-0-168" name="__codelineno-0-168"></a> self.HEADERS,

3472 +<a id="__codelineno-0-169" name="__codelineno-0-169"></a> connector_username,

3473 +<a id="__codelineno-0-170" name="__codelineno-0-170"></a> connector_password,

3474 +<a id="__codelineno-0-171" name="__codelineno-0-171"></a> )

3475 +<a id="__codelineno-0-172" name="__codelineno-0-172"></a> return self._parse_throughput_metrics(throughput_metrics)

3476 +<a id="__codelineno-0-173" name="__codelineno-0-173"></a> except Exception as e:

3477 +<a id="__codelineno-0-174" name="__codelineno-0-174"></a> logger.error(f"Failed to collect throughput usage from Graylog: {e}")

3478 +<a id="__codelineno-0-175" name="__codelineno-0-175"></a> return {

3479 +<a id="__codelineno-0-176" name="__codelineno-0-176"></a> "message": "Failed to collect throughput usage from Graylog",

3480 +<a id="__codelineno-0-177" name="__codelineno-0-177"></a> "success": False,

3481 +<a id="__codelineno-0-178" name="__codelineno-0-178"></a> }

3482 +<a id="__codelineno-0-179" name="__codelineno-0-179"></a>

3483 +<a id="__codelineno-0-180" name="__codelineno-0-180"></a> def _make_throughput_api_call(

3484 +<a id="__codelineno-0-181" name="__codelineno-0-181"></a> self,

3485 +<a id="__codelineno-0-182" name="__codelineno-0-182"></a> connector_url: str,

3486 +<a id="__codelineno-0-183" name="__codelineno-0-183"></a> headers: Dict[str, str],

3487 +<a id="__codelineno-0-184" name="__codelineno-0-184"></a> connector_username: str,

3488 +<a id="__codelineno-0-185" name="__codelineno-0-185"></a> connector_password: str,

3489 +<a id="__codelineno-0-186" name="__codelineno-0-186"></a> ) -> Dict[str, object]:

3490 +<a id="__codelineno-0-187" name="__codelineno-0-187"></a> """

3491 +<a id="__codelineno-0-188" name="__codelineno-0-188"></a> Makes Throughput API call to Graylog.

3492 +<a id="__codelineno-0-189" name="__codelineno-0-189"></a>

3493 +<a id="__codelineno-0-190" name="__codelineno-0-190"></a> Args:

3494 +<a id="__codelineno-0-191" name="__codelineno-0-191"></a> connector_url (str): The URL of the Graylog connector.

3495 +<a id="__codelineno-0-192" name="__codelineno-0-192"></a> headers (dict): The headers for the request.

3496 +<a id="__codelineno-0-193" name="__codelineno-0-193"></a> connector_username (str): The username of the Graylog connector.

3497 +<a id="__codelineno-0-194" name="__codelineno-0-194"></a> connector_password (str): The password of the Graylog connector.

3498 +<a id="__codelineno-0-195" name="__codelineno-0-195"></a>

3499 +<a id="__codelineno-0-196" name="__codelineno-0-196"></a> Returns:

3500 +<a id="__codelineno-0-197" name="__codelineno-0-197"></a> dict: The dictionary containing throughput metrics.

3501 +<a id="__codelineno-0-198" name="__codelineno-0-198"></a> """

3502 +<a id="__codelineno-0-199" name="__codelineno-0-199"></a> throughput = requests.get(

3503 +<a id="__codelineno-0-200" name="__codelineno-0-200"></a> f"{connector_url}/api/system/metrics",

3504 +<a id="__codelineno-0-201" name="__codelineno-0-201"></a> headers=headers,

3505 +<a id="__codelineno-0-202" name="__codelineno-0-202"></a> auth=(connector_username, connector_password),

3506 +<a id="__codelineno-0-203" name="__codelineno-0-203"></a> verify=False,

3507 +<a id="__codelineno-0-204" name="__codelineno-0-204"></a> )

3508 +<a id="__codelineno-0-205" name="__codelineno-0-205"></a> throughput_json = throughput.json()

3509 +<a id="__codelineno-0-206" name="__codelineno-0-206"></a> throughput_metrics = throughput_json["gauges"]

3510 +<a id="__codelineno-0-207" name="__codelineno-0-207"></a> input_output_throughput = throughput_json["counters"]

3511 +<a id="__codelineno-0-208" name="__codelineno-0-208"></a>

3512 +<a id="__codelineno-0-209" name="__codelineno-0-209"></a> throughput_metrics.update(input_output_throughput) # Merge the two dictionaries

3513 +<a id="__codelineno-0-210" name="__codelineno-0-210"></a> return throughput_metrics

3514 +<a id="__codelineno-0-211" name="__codelineno-0-211"></a>

3515 +<a id="__codelineno-0-212" name="__codelineno-0-212"></a> def _parse_throughput_metrics(

3516 +<a id="__codelineno-0-213" name="__codelineno-0-213"></a> self,

3517 +<a id="__codelineno-0-214" name="__codelineno-0-214"></a> throughput_metrics: Dict[str, object],

3518 +<a id="__codelineno-0-215" name="__codelineno-0-215"></a> ) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3519 +<a id="__codelineno-0-216" name="__codelineno-0-216"></a> """

3520 +<a id="__codelineno-0-217" name="__codelineno-0-217"></a> Parses throughput metrics.

3521 +<a id="__codelineno-0-218" name="__codelineno-0-218"></a>

3522 +<a id="__codelineno-0-219" name="__codelineno-0-219"></a> Args:

3523 +<a id="__codelineno-0-220" name="__codelineno-0-220"></a> throughput_metrics (dict): The dictionary containing throughput metrics.

3524 +<a id="__codelineno-0-221" name="__codelineno-0-221"></a>

3525 +<a id="__codelineno-0-222" name="__codelineno-0-222"></a> Returns:

3526 +<a id="__codelineno-0-223" name="__codelineno-0-223"></a> dict: A dictionary containing the success status, a message, and the list of parsed throughput metrics.

3527 +<a id="__codelineno-0-224" name="__codelineno-0-224"></a> """

3528 +<a id="__codelineno-0-225" name="__codelineno-0-225"></a> throughput_metrics_list = []

3529 +<a id="__codelineno-0-226" name="__codelineno-0-226"></a> results = {}

3530 +<a id="__codelineno-0-227" name="__codelineno-0-227"></a>

3531 +<a id="__codelineno-0-228" name="__codelineno-0-228"></a> for metric, data in throughput_metrics.items():

3532 +<a id="__codelineno-0-229" name="__codelineno-0-229"></a> if metric in self.METRIC_NAMES:

3533 +<a id="__codelineno-0-230" name="__codelineno-0-230"></a> value = data["value"] if "value" in data else data["count"]

3534 +<a id="__codelineno-0-231" name="__codelineno-0-231"></a> throughput_metrics_list.append({"metric": metric, "value": value})

3535 +<a id="__codelineno-0-232" name="__codelineno-0-232"></a>

3536 +<a id="__codelineno-0-233" name="__codelineno-0-233"></a> variable_name = self.METRIC_NAMES.get(metric)

3537 +<a id="__codelineno-0-234" name="__codelineno-0-234"></a> if variable_name is not None:

3538 +<a id="__codelineno-0-235" name="__codelineno-0-235"></a> results[variable_name] = value

3539 +<a id="__codelineno-0-236" name="__codelineno-0-236"></a>

3540 +<a id="__codelineno-0-237" name="__codelineno-0-237"></a> logger.info(

3541 +<a id="__codelineno-0-238" name="__codelineno-0-238"></a> f"Received throughput usage from Graylog: {throughput_metrics_list}",

3542 +<a id="__codelineno-0-239" name="__codelineno-0-239"></a> )

3543 +<a id="__codelineno-0-240" name="__codelineno-0-240"></a> return {

3544 +<a id="__codelineno-0-241" name="__codelineno-0-241"></a> "message": "Successfully retrieved throughput usage",

3545 +<a id="__codelineno-0-242" name="__codelineno-0-242"></a> "success": True,

3546 +<a id="__codelineno-0-243" name="__codelineno-0-243"></a> "throughput_metrics": throughput_metrics_list,

3547 +<a id="__codelineno-0-244" name="__codelineno-0-244"></a> }

3548 </code></pre></div></td></tr></table></div>

3549 </details>

3550

@@ -3639,36 +3673,40 @@ This class inherits from SQLAlchemy's Model class.

3673 <a href="#__codelineno-0-92">92</a>

3674 <a href="#__codelineno-0-93">93</a>

3675 <a href="#__codelineno-0-94">94</a>

3642 -<a href="#__codelineno-0-95">95</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-66" name="__codelineno-0-66"></a>def collect_throughput_metrics(self) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3643 -<a id="__codelineno-0-67" name="__codelineno-0-67"></a> """

3644 -<a id="__codelineno-0-68" name="__codelineno-0-68"></a> Collects the following Graylog Metrics:

3645 -<a id="__codelineno-0-69" name="__codelineno-0-69"></a> - Input Usage

3646 -<a id="__codelineno-0-70" name="__codelineno-0-70"></a> - Output Usage

3647 -<a id="__codelineno-0-71" name="__codelineno-0-71"></a> - Processor Usage

3648 -<a id="__codelineno-0-72" name="__codelineno-0-72"></a> - Input 1 Seconds Rate

3649 -<a id="__codelineno-0-73" name="__codelineno-0-73"></a> - Output 1 Seconds Rate

3650 -<a id="__codelineno-0-74" name="__codelineno-0-74"></a> - Total Input

3651 -<a id="__codelineno-0-75" name="__codelineno-0-75"></a> - Total Output

3652 -<a id="__codelineno-0-76" name="__codelineno-0-76"></a>

3653 -<a id="__codelineno-0-77" name="__codelineno-0-77"></a> Returns:

3654 -<a id="__codelineno-0-78" name="__codelineno-0-78"></a> dict: A dictionary containing the success status, a message, and the list of throughput metrics.

3655 -<a id="__codelineno-0-79" name="__codelineno-0-79"></a> """

3656 -<a id="__codelineno-0-80" name="__codelineno-0-80"></a> if (

3657 -<a id="__codelineno-0-81" name="__codelineno-0-81"></a> self.connector_url is None

3658 -<a id="__codelineno-0-82" name="__codelineno-0-82"></a> or self.connector_username is None

3659 -<a id="__codelineno-0-83" name="__codelineno-0-83"></a> or self.connector_password is None

3660 -<a id="__codelineno-0-84" name="__codelineno-0-84"></a> ):

3661 -<a id="__codelineno-0-85" name="__codelineno-0-85"></a> return {"message": "Failed to collect Graylog details", "success": False}

3662 -<a id="__codelineno-0-86" name="__codelineno-0-86"></a>

3663 -<a id="__codelineno-0-87" name="__codelineno-0-87"></a> throughput_usage = self._collect_metrics_throughput_usage(

3664 -<a id="__codelineno-0-88" name="__codelineno-0-88"></a> self.connector_url,

3665 -<a id="__codelineno-0-89" name="__codelineno-0-89"></a> self.connector_username,

3666 -<a id="__codelineno-0-90" name="__codelineno-0-90"></a> self.connector_password,

3667 -<a id="__codelineno-0-91" name="__codelineno-0-91"></a> )

3668 -<a id="__codelineno-0-92" name="__codelineno-0-92"></a>

3669 -<a id="__codelineno-0-93" name="__codelineno-0-93"></a> if throughput_usage["success"] is False:

3670 -<a id="__codelineno-0-94" name="__codelineno-0-94"></a> return throughput_usage

3671 -<a id="__codelineno-0-95" name="__codelineno-0-95"></a> return throughput_usage

3676 +<a href="#__codelineno-0-95">95</a>

3677 +<a href="#__codelineno-0-96">96</a>

3678 +<a href="#__codelineno-0-97">97</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-66" name="__codelineno-0-66"></a>def collect_throughput_metrics(

3679 +<a id="__codelineno-0-67" name="__codelineno-0-67"></a> self,

3680 +<a id="__codelineno-0-68" name="__codelineno-0-68"></a>) -> Dict[str, Union[str, bool, List[Dict[str, Any]]]]:

3681 +<a id="__codelineno-0-69" name="__codelineno-0-69"></a> """

3682 +<a id="__codelineno-0-70" name="__codelineno-0-70"></a> Collects the following Graylog Metrics:

3683 +<a id="__codelineno-0-71" name="__codelineno-0-71"></a> - Input Usage

3684 +<a id="__codelineno-0-72" name="__codelineno-0-72"></a> - Output Usage

3685 +<a id="__codelineno-0-73" name="__codelineno-0-73"></a> - Processor Usage

3686 +<a id="__codelineno-0-74" name="__codelineno-0-74"></a> - Input 1 Seconds Rate

3687 +<a id="__codelineno-0-75" name="__codelineno-0-75"></a> - Output 1 Seconds Rate

3688 +<a id="__codelineno-0-76" name="__codelineno-0-76"></a> - Total Input

3689 +<a id="__codelineno-0-77" name="__codelineno-0-77"></a> - Total Output

3690 +<a id="__codelineno-0-78" name="__codelineno-0-78"></a>

3691 +<a id="__codelineno-0-79" name="__codelineno-0-79"></a> Returns:

3692 +<a id="__codelineno-0-80" name="__codelineno-0-80"></a> dict: A dictionary containing the success status, a message, and the list of throughput metrics.

3693 +<a id="__codelineno-0-81" name="__codelineno-0-81"></a> """

3694 +<a id="__codelineno-0-82" name="__codelineno-0-82"></a> if (

3695 +<a id="__codelineno-0-83" name="__codelineno-0-83"></a> self.connector_url is None

3696 +<a id="__codelineno-0-84" name="__codelineno-0-84"></a> or self.connector_username is None

3697 +<a id="__codelineno-0-85" name="__codelineno-0-85"></a> or self.connector_password is None

3698 +<a id="__codelineno-0-86" name="__codelineno-0-86"></a> ):

3699 +<a id="__codelineno-0-87" name="__codelineno-0-87"></a> return {"message": "Failed to collect Graylog details", "success": False}

3700 +<a id="__codelineno-0-88" name="__codelineno-0-88"></a>

3701 +<a id="__codelineno-0-89" name="__codelineno-0-89"></a> throughput_usage = self._collect_metrics_throughput_usage(

3702 +<a id="__codelineno-0-90" name="__codelineno-0-90"></a> self.connector_url,

3703 +<a id="__codelineno-0-91" name="__codelineno-0-91"></a> self.connector_username,

3704 +<a id="__codelineno-0-92" name="__codelineno-0-92"></a> self.connector_password,

3705 +<a id="__codelineno-0-93" name="__codelineno-0-93"></a> )

3706 +<a id="__codelineno-0-94" name="__codelineno-0-94"></a>

3707 +<a id="__codelineno-0-95" name="__codelineno-0-95"></a> if throughput_usage["success"] is False:

3708 +<a id="__codelineno-0-96" name="__codelineno-0-96"></a> return throughput_usage

3709 +<a id="__codelineno-0-97" name="__codelineno-0-97"></a> return throughput_usage

3710 </code></pre></div></td></tr></table></div>

3711 </details>

3712 </div>

@@ -3817,7 +3855,8 @@ This class inherits from SQLAlchemy's Model class.

3855

3856 <details class="quote">

3857 <summary>Source code in <code>app\services\Graylog\universal.py</code></summary>

3820 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-6"> 6</a>

3858 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-5"> 5</a>

3859 +<a href="#__codelineno-0-6"> 6</a>

3860 <a href="#__codelineno-0-7"> 7</a>

3861 <a href="#__codelineno-0-8"> 8</a>

3862 <a href="#__codelineno-0-9"> 9</a>

@@ -3844,36 +3883,35 @@ This class inherits from SQLAlchemy's Model class.

3883 <a href="#__codelineno-0-30">30</a>

3884 <a href="#__codelineno-0-31">31</a>

3885 <a href="#__codelineno-0-32">32</a>

3847 -<a href="#__codelineno-0-33">33</a>

3848 -<a href="#__codelineno-0-34">34</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-6" name="__codelineno-0-6"></a>class UniversalService:

3849 -<a id="__codelineno-0-7" name="__codelineno-0-7"></a> """

3850 -<a id="__codelineno-0-8" name="__codelineno-0-8"></a> A service class that encapsulates the logic for polling messages from Graylog.

3851 -<a id="__codelineno-0-9" name="__codelineno-0-9"></a> """

3852 -<a id="__codelineno-0-10" name="__codelineno-0-10"></a>

3853 -<a id="__codelineno-0-11" name="__codelineno-0-11"></a> def __init__(self) -> None:

3854 -<a id="__codelineno-0-12" name="__codelineno-0-12"></a> self.collect_graylog_details("Graylog")

3855 -<a id="__codelineno-0-13" name="__codelineno-0-13"></a>

3856 -<a id="__codelineno-0-14" name="__codelineno-0-14"></a> def collect_graylog_details(self, connector_name: str) -> tuple:

3857 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a> """

3858 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> Collects the details of the Graylog connector.

3859 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a>

3860 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> Args:

3861 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> connector_name (str): The name of the Graylog connector.

3862 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a>

3863 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a> Returns:

3864 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> tuple: A tuple containing the connection URL, username, and password.

3865 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a> """

3866 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> connector_instance = connector_factory.create(connector_name, connector_name)

3867 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> connection_successful = connector_instance.verify_connection()

3868 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> if connection_successful:

3869 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

3870 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> return (

3871 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> connection_details.get("connector_url"),

3872 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> connection_details.get("connector_username"),

3873 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> connection_details.get("connector_password"),

3874 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> )

3875 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> else:

3876 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> return None, None, None

3886 +<a href="#__codelineno-0-33">33</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-5" name="__codelineno-0-5"></a>class UniversalService:

3887 +<a id="__codelineno-0-6" name="__codelineno-0-6"></a> """

3888 +<a id="__codelineno-0-7" name="__codelineno-0-7"></a> A service class that encapsulates the logic for polling messages from Graylog.

3889 +<a id="__codelineno-0-8" name="__codelineno-0-8"></a> """

3890 +<a id="__codelineno-0-9" name="__codelineno-0-9"></a>

3891 +<a id="__codelineno-0-10" name="__codelineno-0-10"></a> def __init__(self) -> None:

3892 +<a id="__codelineno-0-11" name="__codelineno-0-11"></a> self.collect_graylog_details("Graylog")

3893 +<a id="__codelineno-0-12" name="__codelineno-0-12"></a>

3894 +<a id="__codelineno-0-13" name="__codelineno-0-13"></a> def collect_graylog_details(self, connector_name: str) -> tuple:

3895 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> """

3896 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> Collects the details of the Graylog connector.

3897 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a>

3898 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a> Args:

3899 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> connector_name (str): The name of the Graylog connector.

3900 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a>

3901 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> Returns:

3902 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> tuple: A tuple containing the connection URL, username, and password.

3903 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> """

3904 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> connector_instance = connector_factory.create(connector_name, connector_name)

3905 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a> connection_successful = connector_instance.verify_connection()

3906 +<a id="__codelineno-0-25" name="__codelineno-0-25"></a> if connection_successful:

3907 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

3908 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> return (

3909 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> connection_details.get("connector_url"),

3910 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> connection_details.get("connector_username"),

3911 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> connection_details.get("connector_password"),

3912 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> )

3913 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> else:

3914 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> return None, None, None

3915 </code></pre></div></td></tr></table></div>

3916 </details>

3917

@@ -3955,7 +3993,8 @@ This class inherits from SQLAlchemy's Model class.

3993

3994 <details class="quote">

3995 <summary>Source code in <code>app\services\Graylog\universal.py</code></summary>

3958 - <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-14">14</a>

3996 + <div class="highlight"><table class="highlighttable"><tr><td class="linenos"><div class="linenodiv"><pre><a href="#__codelineno-0-13">13</a>

3997 +<a href="#__codelineno-0-14">14</a>

3998 <a href="#__codelineno-0-15">15</a>

3999 <a href="#__codelineno-0-16">16</a>

4000 <a href="#__codelineno-0-17">17</a>

@@ -3974,28 +4013,27 @@ This class inherits from SQLAlchemy's Model class.

4013 <a href="#__codelineno-0-30">30</a>

4014 <a href="#__codelineno-0-31">31</a>

4015 <a href="#__codelineno-0-32">32</a>

3977 -<a href="#__codelineno-0-33">33</a>

3978 -<a href="#__codelineno-0-34">34</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-14" name="__codelineno-0-14"></a>def collect_graylog_details(self, connector_name: str) -> tuple:

3979 -<a id="__codelineno-0-15" name="__codelineno-0-15"></a> """

3980 -<a id="__codelineno-0-16" name="__codelineno-0-16"></a> Collects the details of the Graylog connector.

3981 -<a id="__codelineno-0-17" name="__codelineno-0-17"></a>

3982 -<a id="__codelineno-0-18" name="__codelineno-0-18"></a> Args:

3983 -<a id="__codelineno-0-19" name="__codelineno-0-19"></a> connector_name (str): The name of the Graylog connector.

3984 -<a id="__codelineno-0-20" name="__codelineno-0-20"></a>

3985 -<a id="__codelineno-0-21" name="__codelineno-0-21"></a> Returns:

3986 -<a id="__codelineno-0-22" name="__codelineno-0-22"></a> tuple: A tuple containing the connection URL, username, and password.

3987 -<a id="__codelineno-0-23" name="__codelineno-0-23"></a> """

3988 -<a id="__codelineno-0-24" name="__codelineno-0-24"></a> connector_instance = connector_factory.create(connector_name, connector_name)

3989 -<a id="__codelineno-0-25" name="__codelineno-0-25"></a> connection_successful = connector_instance.verify_connection()

3990 -<a id="__codelineno-0-26" name="__codelineno-0-26"></a> if connection_successful:

3991 -<a id="__codelineno-0-27" name="__codelineno-0-27"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

3992 -<a id="__codelineno-0-28" name="__codelineno-0-28"></a> return (

3993 -<a id="__codelineno-0-29" name="__codelineno-0-29"></a> connection_details.get("connector_url"),

3994 -<a id="__codelineno-0-30" name="__codelineno-0-30"></a> connection_details.get("connector_username"),

3995 -<a id="__codelineno-0-31" name="__codelineno-0-31"></a> connection_details.get("connector_password"),

3996 -<a id="__codelineno-0-32" name="__codelineno-0-32"></a> )

3997 -<a id="__codelineno-0-33" name="__codelineno-0-33"></a> else:

3998 -<a id="__codelineno-0-34" name="__codelineno-0-34"></a> return None, None, None

4016 +<a href="#__codelineno-0-33">33</a></pre></div></td><td class="code"><div><pre><code><a id="__codelineno-0-13" name="__codelineno-0-13"></a>def collect_graylog_details(self, connector_name: str) -> tuple:

4017 +<a id="__codelineno-0-14" name="__codelineno-0-14"></a> """

4018 +<a id="__codelineno-0-15" name="__codelineno-0-15"></a> Collects the details of the Graylog connector.

4019 +<a id="__codelineno-0-16" name="__codelineno-0-16"></a>

4020 +<a id="__codelineno-0-17" name="__codelineno-0-17"></a> Args:

4021 +<a id="__codelineno-0-18" name="__codelineno-0-18"></a> connector_name (str): The name of the Graylog connector.

4022 +<a id="__codelineno-0-19" name="__codelineno-0-19"></a>

4023 +<a id="__codelineno-0-20" name="__codelineno-0-20"></a> Returns:

4024 +<a id="__codelineno-0-21" name="__codelineno-0-21"></a> tuple: A tuple containing the connection URL, username, and password.

4025 +<a id="__codelineno-0-22" name="__codelineno-0-22"></a> """

4026 +<a id="__codelineno-0-23" name="__codelineno-0-23"></a> connector_instance = connector_factory.create(connector_name, connector_name)

4027 +<a id="__codelineno-0-24" name="__codelineno-0-24"></a> connection_successful = connector_instance.verify_connection()

4028 +<a id="__codelineno-0-25" name="__codelineno-0-25"></a> if connection_successful:

4029 +<a id="__codelineno-0-26" name="__codelineno-0-26"></a> connection_details = Connector.get_connector_info_from_db(connector_name)

4030 +<a id="__codelineno-0-27" name="__codelineno-0-27"></a> return (

4031 +<a id="__codelineno-0-28" name="__codelineno-0-28"></a> connection_details.get("connector_url"),

4032 +<a id="__codelineno-0-29" name="__codelineno-0-29"></a> connection_details.get("connector_username"),

4033 +<a id="__codelineno-0-30" name="__codelineno-0-30"></a> connection_details.get("connector_password"),

4034 +<a id="__codelineno-0-31" name="__codelineno-0-31"></a> )

4035 +<a id="__codelineno-0-32" name="__codelineno-0-32"></a> else:

4036 +<a id="__codelineno-0-33" name="__codelineno-0-33"></a> return None, None, None

4037 </code></pre></div></td></tr></table></div>

4038 </details>

4039 </div>

backend/site/sitemap.xml.gz

Binary files a/backend/site/sitemap.xml.gz and b/backend/site/sitemap.xml.gz differ