cdramey/qurl
1
0
Fork 0
Code Releases Activity

Fixed bad URL redirect bug

Browse Source
This commit is contained in:
Christopher Ramey 2018-11-24 04:22:41 +00:00 committed by cdramey
parent 0216358dfb
commit 01ef0d5183
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pages/root.go
View File

@ -47,6 +47,7 @@ func (ctx *RootHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
id, err := qurl.FromString(fname[1:len(fname)]) id, err := qurl.FromString(fname[1:len(fname)])
if err != nil { if err != nil {
http.Redirect(w, r, "/", http.StatusTemporaryRedirect) http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
return
} }
q, err := ctx.Storage.GetQURLByID(id) q, err := ctx.Storage.GetQURLByID(id)
@ -56,6 +57,11 @@ func (ctx *RootHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
return return
} }
if q == nil {
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
return
}
http.Redirect(w, r, q.URL, http.StatusPermanentRedirect) http.Redirect(w, r, q.URL, http.StatusPermanentRedirect)
} }
} }