mirror of
https://github.com/Control-D-Inc/ctrld.git
synced 2026-02-03 22:18:39 +00:00
6e10bba7fe
Create separate file for service command handling to improve code organization. Add ServiceCommand struct with Install, Uninstall, Start, Stop, and Status methods to handle service operations with proper error handling and dependency injection.
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/kardianos/service"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// ServiceCommand handles service-related operations
|
|
type ServiceCommand struct {
|
|
serviceManager *ServiceManager
|
|
}
|
|
|
|
// NewServiceCommand creates a new service command handler
|
|
func NewServiceCommand() (*ServiceCommand, error) {
|
|
sm, err := NewServiceManager()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &ServiceCommand{
|
|
serviceManager: sm,
|
|
}, nil
|
|
}
|
|
|
|
// createServiceConfig creates a properly initialized service configuration
|
|
func (sc *ServiceCommand) createServiceConfig() *service.Config {
|
|
return &service.Config{
|
|
Name: ctrldServiceName,
|
|
DisplayName: "Control-D Helper Service",
|
|
Description: "A highly configurable, multi-protocol DNS forwarding proxy",
|
|
Option: service.KeyValue{},
|
|
}
|
|
}
|
|
|
|
// Install installs the service
|
|
func (sc *ServiceCommand) Install(cmd *cobra.Command, args []string) error {
|
|
svcConfig := sc.createServiceConfig()
|
|
|
|
// Set the working directory to the executable's directory
|
|
if exe, err := os.Executable(); err == nil {
|
|
svcConfig.WorkingDirectory = filepath.Dir(exe)
|
|
}
|
|
|
|
if err := sc.serviceManager.svc.Install(); err != nil {
|
|
return fmt.Errorf("failed to install service: %w", err)
|
|
}
|
|
|
|
mainLog.Load().Notice().Msg("Service installed successfully")
|
|
return nil
|
|
}
|
|
|
|
// Uninstall uninstalls the service
|
|
func (sc *ServiceCommand) Uninstall(cmd *cobra.Command, args []string) error {
|
|
if err := sc.serviceManager.svc.Uninstall(); err != nil {
|
|
return fmt.Errorf("failed to uninstall service: %w", err)
|
|
}
|
|
|
|
mainLog.Load().Notice().Msg("Service uninstalled successfully")
|
|
return nil
|
|
}
|
|
|
|
// Start starts the service
|
|
func (sc *ServiceCommand) Start(cmd *cobra.Command, args []string) error {
|
|
if err := sc.serviceManager.svc.Start(); err != nil {
|
|
return fmt.Errorf("failed to start service: %w", err)
|
|
}
|
|
|
|
mainLog.Load().Notice().Msg("Service started successfully")
|
|
return nil
|
|
}
|
|
|
|
// Stop stops the service
|
|
func (sc *ServiceCommand) Stop(cmd *cobra.Command, args []string) error {
|
|
if err := sc.serviceManager.svc.Stop(); err != nil {
|
|
return fmt.Errorf("failed to stop service: %w", err)
|
|
}
|
|
|
|
mainLog.Load().Notice().Msg("Service stopped successfully")
|
|
return nil
|
|
}
|
|
|
|
// Status returns the service status
|
|
func (sc *ServiceCommand) Status(cmd *cobra.Command, args []string) error {
|
|
status, err := sc.serviceManager.Status()
|
|
if err != nil {
|
|
if err == service.ErrNotInstalled {
|
|
mainLog.Load().Warn().Msg("Service not installed")
|
|
return nil
|
|
}
|
|
return fmt.Errorf("failed to get service status: %w", err)
|
|
}
|
|
|
|
switch status {
|
|
case service.StatusRunning:
|
|
mainLog.Load().Notice().Msg("Service is running")
|
|
case service.StatusStopped:
|
|
mainLog.Load().Warn().Msg("Service is stopped")
|
|
default:
|
|
mainLog.Load().Warn().Msgf("Service status: %v", status)
|
|
}
|
|
|
|
return nil
|
|
}
|