update 2021年07月25日 9:10:21

Author: No-Github

.gitignore

@@ -0,0 +1 @@
+logo2.png

README.md

@@ -1,9 +1,37 @@
+# 简介
+
 给文件夹内图片添加水印的工具
 
 源码来自 : https://blog.csdn.net/czy279470138/article/details/96993285
 
 仅作部分修改
 
+## 效果
+
+```
+main.exe -img 11.png -width 300 -hight 300
+```
+
+![](./demo.png)
+
+## 使用命令
+
+单文件添加水印
+```
+./main.exe -img 1.png
+```
+
+文件夹内图片添加水印
+```
+./main.exe -dir tmp
+```
+
+自定义水印 (默认logo.png)
+```
+./main.exe -logo logo2.png -img 1.png
+```
+
+水印区间 (默认皆为600)
 ```
-./main.exe -name "1.png"
+./main.exe -width 500 -hight 500 -img 1.png
 ```

demo.png

@@ -9,9 +9,14 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"io/ioutil"
 )
 
-var Input_name = flag.String("name", "null", "输入文件名")
+var Input_img = flag.String("img", "", "输入文件名")
+var Input_dir = flag.String("dir", "", "输入文件夹名")
+var Input_logo = flag.String("logo", "logo.png", "输入 logo 名(例如:logo.png)")
+var Input_width = flag.Int("width", 600, "输入宽度")
+var Input_hight = flag.Int("hight", 600, "输入高度")
 
 func main() {
 	flag.Parse()
@@ -20,18 +25,26 @@ func main() {
 		fmt.Printf("arg[%d]=%s\n", i, flag.Arg(i))
 	}
 
-	if *Input_name == ""{
-		fmt.Println("文件夹?")
+	if *Input_img == ""{
+		*Input_img="null"
 	}else {
-		fmt.Println("目标图片:",*Input_name)
 		imgwater()
 	}
 
+	if *Input_dir == ""{
+		*Input_dir="null"
+	}else {
+		imgwater_dir()
+	}
+
 }
 
 func imgwater() {
-	srcImg := *Input_name             // 原始图片
-	imgWaterMarkPath := "./logo.png" // 水印图片
+
+	fmt.Println("目标图片:",*Input_img)
+
+	srcImg := *Input_img             // 原始图片
+	imgWaterMarkPath := *Input_logo // 水印图片
 
 	// 原始图片
 	originalImg, err := os.Open(srcImg)
@@ -68,7 +81,7 @@ func imgwater() {
 	imgType := http.DetectContentType(buff)
 
 	if imgType == "image/jpeg" {
-		fmt.Println("这是JPG文件")
+		//fmt.Println("这是JPG文件")
 		imgJpeg, err := jpeg.Decode(originalImg)
 		if err != nil {
 			fmt.Println("把jpeg图片解码为结构体时出错")
@@ -77,29 +90,32 @@ func imgwater() {
 		b := imgJpeg.Bounds()
 		waterMarkWidth := b.Max.X
 		waterMarkHeight := b.Max.Y
-		fmt.Println("jpeg原始图片宽", waterMarkWidth, "jpeg原始图片高", waterMarkHeight)
+		//fmt.Println("jpeg原始图片宽", waterMarkWidth, "jpeg原始图片高", waterMarkHeight)
 
 		m := image.NewRGBA(b)
 
 		// 原始图片
 		draw.Draw(m, b, imgJpeg, image.ZP, draw.Src)
 		// 水印图片
 		for i := 0; i < waterMarkWidth; i++ {
-			offsetWidth := 1000 * i
+			offsetWidth := *Input_width * i
 			if offsetWidth < waterMarkWidth {
 				for j := 0; j < waterMarkHeight; j++ {
-					offsetHeight := 1000 * j
+					offsetHeight := *Input_hight * j
 					if offsetHeight < waterMarkHeight {
 						offset := image.Pt(offsetWidth, offsetHeight)
 						draw.Draw(m, waterMarkImg.Bounds().Add(offset), waterMarkImg, image.ZP, draw.Over)
-						fmt.Println("jpegOffset", offset)
+						//fmt.Println("jpegOffset", offset)
 					}
 				}
 			}
 		}
 
+
 		// 生成新图片new.jpg,并设置图片质量 100
-		imgNew, err := os.Create("new.jpg") // 这里可以设置为上传图片 srcImg
+		imgNew, err := os.Create(*Input_img) // 这里可以设置为上传图片 srcImg
+
+
 		if err != nil {
 			log.Println(err)
 		}
@@ -115,15 +131,15 @@ func imgwater() {
 	}
 
 	if imgType == "image/png" {
-		fmt.Println("这是PNG文件")
+		//fmt.Println("这是PNG文件")
 		imgPng, err := png.Decode(originalImg)
 		if err != nil {
 			fmt.Println("把PNG图片解码为结构体时出错")
 		}
 		b := imgPng.Bounds()
 		waterMarkWidth := b.Max.X
 		waterMarkHeight := b.Max.Y
-		fmt.Println("png原始图片宽", waterMarkWidth, "png原始图片高", waterMarkHeight)
+		//fmt.Println("png原始图片宽", waterMarkWidth, "png原始图片高", waterMarkHeight)
 
 		m := image.NewRGBA(b)
 
@@ -133,22 +149,22 @@ func imgwater() {
 		// 水印图片
 		for i := 0; i < waterMarkWidth; i++ {
 			// resWidth := 0
-			offsetWidth := 1000 * i
+			offsetWidth := *Input_width * i
 			// fmt.Println(offsetWidth)
 			if offsetWidth < waterMarkWidth {
 				for j := 0; j < waterMarkHeight; j++ {
-					offsetHeight := 1000 * j
+					offsetHeight := *Input_hight * j
 					if offsetHeight < waterMarkHeight {
 						offset := image.Pt(offsetWidth, offsetHeight)
 						draw.Draw(m, waterMarkImg.Bounds().Add(offset), waterMarkImg, image.ZP, draw.Over)
-						fmt.Println("pngOffset", offset)
+						//fmt.Println("pngOffset", offset)
 					}
 				}
 			}
 		}
 
 		// 生成新图片new.png
-		imgNew, err := os.Create("./new.png") // 这里可以设置为上传图片 srcImg
+		imgNew, err := os.Create(*Input_img) // 这里可以设置为上传图片 srcImg
 		if err != nil {
 			log.Println(err)
 		}
@@ -164,3 +180,12 @@ func imgwater() {
 		fmt.Println("暂不支持 gif 格式。。。")
 	}
 }
+
+func imgwater_dir() {
+	fmt.Println("目标文件夹:",*Input_dir)
+	files, _ := ioutil.ReadDir(*Input_dir)
+	for _, f := range files {
+		*Input_img=*Input_dir + "/" +f.Name()
+		imgwater()
+	}
+}