celogeek/go-comic-converter
1
0
Fork 0
mirror of https://github.com/celogeek/go-comic-converter.git synced 2025-05-24 07:42:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

support limit and split

Browse Source
This commit is contained in:
Celogeek 2022-12-27 19:59:17 +01:00
parent 6dcaf4b7f3
commit ba7291eb65
Signed by: celogeek
GPG Key ID: E6B7BDCFC446233A
2 changed files with 111 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

108
internal/epub/core.go
View File

@ -20,7 +20,7 @@ import (
imageconverter "go-comic-converter/internal/image-converter"
)
type Images struct {
type Image struct {
Id int
Data []byte
Width int
@ -44,10 +44,17 @@ type EPub struct {
Error error
ImagesCount int
ProcessingImages func() chan *Images
ProcessingImages func() chan *Image
TemplateProcessor *template.Template
}
type EpubPart struct {
Idx int
Cover *Image
Images []*Image
Suffix string
}
func NewEpub(path string) *EPub {
uid, err := uuid.NewV4()
if err != nil {
@ -178,16 +185,16 @@ func (e *EPub) LoadDir(dirname string) *EPub {
todo := make(chan *Todo)
e.ProcessingImages = func() chan *Images {
e.ProcessingImages = func() chan *Image {
wg := &sync.WaitGroup{}
results := make(chan *Images)
results := make(chan *Image)
for i := 0; i < runtime.NumCPU(); i++ {
wg.Add(1)
go func() {
defer wg.Done()
for task := range todo {
data, w, h := imageconverter.Convert(task.Path, e.Crop, e.ViewWidth, e.ViewHeight, e.Quality)
results <- &Images{
results <- &Image{
task.Id,
data,
w,
@ -214,17 +221,8 @@ func (e *EPub) LoadDir(dirname string) *EPub {
return e
}
func (e *EPub) Write() error {
if e.Error != nil {
return e.Error
}
w, err := os.Create(e.Path)
if err != nil {
return err
}
images := make([]*Images, e.ImagesCount)
func (e *EPub) GetParts() <-chan *EpubPart {
images := make([]*Image, e.ImagesCount)
totalSize := 0
bar := progressbar.Default(int64(e.ImagesCount), "Processing")
for img := range e.ProcessingImages() {
@ -234,41 +232,97 @@ func (e *EPub) Write() error {
}
bar.Close()
epubPart := make(chan *EpubPart)
go func() {
defer close(epubPart)
cover := images[0]
images = images[1:]
fmt.Println("Limit: ", e.LimitMb, e.LimitMb == 0)
if e.LimitMb == 0 {
epubPart <- &EpubPart{
Idx: 1,
Cover: cover,
Images: images,
Suffix: "",
}
return
}
fmt.Println(len(images))
fmt.Println("Total Size:", totalSize)
maxSize := e.LimitMb * 1024 * 1024
currentSize := 512*1024 + len(cover.Data)
currentImages := make([]*Image, 0)
part := 1
for _, img := range images {
if len(currentImages) > 0 && currentSize+len(img.Data) > maxSize {
epubPart <- &EpubPart{
Idx: part,
Cover: cover,
Images: currentImages,
Suffix: fmt.Sprintf(" PART_%03d", part),
}
part += 1
currentSize = 512*1024 + len(cover.Data)
currentImages = make([]*Image, 0)
}
currentSize += len(img.Data)
currentImages = append(currentImages, img)
}
if len(currentImages) > 0 {
epubPart <- &EpubPart{
Idx: part,
Cover: cover,
Images: currentImages,
Suffix: fmt.Sprintf(" PART_%03d", part),
}
}
}()
return epubPart
}
func (e *EPub) Write() error {
if e.Error != nil {
return e.Error
}
type ZipContent struct {
Name string
Content any
}
for part := range e.GetParts() {
fmt.Printf("Writing part %d...\n", part.Idx)
ext := filepath.Ext(e.Path)
path := fmt.Sprintf("%s%s%s", e.Path[0:len(e.Path)-len(ext)], part.Suffix, ext)
w, err := os.Create(path)
if err != nil {
return err
}
zipContent := []ZipContent{
{"mimetype", TEMPLATE_MIME_TYPE},
{"META-INF/container.xml", gohtml.Format(TEMPLATE_CONTAINER)},
{"OEBPS/content.opf", e.Render(TEMPLATE_CONTENT, map[string]any{"Info": e, "Images": images})},
{"OEBPS/toc.ncx", e.Render(TEMPLATE_TOC, map[string]any{"Info": e, "Images": images})},
{"OEBPS/nav.xhtml", e.Render(TEMPLATE_NAV, map[string]any{"Info": e, "Images": images})},
{"OEBPS/content.opf", e.Render(TEMPLATE_CONTENT, map[string]any{"Info": e, "Images": part.Images})},
{"OEBPS/toc.ncx", e.Render(TEMPLATE_TOC, map[string]any{"Info": e, "Images": part.Images})},
{"OEBPS/nav.xhtml", e.Render(TEMPLATE_NAV, map[string]any{"Info": e, "Images": part.Images})},
{"OEBPS/Text/style.css", TEMPLATE_STYLE},
{"OEBPS/Text/cover.xhtml", e.Render(TEMPLATE_TEXT, map[string]any{
"Id": "cover",
"Width": cover.Width,
"Height": cover.Height,
"Width": part.Cover.Width,
"Height": part.Cover.Height,
})},
{"OEBPS/Images/cover.jpg", cover.Data},
{"OEBPS/Images/cover.jpg", part.Cover.Data},
}
wz := zip.NewWriter(w)
defer wz.Close()
for _, content := range zipContent {
if err := e.WriteFile(wz, content.Name, content.Content); err != nil {
return err
}
}
for _, img := range images {
for _, img := range part.Images {
text := fmt.Sprintf("OEBPS/Text/%d.xhtml", img.Id)
image := fmt.Sprintf("OEBPS/Images/%d.jpg", img.Id)
if err := e.WriteFile(wz, text, e.Render(TEMPLATE_TEXT, img)); err != nil {
@ -278,6 +332,8 @@ func (e *EPub) Write() error {
return err
}
}
wz.Close()
}
return nil
}

9
main.go
View File

@ -80,7 +80,7 @@ func main() {
flag.StringVar(&opt.Title, "title", "", "Title of the epub")
flag.IntVar(&opt.Quality, "quality", 85, "Quality of the image: Default 75")
flag.BoolVar(&opt.NoCrop, "nocrop", false, "Disable cropping: Default false")
flag.IntVar(&opt.LimitMb, "limitmb", 0, "Limit size of the ePub: Default nolimit")
flag.IntVar(&opt.LimitMb, "limitmb", 0, "Limit size of the ePub: Default nolimit, Minimum 20")
flag.Parse()
if opt.Input == "" || opt.Output == "" {
@ -95,6 +95,12 @@ func main() {
return
}
if opt.LimitMb > 0 && opt.LimitMb < 20 {
fmt.Println("LimitMb should be 0 or >= 20")
flag.Usage()
return
}
if opt.Title == "" {
opt.Title = filepath.Base(opt.Input)
}
@ -105,6 +111,7 @@ func main() {
SetSize(profile.Width, profile.Height).
SetQuality(opt.Quality).
SetCrop(!opt.NoCrop).
SetLimitMb(opt.LimitMb).
SetTitle(opt.Title).
SetAuthor(opt.Author).
LoadDir(opt.Input).