132 lines
3.3 KiB
Go
132 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/coreos/poke/connector"
|
|
"github.com/coreos/poke/connector/github"
|
|
"github.com/coreos/poke/connector/ldap"
|
|
"github.com/coreos/poke/connector/mock"
|
|
"github.com/coreos/poke/storage"
|
|
"github.com/coreos/poke/storage/kubernetes"
|
|
"github.com/coreos/poke/storage/memory"
|
|
)
|
|
|
|
// Config is the config format for the main application.
|
|
type Config struct {
|
|
Issuer string `yaml:"issuer"`
|
|
Storage Storage `yaml:"storage"`
|
|
Connectors []Connector `yaml:"connectors"`
|
|
Web Web `yaml:"web"`
|
|
|
|
StaticClients []storage.Client `yaml:"staticClients"`
|
|
}
|
|
|
|
// Web is the config format for the HTTP server.
|
|
type Web struct {
|
|
HTTP string `yaml:"http"`
|
|
HTTPS string `yaml:"https"`
|
|
TLSCert string `yaml:"tlsCert"`
|
|
TLSKey string `yaml:"tlsKey"`
|
|
}
|
|
|
|
// Storage holds app's storage configuration.
|
|
type Storage struct {
|
|
Type string `yaml:"type"`
|
|
Config StorageConfig `yaml:"config"`
|
|
}
|
|
|
|
// UnmarshalYAML allows Storage to unmarshal its config field dynamically
|
|
// depending on the type of storage.
|
|
func (s *Storage) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
var storageMeta struct {
|
|
Type string `yaml:"type"`
|
|
}
|
|
if err := unmarshal(&storageMeta); err != nil {
|
|
return err
|
|
}
|
|
s.Type = storageMeta.Type
|
|
var c struct {
|
|
Config StorageConfig `yaml:"config"`
|
|
}
|
|
// TODO(ericchiang): replace this with a registration process.
|
|
switch storageMeta.Type {
|
|
case "kubernetes":
|
|
c.Config = &kubernetes.Config{}
|
|
case "memory":
|
|
c.Config = &memory.Config{}
|
|
default:
|
|
return fmt.Errorf("unknown storage type %q", storageMeta.Type)
|
|
}
|
|
if err := unmarshal(c); err != nil {
|
|
return err
|
|
}
|
|
s.Config = c.Config
|
|
return nil
|
|
}
|
|
|
|
// StorageConfig is a configuration that can create a storage.
|
|
type StorageConfig interface {
|
|
Open() (storage.Storage, error)
|
|
}
|
|
|
|
// Connector is a magical type that can unmarshal YAML dynamically. The
|
|
// Type field determines the connector type, which is then customized for Config.
|
|
type Connector struct {
|
|
Type string `yaml:"type"`
|
|
Name string `yaml:"name"`
|
|
ID string `yaml:"id"`
|
|
|
|
Config ConnectorConfig `yaml:"config"`
|
|
}
|
|
|
|
// ConnectorConfig is a configuration that can open a connector.
|
|
type ConnectorConfig interface {
|
|
Open() (connector.Connector, error)
|
|
}
|
|
|
|
// UnmarshalYAML allows Connector to unmarshal its config field dynamically
|
|
// depending on the type of connector.
|
|
func (c *Connector) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
var connectorMetadata struct {
|
|
Type string `yaml:"type"`
|
|
Name string `yaml:"name"`
|
|
ID string `yaml:"id"`
|
|
}
|
|
if err := unmarshal(&connectorMetadata); err != nil {
|
|
return err
|
|
}
|
|
c.Type = connectorMetadata.Type
|
|
c.Name = connectorMetadata.Name
|
|
c.ID = connectorMetadata.ID
|
|
|
|
switch c.Type {
|
|
case "mock":
|
|
var config struct {
|
|
Config mock.Config `yaml:"config"`
|
|
}
|
|
if err := unmarshal(&config); err != nil {
|
|
return err
|
|
}
|
|
c.Config = &config.Config
|
|
case "ldap":
|
|
var config struct {
|
|
Config ldap.Config `yaml:"config"`
|
|
}
|
|
if err := unmarshal(&config); err != nil {
|
|
return err
|
|
}
|
|
c.Config = &config.Config
|
|
case "github":
|
|
var config struct {
|
|
Config github.Config `yaml:"config"`
|
|
}
|
|
if err := unmarshal(&config); err != nil {
|
|
return err
|
|
}
|
|
c.Config = &config.Config
|
|
default:
|
|
return fmt.Errorf("unknown connector type %q", c.Type)
|
|
}
|
|
return nil
|
|
}
|