-
Notifications
You must be signed in to change notification settings - Fork 0
/
qrcode.go
75 lines (62 loc) · 1.98 KB
/
qrcode.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package image
import (
"fmt"
"image/color"
"log"
qrcode "github.com/skip2/go-qrcode"
"github.com/spf13/cobra"
)
var (
size, level int
fileName string
disableBorder bool
)
// qrcodeCmd represents the qrcode command
var qrcodeCmd = &cobra.Command{
Use: "qrcode",
Short: "生成二维码",
Long: `将输入的内容生成二维码, 并生成png文件:
生成二维码
1. 当前目录快速生成二维码, 名字默认为 output.png
knife qrcode "https://clibing.com"
2. 有边框,大小512,recovery level 2 输出到 /tmp/512.png 二维码的内容是 "https://clibing.com"
knife qrcode -l 2 -s 512 -f /tmp/512.png "https://clibing.com"
3. 无边框,大小512,recovery level 2 输出到 /tmp/512.png 二维码的内容是 "https://clibing.com"
knife qrcode -d -l 2 -s 512 -f /tmp/512.png "https://clibing.com"
.`,
Run: func(_ *cobra.Command, args []string) {
if len(args) != 1 {
log.Fatal("请输入二维码的内容")
return
}
recoveryLevel := qrcode.Medium
if level == 2 {
recoveryLevel = qrcode.High
} else if level == 3 {
recoveryLevel = qrcode.Highest
}
q, err := qrcode.New(args[0], recoveryLevel)
if err != nil {
log.Fatalf("创建二维码基础参数异常 %s", err)
return
}
q.BackgroundColor = color.White
q.ForegroundColor = color.Black
q.DisableBorder = disableBorder
err = q.WriteFile(size, fileName)
if err != nil {
log.Fatalf("生成二维码异常 %s", err)
return
}
fmt.Println("二维码生成成功, ", fileName)
},
}
func init() {
qrcodeCmd.Flags().IntVarP(&size, "size", "s", 256, "二维码的size")
qrcodeCmd.Flags().StringVarP(&fileName, "fileName", "f", "./output.png", "输出二维码图片完整路径")
qrcodeCmd.Flags().IntVarP(&level, "level", "l", 1, "生成图片质量,默认是1(15%), 2(25%), 3(30%)")
qrcodeCmd.Flags().BoolVarP(&disableBorder, "disableBorder", "d", false, "是否禁用二维码边框, 默认不禁用")
}
func NewQrcodeCmd() *cobra.Command {
return qrcodeCmd
}