feat: added mid-level API for pools
Signed-off-by: SoXX <soxx@fenpa.ws>
This commit is contained in:
parent
af69de28cf
commit
989f853d72
@ -37,7 +37,7 @@ _Mid Level API_
|
|||||||
| Tag aliases | :x: | |
|
| Tag aliases | :x: | |
|
||||||
| Tag implications | :x: | |
|
| Tag implications | :x: | |
|
||||||
| Notes | :x: | |
|
| Notes | :x: | |
|
||||||
| Pools | :x: | |
|
| Pools | :heavy_check_mark: | |
|
||||||
| Users | :heavy_check_mark: | |
|
| Users | :heavy_check_mark: | |
|
||||||
| Favorites | :x: | |
|
| Favorites | :x: | |
|
||||||
| DB export | :x: | |
|
| DB export | :x: | |
|
||||||
|
57
example/midlevel/pool.go
Normal file
57
example/midlevel/pool.go
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/builder"
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/model"
|
||||||
|
_ "github.com/joho/godotenv/autoload"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Define the request context with essential information.
|
||||||
|
requestContext := model.RequestContext{
|
||||||
|
Client: http.Client{},
|
||||||
|
Host: "https://e621.net",
|
||||||
|
UserAgent: "Go-e621-SDK (@username)",
|
||||||
|
Username: os.Getenv("API_USER"), // Replace with your username
|
||||||
|
APIKey: os.Getenv("API_KEY"), // Replace with your API key
|
||||||
|
}
|
||||||
|
|
||||||
|
// Log: Getting a single pool.
|
||||||
|
log.Println("Getting single pool: ")
|
||||||
|
|
||||||
|
// Specify the pool ID for retrieval.
|
||||||
|
poolID := 36957 // Replace with the desired pool's ID.
|
||||||
|
|
||||||
|
// Call the GetPool function to retrieve the specified pool.
|
||||||
|
getPool := builder.NewGetPoolBuilder(requestContext)
|
||||||
|
pool, err := getPool.ID(poolID).Execute()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
} else {
|
||||||
|
// Log the name of the retrieved pool.
|
||||||
|
log.Println(pool.Name)
|
||||||
|
}
|
||||||
|
log.Println("----------")
|
||||||
|
|
||||||
|
// Log: Getting a list of pools.
|
||||||
|
log.Println("Getting a list of pools: ")
|
||||||
|
|
||||||
|
// Call the GetPools function to retrieve a list of pools based on the query parameters.
|
||||||
|
getPools := builder.NewGetPoolsBuilder(requestContext)
|
||||||
|
pools, err := getPools.SetLimit(5).Active(true).SearchDescritpion("mass effect").Execute()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
} else {
|
||||||
|
// Log the number of pools retrieved.
|
||||||
|
log.Println(len(pools))
|
||||||
|
log.Println(pools[1].Name)
|
||||||
|
log.Println(pools[1].ID)
|
||||||
|
}
|
||||||
|
log.Println("----------")
|
||||||
|
|
||||||
|
}
|
36
pkg/e621/builder/pool.go
Normal file
36
pkg/e621/builder/pool.go
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
package builder
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/endpoints"
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/model"
|
||||||
|
"log"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
type PoolBuilder interface {
|
||||||
|
ID(poolID int) PoolBuilder
|
||||||
|
Execute() (model.Pool, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewGetPoolBuilder(requestContext model.RequestContext) PoolBuilder {
|
||||||
|
return &getPool{requestContext: requestContext}
|
||||||
|
}
|
||||||
|
|
||||||
|
type getPool struct {
|
||||||
|
requestContext model.RequestContext
|
||||||
|
id int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g getPool) ID(poolID int) PoolBuilder {
|
||||||
|
g.id = poolID
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g getPool) Execute() (model.Pool, error) {
|
||||||
|
pool, err := endpoints.GetPool(g.requestContext, strconv.Itoa(g.id))
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return model.Pool{}, err
|
||||||
|
}
|
||||||
|
return pool, nil
|
||||||
|
}
|
56
pkg/e621/builder/pool_test.go
Normal file
56
pkg/e621/builder/pool_test.go
Normal file
@ -0,0 +1,56 @@
|
|||||||
|
package builder
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/model"
|
||||||
|
"github.com/jarcoal/httpmock"
|
||||||
|
"net/http"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestGetPool(t *testing.T) {
|
||||||
|
httpmock.Activate()
|
||||||
|
defer httpmock.DeactivateAndReset()
|
||||||
|
|
||||||
|
poolResponse := model.Pool{
|
||||||
|
ID: 36957,
|
||||||
|
Name: "MLP:His_Nocturnal_Excellency",
|
||||||
|
CreatedAt: "2023-10-13T19:24:00.576-04:00",
|
||||||
|
UpdatedAt: "2023-10-19T07:46:00.267-04:00",
|
||||||
|
CreatorID: 16883,
|
||||||
|
Description: "Two (un)lucky royal guards who saw something they didn’t have to~",
|
||||||
|
IsActive: true,
|
||||||
|
Category: "series",
|
||||||
|
PostIDS: nil,
|
||||||
|
CreatorName: "2DUK",
|
||||||
|
PostCount: 11,
|
||||||
|
}
|
||||||
|
|
||||||
|
jsonResponser, err := httpmock.NewJsonResponder(200, poolResponse)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
httpmock.RegisterResponder("GET", "https://e621.net/pools/36957.json", jsonResponser)
|
||||||
|
|
||||||
|
requestContext := model.RequestContext{
|
||||||
|
Client: http.Client{},
|
||||||
|
Host: "https://e621.net",
|
||||||
|
UserAgent: "Go-e621-SDK (@username)",
|
||||||
|
Username: "memo",
|
||||||
|
APIKey: "123456",
|
||||||
|
}
|
||||||
|
|
||||||
|
getPool := NewGetPoolBuilder(requestContext)
|
||||||
|
pool, err := getPool.ID(36957).Execute()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if pool.ID == poolResponse.ID && pool.Name == poolResponse.Name {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Errorf("Respons did not match mock data:\nOriginal: %v\nMock: %v", pool, poolResponse)
|
||||||
|
|
||||||
|
}
|
87
pkg/e621/builder/pools.go
Normal file
87
pkg/e621/builder/pools.go
Normal file
@ -0,0 +1,87 @@
|
|||||||
|
package builder
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/endpoints"
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/model"
|
||||||
|
"log"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
type PoolsBuilder interface {
|
||||||
|
SearchName(name string) PoolsBuilder
|
||||||
|
SearchID(poolIDs string) PoolsBuilder
|
||||||
|
SearchDescritpion(description string) PoolsBuilder
|
||||||
|
SearchCreatorID(creatorID int) PoolsBuilder
|
||||||
|
searchCreatorName(creatorName string) PoolsBuilder
|
||||||
|
Active(isActive bool) PoolsBuilder
|
||||||
|
SearchCategory(category model.PoolCategory) PoolsBuilder
|
||||||
|
Order(order model.PoolOrder) PoolsBuilder
|
||||||
|
SetLimit(limitPools int) PoolsBuilder
|
||||||
|
Execute() ([]model.Pool, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewGetPoolsBuilder(requstContext model.RequestContext) PoolsBuilder {
|
||||||
|
return &getPools{
|
||||||
|
requestContext: requstContext,
|
||||||
|
query: make(map[string]string),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type getPools struct {
|
||||||
|
requestContext model.RequestContext
|
||||||
|
query map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) SearchName(name string) PoolsBuilder {
|
||||||
|
g.query["search[name_matches]"] = name
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) SearchID(poolIDs string) PoolsBuilder {
|
||||||
|
g.query["search[id]"] = poolIDs
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) SearchDescritpion(description string) PoolsBuilder {
|
||||||
|
g.query["search[description_matches]"] = description
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) SearchCreatorID(creatorID int) PoolsBuilder {
|
||||||
|
g.query["search[creator_id]"] = strconv.Itoa(creatorID)
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) searchCreatorName(creatorName string) PoolsBuilder {
|
||||||
|
g.query["search[creator_name]"] = creatorName
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) Active(isActive bool) PoolsBuilder {
|
||||||
|
g.query["search[is_active]"] = strconv.FormatBool(isActive)
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) SearchCategory(category model.PoolCategory) PoolsBuilder {
|
||||||
|
g.query["search[category]"] = string(category)
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) Order(order model.PoolOrder) PoolsBuilder {
|
||||||
|
g.query["search[order]"] = string(order)
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) SetLimit(limitUser int) PoolsBuilder {
|
||||||
|
g.query["limit"] = strconv.Itoa(limitUser)
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *getPools) Execute() ([]model.Pool, error) {
|
||||||
|
pools, err := endpoints.GetPools(g.requestContext, g.query)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return pools, nil
|
||||||
|
}
|
97
pkg/e621/builder/pools_test.go
Normal file
97
pkg/e621/builder/pools_test.go
Normal file
@ -0,0 +1,97 @@
|
|||||||
|
package builder
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/model"
|
||||||
|
"github.com/jarcoal/httpmock"
|
||||||
|
"net/http"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestGetPools(t *testing.T) {
|
||||||
|
httpmock.Activate()
|
||||||
|
defer httpmock.DeactivateAndReset()
|
||||||
|
|
||||||
|
poolResponse := []model.Pool{
|
||||||
|
{
|
||||||
|
ID: 36957,
|
||||||
|
Name: "MLP:His_Nocturnal_Excellency",
|
||||||
|
CreatedAt: "2023-10-13T19:24:00.576-04:00",
|
||||||
|
UpdatedAt: "2023-10-19T07:46:00.267-04:00",
|
||||||
|
CreatorID: 16883,
|
||||||
|
Description: "Two (un)lucky royal guards who saw something they didn’t have to~",
|
||||||
|
IsActive: true,
|
||||||
|
Category: "series",
|
||||||
|
PostIDS: nil,
|
||||||
|
CreatorName: "2DUK",
|
||||||
|
PostCount: 11,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
ID: 5772,
|
||||||
|
Name: "MLP:His_Nocturnal_Excellency",
|
||||||
|
CreatedAt: "2023-10-13T19:24:00.576-04:00",
|
||||||
|
UpdatedAt: "2023-10-19T07:46:00.267-04:00",
|
||||||
|
CreatorID: 16883,
|
||||||
|
Description: "Two (un)lucky royal guards who saw something they didn’t have to~",
|
||||||
|
IsActive: true,
|
||||||
|
Category: "series",
|
||||||
|
PostIDS: nil,
|
||||||
|
CreatorName: "2DUK",
|
||||||
|
PostCount: 11,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
ID: 453687,
|
||||||
|
Name: "MLP:His_Nocturnal_Excellency",
|
||||||
|
CreatedAt: "2023-10-13T19:24:00.576-04:00",
|
||||||
|
UpdatedAt: "2023-10-19T07:46:00.267-04:00",
|
||||||
|
CreatorID: 16883,
|
||||||
|
Description: "Two (un)lucky royal guards who saw something they didn’t have to~",
|
||||||
|
IsActive: true,
|
||||||
|
Category: "series",
|
||||||
|
PostIDS: nil,
|
||||||
|
CreatorName: "2DUK",
|
||||||
|
PostCount: 11,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
ID: 3456876,
|
||||||
|
Name: "MLP:His_Nocturnal_Excellency",
|
||||||
|
CreatedAt: "2023-10-13T19:24:00.576-04:00",
|
||||||
|
UpdatedAt: "2023-10-19T07:46:00.267-04:00",
|
||||||
|
CreatorID: 16883,
|
||||||
|
Description: "Two (un)lucky royal guards who saw something they didn’t have to~",
|
||||||
|
IsActive: true,
|
||||||
|
Category: "series",
|
||||||
|
PostIDS: nil,
|
||||||
|
CreatorName: "2DUK",
|
||||||
|
PostCount: 11,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
jsonResponser, err := httpmock.NewJsonResponder(200, poolResponse)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
httpmock.RegisterResponder("GET", "https://e621.net/pools.json", jsonResponser)
|
||||||
|
|
||||||
|
requestContext := model.RequestContext{
|
||||||
|
Client: http.Client{},
|
||||||
|
Host: "https://e621.net",
|
||||||
|
UserAgent: "Go-e621-SDK (@username)",
|
||||||
|
Username: "memo",
|
||||||
|
APIKey: "123456",
|
||||||
|
}
|
||||||
|
|
||||||
|
getPools := NewGetPoolsBuilder(requestContext)
|
||||||
|
pools, err := getPools.SetLimit(4).Execute()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(pools) == 4 && pools[0].ID == poolResponse[0].ID && pools[1].Name == poolResponse[1].Name && pools[2].UpdatedAt == poolResponse[2].UpdatedAt {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Errorf("Respons did not match mock data:\nOriginal: %v\nMock: %v", pools, poolResponse)
|
||||||
|
|
||||||
|
}
|
@ -1,5 +1,17 @@
|
|||||||
package model
|
package model
|
||||||
|
|
||||||
|
type PoolCategory string
|
||||||
|
type PoolOrder string
|
||||||
|
|
||||||
|
const (
|
||||||
|
Series PoolCategory = "series"
|
||||||
|
Collection PoolCategory = "collection"
|
||||||
|
Name PoolOrder = "name"
|
||||||
|
CreatedAt PoolOrder = "created_at"
|
||||||
|
UpdatedAt PoolOrder = "updated_at"
|
||||||
|
PostCount PoolOrder = "post_count"
|
||||||
|
)
|
||||||
|
|
||||||
type Pool struct {
|
type Pool struct {
|
||||||
ID int64 `json:"id"`
|
ID int64 `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
@ -8,7 +20,7 @@ type Pool struct {
|
|||||||
CreatorID int64 `json:"creator_id"`
|
CreatorID int64 `json:"creator_id"`
|
||||||
Description string `json:"description"`
|
Description string `json:"description"`
|
||||||
IsActive bool `json:"is_active"`
|
IsActive bool `json:"is_active"`
|
||||||
Category string `json:"category"`
|
Category PoolCategory `json:"category"`
|
||||||
PostIDS []int64 `json:"post_ids"`
|
PostIDS []int64 `json:"post_ids"`
|
||||||
CreatorName string `json:"creator_name"`
|
CreatorName string `json:"creator_name"`
|
||||||
PostCount int64 `json:"post_count"`
|
PostCount int64 `json:"post_count"`
|
||||||
|
Reference in New Issue
Block a user