picoclaw/web/backend/app_runtime.go

package main

import (
	"context"
	"errors"
	"fmt"
	"time"

	"github.com/sipeed/picoclaw/pkg/logger"
	"github.com/sipeed/picoclaw/web/backend/utils"
)

const (
	browserDelay    = 500 * time.Millisecond
	shutdownTimeout = 15 * time.Second
)

// shutdownApp gracefully shuts down all server components and resources.
// It performs the following shutdown sequence:
//   - Shuts down the API handler to close all active SSE (Server-Sent Events) connections
//   - Disables HTTP keep-alive to prevent new connections during shutdown
//   - Attempts graceful HTTP server shutdown with timeout
//   - Logs shutdown status at appropriate log levels
//
// The function handles timeout errors gracefully by logging them at info level
// since context.DeadlineExceeded is expected when there are active long-running
// connections (such as SSE streams).
//
// This function should be called during application termination to ensure
// clean resource cleanup and proper connection closure.
func shutdownApp() {
	// First, shutdown API handler to close all SSE connections
	if apiHandler != nil {
		apiHandler.Shutdown()
	}

	if len(servers) > 0 {
		for _, srv := range servers {
			if srv == nil {
				continue
			}

			// Disable keep-alive to allow graceful shutdown
			srv.SetKeepAlivesEnabled(false)

			ctx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
			err := srv.Shutdown(ctx)
			cancel()

			if err != nil {
				// Context deadline exceeded is expected if there are active connections
				// This is not necessarily an error, so log it at info level
				if errors.Is(err, context.DeadlineExceeded) {
					logger.Infof("Server shutdown timeout after %v, forcing close", shutdownTimeout)
				} else {
					logger.Errorf("Server shutdown error: %v", err)
				}
			} else {
				logger.Infof("Server shutdown completed successfully")
			}
		}
	}
}

func openBrowser() error {
	target := browserLaunchURL
	if target == "" {
		target = serverAddr
	}
	if target == "" {
		return fmt.Errorf("server address not set")
	}
	return utils.OpenBrowser(target)
}