Merge pull request #1924 from josmo/stash-rsa-string

Stash rsa string

drone/server.go

@@ -261,6 +261,11 @@ var serverCmd = cli.Command{
 			Name:   "stash-consumer-rsa",
 			Usage:  "stash oauth1 private key file",
 		},
+		cli.StringFlag{
+			EnvVar: "DRONE_STASH_CONSUMER_RSA_STRING",
+			Name:   "stash-consumer-rsa-string",
+			Usage:  "stash oauth1 private key string",
+		},
 		cli.StringFlag{
 			EnvVar: "DRONE_STASH_GIT_USERNAME",
 			Name:   "stash-git-username",

remote/bitbucketserver/bitbucketserver.go

@@ -32,6 +32,7 @@ type Opts struct {
 	Password          string // Git machine account password.
 	ConsumerKey       string // Oauth1 consumer key.
 	ConsumerRSA       string // Oauth1 consumer key file.
+	ConsumerRSAString string
 	SkipVerify        bool // Skip ssl verification.
 }
 
@@ -60,19 +61,29 @@ func New(opts Opts) (remote.Remote, error) {
 		return nil, fmt.Errorf("Must have a git machine account password")
 	case opts.ConsumerKey == "":
 		return nil, fmt.Errorf("Must have a oauth1 consumer key")
-	case opts.ConsumerRSA == "":
-		return nil, fmt.Errorf("Must have a oauth1 consumer key file")
 	}
 
-	keyFile, err := ioutil.ReadFile(opts.ConsumerRSA)
+	if opts.ConsumerRSA == "" && opts.ConsumerRSAString == "" {
+		return nil, fmt.Errorf("must have CONSUMER_RSA_KEY set to the path of a oauth1 consumer key file or CONSUMER_RSA_KEY_STRING set to the value of a oauth1 consumer key")
+	}
+
+	var keyFileBytes []byte
+	if opts.ConsumerRSA != "" {
+		var err error
+		keyFileBytes, err = ioutil.ReadFile(opts.ConsumerRSA)
 		if err != nil {
 			return nil, err
 		}
-	block, _ := pem.Decode(keyFile)
+	} else {
+		keyFileBytes = []byte(opts.ConsumerRSAString)
+	}
+
+	block, _ := pem.Decode(keyFileBytes)
 	PrivateKey, err := x509.ParsePKCS1PrivateKey(block.Bytes)
 	if err != nil {
 		return nil, err
 	}
+
 	config.Consumer = CreateConsumer(opts.URL, opts.ConsumerKey, PrivateKey)
 	return config, nil
 }
@@ -154,7 +165,7 @@ func (c *Config) File(u *model.User, r *model.Repo, b *model.Build, f string) ([
 }
 
 // Status is not supported by the bitbucketserver driver.
-func (c *Config) Status(u *model.User,r *model.Repo,b *model.Build,link string) error {
+func (c *Config) Status(u *model.User, r *model.Repo, b *model.Build, link string) error {
 	status := internal.BuildStatus{
 		State: convertStatus(b.Status),
 		Desc:  convertDesc(b.Status),

router/middleware/remote.go

@@ -71,6 +71,7 @@ func setupStash(c *cli.Context) (remote.Remote, error) {
 		Password:          c.String("stash-git-password"),
 		ConsumerKey:       c.String("stash-consumer-key"),
 		ConsumerRSA:       c.String("stash-consumer-rsa"),
+		ConsumerRSAString: c.String("stash-consumer-rsa-string"),
 		SkipVerify:        c.Bool("stash-skip-verify"),
 	})
 }