mirror of
https://github.com/Floriansylvain/RenewCMS.git
synced 2026-08-19 19:53:21 +02:00
clean: moved funcs to specific files + GetPage usecase/repo
This commit is contained in:
+9
-120
@@ -1,22 +1,14 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"embed"
|
||||
"encoding/json"
|
||||
"github.com/go-chi/chi/v5"
|
||||
"html/template"
|
||||
"io/fs"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
)
|
||||
|
||||
//go:embed web/templates/*
|
||||
var templateFiles embed.FS
|
||||
|
||||
//go:embed web/static
|
||||
//go:embed static
|
||||
var staticFolder embed.FS
|
||||
|
||||
var contentTypes = map[string]string{
|
||||
@@ -29,110 +21,7 @@ var contentTypes = map[string]string{
|
||||
".ico": "image/x-icon",
|
||||
}
|
||||
|
||||
const loginRoute = "/login"
|
||||
|
||||
type LoginPage struct {
|
||||
IsError bool `json:"isError"`
|
||||
Error string `json:"error"`
|
||||
Username string `json:"username"`
|
||||
}
|
||||
|
||||
func NewLoginPage(error string, username string) *LoginPage {
|
||||
return &LoginPage{
|
||||
IsError: strings.Compare(error, "") != 0,
|
||||
Error: error,
|
||||
Username: username,
|
||||
}
|
||||
}
|
||||
|
||||
func getPage(page string, data interface{}) ([]byte, error) {
|
||||
var processedHTML bytes.Buffer
|
||||
tmpl, err := template.ParseFS(templateFiles, "web/templates/"+page+".html")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
err = tmpl.Execute(&processedHTML, data)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return processedHTML.Bytes(), nil
|
||||
}
|
||||
|
||||
func getLoginPageHandler(loginPage *LoginPage) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
if IsLoggedIn(r) {
|
||||
http.Redirect(w, r, "/home", http.StatusSeeOther)
|
||||
return
|
||||
}
|
||||
if r.Method == http.MethodPost {
|
||||
postLoginPage(w, r)
|
||||
return
|
||||
}
|
||||
bs, err := getPage("login", &loginPage)
|
||||
if err != nil {
|
||||
_, _ = w.Write([]byte(err.Error()))
|
||||
}
|
||||
_, _ = w.Write(bs)
|
||||
}
|
||||
}
|
||||
|
||||
func postLoginPage(w http.ResponseWriter, r *http.Request) {
|
||||
_ = r.ParseForm()
|
||||
|
||||
credentials, err := json.Marshal(&UserLogin{
|
||||
Username: r.FormValue("username"),
|
||||
Password: r.FormValue("password"),
|
||||
})
|
||||
if err != nil {
|
||||
r.Method = http.MethodGet
|
||||
getLoginPageHandler(NewLoginPage("Invalid username or password.", r.FormValue("username")))(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
response, err := http.Post(
|
||||
"http://localhost:"+os.Getenv("PORT")+"/v1/auth/login",
|
||||
"application/json",
|
||||
bytes.NewBuffer(credentials))
|
||||
|
||||
if err != nil || response.StatusCode != http.StatusOK {
|
||||
r.Method = http.MethodGet
|
||||
getLoginPageHandler(NewLoginPage("Invalid username or password.", r.FormValue("username")))(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Set-Cookie", response.Header.Get("Set-Cookie"))
|
||||
|
||||
http.Redirect(w, r, "/home", http.StatusSeeOther)
|
||||
}
|
||||
|
||||
func getHomePage(w http.ResponseWriter, _ *http.Request) {
|
||||
navbarTmpl, _ := getPage("componentNavbar", nil)
|
||||
homeTmpl, _ := getPage("home", map[string]interface{}{
|
||||
"Navbar": template.HTML(navbarTmpl),
|
||||
})
|
||||
_, _ = w.Write(homeTmpl)
|
||||
}
|
||||
|
||||
func getLogin(w http.ResponseWriter, r *http.Request) {
|
||||
http.Redirect(w, r, loginRoute, http.StatusPermanentRedirect)
|
||||
}
|
||||
|
||||
func getLogout(w http.ResponseWriter, r *http.Request) {
|
||||
removeJwtCookie(w)
|
||||
http.Redirect(w, r, loginRoute, http.StatusSeeOther)
|
||||
}
|
||||
|
||||
func IsLoggedInMiddleware(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if !IsLoggedIn(r) {
|
||||
http.Redirect(w, r, loginRoute, http.StatusSeeOther)
|
||||
return
|
||||
}
|
||||
next.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
func staticFileServerWithContentType(fsys http.FileSystem) http.Handler {
|
||||
func StaticFileServerWithContentType(fsys http.FileSystem) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
path := r.URL.Path
|
||||
if ext := filepath.Ext(path); ext != "" {
|
||||
@@ -147,9 +36,9 @@ func staticFileServerWithContentType(fsys http.FileSystem) http.Handler {
|
||||
func NewPageRouter() http.Handler {
|
||||
r := chi.NewRouter()
|
||||
|
||||
contentStatic, _ := fs.Sub(fs.FS(staticFolder), "web/static")
|
||||
contentStatic := fs.FS(staticFolder)
|
||||
|
||||
r.Handle("/static/*", staticFileServerWithContentType(http.FS(contentStatic)))
|
||||
r.Handle("/static/*", StaticFileServerWithContentType(http.FS(contentStatic)))
|
||||
r.HandleFunc("/favicon.ico", func(w http.ResponseWriter, r *http.Request) {
|
||||
_, err := http.FS(staticFolder).Open("favicon.ico")
|
||||
if err != nil {
|
||||
@@ -157,14 +46,14 @@ func NewPageRouter() http.Handler {
|
||||
}
|
||||
})
|
||||
|
||||
r.Get("/", getLogin)
|
||||
r.Get(loginRoute, getLoginPageHandler(NewLoginPage("", "")))
|
||||
r.Post(loginRoute, getLoginPageHandler(NewLoginPage("", "")))
|
||||
r.Get("/logout", getLogout)
|
||||
r.Get("/", GetLogin)
|
||||
r.Get(LoginRoute, GetLoginPageHandler(NewLoginPage("", "")))
|
||||
r.Post(LoginRoute, GetLoginPageHandler(NewLoginPage("", "")))
|
||||
r.Get("/logout", GetLogout)
|
||||
|
||||
r.Group(func(r chi.Router) {
|
||||
r.Use(IsLoggedInMiddleware)
|
||||
r.Get("/home", getHomePage)
|
||||
r.Get("/home", GetHomePage)
|
||||
})
|
||||
|
||||
return r
|
||||
|
||||
Reference in New Issue
Block a user