From 873cb075d8f1be5bb3dd06e025b967e83f797a04 Mon Sep 17 00:00:00 2001
From: Brad Fitzpatrick <brad@danga.com>
Date: Sat, 29 Jan 2011 00:00:59 -0800
Subject: [PATCH] Go blobserver: guess a MIME type for better demos

---
 server/go/blobserver/get.go | 37 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 36 insertions(+), 1 deletion(-)

diff --git a/server/go/blobserver/get.go b/server/go/blobserver/get.go
index 9500264fa..53cee7b3c 100644
--- a/server/go/blobserver/get.go
+++ b/server/go/blobserver/get.go
@@ -17,6 +17,7 @@ limitations under the License.
 package main
 
 import (
+	"bufio"
 	"bytes"
 	"camli/auth"
 	"camli/blobref"
@@ -178,7 +179,29 @@ func handleGet(conn http.ResponseWriter, req *http.Request, fetcher blobref.Fetc
 		remainBytes = reqRange.LimitBytes
 	}
 
-	conn.SetHeader("Content-Type", "application/octet-stream")
+	// Assume this generic content type by default.  For better
+	// demos we'll try to sniff and guess the "right" MIME type in
+	// certain cases (no Range requests, etc) but this isn't part
+	// of the Camli spec at all.  We just do it to ease demos.
+	contentType := "application/octet-stream"
+	if reqRange.IsWholeFile() {
+		const peekSize = 1024
+		bufReader, _ := bufio.NewReaderSize(input, peekSize)
+		header, _ := bufReader.Peek(peekSize)
+		if len(header) >= 8 {
+			switch {
+			case isValidUtf8(string(header)):
+				contentType = "text/plain; charset=utf-8"
+			case bytes.HasPrefix(header, []byte{0xff, 0xd8, 0xff, 0xe2}):
+				contentType = "image/jpeg"
+			case bytes.HasPrefix(header, []byte{0x89, 0x50, 0x4e, 0x47, 0xd, 0xa, 0x1a, 0xa}):
+				contentType = "image/png"
+			}
+		}
+		input = bufReader
+	}
+
+	conn.SetHeader("Content-Type", contentType)
 	if !reqRange.IsWholeFile() {
 		conn.SetHeader("Content-Range",
 			fmt.Sprintf("bytes %d-%d/%d", reqRange.SkipBytes,
@@ -211,3 +234,15 @@ func handleGet(conn http.ResponseWriter, req *http.Request, fetcher blobref.Fetc
 		return
 	}
 }
+
+// TODO: copied this from lib/go/schema, but this might not be ideal.
+// unify and speed up?
+func isValidUtf8(s string) bool {
+	for _, rune := range []int(s) {
+		if rune == 0xfffd {
+			return false
+		}
+	}
+	return true
+
+}