|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
_ "embed"
|
|
|
|
"flag"
|
|
|
|
"fmt"
|
|
|
|
"net/http"
|
|
|
|
"os"
|
|
|
|
"strconv"
|
|
|
|
"sync"
|
|
|
|
"time"
|
|
|
|
"ytdlp-viewer/DirectoryIndexers"
|
|
|
|
"ytdlp-viewer/PageHandlers"
|
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
var path string
|
|
|
|
flag.StringVar(&path, "path", "./", "the full path to the ytdlp archive")
|
|
|
|
// append last slash in case it's not provided
|
|
|
|
if path[len(path)-1:] != "/" {
|
|
|
|
path += "/"
|
|
|
|
}
|
|
|
|
var port int
|
|
|
|
flag.IntVar(&port, "port", 8000, "the port for the web panel to listen on")
|
|
|
|
var autorefresh bool
|
|
|
|
flag.BoolVar(&autorefresh, "refresh", false, "whether should the index be updated every x seconds")
|
|
|
|
var refreshInterval int
|
|
|
|
flag.IntVar(&refreshInterval, "interval", 60, "the interval for the index to update in seconds")
|
|
|
|
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
var FL DirectoryIndexers.FileList
|
|
|
|
FL.RWMutex = &sync.RWMutex{}
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
resultChannel := make(chan DirectoryIndexers.FileList)
|
|
|
|
|
|
|
|
fmt.Println("Starting scanner at", path)
|
|
|
|
go DirectoryIndexers.Index(path, resultChannel, &FL)
|
|
|
|
|
|
|
|
refreshedFL := <-resultChannel
|
|
|
|
FL.Lock()
|
|
|
|
FL.Files = refreshedFL.Files
|
|
|
|
FL.Unlock()
|
|
|
|
|
|
|
|
if !autorefresh {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
time.Sleep(time.Duration(refreshInterval) * time.Second)
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
|
|
|
http.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) {
|
|
|
|
fmt.Println(time.Now().String(),request.URL)
|
|
|
|
PageHandlers.Index(writer, request, &FL)
|
|
|
|
})
|
|
|
|
http.HandleFunc("/search", func(writer http.ResponseWriter, request *http.Request) {
|
|
|
|
fmt.Println(time.Now().String(),request.URL)
|
|
|
|
PageHandlers.SearchHandler(writer, request, &FL)
|
|
|
|
})
|
|
|
|
http.HandleFunc("/view", func(writer http.ResponseWriter, request *http.Request) {
|
|
|
|
fmt.Println(time.Now().String(),request.URL)
|
|
|
|
PageHandlers.View(writer, request, &FL, path)
|
|
|
|
})
|
|
|
|
http.Handle("/videos/", http.StripPrefix("/videos/", http.FileServer(http.Dir(path))))
|
|
|
|
|
|
|
|
err := http.ListenAndServe(":" + strconv.Itoa(port), nil)
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
}
|