package probe

import (
	"fmt"

	"github.com/ultradns/ultradns-go-sdk/pkg/helper"
)

const (
	HTTP     = "HTTP"
	TCP      = "TCP"
	FTP      = "FTP"
	SMTP     = "SMTP"
	PING     = "PING"
	DNS      = "DNS"
	SMTPSend = "SMTP_SEND"
)

type Probe struct {
	ID         string     `json:"id,omitempty"`
	Type       string     `json:"type,omitempty"`
	PoolRecord string     `json:"poolRecord,omitempty"`
	Interval   string     `json:"interval,omitempty"`
	Agents     []string   `json:"agents,omitempty"`
	Threshold  int        `json:"threshold,omitempty"`
	Details    RawDetails `json:"details,omitempty"`
}

type RawDetails interface{}

type Query struct {
	Type       string
	PoolRecord string
}

type ResponseList struct {
	QueryInfo  *helper.QueryInfo  `json:"queryInfo,omitempty"`
	ResultInfo *helper.ResultInfo `json:"resultInfo,omitempty"`
	Probes     []*Probe           `json:"probes,omitempty"`
}

func (q Query) String() string {
	if q.PoolRecord != "" {
		return fmt.Sprintf("poolRecord:%s", q.PoolRecord)
	}

	if q.Type == "" {
		q.Type = "ALL"
	}

	return fmt.Sprintf("type:%s", q.Type)
}