Os login (#50)

Add OS Login support + change key storage format

key/key.go

@@ -0,0 +1,114 @@
+package key
+
+import (
+	"bytes"
+	"crypto/ed25519"
+	"crypto/rand"
+	"encoding/gob"
+	"encoding/pem"
+	"io/ioutil"
+	"os/user"
+	"path/filepath"
+	"strings"
+
+	"github.com/apex/log"
+	"github.com/melbahja/goph"
+	"github.com/mikesmitty/edkey"
+	"golang.org/x/crypto/ssh"
+)
+
+const Comment = "speedrun"
+
+type Key struct {
+	User    string
+	Comment string
+	Key     []byte
+}
+
+func New() (*Key, error) {
+	key := &Key{}
+	log.Info("Generating new private key")
+	_, rawKey, err := ed25519.GenerateKey(rand.Reader)
+	if err != nil {
+		return nil, err
+	}
+
+	pemKey := &pem.Block{
+		Type:  "OPENSSH PRIVATE KEY",
+		Bytes: edkey.MarshalED25519PrivateKey(rawKey),
+	}
+	privateKey := pem.EncodeToMemory(pemKey)
+	key.Key = privateKey
+
+	user, err := user.Current()
+	if err != nil {
+		return nil, err
+	}
+	key.User = user.Username
+	key.Comment = Comment
+	return key, nil
+}
+
+func Read(path string) (*Key, error) {
+	cleanPath := filepath.Clean(path)
+	absPath, err := filepath.Abs(cleanPath)
+	if err != nil {
+		return nil, err
+	}
+
+	file, err := ioutil.ReadFile(absPath)
+	if err != nil {
+		return nil, err
+	}
+
+	key := &Key{}
+	buf := bytes.NewBuffer(file)
+	enc := gob.NewDecoder(buf)
+
+	err = enc.Decode(key)
+	if err != nil {
+		return nil, err
+	}
+
+	return key, nil
+
+}
+
+func (k *Key) Write(path string) error {
+	var buf bytes.Buffer
+	enc := gob.NewEncoder(&buf)
+	err := enc.Encode(k)
+	if err != nil {
+		return err
+	}
+
+	log.Debugf("Writing priviate key to %s", path)
+	err = ioutil.WriteFile(path, buf.Bytes(), 0600)
+	if err != nil {
+		return err
+	}
+	return nil
+
+}
+
+func (k *Key) MarshalAuthorizedKey() (string, error) {
+	privKey, err := ssh.ParsePrivateKey(k.Key)
+	if err != nil {
+		return "", err
+	}
+
+	authorizedKey := ssh.MarshalAuthorizedKey(privKey.PublicKey())
+	trimmedKey := strings.TrimSuffix(string(authorizedKey), "\n")
+	return trimmedKey, nil
+}
+
+func (k *Key) GetAuth() (goph.Auth, error) {
+	privKey, err := ssh.ParsePrivateKey(k.Key)
+	if err != nil {
+		return nil, err
+	}
+
+	return goph.Auth{
+		ssh.PublicKeys(privKey),
+	}, nil
+}