feat(logging): Implement structured logging and debug mode

fix: Windows cannot download novels correctly

.github/workflows/release.yml

@@ -0,0 +1,28 @@
+name: release
+
+on:
+  push:
+    tags:
+      - "v*"
+
+permissions:
+  contents: write
+
+jobs:
+  goreleaser:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v5
+
+      - name: Set up Go
+        uses: actions/setup-go@v5
+
+      - name: Run GoReleaser
+        uses: goreleaser/goreleaser-action@v6
+        with:
+          distribution: goreleaser
+          version: latest
+          args: release --clean
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.goreleaser.yaml

@@ -1,6 +1,8 @@
+version: 2
 project_name: bilinovel-downloader
 before:
   hooks:
+    - go install github.com/a-h/templ/cmd/templ@latest
     - templ generate
 builds:
   - env:
@@ -12,16 +14,15 @@ builds:
     goarch:
       - amd64
       - arm64
-      - arm
       - "386"
     ldflags:
       - -s -w -X bilinovel-downloader/cmd.Version={{ .Version }}
     flags:
       - -trimpath
 archives:
-  - format: tar.gz
+  - formats: ["tar.gz"]
     format_overrides:
-      - format: zip
+      - formats: ["zip"]
         goos: windows
     wrap_in_directory: true
 release:
@@ -29,3 +30,17 @@ release:
 upx:
   - enabled: true
     compress: best
+
+nfpms:
+  - id: bilinovel-downloader
+    homepage: https://github.com/bestnite/bilinovel-downloader
+    maintainer: Nite <admin@nite07.com>
+    license: "MIT"
+    formats:
+      - apk
+      - deb
+      - rpm
+      - termux.deb
+      - archlinux
+    provides:
+      - bilinovel-downloader

.vscode/launch.json

@@ -2,12 +2,31 @@
   "version": "0.2.0",
   "configurations": [
     {
-      "name": "Debug download volume",
+      "name": "novel",
       "type": "go",
       "request": "launch",
       "mode": "auto",
       "program": "${workspaceFolder}",
-      "args": ["download", "volume", "-n", "2013", "-v", "165880"]
+      "args": [
+        "download",
+        "-n",
+        "2727",
+        "--concurrency",
+        "5"
+      ]
+    },
+    {
+      "name": "volume",
+      "type": "go",
+      "request": "launch",
+      "mode": "auto",
+      "program": "${workspaceFolder}",
+      "args": [
+        "download",
+        "-n=2388",
+        "-v=84522",
+        "--debug=true"
+      ]
     }
   ]
 }

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Nite
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,24 +1,24 @@
-# BiliNovel Downloader
+# Bilinovel Downloader
 
-这是一个用于下载和生成轻小说 EPUB 电子书的工具。
+这是一个用于从 Bilinovel 下载和生成轻小说 EPUB 电子书的工具。
+生成的 EPUB 文件完全符合 EPUB 标准，可以在 Calibre 检查中无错误通过。
 
-## 功能特点
+## 使用示例
 
-- 支持下载轻小说并转换为标准 EPUB 格式
-- 自动处理图片和文本内容
-- 生成符合 EPUB 3.0 规范的电子书文件
-- 支持多章节内容的组织和管理
-- 保留原有插图和排版格式
+1. 下载整本 `https://www.bilinovel.com/novel/2388.html`
 
-## 使用说明
+   ```bash
+   bilinovel-downloader download -n 2388
+   ```
 
-1. 确保系统环境满足要求
-2. 运行下载器获取小说内容
-3. 程序会自动处理并生成标准格式的 EPUB 文件
-4. 生成的电子书文件可以在任何支持 EPUB 3.0 的阅读器中打开
+2. 下载单卷 `https://www.bilinovel.com/novel/2388/vol_84522.html`
 
-## 注意事项
+   ```bash
+   bilinovel-downloader download -n 2388 -v 84522
+   ```
 
-- 生成的 EPUB 文件严格遵循 EPUB 3.0 规范
-- 建议使用支持 EPUB 3.0 的阅读器以获得最佳阅读体验
-- 请遵守相关法律法规，合理使用下载的内容
+3. 对自动生成的 epub 格式不满意可以自行修改后使用命令打包
+
+   ```bash
+   bilinovel-downloader pack -d <目录路径>
+   ```

cmd/download.go

@@ -1,9 +1,19 @@
 package cmd
 
 import (
+	"bilinovel-downloader/downloader"
 	"bilinovel-downloader/downloader/bilinovel"
+	"bilinovel-downloader/epub"
+	"bilinovel-downloader/model"
+	"bilinovel-downloader/text"
+	"encoding/json"
 	"fmt"
+	"io"
+	"log/slog"
+	"os"
+	"path/filepath"
 
+	"github.com/playwright-community/playwright-go"
 	"github.com/spf13/cobra"
 )
 
