Upload files transmit refactored

Builder/IISMainHandler/build.txt

@@ -1 +1 @@
-1749
+1751

Common/UploadManager.cs

@@ -129,5 +129,25 @@ namespace FLocal.Common {
 			return upload;
 		}
 
+		public static void WriteUpload(Upload upload, Stream output) {
+			HttpWebRequest request = (HttpWebRequest)WebRequest.Create(Config.instance.UploaderUrl + "Data/" + upload.hash + "." + upload.extension);
+			request.Method = "GET";
+			HttpWebResponse response = null;
+			try {
+				try {
+					response = (HttpWebResponse)request.GetResponse();
+				} catch(WebException e) {
+					response = (HttpWebResponse)e.Response;
+				}
+				using(Stream responseStream = response.GetResponseStream()) {
+					responseStream.WriteTo(output);
+				}
+			} finally {
+				if(response != null) {
+					response.Close();
+				}
+			}
+		}
+
 	}
 }

Core/extensions/Extensions.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Linq;
 using System.Text;
+using System.IO;
 
 namespace FLocal.Core {
 	
@@ -247,5 +248,18 @@ namespace FLocal.Core {
 			return enumerable.Union((IEnumerable<T>)second);
 		}
 
+		/// <summary>
+		/// Code taken from http://msdn.microsoft.com/en-us/library/system.io.stream.write.aspx
+		/// </summary>
+		/// <param name="input"></param>
+		/// <param name="output"></param>
+		public static void WriteTo(this Stream input, Stream output) {
+			const int size = 4096;
+			byte[] bytes = new byte[4096];
+			int numBytes;
+			while((numBytes = input.Read(bytes, 0, size)) > 0)
+				output.Write(bytes, 0, numBytes);
+		}
+
     }
 }

IISMainHandler/handlers/response/UploadHandler.cs

@@ -24,7 +24,22 @@ namespace FLocal.IISHandler.handlers.response {
 				throw new AccessViolationException();
 			}
 
-			return Config.instance.UploaderUrl + "Data/" + this.url.upload.hash + "." + this.url.upload.extension;
+			string mime = Util.getMimeByExtension(this.url.upload.extension);
+			if(mime != null) {
+				context.httpresponse.ContentType = mime;
+			} else {
+				//throw new HttpException(403, "wrong file type");
+				throw new WrongUrlException();
+			}
+			context.httpresponse.AddHeader("content-disposition", "attachment; filename=" + this.url.upload.filename);
+
+			context.httpresponse.Cache.SetExpires(DateTime.Now.AddDays(10));
+			context.httpresponse.Cache.SetLastModified(this.url.upload.uploadDate);
+			context.httpresponse.Cache.SetCacheability(System.Web.HttpCacheability.Public);
+
+			UploadManager.WriteUpload(this.url.upload, context.httpresponse.OutputStream);
+
+			throw new response.SkipXsltTransformException();
 		}
 
 	}