e6dl/e621/e621.go

94 lines
3.8 KiB
Go

package e621
import (
"encoding/json"
"io/ioutil"
"net/http"
"strconv"
"time"
)
// Post represents an e621 post object returned by the e621 API.
type Post struct {
ID int `json:"id"` // The ID of the post
Tags string `json:"tags"` // Space-separated list of tags attached to this post
LockedTags bool `json:"locked_tags"` // (undocumented)
Description string `json:"description"` // The post's description
CreatedAt SerializedDate `json:"created_at"` // When the post was uploaded
CreatorID int `json:"creator_id"` // User ID of the user who uploaded the post
Author string `json:"author"` // Username of the user who uploaded the post
Change int `json:"change"` // (undocumented)
Source string `json:"source"` // URL that the source for this post can be found at
Score int `json:"score"` // The post's score (upvotes - downvotes)
FavoritesCount int `json:"fav_count"` // Amount of users that favorited this post
MD5Hash string `json:"md5"` // MD5-sum of the post's file's content
FileSize int `json:"file_size"` // Size of the post's file
FileURL string `json:"file_url"` // URL to the full-sized file
FileExt string `json:"file_ext"` // File extension
PreviewURL string `json:"preview_url"` // URL to preview-sized version of the file
PreviewHeight int `json:"preview_height"` // Height of the preview
PreviewWidth int `json:"preview_width"` // Width of the preview
Rating string `json:"rating"` // Rating of the file ("safe", "questionable", "explicit")
Status string `json:"status"` // Moderation status ("active" or "pending")
Width int `json:"width"` // Width of the original file
Height int `json:"height"` // Height of the original file
HasComments bool `json:"has_comments"` // True if post has comments
HasNotes bool `json:"has_notes"` // True if post has notes
HasChildren bool `json:"has_children"` // True if post has children
Children string `json:"children"` // Comma-separated list of children post IDs
ParentID int `json:"parent_id"` // ID of the parent post
Artist []string `json:"artist"` // Slice of artist names
Sources []string `json:"sources"` // Slice of source URLs
}
// SerializedDate represents a serialized date passed via JSON
type SerializedDate struct {
JSONClass string `json:"json_class"`
Seconds int64 `json:"s"`
Nanoseconds int64 `json:"n"`
}
// Time returns a time.Time object representing the SerializedDate
func (date *SerializedDate) Time() time.Time {
return time.Unix(0, date.Nanoseconds)
}
// 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
}