mirror of
https://github.com/go-shiori/shiori.git
synced 2025-09-13 08:24:49 +08:00
876d27f337
* refactor: base http server stdlib * refactor: swagger and frontend routes * fix: use global middlewares * refactor: removed gin from testutils * fix: object references in legacy webserver * refactor: legacy, swagger and system handlers * fix: added verbs to handlers * fix: server handlers ordering * refactor: bookmarks handlers * refactor: system api routes * tests: bookmark handlers * refactor: migrated api auth routes * chore: remove unused middlewares * docs: add swagger docs to refactored system api * chore: remove old auth routes * refactor: account apis * chore: removed old handlers * fix: api v1 handlers missing middlewares * refactor: migrated tag list route * refactor: bookmark routes * refactor: remove gin * chore: make styles * test: fixed tests * test: generate binary file without text * fix: global middleware missing from system api handler * fix: incorrect api handler * chore: avoid logging screenshot contents * tests: bookmarks domain * tests: shortcuts * test: missing tests * tests: server tests * test: remove test using syscall to avoid windows errors * chore: added middlewares
91 lines
2.9 KiB
Go
91 lines
2.9 KiB
Go
package core
|
|
|
|
import (
|
|
"os"
|
|
fp "path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
epub "github.com/go-shiori/go-epub"
|
|
"github.com/go-shiori/shiori/internal/model"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// GenerateEbook receives a `ProcessRequest` and generates an ebook file in the destination path specified.
|
|
// The destination path `dstPath` should include file name with ".epub" extension
|
|
// The bookmark model will be used to update the UI based on whether this function is successful or not.
|
|
func GenerateEbook(deps model.Dependencies, req ProcessRequest, dstPath string) (book model.BookmarkDTO, err error) {
|
|
book = req.Bookmark
|
|
|
|
// Make sure bookmark ID is defined
|
|
if book.ID == 0 {
|
|
return book, errors.New("bookmark ID is not valid")
|
|
}
|
|
|
|
// Get current state of bookmark cheak archive and thumb
|
|
strID := strconv.Itoa(book.ID)
|
|
|
|
bookmarkThumbnailPath := model.GetThumbnailPath(&book)
|
|
bookmarkArchivePath := model.GetArchivePath(&book)
|
|
|
|
if deps.Domains().Storage().FileExists(bookmarkThumbnailPath) {
|
|
book.ImageURL = fp.Join("/", "bookmark", strID, "thumb")
|
|
}
|
|
|
|
if deps.Domains().Storage().FileExists(bookmarkArchivePath) {
|
|
book.HasArchive = true
|
|
}
|
|
|
|
// This function create ebook from reader mode of bookmark so
|
|
// we can't create ebook from PDF so we return error here if bookmark is a pdf
|
|
contentType := req.ContentType
|
|
if strings.Contains(contentType, "application/pdf") {
|
|
return book, errors.New("can't create ebook for pdf")
|
|
}
|
|
|
|
// Create temporary epub file
|
|
tmpFile, err := os.CreateTemp("", "ebook")
|
|
if err != nil {
|
|
return book, errors.Wrap(err, "can't create temporary EPUB file")
|
|
}
|
|
defer os.Remove(tmpFile.Name())
|
|
|
|
// Create last line of ebook
|
|
lastline := `<hr/><p style="text-align:center">Generated By <a href="https://github.com/go-shiori/shiori">Shiori</a> From <a href="` + book.URL + `">This Page</a></p>`
|
|
|
|
// Create ebook
|
|
ebook, err := epub.NewEpub(book.Title)
|
|
if err != nil {
|
|
return book, errors.Wrap(err, "can't create EPUB")
|
|
}
|
|
|
|
ebook.SetTitle(book.Title)
|
|
ebook.SetAuthor(book.Author)
|
|
if deps.Domains().Storage().FileExists(bookmarkThumbnailPath) {
|
|
// TODO: Use `deps.Domains.Storage` to retrieve the file.
|
|
absoluteCoverPath := fp.Join(deps.Config().Storage.DataDir, bookmarkThumbnailPath)
|
|
coverPath, _ := ebook.AddImage(absoluteCoverPath, "cover.jpg")
|
|
ebook.SetCover(coverPath, "")
|
|
}
|
|
ebook.SetDescription(book.Excerpt)
|
|
_, err = ebook.AddSection(`<h1 style="text-align:center"> `+book.Title+` </h1>`+book.HTML+lastline, book.Title, "", "")
|
|
if err != nil {
|
|
return book, errors.Wrap(err, "can't add ebook Section")
|
|
}
|
|
ebook.EmbedImages()
|
|
err = ebook.Write(tmpFile.Name())
|
|
if err != nil {
|
|
return book, errors.Wrap(err, "can't create ebook file")
|
|
}
|
|
|
|
defer tmpFile.Close()
|
|
|
|
// If everything go well we move ebook to dstPath
|
|
err = deps.Domains().Storage().WriteFile(dstPath, tmpFile)
|
|
if err != nil {
|
|
return book, errors.Wrap(err, "failed move ebook to destination")
|
|
}
|
|
|
|
book.HasEbook = true
|
|
return book, nil
|
|
}
|