mcchunkie/plugins/pgp.go

package plugins

import (
	"encoding/hex"
	"fmt"
	"net/http"
	"net/url"
	"regexp"
	"strings"

	"github.com/ProtonMail/go-crypto/openpgp"
	"github.com/matrix-org/gomatrix"
)

// PGP is our plugin type
type PGP struct {
}

// SetStore is the setup function for a plugin
func (p *PGP) SetStore(s PluginStore) {
}

// Descr describes this plugin
func (p *PGP) Descr() string {
	return "Queries keys.openpgp.org"
}

// Re is what our pgp request matches
func (p *PGP) Re() string {
	return `(?i)^pgp: (.+@.+\..+|[a-f0-9]+)$`
}

// Match checks for "pgp: " messages
func (p *PGP) Match(_, msg string) bool {
	re := regexp.MustCompile(p.Re())
	return re.MatchString(msg)
}

func (p *PGP) fix(msg string) string {
	re := regexp.MustCompile(p.Re())
	return strings.ToUpper(re.ReplaceAllString(msg, "$1"))
}

func (p *PGP) Process(from, post string) string {
	search := p.fix(post)
	searchURL := "https://keys.openpgp.org//vks/v1/by-fingerprint/%s"

	if strings.ContainsAny(search, "@") {
		searchURL = "https://keys.openpgp.org//vks/v1/by-email/%s"
	}

	escSearch, err := url.Parse(search)
	if err != nil {
		return err.Error()
	}

	u := fmt.Sprintf(searchURL, escSearch)

	resp, err := http.Get(u)
	if err != nil {
		return err.Error()
	}

	defer resp.Body.Close()

	kr, err := openpgp.ReadArmoredKeyRing(resp.Body)
	if err != nil {
		return err.Error()
	}

	var ids []string
	var fps []string
	for _, entity := range kr {
		for _, i := range entity.Identities {
			ids = append(ids, fmt.Sprintf("- %q", i.Name))
		}
		fps = append(fps, fmt.Sprintf("**Fingerprint**: %s",
			hex.EncodeToString(entity.PrimaryKey.Fingerprint[:])))
	}

	return fmt.Sprintf("%s\n\n%s",
		strings.Join(ids, "\n"),
		strings.Join(fps, "\n"))
}

// RespondText to looking up of PGP info
func (p *PGP) RespondText(c *gomatrix.Client, ev *gomatrix.Event, _, post string) error {
	return SendMD(c, ev.RoomID, p.Process(ev.Sender, post))
}

// Name PGP!
func (p *PGP) Name() string {
	return "PGP"
}