Esempio n. 1
0
// Server starts gateway server created from the given configuration.
//
// It returns a function that can be used to explicitely stop
// the kite server.
func (d *Driver) Server(cfg *keygen.Config) (cancel func()) {
	kiteCfg := d.Kite(cfg, "keygen")

	keygen.NewServer(kiteCfg)

	go kiteCfg.Kite.Run()
	<-kiteCfg.Kite.ServerReadyNotify()

	cfg.ServerURL = fmt.Sprintf("http://127.0.0.1:%d/kite", kiteCfg.Kite.Port())

	return kiteCfg.Kite.Close
}
Esempio n. 2
0
// BeforeFunc spies on cfg.BeforeFunc calls, sending passed time.Time
// to the returned channel.
//
// If cfg.BeforeFunc is non-nil, it is called after the value is sent.
func (d *Driver) BeforeFunc(cfg *keygen.Config) <-chan time.Time {
	ch := make(chan time.Time, d.ChanCap)
	fn := cfg.BeforeFunc

	cfg.BeforeFunc = func(t time.Time) bool {
		ch <- t

		if fn != nil {
			return fn(t)
		}

		return keygen.DefaultBefore(t)
	}

	return ch
}
Esempio n. 3
0
// AuthFunc spies on cfg.AuthFunc calls, sending passed AuthRequest
// to the returned channel.
//
// If cfg.AuthFunc is non-nil, it is called after the value is sent.
func (d *Driver) AuthFunc(cfg *keygen.Config) <-chan *keygen.AuthRequest {
	ch := make(chan *keygen.AuthRequest, d.ChanCap)
	fn := cfg.AuthFunc

	cfg.AuthFunc = func(req *keygen.AuthRequest) error {
		ch <- req

		if fn != nil {
			return fn(req)
		}

		return nil
	}

	return ch
}