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) } }