Merge pull request #8 from pelican-dev/push_notifications

Push notification to panel on container state change

remote/http.go

@@ -33,6 +33,7 @@ type Client interface {
 	SetTransferStatus(ctx context.Context, uuid string, successful bool) error
 	ValidateSftpCredentials(ctx context.Context, request SftpAuthRequest) (SftpAuthResponse, error)
 	SendActivityLogs(ctx context.Context, activity []models.Activity) error
+	PushServerStateChange(ctx context.Context, stateChange ServerStateChange) error
 }
 
 type client struct {

remote/servers.go

@@ -211,3 +211,13 @@ func (c *client) getServersPaged(ctx context.Context, page, limit int) ([]RawSer
 	}
 	return r.Data, r.Meta, nil
 }
+
+// PushServerStateChange updates the Panel with state change notifications
+func (c *client) PushServerStateChange(ctx context.Context, sc ServerStateChange) error {
+	resp, err := c.Post(ctx, fmt.Sprintf("/api/remote/servers/%s/container/status", sc.ServerUuid), d{"data": sc})
+	if err != nil {
+		return errors.WithStackIf(err)
+	}
+	_ = resp.Body.Close()
+	return nil
+}

remote/types.go

@@ -174,3 +174,9 @@ type InstallStatusRequest struct {
 	Successful bool `json:"successful"`
 	Reinstall  bool `json:"reinstall"`
 }
+
+type ServerStateChange struct {
+	PrevState  string `json:"previous_state"`
+	NewState   string `json:"new_state"`
+	ServerUuid string `json:"server_uuid"`
+}

server/server.go

@@ -323,6 +323,10 @@ func (s *Server) OnStateChange() {
 		s.Events().Publish(StatusEvent, st)
 	}
 
+	// Push status update to Panel
+	s.Log().Debug("pushing server status change to panel")
+	s.client.PushServerStateChange(s.Context(), remote.ServerStateChange{prevState, st, s.ID()})
+
 	// Reset the resource usage to 0 when the process fully stops so that all the UI
 	// views in the Panel correctly display 0.
 	if st == environment.ProcessOfflineState {