优化图片覆盖逻辑，新增日志详细模式

Author: Ran-Xing

.idea/.gitignore

@@ -25,20 +25,22 @@ import (
 	"time"
 )
 
+// S3 配置及全局变量
 var (
-	Endpoint    = "s3.bitiful.net"
-	Region      = "cn-east-1"
-	SVC         *s3.S3
-	update      bool
-	file        string
-	name        string
-	HttpClient  *http.Client
-	versionData string
-	fileName    string
-	imageType   bool
-	err         error
+	Endpoint    = "s3.bitiful.net" // S3 终端节点
+	Region      = "cn-east-1"      // S3 区域
+	SVC         *s3.S3             // S3 客户端
+	file        string             // 文件路径或URL
+	name        string             // 指定上传后的文件名
+	verbose     bool               // 是否显示详细日志
+	HttpClient  *http.Client       // 全局 HTTP 客户端
+	versionData string             // 版本信息
+	fileName    string             // 实际上传到S3的文件名
+	imageType   bool               // 是否为本地文件（true=本地，false=URL）
+	err         error              // 全局错误变量
 )
 
+// 初始化配置和HTTP客户端
 func init() {
 	dir, err := os.UserConfigDir()
 	if err != nil {
@@ -57,7 +59,7 @@ func init() {
 		Region = viper.GetString("Region")
 	}
 
-	// HttpClient 注意client 本身是连接池，不要每次请求时创建client
+	// 初始化全局 HTTP 客户端，避免频繁创建
 	dialer := &net.Dialer{
 		Resolver: &net.Resolver{
 			PreferGo: true,
@@ -82,58 +84,70 @@ func init() {
 	}
 }
 
+// 主命令定义，支持批量文件/URL上传
 var rootCmd = &cobra.Command{
-	Short:   "TYPORA UPLOAD IAMGE TO AWS",
-	Long:    "优雅的使用 TYPORA 上传图片到 AWS 存储桶",
+	Short:   "Typora 上传图片到 AWS",
+	Long:    "优雅地使用 Typora 上传图片到 AWS 存储桶",
 	Version: versionData,
 
 	Run: func(cmd *cobra.Command, args []string) {
+		// 如果没有图片路径参数，则用 -f 指定的文件
 		if args == nil || len(args) == 0 {
 			args = append(args, file)
 		}
-		//log.Infof("args %v", args)
-		//log.Infof("update {%v} file {%v} name {%v}", update, file, name)
+		log.Debugf("args: %v", args)
+		log.Debugf("file: %v, name: %v", file, name)
 		for _, v := range args {
 			var data []byte
+			log.Debugf("处理文件/URL: %s", v)
 			getImgName(v)
+			log.Debugf("最终上传文件名: %s", fileName)
 			imageType = checkImg(v)
-			// 判断文件是否存在
+			log.Debugf("文件类型判断: %v (true=本地文件, false=URL)", imageType)
+			// 判断是本地文件还是URL，分别处理
 			if !imageType {
+				log.Debugf("尝试下载图片: %s", v)
 				data, err = getImageUrl(v)
 				if err != nil {
 					log.Fatalln(err)
 					continue
 				}
 			} else {
+				log.Debugf("尝试读取本地图片: %s", v)
 				data, err = getImageFile(v)
 				if err != nil {
 					log.Fatalln(err)
 					continue
 				}
 			}
 
+			log.Debugf("开始上传到 S3: %s", fileName)
 			_, err := UploadObject(data)
 			if err != nil {
 				log.Fatalln(err)
 				continue
 			}
+			log.Debugf("上传成功: %s", fileName)
 
 			if imageType {
+				// 上传图片后删除原文件
 				if err := wastebasket.Trash(v); err != nil {
 					log.Errorf("删除文件失败: %v", err)
 					return
 				}
+				log.Debugf("移动图片到垃圾篓成功: %s", fileName)
 			}
 			fmt.Printf("https://%v.%v%v%v?fmt=webp&q=48&w=800\n", viper.GetString("BucketName"), Endpoint, viper.GetString("Path"), fileName)
 		}
 	},
 	Example: `  - bitiful /Users/xxx/xxx.png
   - bitiful "/example/example.png" "/example/example2.png"
-  - bitiful -u -f/Users/xxx/xxx.png -nTest.png
-  - bitiful -u -f /Users/xxx/xxx.png -n Test.png
+  - bitiful -n Test.png /Users/xxx/xxx.png
+  - bitiful -n Test.png https://example.com/xxx.png
 `,
 }
 
+// 创建 S3 会话
 func CreateS3Session() {
 	SVC = s3.New(session.Must(session.NewSession(
 		&aws.Config{
@@ -150,21 +164,26 @@ func CreateS3Session() {
 	)))
 }
 
+// 生成最终上传文件名：有 -n 用 -n，否则用时间戳
 func getImgName(image string) {
-	//filetype := http.DetectContentType(data)
-	fileNameAll := path.Base(image) // 文件全名
-	fileSuffix := path.Ext(image)   // 文件类型
-	fileName = fileNameAll
+	fileSuffix := path.Ext(image) // 文件类型
 	if fileSuffix == "" {
 		fileSuffix = ".png"
 	}
 
-	// 检测是否覆盖
-	if !update {
-		fileName = strings.Replace(time.Now().Format("20060102150405.00000"), ".", "", -1) + fileSuffix
+	// 有 -n 就用 -n，否则一律生成新名字
+	if name != "" {
+		fileName = name
+		if !strings.HasSuffix(fileName, fileSuffix) {
+			fileName += fileSuffix
+		}
+		return
 	}
+
+	fileName = strings.Replace(time.Now().Format("20060102150405.00000"), ".", "", -1) + fileSuffix
 }
 
+// 判断文件是否为本地文件
 func checkImg(file string) bool {
 	if _, err := os.Stat(file); err != nil {
 		if os.IsNotExist(err) {
@@ -176,6 +195,7 @@ func checkImg(file string) bool {
 	return false
 }
 
+// 下载图片（URL）
 func getImageUrl(imgUrl string) ([]byte, error) {
 	resp, err := HttpClient.Get(imgUrl)
 	if err != nil {
@@ -188,18 +208,19 @@ func getImageUrl(imgUrl string) ([]byte, error) {
 		}
 	}(resp.Body)
 
-	// Check Status
+	// 检查HTTP状态码
 	if resp.StatusCode != 200 {
 		return nil, errors.New(fmt.Sprintf("下载图片的状态码异常: %v", resp.Status))
 	}
-	// Read the data
+	// 读取图片数据
 	data, err := io.ReadAll(resp.Body)
 	if err != nil {
 		return nil, errors.New(fmt.Sprintf("读取图片失败: %v", err))
 	}
 	return data, nil
 }
 
+// 读取本地图片文件
 func getImageFile(imgPath string) ([]byte, error) {
 	tempFile, err := os.Open(imgPath)
 	if err != nil {
@@ -217,8 +238,8 @@ func getImageFile(imgPath string) ([]byte, error) {
 	return data, nil
 }
 
+// 上传图片到 S3
 func UploadObject(raw []byte) (output *s3.PutObjectOutput, err error) {
-	// Upload to s3
 	output, err = SVC.PutObjectWithContext(context.TODO(), &s3.PutObjectInput{
 		Bucket: aws.String(viper.GetString("BucketName")),
 		Key:    aws.String(viper.GetString("Path") + fileName),
@@ -234,12 +255,25 @@ func UploadObject(raw []byte) (output *s3.PutObjectOutput, err error) {
 	return output, nil
 }
 
+// 程序入口
 func main() {
-	// upload img/png/jpeg/webp to aws
+	// 初始化 S3 会话
 	CreateS3Session()
-	rootCmd.Flags().BoolVarP(&update, "update", "u", false, "更新您的图片.")
+	// 注册命令行参数
 	rootCmd.Flags().StringVarP(&file, "file", "f", "", "图片 路径|URL .")
 	rootCmd.Flags().StringVarP(&name, "name", "n", "", "被覆盖的图片名称.")
+	rootCmd.Flags().BoolVarP(&verbose, "verbose", "v", false, "显示详细日志")
+
+	// 日志级别初始化
+	cobra.OnInitialize(func() {
+		if verbose {
+			log.SetLevel(log.DebugLevel)
+			log.Debug("已启用详细日志模式")
+		} else {
+			log.SetLevel(log.InfoLevel)
+		}
+	})
+
 	if err := rootCmd.Execute(); err != nil {
 		log.Errorf("参数错误: %v", err)
 	}

.idea/bitiful-typora-upload-cli.iml

@@ -1,33 +1,38 @@
-## Bitiful Typora Upload CLI
+## Bitiful Typora 图片上传 命令行工具
 
 ![show1](https://xrsec.s3.bitiful.net/IMG/2023011808315652353.gif?fmt=webp&q=48)
 
 ![show2](https://xrsec.s3.bitiful.net/IMG/2023011808320175950.gif?fmt=webp&q=48)
 
-## Install & Use
+## 安装与使用
 
-1. [Release Download](https://github.com/XRSec/bitiful-typora-upload-cli/releases)
+1. [发布版下载](https://github.com/XRSec/bitiful-typora-upload-cli/releases)
 
 ```bash
-bitiful # require config file
+bitiful # 需要配置文件
 bitiful "/example/example.png"
 bitiful "https://example.com/example.png"
+bitiful /path/to/1.png /path/to/2.jpg # 支持多文件批量上传
 ```
 
-or dev build
+2. 开发构建
 
 ```bash
 git clone https://github.com/XRSec/bitiful-typora-upload-cli.git
 cd bitiful-typora-upload-cli
-# Edit bitifulUrl in bitiful-typora-upload-cli.go
+# 编辑 bitifulUrl 变量
 CGO_ENABLED=0 go build -o bitiful
 # mv bitiful /usr/local/bin
 # cd .. && rm -rf bitiful-typora-upload-cli
 ```
 
-2. Open Typora and set `cli` -> `Preference` -> `Image` -> `Upload Image` -> `Custom Command` -> `/home/xxx/bitiful`
+3. 打开 Typora，`偏好设置` -> `图片` -> `上传图片` -> `自定义命令` -> `/home/xxx/bitiful`
 
-## Config
+## 配置文件
+
+配置文件路径：`~/Library/Application\ Support/bitiful.yml`
+
+### 配置文件示例
 
 ```yaml
 Endpoint: s3.bitiful.net
@@ -38,14 +43,36 @@ BucketName: "xxxxxxxxxx"
 Path: "/xxxxxxxx/"
 ```
 
-### Cover image
+### 指定文件名（覆盖上传）
+
+如需自定义上传后的文件名（如封面图），可用 `-n` 参数：
 
 ```bash
-bitiful -u -n "20220903" -f https://example.com/20220903.jpg
-bitiful -u -f/Users/xxx/xxx.png -nTest.png
+bitiful -n "20220903.jpg" https://example.com/20220903.jpg
+bitiful -n Test.png /Users/xxx/xxx.png
 ```
-## Doing
 
-- [x] file
-- [x] url
+- 仅当指定 `-n` 时，才会用该名字覆盖同名文件。
+- 未指定 `-n` 时，自动生成唯一文件名，绝不覆盖。
+
+### 多文件批量上传
+
+直接在命令行追加多个文件或URL即可：
+
+```bash
+bitiful /path/to/1.png /path/to/2.jpg
+```
+
+### 日志与调试
+
+加 `-v` 参数可显示详细日志，便于排查问题：
+
+```bash
+bitiful -v /path/to/1.png
+```
+
+## 功能进度
+
+- [x] 本地文件上传
+- [x] URL 上传
 - [?] ftp/smb/afs

.idea/modules.xml

@@ -5,7 +5,6 @@ go 1.18
 require (
 	github.com/Bios-Marcel/wastebasket v0.0.2
 	github.com/aws/aws-sdk-go v1.44.180
-	github.com/shamsher31/goimgtype v1.0.0
 	github.com/sirupsen/logrus v1.9.0
 	github.com/spf13/cobra v1.6.1
 	github.com/spf13/viper v1.14.0
@@ -20,7 +19,6 @@ require (
 	github.com/mitchellh/mapstructure v1.5.0 // indirect
 	github.com/pelletier/go-toml v1.9.5 // indirect
 	github.com/pelletier/go-toml/v2 v2.0.5 // indirect
-	github.com/shamsher31/goimgext v1.0.0 // indirect
 	github.com/spf13/afero v1.9.2 // indirect
 	github.com/spf13/cast v1.5.0 // indirect
 	github.com/spf13/jwalterweatherman v1.1.0 // indirect

.idea/vcs.xml

@@ -158,10 +158,6 @@ github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:
 github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=
 github.com/rogpeppe/go-internal v1.6.1 h1:/FiVV8dS/e+YqF2JvO3yXRFbBLTIuSDkuC7aBOAvL+k=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
-github.com/shamsher31/goimgext v1.0.0 h1:wFKf9GeeE0Xr6UQtliaPgYYgTju2izobM7XpCEgUCC8=
-github.com/shamsher31/goimgext v1.0.0/go.mod h1:rYLKgXuTGBIaH49z+jUVSWz7gUWIZmqvYUsdvJbNNOc=
-github.com/shamsher31/goimgtype v1.0.0 h1:7nmPz5GEb01sjFUojCbZJcBI68A8tmLXoCFbb5I34u4=
-github.com/shamsher31/goimgtype v1.0.0/go.mod h1:OZm3NZQDbK0ezlk9IV5oq3cWQwjF1oVSxKNx19bkf64=
 github.com/sirupsen/logrus v1.9.0 h1:trlNQbNUG3OdDrDil03MCb1H2o9nJ1x4/5LYw7byDE0=
 github.com/sirupsen/logrus v1.9.0/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
 github.com/spf13/afero v1.9.2 h1:j49Hj62F0n+DaZ1dDCvhABaPNSGNkt32oRFxI33IEMw=