From 43a8d256193a573b4726b3be9cb2c7205d966324 Mon Sep 17 00:00:00 2001
From: Arksine <9563098+Arksine@users.noreply.github.com>
Date: Mon, 12 Apr 2021 17:07:16 -0400
Subject: [PATCH] websockets:  Store IP Address in WebRequest object

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
---
 moonraker/app.py        |  5 +++--
 moonraker/websockets.py | 13 ++++++++++---
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/moonraker/app.py b/moonraker/app.py
index 23e5edc..5226603 100644
--- a/moonraker/app.py
+++ b/moonraker/app.py
@@ -430,11 +430,12 @@ class DynamicRequestHandler(AuthorizedRequestHandler):
     async def _do_local_request(self, args, conn):
         return await self.callback(
             WebRequest(self.request.path, args, self.request.method,
-                       conn=conn))
+                       conn=conn, ip_addr=self.request.remote_ip))
 
     async def _do_remote_request(self, args, conn):
         return await self.server.make_request(
-            WebRequest(self.callback, args, conn=conn))
+            WebRequest(self.callback, args, conn=conn,
+                       ip_addr=self.request.remote_ip))
 
     async def _process_http_request(self):
         if self.request.method not in self.methods:
diff --git a/moonraker/websockets.py b/moonraker/websockets.py
index 0629399..49e1c55 100644
--- a/moonraker/websockets.py
+++ b/moonraker/websockets.py
@@ -15,11 +15,13 @@ class Sentinel:
     pass
 
 class WebRequest:
-    def __init__(self, endpoint, args, action="", conn=None):
+    def __init__(self, endpoint, args, action="",
+                 conn=None, ip_addr=""):
         self.endpoint = endpoint
         self.action = action
         self.args = args
         self.conn = conn
+        self.ip_addr = ip_addr
 
     def get_endpoint(self):
         return self.endpoint
@@ -33,6 +35,9 @@ class WebRequest:
     def get_connection(self):
         return self.conn
 
+    def get_ip_address(self):
+        return self.ip_addr
+
     def _get_converted_arg(self, key, default=Sentinel, dtype=str):
         if key not in self.args:
             if default == Sentinel:
@@ -202,14 +207,15 @@ class WebsocketManager:
     def _generate_callback(self, endpoint):
         async def func(ws, **kwargs):
             result = await self.server.make_request(
-                WebRequest(endpoint, kwargs, conn=ws))
+                WebRequest(endpoint, kwargs, conn=ws, ip_addr=ws.ip_addr))
             return result
         return func
 
     def _generate_local_callback(self, endpoint, request_method, callback):
         async def func(ws, **kwargs):
             result = await callback(
-                WebRequest(endpoint, kwargs, request_method, ws))
+                WebRequest(endpoint, kwargs, request_method,
+                           ws, ip_addr=ws.ip_addr))
             return result
         return func
 
@@ -264,6 +270,7 @@ class WebSocket(WebSocketHandler):
         self.rpc = self.wsm.rpc
         self.uid = id(self)
         self.is_closed = False
+        self.ip_addr = self.request.remote_ip
 
     async def open(self):
         await self.wsm.add_websocket(self)