-
Notifications
You must be signed in to change notification settings - Fork 19
/
copy.go
69 lines (59 loc) · 1.95 KB
/
copy.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package cmd
import (
"fmt"
"os"
"github.com/0chain/gosdk/constants"
"github.com/0chain/gosdk/zboxcore/sdk"
"github.com/spf13/cobra"
)
// copyCmd represents copy command
var copyCmd = &cobra.Command{
Use: "copy",
Short: "copy an object(file/folder) to another folder on blobbers",
Long: `copy an object to another folder on blobbers`,
Args: cobra.MinimumNArgs(0),
Run: func(cmd *cobra.Command, args []string) {
fflags := cmd.Flags() // fflags is a *flag.FlagSet
if !fflags.Changed("allocation") { // check if the flag "path" is set
PrintError("Error: allocation flag is missing") // If not, we'll let the user know
os.Exit(1) // and return
}
if !fflags.Changed("remotepath") {
PrintError("Error: remotepath flag is missing")
os.Exit(1)
}
if !fflags.Changed("destpath") {
PrintError("Error: destpath flag is missing")
os.Exit(1)
}
allocationID := cmd.Flag("allocation").Value.String()
allocationObj, err := sdk.GetAllocation(allocationID)
if err != nil {
PrintError("Error fetching the allocation", err)
os.Exit(1)
}
remotePath := cmd.Flag("remotepath").Value.String()
destPath := cmd.Flag("destpath").Value.String()
err = allocationObj.DoMultiOperation([]sdk.OperationRequest{
{
OperationType: constants.FileOperationCopy,
RemotePath: remotePath,
DestPath: destPath,
},
})
if err != nil {
PrintError("Copy failed.", err)
os.Exit(1)
}
fmt.Println(remotePath + " copied")
},
}
func init() {
rootCmd.AddCommand(copyCmd)
copyCmd.PersistentFlags().String("allocation", "", "Allocation ID")
copyCmd.PersistentFlags().String("remotepath", "", "Remote path of object to copy")
copyCmd.PersistentFlags().String("destpath", "", "Destination path for the object. Existing directory the object should be copied to")
copyCmd.MarkFlagRequired("allocation")
copyCmd.MarkFlagRequired("remotepath")
copyCmd.MarkFlagRequired("destpath")
}