core / appview / state / router.go

package state

import (
	"net/http"
	"strings"

	"github.com/go-chi/chi/v5"
	"tangled.org/core/appview/issues"
	"tangled.org/core/appview/knots"
	"tangled.org/core/appview/labels"
	"tangled.org/core/appview/middleware"
	"tangled.org/core/appview/notifications"
	"tangled.org/core/appview/pipelines"
	"tangled.org/core/appview/pulls"
	"tangled.org/core/appview/repo"
	"tangled.org/core/appview/settings"
	"tangled.org/core/appview/signup"
	"tangled.org/core/appview/spindles"
	"tangled.org/core/appview/state/userutil"
	avstrings "tangled.org/core/appview/strings"
	"tangled.org/core/log"
)

func (s *State) Router() http.Handler {
	router := chi.NewRouter()
	middleware := middleware.New(
		s.oauth,
		s.db,
		s.enforcer,
		s.repoResolver,
		s.idResolver,
		s.pages,
	)

	router.Get("/pwa-manifest.json", s.WebAppManifest)
	router.Get("/robots.txt", s.RobotsTxt)

	userRouter := s.UserRouter(&middleware)
	standardRouter := s.StandardRouter(&middleware)

	router.HandleFunc("/*", func(w http.ResponseWriter, r *http.Request) {
		pat := chi.URLParam(r, "*")
		pathParts := strings.SplitN(pat, "/", 2)

		if len(pathParts) > 0 {
			firstPart := pathParts[0]

			// if using a DID or handle, just continue as per usual
			if userutil.IsDid(firstPart) || userutil.IsHandle(firstPart) {
				userRouter.ServeHTTP(w, r)
				return
			}

			// if using a flattened DID (like you would in go modules), unflatten
			if userutil.IsFlattenedDid(firstPart) {
				unflattenedDid := userutil.UnflattenDid(firstPart)
				redirectPath := strings.Join(append([]string{unflattenedDid}, pathParts[1:]...), "/")

				redirectURL := *r.URL
				redirectURL.Path = "/" + redirectPath

				http.Redirect(w, r, redirectURL.String(), http.StatusFound)
				return
			}

			// if using a handle with @, rewrite to work without @
			if normalized := strings.TrimPrefix(firstPart, "@"); userutil.IsHandle(normalized) {
				redirectPath := strings.Join(append([]string{normalized}, pathParts[1:]...), "/")

				redirectURL := *r.URL
				redirectURL.Path = "/" + redirectPath

				http.Redirect(w, r, redirectURL.String(), http.StatusFound)
				return
			}

		}

		standardRouter.ServeHTTP(w, r)
	})

	return router
}

func (s *State) UserRouter(mw *middleware.Middleware) http.Handler {
	r := chi.NewRouter()

	r.With(mw.ResolveIdent()).Route("/{user}", func(r chi.Router) {
		r.Get("/", s.Profile)
		r.Get("/feed.atom", s.AtomFeedPage)

		r.With(mw.ResolveRepo()).Route("/{repo}", func(r chi.Router) {
			r.Use(mw.GoImport())
			r.Mount("/", s.RepoRouter(mw))
			r.Mount("/issues", s.IssuesRouter(mw))
			r.Mount("/pulls", s.PullsRouter(mw))
			r.Mount("/pipelines", s.PipelinesRouter())
			r.Mount("/labels", s.LabelsRouter())

			// These routes get proxied to the knot
			r.Get("/info/refs", s.InfoRefs)
			r.Post("/git-upload-archive", s.UploadArchive)
			r.Post("/git-upload-pack", s.UploadPack)
			r.Post("/git-receive-pack", s.ReceivePack)

		})
	})

	r.NotFound(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusNotFound)
		s.pages.Error404(w)
	})

	return r
}

func (s *State) StandardRouter(mw *middleware.Middleware) http.Handler {
	r := chi.NewRouter()

	r.Handle("/static/*", s.pages.Static())

	r.Get("/", s.HomeOrTimeline)
	r.Get("/timeline", s.Timeline)
	r.Get("/upgradeBanner", s.UpgradeBanner)

	// special-case handler for serving tangled.org/core
	r.Get("/core", s.Core())

	r.Get("/login", s.Login)
	r.Post("/login", s.Login)
	r.Post("/logout", s.Logout)

	r.Route("/repo", func(r chi.Router) {
		r.Route("/new", func(r chi.Router) {
			r.Use(middleware.AuthMiddleware(s.oauth))
			r.Get("/", s.NewRepo)
			r.Post("/", s.NewRepo)
		})
		// r.Post("/import", s.ImportRepo)
	})

	r.With(middleware.Paginate).Get("/goodfirstissues", s.GoodFirstIssues)

	r.With(middleware.AuthMiddleware(s.oauth)).Route("/follow", func(r chi.Router) {
		r.Post("/", s.Follow)
		r.Delete("/", s.Follow)
	})

	r.With(middleware.AuthMiddleware(s.oauth)).Route("/star", func(r chi.Router) {
		r.Post("/", s.Star)
		r.Delete("/", s.Star)
	})

	r.With(middleware.AuthMiddleware(s.oauth)).Route("/react", func(r chi.Router) {
		r.Post("/", s.React)
		r.Delete("/", s.React)
	})

	r.Route("/profile", func(r chi.Router) {
		r.Use(middleware.AuthMiddleware(s.oauth))
		r.Get("/edit-bio", s.EditBioFragment)
		r.Get("/edit-pins", s.EditPinsFragment)
		r.Post("/bio", s.UpdateProfileBio)
		r.Post("/pins", s.UpdateProfilePins)
	})

	r.Mount("/settings", s.SettingsRouter())
	r.Mount("/strings", s.StringsRouter(mw))

	r.Mount("/settings/knots", s.KnotsRouter())
	r.Mount("/settings/spindles", s.SpindlesRouter())

	r.Mount("/notifications", s.NotificationsRouter(mw))

	r.Mount("/signup", s.SignupRouter())
	r.Mount("/", s.oauth.Router())

	r.Get("/keys/{user}", s.Keys)
	r.Get("/terms", s.TermsOfService)
	r.Get("/privacy", s.PrivacyPolicy)
	r.Get("/brand", s.Brand)

	r.NotFound(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusNotFound)
		s.pages.Error404(w)
	})
	return r
}

