asciinema/util/cfg.go

package util

import (
	"errors"
	"fmt"
	"io/ioutil"
	"os"
	"path"
	"path/filepath"

	"github.com/asciinema/asciinema/Godeps/_workspace/src/code.google.com/p/gcfg"
)

const (
	DefaultAPIURL  = "https://asciinema.org"
	DefaultCommand = "/bin/sh"
)

type ConfigAPI struct {
	Token string
	URL   string
}

type ConfigRecord struct {
	Command string
	MaxWait float64
	Yes     bool
}

type ConfigPlay struct {
	MaxWait float64
}

type ConfigUser struct {
	Token string
}

type ConfigFile struct {
	API    ConfigAPI
	Record ConfigRecord
	Play   ConfigPlay
	User   ConfigUser // old location of token
}

type Config struct {
	File *ConfigFile
	Env  map[string]string
}

func (c *Config) ApiUrl() string {
	return FirstNonBlank(c.Env["ASCIINEMA_API_URL"], c.File.API.URL, DefaultAPIURL)
}

func (c *Config) ApiToken() string {
	return FirstNonBlank(c.File.API.Token, c.File.User.Token)
}

func (c *Config) RecordCommand() string {
	return FirstNonBlank(c.File.Record.Command, c.Env["SHELL"], DefaultCommand)
}

func (c *Config) RecordMaxWait() float64 {
	return c.File.Record.MaxWait
}

func (c *Config) RecordYes() bool {
	return c.File.Record.Yes
}

func (c *Config) PlayMaxWait() float64 {
	return c.File.Play.MaxWait
}

func GetConfig(env map[string]string) (*Config, error) {
	cfg, err := loadConfigFile(env)
	if err != nil {
		return nil, err
	}

	return &Config{cfg, env}, nil
}

func loadConfigFile(env map[string]string) (*ConfigFile, error) {
	pathsToCheck := make([]string, 0, 4)
	if env["ASCIINEMA_CONFIG_HOME"] != "" {
		pathsToCheck = append(pathsToCheck,
			filepath.Join(env["ASCIINEMA_CONFIG_HOME"], "config"))
	}
	if env["XDG_CONFIG_HOME"] != "" {
		pathsToCheck = append(pathsToCheck,
			filepath.Join(env["XDG_CONFIG_HOME"], "asciinema", "config"))
	}
	if env["HOME"] != "" {
		pathsToCheck = append(pathsToCheck,
			filepath.Join(env["HOME"], ".config", "asciinema", "config"))
		pathsToCheck = append(pathsToCheck,
			filepath.Join(env["HOME"], ".asciinema", "config"))
	}

	cfgPath := ""
	for _, pathToCheck := range pathsToCheck {
		if _, err := os.Stat(pathToCheck); err == nil {
			cfgPath = pathToCheck
			break
		}
	}

	if cfgPath == "" {
		if len(pathsToCheck) == 0 {
			return nil, errors.New("Need $HOME")
		}
		cfgPath = pathsToCheck[0]
		if err := createConfigFile(cfgPath); err != nil {
			return nil, err
		}
	}

	return readConfigFile(cfgPath)
}

func readConfigFile(cfgPath string) (*ConfigFile, error) {
	var cfg ConfigFile
	if err := gcfg.ReadFileInto(&cfg, cfgPath); err != nil {
		return nil, err
	}

	return &cfg, nil
}

func createConfigFile(cfgPath string) error {
	apiToken := NewUUID().String()
	contents := fmt.Sprintf("[api]\ntoken = %v\n", apiToken)
	os.MkdirAll(path.Dir(cfgPath), 0755)
	return ioutil.WriteFile(cfgPath, []byte(contents), 0644)
}