From fa41183a53fc185a8a78df9b0b4ce5109e8627c1 Mon Sep 17 00:00:00 2001
From: Alexander Neumann <alexander@bumpern.de>
Date: Mon, 15 May 2017 23:37:02 +0200
Subject: [PATCH] s3: Add s3.layout option and layout auto detection

---
 src/restic/backend/s3/config.go |  6 +++
 src/restic/backend/s3/s3.go     | 87 ++++++++++++++++++++++++++++++++-
 2 files changed, 92 insertions(+), 1 deletion(-)

diff --git a/src/restic/backend/s3/config.go b/src/restic/backend/s3/config.go
index 2df02b58c..6cf7db9c1 100644
--- a/src/restic/backend/s3/config.go
+++ b/src/restic/backend/s3/config.go
@@ -6,6 +6,7 @@ import (
 	"strings"
 
 	"restic/errors"
+	"restic/options"
 )
 
 // Config contains all configuration necessary to connect to an s3 compatible
@@ -16,6 +17,11 @@ type Config struct {
 	KeyID, Secret string
 	Bucket        string
 	Prefix        string
+	Layout        string `option:"layout" help:"use this backend layout (default: auto-detect)"`
+}
+
+func init() {
+	options.Register("s3", Config{})
 }
 
 const defaultPrefix = "restic"
diff --git a/src/restic/backend/s3/s3.go b/src/restic/backend/s3/s3.go
index 241bfe323..3b6a60b54 100644
--- a/src/restic/backend/s3/s3.go
+++ b/src/restic/backend/s3/s3.go
@@ -8,6 +8,7 @@ import (
 	"restic"
 	"strings"
 	"sync"
+	"time"
 
 	"restic/backend"
 	"restic/errors"
@@ -30,6 +31,8 @@ type s3 struct {
 	backend.Layout
 }
 
+const defaultLayout = "s3legacy"
+
 // Open opens the S3 backend at bucket and region. The bucket is created if it
 // does not exist yet.
 func Open(cfg Config) (restic.Backend, error) {
@@ -45,11 +48,17 @@ func Open(cfg Config) (restic.Backend, error) {
 		bucketname:   cfg.Bucket,
 		prefix:       cfg.Prefix,
 		cacheObjSize: make(map[string]int64),
-		Layout:       &backend.S3LegacyLayout{Path: cfg.Prefix, Join: path.Join},
 	}
 
 	client.SetCustomTransport(backend.Transport())
 
+	l, err := backend.ParseLayout(be, cfg.Layout, defaultLayout, cfg.Prefix)
+	if err != nil {
+		return nil, err
+	}
+
+	be.Layout = l
+
 	be.createConnections()
 
 	found, err := client.BucketExists(cfg.Bucket)
@@ -76,6 +85,77 @@ func (be *s3) createConnections() {
 	}
 }
 
+// IsNotExist returns true if the error is caused by a not existing file.
+func (be *s3) IsNotExist(err error) bool {
+	debug.Log("IsNotExist(%T, %#v)", err, err)
+	if os.IsNotExist(err) {
+		return true
+	}
+
+	return false
+}
+
+// Join combines path components with slashes.
+func (be *s3) Join(p ...string) string {
+	return path.Join(p...)
+}
+
+type fileInfo struct {
+	name    string
+	size    int64
+	mode    os.FileMode
+	modTime time.Time
+	isDir   bool
+}
+
+func (fi fileInfo) Name() string       { return fi.name }    // base name of the file
+func (fi fileInfo) Size() int64        { return fi.size }    // length in bytes for regular files; system-dependent for others
+func (fi fileInfo) Mode() os.FileMode  { return fi.mode }    // file mode bits
+func (fi fileInfo) ModTime() time.Time { return fi.modTime } // modification time
+func (fi fileInfo) IsDir() bool        { return fi.isDir }   // abbreviation for Mode().IsDir()
+func (fi fileInfo) Sys() interface{}   { return nil }        // underlying data source (can return nil)
+
+// ReadDir returns the entries for a directory.
+func (be *s3) ReadDir(dir string) (list []os.FileInfo, err error) {
+	debug.Log("ReadDir(%v)", dir)
+
+	// make sure dir ends with a slash
+	if dir[len(dir)-1] != '/' {
+		dir += "/"
+	}
+
+	done := make(chan struct{})
+	defer close(done)
+
+	for obj := range be.client.ListObjects(be.bucketname, dir, false, done) {
+		if obj.Key == "" {
+			continue
+		}
+
+		name := strings.TrimPrefix(obj.Key, dir)
+		if name == "" {
+			return nil, errors.Errorf("invalid key name %v, removing prefix %v yielded empty string", obj.Key, dir)
+		}
+		entry := fileInfo{
+			name:    name,
+			size:    obj.Size,
+			modTime: obj.LastModified,
+		}
+
+		if name[len(name)-1] == '/' {
+			entry.isDir = true
+			entry.mode = os.ModeDir | 0755
+			entry.name = name[:len(name)-1]
+		} else {
+			entry.mode = 0644
+		}
+
+		list = append(list, entry)
+	}
+
+	return list, nil
+}
+
 // Location returns this backend's location (the bucket name).
 func (be *s3) Location() string {
 	return be.bucketname
@@ -290,6 +370,11 @@ func (be *s3) List(t restic.FileType, done <-chan struct{}) <-chan string {
 
 	prefix := be.Dirname(restic.Handle{Type: t})
 
+	// make sure prefix ends with a slash
+	if prefix[len(prefix)-1] != '/' {
+		prefix += "/"
+	}
+
 	listresp := be.client.ListObjects(be.bucketname, prefix, true, done)
 
 	go func() {