Add service restart

This commit is contained in:
problematicconsumer
2023-09-10 20:18:33 +03:30
parent 610747f9ad
commit c887815676
4 changed files with 113 additions and 30 deletions

View File

@@ -7,7 +7,16 @@ var commandServer *libbox.CommandServer
type CommandServerHandler struct{}
func (csh *CommandServerHandler) ServiceReload() error {
return nil
propagateStatus(Starting)
if commandServer != nil {
commandServer.SetService(nil)
commandServer = nil
}
if box != nil {
box.Close()
box = nil
}
return startService()
}
func (csh *CommandServerHandler) GetSystemProxyStatus() *libbox.SystemProxyStatus {

15
custom/constant.go Normal file
View File

@@ -0,0 +1,15 @@
package main
const (
Stopped = "Stopped"
Starting = "Starting"
Started = "Started"
Stopping = "Stopping"
)
const (
EmptyConfiguration = "EmptyConfiguration"
StartCommandServer = "StartCommandServer"
CreateService = "CreateService"
StartService = "StartService"
)

View File

@@ -16,6 +16,7 @@ import (
var box *libbox.BoxService
var configOptions *shared.ConfigOptions
var activeConfigPath *string
//export setupOnce
func setupOnce(api unsafe.Pointer) {
@@ -23,8 +24,9 @@ func setupOnce(api unsafe.Pointer) {
}
//export setup
func setup(baseDir *C.char, workingDir *C.char, tempDir *C.char) {
func setup(baseDir *C.char, workingDir *C.char, tempDir *C.char, statusPort C.longlong) {
Setup(C.GoString(baseDir), C.GoString(workingDir), C.GoString(tempDir))
statusPropagationPort = int64(statusPort)
}
//export parse
@@ -46,16 +48,30 @@ func changeConfigOptions(configOptionsJson *C.char) *C.char {
return C.CString("")
}
//export create
func create(configPath *C.char) *C.char {
//export start
func start(configPath *C.char) *C.char {
if status != Stopped {
return C.CString("")
}
propagateStatus(Starting)
path := C.GoString(configPath)
content, err := os.ReadFile(path)
activeConfigPath = &path
err := startService()
if err != nil {
return C.CString(err.Error())
}
return C.CString("")
}
func startService() error {
content, err := os.ReadFile(*activeConfigPath)
if err != nil {
return stopAndAlert(EmptyConfiguration, err)
}
options, err := parseConfig(string(content))
if err != nil {
return C.CString(err.Error())
return stopAndAlert(EmptyConfiguration, err)
}
options = shared.BuildConfig(*configOptions, options)
@@ -63,45 +79,33 @@ func create(configPath *C.char) *C.char {
err = startCommandServer()
if err != nil {
return C.CString(err.Error())
return stopAndAlert(StartCommandServer, err)
}
instance, err := NewService(options)
if err != nil {
return C.CString(err.Error())
return stopAndAlert(CreateService, err)
}
err = instance.Start()
if err != nil {
return stopAndAlert(StartService, err)
}
box = instance
commandServer.SetService(box)
if err != nil {
instance.Close()
box = nil
return C.CString(err.Error())
}
return C.CString("")
}
//export start
func start() *C.char {
if box == nil {
return C.CString("instance not found")
}
err := box.Start()
if err != nil {
return C.CString(err.Error())
}
return C.CString("")
propagateStatus(Started)
return nil
}
//export stop
func stop() *C.char {
if status != Started {
return C.CString("")
}
if box == nil {
return C.CString("instance not found")
}
propagateStatus(Stopping)
commandServer.SetService(nil)
err := box.Close()
@@ -115,6 +119,28 @@ func stop() *C.char {
return C.CString(err.Error())
}
commandServer = nil
propagateStatus(Stopped)
return C.CString("")
}
//export restart
func restart(configPath *C.char) *C.char {
if status != Started {
return C.CString("")
}
if box == nil {
return C.CString("instance not found")
}
err := stop()
if C.GoString(err) != "" {
return err
}
err = start(configPath)
if C.GoString(err) != "" {
return err
}
return C.CString("")
}

33
custom/status.go Normal file
View File

@@ -0,0 +1,33 @@
package main
import "C"
import (
"encoding/json"
"github.com/hiddify/libcore/bridge"
)
var statusPropagationPort int64
var status = Stopped
type StatusMessage struct {
Status string `json:"status"`
Alert *string `json:"alert"`
Message *string `json:"message"`
}
func propagateStatus(newStatus string) {
status = newStatus
msg, _ := json.Marshal(StatusMessage{Status: status})
bridge.SendStringToPort(statusPropagationPort, string(msg))
}
func stopAndAlert(alert string, err error) error {
status = Stopped
message := err.Error()
msg, _ := json.Marshal(StatusMessage{Status: status, Alert: &alert, Message: &message})
bridge.SendStringToPort(statusPropagationPort, string(msg))
return err
}