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