diff --git a/app.go b/app.go index f3c7b1b..0ef08fc 100644 --- a/app.go +++ b/app.go @@ -82,6 +82,7 @@ func (s serviceHandlerDocumentationList) Swap(i, j int) { s[i], s[j] = s[j], s[i type serviceHandler interface { GetDocumentation() serviceHandlerDocumentationList + IsEnabled() bool Handle(ctx context.Context, params []string) (title, text, color string, err error) } @@ -142,7 +143,7 @@ func generateServiceBadge(res http.ResponseWriter, r *http.Request) { defer cancel() handler, ok := serviceHandlers[service] - if !ok { + if !ok || !handler.IsEnabled() { http.Error(res, "Service not found: "+service, http.StatusNotFound) return } @@ -242,6 +243,10 @@ func handleDemoPage(res http.ResponseWriter, r *http.Request) { examples := serviceHandlerDocumentationList{} for register, handler := range serviceHandlers { + if !handler.IsEnabled() { + continue + } + tmps := handler.GetDocumentation() for _, tmp := range tmps { tmp.Register = register diff --git a/service_aur.go b/service_aur.go index be5be7b..7b13594 100644 --- a/service_aur.go +++ b/service_aur.go @@ -69,6 +69,8 @@ func (a aurServiceHandler) GetDocumentation() serviceHandlerDocumentationList { } } +func (aurServiceHandler) IsEnabled() bool { return true } + func (a aurServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) { if len(params) < 2 { return title, text, color, errors.New("No service-command / parameters were given") diff --git a/service_beerpay.go b/service_beerpay.go index dc337f0..c04d668 100644 --- a/service_beerpay.go +++ b/service_beerpay.go @@ -24,6 +24,8 @@ func (s beerpayServiceHandler) GetDocumentation() serviceHandlerDocumentationLis }} } +func (beerpayServiceHandler) IsEnabled() bool { return true } + func (s beerpayServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) { if len(params) < 2 { err = errors.New("You need to provide user and project") diff --git a/service_github.go b/service_github.go index 438100f..ecf4a92 100644 --- a/service_github.go +++ b/service_github.go @@ -71,6 +71,8 @@ func (g githubServiceHandler) GetDocumentation() serviceHandlerDocumentationList } } +func (githubServiceHandler) IsEnabled() bool { return true } + func (g githubServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) { if len(params) < 2 { err = errors.New("No service-command / parameters were given") diff --git a/service_liberapay.go b/service_liberapay.go index b531807..02a129b 100644 --- a/service_liberapay.go +++ b/service_liberapay.go @@ -44,6 +44,8 @@ func (s liberapayServiceHandler) GetDocumentation() serviceHandlerDocumentationL } } +func (liberapayServiceHandler) IsEnabled() bool { return true } + func (s liberapayServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) { if len(params) < 2 { err = errors.New("You need to provide user and payment direction") diff --git a/service_static.go b/service_static.go index 0ba1e90..69386c3 100644 --- a/service_static.go +++ b/service_static.go @@ -20,6 +20,8 @@ func (s staticServiceHandler) GetDocumentation() serviceHandlerDocumentationList }} } +func (staticServiceHandler) IsEnabled() bool { return true } + func (s staticServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) { if len(params) < 2 { err = errors.New("You need to provide title and text") diff --git a/service_travis.go b/service_travis.go index 7816b7b..362485a 100644 --- a/service_travis.go +++ b/service_travis.go @@ -24,6 +24,8 @@ func (t travisServiceHandler) GetDocumentation() serviceHandlerDocumentationList }} } +func (travisServiceHandler) IsEnabled() bool { return true } + func (t travisServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) { if len(params) < 2 { err = errors.New("You need to provide user and repo")