strutil.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. // Copyright 2020 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package strutil
  5. import (
  6. "crypto/rand"
  7. "math/big"
  8. "unicode"
  9. )
  10. // ToUpperFirst returns s with only the first Unicode letter mapped to its upper case.
  11. func ToUpperFirst(s string) string {
  12. for i, v := range s {
  13. return string(unicode.ToUpper(v)) + s[i+1:]
  14. }
  15. return ""
  16. }
  17. // RandomChars returns a generated string in given number of random characters.
  18. func RandomChars(n int) (string, error) {
  19. const alphanum = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
  20. randomInt := func(max *big.Int) (int, error) {
  21. r, err := rand.Int(rand.Reader, max)
  22. if err != nil {
  23. return 0, err
  24. }
  25. return int(r.Int64()), nil
  26. }
  27. buffer := make([]byte, n)
  28. max := big.NewInt(int64(len(alphanum)))
  29. for i := 0; i < n; i++ {
  30. index, err := randomInt(max)
  31. if err != nil {
  32. return "", err
  33. }
  34. buffer[i] = alphanum[index]
  35. }
  36. return string(buffer), nil
  37. }