functions.go

package tools

import (
	"fmt"
	"go/format"
	"log"
	"strings"

	"github.com/huandu/xstrings"
)

func Sformat(buf []byte) []byte {
	src, err := format.Source(buf)
	if err != nil {
		// Should never happen, but can arise when developing this code.
		// The user can compile the output to see the error.
		log.Printf("warning: internal error: invalid Go generated: %s", err)
		log.Printf("warning: compile the package to analyze the error")
		log.Printf("generated: \n%s", string(buf))
		return buf
	}
	return src
}

var normalizationWords = map[string]string{
	"IP": "ip",
}

func NormalizeResourceName(name string) string {
	n := name
	for k, v := range normalizationWords {
		if strings.HasPrefix(name, k) {
			n = strings.Replace(name, k, v, -1)
			break
		}
	}
	return n
}

func ToSnakeCaseName(name string) string {
	return strings.Replace(xstrings.ToSnakeCase(NormalizeResourceName(name)), "-", "_", -1)
}

func ToDashedName(name string) string {
	// From "CamelCase" to "dash-case"
	return strings.Replace(xstrings.ToSnakeCase(NormalizeResourceName(name)), "_", "-", -1)
}

func ToCamelCaseName(name string) string {
	return xstrings.ToCamelCase(strings.Replace(NormalizeResourceName(name), "-", "_", -1))
}

func ToCamelWithFirstLower(name string) string {
	return xstrings.FirstRuneToLower(xstrings.ToCamelCase(strings.Replace(NormalizeResourceName(name), "-", "_", -1)))
}

func ToCLIFlagName(name string) string {
	format := "--%s"
	if len(name) == 1 {
		format = "-%s"
	}
	return fmt.Sprintf(format, ToDashedName(name))
}

func FlattenStringList(list []string) string {
	if len(list) > 0 {
		return fmt.Sprintf("\"%s\"", strings.Join(list, "\",\""))
	}
	return ""
}

func FlattenIntList(list []int) string {
	if len(list) > 0 {
		tmp := []string{}
		for _, s := range list {
			tmp = append(tmp, fmt.Sprintf("%d", s))
		}
		return strings.Join(tmp, ",")
	}
	return ""
}

func FlattenUintList(list []uint) string {
	if len(list) > 0 {
		tmp := []string{}
		for _, s := range list {
			tmp = append(tmp, fmt.Sprintf("%d", s))
		}
		return strings.Join(tmp, ",")
	}
	return ""
}
func FlattenInt64List(list []int64) string {
	if len(list) > 0 {
		tmp := []string{}
		for _, s := range list {
			tmp = append(tmp, fmt.Sprintf("%d", s))
		}
		return strings.Join(tmp, ",")
	}
	return ""
}

func FlattenUint64List(list []uint64) string {
	if len(list) > 0 {
		tmp := []string{}
		for _, s := range list {
			tmp = append(tmp, fmt.Sprintf("%d", s))
		}
		return strings.Join(tmp, ",")
	}
	return ""
}

func FlattenFloatList(list []float32) string {
	if len(list) > 0 {
		tmp := []string{}
		for _, s := range list {
			tmp = append(tmp, fmt.Sprintf("%f", s))
		}
		return strings.Join(tmp, ",")
	}
	return ""
}

func FlattenFloat64List(list []float64) string {
	if len(list) > 0 {
		tmp := []string{}
		for _, s := range list {
			tmp = append(tmp, fmt.Sprintf("%f", s))
		}
		return strings.Join(tmp, ",")
	}
	return ""
}