130 lines
2.5 KiB
Go
130 lines
2.5 KiB
Go
package laniakea
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// MapToStruct unsafe function
|
|
func MapToStruct(m map[string]any, s any) error {
|
|
data, err := json.Marshal(m)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = json.Unmarshal(data, s)
|
|
return err
|
|
}
|
|
|
|
// SliceToStruct unsafe function
|
|
func SliceToStruct(sl []any, s any) error {
|
|
data, err := json.Marshal(sl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = json.Unmarshal(data, s)
|
|
return err
|
|
}
|
|
|
|
// AnyToStruct unsafe function
|
|
func AnyToStruct(src, dest any) error {
|
|
data, err := json.Marshal(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = json.Unmarshal(data, dest)
|
|
return err
|
|
}
|
|
|
|
func MapToJson(m map[string]any) (string, error) {
|
|
data, err := json.Marshal(m)
|
|
return string(data), err
|
|
}
|
|
|
|
func StructToMap(s interface{}) (map[string]interface{}, error) {
|
|
data, err := json.Marshal(s)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
m := make(map[string]interface{})
|
|
err = json.Unmarshal(data, &m)
|
|
return m, err
|
|
}
|
|
|
|
func Map[T, V any](ts []T, fn func(T) V) []V {
|
|
result := make([]V, len(ts))
|
|
for i, t := range ts {
|
|
result[i] = fn(t)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func EscapeMarkdown(s string) string {
|
|
s = strings.ReplaceAll(s, "_", `\_`)
|
|
s = strings.ReplaceAll(s, "*", `\*`)
|
|
s = strings.ReplaceAll(s, "[", `\[`)
|
|
return strings.ReplaceAll(s, "`", "\\`")
|
|
}
|
|
|
|
func EscapeMarkdownV2(s string) string {
|
|
symbols := []string{"_", "*", "[", "]", "(", ")", "~", "`", ">", "#", "+", "-", "=", "|", "{", "}", ".", "!"}
|
|
for _, symbol := range symbols {
|
|
s = strings.ReplaceAll(s, symbol, fmt.Sprintf("\\%s", symbol))
|
|
}
|
|
return s
|
|
}
|
|
|
|
func GetUnclosedTag(markdown string) string {
|
|
// order is important!
|
|
var tags = []string{
|
|
"```",
|
|
"`",
|
|
"*",
|
|
"_",
|
|
}
|
|
var currentTag = ""
|
|
|
|
markdownRunes := []rune(markdown)
|
|
|
|
var i = 0
|
|
outer:
|
|
for i < len(markdownRunes) {
|
|
// skip escaped characters (only outside tags)
|
|
if markdownRunes[i] == '\\' && currentTag == "" {
|
|
i += 2
|
|
continue
|
|
}
|
|
if currentTag != "" {
|
|
if strings.HasPrefix(string(markdownRunes[i:]), currentTag) {
|
|
// turn a tag off
|
|
i += len(currentTag)
|
|
currentTag = ""
|
|
continue
|
|
}
|
|
} else {
|
|
for _, tag := range tags {
|
|
if strings.HasPrefix(string(markdownRunes[i:]), tag) {
|
|
// turn a tag on
|
|
currentTag = tag
|
|
i += len(currentTag)
|
|
continue outer
|
|
}
|
|
}
|
|
}
|
|
i++
|
|
}
|
|
|
|
return currentTag
|
|
}
|
|
func IsValid(markdown string) bool {
|
|
return GetUnclosedTag(markdown) == ""
|
|
}
|
|
|
|
func FixMarkdown(markdown string) string {
|
|
tag := GetUnclosedTag(markdown)
|
|
if tag == "" {
|
|
return markdown
|
|
}
|
|
return markdown + tag
|
|
}
|