Support run and destroy

This commit is contained in:
Quentin 2021-09-23 10:22:19 +02:00
parent a8b9a43c02
commit e03b09b68f
1 changed files with 64 additions and 47 deletions

111
main.go
View File

@ -3,15 +3,17 @@ package main
import (
"bytes"
"io"
"io/ioutil"
"net"
"os"
"fmt"
"errors"
"time"
"golang.org/x/crypto/ssh"
"golang.org/x/crypto/ssh/agent"
"github.com/pkg/sftp"
"github.com/scaleway/scaleway-sdk-go/scw"
"github.com/scaleway/scaleway-sdk-go/api/instance/v1"
)
@ -256,7 +258,42 @@ func (dt *destroyer) onInstance(zone, machine, image, name string) error {
type runner struct { action }
func (r *runner) onInstance(zone, machine, image, name string) error {
func (r *runner) connect(zone, name string) (*ssh.Client, error) {
// Connect to the remote
z, err := scw.ParseZone(zone)
if err != nil {
return nil, err
}
targetServer, err := r.getInstanceByName(z, name)
if err != nil {
return nil, err
}
if targetServer.PublicIP == nil {
return nil, errors.New("run failed: this instance has no public ip.")
}
ip := targetServer.PublicIP.Address
socket := os.Getenv("SSH_AUTH_SOCK")
conn, err := net.Dial("unix", socket)
if err != nil {
return nil, err
}
agentClient := agent.NewClient(conn)
config := &ssh.ClientConfig{
User: "root",
Auth: []ssh.AuthMethod{
ssh.PublicKeysCallback(agentClient.Signers),
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
}
return ssh.Dial("tcp", ip.String()+":22", config)
}
func (r *runner) send(sshClient *ssh.Client) error {
// Source script
if len(os.Args) < 3 {
return errors.New("Missing script to run on the command line")
@ -268,47 +305,6 @@ func (r *runner) onInstance(zone, machine, image, name string) error {
}
defer src.Close()
// Connect to the remote
z, err := scw.ParseZone(zone)
if err != nil {
return err
}
targetServer, err := r.getInstanceByName(z, name)
if err != nil {
return err
}
if targetServer.PublicIP == nil {
return errors.New("run failed: this instance has no public ip.")
}
ip := targetServer.PublicIP.Address
var hostKey ssh.PublicKey
key, err := ioutil.ReadFile(os.Getenv("HOME") + "/.ssh/id_rsa")
if err != nil {
return err
}
signer, err := ssh.ParsePrivateKey(key)
if err != nil {
return err
}
config := &ssh.ClientConfig{
User: "root",
Auth: []ssh.AuthMethod{
ssh.PublicKeys(signer),
},
HostKeyCallback: ssh.FixedHostKey(hostKey),
}
sshClient, err := ssh.Dial("tcp", ip.String()+":22", config)
if err != nil {
return err
}
defer sshClient.Close()
// Send our script to the destination
sftpClient, err := sftp.NewClient(sshClient)
if err != nil {
@ -328,10 +324,10 @@ func (r *runner) onInstance(zone, machine, image, name string) error {
}
_, err = io.Copy(dst, src)
if err != nil {
return err
}
return err
}
func (r *runner) exec(sshClient *ssh.Client) error {
// Run the script
session, err := sshClient.NewSession()
if err != nil {
@ -345,7 +341,7 @@ func (r *runner) onInstance(zone, machine, image, name string) error {
err = session.Run("/tmp/nuage")
if os.Getenv("VERBOSE") != "" {
fmt.Fprintf(os.Stdout, "logs for %v\n", name)
fmt.Fprintf(os.Stdout, "logs\n")
io.Copy(os.Stdout, &sshStdout)
io.Copy(os.Stderr, &sshStderr)
}
@ -356,6 +352,27 @@ func (r *runner) onInstance(zone, machine, image, name string) error {
return nil
}
func (r *runner) onInstance(zone, machine, image, name string) error {
sshClient, err := r.connect(zone, name)
if err != nil {
return err
}
defer sshClient.Close()
err = r.send(sshClient)
if err != nil {
return err
}
err = r.exec(sshClient)
if err != nil {
return err
}
fmt.Fprintf(os.Stdout, "✅ Successfully ran the script on %v (zone %v)\n", name, zone)
return nil
}
/**
* Commands