package main

import (
	"fmt"
	"image/jpeg"
	"os"
	"strconv"
)

func main() {
	if !checkImg() {
		fmt.Println("########################################")
		fmt.Println("输入参数不符合，请按照要求输入")
		fmt.Println("1、第一个参数为原图位置")
		fmt.Println("2、第二个参数为保留图片质量,数字型:0~100")
		fmt.Println("3、第一个参数为压缩图位置")
		fmt.Println("########################################")
		return
	}
	// 打开一张图片
	srcFile, err := os.Open(os.Args[1])
	if err != nil {
		panic(err)
	}
	defer func(srcFile *os.File) {
		_ = srcFile.Close()
	}(srcFile)

	// 解码图片
	srcImg, err := jpeg.Decode(srcFile)
	if err != nil {
		panic(err)
	}

	// 压缩图片到指定质量，例如质量为80
	quality, _ := strconv.Atoi(os.Args[2])
	options := &jpeg.Options{Quality: quality}

	// 创建一个图片写入器
	dstFile, err := os.Create(os.Args[3])
	if err != nil {
		panic(err)
	}
	defer func(dstFile *os.File) {
		_ = dstFile.Close()
	}(dstFile)

	// 编码压缩后的图片
	err = jpeg.Encode(dstFile, srcImg, options)
	if err != nil {
		panic(err)
	}
}

func checkImg() bool {
	if len(os.Args) != 4 {
		return false
	}
	_, err := strconv.Atoi(os.Args[2])
	if err != nil {
		return false
	}
	return true
}