From ca5a1dbda6bea0fc638ff632e1dd404b367d6a30 Mon Sep 17 00:00:00 2001
From: ystyle <lxy5266@live.com>
Date: Wed, 28 Jun 2023 15:11:47 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E7=94=9F=E6=88=90Orl?=
 =?UTF-8?q?y=E9=A3=8E=E6=A0=BC=E7=9A=84=E4=B9=A6=E7=B1=8D=E5=B0=81?=
 =?UTF-8?q?=E9=9D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 convert.go | 22 +++++++++++++---
 tools.go   | 74 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 93 insertions(+), 3 deletions(-)

diff --git a/convert.go b/convert.go
index de76a15..7036684 100644
--- a/convert.go
+++ b/convert.go
@@ -34,6 +34,8 @@ type Book struct {
 	Align          string    // 标题对齐方式
 	UnknowTitle    string    // 未知章节名称
 	Cover          string    // 封面图片
+	CoverOrlyColor string    // 生成封面图片的颜色
+	CoverOrlyIdx   int       // 生成封面图片的动物
 	Bottom         string    // 段阿落间距
 	LineHeight     string    // 行高
 	Tips           bool      // 是否添加教程文本
@@ -119,13 +121,15 @@ func NewBookArgs() *Book {
 	flag.StringVar(&book.UnknowTitle, "unknowtitle", "章节正文", "未知章节默认名称")
 	flag.UintVar(&book.Indent, "indent", 2, "段落缩进字数")
 	flag.StringVar(&book.Align, "align", GetEnv("KAF_CLI_ALIGN", "center"), "标题对齐方式: left、center、righ。环境变量KAF_CLI_ALIGN可修改默认值")
-	flag.StringVar(&book.Cover, "cover", "cover.png", "封面图片")
+	flag.StringVar(&book.Cover, "cover", "cover.png", "封面图片可为: 本地图片, 和orly。 设置为orly时生成orly风格的封面, 需要连接网络。")
+	flag.StringVar(&book.CoverOrlyColor, "coverorlycolor", "", "orly封面的主题色, 可以为1-16和hex格式的颜色代码, 不填时随机")
+	flag.IntVar(&book.CoverOrlyIdx, "coverorlyidx", -1, "orly封面的动物, 可以为0-41, 不填时随机, 具体图案可以查看: https://orly.nanmu.me")
 	flag.StringVar(&book.Bottom, "bottom", "1em", "段落间距(单位可以为em、px)")
 	flag.StringVar(&book.LineHeight, "lineheight", "", "行高(用于设置行间距, 默认为1.5rem)")
 	flag.StringVar(&book.Format, "format", GetEnv("KAF_CLI_FORMAT", "all"), "书籍格式: all、epub、mobi、azw3。环境变量KAF_CLI_FORMAT可修改默认值")
 	flag.StringVar(&book.Lang, "lang", GetEnv("KAF_CLI_LANG", "zh"), "设置语言: en,de,fr,it,es,zh,ja,pt,ru,nl。环境变量KAF_CLI_LANG可修改默认值")
-	flag.BoolVar(&book.Tips, "tips", true, "添加本软件教程")
 	flag.StringVar(&book.Out, "out", "", "输出文件名，不需要包含格式后缀")
+	flag.BoolVar(&book.Tips, "tips", true, "添加本软件教程")
 	flag.Parse()
 	return &book
 }
@@ -167,9 +171,21 @@ func (book *Book) Check(version string) error {
 		book.Out = book.Bookname
 	}
 	book.Lang = parseLang(book.Lang)
-	if exists, _ := isExists(book.Cover); !exists {
+	switch book.Cover {
+	case "none":
 		book.Cover = ""
+	case "gen":
+		cover, err := GenCover(book.Bookname, book.Author, book.CoverOrlyColor, book.CoverOrlyIdx)
+		if err != nil {
+			panic(err)
+		}
+		book.Cover = cover
+	default:
+		if exists, _ := isExists(book.Cover); !exists {
+			book.Cover = ""
+		}
 	}
+
 	// 编译正则表达式
 	if book.Match == "" {
 		book.Match = DefaultMatchTips
diff --git a/tools.go b/tools.go
index f7f702e..3dc4fdb 100644
--- a/tools.go
+++ b/tools.go
@@ -2,12 +2,16 @@ package kafcli
 
 import (
 	"fmt"
+	"io"
 	"io/ioutil"
 	"math/rand"
+	"net/http"
+	"net/url"
 	"os"
 	"os/exec"
 	"path/filepath"
 	"runtime"
+	"strconv"
 	"strings"
 	"time"
 )
@@ -104,3 +108,73 @@ func GetEnv(key, defaultvalue string) string {
 	}
 	return defaultvalue
 }
+
+var colors = []string{
+	"#61005e",
+	"#70706d",
+	"#890029",
+	"#c4000e",
+	"#6d001d",
+	"#6a00bd",
+	"#f10000",
+	"#0071b1",
+	"#f9bc00",
+	"#2c0077",
+	"#ba009a",
+	"#009047",
+	"#009d9e",
+	"#222e85",
+	"#bd002e",
+	"#009d1a",
+	"#75a500",
+}
+
+func ParseInt(v string) int {
+	v = strings.ReplaceAll(v, ",", "")
+	i, err := strconv.ParseInt(v, 0, 32)
+	if err != nil {
+		return 0
+	}
+	return int(i)
+}
+
+func GenCover(title, author, color string, img int) (string, error) {
+	query := url.Values{}
+	query.Add("title", title)
+	query.Add("author", author)
+	query.Add("g_loc", "BR")
+	query.Add("top_text", "kaf")
+	query.Add("g_text", "")
+	if img >= 0 && img <= 41 {
+		query.Add("img_id", fmt.Sprintf("%d", img))
+	} else {
+		query.Add("img_id", fmt.Sprintf("%d", rand.Intn(41)))
+	}
+	if strings.HasPrefix(color, "#") {
+		query.Add("color", strings.TrimLeft(color, "#"))
+	} else {
+		i := ParseInt(color)
+		if i == 0 {
+			i = rand.Intn(17)
+		}
+		color := colors[i]
+		query.Add("color", strings.TrimLeft(color, "#"))
+	}
+
+	uri := fmt.Sprintf("https://orly.nanmu.me/api/generate?%s", query.Encode())
+	res, err := http.Get(uri)
+	if err != nil {
+		return "", err
+	}
+	tempDir, err := os.MkdirTemp("", "kaf-cli")
+	bs, err := io.ReadAll(res.Body)
+	if err != nil {
+		return "", err
+	}
+	coverfile := filepath.Join(tempDir, fmt.Sprintf("%s.jpg", title))
+	err = os.WriteFile(coverfile, bs, 0666)
+	if err != nil {
+		return "", err
+	}
+	return coverfile, nil
+}