package router import ( "fmt" "net/http" "path/filepath" "github.com/debridmediamanager/zurg/internal/config" "github.com/debridmediamanager/zurg/internal/dav" intHttp "github.com/debridmediamanager/zurg/internal/http" "github.com/debridmediamanager/zurg/internal/torrent" "github.com/debridmediamanager/zurg/internal/universal" "github.com/debridmediamanager/zurg/pkg/logutil" "github.com/debridmediamanager/zurg/pkg/realdebrid" "github.com/julienschmidt/httprouter" ) type ZurgRouter struct { getfile *universal.GetFile torMgr *torrent.TorrentManager cfg config.ConfigInterface api *realdebrid.RealDebrid log *logutil.Logger } func ApplyRouteTable(router *httprouter.Router, getfile *universal.GetFile, torMgr *torrent.TorrentManager, cfg config.ConfigInterface, api *realdebrid.RealDebrid, log *logutil.Logger) { zr := &ZurgRouter{ getfile: getfile, torMgr: torMgr, cfg: cfg, api: api, log: log, } // http router router.GET("/http/:directory/:torrent/:file", zr.universalDownloadFileHandler) router.GET("/http/:directory/:torrent/", zr.httpTorrentDirectoryHandler) router.GET("/http/:directory/", zr.httpDirectoryHandler) router.GET("/http/", zr.httpRootHandler) // HEAD route router.HEAD("/http/:directory/:torrent/:file", zr.headFileHandler) // dav router router.GET("/dav/:directory/:torrent/:file", zr.universalDownloadFileHandler) router.GET("/dav/:directory/:torrent/", zr.propfindTorrentHandler) router.GET("/dav/:directory/", zr.propfindDirectoryHandler) router.GET("/dav/", zr.propfindRootHandler) // PROPFIND routes router.Handle("PROPFIND", "/dav/:directory/:torrent/", zr.propfindTorrentHandler) router.Handle("PROPFIND", "/dav/:directory/", zr.propfindDirectoryHandler) router.Handle("PROPFIND", "/dav/", zr.propfindRootHandler) // DELETE routes router.DELETE("/dav/:directory/:torrent/:file", zr.deleteFileHandler) router.DELETE("/dav/:directory/:torrent/", zr.deleteTorrentHandler) // rename sequence router.Handle("PROPFIND", "/dav/:directory/:torrent/:file", zr.propfindFileHandler) router.Handle("MKCOL", "/dav/:directory/:torrent/", zr.mkcolTorrentHandler) router.Handle("MOVE", "/dav/:directory/:torrent/:file", zr.moveFileHandler) // MOVE routes router.Handle("MOVE", "/dav/:directory/:torrent/", zr.moveTorrentHandler) // Global OPTIONS route router.GlobalOPTIONS = http.HandlerFunc(zr.globalOptionsHandler) // root route router.GET("/", zr.rootHandler) // logs route router.GET("/logs", zr.logsHandler) router.GET("/logs/", zr.logsHandler) router.MethodNotAllowed = http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) { zr.log.Debugf("Method Not Allowed: %s %s %v", req.Method, req.URL, req.Header) http.Error(resp, "Method Not Allowed", http.StatusMethodNotAllowed) }) } func (zr *ZurgRouter) httpTorrentDirectoryHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") out, err := intHttp.HandleListFiles(directory, torrentName, zr.torMgr, zr.log) if err != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.Header().Set("Content-Type", "text/html; charset=\"utf-8\"") resp.WriteHeader(http.StatusOK) resp.Write(out) } func (zr *ZurgRouter) httpDirectoryHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") out, err := intHttp.HandleListTorrents(directory, zr.torMgr, zr.log) if err != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.Header().Set("Content-Type", "text/html; charset=\"utf-8\"") resp.WriteHeader(http.StatusOK) resp.Write(out) } func (zr *ZurgRouter) httpRootHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { out, err := intHttp.HandleListDirectories(zr.torMgr) if err != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.Header().Set("Content-Type", "text/html; charset=\"utf-8\"") resp.WriteHeader(http.StatusOK) resp.Write(out) } func (zr *ZurgRouter) propfindFileHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") fileName := params.ByName("file") out, err := dav.HandlePropfindFile(directory, torrentName, fileName, zr.torMgr, zr.log) if err != nil { fmt.Println(">>>>>>>>>>>>>>>>>>>. not found", err) http.Error(resp, "Not Found", http.StatusNotFound) return } fmt.Println(">>>>>>>>>>>>>>>>>>>. found yey") resp.Header().Set("Content-Type", "text/xml; charset=\"utf-8\"") resp.WriteHeader(http.StatusOK) resp.Write(out) } func (zr *ZurgRouter) propfindTorrentHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") out, err := dav.HandleListFiles(directory, torrentName, zr.torMgr, zr.log) if err != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.Header().Set("Content-Type", "text/xml; charset=\"utf-8\"") resp.WriteHeader(http.StatusOK) resp.Write(out) } func (zr *ZurgRouter) propfindDirectoryHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") out, err := dav.HandleListTorrents(directory, zr.torMgr, zr.log) if err != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.Header().Set("Content-Type", "text/xml; charset=\"utf-8\"") resp.WriteHeader(http.StatusOK) resp.Write(out) } func (zr *ZurgRouter) propfindRootHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { out, err := dav.HandleListDirectories(zr.torMgr) if err != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.Header().Set("Content-Type", "text/xml; charset=\"utf-8\"") resp.WriteHeader(http.StatusOK) resp.Write(out) } func (zr *ZurgRouter) deleteFileHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") fileName := params.ByName("file") if dav.HandleDeleteFile(directory, torrentName, fileName, zr.torMgr) != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.WriteHeader(http.StatusNoContent) } func (zr *ZurgRouter) deleteTorrentHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") if dav.HandleDeleteTorrent(directory, torrentName, zr.torMgr) != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.WriteHeader(http.StatusNoContent) } func (zr *ZurgRouter) mkcolTorrentHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { fmt.Println(">>>>>>>>>>>>>>>>>>> mkcolTorrentHandler") resp.WriteHeader(http.StatusNoContent) } func (zr *ZurgRouter) moveFileHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") fileName := params.ByName("file") newName := req.Header.Get("Destination") newName = filepath.Base(newName) fmt.Println(">>>>>>>>>>>>>>>>>>> moveFileHandler", fileName, ">>>>>>>>", newName) if dav.HandleRenameFile(directory, torrentName, fileName, newName, zr.torMgr) != nil { fmt.Println(">>>>>>>>>>>>>>>>>>> moveFileHandler not found") http.Error(resp, "Not Found", http.StatusNotFound) return } fmt.Println(">>>>>>>>>>>>>>>>>>> moveFileHandler yay") resp.WriteHeader(http.StatusNoContent) } func (zr *ZurgRouter) moveTorrentHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") newName := req.Header.Get("Destination") newName = filepath.Base(newName) if dav.HandleRenameTorrent(directory, torrentName, newName, zr.torMgr) != nil { http.Error(resp, "Not Found", http.StatusNotFound) return } resp.WriteHeader(http.StatusNoContent) } func (zr *ZurgRouter) globalOptionsHandler(resp http.ResponseWriter, req *http.Request) { resp.WriteHeader(http.StatusOK) } func (zr *ZurgRouter) universalDownloadFileHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") fileName := params.ByName("file") zr.getfile.HandleGetRequest(directory, torrentName, fileName, resp, req, zr.torMgr, zr.cfg, zr.log) } func (zr *ZurgRouter) headFileHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { directory := params.ByName("directory") torrentName := params.ByName("torrent") fileName := params.ByName("file") universal.HandleHeadRequest(directory, torrentName, fileName, resp, req, zr.torMgr, zr.log) } func (zr *ZurgRouter) logsHandler(resp http.ResponseWriter, req *http.Request, params httprouter.Params) { logs, err := zr.log.GetLogsFromFile() if err != nil { http.Error(resp, err.Error(), http.StatusInternalServerError) return } fmt.Fprint(resp, logs) } func bToMb(b uint64) uint64 { return b / 1024 / 1024 }