package gondole

import (
	"encoding/json"
	"fmt"
	"strings"

	"github.com/sendgrid/rest"
)

// GetTimelines returns a timeline (a list of statuses
// timeline can be "home", "public", or a hashtag (":hashtag")
func (g *Client) GetTimelines(timeline string) ([]Status, error) {
	var endPoint string
	var tl []Status

	if timeline == "home" || timeline == "public" {
		endPoint = "timelines/" + timeline
	} else if strings.HasPrefix(timeline, ":") {
		endPoint = "timelines/tag/" + timeline
	} else {
		return tl, fmt.Errorf("GetTimelines: bad timelines argument")
	}

	req := g.prepareRequest(endPoint)
	r, err := rest.API(req)
	if err != nil {
		return tl, fmt.Errorf("timelines API query: %s", err.Error())
	}

	err = json.Unmarshal([]byte(r.Body), &tl)
	if err != nil {
		var res struct {
			Error string `json:"error"`
		}
		err2 := json.Unmarshal([]byte(r.Body), &res)
		if err2 == nil {
			return tl, fmt.Errorf("%s", res.Error)
		}
		return tl, fmt.Errorf("timelines API: %s", err.Error())
	}

	return tl, nil
}