bb/lib/common.go
Ryan Cavicchioni 139b32094e
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Specify number of splits to make for command arguments
2022-09-06 10:43:30 -05:00

142 lines
2.1 KiB
Go

package lib
import (
"net/url"
"path"
"strconv"
"strings"
)
func Contains[T comparable](s []T, v T) bool {
for _, x := range s {
if x == v {
return true
}
}
return false
}
func JoinInt(a []int, sep string) string {
var b []string
b = make([]string, len(a))
for i, v := range a {
b[i] = strconv.Itoa(v)
}
return strings.Join(b, sep)
}
func SumInt(a []int) int {
var sum int
for _, v := range a {
sum += v
}
return sum
}
func Itob(v int) bool {
if v == 1 {
return true
}
return false
}
func BuildURI(rawuri, rawpath string) string {
u, _ := url.Parse(rawuri)
u.Path = path.Join(u.Path, rawpath)
return u.String()
}
func HasCommand(s, prefix string) bool {
s = strings.TrimSpace(s)
if len(s) == 0 || len(prefix) == 0 {
return false
}
if !strings.HasPrefix(s, prefix) {
return false
}
// remove the command prefix
s = s[len(prefix):]
// multiple assignment trick
cmd, _ := func() (string, string) {
x := strings.SplitN(s, " ", 2)
if len(x) > 1 {
return x[0], x[1]
}
return x[0], ""
}()
return len(cmd) > 0
}
func ContainsCommand(s, prefix, cmd string) bool {
s = strings.TrimSpace(s)
args := strings.Split(s, " ")
s = args[0]
// a command cannot be less than two characters e.g. !x
if len(s) < 2 {
return false
}
if string(s[0]) != prefix {
return false
}
if strings.HasPrefix(s[1:], cmd) {
return true
}
return false
}
func SplitCommandAndArg(s, prefix string) (cmd string, arg string) {
s = strings.TrimSpace(s)
if !strings.HasPrefix(s, prefix) {
return
}
// remove the command prefix
s = s[len(prefix):]
// multiple assignment trick
cmd, arg = func() (string, string) {
x := strings.SplitN(s, " ", 2)
if len(x) > 1 {
return x[0], x[1]
}
return x[0], ""
}()
return cmd, arg
}
func SplitCommandAndArgs(s, prefix string, n int) (cmd string, args []string) {
cmd, arg := SplitCommandAndArg(s, prefix)
if n == 0 {
return cmd, strings.Split(arg, " ")
}
return cmd, strings.SplitN(arg, " ", n)
}
func SplitArgs(s string, n int) (args []string) {
if n > 0 {
args = strings.SplitN(s, " ", n)
} else {
args = strings.Split(s, " ")
}
return
}