pixiv-scrapper/pixiv/ajax.go

71 lines
1.5 KiB
Go
Raw Normal View History

2021-02-20 17:40:56 +00:00
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
}