strings.go 4.3 KB

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