diff --git a/client/extra.go b/client/extra.go index bccd2c8..3b6cd7e 100644 --- a/client/extra.go +++ b/client/extra.go @@ -3,6 +3,7 @@ package client import ( "fmt" "math/rand" + "strings" ) type ExtraGenerator func() string @@ -18,3 +19,54 @@ func UuidV4Generator() ExtraGenerator { return fmt.Sprintf("%08x-%04x-%04x-%04x-%012x", uuid[:4], uuid[4:6], uuid[6:8], uuid[8:10], uuid[10:]) } } + +func IsCommmand(text string) bool { + if text != "" { + if text[0] == '/' { + return true + } + } + return false +} + +func CheckCommand(text string, entities []*TextEntity) string { + if IsCommmand(text) { + // Check text entities and make bot happy! + if len(entities) >= 1 { + // Get first command + if entities[0].Type.TextEntityTypeType() == "textEntityTypeBotCommand" { + // e.g.: { "text": "/hello@world_bot", "textEntity": { offset: 0, length: 16 } } + // Result: "/hello" + if i := strings.Index(text[:entities[0].Length], "@"); i != -1 { + return text[:i] + } + return text[:entities[0].Length] + } + } else { + // Since userbot does not have bot command entities in Private Chat, so make userbot happy too! + // e.g.: ["/hello@world_bot", "/hello@", "/hello@123"] + // Result: "/hello" + if i := strings.Index(text, "@"); i != -1 { + return text[:i] + } + // e.g. ["/hello 123", "/hell o 123"] + // Result: "/hello", "/hell" + if i := strings.Index(text, " "); i != -1 { + return text[:i] + } + return text + } + } + return "" +} + +func CommandArgument(text string) string { + if IsCommmand(text) { + // e.g. ["/hello 123", "/hell o 123"] + // Result: "123", "o 123" + if i := strings.Index(text, " "); i != -1 { + return text[i+1:] + } + } + return "" +}