package gondole

import (

	"encoding/json"

	"fmt"

	"strconv"

	"github.com/sendgrid/rest"

)

// GetNotifications returns the list of the user's notifications

func (g *Client) GetNotifications() ([]Notification, error) {

	var notifications []Notification

	req := g.prepareRequest("notifications")

	r, err := rest.API(req)

	if err != nil {

		return notifications, fmt.Errorf("notifications API query: %s", err.Error())

	}

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

	if err != nil {

		var errorRes Error

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

		if err2 == nil {

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

		}

		return notifications, fmt.Errorf("notifications API: %s", err.Error())

	}

	return notifications, nil

}

// GetNotification returns a notification

func (g *Client) GetNotification(id int) (*Notification, error) {

	var notification Notification

	req := g.prepareRequest("notifications/" + strconv.Itoa(id))

	r, err := rest.API(req)

	if err != nil {

		return &notification, fmt.Errorf("notification API query: %s", err.Error())

	}

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

	if err != nil {

		var errorRes Error

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

		if err2 == nil {

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

		}

		return &notification, fmt.Errorf("notification API: %s", err.Error())

	}

	if notification.ID == 0 {

		return nil, ErrEntityNotFound

	}

	return &notification, nil

}

// ClearNotifications deletes all notifications from the Mastodon server for

// the authenticated user

func (g *Client) ClearNotifications() error {

	req := g.prepareRequest("notifications/clear")

	req.Method = rest.Post

	_, err := rest.API(req)

	if err != nil {

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

	}

	return nil // TODO: check returned object (should be empty)

}