yep good

2019-02-28 11:47:50 -08:00
commit bdc166c445
11 changed files with 357 additions and 0 deletions
--- a/e621.go
+++ b/e621.go
@@ -0,0 +1,87 @@
+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
+}