webdav/lib/locksystem.go

package lib

import (
	"path/filepath"
	"time"

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

var _ webdav.LockSystem = &lockSystem{}

// lockSystem wraps a [webdav.LockSystem], mapping virtual request names to the
// real backing paths via resolve. This allows reusing the same
// [webdav.LockSystem] for multiple users with different base directories,
// meaning we can correctly lock the files across different users.
type lockSystem struct {
	webdav.LockSystem
	resolve func(name string) (string, error)
}

// newLockSystem returns a lockSystem for a single-directory user, resolving
// names relative to directory.
func newLockSystem(ls webdav.LockSystem, directory string) *lockSystem {
	return &lockSystem{
		LockSystem: ls,
		resolve: func(name string) (string, error) {
			return filepath.Join(directory, name), nil
		},
	}
}

// newMultiDirLockSystem returns a lockSystem for a multi-directory user,
// resolving names against the real backing path of each mount.
func newMultiDirLockSystem(ls webdav.LockSystem, mounts DirectoryMounts) *lockSystem {
	return &lockSystem{
		LockSystem: ls,
		resolve: func(name string) (string, error) {
			if cleanName(name) == "/" {
				return "/", nil
			}

			mount, rest, err := multiDir{mounts: mounts}.resolve(name)
			if err != nil {
				return "", err
			}

			return mount.filePath(rest), nil
		},
	}
}

func (l *lockSystem) Confirm(now time.Time, name0, name1 string, conditions ...webdav.Condition) (release func(), err error) {
	if name0 != "" {
		name0, err = l.resolve(name0)
		if err != nil {
			return nil, err
		}
	}

	if name1 != "" {
		name1, err = l.resolve(name1)
		if err != nil {
			return nil, err
		}
	}

	return l.LockSystem.Confirm(now, name0, name1, conditions...)
}

func (l *lockSystem) Create(now time.Time, details webdav.LockDetails) (token string, err error) {
	details.Root, err = l.resolve(details.Root)
	if err != nil {
		return "", err
	}
	return l.LockSystem.Create(now, details)
}