mirror of
https://github.com/rclone/rclone.git
synced 2024-11-30 12:34:04 +08:00
acd: add support for server side DirMove #122
This commit is contained in:
Stefan Breunig
Nick Craig-Wood
parent
2243b065e8
commit
fb1458815a
|
|
@ -13,6 +13,7 @@ we ignore assets completely!
|
|||
*/
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
|
|
@ -630,96 +631,60 @@ func (f *Fs) Move(src fs.Object, remote string) (fs.Object, error) {
|
|||
return nil, fs.ErrorCantMove
|
||||
}
|
||||
fs.Debug(src, "Attempting to move to %q", remote)
|
||||
|
||||
// Temporary Object under construction
|
||||
dstObj := &Object{
|
||||
fs: f,
|
||||
remote: remote,
|
||||
}
|
||||
// check if there's already a directory with the same name
|
||||
pathID, err := f.dirCache.FindDir(remote, false)
|
||||
if err == nil || pathID != "" {
|
||||
fs.Debug(src, "Can't move - there is already a folder with the same name in the target location")
|
||||
return nil, fs.ErrorCantMove
|
||||
}
|
||||
// Check if there's already a file with the same name
|
||||
err = dstObj.readMetaData()
|
||||
switch err {
|
||||
case nil:
|
||||
fs.Debug(src, "Can't move - there is already a file with the same name in the target location")
|
||||
return nil, fs.ErrorCantMove
|
||||
case fs.ErrorObjectNotFound:
|
||||
// Not found so we can move it there
|
||||
default:
|
||||
return nil, err
|
||||
return srcObj.move(remote, false)
|
||||
}
|
||||
|
||||
sLeaf, sDirectoryID, err := f.dirCache.FindPath(srcObj.Remote(), false)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
tLeaf, tDirectoryID, err := f.dirCache.FindPath(remote, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
// DirMove moves src directory to this remote using server side move
|
||||
// operations.
|
||||
//
|
||||
// Will only be called if src.Fs().Name() == f.Name()
|
||||
//
|
||||
// If it isn't possible then return fs.ErrorCantDirMove
|
||||
//
|
||||
// If destination exists then return fs.ErrorDirExists
|
||||
func (f *Fs) DirMove(src fs.Fs) (err error) {
|
||||
fs.Debug(src, "trying to move to %q", f)
|
||||
|
||||
srcFs, ok := src.(*Fs)
|
||||
if !ok {
|
||||
fs.Debug(src, "Can't move - not same remote type")
|
||||
return fs.ErrorCantDirMove
|
||||
}
|
||||
|
||||
if len(srcObj.info.Parents) > 1 && sLeaf != tLeaf {
|
||||
fs.Debug(src, "Can't move - object is attached to multiple parents and should be renamed. This would change the name of the node in all parents.")
|
||||
return nil, fs.ErrorCantMove
|
||||
// Move does not care about this "no op" and will do nothing. DirMove
|
||||
// requires this case to return an error though.
|
||||
if srcFs.root == f.root {
|
||||
return fs.ErrorDirExists
|
||||
}
|
||||
|
||||
if sLeaf != tLeaf {
|
||||
// fs.Debug(src, "renaming")
|
||||
err = srcObj.rename(tLeaf)
|
||||
err = srcFs.dirCache.FindRoot(false)
|
||||
if err != nil {
|
||||
fs.Debug(src, "Move: quick path rename failed: %v", err)
|
||||
goto OnConflict
|
||||
return
|
||||
}
|
||||
}
|
||||
if sDirectoryID != tDirectoryID {
|
||||
// fs.Debug(src, "moving")
|
||||
err = srcObj.replaceParent(sDirectoryID, tDirectoryID)
|
||||
if err != nil {
|
||||
fs.Debug(src, "Move: quick path parent replace failed: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
f.dirCache.Flush()
|
||||
return dstObj, nil
|
||||
node := acd.NodeFromId(srcFs.dirCache.RootID(), f.c.Nodes)
|
||||
|
||||
OnConflict:
|
||||
fs.Debug(src, "Could not directly rename file, presumably because there was a file with the same name already. Instead, the file will now be trashed where such operations do not cause errors. It will be restored to the correct parent after. If any of the subsequent calls fails, the rename/move will be in an invalid state.")
|
||||
var jsonStr string
|
||||
err = f.pacer.Call(func() (bool, error) {
|
||||
jsonStr, err = node.GetMetadata()
|
||||
return f.shouldRetry(nil, err)
|
||||
})
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = json.Unmarshal([]byte(jsonStr), &node)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
// fs.Debug(src, "Trashing file")
|
||||
err = srcObj.Remove()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// fs.Debug(src, "Renaming file")
|
||||
err = srcObj.rename(tLeaf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// note: replacing parent is forbidden by API, modifying them individually is
|
||||
// okay though
|
||||
// fs.Debug(src, "Adding target parent")
|
||||
err = srcObj.addParent(tDirectoryID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// fs.Debug(src, "removing original parent")
|
||||
err = srcObj.removeParent(sDirectoryID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// fs.Debug(src, "Restoring")
|
||||
err = srcObj.restore()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
} else {
|
||||
f.dirCache.Flush()
|
||||
return dstObj, nil
|
||||
srcObj := &Object{
|
||||
fs: srcFs,
|
||||
remote: srcFs.root,
|
||||
info: node,
|
||||
}
|
||||
|
||||
_, err = srcObj.move(f.root, true)
|
||||
srcFs.dirCache.ResetRoot()
|
||||
return
|
||||
}
|
||||
|
||||
// purgeCheck remotes the root directory, if check is set then it
|
||||
|
|
@ -793,51 +758,6 @@ func (f *Fs) Hashes() fs.HashSet {
|
|||
return fs.HashSet(fs.HashMD5)
|
||||
}
|
||||
|
||||
func (f *Fs) Move(src fs.Object, remote string) (fs.Object, error) {
|
||||
srcObj, ok := src.(*Object)
|
||||
if !ok {
|
||||
fs.Debug(src, "Can't move - not same remote type")
|
||||
return nil, fs.ErrorCantMove
|
||||
}
|
||||
|
||||
// Temporary Object under construction
|
||||
dstObj := &Object{
|
||||
fs: f,
|
||||
remote: remote,
|
||||
}
|
||||
|
||||
var err error
|
||||
_, directoryID, err := f.dirCache.FindPath(remote, false)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var info *acd.Node
|
||||
var resp *http.Response
|
||||
if directoryID == srcObj.info.Parents[0] {
|
||||
// Do the rename
|
||||
err = f.pacer.Call(func() (bool, error) {
|
||||
info, resp, err = srcObj.info.Rename(remote)
|
||||
return srcObj.fs.shouldRetry(resp, err)
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
} else {
|
||||
// Do the move
|
||||
err = f.pacer.Call(func() (bool, error) {
|
||||
info, resp, err = srcObj.info.Move(directoryID)
|
||||
return srcObj.fs.shouldRetry(resp, err)
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
dstObj.info = info
|
||||
return dstObj, nil
|
||||
}
|
||||
|
||||
// Copy src to this remote using server side copy operations.
|
||||
//
|
||||
// This is stored with the remote path given
|
||||
|
|
@ -1064,32 +984,129 @@ func (o *Object) rename(newName string) error {
|
|||
|
||||
// Replaces one parent with another, effectively moving the file. Leaves other
|
||||
// parents untouched. ReplaceParent cannot be used when the file is trashed.
|
||||
func (o *Object) replaceParent(oldParentId string, newParentId string) error {
|
||||
fs.Debug(o, "trying parent replace: %s -> %s", oldParentId, newParentId)
|
||||
func (o *Object) replaceParent(oldParentID string, newParentID string) error {
|
||||
fs.Debug(o, "trying parent replace: %s -> %s", oldParentID, newParentID)
|
||||
|
||||
return o.fs.pacer.Call(func() (bool, error) {
|
||||
resp, err := o.info.ReplaceParent(oldParentId, newParentId)
|
||||
resp, err := o.info.ReplaceParent(oldParentID, newParentID)
|
||||
return o.fs.shouldRetry(resp, err)
|
||||
})
|
||||
}
|
||||
|
||||
// Adds one additional parent to object.
|
||||
func (o *Object) addParent(newParentId string) error {
|
||||
func (o *Object) addParent(newParentID string) error {
|
||||
return o.fs.pacer.Call(func() (bool, error) {
|
||||
resp, err := o.info.AddParent(newParentId)
|
||||
resp, err := o.info.AddParent(newParentID)
|
||||
return o.fs.shouldRetry(resp, err)
|
||||
})
|
||||
}
|
||||
|
||||
// Remove given parent from object, leaving the other possible
|
||||
// parents untouched. Object can end up having no parents.
|
||||
func (o *Object) removeParent(parentId string) error {
|
||||
func (o *Object) removeParent(parentID string) error {
|
||||
return o.fs.pacer.Call(func() (bool, error) {
|
||||
resp, err := o.info.RemoveParent(parentId)
|
||||
resp, err := o.info.RemoveParent(parentID)
|
||||
return o.fs.shouldRetry(resp, err)
|
||||
})
|
||||
}
|
||||
|
||||
func (o *Object) move(remote string, useDirErrorMsgs bool) (fs.Object, error) {
|
||||
cantMove := fs.ErrorCantMove
|
||||
existsAlready := fs.ErrorCantMove
|
||||
if useDirErrorMsgs {
|
||||
cantMove = fs.ErrorCantDirMove
|
||||
existsAlready = fs.ErrorDirExists
|
||||
}
|
||||
|
||||
// Temporary Object under construction
|
||||
dstObj := &Object{
|
||||
fs: o.fs,
|
||||
remote: remote,
|
||||
}
|
||||
// check if there's already a directory with the same name
|
||||
pathID, err := o.fs.dirCache.FindDir(remote, false)
|
||||
if err == nil || pathID != "" {
|
||||
fs.Debug(o, "Can't move - there is already a folder with the same name in the target location")
|
||||
return nil, existsAlready
|
||||
}
|
||||
// Check if there's already a file with the same name
|
||||
err = dstObj.readMetaData()
|
||||
switch err {
|
||||
case nil:
|
||||
fs.Debug(o, "Can't move - there is already a file with the same name in the target location")
|
||||
return nil, existsAlready
|
||||
case fs.ErrorObjectNotFound:
|
||||
// Not found so we can move it there
|
||||
default:
|
||||
return nil, err
|
||||
}
|
||||
|
||||
sLeaf, sDirectoryID, err := o.fs.dirCache.FindPath(o.Remote(), false)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
tLeaf, tDirectoryID, err := o.fs.dirCache.FindPath(remote, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(o.info.Parents) > 1 && sLeaf != tLeaf {
|
||||
fs.Debug(o, "Can't move - object is attached to multiple parents and should be renamed. This would change the name of the node in all parents.")
|
||||
return nil, cantMove
|
||||
}
|
||||
|
||||
if sLeaf != tLeaf {
|
||||
// fs.Debug(o, "renaming")
|
||||
err = o.rename(tLeaf)
|
||||
if err != nil {
|
||||
fs.Debug(o, "Move: quick path rename failed: %v", err)
|
||||
goto OnConflict
|
||||
}
|
||||
}
|
||||
if sDirectoryID != tDirectoryID {
|
||||
// fs.Debug(o, "moving")
|
||||
err = o.replaceParent(sDirectoryID, tDirectoryID)
|
||||
if err != nil {
|
||||
// fs.Debug(o, "Move: quick path parent replace failed: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
return dstObj, nil
|
||||
|
||||
OnConflict:
|
||||
fs.Debug(o, "Could not directly rename file, presumably because there was a file with the same name already. Instead, the file will now be trashed where such operations do not cause errors. It will be restored to the correct parent after. If any of the subsequent calls fails, the rename/move will be in an invalid state.")
|
||||
|
||||
// fs.Debug(o, "Trashing file")
|
||||
err = o.Remove()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// fs.Debug(o, "Renaming file")
|
||||
err = o.rename(tLeaf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// note: replacing parent is forbidden by API, modifying them individually is
|
||||
// okay though
|
||||
// fs.Debug(o, "Adding target parent")
|
||||
err = o.addParent(tDirectoryID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// fs.Debug(o, "removing original parent")
|
||||
err = o.removeParent(sDirectoryID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// fs.Debug(o, "Restoring")
|
||||
err = o.restore()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return dstObj, nil
|
||||
}
|
||||
|
||||
// MimeType of an Object if known, "" otherwise
|
||||
func (o *Object) MimeType() string {
|
||||
if o.info.ContentProperties.ContentType != nil {
|
||||
|
|
@ -1104,7 +1121,7 @@ var (
|
|||
_ fs.Purger = (*Fs)(nil)
|
||||
// _ fs.Copier = (*Fs)(nil)
|
||||
_ fs.Mover = (*Fs)(nil)
|
||||
// _ fs.DirMover = (*Fs)(nil)
|
||||
_ fs.DirMover = (*Fs)(nil)
|
||||
_ fs.Object = (*Object)(nil)
|
||||
_ fs.MimeTyper = &Object{}
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user