Merge pull request #62 from drone-plugins/storage-class

Integrate setting for storage class to use

main.go

@@ -107,6 +107,11 @@ func main() {
 			EnvVar: "PLUGIN_CACHE_CONTROL",
 			Value:  &StringMapFlag{},
 		},
+		cli.StringFlag{
+			Name:   "storage-class",
+			Usage:  "set storage class to choose the best backend",
+			EnvVar: "PLUGIN_STORAGE_CLASS",
+		},
 		cli.StringFlag{
 			Name:  "env-file",
 			Usage: "source env file",
@@ -138,6 +143,7 @@ func run(c *cli.Context) error {
 		ContentType:     c.Generic("content-type").(*StringMapFlag).Get(),
 		ContentEncoding: c.Generic("content-encoding").(*StringMapFlag).Get(),
 		CacheControl:    c.Generic("cache-control").(*StringMapFlag).Get(),
+		StorageClass:    c.String("storage-class"),
 		PathStyle:       c.Bool("path-style"),
 		DryRun:          c.Bool("dry-run"),
 	}

plugin.go

@@ -57,6 +57,9 @@ type Plugin struct {
 	// Sets the Cache-Control header on each uploaded object based on a extension map
 	CacheControl map[string]string
 
+	// Sets the storage class, affects the storage backend costs
+	StorageClass string
+
 	// Copies the files from the specified directory.
 	// Regexp matching will apply to match multiple
 	// files
@@ -196,6 +199,10 @@ func (p *Plugin) Exec() error {
 			putObjectInput.ServerSideEncryption = aws.String(p.Encryption)
 		}
 
+		if p.StorageClass != "" {
+			putObjectInput.StorageClass = &(p.StorageClass)
+		}
+
 		_, err = client.PutObject(putObjectInput)
 
 		if err != nil {