Archived
127 lines
3.0 KiB
Go
Executable File
127 lines
3.0 KiB
Go
Executable File
package routes
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"encoding/json"
|
|
"encoding/xml"
|
|
"errors"
|
|
"frostfire.io/dir2opdspse/app/opds"
|
|
"frostfire.io/dir2opdspse/app/utils"
|
|
"github.com/gorilla/mux"
|
|
"github.com/urfave/cli/v2"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"regexp"
|
|
"slices"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
func Init(c *cli.Context, r *mux.Router) {
|
|
r.HandleFunc("/", Home).Methods("GET")
|
|
r.HandleFunc("/stream/{manga}/{page}", Stream).Methods("GET")
|
|
r.HandleFunc("/test/{path:.*}/{page}", Test).Methods("GET")
|
|
}
|
|
|
|
func Home(w http.ResponseWriter, r *http.Request) {
|
|
feed := opds.FeedBuilder.
|
|
ID("dir2opds").
|
|
Title("dir2OPDS").
|
|
Updated(time.Now()).
|
|
AddLink(opds.LinkBuilder.
|
|
Rel("start").
|
|
Href("/").
|
|
Type("application/atom+xml;profile=opds-catalog;kind=navigation").
|
|
Build())
|
|
|
|
dirEntries, _ := os.ReadDir("books")
|
|
for _, entry := range dirEntries {
|
|
if !entry.IsDir() {
|
|
zFile, _ := zip.OpenReader(path.Join("books", entry.Name()))
|
|
//zCount := len(zFile.File)
|
|
_ = zFile.Close()
|
|
|
|
feed = feed.AddEntry(opds.EntryBuilder.
|
|
ID(utils.FileBasename(entry.Name())).
|
|
Title(utils.FileBasename(entry.Name())).
|
|
AddLink(opds.LinkBuilder.
|
|
Rel("http://opds-spec.org/acquisition").
|
|
Href("/manga/" + entry.Name()).
|
|
Type("application/x-cbz").
|
|
Title(utils.FileBasename(entry.Name())).
|
|
Build()).
|
|
AddLink(opds.LinkBuilder.
|
|
Rel("http://opds-spec.org/image").
|
|
Href("/stream/" + entry.Name() + "/0").
|
|
Type("image/png").
|
|
Build()).
|
|
AddLink(opds.LinkBuilder.
|
|
Rel("http://opds-spec.org/image/thumbnail").
|
|
Href("/stream/" + entry.Name() + "/0").
|
|
Type("image/png").
|
|
Build()).
|
|
AddLink(opds.LinkBuilder.
|
|
Rel("http://vaemendis.net/opds-pse/stream").
|
|
Href("/stream/" + entry.Name() + "/{pageNumber}").
|
|
Type("image/png").
|
|
//PseCount(10).
|
|
Build()).
|
|
Build())
|
|
}
|
|
}
|
|
data, err := xml.MarshalIndent(feed.Build(), "", " ")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
http.ServeContent(w, r, "feed.xml", time.Now(), bytes.NewReader(data))
|
|
}
|
|
|
|
func Stream(w http.ResponseWriter, r *http.Request) {
|
|
re := regexp.MustCompile(`(\d+)(.*)`)
|
|
vars := mux.Vars(r)
|
|
if len(re.FindStringIndex(vars["page"])) <= 0 {
|
|
w.WriteHeader(404)
|
|
return
|
|
}
|
|
|
|
pageNumber := re.FindStringSubmatch(vars["page"])[1]
|
|
|
|
mangaPath := path.Join("books", vars["manga"])
|
|
|
|
if _, err := os.Stat(mangaPath); errors.Is(err, os.ErrNotExist) {
|
|
w.WriteHeader(404)
|
|
return
|
|
}
|
|
|
|
zf, err := zip.OpenReader(mangaPath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
pageIndex := slices.IndexFunc(zf.File, func(f *zip.File) bool {
|
|
page, _ := strconv.Atoi(pageNumber)
|
|
return f.Name == (strconv.Itoa(page+1) + ".png")
|
|
})
|
|
|
|
if pageIndex == -1 {
|
|
w.WriteHeader(404)
|
|
return
|
|
}
|
|
|
|
imf, _ := zf.File[pageIndex].Open()
|
|
imageBytes, _ := io.ReadAll(imf)
|
|
_ = imf.Close()
|
|
_ = zf.Close()
|
|
|
|
http.ServeContent(w, r, zf.File[0].Name, time.Now(), bytes.NewReader(imageBytes))
|
|
}
|
|
|
|
func Test(w http.ResponseWriter, r *http.Request) {
|
|
data, _ := json.MarshalIndent(r.URL, "", " ")
|
|
http.ServeContent(w, r, "data.json", time.Now(), bytes.NewReader(data))
|
|
}
|