mirror of
https://github.com/celogeek/go-comic-converter.git
synced 2025-05-24 15:52:38 +02:00
Compare commits
12 Commits
4c9644e9b3
...
f445a26993
Author | SHA1 | Date | |
---|---|---|---|
f445a26993 | |||
0946696a8f | |||
5c1728dff8 | |||
4a12fe5715 | |||
f34ba5a37c | |||
18a31c43f7 | |||
3a0f37d5f4 | |||
66b66a6b76 | |||
9601e0a9b4 | |||
475af92ce1 | |||
80427dca67 | |||
1250deb53f |
@ -14,17 +14,14 @@ import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"regexp"
|
||||
"runtime"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/converter/options"
|
||||
)
|
||||
|
||||
type Converter struct {
|
||||
Options *options.Options
|
||||
type converter struct {
|
||||
Options *converterOptions
|
||||
Cmd *flag.FlagSet
|
||||
|
||||
order []converterOrder
|
||||
@ -33,10 +30,10 @@ type Converter struct {
|
||||
}
|
||||
|
||||
// Create a new parser
|
||||
func New() *Converter {
|
||||
options := options.New()
|
||||
func New() *converter {
|
||||
options := newOptions()
|
||||
cmd := flag.NewFlagSet("go-comic-converter", flag.ExitOnError)
|
||||
conv := &Converter{
|
||||
conv := &converter{
|
||||
Options: options,
|
||||
Cmd: cmd,
|
||||
order: make([]converterOrder, 0),
|
||||
@ -64,169 +61,76 @@ func New() *Converter {
|
||||
}
|
||||
|
||||
// Load default options (config + default)
|
||||
func (c *Converter) LoadConfig() error {
|
||||
func (c *converter) LoadConfig() error {
|
||||
return c.Options.LoadConfig()
|
||||
}
|
||||
|
||||
// Create a new section of config
|
||||
func (c *Converter) AddSection(section string) {
|
||||
c.order = append(c.order, converterOrderSection{value: section})
|
||||
}
|
||||
|
||||
// Add a string parameter
|
||||
func (c *Converter) AddStringParam(p *string, name string, value string, usage string) {
|
||||
c.Cmd.StringVar(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name, isString: true})
|
||||
}
|
||||
|
||||
// Add an integer parameter
|
||||
func (c *Converter) AddIntParam(p *int, name string, value int, usage string) {
|
||||
c.Cmd.IntVar(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name})
|
||||
}
|
||||
|
||||
// Add an float parameter
|
||||
func (c *Converter) AddFloatParam(p *float64, name string, value float64, usage string) {
|
||||
c.Cmd.Float64Var(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name})
|
||||
}
|
||||
|
||||
// Add a boolean parameter
|
||||
func (c *Converter) AddBoolParam(p *bool, name string, value bool, usage string) {
|
||||
c.Cmd.BoolVar(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name})
|
||||
}
|
||||
|
||||
// Initialize the parser with all section and parameter.
|
||||
func (c *Converter) InitParse() {
|
||||
c.AddSection("Output")
|
||||
c.AddStringParam(&c.Options.Input, "input", "", "Source of comic to convert: directory, cbz, zip, cbr, rar, pdf")
|
||||
c.AddStringParam(&c.Options.Output, "output", "", "Output of the EPUB (directory or EPUB): (default [INPUT].epub)")
|
||||
c.AddStringParam(&c.Options.Author, "author", "GO Comic Converter", "Author of the EPUB")
|
||||
c.AddStringParam(&c.Options.Title, "title", "", "Title of the EPUB")
|
||||
func (c *converter) InitParse() {
|
||||
c.addSection("Output")
|
||||
c.addStringParam(&c.Options.Input, "input", "", "Source of comic to convert: directory, cbz, zip, cbr, rar, pdf")
|
||||
c.addStringParam(&c.Options.Output, "output", "", "Output of the EPUB (directory or EPUB): (default [INPUT].epub)")
|
||||
c.addStringParam(&c.Options.Author, "author", "GO Comic Converter", "Author of the EPUB")
|
||||
c.addStringParam(&c.Options.Title, "title", "", "Title of the EPUB")
|
||||
|
||||
c.AddSection("Config")
|
||||
c.AddStringParam(&c.Options.Profile, "profile", c.Options.Profile, fmt.Sprintf("Profile to use: \n%s", c.Options.AvailableProfiles()))
|
||||
c.AddIntParam(&c.Options.Quality, "quality", c.Options.Quality, "Quality of the image")
|
||||
c.AddBoolParam(&c.Options.Grayscale, "grayscale", c.Options.Grayscale, "Grayscale image. Ideal for eInk devices.")
|
||||
c.AddIntParam(&c.Options.GrayscaleMode, "grayscale-mode", c.Options.GrayscaleMode, "Grayscale Mode\n0 = normal\n1 = average\n2 = luminance")
|
||||
c.AddBoolParam(&c.Options.Crop, "crop", c.Options.Crop, "Crop images")
|
||||
c.AddIntParam(&c.Options.CropRatioLeft, "crop-ratio-left", c.Options.CropRatioLeft, "Crop ratio left: ratio of pixels allow to be non blank while cutting on the left.")
|
||||
c.AddIntParam(&c.Options.CropRatioUp, "crop-ratio-up", c.Options.CropRatioUp, "Crop ratio up: ratio of pixels allow to be non blank while cutting on the top.")
|
||||
c.AddIntParam(&c.Options.CropRatioRight, "crop-ratio-right", c.Options.CropRatioRight, "Crop ratio right: ratio of pixels allow to be non blank while cutting on the right.")
|
||||
c.AddIntParam(&c.Options.CropRatioBottom, "crop-ratio-bottom", c.Options.CropRatioBottom, "Crop ratio bottom: ratio of pixels allow to be non blank while cutting on the bottom.")
|
||||
c.AddIntParam(&c.Options.Brightness, "brightness", c.Options.Brightness, "Brightness readjustement: between -100 and 100, > 0 lighter, < 0 darker")
|
||||
c.AddIntParam(&c.Options.Contrast, "contrast", c.Options.Contrast, "Contrast readjustement: between -100 and 100, > 0 more contrast, < 0 less contrast")
|
||||
c.AddBoolParam(&c.Options.AutoContrast, "autocontrast", c.Options.AutoContrast, "Improve contrast automatically")
|
||||
c.AddBoolParam(&c.Options.AutoRotate, "autorotate", c.Options.AutoRotate, "Auto Rotate page when width > height")
|
||||
c.AddBoolParam(&c.Options.AutoSplitDoublePage, "autosplitdoublepage", c.Options.AutoSplitDoublePage, "Auto Split double page when width > height")
|
||||
c.AddBoolParam(&c.Options.KeepDoublePageIfSplitted, "keepdoublepageifsplitted", c.Options.KeepDoublePageIfSplitted, "Keep the double page if splitted")
|
||||
c.AddBoolParam(&c.Options.NoBlankImage, "noblankimage", c.Options.NoBlankImage, "Remove blank image")
|
||||
c.AddBoolParam(&c.Options.Manga, "manga", c.Options.Manga, "Manga mode (right to left)")
|
||||
c.AddBoolParam(&c.Options.HasCover, "hascover", c.Options.HasCover, "Has cover. Indicate if your comic have a cover. The first page will be used as a cover and include after the title.")
|
||||
c.AddIntParam(&c.Options.LimitMb, "limitmb", c.Options.LimitMb, "Limit size of the EPUB: Default nolimit (0), Minimum 20")
|
||||
c.AddBoolParam(&c.Options.StripFirstDirectoryFromToc, "strip", c.Options.StripFirstDirectoryFromToc, "Strip first directory from the TOC if only 1")
|
||||
c.AddIntParam(&c.Options.SortPathMode, "sort", c.Options.SortPathMode, "Sort path mode\n0 = alpha for path and file\n1 = alphanum for path and alpha for file\n2 = alphanum for path and file")
|
||||
c.AddStringParam(&c.Options.ForegroundColor, "foreground-color", c.Options.ForegroundColor, "Foreground color in hexa format RGB. Black=000, White=FFF")
|
||||
c.AddStringParam(&c.Options.BackgroundColor, "background-color", c.Options.BackgroundColor, "Background color in hexa format RGB. Black=000, White=FFF, Light Gray=DDD, Dark Gray=777")
|
||||
c.AddBoolParam(&c.Options.NoResize, "noresize", c.Options.NoResize, "Do not reduce image size if exceed device size")
|
||||
c.AddStringParam(&c.Options.Format, "format", c.Options.Format, "Format of output images: jpeg (lossy), png (lossless)")
|
||||
c.AddFloatParam(&c.Options.AspectRatio, "aspect-ratio", c.Options.AspectRatio, "Aspect ratio (height/width) of the output\n -1 = same as device\n 0 = same as source\n1.6 = amazon advice for kindle")
|
||||
c.AddBoolParam(&c.Options.PortraitOnly, "portrait-only", c.Options.PortraitOnly, "Portrait only: force orientation to portrait only.")
|
||||
c.AddIntParam(&c.Options.TitlePage, "titlepage", c.Options.TitlePage, "Title page\n0 = never\n1 = always\n2 = only if epub is splitted")
|
||||
c.addSection("Config")
|
||||
c.addStringParam(&c.Options.Profile, "profile", c.Options.Profile, fmt.Sprintf("Profile to use: \n%s", c.Options.AvailableProfiles()))
|
||||
c.addIntParam(&c.Options.Quality, "quality", c.Options.Quality, "Quality of the image")
|
||||
c.addBoolParam(&c.Options.Grayscale, "grayscale", c.Options.Grayscale, "Grayscale image. Ideal for eInk devices.")
|
||||
c.addIntParam(&c.Options.GrayscaleMode, "grayscale-mode", c.Options.GrayscaleMode, "Grayscale Mode\n0 = normal\n1 = average\n2 = luminance")
|
||||
c.addBoolParam(&c.Options.Crop, "crop", c.Options.Crop, "Crop images")
|
||||
c.addIntParam(&c.Options.CropRatioLeft, "crop-ratio-left", c.Options.CropRatioLeft, "Crop ratio left: ratio of pixels allow to be non blank while cutting on the left.")
|
||||
c.addIntParam(&c.Options.CropRatioUp, "crop-ratio-up", c.Options.CropRatioUp, "Crop ratio up: ratio of pixels allow to be non blank while cutting on the top.")
|
||||
c.addIntParam(&c.Options.CropRatioRight, "crop-ratio-right", c.Options.CropRatioRight, "Crop ratio right: ratio of pixels allow to be non blank while cutting on the right.")
|
||||
c.addIntParam(&c.Options.CropRatioBottom, "crop-ratio-bottom", c.Options.CropRatioBottom, "Crop ratio bottom: ratio of pixels allow to be non blank while cutting on the bottom.")
|
||||
c.addIntParam(&c.Options.Brightness, "brightness", c.Options.Brightness, "Brightness readjustement: between -100 and 100, > 0 lighter, < 0 darker")
|
||||
c.addIntParam(&c.Options.Contrast, "contrast", c.Options.Contrast, "Contrast readjustement: between -100 and 100, > 0 more contrast, < 0 less contrast")
|
||||
c.addBoolParam(&c.Options.AutoContrast, "autocontrast", c.Options.AutoContrast, "Improve contrast automatically")
|
||||
c.addBoolParam(&c.Options.AutoRotate, "autorotate", c.Options.AutoRotate, "Auto Rotate page when width > height")
|
||||
c.addBoolParam(&c.Options.AutoSplitDoublePage, "autosplitdoublepage", c.Options.AutoSplitDoublePage, "Auto Split double page when width > height")
|
||||
c.addBoolParam(&c.Options.KeepDoublePageIfSplitted, "keepdoublepageifsplitted", c.Options.KeepDoublePageIfSplitted, "Keep the double page if splitted")
|
||||
c.addBoolParam(&c.Options.NoBlankImage, "noblankimage", c.Options.NoBlankImage, "Remove blank image")
|
||||
c.addBoolParam(&c.Options.Manga, "manga", c.Options.Manga, "Manga mode (right to left)")
|
||||
c.addBoolParam(&c.Options.HasCover, "hascover", c.Options.HasCover, "Has cover. Indicate if your comic have a cover. The first page will be used as a cover and include after the title.")
|
||||
c.addIntParam(&c.Options.LimitMb, "limitmb", c.Options.LimitMb, "Limit size of the EPUB: Default nolimit (0), Minimum 20")
|
||||
c.addBoolParam(&c.Options.StripFirstDirectoryFromToc, "strip", c.Options.StripFirstDirectoryFromToc, "Strip first directory from the TOC if only 1")
|
||||
c.addIntParam(&c.Options.SortPathMode, "sort", c.Options.SortPathMode, "Sort path mode\n0 = alpha for path and file\n1 = alphanum for path and alpha for file\n2 = alphanum for path and file")
|
||||
c.addStringParam(&c.Options.ForegroundColor, "foreground-color", c.Options.ForegroundColor, "Foreground color in hexa format RGB. Black=000, White=FFF")
|
||||
c.addStringParam(&c.Options.BackgroundColor, "background-color", c.Options.BackgroundColor, "Background color in hexa format RGB. Black=000, White=FFF, Light Gray=DDD, Dark Gray=777")
|
||||
c.addBoolParam(&c.Options.NoResize, "noresize", c.Options.NoResize, "Do not reduce image size if exceed device size")
|
||||
c.addStringParam(&c.Options.Format, "format", c.Options.Format, "Format of output images: jpeg (lossy), png (lossless)")
|
||||
c.addFloatParam(&c.Options.AspectRatio, "aspect-ratio", c.Options.AspectRatio, "Aspect ratio (height/width) of the output\n -1 = same as device\n 0 = same as source\n1.6 = amazon advice for kindle")
|
||||
c.addBoolParam(&c.Options.PortraitOnly, "portrait-only", c.Options.PortraitOnly, "Portrait only: force orientation to portrait only.")
|
||||
c.addIntParam(&c.Options.TitlePage, "titlepage", c.Options.TitlePage, "Title page\n0 = never\n1 = always\n2 = only if epub is splitted")
|
||||
|
||||
c.AddSection("Default config")
|
||||
c.AddBoolParam(&c.Options.Show, "show", false, "Show your default parameters")
|
||||
c.AddBoolParam(&c.Options.Save, "save", false, "Save your parameters as default")
|
||||
c.AddBoolParam(&c.Options.Reset, "reset", false, "Reset your parameters to default")
|
||||
c.addSection("Default config")
|
||||
c.addBoolParam(&c.Options.Show, "show", false, "Show your default parameters")
|
||||
c.addBoolParam(&c.Options.Save, "save", false, "Save your parameters as default")
|
||||
c.addBoolParam(&c.Options.Reset, "reset", false, "Reset your parameters to default")
|
||||
|
||||
c.AddSection("Shortcut")
|
||||
c.AddBoolParam(&c.Options.Auto, "auto", false, "Activate all automatic options")
|
||||
c.AddBoolParam(&c.Options.NoFilter, "nofilter", false, "Deactivate all filters")
|
||||
c.AddBoolParam(&c.Options.MaxQuality, "maxquality", false, "Max quality: color png + noresize")
|
||||
c.AddBoolParam(&c.Options.BestQuality, "bestquality", false, "Max quality: color jpg q100 + noresize")
|
||||
c.AddBoolParam(&c.Options.GreatQuality, "greatquality", false, "Max quality: grayscale jpg q90 + noresize")
|
||||
c.AddBoolParam(&c.Options.GoodQuality, "goodquality", false, "Max quality: grayscale jpg q90")
|
||||
c.addSection("Shortcut")
|
||||
c.addBoolParam(&c.Options.Auto, "auto", false, "Activate all automatic options")
|
||||
c.addBoolParam(&c.Options.NoFilter, "nofilter", false, "Deactivate all filters")
|
||||
c.addBoolParam(&c.Options.MaxQuality, "maxquality", false, "Max quality: color png + noresize")
|
||||
c.addBoolParam(&c.Options.BestQuality, "bestquality", false, "Max quality: color jpg q100 + noresize")
|
||||
c.addBoolParam(&c.Options.GreatQuality, "greatquality", false, "Max quality: grayscale jpg q90 + noresize")
|
||||
c.addBoolParam(&c.Options.GoodQuality, "goodquality", false, "Max quality: grayscale jpg q90")
|
||||
|
||||
c.AddSection("Compatibility")
|
||||
c.AddBoolParam(&c.Options.AppleBookCompatibility, "applebookcompatibility", c.Options.AppleBookCompatibility, "Apple book compatibility")
|
||||
c.addSection("Compatibility")
|
||||
c.addBoolParam(&c.Options.AppleBookCompatibility, "applebookcompatibility", c.Options.AppleBookCompatibility, "Apple book compatibility")
|
||||
|
||||
c.AddSection("Other")
|
||||
c.AddIntParam(&c.Options.Workers, "workers", runtime.NumCPU(), "Number of workers")
|
||||
c.AddBoolParam(&c.Options.Dry, "dry", false, "Dry run to show all options")
|
||||
c.AddBoolParam(&c.Options.DryVerbose, "dry-verbose", false, "Display also sorted files after the TOC")
|
||||
c.AddBoolParam(&c.Options.Quiet, "quiet", false, "Disable progress bar")
|
||||
c.AddBoolParam(&c.Options.Json, "json", false, "Output progression and information in Json format")
|
||||
c.AddBoolParam(&c.Options.Version, "version", false, "Show current and available version")
|
||||
c.AddBoolParam(&c.Options.Help, "help", false, "Show this help message")
|
||||
}
|
||||
|
||||
// Customize version of FlagSet.PrintDefaults
|
||||
func (c *Converter) Usage(isString bool, f *flag.Flag) string {
|
||||
var b strings.Builder
|
||||
fmt.Fprintf(&b, " -%s", f.Name) // Two spaces before -; see next two comments.
|
||||
name, usage := flag.UnquoteUsage(f)
|
||||
if len(name) > 0 {
|
||||
b.WriteString(" ")
|
||||
b.WriteString(name)
|
||||
}
|
||||
// Print the default value only if it differs to the zero value
|
||||
// for this flag type.
|
||||
if isZero, err := c.isZeroValue(f, f.DefValue); err != nil {
|
||||
c.isZeroValueErrs = append(c.isZeroValueErrs, err)
|
||||
} else if !isZero {
|
||||
if isString {
|
||||
fmt.Fprintf(&b, " (default %q)", f.DefValue)
|
||||
} else {
|
||||
fmt.Fprintf(&b, " (default %v)", f.DefValue)
|
||||
}
|
||||
}
|
||||
|
||||
// Boolean flags of one ASCII letter are so common we
|
||||
// treat them specially, putting their usage on the same line.
|
||||
if b.Len() <= 4 { // space, space, '-', 'x'.
|
||||
b.WriteString("\t")
|
||||
} else {
|
||||
// Four spaces before the tab triggers good alignment
|
||||
// for both 4- and 8-space tab stops.
|
||||
b.WriteString("\n \t")
|
||||
}
|
||||
b.WriteString(strings.ReplaceAll(usage, "\n", "\n \t"))
|
||||
|
||||
return b.String()
|
||||
}
|
||||
|
||||
// Taken from flag package as it is private and needed for usage.
|
||||
//
|
||||
// isZeroValue determines whether the string represents the zero
|
||||
// value for a flag.
|
||||
func (c *Converter) isZeroValue(f *flag.Flag, value string) (ok bool, err error) {
|
||||
// Build a zero value of the flag's Value type, and see if the
|
||||
// result of calling its String method equals the value passed in.
|
||||
// This works unless the Value type is itself an interface type.
|
||||
typ := reflect.TypeOf(f.Value)
|
||||
var z reflect.Value
|
||||
if typ.Kind() == reflect.Pointer {
|
||||
z = reflect.New(typ.Elem())
|
||||
} else {
|
||||
z = reflect.Zero(typ)
|
||||
}
|
||||
// Catch panics calling the String method, which shouldn't prevent the
|
||||
// usage message from being printed, but that we should report to the
|
||||
// user so that they know to fix their code.
|
||||
defer func() {
|
||||
if e := recover(); e != nil {
|
||||
if typ.Kind() == reflect.Pointer {
|
||||
typ = typ.Elem()
|
||||
}
|
||||
err = fmt.Errorf("panic calling String method on zero %v for flag %s: %v", typ, f.Name, e)
|
||||
}
|
||||
}()
|
||||
return value == z.Interface().(flag.Value).String(), nil
|
||||
c.addSection("Other")
|
||||
c.addIntParam(&c.Options.Workers, "workers", runtime.NumCPU(), "Number of workers")
|
||||
c.addBoolParam(&c.Options.Dry, "dry", false, "Dry run to show all options")
|
||||
c.addBoolParam(&c.Options.DryVerbose, "dry-verbose", false, "Display also sorted files after the TOC")
|
||||
c.addBoolParam(&c.Options.Quiet, "quiet", false, "Disable progress bar")
|
||||
c.addBoolParam(&c.Options.Json, "json", false, "Output progression and information in Json format")
|
||||
c.addBoolParam(&c.Options.Version, "version", false, "Show current and available version")
|
||||
c.addBoolParam(&c.Options.Help, "help", false, "Show this help message")
|
||||
}
|
||||
|
||||
// Parse all parameters
|
||||
func (c *Converter) Parse() {
|
||||
func (c *converter) Parse() {
|
||||
c.Cmd.Parse(os.Args[1:])
|
||||
if c.Options.Help {
|
||||
c.Cmd.Usage()
|
||||
@ -277,7 +181,7 @@ func (c *Converter) Parse() {
|
||||
}
|
||||
|
||||
// Check parameters
|
||||
func (c *Converter) Validate() error {
|
||||
func (c *converter) Validate() error {
|
||||
// Check input
|
||||
if c.Options.Input == "" {
|
||||
return errors.New("missing input")
|
||||
@ -393,14 +297,49 @@ func (c *Converter) Validate() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// Customize version of FlagSet.PrintDefaults
|
||||
func (c *converter) Usage(isString bool, f *flag.Flag) string {
|
||||
var b strings.Builder
|
||||
fmt.Fprintf(&b, " -%s", f.Name) // Two spaces before -; see next two comments.
|
||||
name, usage := flag.UnquoteUsage(f)
|
||||
if len(name) > 0 {
|
||||
b.WriteString(" ")
|
||||
b.WriteString(name)
|
||||
}
|
||||
// Print the default value only if it differs to the zero value
|
||||
// for this flag type.
|
||||
if isZero, err := c.isZeroValue(f, f.DefValue); err != nil {
|
||||
c.isZeroValueErrs = append(c.isZeroValueErrs, err)
|
||||
} else if !isZero {
|
||||
if isString {
|
||||
fmt.Fprintf(&b, " (default %q)", f.DefValue)
|
||||
} else {
|
||||
fmt.Fprintf(&b, " (default %v)", f.DefValue)
|
||||
}
|
||||
}
|
||||
|
||||
// Boolean flags of one ASCII letter are so common we
|
||||
// treat them specially, putting their usage on the same line.
|
||||
if b.Len() <= 4 { // space, space, '-', 'x'.
|
||||
b.WriteString("\t")
|
||||
} else {
|
||||
// Four spaces before the tab triggers good alignment
|
||||
// for both 4- and 8-space tab stops.
|
||||
b.WriteString("\n \t")
|
||||
}
|
||||
b.WriteString(strings.ReplaceAll(usage, "\n", "\n \t"))
|
||||
|
||||
return b.String()
|
||||
}
|
||||
|
||||
// Helper to show usage, err and exit 1
|
||||
func (c *Converter) Fatal(err error) {
|
||||
func (c *converter) Fatal(err error) {
|
||||
c.Cmd.Usage()
|
||||
fmt.Fprintf(os.Stderr, "\nError: %s\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
func (c *Converter) Stats() {
|
||||
func (c *converter) Stats() {
|
||||
// Display elapse time and memory usage
|
||||
elapse := time.Since(c.startAt).Round(time.Millisecond)
|
||||
var mem runtime.MemStats
|
||||
|
65
internal/converter/converter_helper.go
Normal file
65
internal/converter/converter_helper.go
Normal file
@ -0,0 +1,65 @@
|
||||
package converter
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"reflect"
|
||||
)
|
||||
|
||||
// Create a new section of config
|
||||
func (c *converter) addSection(section string) {
|
||||
c.order = append(c.order, converterOrderSection{value: section})
|
||||
}
|
||||
|
||||
// Add a string parameter
|
||||
func (c *converter) addStringParam(p *string, name string, value string, usage string) {
|
||||
c.Cmd.StringVar(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name, isString: true})
|
||||
}
|
||||
|
||||
// Add an integer parameter
|
||||
func (c *converter) addIntParam(p *int, name string, value int, usage string) {
|
||||
c.Cmd.IntVar(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name})
|
||||
}
|
||||
|
||||
// Add an float parameter
|
||||
func (c *converter) addFloatParam(p *float64, name string, value float64, usage string) {
|
||||
c.Cmd.Float64Var(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name})
|
||||
}
|
||||
|
||||
// Add a boolean parameter
|
||||
func (c *converter) addBoolParam(p *bool, name string, value bool, usage string) {
|
||||
c.Cmd.BoolVar(p, name, value, usage)
|
||||
c.order = append(c.order, converterOrderName{value: name})
|
||||
}
|
||||
|
||||
// Taken from flag package as it is private and needed for usage.
|
||||
//
|
||||
// isZeroValue determines whether the string represents the zero
|
||||
// value for a flag.
|
||||
func (c *converter) isZeroValue(f *flag.Flag, value string) (ok bool, err error) {
|
||||
// Build a zero value of the flag's Value type, and see if the
|
||||
// result of calling its String method equals the value passed in.
|
||||
// This works unless the Value type is itself an interface type.
|
||||
typ := reflect.TypeOf(f.Value)
|
||||
var z reflect.Value
|
||||
if typ.Kind() == reflect.Pointer {
|
||||
z = reflect.New(typ.Elem())
|
||||
} else {
|
||||
z = reflect.Zero(typ)
|
||||
}
|
||||
// Catch panics calling the String method, which shouldn't prevent the
|
||||
// usage message from being printed, but that we should report to the
|
||||
// user so that they know to fix their code.
|
||||
defer func() {
|
||||
if e := recover(); e != nil {
|
||||
if typ.Kind() == reflect.Pointer {
|
||||
typ = typ.Elem()
|
||||
}
|
||||
err = fmt.Errorf("panic calling String method on zero %v for flag %s: %v", typ, f.Name, e)
|
||||
}
|
||||
}()
|
||||
return value == z.Interface().(flag.Value).String(), nil
|
||||
}
|
@ -1,7 +1,7 @@
|
||||
/*
|
||||
Manage options with default value from config.
|
||||
*/
|
||||
package options
|
||||
package converter
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
@ -10,11 +10,10 @@ import (
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/converter/profiles"
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
type Options struct {
|
||||
type converterOptions struct {
|
||||
// Output
|
||||
Input string `yaml:"-"`
|
||||
Output string `yaml:"-"`
|
||||
@ -75,12 +74,12 @@ type Options struct {
|
||||
Help bool `yaml:"-"`
|
||||
|
||||
// Internal
|
||||
profiles profiles.Profiles
|
||||
profiles converterProfiles
|
||||
}
|
||||
|
||||
// Initialize default options.
|
||||
func New() *Options {
|
||||
return &Options{
|
||||
func newOptions() *converterOptions {
|
||||
return &converterOptions{
|
||||
Profile: "SR",
|
||||
Quality: 85,
|
||||
Grayscale: true,
|
||||
@ -97,15 +96,15 @@ func New() *Options {
|
||||
BackgroundColor: "FFF",
|
||||
Format: "jpeg",
|
||||
TitlePage: 1,
|
||||
profiles: profiles.New(),
|
||||
profiles: newProfiles(),
|
||||
}
|
||||
}
|
||||
|
||||
func (o *Options) Header() string {
|
||||
func (o *converterOptions) Header() string {
|
||||
return "Go Comic Converter\n\nOptions:"
|
||||
}
|
||||
|
||||
func (o *Options) String() string {
|
||||
func (o *converterOptions) String() string {
|
||||
var b strings.Builder
|
||||
b.WriteString(o.Header())
|
||||
for _, v := range []struct {
|
||||
@ -125,7 +124,7 @@ func (o *Options) String() string {
|
||||
return b.String()
|
||||
}
|
||||
|
||||
func (o *Options) MarshalJSON() ([]byte, error) {
|
||||
func (o *converterOptions) MarshalJSON() ([]byte, error) {
|
||||
out := map[string]any{
|
||||
"input": o.Input,
|
||||
"output": o.Output,
|
||||
@ -186,13 +185,13 @@ func (o *Options) MarshalJSON() ([]byte, error) {
|
||||
}
|
||||
|
||||
// Config file: ~/.go-comic-converter.yaml
|
||||
func (o *Options) FileName() string {
|
||||
func (o *converterOptions) FileName() string {
|
||||
home, _ := os.UserHomeDir()
|
||||
return filepath.Join(home, ".go-comic-converter.yaml")
|
||||
}
|
||||
|
||||
// Load config files
|
||||
func (o *Options) LoadConfig() error {
|
||||
func (o *converterOptions) LoadConfig() error {
|
||||
f, err := os.Open(o.FileName())
|
||||
if err != nil {
|
||||
return nil
|
||||
@ -207,7 +206,7 @@ func (o *Options) LoadConfig() error {
|
||||
}
|
||||
|
||||
// Get current settings for fields that can be saved
|
||||
func (o *Options) ShowConfig() string {
|
||||
func (o *converterOptions) ShowConfig() string {
|
||||
var profileDesc string
|
||||
profile := o.GetProfile()
|
||||
if profile != nil {
|
||||
@ -296,13 +295,13 @@ func (o *Options) ShowConfig() string {
|
||||
}
|
||||
|
||||
// reset all settings to default value
|
||||
func (o *Options) ResetConfig() error {
|
||||
New().SaveConfig()
|
||||
func (o *converterOptions) ResetConfig() error {
|
||||
newOptions().SaveConfig()
|
||||
return o.LoadConfig()
|
||||
}
|
||||
|
||||
// save all current settings as futur default value
|
||||
func (o *Options) SaveConfig() error {
|
||||
func (o *converterOptions) SaveConfig() error {
|
||||
f, err := os.Create(o.FileName())
|
||||
if err != nil {
|
||||
return err
|
||||
@ -312,11 +311,11 @@ func (o *Options) SaveConfig() error {
|
||||
}
|
||||
|
||||
// shortcut to get current profile
|
||||
func (o *Options) GetProfile() *profiles.Profile {
|
||||
func (o *converterOptions) GetProfile() *converterProfile {
|
||||
return o.profiles.Get(o.Profile)
|
||||
}
|
||||
|
||||
// all available profiles
|
||||
func (o *Options) AvailableProfiles() string {
|
||||
func (o *converterOptions) AvailableProfiles() string {
|
||||
return o.profiles.String()
|
||||
}
|
@ -1,25 +1,25 @@
|
||||
/*
|
||||
Manage supported profiles for go-comic-converter.
|
||||
*/
|
||||
package profiles
|
||||
package converter
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Profile struct {
|
||||
type converterProfile struct {
|
||||
Code string `json:"code"`
|
||||
Description string `json:"description"`
|
||||
Width int `json:"width"`
|
||||
Height int `json:"height"`
|
||||
}
|
||||
|
||||
type Profiles []Profile
|
||||
type converterProfiles []*converterProfile
|
||||
|
||||
// Initialize list of all supported profiles.
|
||||
func New() Profiles {
|
||||
return []Profile{
|
||||
func newProfiles() converterProfiles {
|
||||
return []*converterProfile{
|
||||
// High Resolution for Tablette
|
||||
{"HR", "High Resolution", 2400, 3840},
|
||||
{"SR", "Standard Resolution", 1200, 1920},
|
||||
@ -55,7 +55,7 @@ func New() Profiles {
|
||||
}
|
||||
}
|
||||
|
||||
func (p Profiles) String() string {
|
||||
func (p converterProfiles) String() string {
|
||||
s := make([]string, 0)
|
||||
for _, v := range p {
|
||||
s = append(s, fmt.Sprintf(
|
||||
@ -69,10 +69,10 @@ func (p Profiles) String() string {
|
||||
}
|
||||
|
||||
// Lookup profile by code
|
||||
func (p Profiles) Get(name string) *Profile {
|
||||
func (p converterProfiles) Get(name string) *converterProfile {
|
||||
for _, profile := range p {
|
||||
if profile.Code == name {
|
||||
return &profile
|
||||
return profile
|
||||
}
|
||||
}
|
||||
return nil
|
@ -15,34 +15,35 @@ import (
|
||||
"text/template"
|
||||
"time"
|
||||
|
||||
epubimage "github.com/celogeek/go-comic-converter/v2/internal/epub/image"
|
||||
epubimageprocessor "github.com/celogeek/go-comic-converter/v2/internal/epub/imageprocessor"
|
||||
epuboptions "github.com/celogeek/go-comic-converter/v2/internal/epub/options"
|
||||
epubprogress "github.com/celogeek/go-comic-converter/v2/internal/epub/progress"
|
||||
epubtemplates "github.com/celogeek/go-comic-converter/v2/internal/epub/templates"
|
||||
epubtree "github.com/celogeek/go-comic-converter/v2/internal/epub/tree"
|
||||
epubzip "github.com/celogeek/go-comic-converter/v2/internal/epub/zip"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubimage"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubimageprocessor"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubprogress"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubtemplates"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubtree"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubzip"
|
||||
"github.com/celogeek/go-comic-converter/v2/pkg/epuboptions"
|
||||
"github.com/gofrs/uuid"
|
||||
)
|
||||
|
||||
type ePub struct {
|
||||
*epuboptions.Options
|
||||
UID string
|
||||
Publisher string
|
||||
UpdatedAt string
|
||||
options *epuboptions.EPUBOptions
|
||||
|
||||
uid string
|
||||
publisher string
|
||||
updatedAt string
|
||||
|
||||
templateProcessor *template.Template
|
||||
imageProcessor *epubimageprocessor.EPUBImageProcessor
|
||||
}
|
||||
|
||||
type epubPart struct {
|
||||
Cover *epubimage.Image
|
||||
Images []*epubimage.Image
|
||||
Cover *epubimage.EPUBImage
|
||||
Images []*epubimage.EPUBImage
|
||||
Reader *zip.ReadCloser
|
||||
}
|
||||
|
||||
// initialize EPUB
|
||||
func New(options *epuboptions.Options) *ePub {
|
||||
func New(options *epuboptions.EPUBOptions) *ePub {
|
||||
uid := uuid.Must(uuid.NewV4())
|
||||
tmpl := template.New("parser")
|
||||
tmpl.Funcs(template.FuncMap{
|
||||
@ -51,10 +52,10 @@ func New(options *epuboptions.Options) *ePub {
|
||||
})
|
||||
|
||||
return &ePub{
|
||||
Options: options,
|
||||
UID: uid.String(),
|
||||
Publisher: "GO Comic Converter",
|
||||
UpdatedAt: time.Now().UTC().Format("2006-01-02T15:04:05Z"),
|
||||
options: options,
|
||||
uid: uid.String(),
|
||||
publisher: "GO Comic Converter",
|
||||
updatedAt: time.Now().UTC().Format("2006-01-02T15:04:05Z"),
|
||||
templateProcessor: tmpl,
|
||||
imageProcessor: epubimageprocessor.New(options),
|
||||
}
|
||||
@ -71,14 +72,14 @@ func (e *ePub) render(templateString string, data map[string]any) string {
|
||||
}
|
||||
|
||||
// write image to the zip
|
||||
func (e *ePub) writeImage(wz *epubzip.EPUBZip, img *epubimage.Image, zipImg *zip.File) error {
|
||||
func (e *ePub) writeImage(wz *epubzip.EPUBZip, img *epubimage.EPUBImage, zipImg *zip.File) error {
|
||||
err := wz.WriteContent(
|
||||
img.EPUBPagePath(),
|
||||
[]byte(e.render(epubtemplates.Text, map[string]any{
|
||||
"Title": fmt.Sprintf("Image %d Part %d", img.Id, img.Part),
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.Image.View.Width, e.Image.View.Height),
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.options.Image.View.Width, e.options.Image.View.Height),
|
||||
"ImagePath": img.ImgPath(),
|
||||
"ImageStyle": img.ImgStyle(e.Image.View.Width, e.Image.View.Height, ""),
|
||||
"ImageStyle": img.ImgStyle(e.options.Image.View.Width, e.options.Image.View.Height, ""),
|
||||
})),
|
||||
)
|
||||
if err == nil {
|
||||
@ -89,18 +90,18 @@ func (e *ePub) writeImage(wz *epubzip.EPUBZip, img *epubimage.Image, zipImg *zip
|
||||
}
|
||||
|
||||
// write blank page
|
||||
func (e *ePub) writeBlank(wz *epubzip.EPUBZip, img *epubimage.Image) error {
|
||||
func (e *ePub) writeBlank(wz *epubzip.EPUBZip, img *epubimage.EPUBImage) error {
|
||||
return wz.WriteContent(
|
||||
img.EPUBSpacePath(),
|
||||
[]byte(e.render(epubtemplates.Blank, map[string]any{
|
||||
"Title": fmt.Sprintf("Blank Page %d", img.Id),
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.Image.View.Width, e.Image.View.Height),
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.options.Image.View.Width, e.options.Image.View.Height),
|
||||
})),
|
||||
)
|
||||
}
|
||||
|
||||
// write title image
|
||||
func (e *ePub) writeCoverImage(wz *epubzip.EPUBZip, img *epubimage.Image, part, totalParts int) error {
|
||||
func (e *ePub) writeCoverImage(wz *epubzip.EPUBZip, img *epubimage.EPUBImage, part, totalParts int) error {
|
||||
title := "Cover"
|
||||
text := ""
|
||||
if totalParts > 1 {
|
||||
@ -112,9 +113,9 @@ func (e *ePub) writeCoverImage(wz *epubzip.EPUBZip, img *epubimage.Image, part,
|
||||
"OEBPS/Text/cover.xhtml",
|
||||
[]byte(e.render(epubtemplates.Text, map[string]any{
|
||||
"Title": title,
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.Image.View.Width, e.Image.View.Height),
|
||||
"ImagePath": fmt.Sprintf("Images/cover.%s", e.Image.Format),
|
||||
"ImageStyle": img.ImgStyle(e.Image.View.Width, e.Image.View.Height, ""),
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.options.Image.View.Width, e.options.Image.View.Height),
|
||||
"ImagePath": fmt.Sprintf("Images/cover.%s", e.options.Image.Format),
|
||||
"ImageStyle": img.ImgStyle(e.options.Image.View.Width, e.options.Image.View.Height, ""),
|
||||
})),
|
||||
); err != nil {
|
||||
return err
|
||||
@ -143,22 +144,22 @@ func (e *ePub) writeCoverImage(wz *epubzip.EPUBZip, img *epubimage.Image, part,
|
||||
}
|
||||
|
||||
// write title image
|
||||
func (e *ePub) writeTitleImage(wz *epubzip.EPUBZip, img *epubimage.Image, title string) error {
|
||||
func (e *ePub) writeTitleImage(wz *epubzip.EPUBZip, img *epubimage.EPUBImage, title string) error {
|
||||
titleAlign := ""
|
||||
if !e.Image.View.PortraitOnly {
|
||||
if e.Image.Manga {
|
||||
if !e.options.Image.View.PortraitOnly {
|
||||
if e.options.Image.Manga {
|
||||
titleAlign = "right:0"
|
||||
} else {
|
||||
titleAlign = "left:0"
|
||||
}
|
||||
}
|
||||
|
||||
if !e.Image.View.PortraitOnly {
|
||||
if !e.options.Image.View.PortraitOnly {
|
||||
if err := wz.WriteContent(
|
||||
"OEBPS/Text/space_title.xhtml",
|
||||
[]byte(e.render(epubtemplates.Blank, map[string]any{
|
||||
"Title": "Blank Page Title",
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.Image.View.Width, e.Image.View.Height),
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.options.Image.View.Width, e.options.Image.View.Height),
|
||||
})),
|
||||
); err != nil {
|
||||
return err
|
||||
@ -169,9 +170,9 @@ func (e *ePub) writeTitleImage(wz *epubzip.EPUBZip, img *epubimage.Image, title
|
||||
"OEBPS/Text/title.xhtml",
|
||||
[]byte(e.render(epubtemplates.Text, map[string]any{
|
||||
"Title": title,
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.Image.View.Width, e.Image.View.Height),
|
||||
"ImagePath": fmt.Sprintf("Images/title.%s", e.Image.Format),
|
||||
"ImageStyle": img.ImgStyle(e.Image.View.Width, e.Image.View.Height, titleAlign),
|
||||
"ViewPort": fmt.Sprintf("width=%d,height=%d", e.options.Image.View.Width, e.options.Image.View.Height),
|
||||
"ImagePath": fmt.Sprintf("Images/title.%s", e.options.Image.Format),
|
||||
"ImageStyle": img.ImgStyle(e.options.Image.View.Width, e.options.Image.View.Height, titleAlign),
|
||||
})),
|
||||
); err != nil {
|
||||
return err
|
||||
@ -199,7 +200,7 @@ func (e *ePub) writeTitleImage(wz *epubzip.EPUBZip, img *epubimage.Image, title
|
||||
}
|
||||
|
||||
// extract image and split it into part
|
||||
func (e *ePub) getParts() (parts []*epubPart, imgStorage *epubzip.EPUBZipStorageImageReader, err error) {
|
||||
func (e *ePub) getParts() (parts []*epubPart, imgStorage *epubzip.EPUBZipImageReader, err error) {
|
||||
images, err := e.imageProcessor.Load()
|
||||
|
||||
if err != nil {
|
||||
@ -216,11 +217,11 @@ func (e *ePub) getParts() (parts []*epubPart, imgStorage *epubzip.EPUBZipStorage
|
||||
|
||||
parts = make([]*epubPart, 0)
|
||||
cover := images[0]
|
||||
if e.Image.HasCover {
|
||||
if e.options.Image.HasCover {
|
||||
images = images[1:]
|
||||
}
|
||||
|
||||
if e.Dry {
|
||||
if e.options.Dry {
|
||||
parts = append(parts, &epubPart{
|
||||
Cover: cover,
|
||||
Images: images,
|
||||
@ -228,19 +229,19 @@ func (e *ePub) getParts() (parts []*epubPart, imgStorage *epubzip.EPUBZipStorage
|
||||
return parts, nil, nil
|
||||
}
|
||||
|
||||
imgStorage, err = epubzip.NewEPUBZipStorageImageReader(e.ImgStorage())
|
||||
imgStorage, err = epubzip.NewImageReader(e.options.Temp())
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
// compute size of the EPUB part and try to be as close as possible of the target
|
||||
maxSize := uint64(e.LimitMb * 1024 * 1024)
|
||||
maxSize := uint64(e.options.LimitMb * 1024 * 1024)
|
||||
xhtmlSize := uint64(1024)
|
||||
// descriptor files + title + cover
|
||||
baseSize := uint64(128*1024) + imgStorage.Size(cover.EPUBImgPath())*2
|
||||
|
||||
currentSize := baseSize
|
||||
currentImages := make([]*epubimage.Image, 0)
|
||||
currentImages := make([]*epubimage.EPUBImage, 0)
|
||||
part := 1
|
||||
|
||||
for _, img := range images {
|
||||
@ -252,7 +253,7 @@ func (e *ePub) getParts() (parts []*epubPart, imgStorage *epubzip.EPUBZipStorage
|
||||
})
|
||||
part += 1
|
||||
currentSize = baseSize
|
||||
currentImages = make([]*epubimage.Image, 0)
|
||||
currentImages = make([]*epubimage.EPUBImage, 0)
|
||||
}
|
||||
currentSize += imgSize
|
||||
currentImages = append(currentImages, img)
|
||||
@ -270,7 +271,7 @@ func (e *ePub) getParts() (parts []*epubPart, imgStorage *epubzip.EPUBZipStorage
|
||||
// create a tree from the directories.
|
||||
//
|
||||
// this is used to simulate the toc.
|
||||
func (e *ePub) getTree(images []*epubimage.Image, skip_files bool) string {
|
||||
func (e *ePub) getTree(images []*epubimage.EPUBImage, skip_files bool) string {
|
||||
t := epubtree.New()
|
||||
for _, img := range images {
|
||||
if skip_files {
|
||||
@ -280,7 +281,7 @@ func (e *ePub) getTree(images []*epubimage.Image, skip_files bool) string {
|
||||
}
|
||||
}
|
||||
c := t.Root()
|
||||
if skip_files && e.StripFirstDirectoryFromToc && len(c.Children) == 1 {
|
||||
if skip_files && e.options.StripFirstDirectoryFromToc && len(c.Children) == 1 {
|
||||
c = c.Children[0]
|
||||
}
|
||||
|
||||
@ -315,21 +316,21 @@ func (e *ePub) computeAspectRatio(epubParts []*epubPart) float64 {
|
||||
}
|
||||
|
||||
func (e *ePub) computeViewPort(epubParts []*epubPart) {
|
||||
if e.Image.View.AspectRatio == -1 {
|
||||
if e.options.Image.View.AspectRatio == -1 {
|
||||
return //keep device size
|
||||
}
|
||||
|
||||
// readjusting view port
|
||||
bestAspectRatio := e.Image.View.AspectRatio
|
||||
bestAspectRatio := e.options.Image.View.AspectRatio
|
||||
if bestAspectRatio == 0 {
|
||||
bestAspectRatio = e.computeAspectRatio(epubParts)
|
||||
}
|
||||
|
||||
viewWidth, viewHeight := int(float64(e.Image.View.Height)/bestAspectRatio), int(float64(e.Image.View.Width)*bestAspectRatio)
|
||||
if viewWidth > e.Image.View.Width {
|
||||
e.Image.View.Height = viewHeight
|
||||
viewWidth, viewHeight := int(float64(e.options.Image.View.Height)/bestAspectRatio), int(float64(e.options.Image.View.Width)*bestAspectRatio)
|
||||
if viewWidth > e.options.Image.View.Width {
|
||||
e.options.Image.View.Height = viewHeight
|
||||
} else {
|
||||
e.Image.View.Width = viewWidth
|
||||
e.options.Image.View.Width = viewWidth
|
||||
}
|
||||
}
|
||||
|
||||
@ -345,12 +346,12 @@ func (e *ePub) Write() error {
|
||||
return err
|
||||
}
|
||||
|
||||
if e.Dry {
|
||||
if e.options.Dry {
|
||||
p := epubParts[0]
|
||||
fmt.Fprintf(os.Stderr, "TOC:\n - %s\n%s\n", e.Title, e.getTree(p.Images, true))
|
||||
if e.DryVerbose {
|
||||
if e.Image.HasCover {
|
||||
fmt.Fprintf(os.Stderr, "Cover:\n%s\n", e.getTree([]*epubimage.Image{p.Cover}, false))
|
||||
fmt.Fprintf(os.Stderr, "TOC:\n - %s\n%s\n", e.options.Title, e.getTree(p.Images, true))
|
||||
if e.options.DryVerbose {
|
||||
if e.options.Image.HasCover {
|
||||
fmt.Fprintf(os.Stderr, "Cover:\n%s\n", e.getTree([]*epubimage.EPUBImage{p.Cover}, false))
|
||||
}
|
||||
fmt.Fprintf(os.Stderr, "Files:\n%s\n", e.getTree(p.Images, false))
|
||||
}
|
||||
@ -368,14 +369,14 @@ func (e *ePub) Write() error {
|
||||
Description: "Writing Part",
|
||||
CurrentJob: 2,
|
||||
TotalJob: 2,
|
||||
Quiet: e.Quiet,
|
||||
Json: e.Json,
|
||||
Quiet: e.options.Quiet,
|
||||
Json: e.options.Json,
|
||||
})
|
||||
|
||||
e.computeViewPort(epubParts)
|
||||
hasTitlePage := e.TitlePage == 1 || (e.TitlePage == 2 && totalParts > 1)
|
||||
hasTitlePage := e.options.TitlePage == 1 || (e.options.TitlePage == 2 && totalParts > 1)
|
||||
for i, part := range epubParts {
|
||||
ext := filepath.Ext(e.Output)
|
||||
ext := filepath.Ext(e.options.Output)
|
||||
suffix := ""
|
||||
if totalParts > 1 {
|
||||
fmtLen := len(fmt.Sprint(totalParts))
|
||||
@ -383,14 +384,14 @@ func (e *ePub) Write() error {
|
||||
suffix = fmt.Sprintf(fmtPart, i+1, totalParts)
|
||||
}
|
||||
|
||||
path := fmt.Sprintf("%s%s%s", e.Output[0:len(e.Output)-len(ext)], suffix, ext)
|
||||
path := fmt.Sprintf("%s%s%s", e.options.Output[0:len(e.options.Output)-len(ext)], suffix, ext)
|
||||
wz, err := epubzip.New(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer wz.Close()
|
||||
|
||||
title := e.Title
|
||||
title := e.options.Title
|
||||
if totalParts > 1 {
|
||||
title = fmt.Sprintf("%s [%d/%d]", title, i+1, totalParts)
|
||||
}
|
||||
@ -401,19 +402,19 @@ func (e *ePub) Write() error {
|
||||
{"OEBPS/content.opf", epubtemplates.Content(&epubtemplates.ContentOptions{
|
||||
Title: title,
|
||||
HasTitlePage: hasTitlePage,
|
||||
UID: e.UID,
|
||||
Author: e.Author,
|
||||
Publisher: e.Publisher,
|
||||
UpdatedAt: e.UpdatedAt,
|
||||
ImageOptions: e.Image,
|
||||
UID: e.uid,
|
||||
Author: e.options.Author,
|
||||
Publisher: e.publisher,
|
||||
UpdatedAt: e.updatedAt,
|
||||
ImageOptions: e.options.Image,
|
||||
Cover: part.Cover,
|
||||
Images: part.Images,
|
||||
Current: i + 1,
|
||||
Total: totalParts,
|
||||
})},
|
||||
{"OEBPS/toc.xhtml", epubtemplates.Toc(title, hasTitlePage, e.StripFirstDirectoryFromToc, part.Images)},
|
||||
{"OEBPS/toc.xhtml", epubtemplates.Toc(title, hasTitlePage, e.options.StripFirstDirectoryFromToc, part.Images)},
|
||||
{"OEBPS/Text/style.css", e.render(epubtemplates.Style, map[string]any{
|
||||
"View": e.Image.View,
|
||||
"View": e.options.Image.View,
|
||||
})},
|
||||
}
|
||||
|
||||
@ -443,9 +444,9 @@ func (e *ePub) Write() error {
|
||||
}
|
||||
|
||||
// Double Page or Last Image that is not a double page
|
||||
if !e.Image.View.PortraitOnly &&
|
||||
if !e.options.Image.View.PortraitOnly &&
|
||||
(img.DoublePage ||
|
||||
(!e.Image.KeepDoublePageIfSplitted && img.Part == 1) ||
|
||||
(!e.options.Image.KeepDoublePageIfSplitted && img.Part == 1) ||
|
||||
(img.Part == 0 && img == lastImage)) {
|
||||
if err := e.writeBlank(wz, img); err != nil {
|
||||
return err
|
||||
@ -455,14 +456,14 @@ func (e *ePub) Write() error {
|
||||
bar.Add(1)
|
||||
}
|
||||
bar.Close()
|
||||
if !e.Json {
|
||||
if !e.options.Json {
|
||||
fmt.Fprintln(os.Stderr)
|
||||
}
|
||||
|
||||
// display corrupted images
|
||||
hasError := false
|
||||
for pId, part := range epubParts {
|
||||
if pId == 0 && e.Image.HasCover && part.Cover.Error != nil {
|
||||
if pId == 0 && e.options.Image.HasCover && part.Cover.Error != nil {
|
||||
hasError = true
|
||||
fmt.Fprintf(os.Stderr, "Error on image %s: %v\n", filepath.Join(part.Cover.Path, part.Cover.Name), part.Cover.Error)
|
||||
}
|
||||
|
@ -1,23 +0,0 @@
|
||||
/*
|
||||
Templates use to create xml files of the EPUB.
|
||||
*/
|
||||
package epubtemplates
|
||||
|
||||
import _ "embed"
|
||||
|
||||
var (
|
||||
//go:embed "epub_templates_container.xml.tmpl"
|
||||
Container string
|
||||
|
||||
//go:embed "epub_templates_applebooks.xml.tmpl"
|
||||
AppleBooks string
|
||||
|
||||
//go:embed "epub_templates_style.css.tmpl"
|
||||
Style string
|
||||
|
||||
//go:embed "epub_templates_text.xhtml.tmpl"
|
||||
Text string
|
||||
|
||||
//go:embed "epub_templates_blank.xhtml.tmpl"
|
||||
Blank string
|
||||
)
|
@ -1,100 +0,0 @@
|
||||
package epubzip
|
||||
|
||||
import (
|
||||
"archive/zip"
|
||||
"image"
|
||||
"os"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type EPUBZipStorageImageWriter struct {
|
||||
fh *os.File
|
||||
fz *zip.Writer
|
||||
format string
|
||||
mut *sync.Mutex
|
||||
}
|
||||
|
||||
func NewEPUBZipStorageImageWriter(filename string, format string) (*EPUBZipStorageImageWriter, error) {
|
||||
fh, err := os.Create(filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
fz := zip.NewWriter(fh)
|
||||
return &EPUBZipStorageImageWriter{fh, fz, format, &sync.Mutex{}}, nil
|
||||
}
|
||||
|
||||
func (e *EPUBZipStorageImageWriter) Close() error {
|
||||
if err := e.fz.Close(); err != nil {
|
||||
e.fh.Close()
|
||||
return err
|
||||
}
|
||||
return e.fh.Close()
|
||||
}
|
||||
|
||||
func (e *EPUBZipStorageImageWriter) Add(filename string, img image.Image, quality int) error {
|
||||
zipImage, err := CompressImage(filename, e.format, img, quality)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
e.mut.Lock()
|
||||
defer e.mut.Unlock()
|
||||
fh, err := e.fz.CreateRaw(zipImage.Header)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = fh.Write(zipImage.Data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type EPUBZipStorageImageReader struct {
|
||||
filename string
|
||||
fh *os.File
|
||||
fz *zip.Reader
|
||||
|
||||
files map[string]*zip.File
|
||||
}
|
||||
|
||||
func NewEPUBZipStorageImageReader(filename string) (*EPUBZipStorageImageReader, error) {
|
||||
fh, err := os.Open(filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
s, err := fh.Stat()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
fz, err := zip.NewReader(fh, s.Size())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
files := map[string]*zip.File{}
|
||||
for _, z := range fz.File {
|
||||
files[z.Name] = z
|
||||
}
|
||||
return &EPUBZipStorageImageReader{filename, fh, fz, files}, nil
|
||||
}
|
||||
|
||||
func (e *EPUBZipStorageImageReader) Get(filename string) *zip.File {
|
||||
return e.files[filename]
|
||||
}
|
||||
|
||||
func (e *EPUBZipStorageImageReader) Size(filename string) uint64 {
|
||||
img := e.Get(filename)
|
||||
if img != nil {
|
||||
return img.CompressedSize64 + 30 + uint64(len(img.Name))
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (e *EPUBZipStorageImageReader) Close() error {
|
||||
return e.fh.Close()
|
||||
}
|
||||
|
||||
func (e *EPUBZipStorageImageReader) Remove() error {
|
||||
return os.Remove(e.filename)
|
||||
}
|
@ -9,7 +9,7 @@ import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Image struct {
|
||||
type EPUBImage struct {
|
||||
Id int
|
||||
Part int
|
||||
Raw image.Image
|
||||
@ -27,47 +27,47 @@ type Image struct {
|
||||
}
|
||||
|
||||
// key name of the blank plage after the image
|
||||
func (i *Image) SpaceKey() string {
|
||||
func (i *EPUBImage) SpaceKey() string {
|
||||
return fmt.Sprintf("space_%d", i.Id)
|
||||
}
|
||||
|
||||
// path of the blank page
|
||||
func (i *Image) SpacePath() string {
|
||||
func (i *EPUBImage) SpacePath() string {
|
||||
return fmt.Sprintf("Text/%s.xhtml", i.SpaceKey())
|
||||
}
|
||||
|
||||
// path of the blank page into the EPUB
|
||||
func (i *Image) EPUBSpacePath() string {
|
||||
func (i *EPUBImage) EPUBSpacePath() string {
|
||||
return fmt.Sprintf("OEBPS/%s", i.SpacePath())
|
||||
}
|
||||
|
||||
// key for page
|
||||
func (i *Image) PageKey() string {
|
||||
func (i *EPUBImage) PageKey() string {
|
||||
return fmt.Sprintf("page_%d_p%d", i.Id, i.Part)
|
||||
}
|
||||
|
||||
// page path linked to the image
|
||||
func (i *Image) PagePath() string {
|
||||
func (i *EPUBImage) PagePath() string {
|
||||
return fmt.Sprintf("Text/%s.xhtml", i.PageKey())
|
||||
}
|
||||
|
||||
// page path into the EPUB
|
||||
func (i *Image) EPUBPagePath() string {
|
||||
func (i *EPUBImage) EPUBPagePath() string {
|
||||
return fmt.Sprintf("OEBPS/%s", i.PagePath())
|
||||
}
|
||||
|
||||
// key for image
|
||||
func (i *Image) ImgKey() string {
|
||||
func (i *EPUBImage) ImgKey() string {
|
||||
return fmt.Sprintf("img_%d_p%d", i.Id, i.Part)
|
||||
}
|
||||
|
||||
// image path
|
||||
func (i *Image) ImgPath() string {
|
||||
func (i *EPUBImage) ImgPath() string {
|
||||
return fmt.Sprintf("Images/%s.%s", i.ImgKey(), i.Format)
|
||||
}
|
||||
|
||||
// image path into the EPUB
|
||||
func (i *Image) EPUBImgPath() string {
|
||||
func (i *EPUBImage) EPUBImgPath() string {
|
||||
return fmt.Sprintf("OEBPS/%s", i.ImgPath())
|
||||
}
|
||||
|
||||
@ -75,7 +75,7 @@ func (i *Image) EPUBImgPath() string {
|
||||
//
|
||||
// center by default.
|
||||
// align to left or right if it's part of the splitted double page.
|
||||
func (i *Image) ImgStyle(viewWidth, viewHeight int, align string) string {
|
||||
func (i *EPUBImage) ImgStyle(viewWidth, viewHeight int, align string) string {
|
||||
relWidth, relHeight := i.RelSize(viewWidth, viewHeight)
|
||||
marginW, marginH := float64(viewWidth-relWidth)/2, float64(viewHeight-relHeight)/2
|
||||
|
||||
@ -100,7 +100,7 @@ func (i *Image) ImgStyle(viewWidth, viewHeight int, align string) string {
|
||||
return strings.Join(style, "; ")
|
||||
}
|
||||
|
||||
func (i *Image) RelSize(viewWidth, viewHeight int) (relWidth, relHeight int) {
|
||||
func (i *EPUBImage) RelSize(viewWidth, viewHeight int) (relWidth, relHeight int) {
|
||||
w, h := viewWidth, viewHeight
|
||||
srcw, srch := i.Width, i.Height
|
||||
|
@ -6,55 +6,54 @@ package epubimageprocessor
|
||||
import (
|
||||
"fmt"
|
||||
"image"
|
||||
"image/color"
|
||||
"image/draw"
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
epubimage "github.com/celogeek/go-comic-converter/v2/internal/epub/image"
|
||||
epubimagefilters "github.com/celogeek/go-comic-converter/v2/internal/epub/imagefilters"
|
||||
epuboptions "github.com/celogeek/go-comic-converter/v2/internal/epub/options"
|
||||
epubprogress "github.com/celogeek/go-comic-converter/v2/internal/epub/progress"
|
||||
epubzip "github.com/celogeek/go-comic-converter/v2/internal/epub/zip"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubimage"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubimagefilters"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubprogress"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubzip"
|
||||
"github.com/celogeek/go-comic-converter/v2/pkg/epuboptions"
|
||||
"github.com/disintegration/gift"
|
||||
)
|
||||
|
||||
type EPUBImageProcessor struct {
|
||||
*epuboptions.Options
|
||||
options *epuboptions.EPUBOptions
|
||||
}
|
||||
|
||||
func New(o *epuboptions.Options) *EPUBImageProcessor {
|
||||
func New(o *epuboptions.EPUBOptions) *EPUBImageProcessor {
|
||||
return &EPUBImageProcessor{o}
|
||||
}
|
||||
|
||||
// extract and convert images
|
||||
func (e *EPUBImageProcessor) Load() (images []*epubimage.Image, err error) {
|
||||
images = make([]*epubimage.Image, 0)
|
||||
func (e *EPUBImageProcessor) Load() (images []*epubimage.EPUBImage, err error) {
|
||||
images = make([]*epubimage.EPUBImage, 0)
|
||||
imageCount, imageInput, err := e.load()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// dry run, skip convertion
|
||||
if e.Dry {
|
||||
if e.options.Dry {
|
||||
for img := range imageInput {
|
||||
images = append(images, &epubimage.Image{
|
||||
images = append(images, &epubimage.EPUBImage{
|
||||
Id: img.Id,
|
||||
Path: img.Path,
|
||||
Name: img.Name,
|
||||
Format: e.Image.Format,
|
||||
Format: e.options.Image.Format,
|
||||
})
|
||||
}
|
||||
|
||||
return images, nil
|
||||
}
|
||||
|
||||
imageOutput := make(chan *epubimage.Image)
|
||||
imageOutput := make(chan *epubimage.EPUBImage)
|
||||
|
||||
// processing
|
||||
bar := epubprogress.New(epubprogress.Options{
|
||||
Quiet: e.Quiet,
|
||||
Json: e.Json,
|
||||
Quiet: e.options.Quiet,
|
||||
Json: e.options.Json,
|
||||
Max: imageCount,
|
||||
Description: "Processing",
|
||||
CurrentJob: 1,
|
||||
@ -62,17 +61,17 @@ func (e *EPUBImageProcessor) Load() (images []*epubimage.Image, err error) {
|
||||
})
|
||||
wg := &sync.WaitGroup{}
|
||||
|
||||
imgStorage, err := epubzip.NewEPUBZipStorageImageWriter(e.ImgStorage(), e.Image.Format)
|
||||
imgStorage, err := epubzip.NewImageWriter(e.options.Temp(), e.options.Image.Format)
|
||||
if err != nil {
|
||||
bar.Close()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
wr := 50
|
||||
if e.Image.Format == "png" {
|
||||
if e.options.Image.Format == "png" {
|
||||
wr = 100
|
||||
}
|
||||
for i := 0; i < e.WorkersRatio(wr); i++ {
|
||||
for i := 0; i < e.options.WorkersRatio(wr); i++ {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
@ -86,7 +85,7 @@ func (e *EPUBImageProcessor) Load() (images []*epubimage.Image, err error) {
|
||||
raw = dst
|
||||
}
|
||||
|
||||
img := &epubimage.Image{
|
||||
img := &epubimage.EPUBImage{
|
||||
Id: input.Id,
|
||||
Part: part,
|
||||
Raw: raw,
|
||||
@ -97,7 +96,7 @@ func (e *EPUBImageProcessor) Load() (images []*epubimage.Image, err error) {
|
||||
DoublePage: part == 0 && src.Bounds().Dx() > src.Bounds().Dy(),
|
||||
Path: input.Path,
|
||||
Name: input.Name,
|
||||
Format: e.Image.Format,
|
||||
Format: e.options.Image.Format,
|
||||
OriginalAspectRatio: float64(src.Bounds().Dy()) / float64(src.Bounds().Dx()),
|
||||
Error: input.Error,
|
||||
}
|
||||
@ -105,12 +104,12 @@ func (e *EPUBImageProcessor) Load() (images []*epubimage.Image, err error) {
|
||||
// do not keep double page if requested
|
||||
if !img.IsCover &&
|
||||
img.DoublePage &&
|
||||
e.Options.Image.AutoSplitDoublePage &&
|
||||
!e.Options.Image.KeepDoublePageIfSplitted {
|
||||
e.options.Image.AutoSplitDoublePage &&
|
||||
!e.options.Image.KeepDoublePageIfSplitted {
|
||||
continue
|
||||
}
|
||||
|
||||
if err = imgStorage.Add(img.EPUBImgPath(), dst, e.Image.Quality); err != nil {
|
||||
if err = imgStorage.Add(img.EPUBImgPath(), dst, e.options.Image.Quality); err != nil {
|
||||
bar.Close()
|
||||
fmt.Fprintf(os.Stderr, "error with %s: %s", input.Name, err)
|
||||
os.Exit(1)
|
||||
@ -131,7 +130,7 @@ func (e *EPUBImageProcessor) Load() (images []*epubimage.Image, err error) {
|
||||
if img.Part == 0 {
|
||||
bar.Add(1)
|
||||
}
|
||||
if e.Image.NoBlankImage && img.IsBlank {
|
||||
if e.options.Image.NoBlankImage && img.IsBlank {
|
||||
continue
|
||||
}
|
||||
images = append(images, img)
|
||||
@ -146,7 +145,7 @@ func (e *EPUBImageProcessor) Load() (images []*epubimage.Image, err error) {
|
||||
}
|
||||
|
||||
func (e *EPUBImageProcessor) createImage(src image.Image, r image.Rectangle) draw.Image {
|
||||
if e.Options.Image.GrayScale {
|
||||
if e.options.Image.GrayScale {
|
||||
return image.NewGray(r)
|
||||
}
|
||||
|
||||
@ -183,13 +182,13 @@ func (e *EPUBImageProcessor) transformImage(src image.Image, srcId int) []image.
|
||||
var images []image.Image
|
||||
|
||||
// Lookup for margin if crop is enable or if we want to remove blank image
|
||||
if e.Image.Crop.Enabled || e.Image.NoBlankImage {
|
||||
if e.options.Image.Crop.Enabled || e.options.Image.NoBlankImage {
|
||||
f := epubimagefilters.AutoCrop(
|
||||
src,
|
||||
e.Image.Crop.Left,
|
||||
e.Image.Crop.Up,
|
||||
e.Image.Crop.Right,
|
||||
e.Image.Crop.Bottom,
|
||||
e.options.Image.Crop.Left,
|
||||
e.options.Image.Crop.Up,
|
||||
e.options.Image.Crop.Right,
|
||||
e.options.Image.Crop.Bottom,
|
||||
)
|
||||
|
||||
// detect if blank image
|
||||
@ -197,42 +196,42 @@ func (e *EPUBImageProcessor) transformImage(src image.Image, srcId int) []image.
|
||||
isBlank := size.Dx() == 0 && size.Dy() == 0
|
||||
|
||||
// crop is enable or if blank image with noblankimage options
|
||||
if e.Image.Crop.Enabled || (e.Image.NoBlankImage && isBlank) {
|
||||
if e.options.Image.Crop.Enabled || (e.options.Image.NoBlankImage && isBlank) {
|
||||
filters = append(filters, f)
|
||||
splitFilters = append(splitFilters, f)
|
||||
}
|
||||
}
|
||||
|
||||
if e.Image.AutoRotate && src.Bounds().Dx() > src.Bounds().Dy() {
|
||||
if e.options.Image.AutoRotate && src.Bounds().Dx() > src.Bounds().Dy() {
|
||||
filters = append(filters, gift.Rotate90())
|
||||
}
|
||||
|
||||
if e.Image.AutoContrast {
|
||||
if e.options.Image.AutoContrast {
|
||||
f := epubimagefilters.AutoContrast()
|
||||
filters = append(filters, f)
|
||||
splitFilters = append(splitFilters, f)
|
||||
}
|
||||
|
||||
if e.Image.Contrast != 0 {
|
||||
f := gift.Contrast(float32(e.Image.Contrast))
|
||||
if e.options.Image.Contrast != 0 {
|
||||
f := gift.Contrast(float32(e.options.Image.Contrast))
|
||||
filters = append(filters, f)
|
||||
splitFilters = append(splitFilters, f)
|
||||
}
|
||||
|
||||
if e.Image.Brightness != 0 {
|
||||
f := gift.Brightness(float32(e.Image.Brightness))
|
||||
if e.options.Image.Brightness != 0 {
|
||||
f := gift.Brightness(float32(e.options.Image.Brightness))
|
||||
filters = append(filters, f)
|
||||
splitFilters = append(splitFilters, f)
|
||||
}
|
||||
|
||||
if e.Image.Resize {
|
||||
f := gift.ResizeToFit(e.Image.View.Width, e.Image.View.Height, gift.LanczosResampling)
|
||||
if e.options.Image.Resize {
|
||||
f := gift.ResizeToFit(e.options.Image.View.Width, e.options.Image.View.Height, gift.LanczosResampling)
|
||||
filters = append(filters, f)
|
||||
}
|
||||
|
||||
if e.Image.GrayScale {
|
||||
if e.options.Image.GrayScale {
|
||||
var f gift.Filter
|
||||
switch e.Image.GrayScaleMode {
|
||||
switch e.options.Image.GrayScaleMode {
|
||||
case 1: // average
|
||||
f = gift.ColorFunc(func(r0, g0, b0, a0 float32) (r float32, g float32, b float32, a float32) {
|
||||
y := (r0 + g0 + b0) / 3
|
||||
@ -261,7 +260,7 @@ func (e *EPUBImageProcessor) transformImage(src image.Image, srcId int) []image.
|
||||
}
|
||||
|
||||
// auto split off
|
||||
if !e.Image.AutoSplitDoublePage {
|
||||
if !e.options.Image.AutoSplitDoublePage {
|
||||
return images
|
||||
}
|
||||
|
||||
@ -271,16 +270,16 @@ func (e *EPUBImageProcessor) transformImage(src image.Image, srcId int) []image.
|
||||
}
|
||||
|
||||
// cover
|
||||
if e.Image.HasCover && srcId == 0 {
|
||||
if e.options.Image.HasCover && srcId == 0 {
|
||||
return images
|
||||
}
|
||||
|
||||
// convert double page
|
||||
for _, b := range []bool{e.Image.Manga, !e.Image.Manga} {
|
||||
for _, b := range []bool{e.options.Image.Manga, !e.options.Image.Manga} {
|
||||
g := gift.New(splitFilters...)
|
||||
g.Add(epubimagefilters.CropSplitDoublePage(b))
|
||||
if e.Image.Resize {
|
||||
g.Add(gift.ResizeToFit(e.Image.View.Width, e.Image.View.Height, gift.LanczosResampling))
|
||||
if e.options.Image.Resize {
|
||||
g.Add(gift.ResizeToFit(e.options.Image.View.Width, e.options.Image.View.Height, gift.LanczosResampling))
|
||||
}
|
||||
dst := e.createImage(src, g.Bounds(src.Bounds()))
|
||||
g.Draw(dst, src)
|
||||
@ -289,55 +288,3 @@ func (e *EPUBImageProcessor) transformImage(src image.Image, srcId int) []image.
|
||||
|
||||
return images
|
||||
}
|
||||
|
||||
type CoverTitleDataOptions struct {
|
||||
Src image.Image
|
||||
Name string
|
||||
Text string
|
||||
Align string
|
||||
PctWidth int
|
||||
PctMargin int
|
||||
MaxFontSize int
|
||||
BorderSize int
|
||||
}
|
||||
|
||||
func (e *EPUBImageProcessor) Cover16LevelOfGray(bounds image.Rectangle) draw.Image {
|
||||
return image.NewPaletted(bounds, color.Palette{
|
||||
color.Gray{0x00},
|
||||
color.Gray{0x11},
|
||||
color.Gray{0x22},
|
||||
color.Gray{0x33},
|
||||
color.Gray{0x44},
|
||||
color.Gray{0x55},
|
||||
color.Gray{0x66},
|
||||
color.Gray{0x77},
|
||||
color.Gray{0x88},
|
||||
color.Gray{0x99},
|
||||
color.Gray{0xAA},
|
||||
color.Gray{0xBB},
|
||||
color.Gray{0xCC},
|
||||
color.Gray{0xDD},
|
||||
color.Gray{0xEE},
|
||||
color.Gray{0xFF},
|
||||
})
|
||||
}
|
||||
|
||||
// create a title page with the cover
|
||||
func (e *EPUBImageProcessor) CoverTitleData(o *CoverTitleDataOptions) (*epubzip.ZipImage, error) {
|
||||
// Create a blur version of the cover
|
||||
g := gift.New(epubimagefilters.CoverTitle(o.Text, o.Align, o.PctWidth, o.PctMargin, o.MaxFontSize, o.BorderSize))
|
||||
var dst draw.Image
|
||||
if o.Name == "cover" && e.Image.GrayScale {
|
||||
dst = e.Cover16LevelOfGray(o.Src.Bounds())
|
||||
} else {
|
||||
dst = e.createImage(o.Src, g.Bounds(o.Src.Bounds()))
|
||||
}
|
||||
g.Draw(dst, o.Src)
|
||||
|
||||
return epubzip.CompressImage(
|
||||
fmt.Sprintf("OEBPS/Images/%s.%s", o.Name, e.Image.Format),
|
||||
e.Image.Format,
|
||||
dst,
|
||||
e.Image.Quality,
|
||||
)
|
||||
}
|
@ -0,0 +1,61 @@
|
||||
package epubimageprocessor
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"image"
|
||||
"image/color"
|
||||
"image/draw"
|
||||
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubimagefilters"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubzip"
|
||||
"github.com/disintegration/gift"
|
||||
)
|
||||
|
||||
type CoverTitleDataOptions struct {
|
||||
Src image.Image
|
||||
Name string
|
||||
Text string
|
||||
Align string
|
||||
PctWidth int
|
||||
PctMargin int
|
||||
MaxFontSize int
|
||||
BorderSize int
|
||||
}
|
||||
|
||||
// create a title page with the cover
|
||||
func (e *EPUBImageProcessor) CoverTitleData(o *CoverTitleDataOptions) (*epubzip.EPUBZipImage, error) {
|
||||
// Create a blur version of the cover
|
||||
g := gift.New(epubimagefilters.CoverTitle(o.Text, o.Align, o.PctWidth, o.PctMargin, o.MaxFontSize, o.BorderSize))
|
||||
var dst draw.Image
|
||||
if o.Name == "cover" && e.options.Image.GrayScale {
|
||||
// 16 shade of gray
|
||||
dst = image.NewPaletted(o.Src.Bounds(), color.Palette{
|
||||
color.Gray{0x00},
|
||||
color.Gray{0x11},
|
||||
color.Gray{0x22},
|
||||
color.Gray{0x33},
|
||||
color.Gray{0x44},
|
||||
color.Gray{0x55},
|
||||
color.Gray{0x66},
|
||||
color.Gray{0x77},
|
||||
color.Gray{0x88},
|
||||
color.Gray{0x99},
|
||||
color.Gray{0xAA},
|
||||
color.Gray{0xBB},
|
||||
color.Gray{0xCC},
|
||||
color.Gray{0xDD},
|
||||
color.Gray{0xEE},
|
||||
color.Gray{0xFF},
|
||||
})
|
||||
} else {
|
||||
dst = e.createImage(o.Src, g.Bounds(o.Src.Bounds()))
|
||||
}
|
||||
g.Draw(dst, o.Src)
|
||||
|
||||
return epubzip.CompressImage(
|
||||
fmt.Sprintf("OEBPS/Images/%s.%s", o.Name, e.options.Image.Format),
|
||||
e.options.Image.Format,
|
||||
dst,
|
||||
e.options.Image.Quality,
|
||||
)
|
||||
}
|
@ -51,7 +51,7 @@ func (e *EPUBImageProcessor) isSupportedImage(path string) bool {
|
||||
|
||||
// Load images from input
|
||||
func (e *EPUBImageProcessor) load() (totalImages int, output chan *tasks, err error) {
|
||||
fi, err := os.Stat(e.Input)
|
||||
fi, err := os.Stat(e.options.Input)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
@ -60,7 +60,7 @@ func (e *EPUBImageProcessor) load() (totalImages int, output chan *tasks, err er
|
||||
if fi.IsDir() {
|
||||
return e.loadDir()
|
||||
} else {
|
||||
switch ext := strings.ToLower(filepath.Ext(e.Input)); ext {
|
||||
switch ext := strings.ToLower(filepath.Ext(e.options.Input)); ext {
|
||||
case ".cbz", ".zip":
|
||||
return e.loadCbz()
|
||||
case ".cbr", ".rar":
|
||||
@ -101,7 +101,7 @@ func (e *EPUBImageProcessor) corruptedImage(path, name string) image.Image {
|
||||
func (e *EPUBImageProcessor) loadDir() (totalImages int, output chan *tasks, err error) {
|
||||
images := make([]string, 0)
|
||||
|
||||
input := filepath.Clean(e.Input)
|
||||
input := filepath.Clean(e.options.Input)
|
||||
err = filepath.WalkDir(input, func(path string, d fs.DirEntry, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
@ -123,7 +123,7 @@ func (e *EPUBImageProcessor) loadDir() (totalImages int, output chan *tasks, err
|
||||
return
|
||||
}
|
||||
|
||||
sort.Sort(sortpath.By(images, e.SortPathMode))
|
||||
sort.Sort(sortpath.By(images, e.options.SortPathMode))
|
||||
|
||||
// Queue all file with id
|
||||
type job struct {
|
||||
@ -139,16 +139,16 @@ func (e *EPUBImageProcessor) loadDir() (totalImages int, output chan *tasks, err
|
||||
}()
|
||||
|
||||
// read in parallel and get an image
|
||||
output = make(chan *tasks, e.Workers)
|
||||
output = make(chan *tasks, e.options.Workers)
|
||||
wg := &sync.WaitGroup{}
|
||||
for j := 0; j < e.WorkersRatio(50); j++ {
|
||||
for j := 0; j < e.options.WorkersRatio(50); j++ {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
for job := range jobs {
|
||||
var img image.Image
|
||||
var err error
|
||||
if !e.Dry {
|
||||
if !e.options.Dry {
|
||||
var f *os.File
|
||||
f, err = os.Open(job.Path)
|
||||
if err == nil {
|
||||
@ -188,7 +188,7 @@ func (e *EPUBImageProcessor) loadDir() (totalImages int, output chan *tasks, err
|
||||
|
||||
// load a zip file that include images
|
||||
func (e *EPUBImageProcessor) loadCbz() (totalImages int, output chan *tasks, err error) {
|
||||
r, err := zip.OpenReader(e.Input)
|
||||
r, err := zip.OpenReader(e.options.Input)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
@ -212,7 +212,7 @@ func (e *EPUBImageProcessor) loadCbz() (totalImages int, output chan *tasks, err
|
||||
for _, img := range images {
|
||||
names = append(names, img.Name)
|
||||
}
|
||||
sort.Sort(sortpath.By(names, e.SortPathMode))
|
||||
sort.Sort(sortpath.By(names, e.options.SortPathMode))
|
||||
|
||||
indexedNames := make(map[string]int)
|
||||
for i, name := range names {
|
||||
@ -231,16 +231,16 @@ func (e *EPUBImageProcessor) loadCbz() (totalImages int, output chan *tasks, err
|
||||
}
|
||||
}()
|
||||
|
||||
output = make(chan *tasks, e.Workers)
|
||||
output = make(chan *tasks, e.options.Workers)
|
||||
wg := &sync.WaitGroup{}
|
||||
for j := 0; j < e.WorkersRatio(50); j++ {
|
||||
for j := 0; j < e.options.WorkersRatio(50); j++ {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
for job := range jobs {
|
||||
var img image.Image
|
||||
var err error
|
||||
if !e.Dry {
|
||||
if !e.options.Dry {
|
||||
var f io.ReadCloser
|
||||
f, err = job.F.Open()
|
||||
if err == nil {
|
||||
@ -275,7 +275,7 @@ func (e *EPUBImageProcessor) loadCbz() (totalImages int, output chan *tasks, err
|
||||
// load a rar file that include images
|
||||
func (e *EPUBImageProcessor) loadCbr() (totalImages int, output chan *tasks, err error) {
|
||||
var isSolid bool
|
||||
files, err := rardecode.List(e.Input)
|
||||
files, err := rardecode.List(e.options.Input)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
@ -296,7 +296,7 @@ func (e *EPUBImageProcessor) loadCbr() (totalImages int, output chan *tasks, err
|
||||
return
|
||||
}
|
||||
|
||||
sort.Sort(sortpath.By(names, e.SortPathMode))
|
||||
sort.Sort(sortpath.By(names, e.options.SortPathMode))
|
||||
|
||||
indexedNames := make(map[string]int)
|
||||
for i, name := range names {
|
||||
@ -312,10 +312,10 @@ func (e *EPUBImageProcessor) loadCbr() (totalImages int, output chan *tasks, err
|
||||
jobs := make(chan *job)
|
||||
go func() {
|
||||
defer close(jobs)
|
||||
if isSolid && !e.Dry {
|
||||
r, rerr := rardecode.OpenReader(e.Input)
|
||||
if isSolid && !e.options.Dry {
|
||||
r, rerr := rardecode.OpenReader(e.options.Input)
|
||||
if rerr != nil {
|
||||
fmt.Fprintf(os.Stderr, "\nerror processing image %s: %s\n", e.Input, rerr)
|
||||
fmt.Fprintf(os.Stderr, "\nerror processing image %s: %s\n", e.options.Input, rerr)
|
||||
os.Exit(1)
|
||||
}
|
||||
defer r.Close()
|
||||
@ -350,16 +350,16 @@ func (e *EPUBImageProcessor) loadCbr() (totalImages int, output chan *tasks, err
|
||||
}()
|
||||
|
||||
// send file to the queue
|
||||
output = make(chan *tasks, e.Workers)
|
||||
output = make(chan *tasks, e.options.Workers)
|
||||
wg := &sync.WaitGroup{}
|
||||
for j := 0; j < e.WorkersRatio(50); j++ {
|
||||
for j := 0; j < e.options.WorkersRatio(50); j++ {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
for job := range jobs {
|
||||
var img image.Image
|
||||
var err error
|
||||
if !e.Dry {
|
||||
if !e.options.Dry {
|
||||
var f io.ReadCloser
|
||||
f, err = job.Open()
|
||||
if err == nil {
|
||||
@ -391,7 +391,7 @@ func (e *EPUBImageProcessor) loadCbr() (totalImages int, output chan *tasks, err
|
||||
|
||||
// extract image from a pdf
|
||||
func (e *EPUBImageProcessor) loadPdf() (totalImages int, output chan *tasks, err error) {
|
||||
pdf := pdfread.Load(e.Input)
|
||||
pdf := pdfread.Load(e.options.Input)
|
||||
if pdf == nil {
|
||||
err = fmt.Errorf("can't read pdf")
|
||||
return
|
||||
@ -406,7 +406,7 @@ func (e *EPUBImageProcessor) loadPdf() (totalImages int, output chan *tasks, err
|
||||
for i := 0; i < totalImages; i++ {
|
||||
var img image.Image
|
||||
var err error
|
||||
if !e.Dry {
|
||||
if !e.options.Dry {
|
||||
img, err = pdfimage.Extract(pdf, i+1)
|
||||
}
|
||||
|
@ -4,6 +4,7 @@ create a progress bar with custom settings.
|
||||
package epubprogress
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"time"
|
||||
@ -20,18 +21,21 @@ type Options struct {
|
||||
TotalJob int
|
||||
}
|
||||
|
||||
type EpubProgress interface {
|
||||
type epubProgress interface {
|
||||
Add(num int) error
|
||||
Close() error
|
||||
}
|
||||
|
||||
func New(o Options) EpubProgress {
|
||||
func New(o Options) epubProgress {
|
||||
if o.Quiet {
|
||||
return progressbar.DefaultSilent(int64(o.Max))
|
||||
}
|
||||
|
||||
if o.Json {
|
||||
return newEpubProgressJson(o)
|
||||
return &epubProgressJson{
|
||||
o: o,
|
||||
e: json.NewEncoder(os.Stdout),
|
||||
}
|
||||
}
|
||||
|
||||
fmtJob := fmt.Sprintf("%%0%dd", len(fmt.Sprint(o.TotalJob)))
|
@ -2,23 +2,15 @@ package epubprogress
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"os"
|
||||
)
|
||||
|
||||
type EpubProgressJson struct {
|
||||
type epubProgressJson struct {
|
||||
o Options
|
||||
e *json.Encoder
|
||||
current int
|
||||
}
|
||||
|
||||
func newEpubProgressJson(o Options) EpubProgress {
|
||||
return &EpubProgressJson{
|
||||
o: o,
|
||||
e: json.NewEncoder(os.Stdout),
|
||||
}
|
||||
}
|
||||
|
||||
func (p *EpubProgressJson) Add(num int) error {
|
||||
func (p *epubProgressJson) Add(num int) error {
|
||||
p.current += num
|
||||
p.e.Encode(map[string]any{
|
||||
"type": "progress",
|
||||
@ -37,6 +29,6 @@ func (p *EpubProgressJson) Add(num int) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *EpubProgressJson) Close() error {
|
||||
func (p *epubProgressJson) Close() error {
|
||||
return nil
|
||||
}
|
23
internal/epubtemplates/epubtemplates.go
Normal file
23
internal/epubtemplates/epubtemplates.go
Normal file
@ -0,0 +1,23 @@
|
||||
/*
|
||||
Templates use to create xml files of the EPUB.
|
||||
*/
|
||||
package epubtemplates
|
||||
|
||||
import _ "embed"
|
||||
|
||||
var (
|
||||
//go:embed "epubtemplates_container.xml.tmpl"
|
||||
Container string
|
||||
|
||||
//go:embed "epubtemplates_applebooks.xml.tmpl"
|
||||
AppleBooks string
|
||||
|
||||
//go:embed "epubtemplates_style.css.tmpl"
|
||||
Style string
|
||||
|
||||
//go:embed "epubtemplates_text.xhtml.tmpl"
|
||||
Text string
|
||||
|
||||
//go:embed "epubtemplates_blank.xhtml.tmpl"
|
||||
Blank string
|
||||
)
|
@ -4,8 +4,8 @@ import (
|
||||
"fmt"
|
||||
|
||||
"github.com/beevik/etree"
|
||||
epubimage "github.com/celogeek/go-comic-converter/v2/internal/epub/image"
|
||||
epuboptions "github.com/celogeek/go-comic-converter/v2/internal/epub/options"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubimage"
|
||||
"github.com/celogeek/go-comic-converter/v2/pkg/epuboptions"
|
||||
)
|
||||
|
||||
type ContentOptions struct {
|
||||
@ -16,8 +16,8 @@ type ContentOptions struct {
|
||||
Publisher string
|
||||
UpdatedAt string
|
||||
ImageOptions *epuboptions.Image
|
||||
Cover *epubimage.Image
|
||||
Images []*epubimage.Image
|
||||
Cover *epubimage.EPUBImage
|
||||
Images []*epubimage.EPUBImage
|
||||
Current int
|
||||
Total int
|
||||
}
|
||||
@ -140,7 +140,7 @@ func getMeta(o *ContentOptions) []tag {
|
||||
|
||||
func getManifest(o *ContentOptions) []tag {
|
||||
var imageTags, pageTags, spaceTags []tag
|
||||
addTag := func(img *epubimage.Image, withSpace bool) {
|
||||
addTag := func(img *epubimage.EPUBImage, withSpace bool) {
|
||||
imageTags = append(imageTags,
|
||||
tag{"item", tagAttrs{"id": img.ImgKey(), "href": img.ImgPath(), "media-type": fmt.Sprintf("image/%s", o.ImageOptions.Format)}, ""},
|
||||
)
|
@ -5,11 +5,11 @@ import (
|
||||
"strings"
|
||||
|
||||
"github.com/beevik/etree"
|
||||
epubimage "github.com/celogeek/go-comic-converter/v2/internal/epub/image"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epubimage"
|
||||
)
|
||||
|
||||
// create toc
|
||||
func Toc(title string, hasTitle bool, stripFirstDirectoryFromToc bool, images []*epubimage.Image) string {
|
||||
func Toc(title string, hasTitle bool, stripFirstDirectoryFromToc bool, images []*epubimage.EPUBImage) string {
|
||||
doc := etree.NewDocument()
|
||||
doc.CreateProcInst("xml", `version="1.0" encoding="UTF-8"`)
|
||||
doc.CreateDirective("DOCTYPE html")
|
@ -63,7 +63,7 @@ func (e *EPUBZip) Copy(fz *zip.File) error {
|
||||
}
|
||||
|
||||
// Write image. They are already compressed, so we write them down directly.
|
||||
func (e *EPUBZip) WriteRaw(raw *ZipImage) error {
|
||||
func (e *EPUBZip) WriteRaw(raw *EPUBZipImage) error {
|
||||
m, err := e.wz.CreateRaw(raw.Header)
|
||||
if err != nil {
|
||||
return err
|
@ -12,13 +12,13 @@ import (
|
||||
"time"
|
||||
)
|
||||
|
||||
type ZipImage struct {
|
||||
type EPUBZipImage struct {
|
||||
Header *zip.FileHeader
|
||||
Data []byte
|
||||
}
|
||||
|
||||
// create gzip encoded jpeg
|
||||
func CompressImage(filename string, format string, img image.Image, quality int) (*ZipImage, error) {
|
||||
func CompressImage(filename string, format string, img image.Image, quality int) (*EPUBZipImage, error) {
|
||||
var (
|
||||
data, cdata bytes.Buffer
|
||||
err error
|
||||
@ -52,7 +52,7 @@ func CompressImage(filename string, format string, img image.Image, quality int)
|
||||
}
|
||||
|
||||
t := time.Now()
|
||||
return &ZipImage{
|
||||
return &EPUBZipImage{
|
||||
&zip.FileHeader{
|
||||
Name: filename,
|
||||
CompressedSize64: uint64(cdata.Len()),
|
54
internal/epubzip/epubzip_image_reader.go
Normal file
54
internal/epubzip/epubzip_image_reader.go
Normal file
@ -0,0 +1,54 @@
|
||||
package epubzip
|
||||
|
||||
import (
|
||||
"archive/zip"
|
||||
"os"
|
||||
)
|
||||
|
||||
type EPUBZipImageReader struct {
|
||||
filename string
|
||||
fh *os.File
|
||||
fz *zip.Reader
|
||||
|
||||
files map[string]*zip.File
|
||||
}
|
||||
|
||||
func NewImageReader(filename string) (*EPUBZipImageReader, error) {
|
||||
fh, err := os.Open(filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
s, err := fh.Stat()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
fz, err := zip.NewReader(fh, s.Size())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
files := map[string]*zip.File{}
|
||||
for _, z := range fz.File {
|
||||
files[z.Name] = z
|
||||
}
|
||||
return &EPUBZipImageReader{filename, fh, fz, files}, nil
|
||||
}
|
||||
|
||||
func (e *EPUBZipImageReader) Get(filename string) *zip.File {
|
||||
return e.files[filename]
|
||||
}
|
||||
|
||||
func (e *EPUBZipImageReader) Size(filename string) uint64 {
|
||||
img := e.Get(filename)
|
||||
if img != nil {
|
||||
return img.CompressedSize64 + 30 + uint64(len(img.Name))
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (e *EPUBZipImageReader) Close() error {
|
||||
return e.fh.Close()
|
||||
}
|
||||
|
||||
func (e *EPUBZipImageReader) Remove() error {
|
||||
return os.Remove(e.filename)
|
||||
}
|
52
internal/epubzip/epubzip_image_writer.go
Normal file
52
internal/epubzip/epubzip_image_writer.go
Normal file
@ -0,0 +1,52 @@
|
||||
package epubzip
|
||||
|
||||
import (
|
||||
"archive/zip"
|
||||
"image"
|
||||
"os"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type EPUBZipImageWriter struct {
|
||||
fh *os.File
|
||||
fz *zip.Writer
|
||||
format string
|
||||
mut *sync.Mutex
|
||||
}
|
||||
|
||||
func NewImageWriter(filename string, format string) (*EPUBZipImageWriter, error) {
|
||||
fh, err := os.Create(filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
fz := zip.NewWriter(fh)
|
||||
return &EPUBZipImageWriter{fh, fz, format, &sync.Mutex{}}, nil
|
||||
}
|
||||
|
||||
func (e *EPUBZipImageWriter) Close() error {
|
||||
if err := e.fz.Close(); err != nil {
|
||||
e.fh.Close()
|
||||
return err
|
||||
}
|
||||
return e.fh.Close()
|
||||
}
|
||||
|
||||
func (e *EPUBZipImageWriter) Add(filename string, img image.Image, quality int) error {
|
||||
zipImage, err := CompressImage(filename, e.format, img, quality)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
e.mut.Lock()
|
||||
defer e.mut.Unlock()
|
||||
fh, err := e.fz.CreateRaw(zipImage.Header)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = fh.Write(zipImage.Data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
40
main.go
40
main.go
@ -14,8 +14,8 @@ import (
|
||||
"runtime/debug"
|
||||
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/converter"
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epub"
|
||||
epuboptions "github.com/celogeek/go-comic-converter/v2/internal/epub/options"
|
||||
"github.com/celogeek/go-comic-converter/v2/pkg/epub"
|
||||
"github.com/celogeek/go-comic-converter/v2/pkg/epuboptions"
|
||||
"github.com/tcnksm/go-latest"
|
||||
)
|
||||
|
||||
@ -106,8 +106,7 @@ $ go install github.com/celogeek/go-comic-converter/v%d@%s
|
||||
}
|
||||
|
||||
profile := cmd.Options.GetProfile()
|
||||
|
||||
if err := epub.New(&epuboptions.Options{
|
||||
options := &epuboptions.EPUBOptions{
|
||||
Input: cmd.Options.Input,
|
||||
Output: cmd.Options.Output,
|
||||
LimitMb: cmd.Options.LimitMb,
|
||||
@ -122,7 +121,13 @@ $ go install github.com/celogeek/go-comic-converter/v%d@%s
|
||||
Quiet: cmd.Options.Quiet,
|
||||
Json: cmd.Options.Json,
|
||||
Image: &epuboptions.Image{
|
||||
Crop: &epuboptions.Crop{Enabled: cmd.Options.Crop, Left: cmd.Options.CropRatioLeft, Up: cmd.Options.CropRatioUp, Right: cmd.Options.CropRatioRight, Bottom: cmd.Options.CropRatioBottom},
|
||||
Crop: &epuboptions.Crop{
|
||||
Enabled: cmd.Options.Crop,
|
||||
Left: cmd.Options.CropRatioLeft,
|
||||
Up: cmd.Options.CropRatioUp,
|
||||
Right: cmd.Options.CropRatioRight,
|
||||
Bottom: cmd.Options.CropRatioBottom,
|
||||
},
|
||||
Quality: cmd.Options.Quality,
|
||||
Brightness: cmd.Options.Brightness,
|
||||
Contrast: cmd.Options.Contrast,
|
||||
@ -133,14 +138,25 @@ $ go install github.com/celogeek/go-comic-converter/v%d@%s
|
||||
NoBlankImage: cmd.Options.NoBlankImage,
|
||||
Manga: cmd.Options.Manga,
|
||||
HasCover: cmd.Options.HasCover,
|
||||
View: &epuboptions.View{Width: profile.Width, Height: profile.Height, AspectRatio: cmd.Options.AspectRatio, PortraitOnly: cmd.Options.PortraitOnly, Color: epuboptions.Color{Foreground: cmd.Options.ForegroundColor, Background: cmd.Options.BackgroundColor}},
|
||||
GrayScale: cmd.Options.Grayscale,
|
||||
GrayScaleMode: cmd.Options.GrayscaleMode,
|
||||
Resize: !cmd.Options.NoResize,
|
||||
Format: cmd.Options.Format,
|
||||
AppleBookCompatibility: cmd.Options.AppleBookCompatibility,
|
||||
View: &epuboptions.View{
|
||||
Width: profile.Width,
|
||||
Height: profile.Height,
|
||||
AspectRatio: cmd.Options.AspectRatio,
|
||||
PortraitOnly: cmd.Options.PortraitOnly,
|
||||
Color: epuboptions.Color{
|
||||
Foreground: cmd.Options.ForegroundColor,
|
||||
Background: cmd.Options.BackgroundColor,
|
||||
},
|
||||
},
|
||||
GrayScale: cmd.Options.Grayscale,
|
||||
GrayScaleMode: cmd.Options.GrayscaleMode,
|
||||
Resize: !cmd.Options.NoResize,
|
||||
Format: cmd.Options.Format,
|
||||
AppleBookCompatibility: cmd.Options.AppleBookCompatibility,
|
||||
},
|
||||
}).Write(); err != nil {
|
||||
}
|
||||
|
||||
if err := epub.Generate(options); err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
10
pkg/epub/epub.go
Normal file
10
pkg/epub/epub.go
Normal file
@ -0,0 +1,10 @@
|
||||
package epub
|
||||
|
||||
import (
|
||||
"github.com/celogeek/go-comic-converter/v2/internal/epub"
|
||||
"github.com/celogeek/go-comic-converter/v2/pkg/epuboptions"
|
||||
)
|
||||
|
||||
func Generate(options *epuboptions.EPUBOptions) error {
|
||||
return epub.New(options).Write()
|
||||
}
|
@ -41,7 +41,7 @@ type Image struct {
|
||||
AppleBookCompatibility bool
|
||||
}
|
||||
|
||||
type Options struct {
|
||||
type EPUBOptions struct {
|
||||
Input string
|
||||
Output string
|
||||
Title string
|
||||
@ -58,7 +58,7 @@ type Options struct {
|
||||
Image *Image
|
||||
}
|
||||
|
||||
func (o *Options) WorkersRatio(pct int) (nbWorkers int) {
|
||||
func (o *EPUBOptions) WorkersRatio(pct int) (nbWorkers int) {
|
||||
nbWorkers = o.Workers * pct / 100
|
||||
if nbWorkers < 1 {
|
||||
nbWorkers = 1
|
||||
@ -66,6 +66,6 @@ func (o *Options) WorkersRatio(pct int) (nbWorkers int) {
|
||||
return
|
||||
}
|
||||
|
||||
func (o *Options) ImgStorage() string {
|
||||
func (o *EPUBOptions) Temp() string {
|
||||
return fmt.Sprintf("%s.tmp", o.Output)
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user