utils

var DelConsulKv = func(client *consul.Client, key string) error {
	log.Printf("consul delete `%s`", key)
	_, err := client.KV().Delete(strings.TrimPrefix(key, "/"), nil)
	return err
}

var DeletePluginData = func(plugin string, packageName string, consulAddress string) error {
	log.Println(color.YellowString("Delete %s for %s package in consul", plugin, packageName))
	consulApi, err := ConsulClient(consulAddress)
	if err != nil {
		return err
	}

	return DelConsulKv(consulApi, "services/data/"+packageName+"/"+plugin)
}

var PutConsulKv = func(client *consul.Client, key string, value string) error {
	log.Printf("consul put `%s`: %s", key, value)
	_, err := client.KV().Put(&consul.KVPair{Key: strings.TrimPrefix(key, "/"), Value: []byte(value)}, nil)
	return err
}

var RandomString = func(length uint) string {
	b := make([]rune, length)
	for i := range b {
		b[i] = allLetters[rand.Intn(len(allLetters))]
	}
	return string(b)
}

var RegisterPluginData = func(plugin string, packageName string, data string, consulAddress string) error {
	consulApi, err := ConsulClient(consulAddress)
	if err != nil {
		return err
	}

	return PutConsulKv(consulApi, "services/data/"+packageName+"/"+plugin, data)
}

var RunCmdWithEnv = func(cmdline string, env map[string]string) error {
	log.Println(color.YellowString("> %s", cmdline))

	cmd := exec.Command("/bin/bash", "-c", cmdline)

	cmd.Env = os.Environ()

	for k, v := range env {
		cmd.Env = append(cmd.Env, fmt.Sprintf("%s=%v", k, v))
	}

	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	return cmd.Run()
}