daemon: allowing adding external http handlers

internals/daemon/api.go

@@ -107,7 +107,7 @@ var (
 	muxVars = mux.Vars
 )
 
-func v1SystemInfo(c *Command, r *http.Request, _ *userState) Response {
+func v1SystemInfo(c *Command, r *http.Request, _ *UserState) Response {
 	state := c.d.overlord.State()
 	state.Lock()
 	defer state.Unlock()
@@ -117,3 +117,7 @@ func v1SystemInfo(c *Command, r *http.Request, _ *userState) Response {
 	}
 	return SyncResponse(result)
 }
+
+func Register(c *Command) {
+	api = append(api, c)
+}

internals/daemon/api_changes.go

@@ -115,7 +115,7 @@ func change2changeInfo(chg *state.Change) *changeInfo {
 	return chgInfo
 }
 
-func v1GetChanges(c *Command, r *http.Request, _ *userState) Response {
+func v1GetChanges(c *Command, r *http.Request, _ *UserState) Response {
 	query := r.URL.Query()
 	qselect := query.Get("select")
 	if qselect == "" {
@@ -170,7 +170,7 @@ func v1GetChanges(c *Command, r *http.Request, _ *userState) Response {
 	return SyncResponse(chgInfos)
 }
 
-func v1GetChange(c *Command, r *http.Request, _ *userState) Response {
+func v1GetChange(c *Command, r *http.Request, _ *UserState) Response {
 	changeID := muxVars(r)["id"]
 	st := c.d.overlord.State()
 	st.Lock()
@@ -183,7 +183,7 @@ func v1GetChange(c *Command, r *http.Request, _ *userState) Response {
 	return SyncResponse(change2changeInfo(chg))
 }
 
-func v1GetChangeWait(c *Command, r *http.Request, _ *userState) Response {
+func v1GetChangeWait(c *Command, r *http.Request, _ *UserState) Response {
 	changeID := muxVars(r)["id"]
 	st := c.d.overlord.State()
 	st.Lock()
@@ -224,7 +224,7 @@ func v1GetChangeWait(c *Command, r *http.Request, _ *userState) Response {
 	return SyncResponse(change2changeInfo(change))
 }
 
-func v1PostChange(c *Command, r *http.Request, _ *userState) Response {
+func v1PostChange(c *Command, r *http.Request, _ *UserState) Response {
 	chID := muxVars(r)["id"]
 	state := c.d.overlord.State()
 	state.Lock()

internals/daemon/api_checks.go

@@ -30,7 +30,7 @@ type checkInfo struct {
 	Threshold int    `json:"threshold"`
 }
 
-func v1GetChecks(c *Command, r *http.Request, _ *userState) Response {
+func v1GetChecks(c *Command, r *http.Request, _ *UserState) Response {
 	query := r.URL.Query()
 	level := plan.CheckLevel(query.Get("level"))
 	switch level {

internals/daemon/api_exec.go

@@ -44,7 +44,7 @@ type execPayload struct {
 	Height         int               `json:"height"`
 }
 
-func v1PostExec(c *Command, req *http.Request, _ *userState) Response {
+func v1PostExec(c *Command, req *http.Request, _ *UserState) Response {
 	var payload execPayload
 	decoder := json.NewDecoder(req.Body)
 	if err := decoder.Decode(&payload); err != nil {

internals/daemon/api_files.go

@@ -37,7 +37,7 @@ import (
 
 const minBoundaryLength = 32
 
-func v1GetFiles(_ *Command, req *http.Request, _ *userState) Response {
+func v1GetFiles(_ *Command, req *http.Request, _ *UserState) Response {
 	query := req.URL.Query()
 	action := query.Get("action")
 	switch action {
@@ -337,7 +337,7 @@ func listFiles(path, pattern string, itself bool) ([]fileInfoResult, error) {
 	return result, nil
 }
 
-func v1PostFiles(_ *Command, req *http.Request, _ *userState) Response {
+func v1PostFiles(_ *Command, req *http.Request, _ *UserState) Response {
 	contentType := req.Header.Get("Content-Type")
 	mediaType, params, err := mime.ParseMediaType(contentType)
 	if err != nil {

internals/daemon/api_health.go

@@ -28,7 +28,7 @@ type healthInfo struct {
 	Healthy bool `json:"healthy"`
 }
 
-func v1Health(c *Command, r *http.Request, _ *userState) Response {
+func v1Health(c *Command, r *http.Request, _ *UserState) Response {
 	query := r.URL.Query()
 	level := plan.CheckLevel(query.Get("level"))
 	switch level {

internals/daemon/api_logs.go

@@ -42,7 +42,7 @@ type serviceManager interface {
 	ServiceLogs(services []string, last int) (map[string]servicelog.Iterator, error)
 }
 
-func v1GetLogs(c *Command, _ *http.Request, _ *userState) Response {
+func v1GetLogs(c *Command, _ *http.Request, _ *UserState) Response {
 	return logsResponse{
 		svcMgr: overlordServiceManager(c.d.overlord),
 	}

internals/daemon/api_plan.go

@@ -24,7 +24,7 @@ import (
 	"github.com/canonical/pebble/internals/plan"
 )
 
-func v1GetPlan(c *Command, r *http.Request, _ *userState) Response {
+func v1GetPlan(c *Command, r *http.Request, _ *UserState) Response {
 	format := r.URL.Query().Get("format")
 	if format != "yaml" {
 		return statusBadRequest("invalid format %q", format)
@@ -42,7 +42,7 @@ func v1GetPlan(c *Command, r *http.Request, _ *userState) Response {
 	return SyncResponse(string(planYAML))
 }
 
-func v1PostLayers(c *Command, r *http.Request, _ *userState) Response {
+func v1PostLayers(c *Command, r *http.Request, _ *UserState) Response {
 	var payload struct {
 		Action  string `json:"action"`
 		Combine bool   `json:"combine"`

internals/daemon/api_services.go

@@ -35,7 +35,7 @@ type serviceInfo struct {
 	CurrentSince *time.Time `json:"current-since,omitempty"` // pointer as omitempty doesn't work with time.Time directly
 }
 
-func v1GetServices(c *Command, r *http.Request, _ *userState) Response {
+func v1GetServices(c *Command, r *http.Request, _ *UserState) Response {
 	names := strutil.MultiCommaSeparatedList(r.URL.Query()["names"])
 
 	servmgr := overlordServiceManager(c.d.overlord)
@@ -59,7 +59,7 @@ func v1GetServices(c *Command, r *http.Request, _ *userState) Response {
 	return SyncResponse(infos)
 }
 
-func v1PostServices(c *Command, r *http.Request, _ *userState) Response {
+func v1PostServices(c *Command, r *http.Request, _ *UserState) Response {
 	var payload struct {
 		Action   string   `json:"action"`
 		Services []string `json:"services"`
@@ -212,11 +212,11 @@ func v1PostServices(c *Command, r *http.Request, _ *userState) Response {
 	return AsyncResponse(nil, change.ID())
 }
 
-func v1GetService(c *Command, r *http.Request, _ *userState) Response {
+func v1GetService(c *Command, r *http.Request, _ *UserState) Response {
 	return statusBadRequest("not implemented")
 }
 
-func v1PostService(c *Command, r *http.Request, _ *userState) Response {
+func v1PostService(c *Command, r *http.Request, _ *UserState) Response {
 	return statusBadRequest("not implemented")
 }
 

internals/daemon/api_signals.go

@@ -24,7 +24,7 @@ type signalsPayload struct {
 	Services []string `json:"services"`
 }
 
-func v1PostSignals(c *Command, req *http.Request, _ *userState) Response {
+func v1PostSignals(c *Command, req *http.Request, _ *UserState) Response {
 	var payload signalsPayload
 	decoder := json.NewDecoder(req.Body)
 	if err := decoder.Decode(&payload); err != nil {

internals/daemon/api_tasks.go

@@ -23,7 +23,7 @@ import (
 	"github.com/canonical/pebble/internals/overlord/state"
 )
 
-func v1GetTaskWebsocket(c *Command, req *http.Request, _ *userState) Response {
+func v1GetTaskWebsocket(c *Command, req *http.Request, _ *UserState) Response {
 	vars := muxVars(req)
 	taskID := vars["task-id"]
 	websocketID := vars["websocket-id"]

internals/daemon/api_warnings.go

@@ -22,7 +22,7 @@ import (
 	"github.com/canonical/pebble/internals/overlord/state"
 )
 
-func v1AckWarnings(c *Command, r *http.Request, _ *userState) Response {
+func v1AckWarnings(c *Command, r *http.Request, _ *UserState) Response {
 	defer r.Body.Close()
 	var op struct {
 		Action    string    `json:"action"`
@@ -43,7 +43,7 @@ func v1AckWarnings(c *Command, r *http.Request, _ *userState) Response {
 	return SyncResponse(n)
 }
 
-func v1GetWarnings(c *Command, r *http.Request, _ *userState) Response {
+func v1GetWarnings(c *Command, r *http.Request, _ *UserState) Response {
 	query := r.URL.Query()
 	var all bool
 	sel := query.Get("select")

internals/daemon/daemon.go

@@ -109,10 +109,10 @@ type Daemon struct {
 }
 
 // XXX Placeholder for now.
-type userState struct{}
+type UserState struct{}
 
 // A ResponseFunc handles one of the individual verbs for a method
-type ResponseFunc func(*Command, *http.Request, *userState) Response
+type ResponseFunc func(*Command, *http.Request, *UserState) Response
 
 // A Command routes a request to an individual per-verb ResponseFUnc
 type Command struct {
@@ -150,7 +150,7 @@ const (
 // - UserOK: any uid on the local system can access GET
 // - AdminOnly: only the administrator can access this
 // - UntrustedOK: can access this via the untrusted socket
-func (c *Command) canAccess(r *http.Request, user *userState) accessResult {
+func (c *Command) canAccess(r *http.Request, user *UserState) accessResult {
 	if c.AdminOnly && (c.UserOK || c.GuestOK || c.UntrustedOK) {
 		logger.Panicf("internal error: command cannot have AdminOnly together with any *OK flag")
 	}
@@ -211,7 +211,7 @@ func (c *Command) canAccess(r *http.Request, user *userState) accessResult {
 	return accessUnauthorized
 }
 
-func userFromRequest(state interface{}, r *http.Request) (*userState, error) {
+func userFromRequest(state interface{}, r *http.Request) (*UserState, error) {
 	return nil, nil
 }
 

internals/daemon/daemon_test.go

@@ -98,6 +98,29 @@ func (h *fakeHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	h.lastMethod = r.Method
 }
 
+func (s *daemonSuite) TestRegisterCommand(c *C) {
+	const endpoint = "/v1/addedendpoint"
+	var handler fakeHandler
+	getCallback := func(c *Command, r *http.Request, s *UserState) Response {
+		handler.cmd = c
+		return &handler
+	}
+	command := Command{
+		Path:    endpoint,
+		GuestOK: true,
+		GET:     getCallback,
+	}
+	Register(&command)
+
+	d := s.newDaemon(c)
+	d.Init()
+	d.Start()
+	defer d.Stop(nil)
+
+	result := d.router.Get(endpoint).GetHandler()
+	c.Assert(result, Equals, &command)
+}
+
 func (s *daemonSuite) TestExplicitPaths(c *C) {
 	s.socketPath = filepath.Join(c.MkDir(), "custom.socket")
 
@@ -120,7 +143,7 @@ func (s *daemonSuite) TestCommandMethodDispatch(c *check.C) {
 
 	cmd := &Command{d: s.newDaemon(c)}
 	handler := &fakeHandler{cmd: cmd}
-	rf := func(innerCmd *Command, req *http.Request, user *userState) Response {
+	rf := func(innerCmd *Command, req *http.Request, user *UserState) Response {
 		c.Assert(cmd, check.Equals, innerCmd)
 		return handler
 	}
@@ -159,7 +182,7 @@ func (s *daemonSuite) TestCommandRestartingState(c *check.C) {
 	d := s.newDaemon(c)
 
 	cmd := &Command{d: d}
-	cmd.GET = func(*Command, *http.Request, *userState) Response {
+	cmd.GET = func(*Command, *http.Request, *UserState) Response {
 		return SyncResponse(nil)
 	}
 	req, err := http.NewRequest("GET", "", nil)
@@ -209,7 +232,7 @@ func (s *daemonSuite) TestFillsWarnings(c *check.C) {
 	d := s.newDaemon(c)
 
 	cmd := &Command{d: d}
-	cmd.GET = func(*Command, *http.Request, *userState) Response {
+	cmd.GET = func(*Command, *http.Request, *UserState) Response {
 		return SyncResponse(nil)
 	}
 	req, err := http.NewRequest("GET", "", nil)
@@ -340,7 +363,7 @@ func (s *daemonSuite) TestUserAccess(c *check.C) {
 func (s *daemonSuite) TestLoggedInUserAccess(c *check.C) {
 	d := s.newDaemon(c)
 
-	user := &userState{}
+	user := &UserState{}
 	get := &http.Request{Method: "GET", RemoteAddr: "pid=100;uid=42;socket=;"}
 	put := &http.Request{Method: "PUT", RemoteAddr: "pid=100;uid=42;socket=;"}
 
@@ -1027,10 +1050,10 @@ func doTestReq(c *check.C, cmd *Command, mth string) *httptest.ResponseRecorder
 func (s *daemonSuite) TestDegradedModeReply(c *check.C) {
 	d := s.newDaemon(c)
 	cmd := &Command{d: d}
-	cmd.GET = func(*Command, *http.Request, *userState) Response {
+	cmd.GET = func(*Command, *http.Request, *UserState) Response {
 		return SyncResponse(nil)
 	}
-	cmd.POST = func(*Command, *http.Request, *userState) Response {
+	cmd.POST = func(*Command, *http.Request, *UserState) Response {
 		return SyncResponse(nil)
 	}