Handle pause

Author: matteosuppo

app/contexts.go

@@ -38,8 +38,12 @@ func main() {
 	var (
 		hibernate = flag.Bool("hibernate", false, "start hibernated")
 	)
+
+	flag.Parse()
+
 	opts := agent.Opts{
 		Hibernate: *hibernate,
 	}
+
 	agent.Start(opts)
 }

app/controllers.go

@@ -38,9 +38,19 @@ var _ = Resource("manage_v1", func() {
 
 	Action("info", func() {
 		Description("Returns the info about the agent")
-		Routing(GET(""))
+		Routing(GET("/info"))
 		Response(OK, ManageInfoV1)
 	})
+	Action("pause", func() {
+		Description("Restarts the agent in hibernation mode")
+		Routing(POST("/pause"))
+		Response(OK)
+	})
+	Action("update", func() {
+		Description("Search for a new version, updates and restarts itself")
+		Routing(POST("/update"))
+		Response(OK)
+	})
 })
 
 var ManageInfoV1 = MediaType("application/vnd.arduino.agent.manage.info+json", func() {

app/test/manage_v1_testing.go

@@ -77,6 +77,12 @@ func Start(opts Opts) {
 	monitor := discovery.New(1 * time.Second)
 	monitor.Start(context.Background())
 
+	// Setup helper functions
+	restart := restartFunc("", !opts.Hibernate)
+	shutdown := func() {
+		os.Exit(0)
+	}
+
 	// Mount middleware
 	service.Use(middleware.RequestID())
 	service.Use(middleware.LogRequest(true))
@@ -92,7 +98,7 @@ func Start(opts Opts) {
 	app.MountDiscoverV1Controller(service, d)
 
 	// Mount "manage" controller
-	m := NewManageV1Controller(service, version, revision)
+	m := NewManageV1Controller(service, version, revision, restart)
 	app.MountManageV1Controller(service, m)
 
 	// Mount "tools" controller
@@ -107,12 +113,6 @@ func Start(opts Opts) {
 	public := NewPublicController(service)
 	app.MountPublicController(service, public)
 
-	// Mount systray
-	restart := restartFunc("", !opts.Hibernate)
-	shutdown := func() {
-		os.Exit(0)
-	}
-
 	// Find boards
 	http, https := findPorts()
 	if opts.HTTPPort != 0 {

cli/agent/main.go

@@ -38,14 +38,16 @@ type ManageV1Controller struct {
 	*goa.Controller
 	Version  string
 	Revision string
+	Restart  func()
 }
 
 // NewManageV1Controller creates a Manage_v1 controller.
-func NewManageV1Controller(service *goa.Service, version, revision string) *ManageV1Controller {
+func NewManageV1Controller(service *goa.Service, version, revision string, restart func()) *ManageV1Controller {
 	return &ManageV1Controller{
 		Controller: service.NewController("ManageV1Controller"),
 		Version:    version,
 		Revision:   revision,
+		Restart:    restart,
 	}
 }
 
@@ -56,3 +58,14 @@ func (c *ManageV1Controller) Info(ctx *app.InfoManageV1Context) error {
 		Revision: c.Revision,
 	})
 }
+
+// Pause runs the Pause action.
+func (c *ManageV1Controller) Pause(ctx *app.PauseManageV1Context) error {
+	c.Restart()
+	return ctx.OK(nil)
+}
+
+// Update runs the Update action.
+func (c *ManageV1Controller) Update(ctx *app.UpdateManageV1Context) error {
+	return ctx.OK(nil)
+}