76 lines
1.3 KiB
Go
76 lines
1.3 KiB
Go
/*
|
|
* Copyright 2026 Oleg Borodin <onborodin@gmail.com>
|
|
*/
|
|
package starter
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"mproxy/app/server"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type Starter struct {
|
|
runAsDaemon bool
|
|
port uint32
|
|
cmd *cobra.Command
|
|
srv *server.Server
|
|
}
|
|
|
|
func NewStarter() *Starter {
|
|
execName := filepath.Base(os.Args[0])
|
|
sta := &Starter{}
|
|
cmd := &cobra.Command{
|
|
Use: execName,
|
|
Short: "\nArtifact storage service",
|
|
SilenceUsage: true,
|
|
RunE: sta.run,
|
|
}
|
|
cmd.CompletionOptions.DisableDefaultCmd = true
|
|
cmd.Flags().BoolVarP(&sta.runAsDaemon, "asDaemon", "D", true, "Run service as daemon")
|
|
cmd.Flags().Uint32VarP(&sta.port, "port", "P", 1025, "Service port")
|
|
|
|
sta.cmd = cmd
|
|
return sta
|
|
}
|
|
|
|
func (sta *Starter) GetCmd() *cobra.Command {
|
|
return sta.cmd
|
|
}
|
|
|
|
func (sta *Starter) run(cmd *cobra.Command, args []string) error {
|
|
var err error
|
|
srv, err := server.NewServer()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = srv.Configure()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
srv.SetAsDaemon(sta.runAsDaemon)
|
|
srv.SetPort(sta.port)
|
|
|
|
err = srv.Daemonize()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = srv.Build()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = srv.Run()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
sta.srv = srv
|
|
return err
|
|
}
|
|
|
|
func (sta *Starter) Exec() error {
|
|
return sta.cmd.Execute()
|
|
}
|