123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109 |
- package util
- import (
- "bytes"
- "encoding/xml"
- "fmt"
- "go/format"
- "io"
- "reflect"
- "regexp"
- "strings"
- "github.com/aws/aws-sdk-go/internal/protocol/xml/xmlutil"
- )
- // GoFmt returns the Go formated string of the input.
- //
- // Panics if the format fails.
- func GoFmt(buf string) string {
- formatted, err := format.Source([]byte(buf))
- if err != nil {
- panic(fmt.Errorf("%s\nOriginal code:\n%s", err.Error(), buf))
- }
- return string(formatted)
- }
- var reTrim = regexp.MustCompile(`\s{2,}`)
- // Trim removes all leading and trailing white space.
- //
- // All consecutive spaces will be reduced to a single space.
- func Trim(s string) string {
- return strings.TrimSpace(reTrim.ReplaceAllString(s, " "))
- }
- // Capitalize capitalizes the first character of the string.
- func Capitalize(s string) string {
- if len(s) == 1 {
- return strings.ToUpper(s)
- }
- return strings.ToUpper(s[0:1]) + s[1:]
- }
- // SortXML sorts the reader's XML elements
- func SortXML(r io.Reader) string {
- var buf bytes.Buffer
- d := xml.NewDecoder(r)
- root, _ := xmlutil.XMLToStruct(d, nil)
- e := xml.NewEncoder(&buf)
- xmlutil.StructToXML(e, root, true)
- return buf.String()
- }
- // PrettyPrint generates a human readable representation of the value v.
- // All values of v are recursively found and pretty printed also.
- func PrettyPrint(v interface{}) string {
- value := reflect.ValueOf(v)
- switch value.Kind() {
- case reflect.Struct:
- str := fullName(value.Type()) + "{\n"
- for i := 0; i < value.NumField(); i++ {
- l := string(value.Type().Field(i).Name[0])
- if strings.ToUpper(l) == l {
- str += value.Type().Field(i).Name + ": "
- str += PrettyPrint(value.Field(i).Interface())
- str += ",\n"
- }
- }
- str += "}"
- return str
- case reflect.Map:
- str := "map[" + fullName(value.Type().Key()) + "]" + fullName(value.Type().Elem()) + "{\n"
- for _, k := range value.MapKeys() {
- str += "\"" + k.String() + "\": "
- str += PrettyPrint(value.MapIndex(k).Interface())
- str += ",\n"
- }
- str += "}"
- return str
- case reflect.Ptr:
- if e := value.Elem(); e.IsValid() {
- return "&" + PrettyPrint(e.Interface())
- }
- return "nil"
- case reflect.Slice:
- str := "[]" + fullName(value.Type().Elem()) + "{\n"
- for i := 0; i < value.Len(); i++ {
- str += PrettyPrint(value.Index(i).Interface())
- str += ",\n"
- }
- str += "}"
- return str
- default:
- return fmt.Sprintf("%#v", v)
- }
- }
- func pkgName(t reflect.Type) string {
- pkg := t.PkgPath()
- c := strings.Split(pkg, "/")
- return c[len(c)-1]
- }
- func fullName(t reflect.Type) string {
- if pkg := pkgName(t); pkg != "" {
- return pkg + "." + t.Name()
- }
- return t.Name()
- }
|