93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
package service
|
|
|
|
import (
|
|
"acc-server-manager/local/model"
|
|
"acc-server-manager/local/repository"
|
|
"acc-server-manager/local/utl/common"
|
|
"errors"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
type ApiService struct {
|
|
repository *repository.ApiRepository
|
|
serverRepository *repository.ServerRepository
|
|
}
|
|
|
|
func NewApiService(repository *repository.ApiRepository,
|
|
serverRepository *repository.ServerRepository) *ApiService {
|
|
return &ApiService{
|
|
repository: repository,
|
|
serverRepository: serverRepository,
|
|
}
|
|
}
|
|
|
|
// GetFirst
|
|
// Gets first row from API table.
|
|
//
|
|
// Args:
|
|
// context.Context: Application context
|
|
// Returns:
|
|
// string: Application version
|
|
func (as ApiService) GetFirst(ctx *fiber.Ctx) *model.ApiModel {
|
|
return as.repository.GetFirst(ctx.UserContext())
|
|
}
|
|
|
|
func (as ApiService) GetStatus(ctx *fiber.Ctx) (string, error) {
|
|
service := ctx.Params("service")
|
|
return as.StatusServer(ctx, service)
|
|
}
|
|
|
|
func (as ApiService) ApiStartServer(ctx *fiber.Ctx) (string, error) {
|
|
service, ok := ctx.Locals("service").(string)
|
|
if !ok {
|
|
return "", errors.New("service name missing")
|
|
}
|
|
return as.StartServer(ctx, service)
|
|
}
|
|
|
|
func (as ApiService) ApiStopServer(ctx *fiber.Ctx) (string, error) {
|
|
service, ok := ctx.Locals("service").(string)
|
|
if !ok {
|
|
return "", errors.New("service name missing")
|
|
}
|
|
return as.StopServer(ctx, service)
|
|
}
|
|
|
|
func (as ApiService) ApiRestartServer(ctx *fiber.Ctx) (string, error) {
|
|
service, ok := ctx.Locals("service").(string)
|
|
if !ok {
|
|
return "", errors.New("service name missing")
|
|
}
|
|
return as.RestartServer(ctx, service)
|
|
}
|
|
|
|
func (as ApiService) StatusServer(ctx *fiber.Ctx, serviceName string) (string, error) {
|
|
return as.ManageService(ctx, "status", serviceName)
|
|
}
|
|
|
|
func (as ApiService) StartServer(ctx *fiber.Ctx, serviceName string) (string, error) {
|
|
return as.ManageService(ctx, "start", serviceName)
|
|
}
|
|
|
|
func (as ApiService) StopServer(ctx *fiber.Ctx, serviceName string) (string, error) {
|
|
return as.ManageService(ctx, "stop", serviceName)
|
|
}
|
|
|
|
func (as ApiService) RestartServer(ctx *fiber.Ctx, serviceName string) (string, error) {
|
|
return as.ManageService(ctx, "restart", serviceName)
|
|
}
|
|
|
|
func (as ApiService) ManageService(ctx *fiber.Ctx, action string, serviceName string) (string, error) {
|
|
server := as.serverRepository.GetFirstByServiceName(ctx.UserContext(), serviceName)
|
|
if server == nil {
|
|
return "", fiber.NewError(404, "Server not found")
|
|
}
|
|
|
|
output, err := common.RunElevatedCommand(action, serviceName)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return output, nil
|
|
}
|