package pm

import (
	"bytes"
	"io/ioutil"
	"strings"

	"golang.org/x/crypto/openpgp/armor"
)

const (
	pgpMessageType      string = "PGP MESSAGE"
	pgpPublicBlockType  string = "PGP PUBLIC KEY BLOCK"
	pgpPrivateBlockType string = "PGP PRIVATE KEY BLOCK"
)

// ArmorWithType make bytes input to armor format
func ArmorWithType(input []byte, armorType string) (string, error) {
	var b bytes.Buffer
	w, err := armor.Encode(&b, armorType, nil)
	if err != nil {
		return "", err
	}
	_, err = w.Write(input)
	if err != nil {
		return "", err
	}
	w.Close()
	return b.String(), nil
}

// UnArmor an armored key to bytes key
func UnArmor(input string) ([]byte, error) {
	b, err := unArmor(input)
	if err != nil {
		return nil, err
	}
	return ioutil.ReadAll(b.Body)
}

func unArmor(input string) (*armor.Block, error) {
	io := strings.NewReader(input)
	b, err := armor.Decode(io)
	if err != nil {
		return nil, err
	}
	return b, nil

}