@@ -11,74 +21,184 @@ var downloadCmd = &cobra.Command{
 	Use:   "download",
 	Short: "Download a novel or volume",
 	Long:  "Download a novel or volume",
+	Run: func(cmd *cobra.Command, args []string) {
+		slog.Info("Installing playwright")
+		err := playwright.Install(&playwright.RunOptions{
+			Browsers: []string{"chromium"},
+			Stdout:   io.Discard,
+		})
+		if err != nil {
+			slog.Error("failed to install playwright")
+			return
 		}
 
-var downloadNovelCmd = &cobra.Command{
-	Use:   "novel",
-	Short: "Download a novel, default download all volumes",
-	Long:  "Download a novel, default download all volumes",
-	RunE:  runDownloadNovel,
+		err = runDownloadNovel()
+		if err != nil {
+			slog.Error("failed to download novel", slog.Any("error", err))
+			return
+		}
+	},
 }
 
-var downloadVolumeCmd = &cobra.Command{
-	Use:   "volume",
-	Short: "Download a volume",
-	Long:  "Download a volume",
-	RunE:  runDownloadVolume,
-}
-
-type downloadNovelArgs struct {
-	NovelId    int `validate:"required"`
-	outputPath string
-}
-
-type downloadVolumeArgs struct {
+type downloadCmdArgs struct {
 	NovelId     int `validate:"required"`
 	VolumeId    int `validate:"required"`
 	outputPath  string
+	outputType  string
+	concurrency int
+	debug       bool
 }
 
 var (
-	novelArgs  downloadNovelArgs
-	volumeArgs downloadVolumeArgs
+	downloadArgs downloadCmdArgs
 )
 
 func init() {
-	downloadNovelCmd.Flags().IntVarP(&novelArgs.NovelId, "novel-id", "n", 0, "novel id")
-	downloadNovelCmd.Flags().StringVarP(&novelArgs.outputPath, "output-path", "o", "./novels", "output path")
-
-	downloadVolumeCmd.Flags().IntVarP(&volumeArgs.NovelId, "novel-id", "n", 0, "novel id")
-	downloadVolumeCmd.Flags().IntVarP(&volumeArgs.VolumeId, "volume-id", "v", 0, "volume id")
-	downloadVolumeCmd.Flags().StringVarP(&volumeArgs.outputPath, "output-path", "o", "./novels", "output path")
-
-	downloadCmd.AddCommand(downloadNovelCmd)
-	downloadCmd.AddCommand(downloadVolumeCmd)
+	downloadCmd.Flags().IntVarP(&downloadArgs.NovelId, "novel-id", "n", 0, "novel id")
+	downloadCmd.Flags().IntVarP(&downloadArgs.VolumeId, "volume-id", "v", 0, "volume id")
+	downloadCmd.Flags().StringVarP(&downloadArgs.outputPath, "output-path", "o", "novels", "output path")
+	downloadCmd.Flags().StringVarP(&downloadArgs.outputType, "output-type", "t", "epub", "output type, epub or text")
+	downloadCmd.Flags().BoolVar(&downloadArgs.debug, "debug", false, "debug mode")
+	downloadCmd.Flags().IntVar(&downloadArgs.concurrency, "concurrency", 3, "concurrency of downloading volumes")
 	RootCmd.AddCommand(downloadCmd)
 }
 
-func runDownloadNovel(cmd *cobra.Command, args []string) error {
-	if novelArgs.NovelId == 0 {
-		return fmt.Errorf("novel id is required")
-	}
-	err := bilinovel.DownloadNovel(novelArgs.NovelId, novelArgs.outputPath)
+func runDownloadNovel() error {
+	downloader, err := bilinovel.New(bilinovel.BilinovelNewOption{
+		Concurrency: downloadArgs.concurrency,
+		Debug:       downloadArgs.debug,
+	})
 	if err != nil {
-		return fmt.Errorf("failed to download novel: %v", err)
+		return fmt.Errorf("failed to create downloader: %v", err)
 	}
-
-	return nil
+	// 确保在函数结束时关闭资源
+	defer func() {
+		if closeErr := downloader.Close(); closeErr != nil {
+			slog.Info("Failed to close downloader", slog.Any("error", closeErr))
 		}
+	}()
 
-func runDownloadVolume(cmd *cobra.Command, args []string) error {
-	if volumeArgs.NovelId == 0 {
+	if downloadArgs.NovelId == 0 {
 		return fmt.Errorf("novel id is required")
 	}
-	if volumeArgs.VolumeId == 0 {
-		return fmt.Errorf("volume id is required")
+
+	if downloadArgs.VolumeId == 0 {
+		// 下载整本小说
+		err := downloadNovel(downloader, downloadArgs.NovelId)
+		if err != nil {
+			return fmt.Errorf("failed to get novel: %v", err)
 		}
-	err := bilinovel.DownloadVolume(volumeArgs.NovelId, volumeArgs.VolumeId, volumeArgs.outputPath)
+	} else {
+		// 下载单卷
+		err = downloadVolume(downloader, downloadArgs.VolumeId)
 		if err != nil {
 			return fmt.Errorf("failed to download volume: %v", err)
 		}
+	}
 
 	return nil
 }
+
+func downloadNovel(downloader downloader.Downloader, novelId int) error {
+	novelInfo, err := downloader.GetNovel(novelId, true, nil)
+	if err != nil {
+		return fmt.Errorf("failed to get novel info: %w", err)
+	}
+	skipVolumes := make([]int, 0)
+	for _, volume := range novelInfo.Volumes {
+		jsonPath := filepath.Join(downloadArgs.outputPath, fmt.Sprintf("volume-%d-%d.json", downloadArgs.NovelId, volume.Id))
+		err = os.MkdirAll(filepath.Dir(jsonPath), 0755)
+		if err != nil {
+			return fmt.Errorf("failed to create directory: %v", err)
+		}
+		_, err = os.Stat(jsonPath)
+		if err == nil {
+			// 已经下载
+			skipVolumes = append(skipVolumes, volume.Id)
+		}
+	}
+	novel, err := downloader.GetNovel(novelId, false, skipVolumes)
+	if err != nil {
+		return fmt.Errorf("failed to download novel: %w", err)
+	}
+	for _, volume := range novel.Volumes {
+		jsonPath := filepath.Join(downloadArgs.outputPath, fmt.Sprintf("volume-%d-%d.json", downloadArgs.NovelId, volume.Id))
+		err = os.MkdirAll(filepath.Dir(jsonPath), 0755)
+		if err != nil {
+			return fmt.Errorf("failed to create directory: %v", err)
+		}
+		jsonFile, err := os.Create(jsonPath)
+		if err != nil {
+			return fmt.Errorf("failed to create json file: %v", err)
+		}
+		err = json.NewEncoder(jsonFile).Encode(volume)
+		if err != nil {
+			return fmt.Errorf("failed to encode json file: %v", err)
+		}
+		switch downloadArgs.outputType {
+		case "epub":
+			err = epub.PackVolumeToEpub(volume, downloadArgs.outputPath, downloader.GetStyleCSS(), downloader.GetExtraFiles())
+			if err != nil {
+				return fmt.Errorf("failed to pack volume: %v", err)
+			}
+		case "text":
+			err = text.PackVolumeToText(volume, downloadArgs.outputPath)
+			if err != nil {
+				return fmt.Errorf("failed to pack volume: %v", err)
+			}
+		}
+	}
+	return nil
+}
+
+func downloadVolume(downloader downloader.Downloader, volumeId int) error {
+	jsonPath := filepath.Join(downloadArgs.outputPath, fmt.Sprintf("volume-%d-%d.json", downloadArgs.NovelId, volumeId))
+	err := os.MkdirAll(filepath.Dir(jsonPath), 0755)
+	if err != nil {
+		return fmt.Errorf("failed to create directory: %v", err)
+	}
+	_, err = os.Stat(jsonPath)
+	volume := &model.Volume{}
+	if err != nil {
+		if os.IsNotExist(err) {
+			volume, err = downloader.GetVolume(downloadArgs.NovelId, volumeId, false)
+			if err != nil {
+				return fmt.Errorf("failed to get volume: %v", err)
+			}
+			jsonFile, err := os.Create(jsonPath)
+			if err != nil {
+				return fmt.Errorf("failed to create json file: %v", err)
+			}
+			err = json.NewEncoder(jsonFile).Encode(volume)
+			if err != nil {
+				return fmt.Errorf("failed to encode json file: %v", err)
+			}
+		} else {
+			return fmt.Errorf("failed to get volume: %v", err)
+		}
+	} else {
+		jsonFile, err := os.Open(jsonPath)
+		if err != nil {
+			return fmt.Errorf("failed to open json file: %v", err)
+		}
+		defer jsonFile.Close()
+		err = json.NewDecoder(jsonFile).Decode(volume)
+		if err != nil {
+			return fmt.Errorf("failed to decode json file: %v", err)
+		}
+	}
+
+	switch downloadArgs.outputType {
+	case "epub":
+		err = epub.PackVolumeToEpub(volume, downloadArgs.outputPath, downloader.GetStyleCSS(), downloader.GetExtraFiles())
+		if err != nil {
+			return fmt.Errorf("failed to pack volume: %v", err)
+		}
+	case "text":
+		err = text.PackVolumeToText(volume, downloadArgs.outputPath)
+		if err != nil {
+			return fmt.Errorf("failed to pack volume: %v", err)
+		}
+	}
+	return nil
+}

cmd/package.go

@@ -1,7 +1,7 @@
 package cmd
 
 import (
-	"bilinovel-downloader/downloader/bilinovel"
+	"bilinovel-downloader/epub"
 	"fmt"
 
 	"github.com/spf13/cobra"
@@ -28,7 +28,7 @@ func init() {
 }
 
 func runPackage(cmd *cobra.Command, args []string) error {
-	err := bilinovel.CreateEpub(pArgs.DirPath)
+	err := epub.PackEpub(pArgs.DirPath)
 	if err != nil {
 		return fmt.Errorf("failed to create epub: %v", err)
 	}

cmd/root.go

@@ -4,4 +4,6 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var RootCmd = &cobra.Command{}
+var RootCmd = &cobra.Command{
+	Use: "bilinovel-downloader",
+}

downloader/bilinovel/epub.go

@@ -1,122 +0,0 @@
-package bilinovel
-
-import (
-	"archive/zip"
-	"io"
-	"log"
-	"os"
-	"path/filepath"
-)
-
-func CreateEpub(path string) error {
-	log.Printf("Creating epub for %s", path)
-
-	savePath := path + ".epub"
-	zipFile, err := os.Create(savePath)
-	if err != nil {
-		return err
-	}
-	defer zipFile.Close()
-
-	zipWriter := zip.NewWriter(zipFile)
-	defer zipWriter.Close()
-
-	err = addStringToZip(zipWriter, "mimetype", "application/epub+zip", zip.Store)
-	if err != nil {
-		return err
-	}
-
-	err = addDirContentToZip(zipWriter, path, zip.Deflate)
-	if err != nil {
-		return err
-	}
-
-	err = addStringToZip(zipWriter, "OEBPS/Styles/style.css", StyleCSS, zip.Deflate)
-	if err != nil {
-		return err
-	}
-
-	return nil
-}
-
-// func addFileToZip(zipWriter *zip.Writer, filename string, relPath string, method uint16) error {
-// 	file, err := os.Open(filename)
-// 	if err != nil {
-// 		return err
-// 	}
-// 	defer file.Close()
-
-// 	info, err := file.Stat()
-// 	if err != nil {
-// 		return err
-// 	}
-
-// 	header, err := zip.FileInfoHeader(info)
-// 	if err != nil {
-// 		return err
-// 	}
-// 	header.Name = relPath
-// 	header.Method = method
-
-// 	writer, err := zipWriter.CreateHeader(header)
-// 	if err != nil {
-// 		return err
-// 	}
-
-// 	_, err = io.Copy(writer, file)
-// 	return err
-// }
-
-func addStringToZip(zipWriter *zip.Writer, relPath, content string, method uint16) error {
-	header := &zip.FileHeader{
-		Name:   relPath,
-		Method: method,
-	}
-	writer, err := zipWriter.CreateHeader(header)
-	if err != nil {
-		return err
-	}
-
-	_, err = writer.Write([]byte(content))
-	return err
-}
-
-func addDirContentToZip(zipWriter *zip.Writer, dirPath string, method uint16) error {
-	return filepath.Walk(dirPath, func(filePath string, info os.FileInfo, err error) error {
-		if filepath.Base(filePath) == "volume.json" {
-			return nil
-		}
-		if err != nil {
-			return err
-		}
-		if info.IsDir() {
-			return nil
-		}
-
-		relPath, err := filepath.Rel(dirPath, filePath)
-		if err != nil {
-			return err
-		}
-
-		file, err := os.Open(filePath)
-		if err != nil {
-			return err
-		}
-		defer file.Close()
-
-		header, err := zip.FileInfoHeader(info)
-		if err != nil {
-			return err
-		}
-		header.Name = relPath
-		header.Method = method
-
-		writer, err := zipWriter.CreateHeader(header)
-		if err != nil {
-			return err
-		}
-
-		_, err = io.Copy(writer, file)
-		return err
-	})
-}

downloader/bilinovel/style.css

@@ -1,15 +1,3 @@
-package bilinovel
-
-const StyleCSS = `
-@font-face {
-  font-family: "MI LANTING";
-  src: url(../Fonts/read.ttf);
-}
-
-.read-font {
-  font-family: "MI LANTING", serif !important;
-}
-
 body > div {
   margin: 0 auto;
   padding: 20px;
@@ -49,4 +37,3 @@ img {
   margin-top: 1em;
   margin-bottom: 1em;
 }
-`

downloader/downloader.go

@@ -0,0 +1,12 @@
+package downloader
+
+import "bilinovel-downloader/model"
+
+type Downloader interface {
+	GetNovel(novelId int, skipChapterContent bool, skipVolumes []int) (*model.Novel, error)
+	GetVolume(novelId int, volumeId int, skipChapterContent bool) (*model.Volume, error)
+	GetChapter(novelId int, volumeId int, chapterId int) (*model.Chapter, error)
+	GetStyleCSS() string
+	GetExtraFiles() []model.ExtraFile
+	Close() error
+}

epub/wrapper.go

@@ -0,0 +1,365 @@
+package epub
+
+import (
+	"archive/zip"
+	"bilinovel-downloader/model"
+	"bilinovel-downloader/template"
+	"bilinovel-downloader/utils"
+	"context"
+	"fmt"
+	"io"
+	"os"
+	"path"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+func PackVolumeToEpub(volume *model.Volume, outputPath string, styleCSS string, extraFiles []model.ExtraFile) error {
+	outputPath = filepath.Join(outputPath, utils.CleanDirName(volume.Title))
+	_, err := os.Stat(outputPath)
+	if err != nil {
+		if os.IsNotExist(err) {
+			err = os.MkdirAll(outputPath, 0755)
+			if err != nil {
+				return fmt.Errorf("failed to create output directory: %v", err)
+			}
+		} else {
+			return fmt.Errorf("failed to get output directory: %v", err)
+		}
+	} else {
+		err = os.RemoveAll(outputPath)
+		if err != nil {
+			return fmt.Errorf("failed to remove output directory: %v", err)
+		}
+		err = os.MkdirAll(outputPath, 0755)
+		if err != nil {
+			return fmt.Errorf("failed to create output directory: %v", err)
+		}
+	}
+
+	// 将文字写入 OEBPS/Text/chapter-%03v.xhtml
+	// 将图片写入 OEBPS/Images/chapter-%03v/
+	for i, chapter := range volume.Chapters {
+		imageNames := make([]string, 0)
+		for imgName, imgData := range chapter.Content.Images {
+			imageNames = append(imageNames, imgName)
+			imgPath := filepath.Join(outputPath, fmt.Sprintf("OEBPS/Images/chapter-%03v/%s", i, imgName))
+			err := os.MkdirAll(filepath.Dir(imgPath), 0755)
+			if err != nil {
+				return fmt.Errorf("failed to create image directory: %v", err)
+			}
+			err = os.WriteFile(imgPath, imgData, 0644)
+			if err != nil {
+				return fmt.Errorf("failed to write image: %v", err)
+			}
+		}
+		chapterPath := filepath.Join(outputPath, fmt.Sprintf("OEBPS/Text/chapter-%03v.xhtml", i))
+		err = os.MkdirAll(filepath.Dir(chapterPath), 0755)
+		if err != nil {
+			return fmt.Errorf("failed to create chapter directory: %v", err)
+		}
+		file, err := os.Create(chapterPath)
+		if err != nil {
+			return fmt.Errorf("failed to create chapter file: %v", err)
+		}
+		defer file.Close()
+		text := chapter.Content.Html
+		for _, imgName := range imageNames {
+			text = strings.ReplaceAll(text, imgName, fmt.Sprintf("../Images/chapter-%03v/%s", i, imgName))
+		}
+		err = template.ContentXHTML(chapter.Title, text).Render(context.Background(), file)
+		if err != nil {
+			return fmt.Errorf("failed to write chapter: %v", err)
+		}
+	}
+
+	// 将 Cover 写入
+	coverPath := filepath.Join(outputPath, fmt.Sprintf("cover%s", filepath.Ext(volume.CoverUrl)))
+	err = os.WriteFile(coverPath, volume.Cover, 0644)
+	if err != nil {
+		return fmt.Errorf("failed to write cover: %v", err)
+	}
+
+	// 将 CoverXHTML 写入 OEBPS/Text/cover.xhtml
+	coverXHTMLPath := filepath.Join(outputPath, "OEBPS/Text/cover.xhtml")
+	file, err := os.Create(coverXHTMLPath)
+	if err != nil {
+		return fmt.Errorf("failed to create cover XHTML file: %v", err)
+	}
+	defer file.Close()
+	err = template.CoverXHTML(fmt.Sprintf("../../%s", filepath.Base(coverPath))).Render(context.Background(), file)
+	if err != nil {
+		return fmt.Errorf("failed to render cover XHTML: %v", err)
+	}
+
+	// OEBPS/Text/contents.xhtml 目录
+	contentsXHTMLPath := filepath.Join(outputPath, "OEBPS/Text/contents.xhtml")
+	file, err = os.Create(contentsXHTMLPath)
+	if err != nil {
+		return fmt.Errorf("failed to create contents XHTML file: %v", err)
+	}
+	defer file.Close()
+	contents := strings.Builder{}
+	contents.WriteString(`<nav epub:type="toc" id="toc">`)
+	contents.WriteString(`<ol>`)
+	for i, chapter := range volume.Chapters {
+		contents.WriteString(fmt.Sprintf(`<li><a href="chapter-%03v.xhtml">%s</a></li>`, i, chapter.Title))
+	}
+	contents.WriteString(`</ol>`)
+	contents.WriteString(`</nav>`)
+	err = template.ContentXHTML("目录", contents.String()).Render(context.Background(), file)
+	if err != nil {
+		return fmt.Errorf("failed to render contents XHTML: %v", err)
+	}
+
+	// ContainerXML
+	containerPath := filepath.Join(outputPath, "META-INF/container.xml")
+	err = os.MkdirAll(filepath.Dir(containerPath), 0755)
+	if err != nil {
+		return fmt.Errorf("failed to create container directory: %v", err)
+	}
+	file, err = os.Create(containerPath)
+	if err != nil {
+		return fmt.Errorf("failed to create container file: %v", err)
+	}
+	defer file.Close()
+	err = template.ContainerXML().Render(context.Background(), file)
+	if err != nil {
+		return fmt.Errorf("failed to render container: %v", err)
+	}
+
+	// ContentOPF
+	u := uuid.New()
+	err = CreateContentOPF(outputPath, u.String(), volume, extraFiles)
+	if err != nil {
+		return fmt.Errorf("failed to create content OPF: %v", err)
+	}
+
+	// 写入 CSS
+	cssPath := filepath.Join(outputPath, "style.css")
+	err = os.WriteFile(cssPath, []byte(styleCSS), 0644)
+	if err != nil {
+		return fmt.Errorf("failed to write CSS: %v", err)
+	}
+
+	// 写入 extraFiles
+	for _, file := range extraFiles {
+		extraFilePath := filepath.Join(outputPath, file.Path)
+		err = os.WriteFile(extraFilePath, file.Data, 0644)
+		if err != nil {
+			return fmt.Errorf("failed to write extra file: %v", err)
+		}
+	}
+
+	// 打包成 epub 文件
+	err = PackEpub(outputPath)
+	if err != nil {
+		return fmt.Errorf("failed to pack epub: %v", err)
+	}
+	return nil
+}
+
+func CreateContentOPF(outputPath string, uuid string, volume *model.Volume, extraFiles []model.ExtraFile) error {
+	creators := make([]model.DCCreator, 0)
+	for _, author := range volume.Authors {
+		creators = append(creators, model.DCCreator{
+			Value: author,
+		})
+	}
+	dc := &model.DublinCoreMetadata{
+		Titles: []model.DCTitle{
+			{
+				Value: volume.Title,
+			},
+		},
+		Identifiers: []model.DCIdentifier{
+			{
+				Value: fmt.Sprintf("urn:uuid:%s", uuid),
+				ID:    "book-id",
+				// Scheme: "UUID",
+			},
+		},
+		Languages: []model.DCLanguage{
+			{
+				Value: "zh-CN",
+			},
+		},
+		Descriptions: []model.DCDescription{
+			{
+				Value: volume.Description,
+			},
+		},
+		Creators: creators,
+		Metas: []model.DublinCoreMeta{
+			{
+				Name:    "cover",
+				Content: "cover",
+			},
+			{
+				Property: "dcterms:modified",
+				Value:    time.Now().UTC().Format("2006-01-02T15:04:05Z"),
+			},
+			{
+				Name:    "calibre:series",
+				Content: volume.NovelTitle,
+			},
+			{
+				Name:    "calibre:series_index",
+				Content: strconv.Itoa(volume.SeriesIdx),
+			},
+		},
+	}
+	manifest := &model.Manifest{
+		Items: make([]model.ManifestItem, 0),
+	}
+	manifest.Items = append(manifest.Items, model.ManifestItem{
+		ID:    "cover.xhtml",
+		Link:  "OEBPS/Text/cover.xhtml",
+		Media: "application/xhtml+xml",
+	})
+	manifest.Items = append(manifest.Items, model.ManifestItem{
+		ID:         "contents.xhtml",
+		Link:       "OEBPS/Text/contents.xhtml",
+		Media:      "application/xhtml+xml",
+		Properties: "nav",
+	})
+	manifest.Items = append(manifest.Items, model.ManifestItem{
+		ID:         "cover",
+		Link:       fmt.Sprintf("cover%s", filepath.Ext(volume.CoverUrl)),
+		Media:      fmt.Sprintf("image/%s", strings.ReplaceAll(strings.TrimPrefix(filepath.Ext(volume.CoverUrl), "."), "jpg", "jpeg")),
+		Properties: "cover-image",
+	})
+	for i, chapter := range volume.Chapters {
+		manifest.Items = append(manifest.Items, model.ManifestItem{
+			ID:    fmt.Sprintf("chapter-%03v.xhtml", i),
+			Link:  fmt.Sprintf("OEBPS/Text/chapter-%03v.xhtml", i),
+			Media: "application/xhtml+xml",
+		})
+		for filename := range chapter.Content.Images {
+			item := model.ManifestItem{
+				ID:    fmt.Sprintf("chapter-%03v-%s", i, filepath.Base(filename)),
+				Link:  fmt.Sprintf("OEBPS/Images/chapter-%03v/%s", i, filepath.Base(filename)),
+				Media: fmt.Sprintf("image/%s", strings.ReplaceAll(strings.TrimPrefix(filepath.Ext(filename), "."), "jpg", "jpeg")),
+			}
+			manifest.Items = append(manifest.Items, item)
+		}
+	}
+	manifest.Items = append(manifest.Items, model.ManifestItem{
+		ID:    "style",
+		Link:  "style.css",
+		Media: "text/css",
+	})
+	// ExtraFiles
+	for _, file := range extraFiles {
+		manifest.Items = append(manifest.Items, file.ManifestItem)
+	}
+
+	spine := &model.Spine{
+		Items: make([]model.SpineItem, 0),
+	}
+	for _, item := range manifest.Items {
+		if filepath.Ext(item.Link) == ".xhtml" {
+			spine.Items = append(spine.Items, model.SpineItem{
+				IDref: item.ID,
+			})
+		}
+	}
+	contentOPFPath := filepath.Join(outputPath, "content.opf")
+	err := os.MkdirAll(path.Dir(contentOPFPath), 0755)
+	if err != nil {
+		return fmt.Errorf("failed to create content directory: %v", err)
+	}
+	file, err := os.Create(contentOPFPath)
+	if err != nil {
+		return fmt.Errorf("failed to create content file: %v", err)
+	}
+	err = template.ContentOPF("book-id", dc, manifest, spine, nil).Render(context.Background(), file)
+	if err != nil {
+		return fmt.Errorf("failed to render content: %v", err)
+	}
+	return nil
+}
+
+func PackEpub(dirPath string) error {
+	savePath := strings.TrimSuffix(dirPath, string(filepath.Separator)) + ".epub"
+	zipFile, err := os.Create(savePath)
+	if err != nil {
+		return err
+	}
+	defer zipFile.Close()
+
+	zipWriter := zip.NewWriter(zipFile)
+	defer zipWriter.Close()
+
+	err = addStringToZip(zipWriter, "mimetype", "application/epub+zip", zip.Store)
+	if err != nil {
+		return err
+	}
+
+	err = addDirContentToZip(zipWriter, dirPath, zip.Deflate)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func addStringToZip(zipWriter *zip.Writer, relPath, content string, method uint16) error {
+	header := &zip.FileHeader{
+		Name:   relPath,
+		Method: method,
+	}
+	writer, err := zipWriter.CreateHeader(header)
+	if err != nil {
+		return err
+	}
+
+	_, err = writer.Write([]byte(content))
+	return err
+}
+
+func addDirContentToZip(zipWriter *zip.Writer, dirPath string, method uint16) error {
+	return filepath.Walk(dirPath, func(filePath string, info os.FileInfo, err error) error {
+		if filepath.Base(filePath) == "volume.json" {
+			return nil
+		}
+		if err != nil {
+			return err
+		}
+		if info.IsDir() {
+			return nil
+		}
+
+		relPath, err := filepath.Rel(dirPath, filePath)
+		if err != nil {
+			return err
+		}
+
+		relPath = filepath.ToSlash(relPath)
+
+		file, err := os.Open(filePath)
+		if err != nil {
+			return err
+		}
+		defer file.Close()
+
+		header, err := zip.FileInfoHeader(info)
+		if err != nil {
+			return err
+		}
+		header.Name = relPath
+		header.Method = method
+
+		writer, err := zipWriter.CreateHeader(header)
+		if err != nil {
+			return err
+		}
+
+		_, err = io.Copy(writer, file)
+		return err
+	})
+}

go.mod

@@ -4,15 +4,22 @@ go 1.24.2
 
 require (
 	github.com/PuerkitoBio/goquery v1.10.3
-	github.com/a-h/templ v0.3.857
+	github.com/a-h/templ v0.3.943
+	github.com/bestnite/font-mapper v0.0.0-20250823155658-56c76d820267
 	github.com/go-resty/resty/v2 v2.16.5
 	github.com/google/uuid v1.6.0
+	github.com/playwright-community/playwright-go v0.5200.1
 	github.com/spf13/cobra v1.9.1
 )
 
 require (
 	github.com/andybalholm/cascadia v1.3.3 // indirect
+	github.com/deckarep/golang-set/v2 v2.8.0 // indirect
+	github.com/go-jose/go-jose/v3 v3.0.4 // indirect
+	github.com/go-stack/stack v1.8.1 // indirect
+	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
-	github.com/spf13/pflag v1.0.6 // indirect
-	golang.org/x/net v0.39.0 // indirect
+	github.com/spf13/pflag v1.0.7 // indirect
+	golang.org/x/image v0.30.0 // indirect
+	golang.org/x/net v0.43.0 // indirect
 )

go.sum

@@ -1,23 +1,48 @@
 github.com/PuerkitoBio/goquery v1.10.3 h1:pFYcNSqHxBD06Fpj/KsbStFRsgRATgnf3LeXiUkhzPo=
 github.com/PuerkitoBio/goquery v1.10.3/go.mod h1:tMUX0zDMHXYlAQk6p35XxQMqMweEKB7iK7iLNd4RH4Y=
-github.com/a-h/templ v0.3.857 h1:6EqcJuGZW4OL+2iZ3MD+NnIcG7nGkaQeF2Zq5kf9ZGg=
-github.com/a-h/templ v0.3.857/go.mod h1:qhrhAkRFubE7khxLZHsBFHfX+gWwVNKbzKeF9GlPV4M=
+github.com/a-h/templ v0.3.943 h1:o+mT/4yqhZ33F3ootBiHwaY4HM5EVaOJfIshvd5UNTY=
+github.com/a-h/templ v0.3.943/go.mod h1:oCZcnKRf5jjsGpf2yELzQfodLphd2mwecwG4Crk5HBo=
 github.com/andybalholm/cascadia v1.3.3 h1:AG2YHrzJIm4BZ19iwJ/DAua6Btl3IwJX+VI4kktS1LM=
 github.com/andybalholm/cascadia v1.3.3/go.mod h1:xNd9bqTn98Ln4DwST8/nG+H0yuB8Hmgu1YHNnWw0GeA=
+github.com/bestnite/font-mapper v0.0.0-20250823155658-56c76d820267 h1:nmUTJV2u/0XmVjQ++VIy/Hu+MtxdpQvOevvcSZtUATA=
+github.com/bestnite/font-mapper v0.0.0-20250823155658-56c76d820267/go.mod h1:cfB1e9YhoI/QWrXPp3h6QVAKU6iCI2ifbjRPHP3xf/0=
 github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/deckarep/golang-set/v2 v2.8.0 h1:swm0rlPCmdWn9mESxKOjWk8hXSqoxOp+ZlfuyaAdFlQ=
+github.com/deckarep/golang-set/v2 v2.8.0/go.mod h1:VAky9rY/yGXJOLEDv3OMci+7wtDpOF4IN+y82NBOac4=
+github.com/go-jose/go-jose/v3 v3.0.4 h1:Wp5HA7bLQcKnf6YYao/4kpRpVMp/yf6+pJKV8WFSaNY=
+github.com/go-jose/go-jose/v3 v3.0.4/go.mod h1:5b+7YgP7ZICgJDBdfjZaIt+H/9L9T/YQrVfLAMboGkQ=
 github.com/go-resty/resty/v2 v2.16.5 h1:hBKqmWrr7uRc3euHVqmh1HTHcKn99Smr7o5spptdhTM=
 github.com/go-resty/resty/v2 v2.16.5/go.mod h1:hkJtXbA2iKHzJheXYvQ8snQES5ZLGKMwQ07xAwp/fiA=
+github.com/go-stack/stack v1.8.1 h1:ntEHSVwIt7PNXNpgPmVfMrNhLtgjlmnZha2kOpuRiDw=
+github.com/go-stack/stack v1.8.1/go.mod h1:dcoOX6HbPZSZptuspn9bctJ+N/CnF5gGygcUP3XYfe4=
+github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g=
+github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k=
+github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
 github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
 github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
 github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/mitchellh/go-ps v1.0.0 h1:i6ampVEEF4wQFF+bkYfwYgY+F/uYJDktmvLPf7qIgjc=
+github.com/mitchellh/go-ps v1.0.0/go.mod h1:J4lOc8z8yJs6vUwklHw2XEIiT4z4C40KtWVN3nvg8Pg=
+github.com/playwright-community/playwright-go v0.5200.1 h1:Sm2oOuhqt0M5Y4kUi/Qh9w4cyyi3ZIWTBeGKImc2UVo=
+github.com/playwright-community/playwright-go v0.5200.1/go.mod h1:UnnyQZaqUOO5ywAZu60+N4EiWReUqX1MQBBA3Oofvf8=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
 github.com/spf13/cobra v1.9.1 h1:CXSaggrXdbHK9CF+8ywj8Amf7PBRmPCOJugH954Nnlo=
 github.com/spf13/cobra v1.9.1/go.mod h1:nDyEzZ8ogv936Cinf6g1RU9MRY64Ir93oCnqb9wxYW0=
-github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=
 github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/spf13/pflag v1.0.7 h1:vN6T9TfwStFPFM5XzjsvmzZkLuaLX+HS+0SeFLRgU6M=
+github.com/spf13/pflag v1.0.7/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
+github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
@@ -25,6 +50,8 @@ golang.org/x/crypto v0.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliY
 golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU=
 golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=
 golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
+golang.org/x/image v0.30.0 h1:jD5RhkmVAnjqaCUXfbGBrn3lpxbknfN9w2UhHHU+5B4=
+golang.org/x/image v0.30.0/go.mod h1:SAEUTxCCMWSrJcCy/4HwavEsfZZJlYxeHLc6tTiAe/c=
 golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
 golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
 golang.org/x/mod v0.12.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
@@ -39,8 +66,8 @@ golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk=
 golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44=
 golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
 golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
-golang.org/x/net v0.39.0 h1:ZCu7HMWDxpXpaiKdhzIfaltL9Lp31x/3fCP11bc6/fY=
-golang.org/x/net v0.39.0/go.mod h1:X7NRbYVEA+ewNkCNyJ513WmMdQ3BineSwVtN2zD/d+E=
+golang.org/x/net v0.43.0 h1:lat02VYK2j4aLzMzecihNvTlJNQUq316m2Mr9rnM6YE=
+golang.org/x/net v0.43.0/go.mod h1:vhO1fvI4dGsIjh73sWfUVjj3N7CA9WkKJNQm2svM6Jg=
 golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
@@ -87,4 +114,6 @@ golang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58
 golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxbQROHiO6hDPo2LHcIPhhQsa9DLh0yGk=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

main.go

@@ -2,11 +2,8 @@ package main
 
 import (
 	"bilinovel-downloader/cmd"
-	"log"
 )
 
 func main() {
-	if err := cmd.RootCmd.Execute(); err != nil {
-		log.Fatalf("Error executing command: %v", err)
-	}
+	_ = cmd.RootCmd.Execute()
 }

model/epub.go

@@ -1,6 +1,14 @@
 package model
 
-import "encoding/xml"
+import (
+	"encoding/xml"
+)
+
+type ExtraFile struct {
+	Data         []byte
+	Path         string
+	ManifestItem ManifestItem
+}
 
 type DublinCoreMetadata struct {
 	XMLName xml.Name `xml:"metadata"`
@@ -158,7 +166,6 @@ type Spine struct {
 }
 
 func (s *Spine) Marshal() (string, error) {
-	s.Toc = "ncx"
 	xmlBytes, err := xml.Marshal(s)
 	if err != nil {
 		return "", err

model/novel.go

@@ -1,13 +1,17 @@
 package model
 
+type ChaperContent struct {
+	Html   string
+	Images map[string][]byte
+}
+
 type Chapter struct {
+	Id       int
+	NovelId  int
+	VolumeId int
 	Title    string
 	Url      string
-	Content         string
-	ImageOEBPSPaths []string
-	ImageFullPaths  []string
-	TextOEBPSPath   string
-	TextFullPath    string
+	Content  *ChaperContent
 }
 
 type Volume struct {
@@ -15,7 +19,8 @@ type Volume struct {
 	SeriesIdx   int
 	Title       string
 	Url         string
-	Cover       string
+	CoverUrl    string
+	Cover       []byte
 	Description string
 	Authors     []string
 	Chapters    []*Chapter

model/toc_ncx.go

@@ -1,47 +0,0 @@
-package model
-
-import "encoding/xml"
-
-type TocNCXHead struct {
-	XMLName xml.Name         `xml:"head"`
-	Meta    []TocNCXHeadMeta `xml:"meta"`
-}
-
-type TocNCXHeadMeta struct {
-	XMLName xml.Name `xml:"meta"`
-	Content string   `xml:"content,attr"`
-	Name    string   `xml:"name,attr"`
-}
-
-func (h *TocNCXHead) Marshal() (string, error) {
-	xmlBytes, err := xml.Marshal(h)
-	if err != nil {
-		return "", err
-	}
-	return string(xmlBytes), nil
-}
-
-type NavPoint struct {
-	Id        string          `xml:"id,attr"`
-	PlayOrder int             `xml:"playOrder,attr"`
-	Label     string          `xml:"navLabel>text"`
-	Content   NavPointContent `xml:"content"`
-	NavPoints []*NavPoint     `xml:"navPoint"`
-}
-
-type NavPointContent struct {
-	Src string `xml:"src,attr"`
-}
-
-type NavMap struct {
-	XMLName xml.Name    `xml:"navMap"`
-	Points  []*NavPoint `xml:"navPoint"`
-}
-
-func (n *NavMap) Marshal() (string, error) {
-	xmlBytes, err := xml.Marshal(n)
-	if err != nil {
-		return "", err
-	}
-	return string(xmlBytes), nil
-}

template/container.xml.templ

@@ -1,10 +1,10 @@
 package template
 
 templ ContainerXML() {
-	@templ.Raw(`<?xml version="1.0"  encoding="UTF-8"?>`)
-	<container version="1.0" xmlns="urn:oasis:names:tc:opendocument:xmlns:container">
+	@templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`)
+	<container xmlns="urn:oasis:names:tc:opendocument:xmlns:container" version="1.0">
 		<rootfiles>
-			<rootfile full-path="OEBPS/content.opf" media-type="application/oebps-package+xml"></rootfile>
+			<rootfile full-path="content.opf" media-type="application/oebps-package+xml"></rootfile>
 		</rootfiles>
 	</container>
 }

template/container.xml_templ.go

@@ -1,6 +1,6 @@
 // Code generated by templ - DO NOT EDIT.
 
-// templ: version: v0.3.857
+// templ: version: v0.3.943
 package template
 
 //lint:file-ignore SA4006 This context is only used if a nested component is present.
@@ -29,11 +29,11 @@ func ContainerXML() templ.Component {
 			templ_7745c5c3_Var1 = templ.NopComponent
 		}
 		ctx = templ.ClearChildren(ctx)
-		templ_7745c5c3_Err = templ.Raw(`<?xml version="1.0"  encoding="UTF-8"?>`).Render(ctx, templ_7745c5c3_Buffer)
+		templ_7745c5c3_Err = templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`).Render(ctx, templ_7745c5c3_Buffer)
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
-		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 1, "<container version=\"1.0\" xmlns=\"urn:oasis:names:tc:opendocument:xmlns:container\"><rootfiles><rootfile full-path=\"OEBPS/content.opf\" media-type=\"application/oebps-package+xml\"></rootfile></rootfiles></container>")
+		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 1, "<container xmlns=\"urn:oasis:names:tc:opendocument:xmlns:container\" version=\"1.0\"><rootfiles><rootfile full-path=\"content.opf\" media-type=\"application/oebps-package+xml\"></rootfile></rootfiles></container>")
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}

template/content.opf.templ

@@ -3,7 +3,7 @@ package template
 import "bilinovel-downloader/model"
 
 templ ContentOPF(uniqueIdentifier string, dc *model.DublinCoreMetadata, manifest *model.Manifest, spine *model.Spine, guide *model.Guide) {
-	@templ.Raw(`<?xml version="1.0"  encoding="UTF-8"?>`)
+	@templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`)
 	<package version="3.0" xmlns="http://www.idpf.org/2007/opf" xmlns:dc="http://purl.org/dc/elements/1.1/" unique-identifier={ uniqueIdentifier }>
 		if dc != nil {
 			{{ metadata, err := dc.Marshal() }}

template/content.opf_templ.go

@@ -1,6 +1,6 @@
 // Code generated by templ - DO NOT EDIT.
 
-// templ: version: v0.3.857
+// templ: version: v0.3.943
 package template
 
 //lint:file-ignore SA4006 This context is only used if a nested component is present.
@@ -31,7 +31,7 @@ func ContentOPF(uniqueIdentifier string, dc *model.DublinCoreMetadata, manifest
 			templ_7745c5c3_Var1 = templ.NopComponent
 		}
 		ctx = templ.ClearChildren(ctx)
-		templ_7745c5c3_Err = templ.Raw(`<?xml version="1.0"  encoding="UTF-8"?>`).Render(ctx, templ_7745c5c3_Buffer)
+		templ_7745c5c3_Err = templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`).Render(ctx, templ_7745c5c3_Buffer)
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}

template/content.xhtml.templ

@@ -1,21 +1,19 @@
 package template
 
-import "bilinovel-downloader/model"
-
-templ ContentXHTML(content *model.Chapter) {
-	@templ.Raw(`<?xml version="1.0" encoding="utf-8" standalone="no"?>`)
-	@templ.Raw(`<!DOCTYPE html>`)
-	<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="zh-CN" xmlns:epub="http://www.idpf.org/2007/ops" xmlns:xml="http://www.w3.org/XML/1998/namespace">
+templ ContentXHTML(title, content string) {
+	@templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`)
+	// @templ.Raw(`<!DOCTYPE html>`)
+	<html xmlns="http://www.w3.org/1999/xhtml" xmlns:epub="http://www.idpf.org/2007/ops" xml:lang="zh-CN">
 		<head>
-			<title>{ content.Title }</title>
-			@templ.Raw(`<link href="../Styles/style.css" rel="stylesheet" type="text/css"/>`)
+			<title>{ title }</title>
+			@templ.Raw(`<link href="../../style.css" rel="stylesheet" type="text/css"/>`)
 		</head>
 		<body>
 			<div class="chapter">
-				<h1>{ content.Title }</h1>
+				<h1>{ title }</h1>
 				@templ.Raw(`<hr/>`)
 				<div class="content">
-					@templ.Raw(content.Content)
+					@templ.Raw(content)
 				</div>
 			</div>
 		</body>

template/content.xhtml_templ.go

@@ -1,6 +1,6 @@
 // Code generated by templ - DO NOT EDIT.
 
-// templ: version: v0.3.857
+// templ: version: v0.3.943
 package template
 
 //lint:file-ignore SA4006 This context is only used if a nested component is present.
@@ -8,9 +8,7 @@ package template
 import "github.com/a-h/templ"
 import templruntime "github.com/a-h/templ/runtime"
 
-import "bilinovel-downloader/model"
-
-func ContentXHTML(content *model.Chapter) templ.Component {
+func ContentXHTML(title, content string) templ.Component {
 	return templruntime.GeneratedTemplate(func(templ_7745c5c3_Input templruntime.GeneratedComponentInput) (templ_7745c5c3_Err error) {
 		templ_7745c5c3_W, ctx := templ_7745c5c3_Input.Writer, templ_7745c5c3_Input.Context
 		if templ_7745c5c3_CtxErr := ctx.Err(); templ_7745c5c3_CtxErr != nil {
@@ -31,22 +29,18 @@ func ContentXHTML(content *model.Chapter) templ.Component {
 			templ_7745c5c3_Var1 = templ.NopComponent
 		}
 		ctx = templ.ClearChildren(ctx)
-		templ_7745c5c3_Err = templ.Raw(`<?xml version="1.0" encoding="utf-8" standalone="no"?>`).Render(ctx, templ_7745c5c3_Buffer)
+		templ_7745c5c3_Err = templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`).Render(ctx, templ_7745c5c3_Buffer)
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
-		templ_7745c5c3_Err = templ.Raw(`<!DOCTYPE html>`).Render(ctx, templ_7745c5c3_Buffer)
-		if templ_7745c5c3_Err != nil {
-			return templ_7745c5c3_Err
-		}
-		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 1, "<html xmlns=\"http://www.w3.org/1999/xhtml\" xml:lang=\"zh-CN\" xmlns:epub=\"http://www.idpf.org/2007/ops\" xmlns:xml=\"http://www.w3.org/XML/1998/namespace\"><head><title>")
+		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 1, "<html xmlns=\"http://www.w3.org/1999/xhtml\" xmlns:epub=\"http://www.idpf.org/2007/ops\" xml:lang=\"zh-CN\"><head><title>")
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
 		var templ_7745c5c3_Var2 string
-		templ_7745c5c3_Var2, templ_7745c5c3_Err = templ.JoinStringErrs(content.Title)
+		templ_7745c5c3_Var2, templ_7745c5c3_Err = templ.JoinStringErrs(title)
 		if templ_7745c5c3_Err != nil {
-			return templ.Error{Err: templ_7745c5c3_Err, FileName: `template/content.xhtml.templ`, Line: 10, Col: 25}
+			return templ.Error{Err: templ_7745c5c3_Err, FileName: `template/content.xhtml.templ`, Line: 8, Col: 17}
 		}
 		_, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString(templ.EscapeString(templ_7745c5c3_Var2))
 		if templ_7745c5c3_Err != nil {
@@ -56,7 +50,7 @@ func ContentXHTML(content *model.Chapter) templ.Component {
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
-		templ_7745c5c3_Err = templ.Raw(`<link href="../Styles/style.css" rel="stylesheet" type="text/css"/>`).Render(ctx, templ_7745c5c3_Buffer)
+		templ_7745c5c3_Err = templ.Raw(`<link href="../../style.css" rel="stylesheet" type="text/css"/>`).Render(ctx, templ_7745c5c3_Buffer)
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
@@ -65,9 +59,9 @@ func ContentXHTML(content *model.Chapter) templ.Component {
 			return templ_7745c5c3_Err
 		}
 		var templ_7745c5c3_Var3 string
-		templ_7745c5c3_Var3, templ_7745c5c3_Err = templ.JoinStringErrs(content.Title)
+		templ_7745c5c3_Var3, templ_7745c5c3_Err = templ.JoinStringErrs(title)
 		if templ_7745c5c3_Err != nil {
-			return templ.Error{Err: templ_7745c5c3_Err, FileName: `template/content.xhtml.templ`, Line: 15, Col: 23}
+			return templ.Error{Err: templ_7745c5c3_Err, FileName: `template/content.xhtml.templ`, Line: 13, Col: 15}
 		}
 		_, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString(templ.EscapeString(templ_7745c5c3_Var3))
 		if templ_7745c5c3_Err != nil {
@@ -85,7 +79,7 @@ func ContentXHTML(content *model.Chapter) templ.Component {
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
-		templ_7745c5c3_Err = templ.Raw(content.Content).Render(ctx, templ_7745c5c3_Buffer)
+		templ_7745c5c3_Err = templ.Raw(content).Render(ctx, templ_7745c5c3_Buffer)
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}

template/cover.xhtml.templ

@@ -0,0 +1,37 @@
+package template
+
+templ CoverXHTML(coverPath string) {
+	@templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`)
+	<html xmlns="http://www.w3.org/1999/xhtml" xmlns:epub="http://www.idpf.org/2007/ops"
+	xml:lang="zh-CN">
+	<head>
+		<title>Cover</title>
+	</head>
+	<style type="text/css">
+		@page {
+		padding: 0pt;
+		margin: 0pt
+		}
+		body {
+		text-align: center;
+		padding: 0pt;
+		margin: 0pt;
+		}
+	</style>
+	<body>
+		<div>
+			<svg
+				xmlns="http://www.w3.org/2000/svg"
+				xmlns:xlink="http://www.w3.org/1999/xlink"
+				version="1.1"
+				width="100%"
+				height="100%"
+				viewBox="0 0 400 581"
+				preserveAspectRatio="none"
+			>
+				<image width="400" height="581" xlink:href={ coverPath }></image>
+			</svg>
+		</div>
+	</body>
+</html>
+}

template/cover.xhtml_templ.go

@@ -1,6 +1,6 @@
 // Code generated by templ - DO NOT EDIT.
 
-// templ: version: v0.3.857
+// templ: version: v0.3.943
 package template
 
 //lint:file-ignore SA4006 This context is only used if a nested component is present.
@@ -8,9 +8,7 @@ package template
 import "github.com/a-h/templ"
 import templruntime "github.com/a-h/templ/runtime"
 
-import "bilinovel-downloader/model"
-
-func TocNCX(title string, head *model.TocNCXHead, navMap *model.NavMap) templ.Component {
+func CoverXHTML(coverPath string) templ.Component {
 	return templruntime.GeneratedTemplate(func(templ_7745c5c3_Input templruntime.GeneratedComponentInput) (templ_7745c5c3_Err error) {
 		templ_7745c5c3_W, ctx := templ_7745c5c3_Input.Writer, templ_7745c5c3_Input.Context
 		if templ_7745c5c3_CtxErr := ctx.Err(); templ_7745c5c3_CtxErr != nil {
@@ -31,54 +29,24 @@ func TocNCX(title string, head *model.TocNCXHead, navMap *model.NavMap) templ.Co
 			templ_7745c5c3_Var1 = templ.NopComponent
 		}
 		ctx = templ.ClearChildren(ctx)
-		templ_7745c5c3_Err = templ.Raw(`<?xml version="1.0" encoding="UTF-8"?>`).Render(ctx, templ_7745c5c3_Buffer)
+		templ_7745c5c3_Err = templ.Raw(`<?xml version='1.0' encoding='utf-8'?>`).Render(ctx, templ_7745c5c3_Buffer)
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
-		templ_7745c5c3_Err = templ.Raw(`<!DOCTYPE ncx PUBLIC "-//NISO//DTD ncx 2005-1//EN" "http://www.daisy.org/z3986/2005/ncx-2005-1.dtd">`).Render(ctx, templ_7745c5c3_Buffer)
-		if templ_7745c5c3_Err != nil {
-			return templ_7745c5c3_Err
-		}
-		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 1, "<ncx xmlns=\"http://www.daisy.org/z3986/2005/ncx/\" version=\"2005-1\">")
-		if templ_7745c5c3_Err != nil {
-			return templ_7745c5c3_Err
-		}
-		if head != nil {
-			head, err := head.Marshal()
-			if err == nil {
-				templ_7745c5c3_Err = templ.Raw(head).Render(ctx, templ_7745c5c3_Buffer)
-				if templ_7745c5c3_Err != nil {
-					return templ_7745c5c3_Err
-				}
-			}
-		}
-		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 2, "<docTitle><text>")
+		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 1, "<html xmlns=\"http://www.w3.org/1999/xhtml\" xmlns:epub=\"http://www.idpf.org/2007/ops\" xml:lang=\"zh-CN\"><head><title>Cover</title></head><style type=\"text/css\">\n\t\t@page {\n\t\tpadding: 0pt;\n\t\tmargin: 0pt\n\t\t}\n\t\tbody {\n\t\ttext-align: center;\n\t\tpadding: 0pt;\n\t\tmargin: 0pt;\n\t\t}\n\t</style><body><div><svg xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\" version=\"1.1\" width=\"100%\" height=\"100%\" viewBox=\"0 0 400 581\" preserveAspectRatio=\"none\"><image width=\"400\" height=\"581\" xlink:href=\"")
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
 		var templ_7745c5c3_Var2 string
-		templ_7745c5c3_Var2, templ_7745c5c3_Err = templ.JoinStringErrs(title)
+		templ_7745c5c3_Var2, templ_7745c5c3_Err = templ.JoinStringErrs(coverPath)
 		if templ_7745c5c3_Err != nil {
-			return templ.Error{Err: templ_7745c5c3_Err, FileName: `template/toc.ncx.templ`, Line: 16, Col: 16}
+			return templ.Error{Err: templ_7745c5c3_Err, FileName: `template/cover.xhtml.templ`, Line: 32, Col: 58}
 		}
 		_, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString(templ.EscapeString(templ_7745c5c3_Var2))
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}
-		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 3, "</text></docTitle> ")
-		if templ_7745c5c3_Err != nil {
-			return templ_7745c5c3_Err
-		}
-		if navMap != nil {
-			navMap, err := navMap.Marshal()
-			if err == nil {
-				templ_7745c5c3_Err = templ.Raw(navMap).Render(ctx, templ_7745c5c3_Buffer)
-				if templ_7745c5c3_Err != nil {
-					return templ_7745c5c3_Err
-				}
-			}
-		}
-		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 4, "</ncx>")
+		templ_7745c5c3_Err = templruntime.WriteString(templ_7745c5c3_Buffer, 2, "\"></image></svg></div></body></html>")
 		if templ_7745c5c3_Err != nil {
 			return templ_7745c5c3_Err
 		}

template/toc.ncx.templ

@@ -1,25 +0,0 @@
-package template
-
-import "bilinovel-downloader/model"
-
-templ TocNCX(title string, head *model.TocNCXHead, navMap *model.NavMap) {
-	@templ.Raw(`<?xml version="1.0" encoding="UTF-8"?>`)
-	@templ.Raw(`<!DOCTYPE ncx PUBLIC "-//NISO//DTD ncx 2005-1//EN" "http://www.daisy.org/z3986/2005/ncx-2005-1.dtd">`)
-	<ncx xmlns="http://www.daisy.org/z3986/2005/ncx/" version="2005-1">
-		if head != nil {
-			{{ head, err := head.Marshal() }}
-			if err == nil {
-				@templ.Raw(head)
-			}
-		}
-		<docTitle>
-			<text>{ title }</text>
-		</docTitle>
-		if navMap != nil {
-			{{ navMap, err := navMap.Marshal() }}
-			if err == nil {
-				@templ.Raw(navMap)
-			}
-		}
-	</ncx>
-}

test/bilinovel_test.go

@@ -0,0 +1,59 @@
+package test
+
+import (
+	"bilinovel-downloader/downloader/bilinovel"
+	"encoding/json"
+	"fmt"
+	"testing"
+)
+
+func TestBilinovel_GetNovel(t *testing.T) {
+	bilinovel, err := bilinovel.New(bilinovel.BilinovelNewOption{Concurrency: 5})
+	bilinovel.SetTextOnly(true)
+	if err != nil {
+		t.Fatalf("failed to create bilinovel: %v", err)
+	}
+	novel, err := bilinovel.GetNovel(2727, false, nil)
+	if err != nil {
+		t.Fatalf("failed to get novel: %v", err)
+	}
+	jsonBytes, err := json.Marshal(novel)
+	if err != nil {
+		t.Fatalf("failed to marshal novel: %v", err)
+	}
+	fmt.Println(string(jsonBytes))
+}
+
+func TestBilinovel_GetVolume(t *testing.T) {
+	bilinovel, err := bilinovel.New(bilinovel.BilinovelNewOption{Concurrency: 1})
+	bilinovel.SetTextOnly(true)
+	if err != nil {
+		t.Fatalf("failed to create bilinovel: %v", err)
+	}
+	volume, err := bilinovel.GetVolume(2727, 129092, false)
+	if err != nil {
+		t.Fatalf("failed to get volume: %v", err)
+	}
+	jsonBytes, err := json.Marshal(volume)
+	if err != nil {
+		t.Fatalf("failed to marshal volume: %v", err)
+	}
+	fmt.Println(string(jsonBytes))
+}
+
+func TestBilinovel_GetChapter(t *testing.T) {
+	bilinovel, err := bilinovel.New(bilinovel.BilinovelNewOption{Concurrency: 1})
+	bilinovel.SetTextOnly(true)
+	if err != nil {
+		t.Fatalf("failed to create bilinovel: %v", err)
+	}
+	chapter, err := bilinovel.GetChapter(2727, 129092, 129094)
+	if err != nil {
+		t.Fatalf("failed to get chapter: %v", err)
+	}
+	jsonBytes, err := json.Marshal(chapter)
+	if err != nil {
+		t.Fatalf("failed to marshal chapter: %v", err)
+	}
+	fmt.Println(string(jsonBytes))
+}

text/wrapper.go

@@ -0,0 +1,55 @@
+package text
+
+import (
+	"bilinovel-downloader/model"
+	"bilinovel-downloader/utils"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/PuerkitoBio/goquery"
+)
+
+func PackVolumeToText(volume *model.Volume, outputPath string) error {
+	outputPath = filepath.Join(outputPath, utils.CleanDirName(volume.Title))
+	_, err := os.Stat(outputPath)
+	if err != nil {
+		if os.IsNotExist(err) {
+			err = os.MkdirAll(outputPath, 0755)
+			if err != nil {
+				return fmt.Errorf("failed to create output directory: %v", err)
+			}
+		} else {
+			return fmt.Errorf("failed to get output directory: %v", err)
+		}
+	} else {
+		err = os.RemoveAll(outputPath)
+		if err != nil {
+			return fmt.Errorf("failed to remove output directory: %v", err)
+		}
+		err = os.MkdirAll(outputPath, 0755)
+		if err != nil {
+			return fmt.Errorf("failed to create output directory: %v", err)
+		}
+	}
+	for i, chapter := range volume.Chapters {
+		chapterPath := filepath.Join(outputPath, fmt.Sprintf("%03d-%s.txt", i, chapter.Title))
+		chapterFile, err := os.Create(chapterPath)
+		if err != nil {
+			return fmt.Errorf("failed to create chapter file: %v", err)
+		}
+		defer chapterFile.Close()
+		doc, err := goquery.NewDocumentFromReader(strings.NewReader(chapter.Content.Html))
+		if err != nil {
+			return fmt.Errorf("failed to create chapter file: %v", err)
+		}
+		doc.Find("img").Remove()
+		text := doc.Text()
+		_, err = chapterFile.WriteString(strings.TrimSpace(text))
+		if err != nil {
+			return fmt.Errorf("failed to write chapter file: %v", err)
+		}
+	}
+	return nil
+}

utils/request.go

@@ -9,11 +9,19 @@ import (
 	"github.com/go-resty/resty/v2"
 )
 
-var client *resty.Client
+type RestyClient struct {
+	client      *resty.Client
+	concurrency int
+	sem         chan struct{}
+}
 
-func init() {
-	client = resty.New()
-	client.SetTransport(&http.Transport{
+func NewRestyClient(concurrency int) *RestyClient {
+	client := &RestyClient{
+		client:      resty.New(),
+		concurrency: concurrency,
+		sem:         make(chan struct{}, concurrency),
+	}
+	client.client.SetTransport(&http.Transport{
 		DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
 			if addr == "www.bilinovel.com:443" {
 				addr = "64.140.161.52:443"
@@ -24,7 +32,16 @@ func init() {
 		},
 		TLSHandshakeTimeout: 10 * time.Second,
 	})
-	client.SetRetryCount(10).
+	client.client.
+		OnBeforeRequest(func(c *resty.Client, req *resty.Request) error {
+			client.sem <- struct{}{}
+			return nil
+		}).
+		OnAfterResponse(func(c *resty.Client, resp *resty.Response) error {
+			<-client.sem
+			return nil
+		})
+	client.client.SetRetryCount(10).
 		SetRetryWaitTime(3 * time.Second).
 		SetRetryAfter(func(client *resty.Client, resp *resty.Response) (time.Duration, error) {
 			if resp.StatusCode() == http.StatusTooManyRequests {
@@ -43,10 +60,13 @@ func init() {
 		AddRetryCondition(func(r *resty.Response, err error) bool {
 			return err != nil || r.StatusCode() == http.StatusTooManyRequests
 		})
+
+	client.client.SetLogger(disableLogger{}).SetHeader("Accept-Charset", "utf-8").SetHeader("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:133.0) Gecko/20100101 Firefox/133.0")
+	return client
 }
 
-func Request() *resty.Request {
-	return client.R().SetLogger(disableLogger{}).SetHeader("Accept-Charset", "utf-8").SetHeader("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:133.0) Gecko/20100101 Firefox/133.0")
+func (c *RestyClient) R() *resty.Request {
+	return c.client.R()
 }
 
 type disableLogger struct{}