mirror of
https://github.com/asciinema/asciinema.git
synced 2025-12-16 11:48:13 +01:00
136 lines
2.7 KiB
Go
136 lines
2.7 KiB
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)
|
|
}
|