forked from
tangled.org/core
Monorepo for Tangled
1package state
2
3import (
4 "database/sql"
5 "errors"
6 "net/http"
7 "strings"
8
9 "github.com/go-chi/chi/v5"
10 "tangled.org/core/appview/db"
11 "tangled.org/core/appview/issues"
12 "tangled.org/core/appview/knots"
13 "tangled.org/core/appview/labels"
14 "tangled.org/core/appview/metrics"
15 "tangled.org/core/appview/middleware"
16 "tangled.org/core/appview/migration"
17 "tangled.org/core/appview/notifications"
18 "tangled.org/core/appview/pipelines"
19 "tangled.org/core/appview/pulls"
20 "tangled.org/core/appview/repo"
21 "tangled.org/core/appview/settings"
22 "tangled.org/core/appview/signup"
23 "tangled.org/core/appview/spindles"
24 "tangled.org/core/appview/state/userutil"
25 avstrings "tangled.org/core/appview/strings"
26 "tangled.org/core/log"
27)
28
29func (s *State) Router() http.Handler {
30 router := chi.NewRouter()
31 middleware := middleware.New(
32 s.oauth,
33 s.db,
34 s.enforcer,
35 s.repoResolver,
36 s.idResolver,
37 s.pages,
38 s.rdb,
39 s.logger,
40 )
41
42 router.Use(metrics.Middleware)
43
44 m := migration.NewMigration(s.db, s.oauth, s.idResolver.Directory(), s.logger)
45 router.Use(m.BackgroundMigrationMiddleware)
46
47 router.Get("/pwa-manifest.json", s.WebAppManifest)
48 router.Get("/robots.txt", s.RobotsTxt)
49 router.Get("/.well-known/security.txt", s.SecurityTxt)
50
51 userRouter := s.UserRouter(&middleware)
52 standardRouter := s.StandardRouter(&middleware)
53
54 router.HandleFunc("/*", func(w http.ResponseWriter, r *http.Request) {
55 pat := chi.URLParam(r, "*")
56 pathParts := strings.SplitN(pat, "/", 2)
57
58 if len(pathParts) > 0 {
59 firstPart := pathParts[0]
60
61 if userutil.IsDid(firstPart) {
62 repo, err := db.GetRepoByDid(s.db, firstPart)
63 switch {
64 case err == nil:
65 remaining := ""
66 if len(pathParts) > 1 {
67 remaining = "/" + pathParts[1]
68 }
69 rewritten := "/" + repo.Did + "/" + repo.Rkey + remaining
70 r2 := r.Clone(r.Context())
71 r2.URL.Path = rewritten
72 r2.URL.RawPath = rewritten
73 userRouter.ServeHTTP(w, r2)
74 case errors.Is(err, sql.ErrNoRows):
75 userRouter.ServeHTTP(w, r)
76 default:
77 s.logger.Error("db error looking up repo DID", "repoDid", firstPart, "err", err)
78 http.Error(w, "internal server error", http.StatusInternalServerError)
79 }
80 return
81 }
82
83 if userutil.IsHandle(firstPart) {
84 userRouter.ServeHTTP(w, r)
85 return
86 }
87
88 // if using a flattened DID (like you would in go modules), unflatten
89 if userutil.IsFlattenedDid(firstPart) {
90 unflattenedDid := userutil.UnflattenDid(firstPart)
91 redirectPath := strings.Join(append([]string{unflattenedDid}, pathParts[1:]...), "/")
92
93 redirectURL := *r.URL
94 redirectURL.Path = "/" + redirectPath
95
96 http.Redirect(w, r, redirectURL.String(), http.StatusFound)
97 return
98 }
99
100 // if using a handle with @, rewrite to work without @
101 if normalized := strings.TrimPrefix(firstPart, "@"); userutil.IsHandle(normalized) {
102 redirectPath := strings.Join(append([]string{normalized}, pathParts[1:]...), "/")
103
104 redirectURL := *r.URL
105 redirectURL.Path = "/" + redirectPath
106
107 http.Redirect(w, r, redirectURL.String(), http.StatusFound)
108 return
109 }
110
111 }
112
113 standardRouter.ServeHTTP(w, r)
114 })
115
116 return router
117}
118
119func (s *State) UserRouter(mw *middleware.Middleware) http.Handler {
120 r := chi.NewRouter()
121
122 r.With(mw.ResolveIdent()).Route("/{user}", func(r chi.Router) {
123 r.Get("/", s.Profile)
124 r.Get("/feed.atom", s.AtomFeedPage)
125
126 r.With(mw.ResolveRepo()).Route("/{repo}", func(r chi.Router) {
127 r.Use(mw.GoImport())
128 r.Mount("/", s.RepoRouter(mw))
129 r.Mount("/issues", s.IssuesRouter(mw))
130 r.Mount("/pulls", s.PullsRouter(mw))
131 r.Mount("/pipelines", s.PipelinesRouter(mw))
132 r.Mount("/labels", s.LabelsRouter())
133
134 // These routes get proxied to the knot
135 r.Get("/info/refs", s.InfoRefs)
136 r.Post("/git-upload-archive", s.UploadArchive)
137 r.Post("/git-upload-pack", s.UploadPack)
138 r.Post("/git-receive-pack", s.ReceivePack)
139
140 })
141 })
142
143 r.NotFound(func(w http.ResponseWriter, r *http.Request) {
144 w.WriteHeader(http.StatusNotFound)
145 s.pages.Error404(w)
146 })
147
148 return r
149}
150
151func (s *State) StandardRouter(mw *middleware.Middleware) http.Handler {
152 r := chi.NewRouter()
153
154 r.Handle("/static/*", s.pages.Static())
155
156 r.Get("/", s.HomeOrTimeline)
157 r.Get("/home", s.Home)
158 r.Get("/timeline", s.Timeline)
159 r.Get("/upgradeBanner", s.UpgradeBanner)
160 r.Post("/newsletter/signup", s.NewsletterSignup)
161 r.Post("/newsletter/dismiss", s.NewsletterDismiss)
162
163 // special-case handler for serving tangled.org/core
164 r.Get("/core", s.Core())
165
166 r.Get("/login", s.Login)
167 r.Post("/login", s.Login)
168 r.Post("/logout", s.Logout)
169
170 r.With(middleware.Paginate).Get("/search", s.Search)
171 r.With(middleware.AuthMiddleware(s.oauth)).Get("/search/quick", s.SearchQuick)
172 r.With(middleware.AuthMiddleware(s.oauth)).Get("/search/quick/mobile", s.SearchQuickMobile)
173
174 r.Post("/account/switch", s.SwitchAccount)
175 r.With(middleware.AuthMiddleware(s.oauth)).Delete("/account/{did}", s.RemoveAccount)
176
177 r.Route("/repo", func(r chi.Router) {
178 r.Route("/new", func(r chi.Router) {
179 r.Use(middleware.AuthMiddleware(s.oauth))
180 r.Get("/", s.NewRepo)
181 r.Post("/", s.NewRepo)
182 })
183 // r.Post("/import", s.ImportRepo)
184 })
185
186 r.With(middleware.Paginate).Get("/goodfirstissues", s.GoodFirstIssues)
187
188 r.With(middleware.AuthMiddleware(s.oauth)).Route("/follow", func(r chi.Router) {
189 r.Post("/", s.Follow)
190 r.Delete("/", s.Follow)
191 })
192
193 r.With(middleware.AuthMiddleware(s.oauth)).Route("/vouch", func(r chi.Router) {
194 r.Post("/", s.Vouch)
195 r.Post("/skip", s.SkipVouchSuggestion)
196 })
197
198 r.With(middleware.AuthMiddleware(s.oauth)).Route("/star", func(r chi.Router) {
199 r.Post("/", s.Star)
200 r.Delete("/", s.Star)
201 })
202
203 r.With(middleware.AuthMiddleware(s.oauth)).Route("/react", func(r chi.Router) {
204 r.Post("/", s.React)
205 r.Delete("/", s.React)
206 })
207
208 r.Get("/profile/popover", s.ProfilePopover)
209
210 r.Route("/profile", func(r chi.Router) {
211 r.Use(middleware.AuthMiddleware(s.oauth))
212 r.Get("/edit-bio", s.EditBioFragment)
213 r.Get("/edit-pins", s.EditPinsFragment)
214 r.Post("/bio", s.UpdateProfileBio)
215 r.Post("/pins", s.UpdateProfilePins)
216 r.Post("/avatar", s.UploadProfileAvatar)
217 r.Delete("/avatar", s.RemoveProfileAvatar)
218 r.Post("/punchcard", s.UpdateProfilePunchcardSetting)
219 })
220
221 r.Mount("/settings", s.SettingsRouter())
222 r.Mount("/strings", s.StringsRouter(mw))
223
224 r.Mount("/settings/knots", s.KnotsRouter())
225 r.Mount("/settings/spindles", s.SpindlesRouter())
226
227 r.Mount("/notifications", s.NotificationsRouter(mw))
228
229 r.Mount("/signup", s.SignupRouter())
230 r.Mount("/", s.oauth.Router())
231
232 r.Get("/keys/{user}", s.Keys)
233 r.Get("/terms", s.TermsOfService)
234 r.Get("/privacy", s.PrivacyPolicy)
235 r.Get("/brand", s.Brand)
236
237 r.NotFound(func(w http.ResponseWriter, r *http.Request) {
238 w.WriteHeader(http.StatusNotFound)
239 s.pages.Error404(w)
240 })
241 return r
242}
243
244// Core serves tangled.org/core go-import meta tags, and redirects
245// to the core repository if accessed normally.
246func (s *State) Core() http.HandlerFunc {
247 return func(w http.ResponseWriter, r *http.Request) {
248 if r.URL.Query().Get("go-get") == "1" {
249 w.Header().Set("Content-Type", "text/html")
250 w.Write([]byte(`<meta name="go-import" content="tangled.org/core git https://tangled.org/@tangled.org/core">`))
251 return
252 }
253
254 http.Redirect(w, r, "/@tangled.org/core", http.StatusFound)
255 }
256}
257
258func (s *State) SettingsRouter() http.Handler {
259 settings := &settings.Settings{
260 Db: s.db,
261 OAuth: s.oauth,
262 Pages: s.pages,
263 Config: s.config,
264 CfClient: s.cfClient,
265 Logger: log.SubLogger(s.logger, "settings"),
266 IdResolver: s.idResolver,
267 }
268
269 return settings.Router()
270}
271
272func (s *State) SpindlesRouter() http.Handler {
273 logger := log.SubLogger(s.logger, "spindles")
274
275 spindles := &spindles.Spindles{
276 Db: s.db,
277 OAuth: s.oauth,
278 Pages: s.pages,
279 Config: s.config,
280 Enforcer: s.enforcer,
281 IdResolver: s.idResolver,
282 Logger: logger,
283 }
284
285 return spindles.Router()
286}
287
288func (s *State) KnotsRouter() http.Handler {
289 logger := log.SubLogger(s.logger, "knots")
290
291 knots := &knots.Knots{
292 Db: s.db,
293 OAuth: s.oauth,
294 Pages: s.pages,
295 Config: s.config,
296 Enforcer: s.enforcer,
297 IdResolver: s.idResolver,
298 Knotstream: s.knotstream,
299 Logger: logger,
300 }
301
302 return knots.Router()
303}
304
305func (s *State) StringsRouter(mw *middleware.Middleware) http.Handler {
306 logger := log.SubLogger(s.logger, "strings")
307
308 strs := &avstrings.Strings{
309 Db: s.db,
310 OAuth: s.oauth,
311 Pages: s.pages,
312 IdResolver: s.idResolver,
313 Notifier: s.notifier,
314 Logger: logger,
315 }
316
317 return strs.Router(mw)
318}
319
320func (s *State) IssuesRouter(mw *middleware.Middleware) http.Handler {
321 issues := issues.New(
322 s.oauth,
323 s.repoResolver,
324 s.enforcer,
325 s.pages,
326 s.idResolver,
327 s.mentionsResolver,
328 s.db,
329 s.config,
330 s.notifier,
331 s.validator,
332 s.indexer.Issues,
333 log.SubLogger(s.logger, "issues"),
334 )
335 return issues.Router(mw)
336}
337
338func (s *State) PullsRouter(mw *middleware.Middleware) http.Handler {
339 pulls := pulls.New(
340 s.oauth,
341 s.repoResolver,
342 s.pages,
343 s.idResolver,
344 s.mentionsResolver,
345 s.db,
346 s.config,
347 s.notifier,
348 s.enforcer,
349 s.validator,
350 s.indexer.Pulls,
351 log.SubLogger(s.logger, "pulls"),
352 )
353 return pulls.Router(mw)
354}
355
356func (s *State) RepoRouter(mw *middleware.Middleware) http.Handler {
357 repo := repo.New(
358 s.oauth,
359 s.repoResolver,
360 s.pages,
361 s.spindlestream,
362 s.idResolver,
363 s.db,
364 s.config,
365 s.notifier,
366 s.enforcer,
367 log.SubLogger(s.logger, "repo"),
368 s.validator,
369 s.cfClient,
370 )
371 return repo.Router(mw)
372}
373
374func (s *State) PipelinesRouter(mw *middleware.Middleware) http.Handler {
375 pipes := pipelines.New(
376 s.oauth,
377 s.repoResolver,
378 s.pages,
379 s.spindlestream,
380 s.idResolver,
381 s.db,
382 s.config,
383 s.enforcer,
384 log.SubLogger(s.logger, "pipelines"),
385 )
386 return pipes.Router(mw)
387}
388
389func (s *State) LabelsRouter() http.Handler {
390 ls := labels.New(
391 s.oauth,
392 s.pages,
393 s.db,
394 s.validator,
395 s.enforcer,
396 s.notifier,
397 log.SubLogger(s.logger, "labels"),
398 )
399 return ls.Router()
400}
401
402func (s *State) NotificationsRouter(mw *middleware.Middleware) http.Handler {
403 notifs := notifications.New(s.db, s.oauth, s.pages, log.SubLogger(s.logger, "notifications"))
404 return notifs.Router(mw)
405}
406
407func (s *State) SignupRouter() http.Handler {
408 sig := signup.New(s.config, s.db, s.posthog, s.idResolver, s.pages, log.SubLogger(s.logger, "signup"))
409 return sig.Router()
410}