-u also accepts an URL from where to fetch the ssh-keys

cache.go

@@ -1,6 +1,7 @@
 package sshvault
 
 import (
+	"crypto/md5"
 	"fmt"
 	"io/ioutil"
 	"log"
@@ -25,12 +26,24 @@ func Cache() *cache {
 
 // Get return ssh-key
 func (c *cache) Get(u string, k int) (string, error) {
-	uKey := fmt.Sprintf("%s/%s.key-%d", c.dir, u, k)
+	var (
+		uKey string
+		hash string
+	)
+	if !isURL.MatchString(u) {
+		uKey = fmt.Sprintf("%s/%s.key-%d", c.dir, u, k)
+	} else {
+		hash = fmt.Sprintf("%x", md5.Sum([]byte(u)))
+		uKey = fmt.Sprintf("%s/%s.key-%d", c.dir, hash, k)
+	}
 	if !c.IsFile(uKey) {
 		keys, err := GetKey(u)
 		if err != nil {
 			return "", err
 		}
+		if isURL.MatchString(u) {
+			u = hash
+		}
 		for k, v := range keys {
 			err = ioutil.WriteFile(fmt.Sprintf("%s/%s.key-%d", c.dir, u, k+1),
 				[]byte(v),

cmd/ssh-vault/main.go

@@ -20,7 +20,7 @@ func exit1(err error) {
 func main() {
 	var (
 		k       = flag.String("k", "~/.ssh/id_rsa.pub", "public `ssh key or index` when using option -u")
-		u       = flag.String("u", "", "GitHub `username`, optional [-k N] where N is the key index to use")
+		u       = flag.String("u", "", "GitHub `username or URL`, optional [-k N] where N is the key index to use")
 		f       = flag.Bool("f", false, "Print ssh key `fingerprint`")
 		options = []string{"create", "edit", "view"}
 		v       = flag.Bool("v", false, fmt.Sprintf("Print version: %s", version))

getkey.go

@@ -14,12 +14,13 @@ const GITHUB = "https://github.com"
 
 // GetKey fetches ssh-key from url
 func GetKey(u string) ([]string, error) {
+	url := u
+	if !isURL.MatchString(u) {
+		url = fmt.Sprintf("%s/%s.keys", GITHUB, u)
+	}
 	client := &http.Client{}
 	// create a new request
-	req, _ := http.NewRequest("GET", fmt.Sprintf("%s/%s.keys",
+	req, _ := http.NewRequest("GET", url, nil)
-		GITHUB,
-		u),
-		nil)
 	req.Header.Set("User-Agent", "ssh-vault")
 	res, err := client.Do(req)
 	if err != nil {

vault.go

@@ -9,6 +9,7 @@ import (
 	"encoding/pem"
 	"fmt"
 	"os/exec"
+	"regexp"
 	"strconv"
 	"strings"
 )
@@ -22,6 +23,8 @@ type vault struct {
 	password    []byte
 }
 
+var isURL = regexp.MustCompile(`^https?://`)
+
 // New initialize vault parameters
 func New(k, u, o, v string) (*vault, error) {
 	var (