formatter.go

package bindingrule

import (
	"bytes"
	"encoding/json"
	"fmt"

	"github.com/hashicorp/consul/api"
)

const (
	PrettyFormat string = "pretty"
	JSONFormat   string = "json"
)

// Formatter defines methods provided by bindingrule command output formatter
type Formatter interface {
	FormatBindingRule(rule *api.ACLBindingRule) (string, error)
	FormatBindingRuleList(rules []*api.ACLBindingRule) (string, error)
}

// GetSupportedFormats returns supported formats
func GetSupportedFormats() []string {
	return []string{PrettyFormat, JSONFormat}
}

// NewFormatter returns Formatter implementation
func NewFormatter(format string, showMeta bool) (formatter Formatter, err error) {
	switch format {
	case PrettyFormat:
		formatter = newPrettyFormatter(showMeta)
	case JSONFormat:
		formatter = newJSONFormatter(showMeta)
	default:
		err = fmt.Errorf("Unknown format: %s", format)
	}

	return formatter, err
}

func newPrettyFormatter(showMeta bool) Formatter {
	return &prettyFormatter{showMeta}
}

type prettyFormatter struct {
	showMeta bool
}

func (f *prettyFormatter) FormatBindingRule(rule *api.ACLBindingRule) (string, error) {
	var buffer bytes.Buffer

	buffer.WriteString(fmt.Sprintf("ID:           %s\n", rule.ID))
	if rule.Partition != "" {
		buffer.WriteString(fmt.Sprintf("Partition:    %s\n", rule.Partition))
	}
	if rule.Namespace != "" {
		buffer.WriteString(fmt.Sprintf("Namespace:    %s\n", rule.Namespace))
	}
	buffer.WriteString(fmt.Sprintf("AuthMethod:   %s\n", rule.AuthMethod))
	buffer.WriteString(fmt.Sprintf("Description:  %s\n", rule.Description))
	buffer.WriteString(fmt.Sprintf("BindType:     %s\n", rule.BindType))
	buffer.WriteString(fmt.Sprintf("BindName:     %s\n", rule.BindName))
	buffer.WriteString(fmt.Sprintf("Selector:     %s\n", rule.Selector))
	if f.showMeta {
		buffer.WriteString(fmt.Sprintf("Create Index: %d\n", rule.CreateIndex))
		buffer.WriteString(fmt.Sprintf("Modify Index: %d\n", rule.ModifyIndex))
	}

	return buffer.String(), nil
}

func (f *prettyFormatter) FormatBindingRuleList(rules []*api.ACLBindingRule) (string, error) {
	var buffer bytes.Buffer

	for _, rule := range rules {
		buffer.WriteString(f.formatBindingRuleListEntry(rule))
	}

	return buffer.String(), nil
}

func (f *prettyFormatter) formatBindingRuleListEntry(rule *api.ACLBindingRule) string {
	var buffer bytes.Buffer

	buffer.WriteString(fmt.Sprintf("%s:\n", rule.ID))
	if rule.Partition != "" {
		buffer.WriteString(fmt.Sprintf("   Partition:    %s\n", rule.Partition))
	}
	if rule.Namespace != "" {
		buffer.WriteString(fmt.Sprintf("   Namespace:    %s\n", rule.Namespace))
	}
	buffer.WriteString(fmt.Sprintf("   AuthMethod:   %s\n", rule.AuthMethod))
	buffer.WriteString(fmt.Sprintf("   Description:  %s\n", rule.Description))
	buffer.WriteString(fmt.Sprintf("   BindType:     %s\n", rule.BindType))
	buffer.WriteString(fmt.Sprintf("   BindName:     %s\n", rule.BindName))
	buffer.WriteString(fmt.Sprintf("   Selector:     %s\n", rule.Selector))
	if f.showMeta {
		buffer.WriteString(fmt.Sprintf("   Create Index: %d\n", rule.CreateIndex))
		buffer.WriteString(fmt.Sprintf("   Modify Index: %d\n", rule.ModifyIndex))
	}

	return buffer.String()
}

func newJSONFormatter(showMeta bool) Formatter {
	return &jsonFormatter{showMeta}
}

type jsonFormatter struct {
	showMeta bool
}

func (f *jsonFormatter) FormatBindingRule(rule *api.ACLBindingRule) (string, error) {
	b, err := json.MarshalIndent(rule, "", "    ")
	if err != nil {
		return "", fmt.Errorf("Failed to marshal binding rule:, %v", err)
	}
	return string(b), nil
}

func (f *jsonFormatter) FormatBindingRuleList(rules []*api.ACLBindingRule) (string, error) {
	b, err := json.MarshalIndent(rules, "", "    ")
	if err != nil {
		return "", fmt.Errorf("Failed to marshal binding rules:, %v", err)

	}
	return string(b), nil
}