91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/zhang2092/go-url-shortener/internal/db"
|
|
"github.com/zhang2092/go-url-shortener/internal/middleware"
|
|
"github.com/zhang2092/go-url-shortener/internal/service"
|
|
"github.com/zhang2092/go-url-shortener/internal/shortener"
|
|
"github.com/zhang2092/go-url-shortener/internal/templ"
|
|
)
|
|
|
|
func CreateShortUrlView(w http.ResponseWriter, r *http.Request) {
|
|
templ.CreateUrl(w, r, "")
|
|
}
|
|
|
|
func CreateShortUrl(store db.Store) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
defer r.Body.Close()
|
|
|
|
if err := r.ParseForm(); err != nil {
|
|
templ.CreateUrl(w, r, "请求参数错误")
|
|
return
|
|
}
|
|
|
|
ctx := r.Context()
|
|
user := middleware.GetUser(ctx)
|
|
longUrl := r.PostFormValue("long_url")
|
|
shortUrl, err := shortener.GenerateShortLink(longUrl, user.ID)
|
|
if err != nil {
|
|
templ.CreateUrl(w, r, "生成短路径错误")
|
|
return
|
|
}
|
|
|
|
_, err = store.CreateUserUrl(ctx, &db.CreateUserUrlParams{
|
|
UserID: user.ID,
|
|
ShortUrl: shortUrl,
|
|
OriginUrl: longUrl,
|
|
ExpireAt: time.Now().Add(time.Hour * 6),
|
|
})
|
|
if err != nil {
|
|
templ.CreateUrl(w, r, "短路径存储错误")
|
|
return
|
|
}
|
|
|
|
err = service.SaveUrlMapping(shortUrl, longUrl, user.ID)
|
|
if err != nil {
|
|
templ.CreateUrl(w, r, "短路径存储错误")
|
|
return
|
|
}
|
|
|
|
http.Redirect(w, r, "/", http.StatusFound)
|
|
}
|
|
}
|
|
|
|
func DeleteShortUrl(store db.Store) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
shorUrl := vars["shortUrl"]
|
|
_, err := store.UpdateStatus(r.Context(), &db.UpdateStatusParams{
|
|
ShortUrl: shorUrl,
|
|
Status: -1,
|
|
})
|
|
if err != nil {
|
|
RespondErr(w, "删除错误", nil)
|
|
return
|
|
}
|
|
err = service.DeleteShortUrl(shorUrl)
|
|
if err != nil {
|
|
RespondErr(w, "删除错误", nil)
|
|
return
|
|
}
|
|
|
|
respond(w, "删除成功", nil, http.StatusOK)
|
|
}
|
|
}
|
|
|
|
func HandleShortUrlRedirect(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
shorUrl := vars["shortUrl"]
|
|
link, err := service.RetrieveInitialUrl(shorUrl)
|
|
if err != nil || len(link) == 0 {
|
|
RespondErr(w, "短链已经失效", nil)
|
|
return
|
|
}
|
|
|
|
http.Redirect(w, r, link, http.StatusFound)
|
|
}
|