aliyunpan/internal/webdav/user.go

package webdav

import (
	"regexp"
	"strings"

	"golang.org/x/net/webdav"
)

// Rule is a dissalow/allow rule.
type Rule struct {
	Regex  bool
	Allow  bool
	Modify bool
	Path   string
	Regexp *regexp.Regexp
}

// User contains the settings of each user.
type User struct {
	Username string
	Password string
	Scope    string
	Modify   bool
	Rules    []*Rule
	Handler  *webdav.Handler
}

// Allowed checks if the user has permission to access a directory/file
func (u User) Allowed(url string, noModification bool) bool {
	var rule *Rule
	i := len(u.Rules) - 1

	for i >= 0 {
		rule = u.Rules[i]

		isAllowed := rule.Allow && (noModification || rule.Modify)
		if rule.Regex {
			if rule.Regexp.MatchString(url) {
				return isAllowed
			}
		} else if strings.HasPrefix(url, rule.Path) {
			return isAllowed
		}

		i--
	}

	return noModification || u.Modify
}