exploitationservice.go

package goitop

import (
	"fmt"
)

type ExploitationService struct {
	Name string
	ID   string
}

func (c *Client) GetExploitationService(name string) (string, error) {

	payload := map[string]interface{}{
		"operation": "core/get",
		"class":     "ExploitationService",
		"key": map[string]string{
			"name": name,
		},
		"output_fields": "id",
	}

	result, err := Request(c, payload)
	if err != nil {
		return "", err
	}

	if result.Code != 0 {
		return "", fmt.Errorf("Get exploitation service request failed with code %d: %s", result.Code, result.Message)
	}

	if len(result.Objects) > 1 {
		return "", fmt.Errorf("Too many objects in get exploitation service response")
	}

	if len(result.Objects) < 1 {
		return "", fmt.Errorf("Exploitation service not found")
	}

	var firstObject APIObject
	for _, object := range result.Objects {
		firstObject = object
		break
	}

	return firstObject.Fields["id"].(string), nil
}

func (c *Client) GetAllExploitationService() ([]ExploitationService, error) {

	payload := map[string]interface{}{
		"operation":     "core/get",
		"class":         "ExploitationService",
		"key":           "SELECT ExploitationService",
		"output_fields": "id,name",
	}

	result, err := Request(c, payload)
	if err != nil {
		return []ExploitationService{}, err
	}

	if result.Code != 0 {
		return []ExploitationService{}, fmt.Errorf("Get all exploitation service request failed with code %d: %s", result.Code, result.Message)
	}

	if len(result.Objects) < 1 {
		return []ExploitationService{}, fmt.Errorf("No exploitation service found")
	}

	var services []ExploitationService
	for _, object := range result.Objects {
		services = append(services, ExploitationService{
			Name: object.Fields["name"].(string),
			ID:   object.Fields["id"].(string),
		})
	}

	return services, nil
}