Add notes/todos

mafredri · mafredri · commit 9fb2c7ba2e2c · 2023-02-13T13:04:54.000Z
diff --git a/agent/agent.go b/agent/agent.go
@@ -1223,9 +1223,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)
@@ -1264,6 +1267,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
