123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183 |
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Linq;
- using System.Text.RegularExpressions;
- using System.Web;
- /// <summary>
- /// UploadHandler 的摘要说明
- /// </summary>
- public class UploadHandler : Handler
- {
- public UploadConfig UploadConfig { get; private set; }
- public UploadResult Result { get; private set; }
- public UploadHandler(HttpContext context, UploadConfig config)
- : base(context)
- {
- this.UploadConfig = config;
- this.Result = new UploadResult() { State = UploadState.Unknown };
- }
- public override void Process()
- {
- byte[] uploadFileBytes = null;
- string uploadFileName = null;
- if (UploadConfig.Base64)
- {
- uploadFileName = UploadConfig.Base64Filename;
- uploadFileBytes = Convert.FromBase64String(Request[UploadConfig.UploadFieldName]);
- }
- else
- {
- var file = Request.Files[UploadConfig.UploadFieldName];
- uploadFileName = file.FileName;
- if (!CheckFileType(uploadFileName))
- {
- Result.State = UploadState.TypeNotAllow;
- WriteResult();
- return;
- }
- if (!CheckFileSize(file.ContentLength))
- {
- Result.State = UploadState.SizeLimitExceed;
- WriteResult();
- return;
- }
- uploadFileBytes = new byte[file.ContentLength];
- try
- {
- file.InputStream.Read(uploadFileBytes, 0, file.ContentLength);
- }
- catch (Exception)
- {
- Result.State = UploadState.NetworkError;
- WriteResult();
- }
- }
- Result.OriginFileName = uploadFileName;
- var savePath = PathFormatter.Format(uploadFileName, UploadConfig.PathFormat);
- var localPath = Server.MapPath(savePath);
- try
- {
- if (!Directory.Exists(Path.GetDirectoryName(localPath)))
- {
- Directory.CreateDirectory(Path.GetDirectoryName(localPath));
- }
- File.WriteAllBytes(localPath, uploadFileBytes);
- Result.Url = savePath;
- Result.State = UploadState.Success;
- }
- catch (Exception e)
- {
- Result.State = UploadState.FileAccessError;
- Result.ErrorMessage = e.Message;
- }
- finally
- {
- WriteResult();
- }
- }
- private void WriteResult()
- {
- this.WriteJson(new
- {
- state = GetStateMessage(Result.State),
- url = Result.Url,
- title = Result.OriginFileName,
- original = Result.OriginFileName,
- error = Result.ErrorMessage
- });
- }
- private string GetStateMessage(UploadState state)
- {
- switch (state)
- {
- case UploadState.Success:
- return "SUCCESS";
- case UploadState.FileAccessError:
- return "文件访问出错,请检查写入权限";
- case UploadState.SizeLimitExceed:
- return "文件大小超出服务器限制";
- case UploadState.TypeNotAllow:
- return "不允许的文件格式";
- case UploadState.NetworkError:
- return "网络错误";
- }
- return "未知错误";
- }
- private bool CheckFileType(string filename)
- {
- var fileExtension = Path.GetExtension(filename).ToLower();
- return UploadConfig.AllowExtensions.Select(x => x.ToLower()).Contains(fileExtension);
- }
- private bool CheckFileSize(int size)
- {
- return size < UploadConfig.SizeLimit;
- }
- }
- public class UploadConfig
- {
- /// <summary>
- /// 文件命名规则
- /// </summary>
- public string PathFormat { get; set; }
- /// <summary>
- /// 上传表单域名称
- /// </summary>
- public string UploadFieldName { get; set; }
- /// <summary>
- /// 上传大小限制
- /// </summary>
- public int SizeLimit { get; set; }
- /// <summary>
- /// 上传允许的文件格式
- /// </summary>
- public string[] AllowExtensions { get; set; }
- /// <summary>
- /// 文件是否以 Base64 的形式上传
- /// </summary>
- public bool Base64 { get; set; }
- /// <summary>
- /// Base64 字符串所表示的文件名
- /// </summary>
- public string Base64Filename { get; set; }
- }
- public class UploadResult
- {
- public UploadState State { get; set; }
- public string Url { get; set; }
- public string OriginFileName { get; set; }
- public string ErrorMessage { get; set; }
- }
- public enum UploadState
- {
- Success = 0,
- SizeLimitExceed = -1,
- TypeNotAllow = -2,
- FileAccessError = -3,
- NetworkError = -4,
- Unknown = 1,
- }
|