http_server.go

package thousandeyes

import (
	"encoding/json"
	"fmt"
)

// HTTPServerResponse - a http server response
type HTTPServerResponse struct {
	tests []HTTPServer
}

// CustomHeaders represents the JSON object exchanged for specifying
// custom HTTP headers in HTTP Server, Page Load, and Web Transaction tests.
type CustomHeaders struct {
	Root    *map[string]string            `json:"root,omitempty"`
	All     *map[string]string            `json:"all,omitempty"`
	Domains *map[string]map[string]string `json:"domains,omitempty"`
}

// HTTPServer - a http server test
type HTTPServer struct {
	// Common test fields
	AlertsEnabled      *bool                `json:"alertsEnabled,omitempty" te:"int-bool"`
	AlertRules         *[]AlertRule         `json:"alertRules,omitempty"`
	APILinks           *[]APILink           `json:"apiLinks,omitempty"`
	CreatedBy          *string              `json:"createdBy,omitempty"`
	CreatedDate        *string              `json:"createdDate,omitempty"`
	Description        *string              `json:"description,omitempty"`
	Enabled            *bool                `json:"enabled,omitempty" te:"int-bool"`
	Groups             *[]GroupLabel        `json:"groups,omitempty"`
	ModifiedBy         *string              `json:"modifiedBy,omitempty"`
	ModifiedDate       *string              `json:"modifiedDate,omitempty"`
	SavedEvent         *bool                `json:"savedEvent,omitempty" te:"int-bool"`
	SharedWithAccounts *[]SharedWithAccount `json:"sharedWithAccounts,omitempty"`
	TestID             *int64               `json:"testId,omitempty"`
	TestName           *string              `json:"testName,omitempty"`
	Type               *string              `json:"type,omitempty"`
	LiveShare          *bool                `json:"liveShare,omitempty" te:"int-bool"`

	// Fields unique to this test
	Agents                *[]Agent       `json:"agents,omitempty"`
	AuthType              *string        `json:"authType,omitempty"`
	BandwidthMeasurements *bool          `json:"bandwidthMeasurements,omitempty" te:"int-bool"`
	BGPMeasurements       *bool          `json:"bgpMeasurements,omitempty" te:"int-bool"`
	BGPMonitors           *[]Monitor     `json:"bgpMonitors,omitempty"`
	ClientCertificate     *string        `json:"clientCertificate,omitempty"`
	ContentRegex          *string        `json:"contentRegex,omitempty"`
	CustomHeaders         *CustomHeaders `json:"customHeaders,omitempty"`
	DesiredStatusCode     *string        `json:"desiredStatusCode,omitempty"`
	DownloadLimit         *int64         `json:"downloadLimit,omitempty"`
	DNSOverride           *string        `json:"dnsOverride,omitempty"`
	FollowRedirects       *bool          `json:"followRedirects,omitempty" te:"int-bool"`
	Headers               *[]string      `json:"headers,omitempty"`
	HTTPVersion           *int           `json:"httpVersion,omitempty"`
	HTTPTargetTime        *int           `json:"httpTargetTime,omitempty"`
	HTTPTimeLimit         *int           `json:"httpTimeLimit,omitempty"`
	Interval              *int           `json:"interval,omitempty"`
	MTUMeasurements       *bool          `json:"mtuMeasurements,omitempty" te:"int-bool"`
	NetworkMeasurements   *bool          `json:"networkMeasurements,omitempty" te:"int-bool"`
	NumPathTraces         *int           `json:"numPathTraces,omitempty"`
	Password              *string        `json:"password,omitempty"`
	PathTraceMode         *string        `json:"pathTraceMode,omitempty"`
	PostBody              *string        `json:"postBody,omitempty"`
	ProbeMode             *string        `json:"probeMode,omitempty"`
	Protocol              *string        `json:"protocol,omitempty"`
	SSLVersion            *string        `json:"sslVersion,omitempty"`
	SSLVersionID          *int64         `json:"sslVersionId,omitempty"`
	URL                   *string        `json:"url,omitempty"`
	UseNTLM               *bool          `json:"useNtlm,omitempty" te:"int-bool"` // TODO: BasicAuth not working
	UserAgent             *string        `json:"userAgent,omitempty"`
	Username              *string        `json:"username,omitempty"`
	VerifyCertificate     *bool          `json:"verifyCertificate,omitempty" te:"int-bool"`
}

// MarshalJSON implements the json.Marshaler interface. It ensures
// that ThousandEyes int fields that only use the values 0 or 1 are
// treated as booleans.
func (t HTTPServer) MarshalJSON() ([]byte, error) {
	type aliasTest HTTPServer

	data, err := json.Marshal((aliasTest)(t))
	if err != nil {
		return nil, err
	}

	return jsonBoolToInt(&t, data)
}

// UnmarshalJSON implements the json.Unmarshaler interface. It ensures
// that ThousandEyes int fields that only use the values 0 or 1 are
// treated as booleans.
func (t *HTTPServer) UnmarshalJSON(data []byte) error {
	type aliasTest HTTPServer
	test := (*aliasTest)(t)

	data, err := jsonIntToBool(t, data)
	if err != nil {
		return err
	}

	return json.Unmarshal(data, &test)
}

// AddAgent - add an agent
func (t *HTTPServer) AddAgent(id int64) {
	agent := Agent{AgentID: Int64(id)}
	*t.Agents = append(*t.Agents, agent)
}

// GetHTTPServer - Get an HTTP Server test
func (c *Client) GetHTTPServer(id int64) (*HTTPServer, error) {
	resp, err := c.get(fmt.Sprintf("/tests/%d", id))
	if err != nil {
		return &HTTPServer{}, err
	}
	var target map[string][]HTTPServer
	if dErr := c.decodeJSON(resp, &target); dErr != nil {
		return nil, fmt.Errorf("Could not decode JSON response: %v", dErr)
	}
	return &target["test"][0], nil
}

// CreateHTTPServer - create a http server
func (c Client) CreateHTTPServer(t HTTPServer) (*HTTPServer, error) {
	resp, err := c.post("/tests/http-server/new", t, nil)
	if err != nil {
		return &t, err
	}
	if resp.StatusCode != 201 {
		return &t, fmt.Errorf("failed to create http server, response code %d", resp.StatusCode)
	}
	var target map[string][]HTTPServer
	if dErr := c.decodeJSON(resp, &target); dErr != nil {
		return nil, fmt.Errorf("Could not decode JSON response: %v", dErr)
	}
	return &target["test"][0], nil
}

// DeleteHTTPServer - delete an http server
func (c *Client) DeleteHTTPServer(id int64) error {
	resp, err := c.post(fmt.Sprintf("/tests/http-server/%d/delete", id), nil, nil)
	if err != nil {
		return err
	}
	if resp.StatusCode != 204 {
		return fmt.Errorf("failed to delete http server, response code %d", resp.StatusCode)
	}
	return nil
}

// UpdateHTTPServer - Update an http server test
func (c *Client) UpdateHTTPServer(id int64, t HTTPServer) (*HTTPServer, error) {
	resp, err := c.post(fmt.Sprintf("/tests/http-server/%d/update", id), t, nil)
	if err != nil {
		return &t, err
	}
	if resp.StatusCode != 200 {
		return &t, fmt.Errorf("failed to update http server, response code %d", resp.StatusCode)
	}
	var target map[string][]HTTPServer
	if dErr := c.decodeJSON(resp, &target); dErr != nil {
		return nil, fmt.Errorf("Could not decode JSON response: %v", dErr)
	}
	return &target["test"][0], nil
}