refactor: move route handlers to named functions
This commit is contained in:
parent
e9aafd0cea
commit
36b012ac92
345
main.go
345
main.go
@ -146,6 +146,175 @@ func handleToAudio(ctx *Context, w http.ResponseWriter) http.ResponseWriter {
|
|||||||
return w
|
return w
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
func handleDownload(w http.ResponseWriter, r *http.Request) {
|
||||||
|
ctx := NewContext(r)
|
||||||
|
if r.Method != "POST" {
|
||||||
|
w.WriteHeader(400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err := r.ParseForm()
|
||||||
|
if err != nil {
|
||||||
|
w.WriteHeader(400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
userURL := r.FormValue("URL")
|
||||||
|
if userURL == "" {
|
||||||
|
w = writeJSONResponse(w, "Provide URL as query")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
toAudio := r.FormValue("toaudio")
|
||||||
|
if toAudio == "on" {
|
||||||
|
log.Println("User requested audio")
|
||||||
|
}
|
||||||
|
|
||||||
|
filenameChan := make(chan string)
|
||||||
|
|
||||||
|
go GetContentFilename(userURL, filenameChan)
|
||||||
|
downloadURL, err := getYoutubeDownloadURL(userURL)
|
||||||
|
filename := <- filenameChan
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
ctx.StatusCode = 500
|
||||||
|
ctx.Err = &err
|
||||||
|
err = templates.ExecuteTemplate(w,"download-result.html", ctx)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.DownloadURL = downloadURL
|
||||||
|
w.Header().Add(
|
||||||
|
"Hx-Redirect",
|
||||||
|
fmt.Sprintf("/download-direct?URL=%s&filename=%s",
|
||||||
|
url.QueryEscape(ctx.DownloadURL),
|
||||||
|
url.QueryEscape(filename)),
|
||||||
|
)
|
||||||
|
|
||||||
|
err = templates.ExecuteTemplate(w,"download-result.html", ctx)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
ctx.StatusCode = 500
|
||||||
|
ctx.Err = &err
|
||||||
|
err = templates.ExecuteTemplate(w,"download-result.html", ctx)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleDirectDownload(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|
||||||
|
ctx := NewContext(r)
|
||||||
|
if r.Method != "GET" {
|
||||||
|
w.WriteHeader(400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
userURL := strings.Trim(r.URL.Query().Get("URL"), "\n")
|
||||||
|
filename := strings.Trim(r.URL.Query().Get("filename"), "\n")
|
||||||
|
|
||||||
|
if userURL == "" {
|
||||||
|
log.Println("Empty URL")
|
||||||
|
w.WriteHeader(400)
|
||||||
|
ctx.StatusCode = 400
|
||||||
|
if err := templates.ExecuteTemplate(w,"download-result.html", ctx); err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ctx.DownloadURL = userURL
|
||||||
|
|
||||||
|
req, err := http.NewRequest("GET", ctx.DownloadURL, nil)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
ctx.StatusCode = 500
|
||||||
|
ctx.Err = &err
|
||||||
|
if err := templates.ExecuteTemplate(w,"download-result.html", ctx); err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/117.0")
|
||||||
|
|
||||||
|
client := http.Client{}
|
||||||
|
|
||||||
|
dataRequest, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
ctx.StatusCode = 500
|
||||||
|
ctx.Err = &err
|
||||||
|
if err := templates.ExecuteTemplate(w,"download-result.html", ctx); err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer dataRequest.Body.Close()
|
||||||
|
|
||||||
|
if dataRequest.StatusCode != 200 {
|
||||||
|
log.Println("Failed to get content for URL", userURL)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
contentLength := dataRequest.Header.Get("Content-Length")
|
||||||
|
if dataRequest.ContentLength == 0 {
|
||||||
|
log.Println("Empty body from content url")
|
||||||
|
w.WriteHeader(500)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
w.Header().Set(
|
||||||
|
"Content-Disposition",
|
||||||
|
fmt.Sprintf("attachment;filename=%s;", filename),
|
||||||
|
)
|
||||||
|
w.Header().Set("Content-Length", contentLength)
|
||||||
|
w.WriteHeader(206)
|
||||||
|
|
||||||
|
n, err := io.Copy(w, dataRequest.Body)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
}
|
||||||
|
log.Printf("Copied %d bytes", n)
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleRoot(w http.ResponseWriter, r *http.Request) {
|
||||||
|
ctx := NewContext(r)
|
||||||
|
formats := []DownloadFormats{}
|
||||||
|
formats = append(formats, DownloadFormats{
|
||||||
|
VideoRes: "720p",
|
||||||
|
audioOnly: false,
|
||||||
|
videoOnly: false,
|
||||||
|
})
|
||||||
|
err := templates.ExecuteTemplate(w, "download.html", ctx)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
w.WriteHeader(500)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleValidLink(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.Method != "POST" {
|
||||||
|
w.WriteHeader(400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
err := r.ParseForm()
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err.Error())
|
||||||
|
w.WriteHeader(400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
url := r.FormValue("URL")
|
||||||
|
|
||||||
|
ctx := URLValidationCtx{
|
||||||
|
URL: url,
|
||||||
|
Valid: isValidURL(url),
|
||||||
|
}
|
||||||
|
|
||||||
|
templates.ExecuteTemplate(w, "url-validation.html", ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
|
|
||||||
log.Println("[ init ] Starting...")
|
log.Println("[ init ] Starting...")
|
||||||
@ -194,179 +363,11 @@ func main() {
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
handler.HandleFunc(
|
handler.HandleFunc("/", handleRoot)
|
||||||
"/download",
|
handler.HandleFunc("/download", handleDownload)
|
||||||
func(w http.ResponseWriter, r *http.Request) {
|
handler.HandleFunc("/download-direct", handleDirectDownload)
|
||||||
ctx := NewContext(r)
|
handler.HandleFunc("/valid-link", handleValidLink)
|
||||||
if r.Method != "POST" {
|
|
||||||
w.WriteHeader(400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
err := r.ParseForm()
|
|
||||||
if err != nil {
|
|
||||||
w.WriteHeader(400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
userURL := r.FormValue("URL")
|
|
||||||
if userURL == "" {
|
|
||||||
w = writeJSONResponse(w, "Provide URL as query")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
toAudio := r.FormValue("toaudio")
|
|
||||||
if toAudio == "on" {
|
|
||||||
log.Println("User requested audio")
|
|
||||||
}
|
|
||||||
|
|
||||||
filenameChan := make(chan string)
|
|
||||||
|
|
||||||
go GetContentFilename(userURL, filenameChan)
|
|
||||||
downloadURL, err := getYoutubeDownloadURL(userURL)
|
|
||||||
filename := <- filenameChan
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
ctx.StatusCode = 500
|
|
||||||
ctx.Err = &err
|
|
||||||
err = templates.ExecuteTemplate(w,"download-result.html", ctx)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx.DownloadURL = downloadURL
|
|
||||||
w.Header().Add(
|
|
||||||
"Hx-Redirect",
|
|
||||||
fmt.Sprintf("/download-direct?URL=%s&filename=%s",
|
|
||||||
url.QueryEscape(ctx.DownloadURL),
|
|
||||||
url.QueryEscape(filename)),
|
|
||||||
)
|
|
||||||
|
|
||||||
err = templates.ExecuteTemplate(w,"download-result.html", ctx)
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
ctx.StatusCode = 500
|
|
||||||
ctx.Err = &err
|
|
||||||
err = templates.ExecuteTemplate(w,"download-result.html", ctx)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
handler.HandleFunc(
|
|
||||||
"/download-direct",
|
|
||||||
func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
|
|
||||||
ctx := NewContext(r)
|
|
||||||
if r.Method != "GET" {
|
|
||||||
w.WriteHeader(400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
userURL := strings.Trim(r.URL.Query().Get("URL"), "\n")
|
|
||||||
filename := strings.Trim(r.URL.Query().Get("filename"), "\n")
|
|
||||||
|
|
||||||
if userURL == "" {
|
|
||||||
log.Println("Empty URL")
|
|
||||||
w.WriteHeader(400)
|
|
||||||
ctx.StatusCode = 400
|
|
||||||
if err := templates.ExecuteTemplate(w,"download-result.html", ctx); err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
ctx.DownloadURL = userURL
|
|
||||||
|
|
||||||
req, err := http.NewRequest("GET", ctx.DownloadURL, nil)
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
ctx.StatusCode = 500
|
|
||||||
ctx.Err = &err
|
|
||||||
if err := templates.ExecuteTemplate(w,"download-result.html", ctx); err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/117.0")
|
|
||||||
|
|
||||||
client := http.Client{}
|
|
||||||
|
|
||||||
dataRequest, err := client.Do(req)
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
ctx.StatusCode = 500
|
|
||||||
ctx.Err = &err
|
|
||||||
if err := templates.ExecuteTemplate(w,"download-result.html", ctx); err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer dataRequest.Body.Close()
|
|
||||||
|
|
||||||
if dataRequest.StatusCode != 200 {
|
|
||||||
log.Println("Failed to get content for URL", userURL)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
contentLength := dataRequest.Header.Get("Content-Length")
|
|
||||||
if dataRequest.ContentLength == 0 {
|
|
||||||
log.Println("Empty body from content url")
|
|
||||||
w.WriteHeader(500)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
w.Header().Set(
|
|
||||||
"Content-Disposition",
|
|
||||||
fmt.Sprintf("attachment;filename=%s;", filename),
|
|
||||||
)
|
|
||||||
w.Header().Set("Content-Length", contentLength)
|
|
||||||
w.WriteHeader(206)
|
|
||||||
|
|
||||||
n, err := io.Copy(w, dataRequest.Body)
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
}
|
|
||||||
log.Printf("Copied %d bytes", n)
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
handler.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
ctx := NewContext(r)
|
|
||||||
formats := []DownloadFormats{}
|
|
||||||
formats = append(formats, DownloadFormats{
|
|
||||||
VideoRes: "720p",
|
|
||||||
audioOnly: false,
|
|
||||||
videoOnly: false,
|
|
||||||
})
|
|
||||||
err := templates.ExecuteTemplate(w, "download.html", ctx)
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
w.WriteHeader(500)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
handler.HandleFunc("/valid-link", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
|
|
||||||
if r.Method != "POST" {
|
|
||||||
w.WriteHeader(400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
err := r.ParseForm()
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err.Error())
|
|
||||||
w.WriteHeader(400)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
url := r.FormValue("URL")
|
|
||||||
|
|
||||||
ctx := URLValidationCtx{
|
|
||||||
URL: url,
|
|
||||||
Valid: isValidURL(url),
|
|
||||||
}
|
|
||||||
|
|
||||||
templates.ExecuteTemplate(w, "url-validation.html", ctx)
|
|
||||||
})
|
|
||||||
|
|
||||||
wrappedHandler := NewLogger(handler)
|
wrappedHandler := NewLogger(handler)
|
||||||
srv := http.Server{
|
srv := http.Server{
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user