import (

	"encoding/json"

	"fmt"

	"strconv"

	"github.com/sendgrid/rest"

)

// queryStatusData queries the statuses API

// The subquery can be empty (the status itself), "context", "card",

// "reblogged_by", "favourited_by".

func (g *Client) queryStatusData(statusID int, subquery string, data interface{}) error {

	endPoint := "statuses/" + strconv.Itoa(statusID)

	if statusID < 1 {

		return ErrInvalidID

	}

	if subquery != "" {

		// TODO: check subquery values?

		endPoint += "/" + subquery

	}

	req := g.prepareRequest(endPoint)

	r, err := rest.API(req)

	if err != nil {

		return fmt.Errorf("status/%s API query: %s", subquery, err.Error())

	}

	err = json.Unmarshal([]byte(r.Body), &data)

	if err != nil {

		var errorRes Error

		err2 := json.Unmarshal([]byte(r.Body), &errorRes)

		if err2 == nil {

			return fmt.Errorf("%s", errorRes.Text)

		}

		return fmt.Errorf("status/%s API: %s", subquery, err.Error())

	}

	return nil

}

// GetStatus returns a status

// The returned status can be nil if there is an error or if the

// requested ID does not exist.

func (g *Client) GetStatus(id int) (*Status, error) {

	var status Status

	if err := g.queryStatusData(id, "", &status); err != nil {

		return nil, err

	}

	if status.ID == 0 {

		return nil, ErrEntityNotFound

	}

	return &status, nil

}

// GetStatusContext returns a status context

func (g *Client) GetStatusContext(id int) (*Context, error) {

	var context Context

	if err := g.queryStatusData(id, "context", &context); err != nil {

		return nil, err

	}

	return &context, nil

}

// GetStatusCard returns a status card

func (g *Client) GetStatusCard(id int) (*Card, error) {

	var card Card

	if err := g.queryStatusData(id, "card", &card); err != nil {

		return nil, err

	}

	return &card, nil

}

// GetStatusRebloggedBy returns a list of the accounts who reblogged a status

func (g *Client) GetStatusRebloggedBy(id int) ([]Account, error) {

	var accounts []Account

	err := g.queryStatusData(id, "reblogged_by", &accounts)

	return accounts, err

}

// GetStatusFavouritedBy returns a list of the accounts who favourited a status

func (g *Client) GetStatusFavouritedBy(id int) ([]Account, error) {

	var accounts []Account

	err := g.queryStatusData(id, "favourited_by", &accounts)

	return accounts, err

}