git.olowe.co

Blame

Date:: Mon Mar 18 03:10:22 2024 UTC
Message:: apub: handle collection recipients mroe reliably When we receive activity that has a collection, we see if any of the other actor recipients have the collection set as their followers. Then we can encode it with plus addressing. If we can't do that we drop it. Not ideal, but it covers most use cases. Next up we can encode arbitrary collections like "some/path/to/thing@example.com".
Actions:: History | Blob | Raw File
  1 1d5ddf5d 2024-02-20 o package apub
  2 1d5ddf5d 2024-02-20 o 
  3 1d5ddf5d 2024-02-20 o import (
  4 1d5ddf5d 2024-02-20 o 	"bytes"
  5 1d5ddf5d 2024-02-20 o 	"crypto/rsa"
  6 1d5ddf5d 2024-02-20 o 	"encoding/json"
  7 d3dfb672 2024-03-18 o 	"errors"
  8 1d5ddf5d 2024-02-20 o 	"fmt"
  9 71191436 2024-02-28 o 	"io"
 10 1d5ddf5d 2024-02-20 o 	"net/http"
 11 d3dfb672 2024-03-18 o 	"net/url"
 12 71191436 2024-02-28 o 	"os"
 13 d3dfb672 2024-03-18 o 	"path"
 14 1d5ddf5d 2024-02-20 o 	"strings"
 15 1d5ddf5d 2024-02-20 o )
 16 1d5ddf5d 2024-02-20 o 
 17 d3dfb672 2024-03-18 o var DefaultClient Client = Client{Client: http.DefaultClient}
 18 1d5ddf5d 2024-02-20 o 
 19 1d5ddf5d 2024-02-20 o func Lookup(id string) (*Activity, error) {
 20 d3dfb672 2024-03-18 o 	return DefaultClient.Lookup(id)
 21 1d5ddf5d 2024-02-20 o }
 22 1d5ddf5d 2024-02-20 o 
 23 1d5ddf5d 2024-02-20 o func LookupActor(id string) (*Actor, error) {
 24 d3dfb672 2024-03-18 o 	return DefaultClient.LookupActor(id)
 25 1d5ddf5d 2024-02-20 o }
 26 1d5ddf5d 2024-02-20 o 
 27 1d5ddf5d 2024-02-20 o type Client struct {
 28 1d5ddf5d 2024-02-20 o 	*http.Client
 29 25fed994 2024-03-12 o 	// Key is a RSA private key which will be used to sign requests.
 30 25fed994 2024-03-12 o 	Key *rsa.PrivateKey
 31 25fed994 2024-03-12 o 	// PubKeyID is a URL where the corresponding public key of Key
 32 25fed994 2024-03-12 o 	// may be accessed. This must be set if Key is also set.
 33 25fed994 2024-03-12 o 	PubKeyID string // actor.PublicKey.ID
 34 1d5ddf5d 2024-02-20 o }
 35 1d5ddf5d 2024-02-20 o 
 36 1d5ddf5d 2024-02-20 o func (c *Client) Lookup(id string) (*Activity, error) {
 37 1d5ddf5d 2024-02-20 o 	if !strings.HasPrefix(id, "http") {
 38 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("id is not a HTTP URL")
 39 1d5ddf5d 2024-02-20 o 	}
 40 1d5ddf5d 2024-02-20 o 	if c.Client == nil {
 41 1d5ddf5d 2024-02-20 o 		c.Client = http.DefaultClient
 42 1d5ddf5d 2024-02-20 o 	}
 43 1d5ddf5d 2024-02-20 o 
 44 d3dfb672 2024-03-18 o 	req, err := newRequest(http.MethodGet, id, nil, c.Key, c.PubKeyID)
 45 1d5ddf5d 2024-02-20 o 	if err != nil {
 46 d3dfb672 2024-03-18 o 		return nil, fmt.Errorf("new request: %w", err)
 47 1d5ddf5d 2024-02-20 o 	}
 48 1d5ddf5d 2024-02-20 o 	resp, err := c.Do(req)
 49 1d5ddf5d 2024-02-20 o 	if err != nil {
 50 1d5ddf5d 2024-02-20 o 		return nil, err
 51 1d5ddf5d 2024-02-20 o 	}
 52 1d5ddf5d 2024-02-20 o 	defer resp.Body.Close()
 53 1d5ddf5d 2024-02-20 o 	if resp.StatusCode == http.StatusNotFound {
 54 71191436 2024-02-28 o 		return nil, ErrNotExist
 55 1d5ddf5d 2024-02-20 o 	} else if resp.StatusCode >= 400 {
 56 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("non-ok response status %s", resp.Status)
 57 1d5ddf5d 2024-02-20 o 	}
 58 1d5ddf5d 2024-02-20 o 	return Decode(resp.Body)
 59 1d5ddf5d 2024-02-20 o }
 60 1d5ddf5d 2024-02-20 o 
 61 1d5ddf5d 2024-02-20 o func (c *Client) LookupActor(id string) (*Actor, error) {
 62 1d5ddf5d 2024-02-20 o 	activity, err := c.Lookup(id)
 63 1d5ddf5d 2024-02-20 o 	if err != nil {
 64 1d5ddf5d 2024-02-20 o 		return nil, err
 65 1d5ddf5d 2024-02-20 o 	}
 66 d3dfb672 2024-03-18 o 	switch activity.Type {
 67 d3dfb672 2024-03-18 o 	case "Application", "Group", "Organization", "Person", "Service":
 68 d3dfb672 2024-03-18 o 		return activityToActor(activity), nil
 69 d3dfb672 2024-03-18 o 	case "Collection", "OrderedCollection":
 70 d3dfb672 2024-03-18 o 		// probably followers. let caller work out what it wants to do
 71 d3dfb672 2024-03-18 o 		return activityToActor(activity), nil
 72 d3dfb672 2024-03-18 o 	}
 73 d3dfb672 2024-03-18 o 	return nil, fmt.Errorf("bad object Type %s", activity.Type)
 74 71191436 2024-02-28 o }
 75 71191436 2024-02-28 o 
 76 71191436 2024-02-28 o func activityToActor(activity *Activity) *Actor {
 77 77918b00 2024-03-07 o 	actor := &Actor{
 78 1d5ddf5d 2024-02-20 o 		AtContext: activity.AtContext,
 79 1d5ddf5d 2024-02-20 o 		ID:        activity.ID,
 80 1d5ddf5d 2024-02-20 o 		Type:      activity.Type,
 81 1d5ddf5d 2024-02-20 o 		Name:      activity.Name,
 82 1d5ddf5d 2024-02-20 o 		Username:  activity.Username,
 83 1d5ddf5d 2024-02-20 o 		Inbox:     activity.Inbox,
 84 1d5ddf5d 2024-02-20 o 		Outbox:    activity.Outbox,
 85 77918b00 2024-03-07 o 		Followers: activity.Followers,
 86 71191436 2024-02-28 o 		Published: activity.Published,
 87 71191436 2024-02-28 o 		Summary:   activity.Summary,
 88 71191436 2024-02-28 o 	}
 89 77918b00 2024-03-07 o 	if activity.PublicKey != nil {
 90 77918b00 2024-03-07 o 		actor.PublicKey = *activity.PublicKey
 91 77918b00 2024-03-07 o 	}
 92 77918b00 2024-03-07 o 	return actor
 93 1d5ddf5d 2024-02-20 o }
 94 1d5ddf5d 2024-02-20 o 
 95 1d5ddf5d 2024-02-20 o func (c *Client) Send(inbox string, activity *Activity) (*Activity, error) {
 96 1d5ddf5d 2024-02-20 o 	b, err := json.Marshal(activity)
 97 1d5ddf5d 2024-02-20 o 	if err != nil {
 98 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("encode outgoing activity: %w", err)
 99 1d5ddf5d 2024-02-20 o 	}
100 d3dfb672 2024-03-18 o 	req, err := newRequest(http.MethodPost, inbox, bytes.NewReader(b), c.Key, c.PubKeyID)
101 1d5ddf5d 2024-02-20 o 	if err != nil {
102 1d5ddf5d 2024-02-20 o 		return nil, err
103 1d5ddf5d 2024-02-20 o 	}
104 1d5ddf5d 2024-02-20 o 	resp, err := c.Do(req)
105 1d5ddf5d 2024-02-20 o 	if err != nil {
106 1d5ddf5d 2024-02-20 o 		return nil, err
107 1d5ddf5d 2024-02-20 o 	}
108 1d5ddf5d 2024-02-20 o 	switch resp.StatusCode {
109 77918b00 2024-03-07 o 	case http.StatusOK, http.StatusAccepted, http.StatusNoContent:
110 1d5ddf5d 2024-02-20 o 		return nil, nil
111 1d5ddf5d 2024-02-20 o 	case http.StatusNotFound:
112 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("no such inbox %s", inbox)
113 1d5ddf5d 2024-02-20 o 	default:
114 1d5ddf5d 2024-02-20 o 		io.Copy(os.Stderr, resp.Body)
115 1d5ddf5d 2024-02-20 o 		resp.Body.Close()
116 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("non-ok response status %s", resp.Status)
117 1d5ddf5d 2024-02-20 o 	}
118 1d5ddf5d 2024-02-20 o }
119 d3dfb672 2024-03-18 o 
120 d3dfb672 2024-03-18 o func newRequest(method, url string, body io.Reader, key *rsa.PrivateKey, pubkeyURL string) (*http.Request, error) {
121 d3dfb672 2024-03-18 o 	req, err := http.NewRequest(method, url, body)
122 d3dfb672 2024-03-18 o 	if err != nil {
123 d3dfb672 2024-03-18 o 		return nil, err
124 d3dfb672 2024-03-18 o 	}
125 d3dfb672 2024-03-18 o 	req.Header.Set("Accept", ContentType)
126 d3dfb672 2024-03-18 o 	if body != nil {
127 d3dfb672 2024-03-18 o 		req.Header.Set("Content-Type", ContentType)
128 d3dfb672 2024-03-18 o 	}
129 d3dfb672 2024-03-18 o 	if key != nil {
130 d3dfb672 2024-03-18 o 		if err := Sign(req, key, pubkeyURL); err != nil {
131 d3dfb672 2024-03-18 o 			return nil, fmt.Errorf("sign request: %w", err)
132 d3dfb672 2024-03-18 o 		}
133 d3dfb672 2024-03-18 o 	}
134 d3dfb672 2024-03-18 o 	return req, nil
135 d3dfb672 2024-03-18 o }