config.go

package main

import (
	"fmt"
	"io/ioutil"
	"log"

	"github.com/BurntSushi/toml"
)

const configVersion = "unstable"

type configV2 struct {
	Version   string
	Generator []Generator
	Includes  struct {
		Before   []string
		Vendored []string
		Packages []string
		After    []string
	}

	Packages map[string]string

	Overrides []struct {
		Prefixes  []string
		Generator string
		Plugins   *[]string

		// TODO(stevvooe): We could probably support overriding of includes and
		// package maps, but they don't seem to be as useful. Likely,
		// overriding the package map is more useful but includes happen
		// project-wide.
	}

	Descriptors []struct {
		Prefix      string
		Target      string
		IgnoreFiles []string `toml:"ignore_files"`
	}
}

type Plugin struct {
	Name    string
	Options map[string]interface{}
}

type Generator struct {
	Name    string
	Options *map[string]interface{}
	Plugin  *[]Plugin
}

type config struct {
	Version   string
	Generator string
	Plugins   []string
	Includes  struct {
		Before   []string
		Vendored []string
		Packages []string
		After    []string
	}

	Packages map[string]string

	Overrides []struct {
		Prefixes  []string
		Generator string
		Plugins   *[]string

		// TODO(stevvooe): We could probably support overriding of includes and
		// package maps, but they don't seem to be as useful. Likely,
		// overriding the package map is more useful but includes happen
		// project-wide.
	}

	Descriptors []struct {
		Prefix      string
		Target      string
		IgnoreFiles []string `toml:"ignore_files"`
	}
}

// NewDefaultConfig V2
func newDefaultConfigV2() configV2 {
	return configV2{
		Generator: []Generator{
			Generator{
				Name:    "go",
				Options: &map[string]interface{}{},
				Plugin:  &[]Plugin{},
			},
		},
		Includes: struct {
			Before   []string
			Vendored []string
			Packages []string
			After    []string
		}{
			Before: []string{"."},
			After:  []string{"/usr/local/include"},
		},
	}
}

// NewDefaultConfig
func newDefaultConfig() config {
	return config{
		Generator: "go",
		Includes: struct {
			Before   []string
			Vendored []string
			Packages []string
			After    []string
		}{
			Before: []string{"."},
			After:  []string{"/usr/local/include"},
		},
	}
}

func readConfigV2(path string) (configV2, error) {
	p, err := ioutil.ReadFile(path)
	if err != nil {
		log.Fatalln(err)
	}
	c := newDefaultConfigV2()
	if err := toml.Unmarshal(p, &c); err != nil {
		log.Fatalln(err)
	}

	if c.Version != configVersion {
		return configV2{}, fmt.Errorf("unknown file version %v; please upgrade to %v", c.Version, configVersion)
	}

	return c, nil
}

func readConfig(path string) (config, error) {
	p, err := ioutil.ReadFile(path)
	if err != nil {
		log.Fatalln(err)
	}
	c := newDefaultConfig()
	if err := toml.Unmarshal(p, &c); err != nil {
		log.Fatalln(err)
	}

	if c.Version != configVersion {
		return config{}, fmt.Errorf("unknown file version %v; please upgrade to %v", c.Version, configVersion)
	}

	return c, nil
}