Mercurial Hosting > luan
annotate src/luan/webserver/handlers/ContentTypeHandler.java @ 1165:668f29bc52ea
clean up content-type
| author | Franklin Schmidt <fschmidt@gmail.com> |
|---|---|
| date | Wed, 07 Feb 2018 23:16:12 -0700 |
| parents | 49fb4e83484f |
| children | 7e7c2d0c3b99 |
| rev | line source |
|---|---|
| 1137 | 1 package luan.webserver.handlers; |
| 2 | |
| 3 import java.util.Map; | |
| 4 import java.util.HashMap; | |
| 5 import luan.webserver.Handler; | |
| 6 import luan.webserver.Request; | |
| 7 import luan.webserver.Response; | |
| 8 | |
| 9 | |
| 10 public class ContentTypeHandler implements Handler { | |
| 11 private final Handler handler; | |
| 12 | |
| 13 // maps extension to content-type | |
| 14 // key must be lower case | |
| 15 public final Map<String,String> map = new HashMap<String,String>(); | |
| 16 | |
| 17 // set to null for none | |
| 18 public String contentTypeForNoExtension; | |
| 19 | |
| 20 public ContentTypeHandler(Handler handler) { | |
|
1165
668f29bc52ea
clean up content-type
Franklin Schmidt <fschmidt@gmail.com>
parents:
1148
diff
changeset
|
21 this(handler,"utf-8"); |
| 1137 | 22 } |
| 23 | |
| 24 public ContentTypeHandler(Handler handler,String charset) { | |
| 25 this.handler = handler; | |
|
1165
668f29bc52ea
clean up content-type
Franklin Schmidt <fschmidt@gmail.com>
parents:
1148
diff
changeset
|
26 String attrs = charset== null ? "" : "; charset="+charset; |
|
668f29bc52ea
clean up content-type
Franklin Schmidt <fschmidt@gmail.com>
parents:
1148
diff
changeset
|
27 String htmlType = "text/html" + attrs; |
|
668f29bc52ea
clean up content-type
Franklin Schmidt <fschmidt@gmail.com>
parents:
1148
diff
changeset
|
28 String textType = "text/plain" + attrs; |
| 1137 | 29 contentTypeForNoExtension = htmlType; |
| 30 map.put( "html", htmlType ); | |
| 31 map.put( "txt", textType ); | |
| 32 // add more as need | |
| 33 } | |
| 34 | |
| 35 public Response handle(Request request) { | |
| 36 Response response = handler.handle(request); | |
|
1165
668f29bc52ea
clean up content-type
Franklin Schmidt <fschmidt@gmail.com>
parents:
1148
diff
changeset
|
37 if( response!=null && !response.headers.containsKey("content-type") ) { |
| 1137 | 38 String path = request.path; |
| 39 int iSlash = path.lastIndexOf('/'); | |
| 40 int iDot = path.lastIndexOf('.'); | |
| 41 String type; | |
| 42 if( iDot < iSlash ) { // no extension | |
| 43 type = contentTypeForNoExtension; | |
| 44 } else { // extension | |
| 45 String extension = path.substring(iDot+1); | |
| 46 type = map.get( extension.toLowerCase() ); | |
| 47 } | |
| 48 if( type != null ) | |
|
1165
668f29bc52ea
clean up content-type
Franklin Schmidt <fschmidt@gmail.com>
parents:
1148
diff
changeset
|
49 response.headers.put("content-type",type); |
| 1137 | 50 } |
| 51 return response; | |
| 52 } | |
| 53 } |