1
0
Fork 0
mirror of https://github.com/Luzifer/dbx-sync.git synced 2024-12-22 19:01:17 +00:00

Initial working version

This commit is contained in:
Knut Ahlers 2016-10-10 16:50:27 +02:00
commit 25ee865326
Signed by: luzifer
GPG key ID: DC2729FDD34BE99E
2 changed files with 154 additions and 0 deletions

2
.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
.env
dbx-sync

152
main.go Normal file
View file

@ -0,0 +1,152 @@
package main
import (
"fmt"
"log"
"os"
"path"
"path/filepath"
"strings"
"time"
"github.com/Luzifer/rconfig"
"github.com/dropbox/dropbox-sdk-go-unofficial/dropbox"
"github.com/dropbox/dropbox-sdk-go-unofficial/dropbox/files"
)
var (
cfg = struct {
DropboxToken string `flag:"dropbox-token" env:"DROPBOX_TOKEN" description:"Dropbox token to use to access API"`
ForceOverwrite bool `flag:"force-overwrite" default:"false" description:"Upload files even when they exist on target"`
Verbose bool `flag:"verbose,v" default:"false" description:"Enable verbose logging"`
VersionAndExit bool `flag:"version" default:"false" description:"Prints current version and exits"`
}{}
version = "dev"
)
func init() {
if err := rconfig.Parse(&cfg); err != nil {
log.Fatalf("Unable to parse commandline options: %s", err)
}
if cfg.VersionAndExit {
fmt.Printf("dbx-sync %s\n", version)
os.Exit(0)
}
}
func main() {
if len(rconfig.Args()) != 3 {
log.Fatalf("Usage: dbx-sync <source path> <destination prefix>")
}
sourcePath := rconfig.Args()[1]
destPrefix := rconfig.Args()[2]
if f, err := os.Stat(sourcePath); err != nil || !f.IsDir() {
log.Fatalf("Source path %q does not exist or is not a directory.", sourcePath)
}
if destPrefix[0] != '/' {
log.Fatalf("Destination prefix must start with '/': %q", destPrefix)
}
config := dropbox.Config{Token: cfg.DropboxToken, Verbose: cfg.Verbose}
dbx := files.New(config)
remoteFilesPresent, err := getRemoteFileList(dbx, rconfig.Args()[2])
if err != nil {
log.Fatalf("Error while doing getRemoteFileList request: %s", err)
}
localFilesPresent, err := getLocalFileList(rconfig.Args()[1])
if err != nil {
log.Fatalf("Error while reading local file list: %s", err)
}
for f, lmt := range localFilesPresent {
dest := path.Join(destPrefix, strings.Replace(f, sourcePath, "", -1))
if rmt, ok := remoteFilesPresent[dest]; ok && !cfg.ForceOverwrite && !rmt.Before(lmt) {
if cfg.Verbose {
log.Printf("Remote %q already there, not uploading.", dest)
}
continue
}
log.Printf("Uploading %q to %q", f, dest)
if err := uploadFile(dbx, f, dest); err != nil {
log.Fatalf("Error while uploading %q: %s", dest, err)
}
}
}
func uploadFile(dbx files.Client, sourceFile, destinationFile string) error {
info, err := os.Stat(sourceFile)
if err != nil {
return err
}
content, err := os.Open(sourceFile)
if err != nil {
return err
}
defer content.Close()
modTime := time.Unix(info.ModTime().Unix(), 0)
writeMode := &files.WriteMode{}
writeMode.Tag = files.WriteModeOverwrite
_, err = dbx.Upload(&files.CommitInfo{
Path: destinationFile,
Mode: writeMode,
ClientModified: modTime,
}, content)
return err
}
func getRemoteFileList(dbx files.Client, prefix string) (map[string]time.Time, error) {
remoteFilesPresent := map[string]time.Time{}
lfr, err := dbx.ListFolder(&files.ListFolderArg{
Path: prefix,
Recursive: true,
})
if err != nil {
if strings.Contains(err.Error(), "path/not_found/") {
return remoteFilesPresent, nil
}
return remoteFilesPresent, err
}
for {
for _, f := range lfr.Entries {
if fm, ok := f.(*files.FileMetadata); ok {
remoteFilesPresent[fm.PathDisplay] = fm.ClientModified
}
}
if !lfr.HasMore {
break
}
lfr, err = dbx.ListFolderContinue(&files.ListFolderContinueArg{Cursor: lfr.Cursor})
if err != nil {
return remoteFilesPresent, err
}
}
return remoteFilesPresent, nil
}
func getLocalFileList(prefix string) (map[string]time.Time, error) {
fileList := map[string]time.Time{}
return fileList, filepath.Walk(prefix, func(path string, f os.FileInfo, err error) error {
if !f.IsDir() {
fileList[path] = f.ModTime()
}
return nil
})
}