100 lines
2.1 KiB
Go
100 lines
2.1 KiB
Go
package api
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"git.dubyatp.xyz/chat-api-server/db"
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/go-chi/docgen"
|
|
"github.com/go-chi/render"
|
|
)
|
|
|
|
var routes = flag.Bool("routes", false, "Generate API route documentation")
|
|
|
|
func Start() {
|
|
|
|
db.InitScyllaDB()
|
|
defer db.CloseScyllaDB()
|
|
|
|
flag.Parse()
|
|
|
|
r := chi.NewRouter()
|
|
|
|
r.Use(middleware.RequestID)
|
|
r.Use(middleware.Logger)
|
|
r.Use(middleware.Recoverer)
|
|
r.Use(middleware.URLFormat)
|
|
r.Use(render.SetContentType(render.ContentTypeJSON))
|
|
|
|
r.Get("/", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("hello world"))
|
|
})
|
|
|
|
r.Get("/ping", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("pong"))
|
|
})
|
|
|
|
r.Get("/panic", func(w http.ResponseWriter, r *http.Request) {
|
|
panic("oh no")
|
|
})
|
|
|
|
r.Route("/whoami", func(r chi.Router) {
|
|
r.Use(SessionAuthMiddleware)
|
|
r.Use(LoginCtx)
|
|
r.Get("/", Whoami)
|
|
})
|
|
|
|
r.Route("/messages", func(r chi.Router) {
|
|
r.Use(SessionAuthMiddleware) // Protect with authentication
|
|
|
|
r.Get("/", ListMessages)
|
|
r.Route("/{messageID}", func(r chi.Router) {
|
|
r.Use(MessageCtx) // Load message
|
|
r.Get("/", GetMessage)
|
|
r.Delete("/", DeleteMessage)
|
|
r.Post("/edit", EditMessage)
|
|
})
|
|
r.Route("/new", func(r chi.Router) {
|
|
r.Use(LoginCtx)
|
|
r.Post("/", NewMessage)
|
|
})
|
|
})
|
|
|
|
r.Route("/users", func(r chi.Router) {
|
|
r.Use(SessionAuthMiddleware) // Protect with authentication
|
|
|
|
r.Get("/", ListUsers)
|
|
r.Route("/{userID}", func(r chi.Router) {
|
|
r.Use(UserCtx) // Load user
|
|
r.Get("/", GetUser)
|
|
})
|
|
})
|
|
|
|
r.Route("/login", func(r chi.Router) {
|
|
r.Post("/", Login)
|
|
})
|
|
|
|
r.Route("/logout", func(r chi.Router) {
|
|
r.Use(SessionAuthMiddleware)
|
|
|
|
r.Post("/", Logout)
|
|
})
|
|
|
|
r.Route("/register", func(r chi.Router) {
|
|
r.Post("/", NewUser)
|
|
})
|
|
|
|
if *routes {
|
|
fmt.Println(docgen.MarkdownRoutesDoc(r, docgen.MarkdownOpts{
|
|
ProjectPath: "git.dubyatp.xyz/chat-api-server",
|
|
Intro: "Welcome to the chat API server. This is a simple API for sending and receiving messages.",
|
|
}))
|
|
return
|
|
}
|
|
|
|
http.ListenAndServe(":3000", r)
|
|
}
|