browser to server logging

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/api/api.py

@@ -35,7 +35,8 @@ class Api:
 
         # server api
         self.add_api_route("/sdapi/v1/motd", server.get_motd, methods=["GET"], response_model=str)
-        self.add_api_route("/sdapi/v1/log", server.get_log_buffer, methods=["GET"], response_model=List[str])
+        self.add_api_route("/sdapi/v1/log", server.get_log, methods=["GET"], response_model=List[str])
+        self.add_api_route("/sdapi/v1/log", server.post_log, methods=["POST"])
         self.add_api_route("/sdapi/v1/start", self.get_session_start, methods=["GET"])
         self.add_api_route("/sdapi/v1/version", server.get_version, methods=["GET"])
         self.add_api_route("/sdapi/v1/status", server.get_status, methods=["GET"], response_model=models.ResStatus)

modules/api/models.py

@@ -286,10 +286,16 @@ class ResImageInfo(BaseModel):
     items: dict = Field(title="Items", description="A dictionary containing all the other fields the image had")
     parameters: dict = Field(title="Parameters", description="A dictionary with parsed generation info fields")
 
-class ReqLog(BaseModel):
+class ReqGetLog(BaseModel):
     lines: int = Field(default=100, title="Lines", description="How many lines to return")
     clear: bool = Field(default=False, title="Clear", description="Should the log be cleared after returning the lines?")
 
+
+class ReqPostLog(BaseModel):
+    message: Optional[str] = Field(title="Message", description="The info message to log")
+    debug: Optional[str] = Field(title="Debug message", description="The debug message to log")
+    error: Optional[str] = Field(title="Error message", description="The error message to log")
+
 class ReqProgress(BaseModel):
     skip_current_image: bool = Field(default=False, title="Skip current image", description="Skip current image serialization")
 

modules/api/server.py

@@ -37,12 +37,22 @@ def get_platform():
     from modules.loader import get_packages as loader_get_packages
     return { **installer_get_platform(), **loader_get_packages() }
 
-def get_log_buffer(req: models.ReqLog = Depends()):
+def get_log(req: models.ReqGetLog = Depends()):
     lines = shared.log.buffer[:req.lines] if req.lines > 0 else shared.log.buffer.copy()
     if req.clear:
         shared.log.buffer.clear()
     return lines
 
+def post_log(req: models.ReqPostLog):
+    if req.message is not None:
+        shared.log.info(f'UI: {req.message}')
+    if req.debug is not None:
+        shared.log.debug(f'UI: {req.debug}')
+    if req.error is not None:
+        shared.log.error(f'UI: {req.error}')
+    return {}
+
+
 def get_config():
     options = {}
     for k in shared.opts.data.keys():