PUT /api/permission add recursive and owner check

Author: lovehunter9

pkg/fileutils/file.go

@@ -206,6 +206,22 @@ func CommonPrefix(sep byte, paths ...string) string {
 	return string(c)
 }
 
+func ChownRecursive(fs afero.Fs, path string, uid, gid int) error {
+	return filepath.Walk(path, func(name string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		if err := Chown(fs, name, uid, gid); err != nil {
+			fmt.Printf("Failed to chown %s: %v\n", name, err)
+			return err
+		}
+
+		fmt.Printf("Chowned %s\n", name)
+		return nil
+	})
+}
+
 func Chown(fs afero.Fs, path string, uid, gid int) error {
 	start := time.Now()
 	klog.Infoln("Function Chown starts at", start)

pkg/http/data.go

@@ -46,8 +46,17 @@ func handle(fn handleFunc, prefix string, server *settings.Server) http.Handler
 	return stripPrefix(prefix, handler)
 }
 
+func NeedCheckPrefix(prefix string) bool {
+	switch prefix {
+	case "/api/resources", "/api/raw", "/api/preview", "/api/paste", "/api/permission", "/api/share":
+		return true
+	default:
+		return false
+	}
+}
+
 func CheckPathOwner(r *http.Request, prefix string) (bool, error) {
-	if prefix != "/api/resources" && prefix != "/api/raw" && prefix != "/api/preview" && prefix != "/api/paste" {
+	if NeedCheckPrefix(prefix) {
 		return true, nil
 	}
 

pkg/http/permission.go

@@ -46,7 +46,20 @@ func permissionPutHandler(w http.ResponseWriter, r *http.Request, d *common.Data
 	}
 	gid := uid
 
-	err = fileutils.Chown(files.DefaultFs, r.URL.Path, uid, gid)
+	recursiveStr := r.URL.Query().Get("recursive")
+	recursive := 0
+	if recursiveStr != "" {
+		recursive, err = strconv.Atoi(recursiveStr)
+		if err != nil {
+			return http.StatusBadRequest, err
+		}
+	}
+
+	if recursive == 0 {
+		err = fileutils.Chown(files.DefaultFs, r.URL.Path, uid, gid)
+	} else {
+		err = fileutils.ChownRecursive(files.DefaultFs, r.URL.Path, uid, gid)
+	}
 	if err != nil {
 		return http.StatusInternalServerError, err
 	}