AverageMarcus/dir2opds
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix hrel url generation and

Browse Source 
- add debug flag
 - add support for comics via mimetypes
This commit is contained in:
Sinuhe Tellez 2020-05-15 13:38:03 -04:00
parent db39591697
commit b97ca396c2
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
main.go
View File

@ -42,6 +42,7 @@ var (
author = flag.String("author", "", "The server Feed author") author = flag.String("author", "", "The server Feed author")
authorURI = flag.String("uri", "", "The feed's author uri") authorURI = flag.String("uri", "", "The feed's author uri")
authorEmail = flag.String("email", "", "The feed's author email") authorEmail = flag.String("email", "", "The feed's author email")
debug = flag.Bool("debug", false, "If it is set it will log the requests")
) )
type acquisitionFeed struct { type acquisitionFeed struct {
@ -51,14 +52,21 @@ type acquisitionFeed struct {
} }
func init() { func init() {
mime.AddExtensionType(".mobi", "application/x-mobipocket-ebook") _ = mime.AddExtensionType(".mobi", "application/x-mobipocket-ebook")
mime.AddExtensionType(".epub", "application/epub+zip") _ = mime.AddExtensionType(".epub", "application/epub+zip")
mime.AddExtensionType(".fb2", "text/fb2+xml") _ = mime.AddExtensionType(".cbz", "application/x-cbz")
_ = mime.AddExtensionType(".cbr", "application/x-cbr")
_ = mime.AddExtensionType(".fb2", "text/fb2+xml")
} }
func main() { func main() {
flag.Parse() flag.Parse()
if *debug {
log.SetOutput(os.Stdout)
}
fmt.Println(startValues()) fmt.Println(startValues())
http.HandleFunc("/", errorHandler(handler)) http.HandleFunc("/", errorHandler(handler))
@ -67,14 +75,13 @@ func main() {
} }
func startValues() string { func startValues() string {
var result string result := fmt.Sprintf("listening in: %s:%s", *host, *port)
result = fmt.Sprintf("listening in: %s:%s", *host, *port)
return result return result
} }
func handler(w http.ResponseWriter, req *http.Request) error { func handler(w http.ResponseWriter, req *http.Request) error {
fpath := filepath.Join(*dirRoot, req.URL.Path) fpath := filepath.Join(*dirRoot, req.URL.Path)
log.Printf("fpath:'%s'", fpath)
fi, err := os.Stat(fpath) fi, err := os.Stat(fpath)
if err != nil { if err != nil {
return err return err
@ -158,7 +165,7 @@ func getType(name string, pathType int) string {
} }
func getHref(req *http.Request, name string) string { func getHref(req *http.Request, name string) string {
return filepath.Join(req.URL.EscapedPath(), url.PathEscape(name)) return filepath.Join(req.URL.RequestURI(), url.PathEscape(name))
} }
const ( const (