package sshvault

import (
	"bufio"
	"bytes"
	"fmt"
	"io"
	"net/http"
	"net/textproto"
	"strings"
)

// SSHKEYS_ONLINE create new pair of keys online
const SSHKEYS_ONLINE = "https://ssh-keys.online/new"

// Schlosser interface
type Schlosser interface {
	GetKey(string) ([]string, error)
}

// Locksmith implements Schlosser
type Locksmith struct {
	URL string
}

// GetKey fetches ssh-key from url
func (l Locksmith) GetKey(u string) ([]string, error) {
	url := u
	if !isURL.MatchString(u) {
		switch u {
		case "new":
			url = SSHKEYS_ONLINE
		default:
			url = fmt.Sprintf("%s/%s.keys", l.URL, u)
		}
	}
	client := &http.Client{}
	// create a new request
	req, _ := http.NewRequest("GET", url, nil)
	req.Header.Set("User-Agent", "ssh-vault")
	res, err := client.Do(req)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()
	reader := bufio.NewReader(res.Body)
	tp := textproto.NewReader(reader)
	keys := []string{}
	rsa := bytes.Buffer{}
	isRSA := false
	for {
		if line, err := tp.ReadLine(); err != nil {
			if err == io.EOF {
				if len(keys) == 0 {
					return nil, fmt.Errorf("key %q not found", u)
				}
				return keys, nil
			}
			return nil, err
		} else if strings.HasPrefix(line, "ssh-rsa") {
			keys = append(keys, line)
		} else if strings.HasPrefix(line, "-----BEGIN RSA PRIVATE KEY-----") {
			isRSA = true
			if _, err := rsa.WriteString(line + "\n"); err != nil {
				return nil, err
			}
		} else if strings.HasPrefix(line, "-----END RSA PRIVATE KEY-----") {
			if _, err := rsa.WriteString(line); err != nil {
				return nil, err
			}
			return []string{rsa.String()}, nil
		} else if isRSA {
			if _, err := rsa.WriteString(line + "\n"); err != nil {
				return nil, err
			}
		}
	}
}