Add toolkit command for Vespa CLI with server implementations

client/go/internal/cli/cmd/root.go

@@ -275,6 +275,7 @@ func (c *CLI) configureCommands() {
 	rootCmd.AddCommand(newVersionCmd(c))            // version
 	rootCmd.AddCommand(newVisitCmd(c))              // visit
 	rootCmd.AddCommand(newFeedCmd(c))               // feed
+	rootCmd.AddCommand(newToolkitCmd(c))            // toolkit
 }
 
 func (c *CLI) bindWaitFlag(cmd *cobra.Command, defaultSecs int, value *int) {

client/go/internal/cli/cmd/toolkit.go

@@ -0,0 +1,107 @@
+package cmd
+
+import (
+	"bufio"
+	"context"
+	"errors"
+	"fmt"
+	"github.com/spf13/cobra"
+	"github.com/vespa-engine/vespa/client/go/internal/toolkit"
+	"github.com/vespa-engine/vespa/client/go/internal/vespa"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"strings"
+	"sync"
+	"time"
+)
+
+func newToolkitCmd(cli *CLI) *cobra.Command {
+	return &cobra.Command{
+		Use:   "toolkit",
+		Short: "Starts the Vespa toolkit servers",
+		Long: `The 'vespa toolkit' command initializes and starts both the frontend and proxy servers. 
+The frontend server serves the Vespa toolkit's user interface, while the proxy server acts 
+as a reverse proxy to the Vespa service, especially for cloud instances. This toolkit provides 
+a convenient interface for interacting with the Vespa service.`,
+		DisableAutoGenTag: true,
+		SilenceUsage:      true,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			fmt.Println("This command will start the Vespa toolkit which includes frontend and proxy servers.")
+			fmt.Print("Do you want to continue? (Y/n): ")
+			reader := bufio.NewReader(os.Stdin)
+			response, err := reader.ReadString('\n')
+			if err != nil {
+				return fmt.Errorf("failed to read user input: %w", err)
+			}
+			response = strings.ToLower(strings.TrimSpace(response))
+
+			// Treat empty response as "yes" since "Y" is the default
+			if response == "" || response == "y" || response == "yes" {
+				target, err := cli.target(targetOptions{})
+				if err != nil {
+					return err
+				}
+				service, err := target.Service(vespa.QueryService, time.Duration(0)*time.Second, 0, cli.config.cluster())
+				if err != nil {
+					return err
+				}
+
+				var wg sync.WaitGroup
+
+				// Initialize and start the FrontendServer
+				var frontendServer *toolkit.FrontendServer
+				frontendServer = toolkit.NewFrontendServer()
+				wg.Add(1)
+				go func() {
+					defer wg.Done()
+					fmt.Printf("Frontend server is running on http://localhost:%d\n", frontendServer.Port)
+					if err := frontendServer.Start(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+						log.Printf("Frontend server stopped with error: %v\n", err)
+					}
+				}()
+
+				// Only initialize and start the ProxyServer if target is cloud
+				var proxyServer *toolkit.ProxyServer
+				proxyServer, err = toolkit.NewProxyServer(service, target)
+				if err != nil {
+					return fmt.Errorf("could not initialize proxy server: %w", err)
+				}
+				wg.Add(1)
+				go func() {
+					defer wg.Done()
+					fmt.Printf("Proxy server is running on http://localhost:%d\n", proxyServer.Port)
+					if err := proxyServer.Start(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+						log.Printf("Proxy server stopped with error: %v\n", err)
+					}
+				}()
+
+				// Signal listener for graceful shutdown
+				c := make(chan os.Signal, 1)
+				signal.Notify(c, os.Interrupt)
+				go func() {
+					<-c
+					ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+					defer cancel()
+					// Stop frontend server
+					if err := frontendServer.Shutdown(ctx); err != nil {
+						log.Printf("Error stopping frontend server: %v\n", err)
+					}
+					// Stop proxy server
+					if err := proxyServer.Shutdown(ctx); err != nil {
+						log.Printf("Error stopping proxy server: %v\n", err)
+					}
+					// Decrement the wait group for each server to allow the main goroutine to exit
+					wg.Done()
+					wg.Done()
+				}()
+
+				wg.Wait()
+				return nil
+			} else {
+				return nil // Exit if the user chooses "no"
+			}
+		},
+	}
+}