aboutsummaryrefslogtreecommitdiff
path: root/http/router.go
blob: 4c251df1578bb9a08d658765a0cc1443fd377b48 (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
package http

import (
	"bytes"
	"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 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()
}