NanoBot/helper.go

package nano

import (
	"encoding/base64"
	"encoding/hex"
	"hash/crc64"
	"net/url"
	"regexp"
	"runtime"
	"strings"
	"unsafe"
)

func getFuncAndFileNameWithSkip(n int) (string, string) {
	pc, fn, _, ok := runtime.Caller(n)
	if !ok {
		return "", ""
	}
	i := strings.LastIndex(fn, "/") + 1
	if i > 0 {
		fn = strings.TrimSuffix(fn[i:], ".go")
	}
	fullname := runtime.FuncForPC(pc).Name()
	i = strings.LastIndex(fullname, ".") + 1
	if i <= 0 || i >= len(fullname) {
		return fullname, fn
	}
	return fullname[i:], fn
}

// getThisFuncName 获取正在执行的函数名
func getThisFuncName() string {
	x, _ := getFuncAndFileNameWithSkip(2)
	return x
}

// getCallerFuncName 获取调用者函数名
func getCallerFuncName() string {
	x, _ := getFuncAndFileNameWithSkip(3)
	return x
}

// getLogHeader [文件名.函数名]
func getLogHeader() string {
	funcname, filename := getFuncAndFileNameWithSkip(2)
	return "[" + filename + "." + funcname + "]"
}

// MessageEscape 消息转义
//
// https://bot.q.qq.com/wiki/develop/api/openapi/message/message_format.html
func MessageEscape(text string) string {
	text = strings.ReplaceAll(text, "&", "&amp;")
	text = strings.ReplaceAll(text, "<", "&lt;")
	text = strings.ReplaceAll(text, ">", "&gt;")
	return text
}

// MessageUnescape 消息解转义
//
// https://bot.q.qq.com/wiki/develop/api/openapi/message/message_format.html
func MessageUnescape(text string) string {
	text = strings.ReplaceAll(text, "&amp;", "&")
	text = strings.ReplaceAll(text, "&lt;", "<")
	text = strings.ReplaceAll(text, "&gt;", ">")
	return text
}

// HideURL 转义 URL 以避免审核
func HideURL(s string) string {
	s = strings.ReplaceAll(s, ".", "…")
	s = strings.ReplaceAll(s, "http://", "🔗📄:")
	s = strings.ReplaceAll(s, "https://", "🔗🔒:")
	return s
}

// UnderlineToCamel convert abc_def to AbcDef
func UnderlineToCamel(s string) string {
	sb := strings.Builder{}
	isnextupper := true
	for _, c := range []byte(strings.ToLower(s)) {
		if c == '_' {
			isnextupper = true
			continue
		}
		if isnextupper {
			sb.WriteString(strings.ToUpper(string(c)))
			isnextupper = false
			continue
		}
		sb.WriteByte(c)
	}
	return sb.String()
}

// resolveURI github.com/wdvxdr1123/ZeroBot/driver/uri.go
func resolveURI(addr string) (network, address string) {
	network, address = "tcp", addr
	uri, err := url.Parse(addr)
	if err == nil && uri.Scheme != "" {
		scheme, ext, _ := strings.Cut(uri.Scheme, "+")
		if ext != "" {
			network = ext
			uri.Scheme = scheme // remove `+unix`/`+tcp4`
			if ext == "unix" {
				uri.Host, uri.Path, _ = strings.Cut(uri.Path, ":")
				uri.Host = base64.StdEncoding.EncodeToString(StringToBytes(uri.Host)) // special handle for unix
			}
			address = uri.String()
		}
	}
	return
}

// slice is the runtime representation of a slice.
// It cannot be used safely or portably and its representation may
// change in a later release.
//
// Unlike reflect.SliceHeader, its Data field is sufficient to guarantee the
// data it references will not be garbage collected.
type slice struct {
	data unsafe.Pointer
	len  int
	cap  int
}

// BytesToString 没有内存开销的转换
func BytesToString(b []byte) string {
	return *(*string)(unsafe.Pointer(&b))
}

// StringToBytes 没有内存开销的转换
func StringToBytes(s string) (b []byte) {
	bh := (*slice)(unsafe.Pointer(&b))
	sh := (*slice)(unsafe.Pointer(&s))
	bh.data = sh.data
	bh.len = sh.len
	bh.cap = sh.len
	return b
}

// DigestID 归一化 id 为 uint64
func DigestID(id string) uint64 {
	b, err := hex.DecodeString(id)
	if err != nil || len(b) < 8 {
		return 0
	}
	return crc64.Checksum(b, crc64.MakeTable(crc64.ECMA))
}

const mediafilebed = `https://multimedia.nt.qq.com.cn`

var mediafileinfourlre = regexp.MustCompile(`/download\?appid=\d+&fileid=[0-9A-Za-z-_]+&rkey=[0-9A-Za-z-_]+`)

// mediaURL 从 fileinfo 得到 URL
func mediaURL(fileinfo string) (string, error) {
	sb := strings.Builder{}
	data, err := base64.StdEncoding.DecodeString(fileinfo)
	if err != nil {
		return "", err
	}
	sb.WriteString(mediafilebed)
	sb.Write(mediafileinfourlre.Find(data))
	return sb.String(), nil
}