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 1d5ddf5d 2024-02-20 o 	"fmt"
  8 71191436 2024-02-28 o 	"io"
  9 1d5ddf5d 2024-02-20 o 	"net/http"
 10 71191436 2024-02-28 o 	"os"
 11 1d5ddf5d 2024-02-20 o 	"strings"
 12 1d5ddf5d 2024-02-20 o )
 13 1d5ddf5d 2024-02-20 o 
 14 1d5ddf5d 2024-02-20 o var defaultClient Client = Client{Client: http.DefaultClient}
 15 1d5ddf5d 2024-02-20 o 
 16 1d5ddf5d 2024-02-20 o func Lookup(id string) (*Activity, error) {
 17 1d5ddf5d 2024-02-20 o 	return defaultClient.Lookup(id)
 18 1d5ddf5d 2024-02-20 o }
 19 1d5ddf5d 2024-02-20 o 
 20 1d5ddf5d 2024-02-20 o func LookupActor(id string) (*Actor, error) {
 21 1d5ddf5d 2024-02-20 o 	return defaultClient.LookupActor(id)
 22 1d5ddf5d 2024-02-20 o }
 23 1d5ddf5d 2024-02-20 o 
 24 1d5ddf5d 2024-02-20 o type Client struct {
 25 1d5ddf5d 2024-02-20 o 	*http.Client
 26 1d5ddf5d 2024-02-20 o 	Key   *rsa.PrivateKey
 27 1d5ddf5d 2024-02-20 o 	Actor *Actor
 28 1d5ddf5d 2024-02-20 o }
 29 1d5ddf5d 2024-02-20 o 
 30 1d5ddf5d 2024-02-20 o func (c *Client) Lookup(id string) (*Activity, error) {
 31 1d5ddf5d 2024-02-20 o 	if !strings.HasPrefix(id, "http") {
 32 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("id is not a HTTP URL")
 33 1d5ddf5d 2024-02-20 o 	}
 34 1d5ddf5d 2024-02-20 o 	if c.Client == nil {
 35 1d5ddf5d 2024-02-20 o 		c.Client = http.DefaultClient
 36 1d5ddf5d 2024-02-20 o 	}
 37 1d5ddf5d 2024-02-20 o 
 38 1d5ddf5d 2024-02-20 o 	req, err := http.NewRequest(http.MethodGet, id, nil)
 39 1d5ddf5d 2024-02-20 o 	if err != nil {
 40 1d5ddf5d 2024-02-20 o 		return nil, err
 41 1d5ddf5d 2024-02-20 o 	}
 42 1d5ddf5d 2024-02-20 o 	req.Header.Set("Accept", ContentType)
 43 1d5ddf5d 2024-02-20 o 	if c.Key != nil && c.Actor != nil {
 44 1d5ddf5d 2024-02-20 o 		if err := Sign(req, c.Key, c.Actor.PublicKey.ID); err != nil {
 45 1d5ddf5d 2024-02-20 o 			return nil, fmt.Errorf("sign http request: %w", err)
 46 1d5ddf5d 2024-02-20 o 		}
 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 71191436 2024-02-28 o 	return activityToActor(activity), nil
 67 71191436 2024-02-28 o }
 68 71191436 2024-02-28 o 
 69 71191436 2024-02-28 o func activityToActor(activity *Activity) *Actor {
 70 77918b00 2024-03-07 o 	actor := &Actor{
 71 1d5ddf5d 2024-02-20 o 		AtContext: activity.AtContext,
 72 1d5ddf5d 2024-02-20 o 		ID:        activity.ID,
 73 1d5ddf5d 2024-02-20 o 		Type:      activity.Type,
 74 1d5ddf5d 2024-02-20 o 		Name:      activity.Name,
 75 1d5ddf5d 2024-02-20 o 		Username:  activity.Username,
 76 1d5ddf5d 2024-02-20 o 		Inbox:     activity.Inbox,
 77 1d5ddf5d 2024-02-20 o 		Outbox:    activity.Outbox,
 78 77918b00 2024-03-07 o 		Followers: activity.Followers,
 79 71191436 2024-02-28 o 		Published: activity.Published,
 80 71191436 2024-02-28 o 		Summary:   activity.Summary,
 81 71191436 2024-02-28 o 	}
 82 77918b00 2024-03-07 o 	if activity.PublicKey != nil {
 83 77918b00 2024-03-07 o 		actor.PublicKey = *activity.PublicKey
 84 77918b00 2024-03-07 o 	}
 85 77918b00 2024-03-07 o 	return actor
 86 1d5ddf5d 2024-02-20 o }
 87 1d5ddf5d 2024-02-20 o 
 88 1d5ddf5d 2024-02-20 o func (c *Client) Send(inbox string, activity *Activity) (*Activity, error) {
 89 1d5ddf5d 2024-02-20 o 	b, err := json.Marshal(activity)
 90 1d5ddf5d 2024-02-20 o 	if err != nil {
 91 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("encode outgoing activity: %w", err)
 92 1d5ddf5d 2024-02-20 o 	}
 93 1d5ddf5d 2024-02-20 o 	req, err := http.NewRequest(http.MethodPost, inbox, bytes.NewReader(b))
 94 1d5ddf5d 2024-02-20 o 	if err != nil {
 95 1d5ddf5d 2024-02-20 o 		return nil, err
 96 1d5ddf5d 2024-02-20 o 	}
 97 1d5ddf5d 2024-02-20 o 	req.Header.Set("Content-Type", ContentType)
 98 1d5ddf5d 2024-02-20 o 	if err := Sign(req, c.Key, c.Actor.PublicKey.ID); err != nil {
 99 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("sign outgoing request: %w", err)
100 1d5ddf5d 2024-02-20 o 	}
101 1d5ddf5d 2024-02-20 o 	resp, err := c.Do(req)
102 1d5ddf5d 2024-02-20 o 	if err != nil {
103 1d5ddf5d 2024-02-20 o 		return nil, err
104 1d5ddf5d 2024-02-20 o 	}
105 1d5ddf5d 2024-02-20 o 	switch resp.StatusCode {
106 77918b00 2024-03-07 o 	case http.StatusOK, http.StatusAccepted, http.StatusNoContent:
107 1d5ddf5d 2024-02-20 o 		return nil, nil
108 1d5ddf5d 2024-02-20 o 	case http.StatusNotFound:
109 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("no such inbox %s", inbox)
110 1d5ddf5d 2024-02-20 o 	default:
111 1d5ddf5d 2024-02-20 o 		io.Copy(os.Stderr, resp.Body)
112 1d5ddf5d 2024-02-20 o 		resp.Body.Close()
113 1d5ddf5d 2024-02-20 o 		return nil, fmt.Errorf("non-ok response status %s", resp.Status)
114 1d5ddf5d 2024-02-20 o 	}
115 1d5ddf5d 2024-02-20 o }