astconf

package main

import (
	"bytes"
	"fmt"
	"strings"

	"github.com/scjalliance/astconf"
)

type Name string

func (n Name) SectionName() string {
	return strings.ToLower(strings.Replace(string(n), " ", "_", -1))
}

type Elephant struct {
	Name Name `astconf:"elephant_name"`
	Age  int  `astconf:"age"`
}

func (elephant *Elephant) SectionName() string {
	return "elephant." + elephant.Name.SectionName()
}

func (elephant *Elephant) MarshalAsteriskPreamble(e *astconf.Encoder) error {
	var kind string
	switch {
	case elephant.Age > 50:
		kind = "old_elephant"
	case elephant.Age < 10:
		kind = "young_elephant"
	default:
		kind = "elephant"
	}
	return e.Printer().Setting("type", kind)
}

type Zookeeper struct {
	Name           Name     `astconf:"zookeeper_name"`
	Experience     int      `astconf:"experience_level"`
	FavoriteColors []string `astconf:"favorite_colors,commaseparated,omitempty"`
}

func (zk Zookeeper) SectionName() string {
	return "zookeeper." + zk.Name.SectionName()
}

func (zk Zookeeper) SectionTemplates() []string {
	return []string{"senior-management"}
}

func (*Zookeeper) MarshalAsteriskPreamble(e *astconf.Encoder) error {
	return e.Printer().Setting("type", "zookeeper")
}

type Zoo struct {
	Name        Name `astconf:"zoo_name"`
	Maintainers []Zookeeper
	//Maintainers []*Zookeeper // FIXME: This use of a slice of references breaks the marshaler and causes the test to fail
	Elephants []Elephant
}

func (zoo *Zoo) SectionName() string {
	return "zoo"
}

func (*Zoo) MarshalAsteriskPreamble(e *astconf.Encoder) error {
	return e.Printer().Setting("type", "zoo")
}

func main() {
	var buf bytes.Buffer
	e := astconf.NewEncoder(&buf, astconf.AlignRight)
	err := e.Encode(&Zoo{
		Name: "Malarky McFee's Mighty Jungle",
		Elephants: []Elephant{
			{Name: "Matilda", Age: 47},
			{Name: "Franklin", Age: 52},
			{Name: "Georgey the Kid", Age: 5},
		},
		Maintainers: []Zookeeper{
			{
				Name:           "Gershwin McFee",
				Experience:     8000,
				FavoriteColors: []string{"blue", "yellow"},
			},
		},
	})
	if err != nil {
		fmt.Println(err)
	} else {
		fmt.Print(buf.String())
	}
}

Output:

[zoo]
    type = zoo
zoo_name = Malarky McFee's Mighty Jungle

[zookeeper.gershwin_mcfee](senior-management)
            type = zookeeper
  zookeeper_name = Gershwin McFee
experience_level = 8000
 favorite_colors = blue,yellow

[elephant.matilda]
         type = elephant
elephant_name = Matilda
          age = 47

[elephant.franklin]
         type = old_elephant
elephant_name = Franklin
          age = 52

[elephant.georgey_the_kid]
         type = young_elephant
elephant_name = Georgey the Kid
          age = 5