// NewServerKite creates a server kite for the given server.
func NewServerKite(s *Server, name, version string) (*kite.Kite, error) {
k := kite.New(name, version)
if s.opts.Config == nil {
cfg, err := config.Get()
if err != nil {
return nil, err
}
s.opts.Config = cfg
}
s.opts.Config.KontrolURL = s.opts.kontrolURL()
s.opts.Config.Transport = config.XHRPolling
if s.opts.Port != 0 {
s.opts.Config.Port = s.opts.Port
}
if s.opts.Region != "" {
s.opts.Config.Region = s.opts.Region
}
if s.opts.Environment != "" {
s.opts.Config.Environment = s.opts.Environment
}
if s.opts.Test {
s.opts.Config.DisableAuthentication = true
}
if s.opts.Debug {
k.SetLogLevel(kite.DEBUG)
}
k.Log = s.opts.Log
k.Config = s.opts.Config
k.ClientFunc = httputil.ClientFunc(s.opts.Debug)
if fn := s.metricsFunc(); fn != nil {
k.PreHandleFunc(fn)
}
k.HandleFunc("register", s.Register)
k.HandleFunc("registerServices", s.RegisterServices)
k.HandleHTTPFunc("/healthCheck", artifact.HealthCheckHandler(name))
k.HandleHTTPFunc("/version", artifact.VersionHandler())
// Tunnel helper methods, like ports, stats etc.
k.HandleHTTPFunc("/-/discover/{service}", s.discoverHandler())
// Route all the rest requests (match all paths that does not begin with /-/).
k.HandleHTTP(`/{rest:.?$|[^\/].+|\/[^-].+|\/-[^\/].*}`, s.serverHandler())
u, err := url.Parse(s.opts.registerURL())
if err != nil {
return nil, fmt.Errorf("error parsing registerURL: %s", err)
}
if err := k.RegisterForever(u); err != nil {
return nil, fmt.Errorf("error registering to Kontrol: %s", err)
}
return k, nil
}
func main() {
loader := multiconfig.MultiLoader(
&multiconfig.TagLoader{},
&multiconfig.EnvironmentLoader{Prefix: "kontrol"},
&multiconfig.EnvironmentLoader{Prefix: "KONFIG_KONTROL"},
&multiconfig.FlagLoader{EnvPrefix: "kontrol"},
)
conf := new(kontrol.Config)
// Load the config, it's reads from the file, environment variables and
// lastly from flags in order
if err := loader.Load(conf); err != nil {
panic(err)
}
if err := multiconfig.MultiValidator(&multiconfig.RequiredValidator{}).Validate(conf); err != nil {
panic(err)
}
fmt.Printf("Kontrol loaded with following variables: %+v\n", conf)
k := kontrol.New(conf)
k.Kite.HandleHTTPFunc("/healthCheck", artifact.HealthCheckHandler(Name))
k.Kite.HandleHTTPFunc("/version", artifact.VersionHandler())
if conf.Debug {
k.Kite.SetLogLevel(kite.DEBUG)
}
go func() {
// Kloud runs on 6060, so we choose 6061 for kontrol
err := http.ListenAndServe("0.0.0.0:6061", nil)
k.Kite.Log.Error(err.Error())
}()
k.Run()
}
func (m *Mux) addDefaultHandlers() *tigertonic.TrieServeMux {
m.AddUnscopedHandler(handler.Request{
Type: handler.GetRequest,
Endpoint: "/version",
Handler: artifact.VersionHandler(),
})
m.AddUnscopedHandler(handler.Request{
Type: handler.GetRequest,
Endpoint: "/healthCheck",
Handler: artifact.HealthCheckHandler(m.config.Name),
})
m.AddUnscopedHandler(handler.Request{
Type: handler.GetRequest,
Endpoint: "/",
Handler: func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Hello from %s", m.config.Name)
},
})
return m.mux
}
// NewKite creates a new kite for serving terraformer
func NewKite(t *Terraformer, conf *Config) (*kite.Kite, error) {
var err error
k := kite.New(Name, Version)
k.Config, err = kiteconfig.Get()
if err != nil {
return nil, err
}
k = setupKite(k, conf)
// handle current status of terraformer
k.PostHandleFunc(t.handleState)
// track every kind of call
k.PreHandleFunc(createTracker(t.Metrics))
// Terraformer handling methods
k.HandleFunc("apply", t.Apply)
k.HandleFunc("destroy", t.Destroy)
k.HandleFunc("plan", t.Plan)
// artifact handling
k.HandleHTTPFunc("/healthCheck", artifact.HealthCheckHandler(Name))
k.HandleHTTPFunc("/version", artifact.VersionHandler())
secretKey := conf.SecretKey
// allow kloud to make calls to us
k.Authenticators["kloud"] = func(r *kite.Request) error {
if r.Auth.Key != secretKey {
return errors.New("wrong secret key passed, you are not authenticated")
}
return nil
}
return k, nil
}
// New gives new, registered kloud kite.
//
// If conf contains invalid or missing configuration, it return non-nil error.
func New(conf *Config) (*Kloud, error) {
k := kite.New(stack.NAME, stack.VERSION)
k.Config = kiteconfig.MustGet()
k.Config.Port = conf.Port
k.ClientFunc = httputil.ClientFunc(conf.DebugMode)
if conf.DebugMode {
k.SetLogLevel(kite.DEBUG)
}
if conf.Region != "" {
k.Config.Region = conf.Region
}
if conf.Environment != "" {
k.Config.Environment = conf.Environment
}
// TODO(rjeczalik): refactor modelhelper methods to not use global DB
modelhelper.Initialize(conf.MongoURL)
sess, err := newSession(conf, k)
if err != nil {
return nil, err
}
e := newEndpoints(conf)
sess.Log.Debug("Konfig.Endpoints: %s", util.LazyJSON(e))
authUsers := map[string]string{
"kloudctl": conf.KloudSecretKey,
}
restClient := httputil.DefaultRestClient(conf.DebugMode)
storeOpts := &credential.Options{
MongoDB: sess.DB,
Log: sess.Log.New("stackcred"),
Client: restClient,
}
if !conf.NoSneaker {
storeOpts.CredURL = e.Social().WithPath("/credential").Private.URL
}
sess.Log.Debug("storeOpts: %+v", storeOpts)
userPrivateKey, userPublicKey := userMachinesKeys(conf.UserPublicKey, conf.UserPrivateKey)
stacker := &provider.Stacker{
DB: sess.DB,
Log: sess.Log,
Kite: sess.Kite,
Userdata: sess.Userdata,
Debug: conf.DebugMode,
KloudSecretKey: conf.KloudSecretKey,
CredStore: credential.NewStore(storeOpts),
TunnelURL: conf.TunnelURL,
SSHKey: &publickeys.Keys{
KeyName: publickeys.DeployKeyName,
PrivateKey: userPrivateKey,
PublicKey: userPublicKey,
},
}
stats := common.MustInitMetrics(Name)
kloud := &Kloud{
Kite: k,
Stack: stack.New(),
Queue: &queue.Queue{
Interval: 5 * time.Second,
Log: sess.Log.New("queue"),
Kite: k,
MongoDB: sess.DB,
},
}
authFn := func(opts *api.AuthOptions) (*api.Session, error) {
s, err := modelhelper.FetchOrCreateSession(opts.User.Username, opts.User.Team)
if err != nil {
return nil, err
}
return &api.Session{
ClientID: s.ClientId,
User: &api.User{
Username: s.Username,
Team: s.GroupName,
},
}, nil
}
transport := &api.Transport{
RoundTripper: storeOpts.Client.Transport,
AuthFunc: api.NewCache(authFn).Auth,
Debug: conf.DebugMode,
}
if conf.DebugMode {
transport.Log = sess.Log
}
kloud.Stack.Endpoints = e
kloud.Stack.Userdata = sess.Userdata
kloud.Stack.DescribeFunc = provider.Desc
kloud.Stack.CredClient = credential.NewClient(storeOpts)
kloud.Stack.MachineClient = machine.NewClient(machine.NewMongoDatabase())
kloud.Stack.TeamClient = team.NewClient(team.NewMongoDatabase())
kloud.Stack.PresenceClient = client.NewInternal(e.Social().Private.String())
kloud.Stack.PresenceClient.HTTPClient = restClient
kloud.Stack.RemoteClient = &remoteapi.Client{
Client: storeOpts.Client,
Transport: transport,
Endpoint: e.Koding.Private.URL,
}
kloud.Stack.ContextCreator = func(ctx context.Context) context.Context {
return session.NewContext(ctx, sess)
}
kloud.Stack.Metrics = stats
// RSA key pair that we add to the newly created machine for
// provisioning.
kloud.Stack.PublicKeys = stacker.SSHKey
kloud.Stack.DomainStorage = sess.DNSStorage
kloud.Stack.Domainer = sess.DNSClient
kloud.Stack.Locker = stacker
kloud.Stack.Log = sess.Log
kloud.Stack.SecretKey = conf.KloudSecretKey
for _, p := range provider.All() {
s := stacker.New(p)
if err = kloud.Stack.AddProvider(p.Name, s); err != nil {
return nil, err
}
kloud.Queue.Register(s)
sess.Log.Debug("registering %q provider", p.Name)
}
go kloud.Queue.Run()
if conf.KeygenAccessKey != "" && conf.KeygenSecretKey != "" {
cfg := &keygen.Config{
AccessKey: conf.KeygenAccessKey,
SecretKey: conf.KeygenSecretKey,
Region: conf.KeygenRegion,
Bucket: conf.KeygenBucket,
AuthExpire: conf.KeygenTokenTTL,
AuthFunc: kloud.Stack.ValidateUser,
Kite: k,
}
kloud.Keygen = keygen.NewServer(cfg)
} else {
k.Log.Warning(`disabling "keygen" methods due to missing S3/STS credentials`)
}
// Teams/stack handling methods.
k.HandleFunc("plan", kloud.Stack.Plan)
k.HandleFunc("apply", kloud.Stack.Apply)
k.HandleFunc("describeStack", kloud.Stack.Status)
k.HandleFunc("authenticate", kloud.Stack.Authenticate)
k.HandleFunc("bootstrap", kloud.Stack.Bootstrap)
k.HandleFunc("import", kloud.Stack.Import)
// Credential handling.
k.HandleFunc("credential.describe", kloud.Stack.CredentialDescribe)
k.HandleFunc("credential.list", kloud.Stack.CredentialList)
k.HandleFunc("credential.add", kloud.Stack.CredentialAdd)
// Authorization handling.
k.HandleFunc("auth.login", kloud.Stack.AuthLogin)
k.HandleFunc("auth.passwordLogin", kloud.Stack.AuthPasswordLogin).DisableAuthentication()
// Configuration handling.
k.HandleFunc("config.metadata", kloud.Stack.ConfigMetadata)
// Team handling.
k.HandleFunc("team.list", kloud.Stack.TeamList)
k.HandleFunc("team.whoami", kloud.Stack.TeamWhoami)
// Machine handling.
k.HandleFunc("machine.list", kloud.Stack.MachineList)
// Single machine handling.
k.HandleFunc("stop", kloud.Stack.Stop)
k.HandleFunc("start", kloud.Stack.Start)
k.HandleFunc("info", kloud.Stack.Info)
k.HandleFunc("event", kloud.Stack.Event)
// Klient proxy methods.
k.HandleFunc("admin.add", kloud.Stack.AdminAdd)
k.HandleFunc("admin.remove", kloud.Stack.AdminRemove)
k.HandleHTTPFunc("/healthCheck", artifact.HealthCheckHandler(Name))
k.HandleHTTPFunc("/version", artifact.VersionHandler())
for worker, key := range authUsers {
worker, key := worker, key
k.Authenticators[worker] = func(r *kite.Request) error {
if r.Auth.Key != key {
return errors.New("wrong secret key passed, you are not authenticated")
}
return nil
}
}
if conf.DebugMode {
// This should be actually debug level 2. It outputs every single Kite
// message and enables the kite debugging system. So enable it only if
// you need it.
// k.SetLogLevel(kite.DEBUG)
k.Log.Info("Debug mode enabled")
}
if conf.TestMode {
k.Log.Info("Test mode enabled")
}
registerURL := k.RegisterURL(!conf.Public)
if conf.RegisterURL != "" {
u, err := url.Parse(conf.RegisterURL)
if err != nil {
return nil, fmt.Errorf("Couldn't parse register url: %s", err)
}
registerURL = u
}
if err := k.RegisterForever(registerURL); err != nil {
return nil, err
}
return kloud, nil
}