From 1e841569ea83301ce9e974617c89eb710471f6c9 Mon Sep 17 00:00:00 2001
From: Anthony Wang <a@exozy.me>
Date: Sat, 1 Oct 2022 09:22:53 -0400
Subject: [PATCH] Basic inbox implementation

---
 .gitignore |  1 +
 inbox.py   | 10 ++++++++++
 server.py  |  8 +++++++-
 3 files changed, 18 insertions(+), 1 deletion(-)
 create mode 100644 .gitignore
 create mode 100644 inbox.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..bee8a64
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+__pycache__
diff --git a/inbox.py b/inbox.py
new file mode 100644
index 0000000..9799b9b
--- /dev/null
+++ b/inbox.py
@@ -0,0 +1,10 @@
+inbox = {}
+
+def Send(data, recipient):
+    if recipient not in inbox.keys():
+        inbox[recipient] = []
+    
+    inbox[recipient].append(data)
+
+def Fetch(user):
+    return inbox[user]
diff --git a/server.py b/server.py
index 747baed..da643e2 100644
--- a/server.py
+++ b/server.py
@@ -1,4 +1,5 @@
 from http.server import HTTPServer, BaseHTTPRequestHandler
+from inbox import Send, Fetch
 
 
 class HTTPRequestHandler(BaseHTTPRequestHandler):
@@ -6,7 +7,12 @@ class HTTPRequestHandler(BaseHTTPRequestHandler):
         self.send_response(200)
         self.send_header("Content-type", "text/html")
         self.end_headers()
-        self.wfile.write(self.path.encode('utf-8'))
+        self.wfile.write(str(Fetch(self.path)).encode('utf-8'))
+
+    def do_POST(self):
+        Send(self.rfile.read(), self.path)
+        self.send_response(200)
+        self.end_headers()
 
 
 server = HTTPServer(("0.0.0.0", 8000), HTTPRequestHandler)