irrelevant/webtoon-dl-gif-support
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

save cbz image with png ext if it is a png

Browse source
This commit is contained in:
Peter Kaufman 2024-08-28 22:33:20 -04:00
parent f3643c85a9
commit add766c765
1 changed files with 17 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

25
main.go
View file

@ -6,8 +6,6 @@ import (
"encoding/json" "encoding/json"
"flag" "flag"
"fmt" "fmt"
"github.com/anaskhan96/soup"
"github.com/signintech/gopdf"
"image" "image"
"io" "io"
"math" "math"
@ -18,6 +16,9 @@ import (
"strconv" "strconv"
"strings" "strings"
"time" "time"
"github.com/anaskhan96/soup"
"github.com/signintech/gopdf"
) )
type MotiontoonJson struct { type MotiontoonJson struct {
@ -33,7 +34,7 @@ type EpisodeBatch struct {
} }
type ComicFile interface { type ComicFile interface {
addImage([]byte) error addImage([]byte, string) error
save(outFile string) error save(outFile string) error
} }
@ -50,7 +51,7 @@ func newPDFComicFile() *PDFComicFile {
return &PDFComicFile{pdf: &pdf} return &PDFComicFile{pdf: &pdf}
} }
func (c *PDFComicFile) addImage(img []byte) error { func (c *PDFComicFile) addImage(img []byte, ext string) error {
holder, err := gopdf.ImageHolderByBytes(img) holder, err := gopdf.ImageHolderByBytes(img)
if err != nil { if err != nil {
return err return err
@ -91,8 +92,8 @@ func newCBZComicFile() (*CBZComicFile, error) {
return &CBZComicFile{zipWriter: zipWriter, buffer: buffer, numFiles: 0}, nil return &CBZComicFile{zipWriter: zipWriter, buffer: buffer, numFiles: 0}, nil
} }
func (c *CBZComicFile) addImage(img []byte) error { func (c *CBZComicFile) addImage(img []byte, ext string) error {
f, err := c.zipWriter.Create(fmt.Sprintf("%010d.jpg", c.numFiles)) f, err := c.zipWriter.Create(fmt.Sprintf("%010d.%s", c.numFiles, ext))
if err != nil { if err != nil {
return err return err
} }
@ -431,12 +432,20 @@ func main() {
var err error var err error
outFile := getOutFile(opts, episodeBatch) outFile := getOutFile(opts, episodeBatch)
comicFile := getComicFile(opts.format) comicFile := getComicFile(opts.format)
var lowercaseImgLink string
for idx, imgLink := range episodeBatch.imgLinks { for idx, imgLink := range episodeBatch.imgLinks {
if strings.Contains(imgLink, ".gif") { lowercaseImgLink = strings.ToLower(imgLink)
if strings.Contains(lowercaseImgLink, ".gif") {
fmt.Println(fmt.Sprintf("WARNING: skipping gif %s", imgLink)) fmt.Println(fmt.Sprintf("WARNING: skipping gif %s", imgLink))
continue continue
} }
err := comicFile.addImage(fetchImage(imgLink))
if strings.Contains(lowercaseImgLink, ".png") {
err = comicFile.addImage(fetchImage(imgLink), "png")
} else {
err = comicFile.addImage(fetchImage(imgLink), "jpg")
}
if err != nil { if err != nil {
fmt.Println(err.Error()) fmt.Println(err.Error())
os.Exit(1) os.Exit(1)