From 658763dd5baf5088fa6635083a7377e70d4b6ab2 Mon Sep 17 00:00:00 2001
From: David Bailey <techdavid@users.noreply.github.com>
Date: Sat, 19 Aug 2017 23:24:53 +0100
Subject: [PATCH] Allow other add-ons to add headers

---
 AnkiConnect.py | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/AnkiConnect.py b/AnkiConnect.py
index 9fbe5a0..a4acd20 100644
--- a/AnkiConnect.py
+++ b/AnkiConnect.py
@@ -199,6 +199,19 @@ class AjaxServer:
         self.handler = handler
         self.clients = []
         self.sock = None
+        self.resetHeaders()
+
+
+    def addHeader(self, name, value):
+        self.headers.append([name, value])
+    
+    
+    def resetHeaders(self):
+        self.headers = [
+            ['HTTP/1.1 200 OK', None],
+            ['Content-Type', 'text/json'],
+            ['Content-Length', '']
+        ]
 
 
     def advance(self):
@@ -243,11 +256,9 @@ class AjaxServer:
                 body = json.dumps(None);
 
         resp = bytes()
-        headers = [
-            ['HTTP/1.1 200 OK', None],
-            ['Content-Type', 'text/json'],
-            ['Content-Length', str(len(body))]
-        ]
+        
+        headers = self.headers
+        headers[2][1] = str(len(body))
 
         for key, value in headers:
             if value is None: