forked from snapcore/snapd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
configmgr.go
70 lines (61 loc) · 2.13 KB
/
configmgr.go
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
60
61
62
63
64
65
66
67
68
69
70
// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2016-2017 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package configstate
import (
"regexp"
"github.com/snapcore/snapd/overlord/configstate/config"
"github.com/snapcore/snapd/overlord/configstate/configcore"
"github.com/snapcore/snapd/overlord/hookstate"
"github.com/snapcore/snapd/overlord/snapstate"
"github.com/snapcore/snapd/overlord/state"
"github.com/snapcore/snapd/sysconfig"
)
var configcoreRun = configcore.Run
func MockConfigcoreRun(f func(sysconfig.Device, config.Conf) error) (restore func()) {
origConfigcoreRun := configcoreRun
configcoreRun = f
return func() {
configcoreRun = origConfigcoreRun
}
}
func Init(st *state.State, hookManager *hookstate.HookManager) error {
delayedCrossMgrInit()
// Most configuration is handled via the "configure" hook of the
// snaps. However some configuration is internally handled
hookManager.Register(regexp.MustCompile("^configure$"), newConfigureHandler)
// Ensure that we run configure for the core snap internally.
// Note that we use the func() indirection so that mocking configcoreRun
// in tests works correctly.
hookManager.RegisterHijack("configure", "core", func(ctx *hookstate.Context) error {
dev, tr, err := func() (sysconfig.Device, config.Conf, error) {
ctx.Lock()
defer ctx.Unlock()
task, _ := ctx.Task()
dev, err := snapstate.DeviceCtx(ctx.State(), task, nil)
if err != nil {
return nil, nil, err
}
return dev, ContextTransaction(ctx), nil
}()
if err != nil {
return err
}
return configcoreRun(dev, tr)
})
return nil
}