From 43bb61b4e35fb93d2d76a7fe347c3cb9d86bb22d Mon Sep 17 00:00:00 2001
From: Darshit Shah <darnir@gmail.com>
Date: Fri, 13 Sep 2013 14:19:37 +0530
Subject: [PATCH] Smartly guess content type header

---
 testenv/ChangeLog     | 10 ++++++++++
 testenv/HTTPServer.py | 24 ++++++++++++++++++++++--
 2 files changed, 32 insertions(+), 2 deletions(-)

diff --git a/testenv/ChangeLog b/testenv/ChangeLog
index 7bb8b058..3d4093b1 100644
--- a/testenv/ChangeLog
+++ b/testenv/ChangeLog
@@ -1,3 +1,13 @@
+2013-09-13  Darshit Shah  <darnir@gmail.com>
+
+	* HTTPServer.py (_Handler.do_HEAD): If requested path is /, respond
+	with /index.html
+	(_Handler.do_HEAD): Smartly guess value of Content-Type Header from
+	file extension
+	(_Handler.guess_type): Use a preset list of extensions and
+	Content-Type strings. If the extension matches one in the list, use
+	that string, else default to "text/plain"
+
 2013-09-13  Darshit Shah  <darnir@gmail.com>
 
 	* WgetTest.py (CommonMethods._replace_substring): New method that will
diff --git a/testenv/HTTPServer.py b/testenv/HTTPServer.py
index ed78c6a8..e1268ec1 100644
--- a/testenv/HTTPServer.py
+++ b/testenv/HTTPServer.py
@@ -1,4 +1,5 @@
 from http.server import HTTPServer, BaseHTTPRequestHandler
+from posixpath import basename, splitext
 from base64 import b64encode
 from random import random
 from hashlib import md5
@@ -328,7 +329,11 @@ class _Handler (WgetHTTPRequestHandler):
         """ Common code for GET and HEAD Commands.
         This method is overriden to use the fileSys dict.
         """
-        path = self.path[1:]
+
+        if self.path == "/":
+            path = "index.html"
+        else:
+            path = self.path[1:]
 
         if path in self.server.fileSys:
             self.rules = self.server.server_configs.get (path)
@@ -368,7 +373,8 @@ class _Handler (WgetHTTPRequestHandler):
                                                       content_length - 1,
                                                       content_length))
                 content_length -= self.range_begin
-            self.send_header ("Content-type", "text/plain")
+            cont_type = self.guess_type (path)
+            self.send_header ("Content-type", cont_type)
             self.send_header ("Content-Length", content_length)
             self.finish_headers ()
             return (content, self.range_begin)
@@ -376,6 +382,20 @@ class _Handler (WgetHTTPRequestHandler):
             self.send_error (404, "Not Found")
             return (None, None)
 
+    def guess_type (self, path):
+        base_name = basename ("/" + path)
+        name, ext = splitext (base_name)
+        extension_map = {
+        ".txt"   :   "text/plain",
+        ".css"   :   "text/css",
+        ".html"  :   "text/html"
+        }
+        if ext in extension_map:
+            return extension_map[ext]
+        else:
+            return "text/plain"
+
+
 class HTTPd (threading.Thread):
     server_class = StoppableHTTPServer
     handler = _Handler