summaryrefslogtreecommitdiffstats
path: root/app/options.go
blob: 121bbbf8035c25d0dfd33ff4b2f43fbddecb929d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package app

import (
	"github.com/mattermost/mattermost-server/model"
	"github.com/mattermost/mattermost-server/store"
)

type Option func(a *App)

// By default, the app will use a global configuration file. This allows you to override all or part
// of that configuration.
//
// The override parameter must be a *model.Config, func(*model.Config), or func(*model.Config) *model.Config.
//
// XXX: Most code will not respect this at the moment. (We need to eliminate utils.Cfg first.)
func ConfigOverride(override interface{}) Option {
	return func(a *App) {
		switch o := override.(type) {
		case *model.Config:
			a.configOverride = func(*model.Config) *model.Config {
				return o
			}
		case func(*model.Config):
			a.configOverride = func(cfg *model.Config) *model.Config {
				ret := *cfg
				o(&ret)
				return &ret
			}
		case func(*model.Config) *model.Config:
			a.configOverride = o
		default:
			panic("invalid ConfigOverride")
		}
	}
}

// By default, the app will use the store specified by the configuration. This allows you to
// construct an app with a different store.
//
// The override parameter must be either a store.Store or func(App) store.Store.
func StoreOverride(override interface{}) Option {
	return func(a *App) {
		switch o := override.(type) {
		case store.Store:
			a.newStore = func() store.Store {
				return o
			}
		case func(*App) store.Store:
			a.newStore = func() store.Store {
				return o(a)
			}
		default:
			panic("invalid StoreOverride")
		}
	}
}