efb9da57c1
- Introduce ProcessPinyin function to convert Chinese characters to Pinyin with customizable options for tone and separator. - Register the new Pinyin process in the utils package for easy access. - Add comprehensive unit tests for various Pinyin conversion scenarios to ensure accuracy and reliability.
91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
package str
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/mozillazg/go-pinyin"
|
|
"github.com/yaoapp/gou/process"
|
|
"github.com/yaoapp/kun/maps"
|
|
)
|
|
|
|
// ProcessJoin utils.str.Join
|
|
func ProcessJoin(process *process.Process) interface{} {
|
|
process.ValidateArgNums(2)
|
|
args := process.ArgsArray(0)
|
|
sep := process.ArgsString(1)
|
|
strs := []string{}
|
|
for i := range args {
|
|
strs = append(strs, fmt.Sprintf("%v", args[i]))
|
|
}
|
|
return strings.Join(strs, sep)
|
|
}
|
|
|
|
// ProcessJoinPath utils.str.JoinPath
|
|
func ProcessJoinPath(process *process.Process) interface{} {
|
|
process.ValidateArgNums(2)
|
|
paths := []string{}
|
|
for _, arg := range process.Args {
|
|
paths = append(paths, fmt.Sprintf("%v", arg))
|
|
}
|
|
return filepath.Join(paths...)
|
|
}
|
|
|
|
// ProcessUUID utils.str.uuid
|
|
func ProcessUUID(process *process.Process) interface{} {
|
|
uuid := uuid.New()
|
|
return uuid.String()
|
|
}
|
|
|
|
// ProcessPinyin utils.str.Pinyin converts Chinese characters to Pinyin
|
|
// Args:
|
|
// - arg[0]: string, the Chinese characters to convert
|
|
// - arg[1]: map (optional) configuration options
|
|
// {
|
|
// "tone": bool, // whether to include tone marks, default: false
|
|
// "separator": string // separator between pinyin, default: " "
|
|
// }
|
|
func ProcessPinyin(process *process.Process) interface{} {
|
|
process.ValidateArgNums(1)
|
|
seed := process.ArgsString(0)
|
|
|
|
// Create converter with settings
|
|
a := pinyin.NewArgs()
|
|
a.Style = pinyin.Normal // default style
|
|
separator := " " // default separator
|
|
|
|
// Apply custom settings if provided
|
|
if process.NumOfArgs() > 1 {
|
|
confMap := process.ArgsMap(1, maps.MapStrAny{})
|
|
|
|
// Check if tone is enabled
|
|
toneVal, hasTone := confMap["tone"]
|
|
|
|
if hasTone {
|
|
if tone, ok := toneVal.(bool); ok && tone {
|
|
a.Style = pinyin.Tone
|
|
}
|
|
}
|
|
|
|
// Check custom separator
|
|
if sep, ok := confMap["separator"].(string); ok {
|
|
separator = sep
|
|
}
|
|
}
|
|
|
|
// Convert to Pinyin
|
|
result := pinyin.Pinyin(seed, a)
|
|
|
|
// Convert 2D array to 1D array
|
|
pinyinStr := make([]string, 0, len(result))
|
|
for _, py := range result {
|
|
if len(py) > 0 {
|
|
pinyinStr = append(pinyinStr, py[0])
|
|
}
|
|
}
|
|
|
|
final := strings.Join(pinyinStr, separator)
|
|
return final
|
|
}
|