scp.go

package main

import (
	"fmt"
	"log"
	"os"

	scp "github.com/jhand2/go-scp"
	"github.com/jhand2/go-scp/auth"
	"golang.org/x/crypto/ssh"
)

type ScpCopier struct {
	client scp.Client
}

func (s *ScpCopier) Init(username string, privkey string, server string) {
	// TODO: Allow to use specific privkey path or ssh agent.
	// Get password for privkey if ssh-agent fails
	//fmt.Print("Enter Password: ")
	//pw, err := terminal.ReadPassword(int(syscall.Stdin))
	//if err != nil {
	//log.Fatal("Could not read password. Error: ", err)
	//}

	//fmt.Print("\n")

	//conf, err := auth.PrivateKeyWithPassphrase(username, pw, privkey, ssh.InsecureIgnoreHostKey())
	//if err != nil {
	//log.Fatal("Could not use private key. Error: ", err)
	//}

	conf, err := auth.SshAgent(username, ssh.InsecureIgnoreHostKey())
	if err != nil {
		log.Fatal("Could not use ssh agent. Error: ", err)
	}

	s.client = scp.NewClient(server, &conf)
}

func (s *ScpCopier) copy_file(src string, dst string) error {
	fmt.Printf("Copying %s to %s\n", src, dst)
	err := s.client.Connect()
	if err != nil {
		return err
	}

	f, err := os.Open(src)
	if err != nil {
		return err
	}

	defer s.client.Close()
	defer f.Close()

	// TODO: Change file permissions?
	err = s.client.CopyFile(f, dst, "0655")
	if err != nil {
		return err
	}

	return nil
}