From cb53ddc8e8e71f5fedcac2a2af5413fa650948cf Mon Sep 17 00:00:00 2001
From: Xhofe <i@nn.ci>
Date: Wed, 16 Feb 2022 16:10:39 +0800
Subject: [PATCH] :bug: fix nil pointer

---
 drivers/teambition/teambition.go | 4 +++-
 server/webdav/webdav.go          | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/drivers/teambition/teambition.go b/drivers/teambition/teambition.go
index a1996985..e44e251c 100644
--- a/drivers/teambition/teambition.go
+++ b/drivers/teambition/teambition.go
@@ -157,8 +157,10 @@ func (driver Teambition) upload(file *model.FileStream, token string, account *m
 
 func (driver Teambition) chunkUpload(file *model.FileStream, token string, account *model.Account) (*FileUpload, error) {
 	prefix := "tcs"
+	referer := "https://www.teambition.com/"
 	if account.InternalType == "International" {
 		prefix = "us-tcs"
+		referer = "https://us.teambition.com/"
 	}
 	var newChunk ChunkUpload
 	_, err := base.RestyClient.R().SetResult(&newChunk).SetHeader("Authorization", token).
@@ -187,7 +189,7 @@ func (driver Teambition) chunkUpload(file *model.FileStream, token string, accou
 		res, err := base.RestyClient.R().SetHeaders(map[string]string{
 			"Authorization": token,
 			"Content-Type":  "application/octet-stream",
-			"Referer":       "https://www.teambition.com/",
+			"Referer":       referer,
 		}).SetBody(chunkData).Post(u)
 		if err != nil {
 			return nil, err
diff --git a/server/webdav/webdav.go b/server/webdav/webdav.go
index cc24309b..3db500e3 100644
--- a/server/webdav/webdav.go
+++ b/server/webdav/webdav.go
@@ -238,8 +238,8 @@ func (h *Handler) handleGetHeadPost(w http.ResponseWriter, r *http.Request, fs *
 	w.Header().Set("ETag", etag)
 	log.Debugf("url: %+v", r.URL)
 	link, err := fs.Link(w, r, reqPath)
-	log.Debugf("webdav link error: %s", err.Error())
 	if err != nil {
+		log.Debugf("webdav link error: %s", err.Error())
 		return http.StatusInternalServerError, err
 	}
 	if link != "" {