95 lines
1.9 KiB
Go
95 lines
1.9 KiB
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"
|
|
}
|