loader.go

package samb

import (
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"strings"

	"github.com/recoye/config"
	"gopkg.in/yaml.v2"
)

// Load opens a project based on the filesystem
// path supplied.
func Load(path string) (*Project, error) {

	// err will be used to check
	// for errors on file load.
	var err error
	var env = &Project{}

	var ext = filepath.Ext(path)

	switch ext {
		case ".yml":
			err = LoadYAML(path, env)
		default:
			err = LoadSE(path, env)
	}

	if err != nil {
		return nil, err
	}

	chDirRootOf(path)

	env.ProcessImports()

	return env, nil
}

// LoadSE loads .se file with SAMB directives.
func LoadSE(path string, p *Project) error {
	var conf = config.New(path)

	return conf.Unmarshal(p)
}

// LoadYAML loads a YAML file with SAMB
// directives.
func LoadYAML(path string, p *Project) error {

	data, err := ioutil.ReadFile(path)

	if err != nil {
		return err
	}

	return yaml.Unmarshal(data, p)
}

// chDirRootOf will change the working directory to the directory
// of the specified SAMB
// source file path.
func chDirRootOf(file string) {

	parts := strings.Split(file, "/")
	file = strings.Replace(file, parts[len(parts)-1], "", -1)

	if file == "" {
		file = "./"
	}

	log.Println("Changing Directory to ", file)

	err := os.Chdir(file)

	if err != nil {
		panic(err)
	}
}