http.go

package resource

import (
	"context"
	"fmt"
	"time"

	"github.com/goss-org/goss/system"
	"github.com/goss-org/goss/util"
)

type HTTP struct {
	Title              string            `json:"title,omitempty" yaml:"title,omitempty"`
	Meta               meta              `json:"meta,omitempty" yaml:"meta,omitempty"`
	id                 string            `json:"-" yaml:"-"`
	URL                string            `json:"url,omitempty" yaml:"url,omitempty"`
	Method             string            `json:"method,omitempty" yaml:"method,omitempty"`
	Status             matcher           `json:"status" yaml:"status"`
	AllowInsecure      bool              `json:"allow-insecure" yaml:"allow-insecure"`
	NoFollowRedirects  bool              `json:"no-follow-redirects" yaml:"no-follow-redirects"`
	Timeout            int               `json:"timeout,omitempty" yaml:"timeout,omitempty"`
	RequestHeader      []string          `json:"request-headers,omitempty" yaml:"request-headers,omitempty"`
	RequestBody        string            `json:"request-body,omitempty" yaml:"request-body,omitempty"`
	RequestQueryParams map[string]string `json:"request-query-params,omitempty" yaml:"request-query-params,omitempty"`
	Headers            matcher           `json:"headers,omitempty" yaml:"headers,omitempty"`
	Body               matcher           `json:"body,omitempty" yaml:"body,omitempty"`
	Username           string            `json:"username,omitempty" yaml:"username,omitempty"`
	Password           string            `json:"password,omitempty" yaml:"password,omitempty"`
	CAFile             string            `json:"ca-file,omitempty" yaml:"ca-file,omitempty"`
	CertFile           string            `json:"cert-file,omitempty" yaml:"cert-file,omitempty"`
	KeyFile            string            `json:"key-file,omitempty" yaml:"key-file,omitempty"`
	Skip               bool              `json:"skip,omitempty" yaml:"skip,omitempty"`
	Proxy              string            `json:"proxy,omitempty" yaml:"proxy,omitempty"`
}

const (
	HTTPResourceKey  = "http"
	HTTPResourceName = "HTTP"
)

func init() {
	registerResource(HTTPResourceKey, &HTTP{})
}

func (h *HTTP) ID() string {
	if h.URL != "" && h.URL != h.id {
		return fmt.Sprintf("%s: %s", h.id, h.URL)
	}
	return h.id
}
func (h *HTTP) SetID(id string)  { h.id = id }
func (u *HTTP) SetSkip()         { u.Skip = true }
func (u *HTTP) TypeKey() string  { return HTTPResourceKey }
func (u *HTTP) TypeName() string { return HTTPResourceName }

// FIXME: Can this be refactored?
func (r *HTTP) GetTitle() string { return r.Title }
func (r *HTTP) GetMeta() meta    { return r.Meta }
func (r *HTTP) getURL() string {
	if r.URL != "" {
		return r.URL
	}
	return r.id
}

func (u *HTTP) Validate(sys *system.System) []TestResult {
	ctx := context.WithValue(context.Background(), idKey{}, u.ID())
	skip := u.Skip
	if u.Timeout == 0 {
		u.Timeout = 5000
	}
	sysHTTP := sys.NewHTTP(ctx, u.getURL(), sys, util.Config{
		AllowInsecure:     u.AllowInsecure,
		CAFile:            u.CAFile,
		CertFile:          u.CertFile,
		KeyFile:           u.KeyFile,
		NoFollowRedirects: u.NoFollowRedirects,
		Timeout:           time.Duration(u.Timeout) * time.Millisecond, Username: u.Username, Password: u.Password, Proxy: u.Proxy,
		RequestHeader: u.RequestHeader, RequestBody: u.RequestBody, RequestQueryParams: u.RequestQueryParams, Method: u.Method})
	sysHTTP.SetAllowInsecure(u.AllowInsecure)
	sysHTTP.SetNoFollowRedirects(u.NoFollowRedirects)

	var results []TestResult
	results = append(results, ValidateValue(u, "status", u.Status, sysHTTP.Status, skip))
	if shouldSkip(results) {
		skip = true
	}
	if isSet(u.Headers) {
		results = append(results, ValidateValue(u, "Headers", u.Headers, sysHTTP.Headers, skip))
	}
	if isSet(u.Body) {
		results = append(results, ValidateValue(u, "Body", u.Body, sysHTTP.Body, skip))
	}

	return results
}

func NewHTTP(sysHTTP system.HTTP, config util.Config) (*HTTP, error) {
	http := sysHTTP.HTTP()
	status, err := sysHTTP.Status()
	u := &HTTP{
		id:                 http,
		Status:             status,
		RequestHeader:      []string{},
		RequestQueryParams: nil,
		Headers:            nil,
		Body:               []string{},
		AllowInsecure:      config.AllowInsecure,
		NoFollowRedirects:  config.NoFollowRedirects,
		Timeout:            config.TimeOutMilliSeconds(),
		Username:           config.Username,
		Password:           config.Password,
		Proxy:              config.Proxy,
	}
	return u, err
}