Add notes/todos

mafredri · mafredri · commit ee7179ddd120 · 2023-02-10T09:31:14.000Z
diff --git a/agent/agent.go b/agent/agent.go
@@ -1213,9 +1213,12 @@ func (a *agent) Close() error {
 
 	// Close services before running shutdown script.
 	// TODO(mafredri): Gracefully shutdown:
+	// - Cancel startup script, if running
 	// - Close active SSH server connections
 	// - Close processes (send HUP, wait, etc.)
 
+	// TODO(mafredri): Only run shutdown script if the agent is 'ready'?
+
 	lifecycleState := codersdk.WorkspaceAgentLifecycleOff
 	if metadata, ok := a.metadata.Load().(agentsdk.Metadata); ok {
 		scriptDone := make(chan error, 1)
@@ -1254,6 +1257,7 @@ func (a *agent) Close() error {
 	// Set final state and wait for it to be reported because context
 	// cancellation will stop the report loop.
 	a.setLifecycle(ctx, lifecycleState)
+	// TODO(mafredri): What if the agent token is revoked, build outdated, etc.?
 	for s := range a.lifecycleReported {
 		if s == lifecycleState {
 			break
