conf.go

package main

import (
	"errors"
	"gopkg.in/yaml.v1"
	"io/ioutil"
	"github.com/martini-contrib/oauth2"
)

const (
	noAuthServiceName = "nothing" // for testing only (undocumented)
)

type Conf struct {
	Addr         string      `yaml:"address"`
	SSL          SSLConf     `yaml:"ssl"`
	Auth         AuthConf    `yaml:"auth"`
	Restrictions []string    `yaml:"restrictions"`
	Proxies      []ProxyConf `yaml:"proxy"`
	Paths        PathConf    `yaml:"paths"`
	Htdocs       string      `yaml:"htdocs"`
}

type SSLConf struct {
	Cert string `yaml:"cert"`
	Key  string `yaml:"key"`
}

type AuthConf struct {
	Session AuthSessionConf `yaml:"session"`
	Info    AuthInfoConf    `yaml:"info"`
}

type AuthSessionConf struct {
	Key          string `yaml:"key"`
	CookieDomain string `yaml:"cookie_domain"`
}

type AuthInfoConf struct {
	Service      string `yaml:"service"`
	ClientId     string `yaml:"client_id"`
	ClientSecret string `yaml:"client_secret"`
	RedirectURL  string `yaml:"redirect_url"`
	Endpoint     string `yaml:"endpoint"`
	ApiEndpoint  string `yaml:"api_endpoint"`
}

type ProxyConf struct {
	Path  string `yaml:"path"`
	Dest  string `yaml:"dest"`
	Strip bool   `yaml:"strip_path"`
	Host  string `yaml:"host"`
}

type PathConf struct {
	Login    string `yaml:"login"`
	Logout   string `yaml:"logout"`
	Callback string `yaml:"callback"`
	Error    string `yaml:"error"`
}

func ParseConf(path string) (*Conf, error) {
	data, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, err
	}

	c := &Conf{}
	if err := yaml.Unmarshal(data, c); err != nil {
		return nil, err
	}

	if c.Addr == "" {
		return nil, errors.New("address config is required")
	}

	if c.Auth.Session.Key == "" {
		return nil, errors.New("auth.session.key config is required")
	}
	if c.Auth.Info.Service == "" {
		return nil, errors.New("auth.info.service config is required")
	}
	if c.Auth.Info.ClientId == "" {
		return nil, errors.New("auth.info.client_id config is required")
	}
	if c.Auth.Info.ClientSecret == "" {
		return nil, errors.New("auth.info.client_secret config is required")
	}
	if c.Auth.Info.RedirectURL == "" {
		return nil, errors.New("auth.info.redirect_url config is required")
	}

	if c.Htdocs == "" {
		c.Htdocs = "."
	}

	if c.Auth.Info.Service == "github" && c.Auth.Info.Endpoint == "" {
		c.Auth.Info.Endpoint = "https://github.com"
	}
	if c.Auth.Info.Service == "github" && c.Auth.Info.ApiEndpoint == "" {
		c.Auth.Info.ApiEndpoint = "https://api.github.com"
	}

	return c, nil
}

func (c *Conf) SetOAuth2Paths() {
	if c.Paths.Login != "" {
		oauth2.PathLogin = c.Paths.Login
	}
	if c.Paths.Logout != "" {
		oauth2.PathLogout = c.Paths.Logout
	}
	if c.Paths.Callback != "" {
		oauth2.PathCallback = c.Paths.Callback
	}
	if c.Paths.Error != "" {
		oauth2.PathError = c.Paths.Error
	}
}