aboutsummaryrefslogtreecommitdiff
path: root/http/router.go
blob: 6a434991133d75a2fc140f3937b79822219fef3f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
package http

import (
	"bytes"
	"fmt"
	"io"
	"net/http"

	"git.capotej.com/capotej/communique/registry"
	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
)

type Router struct {
	registry *registry.Registry
	log      *zap.SugaredLogger
}

func NewRouter(registry *registry.Registry, log *zap.SugaredLogger) *Router {
	return &Router{registry: registry, log: log}
}

func render(c *gin.Context, resource map[string]interface{}, err error) {
	if resource == nil && err == nil {
		c.JSON(http.StatusNotFound, map[string]interface{}{})
	} else if err != nil {
		jsonErr := c.Error(err)
		c.JSON(http.StatusInternalServerError, jsonErr.JSON())
	} else {
		c.Writer.Header().Set("Content-Type", "application/activity+json")
		c.JSON(http.StatusOK, resource)
	}
}

func (s *Router) Start(zapWriter io.Writer) {
	router := gin.Default()
	router.SetTrustedProxies(nil)
	gin.DisableConsoleColor()
	gin.DefaultWriter = zapWriter // send gin logs to zap

	// Webfinger
	router.GET("/.well-known/webfinger", func(c *gin.Context) {
		resourceParam := c.Query("resource")
		resource, _ := s.registry.Webfinger(resourceParam)
		if resource != nil {
			c.Writer.Header().Set("Content-Type", "application/jrd+json")
			c.JSON(http.StatusOK, resource)
		} else {
			c.JSON(http.StatusNotFound, map[string]interface{}{})
		}

	})

	// Actor
	router.GET("/actors/:actor", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.Actor(actorParam)
		render(c, resource, err)
	})

	// Actor avatar
	router.GET("/actors/:actor/avatar", func(c *gin.Context) {
		actorParam := c.Param("actor")
		avatarBytes, mediaType, err := s.registry.ActorAvatar(actorParam)
		if err != nil || avatarBytes == nil || mediaType == "" {
			c.Data(404, "text/plain", []byte("404 page not found"))
		}
		c.Header("Content-Length", fmt.Sprintf("%d", len(avatarBytes)))
		c.Data(200, mediaType, avatarBytes)
	})

	// Actor Followers
	router.GET("/actors/:actor/followers", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.Followers(actorParam)
		render(c, resource, err)
	})

	// Actor Following
	router.GET("/actors/:actor/following", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.Following(actorParam)
		render(c, resource, err)
	})

	// Inbox
	router.POST("/actors/:actor/inbox", func(c *gin.Context) {
		log := s.log.With("type", "inbox")
		buf := new(bytes.Buffer)
		buf.ReadFrom(c.Request.Body)
		payload := buf.String()
		log.With(
			"signature",
			c.GetHeader("Signature"),
		).With(
			"content-type",
			c.GetHeader("Content-Type"),
		).With(
			"digest",
			c.GetHeader("digest"),
		).With(
			"payload",
			payload,
		).Debug("received inbox item")
		actorParam := c.Param("actor")
		err := s.registry.Inbox(actorParam, c.Request, buf.Bytes())
		resource := map[string]interface{}{}
		render(c, resource, err)
	})

	// Actor Outbox
	router.GET("/actors/:actor/outbox", func(c *gin.Context) {
		actorParam := c.Param("actor")
		resource, err := s.registry.OutboxCollection(actorParam)
		render(c, resource, err)
	})

	// Actor Activity
	router.GET("/actors/:actor/activity/:id", func(c *gin.Context) {
		actorParam := c.Param("actor")
		idParam := c.Param("id")
		resource, err := s.registry.ActivityOrNote("activity", actorParam, idParam)
		render(c, resource, err)
	})

	// Activity Note
	router.GET("/actors/:actor/activity/:id/note", func(c *gin.Context) {
		actorParam := c.Param("actor")
		idParam := c.Param("id")
		resource, err := s.registry.ActivityOrNote("note", actorParam, idParam)
		render(c, resource, err)
	})

	router.Run()
}