Merge pull request #24 from stephanwilliams/import-options

solher · web-flow · commit 83b1466b1755 · 2017-07-24T12:10:33.000+02:00
Add more collection import options
diff --git a/requests/import.go b/requests/import.go
@@ -1,15 +1,50 @@
 package requests
 
-import "fmt"
-
 // ImportCollection imports data to a collection
 type ImportCollection struct {
 	CollectionName string
 	Data           []byte
+	FromPrefix     string
+	ToPrefix       string
+	Overwrite      bool
+	WaitForSync    bool
+	OnDuplicate    string
+	Complete       bool
+	Details        bool
 }
 
 func (c *ImportCollection) Path() string {
-	return fmt.Sprintf("/_api/import/?type=auto&collection=%s", c.CollectionName)
+	path := "/_api/import/?type=auto&collection=" + c.CollectionName
+
+	if c.FromPrefix != "" {
+		path += "&fromPrefix=" + c.FromPrefix
+	}
+
+	if c.ToPrefix != "" {
+		path += "&toPrefix=" + c.ToPrefix
+	}
+
+	if c.Overwrite {
+		path += "&overwrite=yes"
+	}
+
+	if c.WaitForSync {
+		path += "&waitForSync=yes"
+	}
+
+	if c.OnDuplicate != "" {
+		path += "&onDuplicate=" + c.OnDuplicate
+	}
+
+	if c.Complete {
+		path += "&complete=yes"
+	}
+
+	if c.Details {
+		path += "&details=yes"
+	}
+
+	return path
 }
 
 func (c *ImportCollection) Method() string {
