Commit of all existing code.

2024-12-06 11:55:31 -07:00
commit 1f7234997b
17 changed files with 498 additions and 0 deletions
@@ -0,0 +1,126 @@
+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))
+}