From 3ff2d2769764dcc4bd46b9a0b5e95249b74b01ef Mon Sep 17 00:00:00 2001
From: Bob Mottram <bob@freedombone.net>
Date: Wed, 4 Dec 2019 13:54:59 +0000
Subject: [PATCH] Include length and etag in head

---
 daemon.py | 85 +++++++++++++++++++++++++++++++++++++++----------------
 1 file changed, 60 insertions(+), 25 deletions(-)

diff --git a/daemon.py b/daemon.py
index dfff1539..c21c69bf 100644
--- a/daemon.py
+++ b/daemon.py
@@ -386,6 +386,12 @@ class PubServer(BaseHTTPRequestHandler):
         self._set_headers_base(fileFormat,length,cookie)
         self.end_headers()
 
+    def _set_headers_head(self,fileFormat: str,length: int,etag: str) -> None:
+        self._set_headers_base(fileFormat,length,None)
+        if etag:
+            self.send_header('ETag',etag)
+        self.end_headers()
+
     def _set_headers_etag(self,mediaFilename: str,fileFormat: str, \
                           data,cookie: str) -> None:
         self._set_headers_base(fileFormat,len(data),cookie)
@@ -3507,31 +3513,60 @@ class PubServer(BaseHTTPRequestHandler):
         self._benchmarkGETtimings(GETstartTime,GETtimings,55)
 
     def do_HEAD(self):
-        if self.path.endswith('.png'):
-            self._set_headers('image/png',-1,None)
-            return
-        if self.path.endswith('.jpg') or self.path.endswith('.jpeg'):
-            self._set_headers('image/jpeg',-1,None)
-            return
-        if self.path.endswith('.gif'):
-            self._set_headers('image/gif',-1,None)
-            return
-        if self.path.endswith('.webp'):
-            self._set_headers('image/webp',-1,None)
-            return
-        if self.path.endswith('.mp4'):
-            self._set_headers('video/mp4',-1,None)
-            return
-        if self.path.endswith('.ogv'):
-            self._set_headers('video/ogv',-1,None)
-            return
-        if self.path.endswith('.mp3'):
-            self._set_headers('audio/mpeg',-1,None)
-            return
-        if self.path.endswith('.ogg'):
-            self._set_headers('audio/ogg',-1,None)
-            return        
-        self._set_headers('application/json',-1,None)        
+        checkPath=self.path
+        etag=None
+        fileLength=-1
+
+        if '/media/' in self.path:
+            if self.path.endswith('.png') or \
+               self.path.endswith('.jpg') or \
+               self.path.endswith('.gif') or \
+               self.path.endswith('.webp') or \
+               self.path.endswith('.mp4') or \
+               self.path.endswith('.ogv') or \
+               self.path.endswith('.mp3') or \
+               self.path.endswith('.ogg'):
+                mediaStr=self.path.split('/media/')[1]
+                mediaFilename= \
+                    self.server.baseDir+'/media/'+mediaStr
+                if os.path.isfile(mediaFilename):
+                    checkPath=mediaFilename
+                    fileLength=os.path.getsize(mediaFilename)
+                    if os.path.isfile(mediaFilename+'.etag'):
+                        try:
+                            with open(mediaFilename+'.etag', 'r') as etagFile:
+                                etag = etagFile.read()
+                        except:
+                            pass
+                    else:
+                        with open(mediaFilename, 'rb') as avFile:
+                            mediaBinary = avFile.read()
+                            etag=sha1(data).hexdigest()
+                            try:
+                                with open(mediaFilename+'.etag', 'w') as etagFile:
+                                    etagFile.write(etag)
+                            except:
+                                pass
+
+        mediaFileType='application/json'
+        if checkPath.endswith('.png'):
+            mediaFileType='image/png'
+        elif checkPath.endswith('.jpg'):
+            mediaFileType='image/jpeg'
+        elif checkPath.endswith('.gif'):
+            mediaFileType='image/gif'
+        elif checkPath.endswith('.webp'):
+            mediaFileType='image/webp'
+        elif checkPath.endswith('.mp4'):
+            mediaFileType='video/mp4'
+        elif checkPath.endswith('.ogv'):
+            mediaFileType='video/ogv'
+        elif checkPath.endswith('.mp3'):
+            mediaFileType='audio/mpeg'
+        elif checkPath.endswith('.ogg'):
+            mediaFileType='audio/ogg'
+
+        self._set_headers_head(mediaFileType,fileLength,etag)
 
     def _receiveNewPostProcess(self,authorized: bool, \
                                postType: str,path: str,headers: {},