add admin interface to zos-api-light

Author: Eslam-Nawara

pkg/zos_api_light/admin.go

@@ -4,6 +4,11 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/rs/zerolog/log"
+	"github.com/threefoldtech/zosbase/pkg/zinit"
 )
 
 func (g *ZosAPI) adminInterfacesHandler(ctx context.Context, payload []byte) (interface{}, error) {
@@ -44,5 +49,85 @@ func (g *ZosAPI) adminSetPublicNICHandler(ctx context.Context, payload []byte) (
 		return nil, fmt.Errorf("failed to decode input, expecting string: %w", err)
 	}
 	return nil, fmt.Errorf("not supported")
+}
+
+func (g *ZosAPI) adminRebootHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	zinit := zinit.Default()
+
+	return nil, zinit.Reboot()
+}
+
+func (g *ZosAPI) adminRestartServiceHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	var service string
+	if err := json.Unmarshal(payload, &service); err != nil {
+		return nil, fmt.Errorf("failed to decode input, expecting string: %w", err)
+	}
+
+	zinit := zinit.Default()
+
+	return nil, zinit.Restart(service)
+}
+
+func (g *ZosAPI) adminRestartAllHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	zinit := zinit.Default()
+
+	services, err := zinit.List()
+	if err != nil {
+		return nil, fmt.Errorf("failed to list node services, expecting string: %w", err)
+	}
+
+	for service := range services {
+		log.Debug().Str("service", service).Send()
+		if err := zinit.Restart(service); err != nil {
+			return nil, fmt.Errorf("failed to reboot service %s, expecting string: %w", service, err)
+		}
+	}
+
+	return nil, nil
+}
+
+func (g *ZosAPI) adminShowLogsHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	var n int
+	if err := json.Unmarshal(payload, &n); err != nil {
+		return nil, fmt.Errorf("failed to decode input, expecting string: %w", err)
+	}
+
+	zinit := zinit.Default()
+
+	return zinit.Log(n)
+}
+
+func (g *ZosAPI) adminShowResolveHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	path := filepath.Join("/etc", "resolv.conf")
+	return os.ReadFile(path)
+}
+
+func (g *ZosAPI) adminShowOpenConnectionsHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	return g.statisticsStub.OpenConnections(ctx)
+}
+
+func (g *ZosAPI) adminStopWorkloadHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	var args struct {
+		TwinID     uint32 `json:"twin_id"`
+		WorkloadID uint64 `json:"workload_id"`
+	}
+
+	if err := json.Unmarshal(payload, &args); err != nil {
+		return nil, fmt.Errorf("failed to decode input, expecting twin id and workload id: %w", err)
+	}
+
+	return nil, g.provisionStub.Pause(ctx, args.TwinID, args.WorkloadID)
+}
+
+func (g *ZosAPI) adminResumeWorkloadHandler(ctx context.Context, payload []byte) (interface{}, error) {
+	var args struct {
+		TwinID     uint32 `json:"twin_id"`
+		WorkloadID uint64 `json:"workload_id"`
+	}
+
+	if err := json.Unmarshal(payload, &args); err != nil {
+		return nil, fmt.Errorf("failed to decode input, expecting twin id and workload id: %w", err)
+	}
 
+	return nil, g.provisionStub.Resume(ctx, args.TwinID, args.WorkloadID)
 }

pkg/zos_api_light/routes.go

@@ -49,6 +49,17 @@ func (g *ZosAPI) SetupRoutes(router *peer.Router) {
 	admin.WithHandler("set_public_nic", g.adminSetPublicNICHandler)
 	admin.WithHandler("get_public_nic", g.adminGetPublicNICHandler)
 
+	admin.WithHandler("reboot", g.adminRebootHandler)
+	admin.WithHandler("restart", g.adminRestartServiceHandler)
+	admin.WithHandler("restart_all", g.adminRestartAllHandler)
+
+	admin.WithHandler("show_logs", g.adminShowLogsHandler)
+	admin.WithHandler("show_resolve", g.adminShowResolveHandler)
+	admin.WithHandler("show_open_connections", g.adminShowOpenConnectionsHandler)
+
+	admin.WithHandler("stop_workload", g.adminStopWorkloadHandler)
+	admin.WithHandler("resume_workload", g.adminResumeWorkloadHandler)
+
 	location := root.SubRoute("location")
 	location.WithHandler("get", g.locationGet)
 }