mirror of
https://shylinux.com/x/icebergs
synced 2025-04-26 17:44:05 +08:00
117 lines
2.7 KiB
Go
117 lines
2.7 KiB
Go
package cli
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"github.com/skip2/go-qrcode"
|
|
"image/color"
|
|
"math/rand"
|
|
"strconv"
|
|
"strings"
|
|
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/aaa"
|
|
kit "github.com/shylinux/toolkits"
|
|
)
|
|
|
|
var _trans_web = map[string]color.Color{
|
|
"black": color.RGBA{0, 0, 0, 255},
|
|
"red": color.RGBA{255, 0, 0, 255},
|
|
"green": color.RGBA{0, 255, 0, 255},
|
|
"yellow": color.RGBA{255, 255, 0, 255},
|
|
"blue": color.RGBA{0, 0, 255, 255},
|
|
"magenta": color.RGBA{255, 0, 255, 255},
|
|
"cyan": color.RGBA{0, 255, 255, 255},
|
|
"white": color.RGBA{255, 255, 255, 255},
|
|
}
|
|
|
|
func _parse_color(str string) color.Color {
|
|
if str == "random" {
|
|
list := []string{}
|
|
for k := range _trans_web {
|
|
list = append(list, k)
|
|
}
|
|
str = list[rand.Intn(len(list))]
|
|
}
|
|
if strings.HasPrefix(str, "#") {
|
|
if len(str) == 7 {
|
|
str += "ff"
|
|
}
|
|
if u, e := strconv.ParseUint(str[1:], 16, 64); e == nil {
|
|
return color.RGBA{
|
|
uint8((u & 0xFF000000) >> 24),
|
|
uint8((u & 0x00FF0000) >> 16),
|
|
uint8((u & 0x0000FF00) >> 8),
|
|
uint8((u & 0x000000FF) >> 0),
|
|
}
|
|
}
|
|
}
|
|
return _trans_web[str]
|
|
}
|
|
func _trans_cli(str string) string {
|
|
res := 0
|
|
r, g, b, _ := _parse_color(str).RGBA()
|
|
if r > 128 {
|
|
res += 1
|
|
}
|
|
if g > 128 {
|
|
res += 2
|
|
}
|
|
if b > 128 {
|
|
res += 4
|
|
}
|
|
return kit.Format(res)
|
|
}
|
|
|
|
func _qrcode_cli(m *ice.Message, text string, arg ...string) {
|
|
qr, _ := qrcode.New(text, qrcode.Medium)
|
|
fg := _trans_cli(m.Option("fg"))
|
|
bg := _trans_cli(m.Option("bg"))
|
|
|
|
data := qr.Bitmap()
|
|
for i, row := range data {
|
|
if n := len(data); i < 3 || i >= n-3 {
|
|
continue
|
|
}
|
|
for i, col := range row {
|
|
if n := len(row); i < 3 || i >= n-3 {
|
|
continue
|
|
}
|
|
|
|
m.Echo("\033[4%sm \033[0m", kit.Select(bg, fg, col))
|
|
}
|
|
m.Echo("\n")
|
|
}
|
|
}
|
|
func _qrcode_web(m *ice.Message, text string, arg ...string) {
|
|
qr, _ := qrcode.New(text, qrcode.Medium)
|
|
qr.ForegroundColor = _parse_color(m.Option("fg"))
|
|
qr.BackgroundColor = _parse_color(m.Option("bg"))
|
|
|
|
if data, err := qr.PNG(kit.Int(m.Option("size"))); m.Assert(err) {
|
|
m.Echo(`<img src="data:image/png;base64,%s" title='%s'>`, base64.StdEncoding.EncodeToString(data), text)
|
|
}
|
|
}
|
|
|
|
const QRCODE = "qrcode"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
QRCODE: {Name: "qrcode", Help: "二维码", Value: kit.Data()},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
QRCODE: {Name: "qrcode text fg bg size auto", Help: "二维码", Action: map[string]*ice.Action{}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
m.Option("fg", kit.Select("blue", arg, 1))
|
|
m.Option("bg", kit.Select("white", arg, 2))
|
|
m.Option("size", kit.Select("240", arg, 3))
|
|
|
|
if aaa.SessIsCli(m) {
|
|
_qrcode_cli(m, arg[0])
|
|
} else {
|
|
_qrcode_web(m, arg[0])
|
|
}
|
|
}},
|
|
},
|
|
})
|
|
}
|