368 lines
9.5 KiB
Go
368 lines
9.5 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"html/template"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
DEFAULT_HTTP_PORT = "8080"
|
|
DEFAULT_HTTPS_PORT = "4433"
|
|
)
|
|
|
|
type DownloadFormats struct {
|
|
VideoRes string
|
|
videoOnly bool
|
|
audioOnly bool
|
|
}
|
|
|
|
type URLValidationCtx struct {
|
|
URL string
|
|
Valid bool
|
|
}
|
|
|
|
type apiMessageResponse struct {
|
|
Message string
|
|
}
|
|
|
|
type Logger struct {
|
|
Handler http.Handler
|
|
}
|
|
|
|
func (l *Logger) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
|
|
tNow := time.Now().UTC()
|
|
l.Handler.ServeHTTP(w, r)
|
|
methodString := l.getMethodLogString(r.Method)
|
|
remote := r.RemoteAddr
|
|
realIP := r.Header.Get("X-Real-IP")
|
|
if realIP != "" {
|
|
remote = realIP
|
|
}
|
|
log.Printf(" %s %s %s %v", remote, methodString, r.URL, time.Since(tNow))
|
|
}
|
|
|
|
func (l *Logger) getMethodLogString(method string) string {
|
|
colorBlue := "\033[34m"
|
|
colorReset := "\033[0m"
|
|
colorRed := "\033[31m"
|
|
colorGreen := "\033[32m"
|
|
colorYellow := "\033[33m"
|
|
// colorPurple := "\033[35m"
|
|
// colorCyan := "\033[36m"
|
|
// colorWhite := "\033[37m"
|
|
switch method {
|
|
case "GET": return colorBlue + "GET" + colorReset
|
|
case "POST": return colorGreen + "POST" + colorReset
|
|
case "DELETE": return colorRed + "DELETE" + colorReset
|
|
case "PUT": return colorYellow + "PUT" + colorReset
|
|
default: return method
|
|
}
|
|
}
|
|
|
|
func NewLogger(handler http.Handler) *Logger {
|
|
return &Logger{Handler: handler}
|
|
}
|
|
|
|
func writeJSONResponse(w http.ResponseWriter, s string) http.ResponseWriter {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
jsonResp, err := json.Marshal(apiMessageResponse{Message: s})
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return w
|
|
}
|
|
_, err = w.Write(jsonResp)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return w
|
|
}
|
|
|
|
return w
|
|
|
|
}
|
|
|
|
var templates *template.Template
|
|
|
|
// TODO: Change all this to have a unified context
|
|
type Context struct {
|
|
request *http.Request
|
|
Formats []DownloadFormats
|
|
AppURL string
|
|
StatusCode int
|
|
Err *error
|
|
IsTLS bool
|
|
DownloadURL string
|
|
}
|
|
|
|
func NewContext(r *http.Request) *Context {
|
|
isTLS := false
|
|
tls := os.Getenv("TLS")
|
|
if tls != "" {
|
|
isTLS = true
|
|
}
|
|
|
|
return &Context{
|
|
request: r,
|
|
StatusCode: 200,
|
|
Formats: []DownloadFormats{
|
|
{
|
|
VideoRes: "720p",
|
|
videoOnly: false,
|
|
audioOnly: false,
|
|
},
|
|
},
|
|
AppURL: r.Host,
|
|
IsTLS: isTLS,
|
|
}
|
|
}
|
|
|
|
func handleToAudio(ctx *Context, w http.ResponseWriter) http.ResponseWriter {
|
|
|
|
log.Println("User requested audio")
|
|
w.WriteHeader(400)
|
|
|
|
jsonMsg, err := json.Marshal(
|
|
apiMessageResponse{
|
|
Message: "Audio only is not implemented",
|
|
},
|
|
)
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
w.WriteHeader(500)
|
|
}
|
|
w.Write(jsonMsg)
|
|
|
|
return w
|
|
}
|
|
|
|
func init() {
|
|
|
|
log.Println("[ init ] Starting...")
|
|
templates = template.Must(template.ParseFS(TemplatesFS , "templates/*.html"))
|
|
log.Println("[ init ] Templates Loaded")
|
|
|
|
}
|
|
|
|
func main() {
|
|
|
|
handler := http.NewServeMux()
|
|
handler.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.FS(PublicFS))))
|
|
handler.Handle("/assets/", http.FileServer(http.FS(AssetsFS)))
|
|
|
|
handler.HandleFunc(
|
|
"/robots.txt",
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
robotsFile, err := RobotsFS.ReadFile("static/robots.txt")
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
w.WriteHeader(500)
|
|
return
|
|
}
|
|
w.Write(robotsFile)
|
|
},
|
|
)
|
|
|
|
handler.HandleFunc(
|
|
"/sitemap.txt",
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
sitemapFile, err := AssetsFS.ReadFile("assets/sitemap.txt")
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
w.WriteHeader(500)
|
|
return
|
|
}
|
|
w.Write(sitemapFile)
|
|
},
|
|
)
|
|
|
|
|
|
handler.HandleFunc(
|
|
"/download",
|
|
func(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")
|
|
}
|
|
|
|
downloadURL, err := getYoutubeDownloadURL(userURL)
|
|
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", ctx.DownloadURL))
|
|
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 := r.URL.Query().Get("URL")
|
|
urlRaw := strings.TrimLeft(r.URL.RawQuery, "URL=")
|
|
urlRaw, err := url.QueryUnescape(urlRaw)
|
|
if err != nil {
|
|
log.Println("Can not unescape url")
|
|
w.WriteHeader(400)
|
|
return
|
|
}
|
|
userURL = urlRaw
|
|
|
|
if userURL == "" {
|
|
w.WriteHeader(400)
|
|
ctx.StatusCode = 400
|
|
if err := templates.ExecuteTemplate(w,"download-result.html", ctx); err != nil {
|
|
log.Println(err.Error())
|
|
}
|
|
return
|
|
}
|
|
log.Println("Got url: ", userURL, "for direct download")
|
|
ctx.DownloadURL = userURL
|
|
|
|
req, err := http.NewRequest("GET", ctx.DownloadURL, nil)
|
|
if err != nil {
|
|
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()
|
|
|
|
log.Printf("HTTP Client response: %v", dataRequest)
|
|
|
|
if dataRequest.StatusCode != 200 {
|
|
log.Println("Failed to get content for URL", userURL)
|
|
}
|
|
|
|
w.Header().Set("Content-Disposition", "attachment;filename=")
|
|
w.WriteHeader(206)
|
|
|
|
if dataRequest.ContentLength == 0 {
|
|
log.Println("Empty body from content url")
|
|
w.WriteHeader(500)
|
|
return
|
|
}
|
|
|
|
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)
|
|
srv := http.Server{
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
Addr: ":" + DEFAULT_HTTP_PORT,
|
|
Handler: wrappedHandler,
|
|
}
|
|
|
|
log.Printf("Starting HTTP on %s", DEFAULT_HTTP_PORT)
|
|
log.Fatalln(srv.ListenAndServe())
|
|
}
|
|
|
|
|
|
|
|
|