nfsense/internal/api/object/services.go

package object

import (
	"context"
	"fmt"

	"nfsense.net/nfsense/internal/definitions/object"
)

type GetServiceParameters struct {
	ID string
}

type GetServiceResult struct {
	Name string `json:"name"`
	object.Service
}

func (f *Object) GetService(ctx context.Context, params GetServiceParameters) (GetServiceResult, error) {
	_, ok := f.ConfigManager.GetPendingConfig().Object.Services[params.ID]
	if !ok {
		return GetServiceResult{}, fmt.Errorf("Service does not Exist")
	}

	return GetServiceResult{
		Name:    params.ID,
		Service: f.ConfigManager.GetPendingConfig().Object.Services[params.ID],
	}, nil
}

type GetServicesResult struct {
	Services map[string]object.Service
}

func (f *Object) GetServices(ctx context.Context, params struct{}) (GetServicesResult, error) {
	return GetServicesResult{
		Services: f.ConfigManager.GetPendingConfig().Object.Services,
	}, nil
}

type CreateServiceParameters struct {
	Name string
	object.Service
}

func (f *Object) CreateService(ctx context.Context, params CreateServiceParameters) (struct{}, error) {
	_, ok := f.ConfigManager.GetPendingConfig().Object.Services[params.Name]
	if ok {
		return struct{}{}, fmt.Errorf("Service already Exists")
	}

	t, conf := f.ConfigManager.StartTransaction()
	defer t.Discard()

	conf.Object.Services[params.Name] = params.Service
	return struct{}{}, t.Commit()
}

type UpdateServiceParameters struct {
	Name string
	object.Service
}

func (f *Object) UpdateService(ctx context.Context, params UpdateServiceParameters) (struct{}, error) {
	_, ok := f.ConfigManager.GetPendingConfig().Object.Services[params.Name]
	if !ok {
		return struct{}{}, fmt.Errorf("Service does not Exist")
	}

	t, conf := f.ConfigManager.StartTransaction()
	defer t.Discard()

	conf.Object.Services[params.Name] = params.Service
	return struct{}{}, t.Commit()
}

type DeleteServiceParameters struct {
	Name string
}

func (f *Object) DeleteService(ctx context.Context, params DeleteServiceParameters) (struct{}, error) {
	_, ok := f.ConfigManager.GetPendingConfig().Object.Services[params.Name]
	if !ok {
		return struct{}{}, fmt.Errorf("Interface does not Exist")
	}

	t, conf := f.ConfigManager.StartTransaction()
	defer t.Discard()

	delete(conf.Object.Services, params.Name)
	return struct{}{}, t.Commit()
}