88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
// Post represents an e621 post object returned by the e621 API.
|
|
type Post struct {
|
|
ID int `json:"id"`
|
|
Tags string `json:"tags"`
|
|
LockedTags bool `json:"locked_tags"`
|
|
Description string `json:"description"`
|
|
CreatedAt SerializedDate `json:"created_at"`
|
|
CreatorID int `json:"creator_id"`
|
|
Author string `json:"author"`
|
|
Change int `json:"change"`
|
|
Source string `json:"source"`
|
|
Score int `json:"score"`
|
|
FavoritesCount int `json:"fav_count"`
|
|
MD5Hash string `json:"md5"`
|
|
FileSize int `json:"file_size"`
|
|
FileURL string `json:"file_url"`
|
|
FileExt string `json:"file_ext"`
|
|
PreviewURL string `json:"preview_url"`
|
|
PreviewHeight int `json:"preview_height"`
|
|
PreviewWidth int `json:"preview_width"`
|
|
Rating string `json:"rating"`
|
|
Status string `json:"status"`
|
|
Width int `json:"width"`
|
|
Height int `json:"height"`
|
|
HasComments bool `json:"has_comments"`
|
|
HasNotes bool `json:"has_notes"`
|
|
HasChildren bool `json:"has_children"`
|
|
Children string `json:"children"`
|
|
ParentID int `json:"parent_id"`
|
|
Artist []string `json:"artist"`
|
|
Sources []string `json:"sources"`
|
|
}
|
|
|
|
// SerializedDate represents a serialized date passed via JSON
|
|
type SerializedDate struct {
|
|
JSONClass string `json:"json_class"`
|
|
Seconds int `json:"s"`
|
|
Nanoseconds int `json:"n"`
|
|
}
|
|
|
|
// GetPostsForTags gets a list of e621 Posts
|
|
func GetPostsForTags(tags string, limit int, sfw bool) ([]Post, error) {
|
|
client := &http.Client{}
|
|
|
|
var domain string
|
|
|
|
if sfw {
|
|
domain = "e926.net"
|
|
} else {
|
|
domain = "e621.net"
|
|
}
|
|
|
|
req, _ := http.NewRequest("GET", "https://"+domain+"/post/index.json", nil)
|
|
req.Header.Set("User-Agent", "e6dl: go edition (@tjhorner on Telegram)")
|
|
|
|
qs := req.URL.Query()
|
|
qs.Add("tags", tags)
|
|
qs.Add("limit", strconv.Itoa(limit))
|
|
|
|
req.URL.RawQuery = qs.Encode()
|
|
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var posts []Post
|
|
json.Unmarshal(body, &posts)
|
|
|
|
return posts, nil
|
|
}
|