Sillyv2/core/node_strings.go

package core

import (
	"bytes"
	"fmt"
	"math/rand"
	"net/url"
	"path/filepath"
	"regexp"
	"strconv"
	"strings"
	"time"

	"github.com/cdle/sillyplus/emoji"
	"github.com/cdle/sillyplus/utils"
)

type Strings struct {
}

func (sender *Strings) Random(length int, substr string) string {
	ws := "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
	if substr != "" {
		ws = substr
	}
	rand.Seed(time.Now().UnixNano())
	letters := []rune(ws)
	b := make([]rune, length)
	for i := range b {
		b[i] = letters[rand.Intn(len(letters))]
	}
	return string(b)
}

func (sender *Strings) JoinFilepath(elem ...string) string {
	return filepath.Join(elem...)
}

func (sender *Strings) Trim(s, cutset string) string {
	return strings.Trim(s, cutset)
}
func (sender *Strings) TrimLeft(s, cutset string) string {
	return strings.TrimLeft(s, cutset)
}

func (sender *Strings) TrimRight(s, cutset string) string {
	return strings.TrimRight(s, cutset)
}

func (sender *Strings) Filename(path string) string {
	re := regexp.MustCompile(`[\\/]+`)
	parts := re.Split(path, -1)
	filename := parts[len(parts)-1]
	return filename
}

func (sender *Strings) Dir(path string) string {
	re := regexp.MustCompile(`[\\/]+`)
	parts := re.Split(path, -1)
	filename := parts[len(parts)-1]
	dir := path[:len(path)-len(filename)]
	dir = filepath.Clean(dir)
	return dir
}

func (sender *Strings) Contains(s, substr string) bool {
	return strings.Contains(s, substr)
}
func (sender *Strings) Remove(ss []string, s string) []string {
	return utils.Remove(ss, s)
}
func (sender *Strings) HasPrefix(s, substr string) bool {
	return strings.HasPrefix(s, substr)
}
func (sender *Strings) HasSuffix(s, substr string) bool {
	return strings.HasSuffix(s, substr)
}

func (sender *Strings) Replace(s string, old string, new string, n int) string {
	if n == 0 {
		n = -1
	}
	return strings.Replace(s, old, new, n)
}

func (sender *Strings) ReplaceAll(s string, old string, new string) string {
	return strings.ReplaceAll(s, old, new)
}

func (sender *Strings) Split(s string, sep string, n int) []string {
	return strings.SplitN(s, sep, n)
}

func (sender *Strings) EncodeQueryString(params map[string]interface{}) string {
	var buf bytes.Buffer
	for key, value := range params {
		if buf.Len() > 0 {
			buf.WriteByte('&')
		}
		buf.WriteString(url.QueryEscape(key))
		buf.WriteByte('=')
		switch v := value.(type) {
		case string:
			buf.WriteString(url.QueryEscape(v))
		case int:
			buf.WriteString(strconv.Itoa(v))
		case int64:
			buf.WriteString(strconv.Itoa(int(v)))
		case int32:
			buf.WriteString(strconv.Itoa(int(v)))
		case bool:
			buf.WriteString(strconv.FormatBool(v))
		default:
			buf.WriteString(url.QueryEscape(fmt.Sprintf("%v", v)))
		}
	}
	return buf.String()
}

func (sender *Strings) DecodeQueryString(querystring string) map[string]interface{} {
	values, err := url.ParseQuery(querystring)
	if err != nil {
		panic(err)
	}
	params := make(map[string]interface{})
	for key, values := range values {
		if len(values) > 0 {
			value := values[0]
			if intValue, err := strconv.Atoi(value); err == nil {
				params[key] = intValue
			} else if boolValue, err := strconv.ParseBool(value); err == nil {
				params[key] = boolValue
			} else {
				params[key] = value
			}
		}
	}
	return params
}

