Island: Add AgentID type for agent_id to AgentLogs endpoint

This commit is contained in:
Ilija Lazoroski 2022-09-28 16:21:39 +02:00
parent fa13ca8df8
commit b705e33af3
1 changed files with 7 additions and 5 deletions

View File

@ -2,25 +2,27 @@ from http import HTTPStatus
from flask import request from flask import request
from common.types import AgentID
from monkey_island.cc.repository import IAgentLogRepository from monkey_island.cc.repository import IAgentLogRepository
from monkey_island.cc.resources.AbstractResource import AbstractResource from monkey_island.cc.resources.AbstractResource import AbstractResource
from monkey_island.cc.resources.request_authentication import jwt_required from monkey_island.cc.resources.request_authentication import jwt_required
class AgentLogs(AbstractResource): class AgentLogs(AbstractResource):
urls = ["/api/agent-logs/<string:agent_id>"] urls = ["/api/agent-logs/<uuid:agent_id>"]
def __init__(self, agent_log_repository: IAgentLogRepository): def __init__(self, agent_log_repository: IAgentLogRepository):
self._agent_log_repository = agent_log_repository self._agent_log_repository = agent_log_repository
@jwt_required @jwt_required
def get(self, agent_id: str): def get(self, agent_id: AgentID):
log_contents = self._agent_log_repository.get_agent_log(agent_id) log_contents = self._agent_log_repository.get_agent_log(agent_id)
return log_contents, HTTPStatus.OK return log_contents, HTTPStatus.OK
def put(self, agent_id: str): def put(self, agent_id: AgentID):
log_contents = request.json["log_contents"] log_contents = request.json
self._agent_log_repository.upsert_agent_log(agent_id, log_contents)
self._agent_log_repository.upsert_agent_log(agent_id, agent_data)
return {}, HTTPStatus.NO_CONTENT return {}, HTTPStatus.NO_CONTENT