package pixiv

import (
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
)

type ajaxResp struct {
	Error   bool            `json:"error"`
	Message string          `json:"message"`
	Body    json.RawMessage `json:"body"`
}

func (p *Pixiv) ajaxRequest(addr string) (r json.RawMessage, err error) {
	req, err := http.NewRequest("GET", addr, nil)
	if err != nil {
		err = fmt.Errorf("NewRequest failed (%s)", addr)
		return
	}
	req.AddCookie(&p.phpsessid)
	req.Header.Set("User-Agent", p.Ua)
	req.Header.Set("Referer", "https://www.pixiv.net")
	var resp *http.Response
	for i := 0; i < p.RetryCount; i++ {
		resp, err = p.client.Do(req)
		if err != nil {
			log.Printf("%s", err.Error())
			log.Printf("Retry %d of %d...", i, p.RetryCount)
			continue
		}
		defer resp.Body.Close()
		if resp.StatusCode != http.StatusOK {
			log.Printf("GET %s failed: bad status code: %d", addr, resp.StatusCode)
			log.Printf("Retry %d of %d...", i, p.RetryCount)
			continue
		}
		break
	}

	if err != nil {
		log.Printf("%s", err.Error())
		return
	}
	if resp.StatusCode != http.StatusOK {
		log.Printf("GET %s failed: bad status code: %d", addr, resp.StatusCode)
		return
	}
	respData, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Println("ioutil.ReadAll(resp.Body) failed")
		return
	}
	var ajax ajaxResp
	err = json.Unmarshal(respData, &ajax)
	if err != nil {
		log.Println("json.Unmarshal(respData, &ajax) failed")
		return
	}
	if ajax.Error {
		err = errors.New(ajax.Message)
		log.Println("ajax error")
		return
	}
	r = ajax.Body
	return
}