// Core serves tangled.org/core go-import meta tags, and redirects
// to the core repository if accessed normally.
func (s *State) Core() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Query().Get("go-get") == "1" {
			w.Header().Set("Content-Type", "text/html")
			w.Write([]byte(`<meta name="go-import" content="tangled.org/core git https://tangled.org/@tangled.org/core">`))
			return
		}

		http.Redirect(w, r, "/@tangled.org/core", http.StatusFound)
	}
}

func (s *State) SettingsRouter() http.Handler {
	settings := &settings.Settings{
		Db:     s.db,
		OAuth:  s.oauth,
		Pages:  s.pages,
		Config: s.config,
	}

	return settings.Router()
}

func (s *State) SpindlesRouter() http.Handler {
	logger := log.SubLogger(s.logger, "spindles")

	spindles := &spindles.Spindles{
		Db:         s.db,
		OAuth:      s.oauth,
		Pages:      s.pages,
		Config:     s.config,
		Enforcer:   s.enforcer,
		IdResolver: s.idResolver,
		Logger:     logger,
	}

	return spindles.Router()
}

func (s *State) KnotsRouter() http.Handler {
	logger := log.SubLogger(s.logger, "knots")

	knots := &knots.Knots{
		Db:         s.db,
		OAuth:      s.oauth,
		Pages:      s.pages,
		Config:     s.config,
		Enforcer:   s.enforcer,
		IdResolver: s.idResolver,
		Knotstream: s.knotstream,
		Logger:     logger,
	}

	return knots.Router()
}

func (s *State) StringsRouter(mw *middleware.Middleware) http.Handler {
	logger := log.SubLogger(s.logger, "strings")

	strs := &avstrings.Strings{
		Db:         s.db,
		OAuth:      s.oauth,
		Pages:      s.pages,
		IdResolver: s.idResolver,
		Notifier:   s.notifier,
		Logger:     logger,
	}

	return strs.Router(mw)
}

func (s *State) IssuesRouter(mw *middleware.Middleware) http.Handler {
	issues := issues.New(
		s.oauth,
		s.repoResolver,
		s.enforcer,
		s.pages,
		s.idResolver,
		s.mentionsResolver,
		s.db,
		s.config,
		s.notifier,
		s.validator,
		s.indexer.Issues,
		log.SubLogger(s.logger, "issues"),
	)
	return issues.Router(mw)
}

func (s *State) PullsRouter(mw *middleware.Middleware) http.Handler {
	pulls := pulls.New(
		s.oauth,
		s.repoResolver,
		s.pages,
		s.idResolver,
		s.mentionsResolver,
		s.db,
		s.config,
		s.notifier,
		s.enforcer,
		s.validator,
		s.indexer.Pulls,
		log.SubLogger(s.logger, "pulls"),
	)
	return pulls.Router(mw)
}

func (s *State) RepoRouter(mw *middleware.Middleware) http.Handler {
	repo := repo.New(
		s.oauth,
		s.repoResolver,
		s.pages,
		s.spindlestream,
		s.idResolver,
		s.db,
		s.config,
		s.notifier,
		s.enforcer,
		log.SubLogger(s.logger, "repo"),
		s.validator,
	)
	return repo.Router(mw)
}

func (s *State) PipelinesRouter() http.Handler {
	pipes := pipelines.New(
		s.oauth,
		s.repoResolver,
		s.pages,
		s.spindlestream,
		s.idResolver,
		s.db,
		s.config,
		s.enforcer,
		log.SubLogger(s.logger, "pipelines"),
	)
	return pipes.Router()
}

func (s *State) LabelsRouter() http.Handler {
	ls := labels.New(
		s.oauth,
		s.pages,
		s.db,
		s.validator,
		s.enforcer,
		log.SubLogger(s.logger, "labels"),
	)
	return ls.Router()
}

func (s *State) NotificationsRouter(mw *middleware.Middleware) http.Handler {
	notifs := notifications.New(s.db, s.oauth, s.pages, log.SubLogger(s.logger, "notifications"))
	return notifs.Router(mw)
}

func (s *State) SignupRouter() http.Handler {
	sig := signup.New(s.config, s.db, s.posthog, s.idResolver, s.pages, log.SubLogger(s.logger, "signup"))
	return sig.Router()
}