Refine the rest structure.

code/rest/dav_controller.go

@@ -0,0 +1,174 @@
+package rest
+
+import (
+	"bytes"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"regexp"
+	"strings"
+	"tank/code/tool/result"
+	"tank/code/tool/util"
+)
+
+/**
+ *
+ * WebDav协议文档
+ * https://tools.ietf.org/html/rfc4918
+ * http://www.webdav.org/specs/rfc4918.html
+ *
+ */
+
+type DavController struct {
+	BaseController
+	uploadTokenDao    *UploadTokenDao
+	downloadTokenDao  *DownloadTokenDao
+	matterDao         *MatterDao
+	matterService     *MatterService
+	imageCacheDao     *ImageCacheDao
+	imageCacheService *ImageCacheService
+	davService        *DavService
+}
+
+//初始化方法
+func (this *DavController) Init() {
+	this.BaseController.Init()
+
+	//手动装填本实例的Bean.
+	b := CONTEXT.GetBean(this.uploadTokenDao)
+	if c, ok := b.(*UploadTokenDao); ok {
+		this.uploadTokenDao = c
+	}
+
+	b = CONTEXT.GetBean(this.downloadTokenDao)
+	if c, ok := b.(*DownloadTokenDao); ok {
+		this.downloadTokenDao = c
+	}
+
+	b = CONTEXT.GetBean(this.matterDao)
+	if c, ok := b.(*MatterDao); ok {
+		this.matterDao = c
+	}
+
+	b = CONTEXT.GetBean(this.matterService)
+	if c, ok := b.(*MatterService); ok {
+		this.matterService = c
+	}
+
+	b = CONTEXT.GetBean(this.imageCacheDao)
+	if c, ok := b.(*ImageCacheDao); ok {
+		this.imageCacheDao = c
+	}
+
+	b = CONTEXT.GetBean(this.imageCacheService)
+	if c, ok := b.(*ImageCacheService); ok {
+		this.imageCacheService = c
+	}
+
+	b = CONTEXT.GetBean(this.davService)
+	if c, ok := b.(*DavService); ok {
+		this.davService = c
+	}
+}
+
+//通过BasicAuth的方式授权。
+func (this *DavController) CheckCurrentUser(writer http.ResponseWriter, request *http.Request) *User {
+
+	username, password, ok := request.BasicAuth()
+	if !ok {
+		//要求前端使用Basic的形式授权
+		writer.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
+
+		panic(result.ConstWebResult(result.CODE_WRAPPER_LOGIN))
+
+	}
+
+	user := this.userDao.FindByUsername(username)
+	if user == nil {
+		panic(result.BadRequest("邮箱或密码错误"))
+	} else {
+		if !util.MatchBcrypt(password, user.Password) {
+			panic(result.BadRequest("邮箱或密码错误"))
+		}
+	}
+
+	return user
+}
+
+//注册自己的路由。
+func (this *DavController) RegisterRoutes() map[string]func(writer http.ResponseWriter, request *http.Request) {
+
+	routeMap := make(map[string]func(writer http.ResponseWriter, request *http.Request))
+
+	return routeMap
+}
+
+//处理一些特殊的接口，比如参数包含在路径中,一般情况下，controller不将参数放在url路径中
+func (this *DavController) HandleRoutes(writer http.ResponseWriter, request *http.Request) (func(writer http.ResponseWriter, request *http.Request), bool) {
+
+	path := request.URL.Path
+
+	//匹配 /api/dav{subPath}
+	pattern := fmt.Sprintf(`^%s(.*)$`, WEBDAV_PREFFIX)
+	reg := regexp.MustCompile(pattern)
+	strs := reg.FindStringSubmatch(path)
+	if len(strs) == 2 {
+		var f = func(writer http.ResponseWriter, request *http.Request) {
+			subPath := strs[1]
+			//保证subPath不是以/结尾的。
+			//最后多余的/要去掉
+			if strings.HasSuffix(subPath, "/") {
+				subPath = subPath[0 : len(subPath)-1]
+			}
+			this.Index(writer, request, subPath)
+		}
+		return f, true
+	}
+
+	return nil, false
+}
+
+//完成系统安装
+func (this *DavController) Index(writer http.ResponseWriter, request *http.Request, subPath string) {
+
+	/*打印所有HEADER以及请求参数*/
+
+	fmt.Printf("\n------ 请求： %s  --  %s  ------\n", request.URL, subPath)
+
+	fmt.Printf("\n------Method：------\n")
+	fmt.Println(request.Method)
+
+	fmt.Printf("\n------Header：------\n")
+	for key, value := range request.Header {
+		fmt.Printf("%s = %s\n", key, value)
+	}
+
+	fmt.Printf("\n------请求参数：------\n")
+	for key, value := range request.Form {
+		fmt.Printf("%s = %s\n", key, value)
+	}
+
+	fmt.Printf("\n------Body：------\n")
+	//ioutil.ReadAll 不可重复读，第二次读的时候就什么都没有了。
+
+	bodyBytes, err := ioutil.ReadAll(request.Body)
+	if err != nil {
+		fmt.Println("读取body时出错" + err.Error())
+	}
+	fmt.Println(string(bodyBytes))
+
+	//关闭之后再重新赋值
+	err = request.Body.Close()
+	if err != nil {
+		panic(err)
+	}
+	request.Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes))
+
+	fmt.Println("------------------")
+
+	//获取请求者
+	user := this.CheckCurrentUser(writer, request)
+
+	this.davService.HandleDav(writer, request, user, subPath)
+
+}