cicd/drone-ssh
1
0
Fork 0
mirror of https://github.com/appleboy/drone-ssh.git synced 2025-05-09 18:23:21 +08:00
Code Issues Packages Projects Releases 1 Wiki Activity

support password flag.

Browse Source 
Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu 2016-12-27 16:23:52 +08:00
parent 4c85fe6ea9
commit d55423fa68
3 changed files with 37 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -24,7 +24,7 @@ Build the docker image with the following commands:
``` ```
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -a -tags netgo CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -a -tags netgo
docker build --rm=true -t plugins/ssh . docker build -t plugins/ssh .
``` ```
Please note incorrectly building the image for the correct x64 linux and with Please note incorrectly building the image for the correct x64 linux and with

8
main.go
View File

@ -6,6 +6,7 @@ import (
"github.com/Sirupsen/logrus" "github.com/Sirupsen/logrus"
"github.com/joho/godotenv" "github.com/joho/godotenv"
_ "github.com/joho/godotenv/autoload"
"github.com/urfave/cli" "github.com/urfave/cli"
) )
@ -29,6 +30,11 @@ func main() {
EnvVar: "PLUGIN_USER,SSH_USER", EnvVar: "PLUGIN_USER,SSH_USER",
Value: "root", Value: "root",
}, },
cli.StringFlag{
Name: "password",
Usage: "user password",
EnvVar: "PLUGIN_PASSWORD,SSH_PASSWORD",
},
cli.StringSliceFlag{ cli.StringSliceFlag{
Name: "host", Name: "host",
Usage: "connect to host", Usage: "connect to host",
@ -58,6 +64,7 @@ func main() {
cli.StringFlag{ cli.StringFlag{
Name: "env-file", Name: "env-file",
Usage: "source env file", Usage: "source env file",
EnvVar: "ENV_FILE",
}, },
} }
@ -75,6 +82,7 @@ func run(c *cli.Context) error {
Config: Config{ Config: Config{
Key: c.String("ssh-key"), Key: c.String("ssh-key"),
User: c.String("user"), User: c.String("user"),
Password: c.String("password"),
Host: c.StringSlice("host"), Host: c.StringSlice("host"),
Port: c.Int("port"), Port: c.Int("port"),
Sleep: c.Int("sleep"), Sleep: c.Int("sleep"),

35
plugin.go
View File

@ -13,13 +13,14 @@ import (
type ( type (
Config struct { Config struct {
Key string `json:"key"` Key string
User string `json:"user"` User string
Host []string `json:"host"` Password string
Port int `json:"port"` Host []string
Sleep int `json:"sleep"` Port int
Timeout time.Duration `json:"timeout"` Sleep int
Script []string `json:"script"` Timeout time.Duration
Script []string
} }
Plugin struct { Plugin struct {
@ -28,8 +29,8 @@ type (
) )
func (p Plugin) Exec() error { func (p Plugin) Exec() error {
if p.Config.Key == "" { if p.Config.Key == "" && p.Config.Password == "" {
return fmt.Errorf("Error: Can't connect without a private SSH key.") return fmt.Errorf("Error: Can't connect without a private SSH key or password.")
} }
for i, host := range p.Config.Host { for i, host := range p.Config.Host {
@ -38,18 +39,28 @@ func (p Plugin) Exec() error {
strconv.Itoa(p.Config.Port), strconv.Itoa(p.Config.Port),
) )
// auths holds the detected ssh auth methods
auths := []ssh.AuthMethod{}
if p.Config.Key != "" {
signer, err := ssh.ParsePrivateKey([]byte(p.Config.Key)) signer, err := ssh.ParsePrivateKey([]byte(p.Config.Key))
if err != nil { if err != nil {
return fmt.Errorf("Error: Failed to parse private key. %s", err) return fmt.Errorf("Error: Failed to parse private key. %s", err)
} }
auths = append(auths, ssh.PublicKeys(signer))
}
// figure out what auths are requested, what is supported
if p.Config.Password != "" {
auths = append(auths, ssh.Password(p.Config.Password))
}
config := &ssh.ClientConfig{ config := &ssh.ClientConfig{
Timeout: p.Config.Timeout, Timeout: p.Config.Timeout,
User: p.Config.User, User: p.Config.User,
Auth: []ssh.AuthMethod{ Auth: auths,
ssh.PublicKeys(signer),
},
} }
fmt.Printf("+ ssh %s@%s -p %d\n", p.Config.User, addr, p.Config.Port) fmt.Printf("+ ssh %s@%s -p %d\n", p.Config.User, addr, p.Config.Port)