mirror of
https://github.com/mgerb/go-discord-bot
synced 2026-01-11 01:22:48 +00:00
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
|
|
"github.com/mgerb/chi_auth_server/response"
|
|
)
|
|
|
|
// Downloader -
|
|
func Downloader(w http.ResponseWriter, r *http.Request) {
|
|
url := r.FormValue("url")
|
|
fileType := r.FormValue("fileType")
|
|
|
|
// create youtube folder if it does not exist
|
|
if _, err := os.Stat("youtube"); os.IsNotExist(err) {
|
|
os.Mkdir("youtube", os.ModePerm)
|
|
}
|
|
|
|
// get the video title
|
|
titleCmd := exec.Command("youtube-dl", "--get-title", url)
|
|
var titleOut bytes.Buffer
|
|
titleCmd.Stdout = &titleOut
|
|
|
|
err := titleCmd.Run()
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
response.ERR(w, http.StatusInternalServerError, response.DefaultInternalError)
|
|
return
|
|
}
|
|
|
|
// TODO add video id to tile to not get collisions
|
|
|
|
// ------------------------------------------------
|
|
|
|
// remove all special characters from title
|
|
cleanTitle := cleanseTitle(titleOut.String())
|
|
log.Println(cleanTitle)
|
|
|
|
cmd := exec.Command("youtube-dl", "-x", "--audio-format", "mp3", "-o", "./youtube/"+cleanTitle+".%(ext)s", url)
|
|
|
|
var out bytes.Buffer
|
|
cmd.Stdout = &out
|
|
|
|
err = cmd.Run()
|
|
|
|
if err != nil {
|
|
log.Println(out.String())
|
|
log.Println(err)
|
|
response.ERR(w, http.StatusInternalServerError, response.DefaultInternalError)
|
|
return
|
|
}
|
|
|
|
response.JSON(w, map[string]interface{}{"fileName": cleanTitle + "." + fileType})
|
|
}
|
|
|
|
func cleanseTitle(title string) string {
|
|
|
|
// Make a Regex to say we only want
|
|
reg, err := regexp.Compile("[^a-zA-Z0-9]+")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return reg.ReplaceAllString(title, "")
|
|
}
|