mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-02 04:20:50 +08:00
41 lines
1.2 KiB
Go
41 lines
1.2 KiB
Go
// Package cmd provides command-line interface functionality for the CLI Proxy API server.
|
|
// It includes authentication flows for various AI service providers, service startup,
|
|
// and other command-line operations.
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/config"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// StartService builds and runs the proxy service using the exported SDK.
|
|
// It creates a new proxy service instance, sets up signal handling for graceful shutdown,
|
|
// and starts the service with the provided configuration.
|
|
//
|
|
// Parameters:
|
|
// - cfg: The application configuration
|
|
// - configPath: The path to the configuration file
|
|
func StartService(cfg *config.Config, configPath string) {
|
|
service, err := cliproxy.NewBuilder().
|
|
WithConfig(cfg).
|
|
WithConfigPath(configPath).
|
|
Build()
|
|
if err != nil {
|
|
log.Fatalf("failed to build proxy service: %v", err)
|
|
}
|
|
|
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
err = service.Run(ctx)
|
|
if err != nil && !errors.Is(err, context.Canceled) {
|
|
log.Fatalf("proxy service exited with error: %v", err)
|
|
}
|
|
}
|