File: api_console_conf.go

package info (click to toggle)
snapd 2.71-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 79,536 kB
  • sloc: ansic: 16,114; sh: 16,105; python: 9,941; makefile: 1,890; exp: 190; awk: 40; xml: 22
file content (96 lines) | stat: -rw-r--r-- 2,982 bytes parent folder | download | duplicates (3)
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
// -*- Mode: Go; indent-tabs-mode: t -*-

/*
 * Copyright (C) 2020 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 daemon

import (
	"encoding/json"
	"io"
	"net/http"
	"time"

	"github.com/snapcore/snapd/logger"
	"github.com/snapcore/snapd/overlord/auth"
)

var (
	routineConsoleConfStartCmd = &Command{
		Path:        "/v2/internal/console-conf-start",
		POST:        consoleConfStartRoutine,
		WriteAccess: authenticatedAccess{},
	}
)

var delayTime = 20 * time.Minute

// consoleConfStartRoutineResult is the result of running the console-conf start
// routine..
type consoleConfStartRoutineResult struct {
	ActiveAutoRefreshChanges []string `json:"active-auto-refreshes,omitempty"`
	ActiveAutoRefreshSnaps   []string `json:"active-auto-refresh-snaps,omitempty"`
}

func consoleConfStartRoutine(c *Command, r *http.Request, _ *auth.UserState) Response {
	// no body expected, error if we were provided anything
	defer r.Body.Close()
	var routineBody struct{}
	decoder := json.NewDecoder(r.Body)
	if err := decoder.Decode(&routineBody); err != nil && err != io.EOF {
		return BadRequest("cannot decode request body into console-conf operation: %v", err)
	}

	// now run the start routine first by trying to grab a lock on the refreshes
	// for all snaps, which fails if there are any active changes refreshing
	// snaps
	st := c.d.overlord.State()
	st.Lock()
	defer st.Unlock()

	snapAutoRefreshChanges, err := c.d.overlord.SnapManager().EnsureAutoRefreshesAreDelayed(delayTime)
	if err != nil {
		return InternalError(err.Error())
	}

	logger.Debugf("Ensured that new auto refreshes are delayed by %s to allow console-conf to run", delayTime)

	if len(snapAutoRefreshChanges) == 0 {
		// no changes yet, and we delayed the refresh successfully so
		// console-conf is okay to run normally
		return SyncResponse(&consoleConfStartRoutineResult{})
	}

	chgIds := make([]string, 0, len(snapAutoRefreshChanges))
	snapNames := make([]string, 0)
	for _, chg := range snapAutoRefreshChanges {
		chgIds = append(chgIds, chg.ID())
		var updatedSnaps []string
		err := chg.Get("snap-names", &updatedSnaps)
		if err != nil {
			return InternalError(err.Error())
		}
		snapNames = append(snapNames, updatedSnaps...)
	}

	// we have changes that the client should wait for before being ready
	return SyncResponse(&consoleConfStartRoutineResult{
		ActiveAutoRefreshChanges: chgIds,
		ActiveAutoRefreshSnaps:   snapNames,
	})

}