package util

import (
	"crypto/md5"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"math/rand"
	"time"

	"github.com/kr/pretty"

	"github.bus.zalan.do/acid/postgres-operator/pkg/spec"
	"k8s.io/client-go/pkg/api/v1"
	"k8s.io/client-go/pkg/types"
)

var passwordChars = []byte("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")

func init() {
	rand.Seed(int64(time.Now().Unix()))
}

func RandomPassword(n int) string {
	b := make([]byte, n)
	for i := range b {
		b[i] = passwordChars[rand.Intn(len(passwordChars))]
	}

	return string(b)
}

func NameFromMeta(meta v1.ObjectMeta) types.NamespacedName {
	obj := types.NamespacedName{
		Namespace: meta.Namespace,
		Name:      meta.Name,
	}

	return obj
}

func PodClusterName(pod *v1.Pod) spec.ClusterName {
	if name, ok := pod.Labels["spilo-cluster"]; ok {
		return spec.ClusterName{
			Namespace: pod.Namespace,
			Name:      name,
		}
	}

	return spec.ClusterName{}
}

func PodSpiloRole(pod *v1.Pod) string {
	return pod.Labels["spilo-role"]
}

func ClusterDNSName(clusterName, teamName, hostedZone string) string {
	return fmt.Sprintf("%s.%s.%s", clusterName, teamName, hostedZone)
}

func PGUserPassword(user spec.PgUser) string {
	s := md5.Sum([]byte(user.Password + user.Name))

	return "md5" + hex.EncodeToString(s[:])
}

func Pretty(x interface {}) (f fmt.Formatter) {
	return pretty.Formatter(x)
}

func PrettyDiff(a, b interface{}) (result string) {
	diff := pretty.Diff(a, b)
	json, err := json.MarshalIndent(diff, "", " ")
	if err != nil {
		result = fmt.Sprintf("%v", diff)
	} else {
		result = string(json)
	}
	return
}