strings.go 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228
  1. package sprig
  2. import (
  3. "encoding/base32"
  4. "encoding/base64"
  5. "fmt"
  6. "reflect"
  7. "strconv"
  8. "strings"
  9. util "github.com/Masterminds/goutils"
  10. )
  11. func base64encode(v string) string {
  12. return base64.StdEncoding.EncodeToString([]byte(v))
  13. }
  14. func base64decode(v string) string {
  15. data, err := base64.StdEncoding.DecodeString(v)
  16. if err != nil {
  17. return err.Error()
  18. }
  19. return string(data)
  20. }
  21. func base32encode(v string) string {
  22. return base32.StdEncoding.EncodeToString([]byte(v))
  23. }
  24. func base32decode(v string) string {
  25. data, err := base32.StdEncoding.DecodeString(v)
  26. if err != nil {
  27. return err.Error()
  28. }
  29. return string(data)
  30. }
  31. func abbrev(width int, s string) string {
  32. if width < 4 {
  33. return s
  34. }
  35. r, _ := util.Abbreviate(s, width)
  36. return r
  37. }
  38. func abbrevboth(left, right int, s string) string {
  39. if right < 4 || left > 0 && right < 7 {
  40. return s
  41. }
  42. r, _ := util.AbbreviateFull(s, left, right)
  43. return r
  44. }
  45. func initials(s string) string {
  46. return util.Initials(s)
  47. }
  48. func randAlphaNumeric(count int) string {
  49. r, _ := util.CryptoRandomAlphaNumeric(count)
  50. return r
  51. }
  52. func randAlpha(count int) string {
  53. r, _ := util.CryptoRandomAlphabetic(count)
  54. return r
  55. }
  56. func randAscii(count int) string {
  57. r, _ := util.CryptoRandomAscii(count)
  58. return r
  59. }
  60. func randNumeric(count int) string {
  61. r, _ := util.CryptoRandomNumeric(count)
  62. return r
  63. }
  64. func untitle(str string) string {
  65. return util.Uncapitalize(str)
  66. }
  67. func quote(str ...interface{}) string {
  68. out := make([]string, 0, len(str))
  69. for _, s := range str {
  70. if s != nil {
  71. out = append(out, fmt.Sprintf("%q", strval(s)))
  72. }
  73. }
  74. return strings.Join(out, " ")
  75. }
  76. func squote(str ...interface{}) string {
  77. out := make([]string, 0, len(str))
  78. for _, s := range str {
  79. if s != nil {
  80. out = append(out, fmt.Sprintf("'%v'", s))
  81. }
  82. }
  83. return strings.Join(out, " ")
  84. }
  85. func cat(v ...interface{}) string {
  86. v = removeNilElements(v)
  87. r := strings.TrimSpace(strings.Repeat("%v ", len(v)))
  88. return fmt.Sprintf(r, v...)
  89. }
  90. func indent(spaces int, v string) string {
  91. pad := strings.Repeat(" ", spaces)
  92. return pad + strings.Replace(v, "\n", "\n"+pad, -1)
  93. }
  94. func nindent(spaces int, v string) string {
  95. return "\n" + indent(spaces, v)
  96. }
  97. func replace(old, new, src string) string {
  98. return strings.Replace(src, old, new, -1)
  99. }
  100. func plural(one, many string, count int) string {
  101. if count == 1 {
  102. return one
  103. }
  104. return many
  105. }
  106. func strslice(v interface{}) []string {
  107. switch v := v.(type) {
  108. case []string:
  109. return v
  110. case []interface{}:
  111. b := make([]string, 0, len(v))
  112. for _, s := range v {
  113. if s != nil {
  114. b = append(b, strval(s))
  115. }
  116. }
  117. return b
  118. default:
  119. val := reflect.ValueOf(v)
  120. switch val.Kind() {
  121. case reflect.Array, reflect.Slice:
  122. l := val.Len()
  123. b := make([]string, 0, l)
  124. for i := 0; i < l; i++ {
  125. value := val.Index(i).Interface()
  126. if value != nil {
  127. b = append(b, strval(value))
  128. }
  129. }
  130. return b
  131. default:
  132. if v == nil {
  133. return []string{}
  134. }
  135. return []string{strval(v)}
  136. }
  137. }
  138. }
  139. func removeNilElements(v []interface{}) []interface{} {
  140. newSlice := make([]interface{}, 0, len(v))
  141. for _, i := range v {
  142. if i != nil {
  143. newSlice = append(newSlice, i)
  144. }
  145. }
  146. return newSlice
  147. }
  148. func strval(v interface{}) string {
  149. switch v := v.(type) {
  150. case string:
  151. return v
  152. case []byte:
  153. return string(v)
  154. case error:
  155. return v.Error()
  156. case fmt.Stringer:
  157. return v.String()
  158. default:
  159. return fmt.Sprintf("%v", v)
  160. }
  161. }
  162. func trunc(c int, s string) string {
  163. if c < 0 && len(s)+c > 0 {
  164. return s[len(s)+c:]
  165. }
  166. if c >= 0 && len(s) > c {
  167. return s[:c]
  168. }
  169. return s
  170. }
  171. func join(sep string, v interface{}) string {
  172. return strings.Join(strslice(v), sep)
  173. }
  174. func split(sep, orig string) map[string]string {
  175. parts := strings.Split(orig, sep)
  176. res := make(map[string]string, len(parts))
  177. for i, v := range parts {
  178. res["_"+strconv.Itoa(i)] = v
  179. }
  180. return res
  181. }
  182. func splitn(sep string, n int, orig string) map[string]string {
  183. parts := strings.SplitN(orig, sep, n)
  184. res := make(map[string]string, len(parts))
  185. for i, v := range parts {
  186. res["_"+strconv.Itoa(i)] = v
  187. }
  188. return res
  189. }
  190. func substring(start, end int, s string) string {
  191. if start < 0 {
  192. return s[:end]
  193. }
  194. if end < 0 || end > len(s) {
  195. return s[start:]
  196. }
  197. return s[start:end]
  198. }