mirror of
https://github.com/EnumeratedDev/enit.git
synced 2026-09-26 15:36:12 +00:00
Compare commits
3
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2b4feb60ec
|
||
|
|
085bc676a0
|
||
|
|
d83c822c73
|
+1
-1
@@ -128,7 +128,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Get service stage
|
// Get service stage
|
||||||
stage := 2
|
stage := 3
|
||||||
if len(flag.Args()) > 3 {
|
if len(flag.Args()) > 3 {
|
||||||
flagStr := flag.Arg(3)
|
flagStr := flag.Arg(3)
|
||||||
_stage, err := strconv.ParseInt(flagStr, 10, 32)
|
_stage, err := strconv.ParseInt(flagStr, 10, 32)
|
||||||
|
|||||||
+1
-1
@@ -211,7 +211,7 @@ func Init() {
|
|||||||
// Start enabled services
|
// Start enabled services
|
||||||
stages := slices.Collect(maps.Keys(EnabledServices))
|
stages := slices.Collect(maps.Keys(EnabledServices))
|
||||||
slices.Sort(stages)
|
slices.Sort(stages)
|
||||||
for stage := 0; stage <= stages[len(stages)-1]; stage++ {
|
for stage := 1; stage <= stages[len(stages)-1]; stage++ {
|
||||||
logger.Printf("Starting stage %d services...", stage)
|
logger.Printf("Starting stage %d services...", stage)
|
||||||
|
|
||||||
services := EnabledServices[stage]
|
services := EnabledServices[stage]
|
||||||
|
|||||||
+4
-1
@@ -346,14 +346,17 @@ func (service *EnitService) SetEnabled(stage int) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Remove service from current stage
|
// Remove service from current stage
|
||||||
if s != 0 {
|
|
||||||
EnabledServices[s] = slices.DeleteFunc(EnabledServices[s], func(name string) bool {
|
EnabledServices[s] = slices.DeleteFunc(EnabledServices[s], func(name string) bool {
|
||||||
return name == service.Name
|
return name == service.Name
|
||||||
})
|
})
|
||||||
|
if len(EnabledServices[s]) == 0 {
|
||||||
|
delete(EnabledServices, s)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Add service to stage
|
// Add service to stage
|
||||||
|
if stage != 0 {
|
||||||
EnabledServices[stage] = append(EnabledServices[stage], service.Name)
|
EnabledServices[stage] = append(EnabledServices[stage], service.Name)
|
||||||
|
}
|
||||||
|
|
||||||
// Save enabled services to file
|
// Save enabled services to file
|
||||||
data, err := yaml.Marshal(EnabledServices)
|
data, err := yaml.Marshal(EnabledServices)
|
||||||
|
|||||||
@@ -197,7 +197,12 @@ func handleSetEnabledServiceCommand(conn net.Conn, jsonData map[string]any) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if serviceStage == 0 {
|
||||||
|
conn.Write(wrapSuccessMsgInJson(fmt.Sprintf("Service (%s) was disabled sucessfully", serviceName.(string))))
|
||||||
|
} else {
|
||||||
conn.Write(wrapSuccessMsgInJson(fmt.Sprintf("Service (%s) was enabled sucessfully", serviceName.(string))))
|
conn.Write(wrapSuccessMsgInJson(fmt.Sprintf("Service (%s) was enabled sucessfully", serviceName.(string))))
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleStatusServiceCommand(conn net.Conn, jsonData map[string]any) {
|
func handleStatusServiceCommand(conn net.Conn, jsonData map[string]any) {
|
||||||
|
|||||||
Reference in New Issue
Block a user