func (sender *Strings) HideCQEmoji(text string) map[string]interface{} {
	i := 0
	var ms = map[string]string{}
	text = regexp.MustCompile(`\[CQ:(\w+)(.*?)\]`).ReplaceAllStringFunc(text, func(s string) string {
		v := fmt.Sprintf("#%d#", i)
		i++
		ms[v] = s
		return v
	})
	text = emoji.ReplaceEmojisWithFunc(text, func(e emoji.Emoji) string {
		v := fmt.Sprintf("#%d#", i)
		i++
		ms[v] = e.Character
		return v
	})
	return map[string]interface{}{
		"text": text,
		"recover": func(text string) string {
			return regexp.MustCompile(`#\d{1,4}#`).ReplaceAllStringFunc(text, func(s string) string {
				return ms[s]
			})
		},
	}
}

// 将含有 CQ码 的文本解析成文本和 CQ 对象数组
func (sender *Strings) ParseCQText(text string) []interface{} {
	cqRegex := regexp.MustCompile(`\[CQ:(\w+)(.*?)\]`)
	cqMatches := cqRegex.FindAllStringSubmatch(text, -1)
	result := make([]interface{}, 0, len(cqMatches)*2+1)
	// 依次解析 CQ 码和文本
	lastIndex := 0
	for _, match := range cqMatches {
		// 添加 CQ 码前的文本
		if matchIndex := strings.Index(text[lastIndex:], match[0]); matchIndex > 0 {
			result = append(result, text[lastIndex:lastIndex+matchIndex])
			lastIndex += matchIndex
		}

		// 解析 CQ 码
		params := make(map[string]string)
		paramRegex := regexp.MustCompile(`(\w+)=([^,]+)`)
		paramMatches := paramRegex.FindAllStringSubmatch(match[2], -1)
		for _, paramMatch := range paramMatches {
			params[paramMatch[1]] = strings.TrimSpace(paramMatch[2])
		}
		result = append(result, CQ{
			Type:   match[1],
			Params: params,
		})

		lastIndex += len(match[0])
	}

	// 添加最后一个 CQ 码后的文本
	if lastIndex < len(text) {
		result = append(result, text[lastIndex:])
	}

	return result
}

// CQ 对象
type CQ struct {
	Type   string
	Params map[string]string
}

// 将 CQ 对象数组转换回原始文本
func (sender *Strings) StringifyCQText(cqList []interface{}) string {
	var sb strings.Builder
	for _, item := range cqList {
		switch item := item.(type) {
		case string:
			sb.WriteString(item)
		case CQ:
			sb.WriteString(fmt.Sprintf("[CQ:%s", item.Type))
			for k, v := range item.Params {
				sb.WriteString(fmt.Sprintf(",%s=%s", k, v))
			}
			sb.WriteString("]")
		case map[string]interface{}:
			cq := CQ{
				Type:   item["Type"].(string),
				Params: convertParams(item["Params"].(map[string]interface{})),
			}
			sb.WriteString(fmt.Sprintf("[CQ:%s", cq.Type))
			for k, v := range cq.Params {
				sb.WriteString(fmt.Sprintf(",%s=%s", k, v))
			}
			sb.WriteString("]")
		}
	}
	return sb.String()
}

// 将 map[string]interface{} 类型的 params 转换为 map[string]string 类型
func convertParams(params map[string]interface{}) map[string]string {
	result := make(map[string]string)
	for k, v := range params {
		if s, ok := v.(string); ok {
			result[k] = s
		}
	}
	return result
}

func (sender *Strings) ReplaceEmojis(str string, f func([]string) string) string {
	return emoji.ReplaceEmojisWithFunc(str, func(e emoji.Emoji) string {
		return f(e.CodePoint2)
	})
}

// `\[emoji=([0-9A-Z]{4})\]`
func (sender *Strings) ReplaceToEmojis(str string, pattern string) string {
	return emoji.ReplaceToEmojis(str, pattern)
}

func (sender *Strings) ExtractAddress(input string) string {
	return regexp.MustCompile(`http[s]?://[\w.]+:?\d*`).FindString(input)
}

func (sender *Strings) Unique(str ...interface{}) []string {
	return utils.Unique(str...)
}