/
main.go
83 lines (72 loc) · 1.81 KB
/
main.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
76
77
78
79
80
81
82
83
package main
import (
"flag"
"fmt"
"io"
"os"
"runtime"
"strings"
"github.com/mattn/go-colorable"
"github.com/mdp/qrterminal/v3"
"rsc.io/qr"
)
var verboseFlag bool
var levelFlag string
var quietZoneFlag int
var sixelDisableFlag bool
func getLevel(s string) qr.Level {
switch l := strings.ToLower(s); l {
case "l":
return qr.L
case "m":
return qr.M
case "h":
return qr.H
default:
return -1
}
}
func main() {
flag.BoolVar(&verboseFlag, "v", false, "Output debugging information")
flag.StringVar(&levelFlag, "l", "L", "Error correction level")
flag.IntVar(&quietZoneFlag, "q", 2, "Size of quietzone border")
flag.BoolVar(&sixelDisableFlag, "s", false, "disable sixel format for output")
flag.Parse()
level := getLevel(levelFlag)
content := strings.Join(flag.Args(), " ")
if len(content) < 1 {
// Get input from stdin until EOF
stdin, err := io.ReadAll(os.Stdin)
if err != nil {
panic(err)
}
content = string(stdin)
} else if level < 0 {
fmt.Fprintf(os.Stderr, "Invalid error correction level: %s\n", levelFlag)
fmt.Fprintf(os.Stderr, "Valid options are [L, M, H]\n")
os.Exit(1)
}
cfg := qrterminal.Config{
Level: level,
Writer: os.Stdout,
QuietZone: quietZoneFlag,
BlackChar: qrterminal.BLACK,
WhiteChar: qrterminal.WHITE,
}
if !sixelDisableFlag {
cfg.WithSixel = qrterminal.IsSixelSupported(os.Stdout)
}
if verboseFlag {
fmt.Fprintf(os.Stdout, "Level: %s \n", levelFlag)
fmt.Fprintf(os.Stdout, "Quietzone Border Size: %d \n", quietZoneFlag)
fmt.Fprintf(os.Stdout, "Encoded data: %s \n", strings.Join(flag.Args(), "\n"))
fmt.Println("")
}
if runtime.GOOS == "windows" {
cfg.Writer = colorable.NewColorableStdout()
cfg.BlackChar = qrterminal.BLACK
cfg.WhiteChar = qrterminal.WHITE
}
fmt.Fprint(os.Stdout, "\n")
qrterminal.GenerateWithConfig(content, cfg)
}