2019-06-03 18:44:43 +01:00
|
|
|
package server
|
|
|
|
|
|
|
|
import (
|
|
|
|
"crypto/tls"
|
2020-04-07 19:29:26 +01:00
|
|
|
"net/http"
|
2019-10-09 16:42:05 +01:00
|
|
|
|
2020-07-27 13:22:00 +01:00
|
|
|
"github.com/micro/go-micro/v3/api/resolver"
|
|
|
|
"github.com/micro/go-micro/v3/api/server/acme"
|
2019-06-03 18:44:43 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
type Option func(o *Options)
|
|
|
|
|
|
|
|
type Options struct {
|
2020-04-07 19:29:26 +01:00
|
|
|
EnableACME bool
|
|
|
|
EnableCORS bool
|
|
|
|
ACMEProvider acme.Provider
|
|
|
|
EnableTLS bool
|
|
|
|
ACMEHosts []string
|
|
|
|
TLSConfig *tls.Config
|
|
|
|
Resolver resolver.Resolver
|
|
|
|
Wrappers []Wrapper
|
|
|
|
}
|
|
|
|
|
|
|
|
type Wrapper func(h http.Handler) http.Handler
|
|
|
|
|
2020-08-10 22:38:54 +01:00
|
|
|
func WrapHandler(w ...Wrapper) Option {
|
2020-04-07 19:29:26 +01:00
|
|
|
return func(o *Options) {
|
2020-08-10 22:38:54 +01:00
|
|
|
o.Wrappers = append(o.Wrappers, w...)
|
2020-04-07 19:29:26 +01:00
|
|
|
}
|
2019-10-09 16:42:05 +01:00
|
|
|
}
|
|
|
|
|
2020-03-04 11:40:53 +00:00
|
|
|
func EnableCORS(b bool) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.EnableCORS = b
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-10-09 16:42:05 +01:00
|
|
|
func EnableACME(b bool) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.EnableACME = b
|
|
|
|
}
|
2019-06-03 18:44:43 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func ACMEHosts(hosts ...string) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.ACMEHosts = hosts
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-10-09 16:42:05 +01:00
|
|
|
func ACMEProvider(p acme.Provider) Option {
|
2019-06-03 18:44:43 +01:00
|
|
|
return func(o *Options) {
|
2019-10-09 16:42:05 +01:00
|
|
|
o.ACMEProvider = p
|
2019-06-03 18:44:43 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func EnableTLS(b bool) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.EnableTLS = b
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TLSConfig(t *tls.Config) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.TLSConfig = t
|
|
|
|
}
|
|
|
|
}
|
2020-04-02 17:44:48 +01:00
|
|
|
|
2020-04-07 16:24:51 +01:00
|
|
|
func Resolver(r resolver.Resolver) Option {
|
2020-04-02 17:44:48 +01:00
|
|
|
return func(o *Options) {
|
2020-04-07 16:24:51 +01:00
|
|
|
o.Resolver = r
|
2020-04-02 17:44:48 +01:00
|
|
|
}
|
|
|
|
}
|