/* Copyright (C) 2017 Sinuhé Téllez Rivera dir2opds is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. dir2opds is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with dir2opds. If not, see . */ package main import ( "bytes" "encoding/xml" "flag" "fmt" "io/ioutil" "log" "mime" "net/http" "net/url" "os" "path/filepath" "time" "github.com/dubyte/dir2opds/opds" "golang.org/x/tools/blog/atom" ) var ( port = flag.String("port", "8080", "The server will listen in this port") host = flag.String("host", "0.0.0.0", "The server will listen in this host") dirRoot = flag.String("dir", "./books", "A directory with books") author = flag.String("author", "", "The server Feed author") authorURI = flag.String("uri", "", "The feed's author uri") authorEmail = flag.String("email", "", "The feed's author email") ) const acquisitionType = "application/atom+xml;profile=opds-catalog;kind=acquisition" const navegationType = "application/atom+xml;profile=opds-catalog;kind=navigation" type acquisitionFeed struct { *atom.Feed Dc string `xml:"xmlns:dc,attr"` Opds string `xml:"xmlns:opds,attr"` } func init() { mime.AddExtensionType(".mobi", "application/x-mobipocket-ebook") mime.AddExtensionType(".epub", "application/epub+zip") mime.AddExtensionType(".fb2", "txt/xml") } func main() { flag.Parse() fmt.Println(startValues()) http.HandleFunc("/", errorHandler(handler)) log.Fatal(http.ListenAndServe(*host+":"+*port, nil)) } func startValues() string { var result string result = fmt.Sprintf("dir2opds started to listen in: %s:%s", *host, *port) return result } func handler(w http.ResponseWriter, req *http.Request) error { fpath := filepath.Join(*dirRoot, req.URL.Path) fi, err := os.Stat(fpath) if err != nil { return err } if isFile(fi) { http.ServeFile(w, req, fpath) return nil } content, err := getContent(req, fpath) if err != nil { return err } http.ServeContent(w, req, "feed.xml", time.Now(), bytes.NewReader(content)) return nil } func getContent(req *http.Request, dirpath string) (result []byte, err error) { feed := makeFeed(dirpath, req) if isAcquisition(dirpath) { acFeed := &acquisitionFeed{&feed, "http://purl.org/dc/terms/", "http://opds-spec.org/2010/catalog"} result, err = xml.MarshalIndent(acFeed, " ", " ") } else { result, err = xml.MarshalIndent(feed, " ", " ") } return } func makeFeed(dirpath string, req *http.Request) atom.Feed { feedBuilder := opds.FeedBuilder. ID(req.URL.Path). Title("Catalog in " + req.URL.Path). Author(opds.AuthorBuilder.Name(*author).Email(*authorEmail).URI(*authorURI).Build()). Updated(time.Now()). AddLink(opds.LinkBuilder.Rel("start").Href("/").Type(navegationType).Build()) fis, _ := ioutil.ReadDir(dirpath) for _, fi := range fis { linkIsAcquisition := isAcquisition(filepath.Join(dirpath, fi.Name())) feedBuilder = feedBuilder. AddEntry(opds.EntryBuilder. ID(req.URL.Path + fi.Name()). Title(fi.Name()). Updated(time.Now()). Published(time.Now()). AddLink(opds.LinkBuilder. Rel(getRel(fi.Name(), linkIsAcquisition)). Title(fi.Name()). Href(getHref(req, fi.Name())). Type(getType(fi.Name(), linkIsAcquisition)). Build()). Build()) } return feedBuilder.Build() } func getRel(name string, acquisition bool) (rel string) { rel = "subsection" if !acquisition { return } ext := filepath.Ext(name) if rel = "http://opds-spec.org/acquisition"; ext == ".png" || ext == ".jpg" || ext == ".jpeg" || ext == ".gift" { rel = "http://opds-spec.org/image/thumbnail" } return } func getType(name string, acquisition bool) (linkType string) { linkType = acquisitionType if !acquisition { return } ext := filepath.Ext(name) linkType = mime.TypeByExtension(ext) return } func getHref(req *http.Request, name string) string { return filepath.Join(req.URL.EscapedPath(), url.PathEscape(name)) } func isAcquisition(dirpath string) bool { fi, _ := os.Stat(dirpath) if isFile(fi) { return false } fis, _ := ioutil.ReadDir(dirpath) for _, fi := range fis { if isFile(fi) { return true } } return false } func isFile(fi os.FileInfo) bool { return !fi.IsDir() } func errorHandler(f func(http.ResponseWriter, *http.Request) error) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { err := f(w, r) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) log.Printf("handling %q: %v", r.RequestURI, err) } } }