2024-12-11 17:11:53 +08:00

39 lines
1.2 KiB
Go

package templ
import (
"net/http"
"github.com/zhang2092/go-url-shortener/internal/db"
"github.com/zhang2092/go-url-shortener/internal/templ/auth"
"github.com/zhang2092/go-url-shortener/internal/templ/err"
"github.com/zhang2092/go-url-shortener/internal/templ/home"
"github.com/zhang2092/go-url-shortener/internal/templ/model"
"github.com/zhang2092/go-url-shortener/internal/templ/url"
)
func Login(w http.ResponseWriter, r *http.Request, form *model.LoginPageData) {
checkErr(w, auth.Login(r, "login", form).Render(r.Context(), w))
}
func Register(w http.ResponseWriter, r *http.Request, form *model.RegisterPageData) {
checkErr(w, auth.Register(r, "register", form).Render(r.Context(), w))
}
func Home(w http.ResponseWriter, r *http.Request, data []*db.UserRelateUrl) {
checkErr(w, home.Home(r, "home", data).Render(r.Context(), w))
}
func CreateUrl(w http.ResponseWriter, r *http.Request, errorMsg string) {
checkErr(w, url.CreateUrl(r, "create_url", errorMsg).Render(r.Context(), w))
}
func Error404(w http.ResponseWriter, r *http.Request) {
checkErr(w, err.Error404().Render(r.Context(), w))
}
func checkErr(w http.ResponseWriter, err error) {
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}