Compare commits
8 Commits
e2bce2bda7
...
8c662f78fd
Author | SHA1 | Date | |
---|---|---|---|
8c662f78fd | |||
9cee99ecb5 | |||
c6be78419c | |||
208b9a3baa | |||
95e1db6ce6 | |||
9669eb8ddf | |||
8e76fa709d | |||
6124a3fc61 |
46
example/highlevel/dbexport.go
Normal file
46
example/highlevel/dbexport.go
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621"
|
||||||
|
_ "github.com/joho/godotenv/autoload"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
client := e621.NewClient(os.Getenv("API_USER"), os.Getenv("API_KEY"))
|
||||||
|
|
||||||
|
{
|
||||||
|
fileName, _, err := client.GetLatestPoolsDBExportDataAsBytes()
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
log.Println(fileName)
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
latestDBPoolExport, err := client.GetLatestPoolsDBExportDataAsStruct()
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
log.Println(latestDBPoolExport[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
fileName, _, err := client.GetLatestPostsDBExportDataAsBytes()
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
log.Println(fileName)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
latestDBPoolExport, err := client.GetLatestPostsDBExportDataAsStruct()
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
log.Println(latestDBPoolExport[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
50
example/lowlevel/dbexport.go
Normal file
50
example/lowlevel/dbexport.go
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/endpoints"
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
|
"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.Println("Getting a list of DB Exports: ")
|
||||||
|
dbExportFiles, err := endpoints.GetDBExportList(requestContext)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("%d files found", len(dbExportFiles))
|
||||||
|
for _, v := range dbExportFiles {
|
||||||
|
log.Printf("File found: %s", v)
|
||||||
|
}
|
||||||
|
|
||||||
|
exportFileName := dbExportFiles[0]
|
||||||
|
log.Println("Downloading DB export")
|
||||||
|
log.Printf("File to download: %s", exportFileName)
|
||||||
|
rawFile, err := endpoints.GetDBExportFile(requestContext, exportFileName)
|
||||||
|
|
||||||
|
file, err := os.Create(exportFileName)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
err = os.WriteFile(exportFileName, rawFile, 0644)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
log.Printf("File %s downloaded", exportFileName)
|
||||||
|
|
||||||
|
}
|
55
example/midlevel/dbexport.go
Normal file
55
example/midlevel/dbexport.go
Normal file
@ -0,0 +1,55 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/builder"
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
|
"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.Println("Getting a list of DB Exports: ")
|
||||||
|
getDBExportList := builder.NewGetDBExportListBuilder(requestContext)
|
||||||
|
dbExportFiles, err := getDBExportList.Execute()
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("%d files found", len(dbExportFiles))
|
||||||
|
for _, v := range dbExportFiles {
|
||||||
|
log.Printf("File found: %s", v)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println(dbExportFiles)
|
||||||
|
|
||||||
|
exportFileName := dbExportFiles[0]
|
||||||
|
log.Println("Downloading DB export")
|
||||||
|
log.Printf("File to download: %s", exportFileName)
|
||||||
|
getDBExportFile := builder.NewGetDBExportFileBuilder(requestContext)
|
||||||
|
|
||||||
|
rawFile, err := getDBExportFile.SetFile(exportFileName).Execute()
|
||||||
|
|
||||||
|
file, err := os.Create(exportFileName)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
err = os.WriteFile(exportFileName, rawFile, 0644)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
log.Printf("File %s downloaded", exportFileName)
|
||||||
|
|
||||||
|
}
|
5
go.mod
5
go.mod
@ -8,4 +8,7 @@ require (
|
|||||||
golang.org/x/time v0.3.0
|
golang.org/x/time v0.3.0
|
||||||
)
|
)
|
||||||
|
|
||||||
require golang.org/x/net v0.18.0 // indirect
|
require (
|
||||||
|
github.com/gocarina/gocsv v0.0.0-20230616125104-99d496ca653d // indirect
|
||||||
|
golang.org/x/net v0.18.0 // indirect
|
||||||
|
)
|
||||||
|
2
go.sum
2
go.sum
@ -1,5 +1,7 @@
|
|||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/gocarina/gocsv v0.0.0-20230616125104-99d496ca653d h1:KbPOUXFUDJxwZ04vbmDOc3yuruGvVO+LOa7cVER3yWw=
|
||||||
|
github.com/gocarina/gocsv v0.0.0-20230616125104-99d496ca653d/go.mod h1:5YoVOkjYAQumqlV356Hj3xeYh4BdZuLE0/nRkf2NKkI=
|
||||||
github.com/jarcoal/httpmock v1.3.1 h1:iUx3whfZWVf3jT01hQTO/Eo5sAYtB2/rqaUuOtpInww=
|
github.com/jarcoal/httpmock v1.3.1 h1:iUx3whfZWVf3jT01hQTO/Eo5sAYtB2/rqaUuOtpInww=
|
||||||
github.com/jarcoal/httpmock v1.3.1/go.mod h1:3yb8rc4BI7TCBhFY8ng0gjuLKJNquuDNiPaZjnENuYg=
|
github.com/jarcoal/httpmock v1.3.1/go.mod h1:3yb8rc4BI7TCBhFY8ng0gjuLKJNquuDNiPaZjnENuYg=
|
||||||
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
||||||
|
54
pkg/e621/builder/db_export_file.go
Normal file
54
pkg/e621/builder/db_export_file.go
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
package builder
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/endpoints"
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DBExportFileBuilder defines an interface for building and executing requests to retrieve a specific exported database file.
|
||||||
|
type DBExportFileBuilder interface {
|
||||||
|
// SetFile sets the name of the file to be retrieved.
|
||||||
|
SetFile(fileName string) DBExportFileBuilder
|
||||||
|
// Execute sends the request and returns the content of the exported database file or an error if the request fails.
|
||||||
|
Execute() ([]byte, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewGetDBExportFileBuilder creates a new instance of DBExportFileBuilder.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - DBExportFileBuilder: An instance of the DBExportFileBuilder interface.
|
||||||
|
func NewGetDBExportFileBuilder(requestContext model.RequestContext) DBExportFileBuilder {
|
||||||
|
return &getDBExportFile{
|
||||||
|
requestContext: requestContext,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getDBExportFile is an implementation of the DBExportFileBuilder interface.
|
||||||
|
type getDBExportFile struct {
|
||||||
|
requestContext model.RequestContext
|
||||||
|
fileName string
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetFile sets the name of the file to be retrieved.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - fileName: The name of the exported database file.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - DBExportFileBuilder: The instance of the DBExportFileBuilder for method chaining.
|
||||||
|
func (g *getDBExportFile) SetFile(fileName string) DBExportFileBuilder {
|
||||||
|
g.fileName = fileName
|
||||||
|
return g
|
||||||
|
}
|
||||||
|
|
||||||
|
// Execute sends the API request to retrieve the content of a specific exported database file.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []byte: The content of the exported database file.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
|
func (g *getDBExportFile) Execute() ([]byte, error) {
|
||||||
|
return endpoints.GetDBExportFile(g.requestContext, g.fileName)
|
||||||
|
}
|
39
pkg/e621/builder/db_export_list.go
Normal file
39
pkg/e621/builder/db_export_list.go
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
package builder
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/endpoints"
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DBExportListBuilder defines an interface for building and executing requests to retrieve a list of exported databases.
|
||||||
|
type DBExportListBuilder interface {
|
||||||
|
// Execute sends the request and returns a list of exported databases or an error if the request fails.
|
||||||
|
Execute() ([]string, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewGetDBExportListBuilder creates a new instance of DBExportListBuilder.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - DBExportListBuilder: An instance of the DBExportListBuilder interface.
|
||||||
|
func NewGetDBExportListBuilder(requestContext model.RequestContext) DBExportListBuilder {
|
||||||
|
return &getDBExportList{
|
||||||
|
requestContext: requestContext,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getDBExportList is an implementation of the DBExportListBuilder interface.
|
||||||
|
type getDBExportList struct {
|
||||||
|
requestContext model.RequestContext
|
||||||
|
}
|
||||||
|
|
||||||
|
// Execute sends the API request to retrieve a list of exported databases.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []string: A list of exported databases.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
|
func (g *getDBExportList) Execute() ([]string, error) {
|
||||||
|
return endpoints.GetDBExportList(g.requestContext)
|
||||||
|
}
|
@ -19,7 +19,13 @@ type FavoritesBuilder interface {
|
|||||||
Execute() ([]model.Post, error)
|
Execute() ([]model.Post, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetFavoritesBuilder creates a new FavoritesBuilder with the provided RequestContext.
|
// NewGetFavoritesBuilder creates a new instance of FavoritesBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - FavoritesBuilder: An instance of the FavoritesBuilder interface.
|
||||||
func NewGetFavoritesBuilder(requestContext model.RequestContext) FavoritesBuilder {
|
func NewGetFavoritesBuilder(requestContext model.RequestContext) FavoritesBuilder {
|
||||||
return &getFavorites{
|
return &getFavorites{
|
||||||
requestContext: requestContext,
|
requestContext: requestContext,
|
||||||
@ -27,30 +33,53 @@ func NewGetFavoritesBuilder(requestContext model.RequestContext) FavoritesBuilde
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getFavorites is an implementation of the FavoritesBuilder interface.
|
||||||
type getFavorites struct {
|
type getFavorites struct {
|
||||||
query map[string]string
|
query map[string]string
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetUserID sets the user ID for the query.
|
// SetUserID sets the user ID for the query.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - userID: The ID of the user whose favorite posts are to be retrieved.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - FavoritesBuilder: The instance of the FavoritesBuilder for method chaining.
|
||||||
func (g *getFavorites) SetUserID(userID model.UserID) FavoritesBuilder {
|
func (g *getFavorites) SetUserID(userID model.UserID) FavoritesBuilder {
|
||||||
g.query["user_id"] = strconv.Itoa(int(userID))
|
g.query["user_id"] = strconv.Itoa(int(userID))
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetLimit sets the maximum number of favorite posts to retrieve.
|
// SetLimit sets the maximum number of favorite posts to retrieve.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - limitFavorites: The maximum number of favorite posts to retrieve.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - FavoritesBuilder: The instance of the FavoritesBuilder for method chaining.
|
||||||
func (g *getFavorites) SetLimit(limitFavorites int) FavoritesBuilder {
|
func (g *getFavorites) SetLimit(limitFavorites int) FavoritesBuilder {
|
||||||
g.query["limit"] = strconv.Itoa(limitFavorites)
|
g.query["limit"] = strconv.Itoa(limitFavorites)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Page sets the page number for paginated results.
|
// Page sets the page number for paginated results.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - pageNumber: The page number for paginated results.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - FavoritesBuilder: The instance of the FavoritesBuilder for method chaining.
|
||||||
func (g *getFavorites) Page(pageNumber int) FavoritesBuilder {
|
func (g *getFavorites) Page(pageNumber int) FavoritesBuilder {
|
||||||
g.query["page"] = strconv.Itoa(pageNumber)
|
g.query["page"] = strconv.Itoa(pageNumber)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns a slice of favorite posts and an error if any.
|
// Execute sends the constructed query and returns a slice of favorite posts and an error if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []model.Post: A slice of favorite posts.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getFavorites) Execute() ([]model.Post, error) {
|
func (g *getFavorites) Execute() ([]model.Post, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -15,23 +15,40 @@ type NoteBuilder interface {
|
|||||||
Execute() (*model.Note, error)
|
Execute() (*model.Note, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetNoteBuilder creates a new NoteBuilder with the provided RequestContext.
|
// NewGetNoteBuilder creates a new instance of NoteBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NoteBuilder: An instance of the NoteBuilder interface.
|
||||||
func NewGetNoteBuilder(requestContext model.RequestContext) NoteBuilder {
|
func NewGetNoteBuilder(requestContext model.RequestContext) NoteBuilder {
|
||||||
return &getNote{requestContext: requestContext}
|
return &getNote{requestContext: requestContext}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getNote is an implementation of the NoteBuilder interface.
|
||||||
type getNote struct {
|
type getNote struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
noteID int
|
noteID int
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetNoteID sets the note ID for the query.
|
// SetNoteID sets the note ID for the query.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - noteID: The ID of the note to be retrieved.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NoteBuilder: The instance of the NoteBuilder for method chaining.
|
||||||
func (g *getNote) SetNoteID(noteID int) NoteBuilder {
|
func (g *getNote) SetNoteID(noteID int) NoteBuilder {
|
||||||
g.noteID = noteID
|
g.noteID = noteID
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns the requested note and an error, if any.
|
// Execute sends the constructed query and returns the requested note and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - *model.Note: The requested note.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getNote) Execute() (*model.Note, error) {
|
func (g *getNote) Execute() (*model.Note, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -27,7 +27,13 @@ type NotesBuilder interface {
|
|||||||
Execute() ([]model.Note, error)
|
Execute() ([]model.Note, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetNotesBuilder creates a new NotesBuilder with the provided RequestContext.
|
// NewGetNotesBuilder creates a new instance of NotesBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: An instance of the NotesBuilder interface.
|
||||||
func NewGetNotesBuilder(requestContext model.RequestContext) NotesBuilder {
|
func NewGetNotesBuilder(requestContext model.RequestContext) NotesBuilder {
|
||||||
return &getNotes{
|
return &getNotes{
|
||||||
requestContext: requestContext,
|
requestContext: requestContext,
|
||||||
@ -35,54 +41,101 @@ func NewGetNotesBuilder(requestContext model.RequestContext) NotesBuilder {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getNotes is an implementation of the NotesBuilder interface.
|
||||||
type getNotes struct {
|
type getNotes struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
query map[string]string
|
query map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchInBody sets the query to search for notes containing a specific text in their body.
|
// SearchInBody sets the query to search for notes containing a specific text in their body.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - body: The text to search for in the notes' body.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: The instance of the NotesBuilder for method chaining.
|
||||||
func (g *getNotes) SearchInBody(body string) NotesBuilder {
|
func (g *getNotes) SearchInBody(body string) NotesBuilder {
|
||||||
g.query["search[body_matches]"] = body
|
g.query["search[body_matches]"] = body
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// NoteID sets the query to search for a note with a specific note ID.
|
// NoteID sets the query to search for a note with a specific note ID.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - noteID: The ID of the note to be retrieved.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: The instance of the NotesBuilder for method chaining.
|
||||||
func (g *getNotes) NoteID(noteID int) NotesBuilder {
|
func (g *getNotes) NoteID(noteID int) NotesBuilder {
|
||||||
g.query["search[post_id]"] = strconv.Itoa(noteID)
|
g.query["search[post_id]"] = strconv.Itoa(noteID)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchTags sets the query to search for notes containing specific tags.
|
// SearchTags sets the query to search for notes containing specific tags.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - tags: The tags to search for in the notes.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: The instance of the NotesBuilder for method chaining.
|
||||||
func (g *getNotes) SearchTags(tags string) NotesBuilder {
|
func (g *getNotes) SearchTags(tags string) NotesBuilder {
|
||||||
g.query["search[post_tags_match]"] = tags
|
g.query["search[post_tags_match]"] = tags
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchCreatorID sets the query to search for notes created by a specific user (by their user ID).
|
// SearchCreatorID sets the query to search for notes created by a specific user (by their user ID).
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - creatorID: The ID of the user who created the notes.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: The instance of the NotesBuilder for method chaining.
|
||||||
func (g *getNotes) SearchCreatorID(creatorID int) NotesBuilder {
|
func (g *getNotes) SearchCreatorID(creatorID int) NotesBuilder {
|
||||||
g.query["search[creator_id]"] = strconv.Itoa(creatorID)
|
g.query["search[creator_id]"] = strconv.Itoa(creatorID)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchCreatorName sets the query to search for notes created by a specific user (by their username).
|
// SearchCreatorName sets the query to search for notes created by a specific user (by their username).
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - creatorName: The username of the user who created the notes.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: The instance of the NotesBuilder for method chaining.
|
||||||
func (g *getNotes) SearchCreatorName(creatorName string) NotesBuilder {
|
func (g *getNotes) SearchCreatorName(creatorName string) NotesBuilder {
|
||||||
g.query["search[creator_name]"] = creatorName
|
g.query["search[creator_name]"] = creatorName
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Active sets whether to search for active or inactive notes.
|
// Active sets whether to search for active or inactive notes.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - isActive: A boolean indicating whether to search for active notes (true) or inactive notes (false).
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: The instance of the NotesBuilder for method chaining.
|
||||||
func (g *getNotes) Active(isActive bool) NotesBuilder {
|
func (g *getNotes) Active(isActive bool) NotesBuilder {
|
||||||
g.query["search[is_active]"] = strconv.FormatBool(isActive)
|
g.query["search[is_active]"] = strconv.FormatBool(isActive)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetLimit sets the maximum number of notes to retrieve.
|
// SetLimit sets the maximum number of notes to retrieve.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - limitNotes: The maximum number of notes to retrieve.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - NotesBuilder: The instance of the NotesBuilder for method chaining.
|
||||||
func (g *getNotes) SetLimit(limitNotes int) NotesBuilder {
|
func (g *getNotes) SetLimit(limitNotes int) NotesBuilder {
|
||||||
g.query["limit"] = strconv.Itoa(limitNotes)
|
g.query["limit"] = strconv.Itoa(limitNotes)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns a slice of notes and an error, if any.
|
// Execute sends the constructed query and returns a slice of notes and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []model.Note: A slice of notes.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getNotes) Execute() ([]model.Note, error) {
|
func (g *getNotes) Execute() ([]model.Note, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -15,23 +15,40 @@ type PoolBuilder interface {
|
|||||||
Execute() (model.Pool, error)
|
Execute() (model.Pool, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetPoolBuilder creates a new PoolBuilder with the provided RequestContext.
|
// NewGetPoolBuilder creates a new instance of PoolBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolBuilder: An instance of the PoolBuilder interface.
|
||||||
func NewGetPoolBuilder(requestContext model.RequestContext) PoolBuilder {
|
func NewGetPoolBuilder(requestContext model.RequestContext) PoolBuilder {
|
||||||
return &getPool{requestContext: requestContext}
|
return &getPool{requestContext: requestContext}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getPool is an implementation of the PoolBuilder interface.
|
||||||
type getPool struct {
|
type getPool struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
id int
|
id int
|
||||||
}
|
}
|
||||||
|
|
||||||
// ID sets the pool ID for the query.
|
// ID sets the pool ID for the query.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - poolID: The ID of the pool to be retrieved.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolBuilder: The instance of the PoolBuilder for method chaining.
|
||||||
func (g *getPool) ID(poolID int) PoolBuilder {
|
func (g *getPool) ID(poolID int) PoolBuilder {
|
||||||
g.id = poolID
|
g.id = poolID
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns the requested pool and an error, if any.
|
// Execute sends the constructed query and returns the requested pool and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - model.Pool: The requested pool.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getPool) Execute() (model.Pool, error) {
|
func (g *getPool) Execute() (model.Pool, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -31,7 +31,13 @@ type PoolsBuilder interface {
|
|||||||
Execute() ([]model.Pool, error)
|
Execute() ([]model.Pool, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetPoolsBuilder creates a new PoolsBuilder with the provided RequestContext.
|
// NewGetPoolsBuilder creates a new instance of PoolsBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: An instance of the PoolsBuilder interface.
|
||||||
func NewGetPoolsBuilder(requestContext model.RequestContext) PoolsBuilder {
|
func NewGetPoolsBuilder(requestContext model.RequestContext) PoolsBuilder {
|
||||||
return &getPools{
|
return &getPools{
|
||||||
requestContext: requestContext,
|
requestContext: requestContext,
|
||||||
@ -39,66 +45,125 @@ func NewGetPoolsBuilder(requestContext model.RequestContext) PoolsBuilder {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getPools is an implementation of the PoolsBuilder interface.
|
||||||
type getPools struct {
|
type getPools struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
query map[string]string
|
query map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchName sets the query to search for pools with a specific name.
|
// SearchName sets the query to search for pools with a specific name.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - name: The name to search for in the pools.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) SearchName(name string) PoolsBuilder {
|
func (g *getPools) SearchName(name string) PoolsBuilder {
|
||||||
g.query["search[name_matches]"] = name
|
g.query["search[name_matches]"] = name
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchID sets the query to search for pools with specific pool IDs.
|
// SearchID sets the query to search for pools with specific pool IDs.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - poolIDs: The pool IDs to search for, separated by commas.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) SearchID(poolIDs string) PoolsBuilder {
|
func (g *getPools) SearchID(poolIDs string) PoolsBuilder {
|
||||||
g.query["search[id]"] = poolIDs
|
g.query["search[id]"] = poolIDs
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchDescription sets the query to search for pools with specific descriptions.
|
// SearchDescription sets the query to search for pools with specific descriptions.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - description: The description to search for in the pools.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) SearchDescription(description string) PoolsBuilder {
|
func (g *getPools) SearchDescription(description string) PoolsBuilder {
|
||||||
g.query["search[description_matches]"] = description
|
g.query["search[description_matches]"] = description
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchCreatorID sets the query to search for pools created by a specific user (by their user ID).
|
// SearchCreatorID sets the query to search for pools created by a specific user (by their user ID).
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - creatorID: The ID of the user who created the pools.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) SearchCreatorID(creatorID int) PoolsBuilder {
|
func (g *getPools) SearchCreatorID(creatorID int) PoolsBuilder {
|
||||||
g.query["search[creator_id]"] = strconv.Itoa(creatorID)
|
g.query["search[creator_id]"] = strconv.Itoa(creatorID)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchCreatorName sets the query to search for pools created by a specific user (by their username).
|
// SearchCreatorName sets the query to search for pools created by a specific user (by their username).
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - creatorName: The username of the user who created the pools.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) SearchCreatorName(creatorName string) PoolsBuilder {
|
func (g *getPools) SearchCreatorName(creatorName string) PoolsBuilder {
|
||||||
g.query["search[creator_name]"] = creatorName
|
g.query["search[creator_name]"] = creatorName
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Active sets whether to search for active or inactive pools.
|
// Active sets whether to search for active or inactive pools.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - isActive: A boolean indicating whether to search for active pools (true) or inactive pools (false).
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) Active(isActive bool) PoolsBuilder {
|
func (g *getPools) Active(isActive bool) PoolsBuilder {
|
||||||
g.query["search[is_active]"] = strconv.FormatBool(isActive)
|
g.query["search[is_active]"] = strconv.FormatBool(isActive)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchCategory sets the query to search for pools in a specific category.
|
// SearchCategory sets the query to search for pools in a specific category.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - category: The category to search for in the pools.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) SearchCategory(category model.PoolCategory) PoolsBuilder {
|
func (g *getPools) SearchCategory(category model.PoolCategory) PoolsBuilder {
|
||||||
g.query["search[category]"] = string(category)
|
g.query["search[category]"] = string(category)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Order sets the ordering of the retrieved pools.
|
// Order sets the ordering of the retrieved pools.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - order: The order to apply to the retrieved pools.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) Order(order model.PoolOrder) PoolsBuilder {
|
func (g *getPools) Order(order model.PoolOrder) PoolsBuilder {
|
||||||
g.query["search[order]"] = string(order)
|
g.query["search[order]"] = string(order)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetLimit sets the maximum number of pools to retrieve.
|
// SetLimit sets the maximum number of pools to retrieve.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - limitPools: The maximum number of pools to retrieve.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PoolsBuilder: The instance of the PoolsBuilder for method chaining.
|
||||||
func (g *getPools) SetLimit(limitPools int) PoolsBuilder {
|
func (g *getPools) SetLimit(limitPools int) PoolsBuilder {
|
||||||
g.query["limit"] = strconv.Itoa(limitPools)
|
g.query["limit"] = strconv.Itoa(limitPools)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns a slice of pools and an error, if any.
|
// Execute sends the constructed query and returns a slice of pools and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []model.Pool: A slice of pools.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getPools) Execute() ([]model.Pool, error) {
|
func (g *getPools) Execute() ([]model.Pool, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -15,23 +15,40 @@ type PostBuilder interface {
|
|||||||
Execute() (*model.Post, error)
|
Execute() (*model.Post, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetPostBuilder creates a new PostBuilder with the provided RequestContext.
|
// NewGetPostBuilder creates a new instance of PostBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostBuilder: An instance of the PostBuilder interface.
|
||||||
func NewGetPostBuilder(requestContext model.RequestContext) PostBuilder {
|
func NewGetPostBuilder(requestContext model.RequestContext) PostBuilder {
|
||||||
return &getPost{requestContext: requestContext}
|
return &getPost{requestContext: requestContext}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getPost is an implementation of the PostBuilder interface.
|
||||||
type getPost struct {
|
type getPost struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
postID model.PostID
|
postID model.PostID
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetPostID sets the post ID for the query.
|
// SetPostID sets the post ID for the query.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - postID: The ID of the post to be retrieved.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostBuilder: The instance of the PostBuilder for method chaining.
|
||||||
func (g *getPost) SetPostID(postID model.PostID) PostBuilder {
|
func (g *getPost) SetPostID(postID model.PostID) PostBuilder {
|
||||||
g.postID = postID
|
g.postID = postID
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns the requested post and an error, if any.
|
// Execute sends the constructed query and returns the requested post and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - *model.Post: The requested post.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getPost) Execute() (*model.Post, error) {
|
func (g *getPost) Execute() (*model.Post, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -24,7 +24,13 @@ type PostsBuilder interface {
|
|||||||
Execute() ([]model.Post, error)
|
Execute() ([]model.Post, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetPostsBuilder creates a new PostsBuilder with the provided RequestContext.
|
// NewGetPostsBuilder creates a new instance of PostsBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostsBuilder: An instance of the PostsBuilder interface.
|
||||||
func NewGetPostsBuilder(requestContext model.RequestContext) PostsBuilder {
|
func NewGetPostsBuilder(requestContext model.RequestContext) PostsBuilder {
|
||||||
postBuilder := &getPosts{
|
postBuilder := &getPosts{
|
||||||
requestContext: requestContext,
|
requestContext: requestContext,
|
||||||
@ -34,42 +40,77 @@ func NewGetPostsBuilder(requestContext model.RequestContext) PostsBuilder {
|
|||||||
return postBuilder.SetLimit(utils.E621_MAX_POST_COUNT)
|
return postBuilder.SetLimit(utils.E621_MAX_POST_COUNT)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getPosts is an implementation of the PostsBuilder interface.
|
||||||
type getPosts struct {
|
type getPosts struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
query map[string]string
|
query map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
// Tags sets the query to search for posts with specific tags.
|
// Tags sets the query to search for posts with specific tags.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - tags: The tags to search for in the posts.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostsBuilder: The instance of the PostsBuilder for method chaining.
|
||||||
func (g *getPosts) Tags(tags string) PostsBuilder {
|
func (g *getPosts) Tags(tags string) PostsBuilder {
|
||||||
g.query["tags"] = tags
|
g.query["tags"] = tags
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// PageAfter sets the query to retrieve posts after a specific post ID.
|
// PageAfter sets the query to retrieve posts after a specific post ID.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - postID: The ID of the post after which to retrieve other posts.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostsBuilder: The instance of the PostsBuilder for method chaining.
|
||||||
func (g *getPosts) PageAfter(postID model.PostID) PostsBuilder {
|
func (g *getPosts) PageAfter(postID model.PostID) PostsBuilder {
|
||||||
g.query["page"] = "a" + strconv.Itoa(int(postID))
|
g.query["page"] = "a" + strconv.Itoa(int(postID))
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// PageBefore sets the query to retrieve posts before a specific post ID.
|
// PageBefore sets the query to retrieve posts before a specific post ID.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - postID: The ID of the post before which to retrieve other posts.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostsBuilder: The instance of the PostsBuilder for method chaining.
|
||||||
func (g *getPosts) PageBefore(postID model.PostID) PostsBuilder {
|
func (g *getPosts) PageBefore(postID model.PostID) PostsBuilder {
|
||||||
g.query["page"] = "b" + strconv.Itoa(int(postID))
|
g.query["page"] = "b" + strconv.Itoa(int(postID))
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// PageNumber sets the query to retrieve posts on a specific page number.
|
// PageNumber sets the query to retrieve posts on a specific page number.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - number: The page number to retrieve posts from.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostsBuilder: The instance of the PostsBuilder for method chaining.
|
||||||
func (g *getPosts) PageNumber(number int) PostsBuilder {
|
func (g *getPosts) PageNumber(number int) PostsBuilder {
|
||||||
g.query["page"] = strconv.Itoa(number)
|
g.query["page"] = strconv.Itoa(number)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetLimit sets the maximum number of posts to retrieve.
|
// SetLimit sets the maximum number of posts to retrieve.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - limitUser: The maximum number of posts to retrieve.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - PostsBuilder: The instance of the PostsBuilder for method chaining.
|
||||||
func (g *getPosts) SetLimit(limitUser int) PostsBuilder {
|
func (g *getPosts) SetLimit(limitUser int) PostsBuilder {
|
||||||
g.query["limit"] = strconv.Itoa(limitUser)
|
g.query["limit"] = strconv.Itoa(limitUser)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns a slice of posts and an error, if any.
|
// Execute sends the constructed query and returns a slice of posts and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []model.Post: A slice of posts.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getPosts) Execute() ([]model.Post, error) {
|
func (g *getPosts) Execute() ([]model.Post, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -15,23 +15,40 @@ type TagBuilder interface {
|
|||||||
Execute() (model.Tag, error)
|
Execute() (model.Tag, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetTagBuilder creates a new TagBuilder with the provided RequestContext.
|
// NewGetTagBuilder creates a new instance of TagBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagBuilder: An instance of the TagBuilder interface.
|
||||||
func NewGetTagBuilder(requestContext model.RequestContext) TagBuilder {
|
func NewGetTagBuilder(requestContext model.RequestContext) TagBuilder {
|
||||||
return &getTag{requestContext: requestContext}
|
return &getTag{requestContext: requestContext}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getTag is an implementation of the TagBuilder interface.
|
||||||
type getTag struct {
|
type getTag struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
tagID int
|
tagID int
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetTagID sets the tag ID for the query.
|
// SetTagID sets the tag ID for the query.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - tagID: The ID of the tag to be retrieved.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagBuilder: The instance of the TagBuilder for method chaining.
|
||||||
func (g *getTag) SetTagID(tagID int) TagBuilder {
|
func (g *getTag) SetTagID(tagID int) TagBuilder {
|
||||||
g.tagID = tagID
|
g.tagID = tagID
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns the requested tag and an error, if any.
|
// Execute sends the constructed query and returns the requested tag and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - model.Tag: The requested tag.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getTag) Execute() (model.Tag, error) {
|
func (g *getTag) Execute() (model.Tag, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -29,65 +29,124 @@ type TagsBuilder interface {
|
|||||||
Execute() ([]model.Tag, error)
|
Execute() ([]model.Tag, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetTagsBuilder creates a new TagsBuilder with the provided RequestContext.
|
// NewGetTagsBuilder creates a new instance of TagsBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: An instance of the TagsBuilder interface.
|
||||||
func NewGetTagsBuilder(requestContext model.RequestContext) TagsBuilder {
|
func NewGetTagsBuilder(requestContext model.RequestContext) TagsBuilder {
|
||||||
return &getTags{requestContext: requestContext, query: make(map[string]string)}
|
return &getTags{requestContext: requestContext, query: make(map[string]string)}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getTags is an implementation of the TagsBuilder interface.
|
||||||
type getTags struct {
|
type getTags struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
query map[string]string
|
query map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchName sets the query to search for tags with specific names.
|
// SearchName sets the query to search for tags with specific names.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - name: The name to search for.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) SearchName(name string) TagsBuilder {
|
func (g *getTags) SearchName(name string) TagsBuilder {
|
||||||
g.query["search[name_matches]"] = name
|
g.query["search[name_matches]"] = name
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchCategory sets the query to search for tags in a specific category.
|
// SearchCategory sets the query to search for tags in a specific category.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - category: The category of tags to search for.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) SearchCategory(category model.TagCategory) TagsBuilder {
|
func (g *getTags) SearchCategory(category model.TagCategory) TagsBuilder {
|
||||||
g.query["search[category]"] = strconv.Itoa(int(category))
|
g.query["search[category]"] = strconv.Itoa(int(category))
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Order sets the query to order tags by a specific criterion.
|
// Order sets the query to order tags by a specific criterion.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - order: The criterion to order tags by.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) Order(order string) TagsBuilder {
|
func (g *getTags) Order(order string) TagsBuilder {
|
||||||
g.query["search[order]"] = order
|
g.query["search[order]"] = order
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// HideEmpty sets the query to filter out tags that are empty.
|
// HideEmpty sets the query to filter out tags that are empty.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - hideEmpty: A boolean indicating whether to filter out empty tags.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) HideEmpty(hideEmpty bool) TagsBuilder {
|
func (g *getTags) HideEmpty(hideEmpty bool) TagsBuilder {
|
||||||
g.query["search[hide_empty]"] = strconv.FormatBool(hideEmpty)
|
g.query["search[hide_empty]"] = strconv.FormatBool(hideEmpty)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Wiki sets the query to filter tags that have a wiki.
|
// Wiki sets the query to filter tags that have a wiki.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - hasWiki: A boolean indicating whether to filter tags with a wiki.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) Wiki(hasWiki bool) TagsBuilder {
|
func (g *getTags) Wiki(hasWiki bool) TagsBuilder {
|
||||||
g.query["search[has_wiki]"] = strconv.FormatBool(hasWiki)
|
g.query["search[has_wiki]"] = strconv.FormatBool(hasWiki)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Artist sets the query to filter tags that have an artist page.
|
// Artist sets the query to filter tags that have an artist page.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - hasArtistPage: A boolean indicating whether to filter tags with an artist page.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) Artist(hasArtistPage bool) TagsBuilder {
|
func (g *getTags) Artist(hasArtistPage bool) TagsBuilder {
|
||||||
g.query["search[has_artist]"] = strconv.FormatBool(hasArtistPage)
|
g.query["search[has_artist]"] = strconv.FormatBool(hasArtistPage)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetPage sets the query to retrieve tags from a specific page number.
|
// SetPage sets the query to retrieve tags from a specific page number.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - pageNumber: The page number to retrieve tags from.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) SetPage(pageNumber int) TagsBuilder {
|
func (g *getTags) SetPage(pageNumber int) TagsBuilder {
|
||||||
g.query["page"] = strconv.Itoa(pageNumber)
|
g.query["page"] = strconv.Itoa(pageNumber)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetLimit sets the maximum number of tags to retrieve.
|
// SetLimit sets the maximum number of tags to retrieve.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - limitUser: The maximum number of tags to retrieve.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - TagsBuilder: The instance of the TagsBuilder for method chaining.
|
||||||
func (g *getTags) SetLimit(limitUser int) TagsBuilder {
|
func (g *getTags) SetLimit(limitUser int) TagsBuilder {
|
||||||
g.query["limit"] = strconv.Itoa(limitUser)
|
g.query["limit"] = strconv.Itoa(limitUser)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns a slice of tags and an error, if any.
|
// Execute sends the constructed query and returns a slice of tags and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []model.Tag: A slice of tags.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getTags) Execute() ([]model.Tag, error) {
|
func (g *getTags) Execute() ([]model.Tag, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -14,23 +14,40 @@ type UserBuilder interface {
|
|||||||
Execute() (model.User, error)
|
Execute() (model.User, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewGetUserBuilder creates a new UserBuilder with the provided RequestContext.
|
// NewGetUserBuilder creates a new instance of UserBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UserBuilder: An instance of the UserBuilder interface.
|
||||||
func NewGetUserBuilder(requestContext model.RequestContext) UserBuilder {
|
func NewGetUserBuilder(requestContext model.RequestContext) UserBuilder {
|
||||||
return &getUser{requestContext: requestContext}
|
return &getUser{requestContext: requestContext}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getUser is an implementation of the UserBuilder interface.
|
||||||
type getUser struct {
|
type getUser struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
username string
|
username string
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetUsername sets the username for the query to retrieve user information.
|
// SetUsername sets the username for the query to retrieve user information.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - username: The username to retrieve information for.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UserBuilder: The instance of the UserBuilder for method chaining.
|
||||||
func (g *getUser) SetUsername(username string) UserBuilder {
|
func (g *getUser) SetUsername(username string) UserBuilder {
|
||||||
g.username = username
|
g.username = username
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns the requested user information and an error, if any.
|
// Execute sends the constructed query and returns the requested user information and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - model.User: The user information.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getUser) Execute() (model.User, error) {
|
func (g *getUser) Execute() (model.User, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -36,82 +36,159 @@ type UsersBuilder interface {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// NewGetUsersBuilder creates a new UsersBuilder with the provided RequestContext.
|
// NewGetUsersBuilder creates a new UsersBuilder with the provided RequestContext.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: An instance of the UsersBuilder interface.
|
||||||
func NewGetUsersBuilder(requestContext model.RequestContext) UsersBuilder {
|
func NewGetUsersBuilder(requestContext model.RequestContext) UsersBuilder {
|
||||||
return &getUsers{requestContext: requestContext, query: make(map[string]string)}
|
return &getUsers{requestContext: requestContext, query: make(map[string]string)}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// getUsers is an implementation of the UsersBuilder interface.
|
||||||
type getUsers struct {
|
type getUsers struct {
|
||||||
requestContext model.RequestContext
|
requestContext model.RequestContext
|
||||||
query map[string]string
|
query map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByName specifies a username to search for.
|
// SearchByName specifies a username to search for.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - userName: The username to search for.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByName(userName string) UsersBuilder {
|
func (g *getUsers) SearchByName(userName string) UsersBuilder {
|
||||||
g.query["search[name_matches]"] = userName
|
g.query["search[name_matches]"] = userName
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByAbout specifies a text to search in the 'about' field of user profiles.
|
// SearchByAbout specifies a text to search in the 'about' field of user profiles.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - about: The text to search for in the 'about' field of user profiles.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByAbout(about string) UsersBuilder {
|
func (g *getUsers) SearchByAbout(about string) UsersBuilder {
|
||||||
g.query["search[about_me]"] = about
|
g.query["search[about_me]"] = about
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByAvatarID specifies an avatar ID to search for.
|
// SearchByAvatarID specifies an avatar ID to search for.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - avatarID: The avatar ID to search for.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByAvatarID(avatarID model.PostID) UsersBuilder {
|
func (g *getUsers) SearchByAvatarID(avatarID model.PostID) UsersBuilder {
|
||||||
g.query["search[avatar_id]"] = strconv.FormatInt(int64(avatarID), 10)
|
g.query["search[avatar_id]"] = strconv.FormatInt(int64(avatarID), 10)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByLevel specifies a user level to filter by.
|
// SearchByLevel specifies a user level to filter by.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - level: The user level to filter by.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByLevel(level model.UserLevel) UsersBuilder {
|
func (g *getUsers) SearchByLevel(level model.UserLevel) UsersBuilder {
|
||||||
g.query["search[level]"] = strconv.Itoa(int(level))
|
g.query["search[level]"] = strconv.Itoa(int(level))
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByMinLevel specifies the minimum user level to filter by.
|
// SearchByMinLevel specifies the minimum user level to filter by.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - minLevel: The minimum user level to filter by.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByMinLevel(minLevel model.UserLevel) UsersBuilder {
|
func (g *getUsers) SearchByMinLevel(minLevel model.UserLevel) UsersBuilder {
|
||||||
g.query["search[min_level]"] = strconv.Itoa(int(minLevel))
|
g.query["search[min_level]"] = strconv.Itoa(int(minLevel))
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByMaxLevel specifies the maximum user level to filter by.
|
// SearchByMaxLevel specifies the maximum user level to filter by.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - maxLevel: The maximum user level to filter by.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByMaxLevel(maxLevel model.UserLevel) UsersBuilder {
|
func (g *getUsers) SearchByMaxLevel(maxLevel model.UserLevel) UsersBuilder {
|
||||||
g.query["search[max_level]"] = strconv.Itoa(int(maxLevel))
|
g.query["search[max_level]"] = strconv.Itoa(int(maxLevel))
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByCanUpload specifies whether users can upload free content.
|
// SearchByCanUpload specifies whether users can upload free content.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - canUpload: Whether users can upload free content.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByCanUpload(canUpload bool) UsersBuilder {
|
func (g *getUsers) SearchByCanUpload(canUpload bool) UsersBuilder {
|
||||||
g.query["search[can_upload_free]"] = strconv.FormatBool(canUpload)
|
g.query["search[can_upload_free]"] = strconv.FormatBool(canUpload)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByIsApprover specifies whether users can approve posts.
|
// SearchByIsApprover specifies whether users can approve posts.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - isApprover: Whether users can approve posts.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByIsApprover(isApprover bool) UsersBuilder {
|
func (g *getUsers) SearchByIsApprover(isApprover bool) UsersBuilder {
|
||||||
g.query["search[can_approve_posts]"] = strconv.FormatBool(isApprover)
|
g.query["search[can_approve_posts]"] = strconv.FormatBool(isApprover)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SearchByOrder specifies the order in which users are retrieved.
|
// SearchByOrder specifies the order in which users are retrieved.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - order: The order in which users are retrieved.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SearchByOrder(order model.Order) UsersBuilder {
|
func (g *getUsers) SearchByOrder(order model.Order) UsersBuilder {
|
||||||
g.query["search[order]"] = string(order)
|
g.query["search[order]"] = string(order)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetPage sets the page number for paginated results.
|
// SetPage sets the page number for paginated results.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - pageNumber: The page number for paginated results.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SetPage(pageNumber int) UsersBuilder {
|
func (g *getUsers) SetPage(pageNumber int) UsersBuilder {
|
||||||
g.query["page"] = strconv.Itoa(pageNumber)
|
g.query["page"] = strconv.Itoa(pageNumber)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetLimit sets the maximum number of users to retrieve.
|
// SetLimit sets the maximum number of users to retrieve.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - limitUser: The maximum number of users to retrieve.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - UsersBuilder: The instance of the UsersBuilder for method chaining.
|
||||||
func (g *getUsers) SetLimit(limitUser int) UsersBuilder {
|
func (g *getUsers) SetLimit(limitUser int) UsersBuilder {
|
||||||
g.query["limit"] = strconv.Itoa(limitUser)
|
g.query["limit"] = strconv.Itoa(limitUser)
|
||||||
return g
|
return g
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute sends the constructed query and returns the requested user information and an error, if any.
|
// Execute sends the constructed query and returns the requested user information and an error, if any.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - []model.User: A slice of user information.
|
||||||
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func (g *getUsers) Execute() ([]model.User, error) {
|
func (g *getUsers) Execute() ([]model.User, error) {
|
||||||
if g.requestContext.RateLimiter != nil {
|
if g.requestContext.RateLimiter != nil {
|
||||||
err := g.requestContext.RateLimiter.Wait(context.Background())
|
err := g.requestContext.RateLimiter.Wait(context.Background())
|
||||||
|
@ -1,15 +1,23 @@
|
|||||||
package e621
|
package e621
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"compress/gzip"
|
||||||
|
"encoding/csv"
|
||||||
"fmt"
|
"fmt"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/builder"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/builder"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
||||||
|
"github.com/gocarina/gocsv"
|
||||||
_ "github.com/joho/godotenv/autoload"
|
_ "github.com/joho/godotenv/autoload"
|
||||||
"golang.org/x/time/rate"
|
"golang.org/x/time/rate"
|
||||||
|
"log"
|
||||||
"math"
|
"math"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Client is the main client for interacting with the e621 API.
|
// Client is the main client for interacting with the e621 API.
|
||||||
@ -242,3 +250,132 @@ func (c *Client) GetAllPosts(postBuilder builder.PostsBuilder) ([]model.Post, er
|
|||||||
// Retrieves all available posts using the provided post builder.
|
// Retrieves all available posts using the provided post builder.
|
||||||
return c.GetNPosts(math.MaxInt, postBuilder)
|
return c.GetNPosts(math.MaxInt, postBuilder)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *Client) GetLatestPoolsDBExportDataAsBytes() (string, []byte, error) {
|
||||||
|
dbExportFileNameList, err := builder.NewGetDBExportListBuilder(c.RequestContext).Execute()
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
getDBExportFile := builder.NewGetDBExportFileBuilder(c.RequestContext)
|
||||||
|
|
||||||
|
filter := func(s string) bool { return strings.HasPrefix(s, "pools") }
|
||||||
|
filteredFileNameList := utils.SliceFilter(dbExportFileNameList, filter)
|
||||||
|
|
||||||
|
regex, err := regexp.Compile("\\d{4}\\-(0?[1-9]|1[012])\\-(0?[1-9]|[12][0-9]|3[01])*")
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
currentDate, err := time.Parse("2006-04-02", time.Now().Format("2006-01-02"))
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
duration := math.MaxFloat64
|
||||||
|
var fileName string
|
||||||
|
for _, listFileName := range filteredFileNameList {
|
||||||
|
if !regex.MatchString(listFileName) {
|
||||||
|
return "", nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
fileDate, err := time.Parse("2006-04-02", regex.FindString(listFileName))
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if currentDate.Sub(fileDate).Seconds() < duration {
|
||||||
|
duration = currentDate.Sub(fileDate).Seconds()
|
||||||
|
fileName = listFileName
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
rawFile, err := getDBExportFile.SetFile(fileName).Execute()
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return fileName, rawFile, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) GetLatestPoolsDBExportDataAsStruct() ([]*model.Pool, error) {
|
||||||
|
var pools []*model.Pool
|
||||||
|
|
||||||
|
_, data, err := c.GetLatestPoolsDBExportDataAsBytes()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
zipReader, err := gzip.NewReader(bytes.NewReader(data))
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer zipReader.Close()
|
||||||
|
|
||||||
|
// Create a CSV reader
|
||||||
|
reader := csv.NewReader(zipReader)
|
||||||
|
|
||||||
|
err = gocsv.UnmarshalCSV(reader, &pools)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return pools, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) GetLatestPostsDBExportDataAsBytes() (string, []byte, error) {
|
||||||
|
log.Println("Please wait while the download is in progress for the post export... (file over 1GB) ")
|
||||||
|
dbExportFileNameList, err := builder.NewGetDBExportListBuilder(c.RequestContext).Execute()
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
getDBExportFile := builder.NewGetDBExportFileBuilder(c.RequestContext)
|
||||||
|
|
||||||
|
filter := func(s string) bool { return strings.HasPrefix(s, "posts") }
|
||||||
|
filteredFileNameList := utils.SliceFilter(dbExportFileNameList, filter)
|
||||||
|
|
||||||
|
regex, err := regexp.Compile("\\d{4}\\-(0?[1-9]|1[012])\\-(0?[1-9]|[12][0-9]|3[01])*")
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
currentDate, err := time.Parse("2006-04-02", time.Now().Format("2006-01-02"))
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
duration := math.MaxFloat64
|
||||||
|
var fileName string
|
||||||
|
for _, listFileName := range filteredFileNameList {
|
||||||
|
if !regex.MatchString(listFileName) {
|
||||||
|
return "", nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
fileDate, err := time.Parse("2006-04-02", regex.FindString(listFileName))
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if currentDate.Sub(fileDate).Seconds() < duration {
|
||||||
|
duration = currentDate.Sub(fileDate).Seconds()
|
||||||
|
fileName = listFileName
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
rawFile, err := getDBExportFile.SetFile(fileName).Execute()
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return fileName, rawFile, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) GetLatestPostsDBExportDataAsStruct() ([]*model.Post, error) {
|
||||||
|
var post []*model.Post
|
||||||
|
|
||||||
|
// TODO: Implement this function; tags in CSV are just a string with no categories assignment, needs special parsing
|
||||||
|
|
||||||
|
return post, nil
|
||||||
|
}
|
||||||
|
@ -5,6 +5,7 @@ import (
|
|||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
||||||
"golang.org/x/net/html"
|
"golang.org/x/net/html"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
@ -15,7 +16,7 @@ import (
|
|||||||
// the HTML content to extract the links to export files with the ".csv.gz" extension.
|
// the HTML content to extract the links to export files with the ".csv.gz" extension.
|
||||||
//
|
//
|
||||||
// Parameters:
|
// Parameters:
|
||||||
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
// - model.RequestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
//
|
//
|
||||||
// Returns:
|
// Returns:
|
||||||
// - []string: A slice of file names with the ".csv.gz" extension.
|
// - []string: A slice of file names with the ".csv.gz" extension.
|
||||||
@ -79,13 +80,13 @@ func GetDBExportList(requestContext model.RequestContext) ([]string, error) {
|
|||||||
// particular file identified by its name.
|
// particular file identified by its name.
|
||||||
//
|
//
|
||||||
// Parameters:
|
// Parameters:
|
||||||
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
// - model.RequestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
// - file: The name of the file to be fetched from the database export.
|
// - file: The name of the file to be fetched from the database export.
|
||||||
//
|
//
|
||||||
// Returns:
|
// Returns:
|
||||||
// - *http.Response: The HTTP response containing the requested file (probably a csv.gz).
|
// - []byte: A byte array representation of the file.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetDBExportFile(requestContext model.RequestContext, file string) (*http.Response, error) {
|
func GetDBExportFile(requestContext model.RequestContext, file string) ([]byte, error) {
|
||||||
if file == "" {
|
if file == "" {
|
||||||
return nil, fmt.Errorf("no file specified")
|
return nil, fmt.Errorf("no file specified")
|
||||||
}
|
}
|
||||||
@ -112,6 +113,6 @@ func GetDBExportFile(requestContext model.RequestContext, file string) (*http.Re
|
|||||||
return nil, utils.StatusCodesToError(resp.StatusCode)
|
return nil, utils.StatusCodesToError(resp.StatusCode)
|
||||||
}
|
}
|
||||||
|
|
||||||
return resp, nil
|
return io.ReadAll(resp.Body)
|
||||||
|
|
||||||
}
|
}
|
89
pkg/e621/endpoints/executer.go
Normal file
89
pkg/e621/endpoints/executer.go
Normal file
@ -0,0 +1,89 @@
|
|||||||
|
package endpoints
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RequestTypes interface {
|
||||||
|
model.User |
|
||||||
|
model.Tag |
|
||||||
|
model.Post |
|
||||||
|
model.PostResponse |
|
||||||
|
model.Pool |
|
||||||
|
model.Note |
|
||||||
|
[]model.User |
|
||||||
|
[]model.Tag |
|
||||||
|
[]model.Post |
|
||||||
|
[]model.Pool |
|
||||||
|
[]model.Note
|
||||||
|
}
|
||||||
|
|
||||||
|
// getRequest performs an HTTP GET request to the specified e621 API endpoint and unmarshals the JSON response into the provided type T.
|
||||||
|
//
|
||||||
|
// This generic function is designed to fetch data from the e621 API using an HTTP GET request.
|
||||||
|
// It supports generic types and can unmarshal the JSON response into the provided type T.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - requestContext: The context for the API request, including the host, user agent, username, and API key.
|
||||||
|
// - e621Endpoint: The specific e621 API endpoint to be called.
|
||||||
|
// - query: A map containing additional query parameters for the API request. (Optional)
|
||||||
|
//
|
||||||
|
// Type Parameters:
|
||||||
|
// - T: The type into which the JSON response will be unmarshaled.
|
||||||
|
// - supported types are (also as slice): model.User, model.Tag, model.Post, model.PostResponse, model.Pool, model.Note
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - T: The result of unmarshaling the JSON response into the provided type.
|
||||||
|
// - error: An error, if any, encountered during the API request, response handling, or JSON unmarshaling.
|
||||||
|
func getRequest[T RequestTypes](requestContext model.RequestContext, e621Endpoint string, query map[string]string) (T, error) {
|
||||||
|
var base T
|
||||||
|
// Create a new HTTP GET request to fetch user information from the specified 'host' and 'username'.
|
||||||
|
r, err := http.NewRequest("GET", fmt.Sprintf("%s/%s", requestContext.Host, e621Endpoint), nil)
|
||||||
|
if err != nil {
|
||||||
|
// Log the error and return an empty User struct and the error.
|
||||||
|
|
||||||
|
return base, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Append query parameters to the request URL.
|
||||||
|
if query != nil {
|
||||||
|
q := r.URL.Query()
|
||||||
|
for k, v := range query {
|
||||||
|
q.Add(k, v)
|
||||||
|
}
|
||||||
|
r.URL.RawQuery = q.Encode()
|
||||||
|
}
|
||||||
|
|
||||||
|
r.Header.Set("User-Agent", requestContext.UserAgent)
|
||||||
|
r.Header.Add("Accept", "application/json")
|
||||||
|
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
||||||
|
|
||||||
|
// Send the request using the provided http.Client.
|
||||||
|
resp, err := requestContext.Client.Do(r)
|
||||||
|
if err != nil {
|
||||||
|
// Log the error and return an empty User struct and the error.
|
||||||
|
|
||||||
|
return base, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if the HTTP response status code indicates success (2xx range).
|
||||||
|
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
||||||
|
// If the status code is outside the 2xx range, return an error based on the status code.
|
||||||
|
return base, utils.StatusCodesToError(resp.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode the JSON response into the User struct.
|
||||||
|
err = json.NewDecoder(resp.Body).Decode(&base)
|
||||||
|
if err != nil {
|
||||||
|
// Log the error and return an empty User struct and the error.
|
||||||
|
|
||||||
|
return base, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Return the user information and no error (nil).
|
||||||
|
return base, nil
|
||||||
|
}
|
@ -1,11 +1,7 @@
|
|||||||
package endpoints
|
package endpoints
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
|
||||||
"net/http"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetFavorites retrieves a user's favorite posts from the e621 API.
|
// GetFavorites retrieves a user's favorite posts from the e621 API.
|
||||||
@ -20,45 +16,7 @@ import (
|
|||||||
// - []model.Post: A slice of favorite posts.
|
// - []model.Post: A slice of favorite posts.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetFavorites(requestContext model.RequestContext, query map[string]string) ([]model.Post, error) {
|
func GetFavorites(requestContext model.RequestContext, query map[string]string) ([]model.Post, error) {
|
||||||
// Create a new HTTP GET request.
|
endpoint := "favorites.json"
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/favorites.json", requestContext.Host), nil)
|
data, err := getRequest[model.PostResponse](requestContext, endpoint, query)
|
||||||
if err != nil {
|
return data.Posts, err
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Append query parameters to the request URL.
|
|
||||||
q := r.URL.Query()
|
|
||||||
for k, v := range query {
|
|
||||||
q.Add(k, v)
|
|
||||||
}
|
|
||||||
r.URL.RawQuery = q.Encode()
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return nil, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a User struct to store the response data.
|
|
||||||
var favoriteResponse model.PostResponse
|
|
||||||
|
|
||||||
// Decode the JSON response into the user struct.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&favoriteResponse)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty User struct and the error.
|
|
||||||
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return favoriteResponse.Posts, nil
|
|
||||||
}
|
}
|
||||||
|
@ -1,13 +1,8 @@
|
|||||||
package endpoints
|
package endpoints
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"strings"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetNote retrieves a single note by its ID from the e621 API.
|
// GetNote retrieves a single note by its ID from the e621 API.
|
||||||
@ -20,45 +15,8 @@ import (
|
|||||||
// - model.Note: The retrieved note.
|
// - model.Note: The retrieved note.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetNote(requestContext model.RequestContext, ID string) (model.Note, error) {
|
func GetNote(requestContext model.RequestContext, ID string) (model.Note, error) {
|
||||||
// Create a new HTTP GET request to fetch the note information.
|
endpoint := fmt.Sprintf("notes/%s.json", ID)
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/notes/%s.json", requestContext.Host, ID), nil)
|
return getRequest[model.Note](requestContext, endpoint, nil)
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Note struct and the error.
|
|
||||||
|
|
||||||
return model.Note{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Note struct and the error.
|
|
||||||
|
|
||||||
return model.Note{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return model.Note{}, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a Note struct to store the response data.
|
|
||||||
var noteResponse model.Note
|
|
||||||
|
|
||||||
// Decode the JSON response into the Note struct.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(¬eResponse)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Note struct and the error.
|
|
||||||
|
|
||||||
return model.Note{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the note information and no error (nil).
|
|
||||||
return noteResponse, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNotes retrieves a list of notes from the e621 API based on query parameters.
|
// GetNotes retrieves a list of notes from the e621 API based on query parameters.
|
||||||
@ -71,51 +29,6 @@ func GetNote(requestContext model.RequestContext, ID string) (model.Note, error)
|
|||||||
// - []model.Note: A slice of notes.
|
// - []model.Note: A slice of notes.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetNotes(requestContext model.RequestContext, query map[string]string) ([]model.Note, error) {
|
func GetNotes(requestContext model.RequestContext, query map[string]string) ([]model.Note, error) {
|
||||||
// Create a new HTTP GET request.
|
endpoint := "notes.json"
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/notes.json", requestContext.Host), nil)
|
return getRequest[[]model.Note](requestContext, endpoint, query)
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Append query parameters to the request URL.
|
|
||||||
q := r.URL.Query()
|
|
||||||
for k, v := range query {
|
|
||||||
q.Add(k, v)
|
|
||||||
}
|
|
||||||
r.URL.RawQuery = q.Encode()
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return nil, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
respBodyBytes, err := io.ReadAll(resp.Body)
|
|
||||||
if strings.Contains(string(respBodyBytes), "{\"notes\":[]}") {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a slice of Note struct to store the response data.
|
|
||||||
var notesResponse []model.Note
|
|
||||||
|
|
||||||
// Decode the JSON response into the slice of Note structs.
|
|
||||||
err = json.Unmarshal(respBodyBytes, ¬esResponse)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty slice and the error.
|
|
||||||
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the list of notes and no error (nil).
|
|
||||||
return notesResponse, nil
|
|
||||||
}
|
}
|
||||||
|
@ -1,11 +1,8 @@
|
|||||||
package endpoints
|
package endpoints
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
|
||||||
"net/http"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetPool retrieves a pool by its ID from the e621 API.
|
// GetPool retrieves a pool by its ID from the e621 API.
|
||||||
@ -18,45 +15,8 @@ import (
|
|||||||
// - model.Pool: The retrieved pool.
|
// - model.Pool: The retrieved pool.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetPool(requestContext model.RequestContext, ID string) (model.Pool, error) {
|
func GetPool(requestContext model.RequestContext, ID string) (model.Pool, error) {
|
||||||
// Create a new HTTP GET request to fetch the pool information.
|
endpoint := fmt.Sprintf("pools/%s.json", ID)
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/pools/%s.json", requestContext.Host, ID), nil)
|
return getRequest[model.Pool](requestContext, endpoint, nil)
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Pool struct and the error.
|
|
||||||
|
|
||||||
return model.Pool{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Pool struct and the error.
|
|
||||||
|
|
||||||
return model.Pool{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return model.Pool{}, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a Pool struct to store the response data.
|
|
||||||
var poolResponse model.Pool
|
|
||||||
|
|
||||||
// Decode the JSON response into the Pool struct.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&poolResponse)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Pool struct and the error.
|
|
||||||
|
|
||||||
return model.Pool{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the pool information and no error (nil).
|
|
||||||
return poolResponse, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetPools retrieves a list of pools from the e621 API based on query parameters.
|
// GetPools retrieves a list of pools from the e621 API based on query parameters.
|
||||||
@ -69,46 +29,6 @@ func GetPool(requestContext model.RequestContext, ID string) (model.Pool, error)
|
|||||||
// - []model.Pool: A slice of pools.
|
// - []model.Pool: A slice of pools.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetPools(requestContext model.RequestContext, query map[string]string) ([]model.Pool, error) {
|
func GetPools(requestContext model.RequestContext, query map[string]string) ([]model.Pool, error) {
|
||||||
// Create a new HTTP GET request.
|
endpoint := "pools.json"
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/pools.json", requestContext.Host), nil)
|
return getRequest[[]model.Pool](requestContext, endpoint, query)
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Append query parameters to the request URL.
|
|
||||||
q := r.URL.Query()
|
|
||||||
for k, v := range query {
|
|
||||||
q.Add(k, v)
|
|
||||||
}
|
|
||||||
r.URL.RawQuery = q.Encode()
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return nil, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a slice of Pool struct to store the response data.
|
|
||||||
var poolsResponse []model.Pool
|
|
||||||
|
|
||||||
// Decode the JSON response into the slice of Pool structs.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&poolsResponse)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty slice and the error.
|
|
||||||
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the list of pools and no error (nil).
|
|
||||||
return poolsResponse, nil
|
|
||||||
}
|
}
|
||||||
|
@ -1,11 +1,8 @@
|
|||||||
package endpoints
|
package endpoints
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
|
||||||
"net/http"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetPost retrieves a single post by its ID from the e621 API.
|
// GetPost retrieves a single post by its ID from the e621 API.
|
||||||
@ -17,45 +14,8 @@ import (
|
|||||||
// - model.Post: The retrieved post.
|
// - model.Post: The retrieved post.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetPost(requestContext model.RequestContext, ID string) (model.Post, error) {
|
func GetPost(requestContext model.RequestContext, ID string) (model.Post, error) {
|
||||||
// Create a new HTTP GET request to fetch the post information.
|
endpoint := fmt.Sprintf("posts/%s.json", ID)
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/posts/%s.json", requestContext.Host, ID), nil)
|
return getRequest[model.Post](requestContext, endpoint, nil)
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Post struct and the error.
|
|
||||||
|
|
||||||
return model.Post{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Post struct and the error.
|
|
||||||
|
|
||||||
return model.Post{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return model.Post{}, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a Post struct to store the response data.
|
|
||||||
var postResponse model.PostResponse
|
|
||||||
|
|
||||||
// Decode the JSON response into the PostResponse struct.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&postResponse)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Post struct and the error.
|
|
||||||
|
|
||||||
return model.Post{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the post information and no error (nil).
|
|
||||||
return postResponse.Post, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetPosts retrieves a list of posts from the e621 API based on query parameters.
|
// GetPosts retrieves a list of posts from the e621 API based on query parameters.
|
||||||
@ -68,46 +28,7 @@ func GetPost(requestContext model.RequestContext, ID string) (model.Post, error)
|
|||||||
// - []model.Post: A slice of posts.
|
// - []model.Post: A slice of posts.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetPosts(requestContext model.RequestContext, query map[string]string) ([]model.Post, error) {
|
func GetPosts(requestContext model.RequestContext, query map[string]string) ([]model.Post, error) {
|
||||||
// Create a new HTTP GET request.
|
endpoint := "posts.json"
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/posts.json", requestContext.Host), nil)
|
data, err := getRequest[model.PostResponse](requestContext, endpoint, query)
|
||||||
if err != nil {
|
return data.Posts, err
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Append query parameters to the request URL.
|
|
||||||
q := r.URL.Query()
|
|
||||||
for k, v := range query {
|
|
||||||
q.Add(k, v)
|
|
||||||
}
|
|
||||||
r.URL.RawQuery = q.Encode()
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return nil, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a slice of Post struct to store the response data.
|
|
||||||
var postResponse model.PostResponse
|
|
||||||
|
|
||||||
// Decode the JSON response into the PostResponse struct.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&postResponse)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty slice and the error.
|
|
||||||
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the list of posts and no error (nil).
|
|
||||||
return postResponse.Posts, nil
|
|
||||||
}
|
}
|
||||||
|
@ -1,11 +1,8 @@
|
|||||||
package endpoints
|
package endpoints
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
|
||||||
"net/http"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetTag retrieves a tag by its ID from the e621 API.
|
// GetTag retrieves a tag by its ID from the e621 API.
|
||||||
@ -18,46 +15,8 @@ import (
|
|||||||
// - model.Tag: The retrieved tag.
|
// - model.Tag: The retrieved tag.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetTag(requestContext model.RequestContext, ID string) (model.Tag, error) {
|
func GetTag(requestContext model.RequestContext, ID string) (model.Tag, error) {
|
||||||
// Create a new HTTP GET request to fetch tag information.
|
endpoint := fmt.Sprintf("tags/%s.json", ID)
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/tags/%s.json", requestContext.Host, ID), nil)
|
return getRequest[model.Tag](requestContext, endpoint, nil)
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Tag struct and the error.
|
|
||||||
|
|
||||||
return model.Tag{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application.json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Tag struct and the error.
|
|
||||||
|
|
||||||
return model.Tag{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return model.Tag{}, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a Tag struct to store the response data.
|
|
||||||
var tag model.Tag
|
|
||||||
|
|
||||||
// Decode the JSON response into the Tag struct.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&tag)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty Tag struct and the error.
|
|
||||||
|
|
||||||
return model.Tag{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the tag information and no error (nil).
|
|
||||||
return tag, nil
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetTags retrieves a list of tags from the e621 API based on query parameters.
|
// GetTags retrieves a list of tags from the e621 API based on query parameters.
|
||||||
@ -80,46 +39,6 @@ func GetTag(requestContext model.RequestContext, ID string) (model.Tag, error) {
|
|||||||
// - []model.Tag: A slice of tags.
|
// - []model.Tag: A slice of tags.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetTags(requestContext model.RequestContext, query map[string]string) ([]model.Tag, error) {
|
func GetTags(requestContext model.RequestContext, query map[string]string) ([]model.Tag, error) {
|
||||||
// Create a new HTTP GET request.
|
endpoint := "tags.json"
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/tags.json", requestContext.Host), nil)
|
return getRequest[[]model.Tag](requestContext, endpoint, query)
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Append query parameters to the request URL.
|
|
||||||
q := r.URL.Query()
|
|
||||||
for k, v := range query {
|
|
||||||
q.Add(k, v)
|
|
||||||
}
|
|
||||||
r.URL.RawQuery = q.Encode()
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return []model.Tag{}, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a slice of Tag struct to store the response data.
|
|
||||||
var tags []model.Tag
|
|
||||||
|
|
||||||
// Decode the JSON response into the slice of Tag structs.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&tags)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty slice and the error.
|
|
||||||
|
|
||||||
return []model.Tag{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the list of tags and no error (nil).
|
|
||||||
return tags, nil
|
|
||||||
}
|
}
|
||||||
|
@ -1,11 +1,8 @@
|
|||||||
package endpoints
|
package endpoints
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
|
||||||
"net/http"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetUser retrieves user information from e621.net based on the provided username.
|
// GetUser retrieves user information from e621.net based on the provided username.
|
||||||
@ -18,45 +15,8 @@ import (
|
|||||||
// - model.User: The retrieved user.
|
// - model.User: The retrieved user.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetUser(requestContext model.RequestContext, username string) (model.User, error) {
|
func GetUser(requestContext model.RequestContext, username string) (model.User, error) {
|
||||||
// Create a new HTTP GET request to fetch user information from the specified 'host' and 'username'.
|
endpoint := fmt.Sprintf("users/%s.json", username)
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/users/%s.json", requestContext.Host, username), nil)
|
return getRequest[model.User](requestContext, endpoint, nil)
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty User struct and the error.
|
|
||||||
|
|
||||||
return model.User{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty User struct and the error.
|
|
||||||
|
|
||||||
return model.User{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return model.User{}, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a User struct to store the response data.
|
|
||||||
var user model.User
|
|
||||||
|
|
||||||
// Decode the JSON response into the User struct.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&user)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty User struct and the error.
|
|
||||||
|
|
||||||
return model.User{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the user information and no error (nil).
|
|
||||||
return user, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetUsers retrieves a list of users from e621.net based on query parameters.
|
// GetUsers retrieves a list of users from e621.net based on query parameters.
|
||||||
@ -69,46 +29,6 @@ func GetUser(requestContext model.RequestContext, username string) (model.User,
|
|||||||
// - []model.User: A slice of users.
|
// - []model.User: A slice of users.
|
||||||
// - error: An error, if any, encountered during the API request or response handling.
|
// - error: An error, if any, encountered during the API request or response handling.
|
||||||
func GetUsers(requestContext model.RequestContext, query map[string]string) ([]model.User, error) {
|
func GetUsers(requestContext model.RequestContext, query map[string]string) ([]model.User, error) {
|
||||||
// Create a new HTTP GET request.
|
endpoint := "users.json"
|
||||||
r, err := http.NewRequest("GET", fmt.Sprintf("%s/users.json", requestContext.Host), nil)
|
return getRequest[[]model.User](requestContext, endpoint, query)
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Append query parameters to the request URL.
|
|
||||||
q := r.URL.Query()
|
|
||||||
for k, v := range query {
|
|
||||||
q.Add(k, v)
|
|
||||||
}
|
|
||||||
r.URL.RawQuery = q.Encode()
|
|
||||||
|
|
||||||
r.Header.Set("User-Agent", requestContext.UserAgent)
|
|
||||||
r.Header.Add("Accept", "application/json")
|
|
||||||
r.SetBasicAuth(requestContext.Username, requestContext.APIKey)
|
|
||||||
|
|
||||||
// Send the request using the provided http.Client.
|
|
||||||
resp, err := requestContext.Client.Do(r)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the HTTP response status code indicates success (2xx range).
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 300 {
|
|
||||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
|
||||||
return []model.User{}, utils.StatusCodesToError(resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize a slice of User struct to store the response data.
|
|
||||||
var users []model.User
|
|
||||||
|
|
||||||
// Decode the JSON response into the slice of User structs.
|
|
||||||
err = json.NewDecoder(resp.Body).Decode(&users)
|
|
||||||
if err != nil {
|
|
||||||
// Log the error and return an empty slice and the error.
|
|
||||||
|
|
||||||
return []model.User{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Return the list of users and no error (nil).
|
|
||||||
return users, nil
|
|
||||||
}
|
}
|
||||||
|
@ -1,7 +1,14 @@
|
|||||||
package model
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
type PoolCategory string
|
type PoolCategory string
|
||||||
type PoolOrder string
|
type PoolOrder string
|
||||||
|
type PoolIDs []int64
|
||||||
|
|
||||||
const (
|
const (
|
||||||
Series PoolCategory = "series"
|
Series PoolCategory = "series"
|
||||||
@ -16,15 +23,52 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Pool struct {
|
type Pool struct {
|
||||||
ID int64 `json:"id"`
|
ID int64 `json:"id" csv:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name" csv:"name"`
|
||||||
CreatedAt string `json:"created_at"`
|
CreatedAt string `json:"created_at" csv:"created_at"`
|
||||||
UpdatedAt string `json:"updated_at"`
|
UpdatedAt string `json:"updated_at" csv:"updated_at"`
|
||||||
CreatorID int64 `json:"creator_id"`
|
CreatorID int64 `json:"creator_id" csv:"creator_id"`
|
||||||
Description string `json:"description"`
|
Description string `json:"description" csv:"description"`
|
||||||
IsActive bool `json:"is_active"`
|
IsActive bool `json:"is_active" csv:"is_active"`
|
||||||
Category PoolCategory `json:"category"`
|
Category PoolCategory `json:"category" csv:"category"`
|
||||||
PostIDS []int64 `json:"post_ids"`
|
PostIDS PoolIDs `json:"post_ids" csv:"post_ids"`
|
||||||
CreatorName string `json:"creator_name"`
|
CreatorName string `json:"creator_name" csv:"-"`
|
||||||
PostCount int64 `json:"post_count"`
|
PostCount int64 `json:"post_count" csv:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnmarshalCSV parses a CSV-formatted string containing pool IDs and populates the PoolIDs receiver.
|
||||||
|
//
|
||||||
|
// This method is designed to unmarshal a CSV-formatted string, where pool IDs are separated by commas.
|
||||||
|
// It trims the surrounding curly braces and splits the string to extract individual pool IDs.
|
||||||
|
// The parsed IDs are then converted to int64 and assigned to the PoolIDs receiver.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - csv: The CSV-formatted string containing pool IDs.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - error: An error encountered during the unmarshaling process, if any.
|
||||||
|
func (poolIDs *PoolIDs) UnmarshalCSV(csv string) error {
|
||||||
|
// Trim the surrounding curly braces
|
||||||
|
csv = strings.TrimPrefix(csv, "{")
|
||||||
|
csv = strings.TrimSuffix(csv, "}")
|
||||||
|
|
||||||
|
// Split the CSV string into individual pool IDs
|
||||||
|
ids := strings.Split(csv, ",")
|
||||||
|
|
||||||
|
var localPoolIDs PoolIDs
|
||||||
|
|
||||||
|
// Iterate through each ID, parse it to int64, and append to the localPoolIDs
|
||||||
|
for _, id := range ids {
|
||||||
|
if id != "" {
|
||||||
|
int64ID, err := strconv.ParseInt(id, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to parse pool ID '%s': %v", id, err)
|
||||||
|
}
|
||||||
|
localPoolIDs = append(localPoolIDs, int64ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Assign the parsed IDs to the receiver
|
||||||
|
*poolIDs = localPoolIDs
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -8,86 +8,86 @@ type PostResponse struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Post struct {
|
type Post struct {
|
||||||
ID PostID `json:"id"`
|
ID PostID `json:"id" csv:"id"`
|
||||||
CreatedAt string `json:"created_at"`
|
CreatedAt string `json:"created_at" csv:"created_at"`
|
||||||
UpdatedAt string `json:"updated_at"`
|
UpdatedAt string `json:"updated_at" csv:"updated_at"`
|
||||||
File File `json:"file"`
|
File File `json:"file" csv:"file"`
|
||||||
Preview Preview `json:"preview"`
|
Preview Preview `json:"preview" csv:"-"`
|
||||||
Sample Sample `json:"sample"`
|
Sample Sample `json:"sample" csv:"-"`
|
||||||
Score Score `json:"score"`
|
Score Score `json:"score" csv:"score"`
|
||||||
Tags Tags `json:"tags"`
|
Tags Tags `json:"tags" csv:"tag_string"`
|
||||||
LockedTags []interface{} `json:"locked_tags"`
|
LockedTags []interface{} `json:"locked_tags" csv:"locked_tags"`
|
||||||
ChangeSeq int64 `json:"change_seq"`
|
ChangeSeq int64 `json:"change_seq" csv:"change_seq"`
|
||||||
Flags Flags `json:"flags"`
|
Flags Flags `json:"flags" csv:"-"`
|
||||||
Rating string `json:"rating"`
|
Rating string `json:"rating" csv:"rating"`
|
||||||
FavCount int64 `json:"fav_count"`
|
FavCount int64 `json:"fav_count" csv:"fav_count"`
|
||||||
Sources []string `json:"sources"`
|
Sources []string `json:"sources" csv:"source"`
|
||||||
Pools []interface{} `json:"pools"`
|
Pools []interface{} `json:"pools" csv:"-"`
|
||||||
Relationships Relationships `json:"relationships"`
|
Relationships Relationships `json:"relationships" csv:"-"`
|
||||||
ApproverID interface{} `json:"approver_id"`
|
ApproverID UserID `json:"approver_id" csv:"approver_id"`
|
||||||
UploaderID int64 `json:"uploader_id"`
|
UploaderID UserID `json:"uploader_id" csv:"uploader_id"`
|
||||||
Description string `json:"description"`
|
Description string `json:"description" csv:"description"`
|
||||||
CommentCount int64 `json:"comment_count"`
|
CommentCount int64 `json:"comment_count" csv:"comment_count"`
|
||||||
IsFavorited bool `json:"is_favorited"`
|
IsFavorited bool `json:"is_favorited" csv:"-"`
|
||||||
HasNotes bool `json:"has_notes"`
|
HasNotes bool `json:"has_notes" csv:"-"`
|
||||||
Duration interface{} `json:"duration"`
|
Duration interface{} `json:"duration" csv:"duration"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type File struct {
|
type File struct {
|
||||||
Width int64 `json:"width"`
|
Width int64 `json:"width" csv:"image_width"`
|
||||||
Height int64 `json:"height"`
|
Height int64 `json:"height" csv:"image_height"`
|
||||||
EXT string `json:"ext"`
|
EXT string `json:"ext" csv:"file_ext"`
|
||||||
Size int64 `json:"size"`
|
Size int64 `json:"size" csv:"file_size"`
|
||||||
Md5 string `json:"md5"`
|
Md5 string `json:"md5" csv:"md5"`
|
||||||
URL string `json:"url"`
|
URL string `json:"url" csv:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Flags struct {
|
type Flags struct {
|
||||||
Pending bool `json:"pending"`
|
Pending bool `json:"pending" csv:"is_pending"`
|
||||||
Flagged bool `json:"flagged"`
|
Flagged bool `json:"flagged" csv:"is_flagged"`
|
||||||
NoteLocked bool `json:"note_locked"`
|
NoteLocked bool `json:"note_locked" csv:"is_note_locked"`
|
||||||
StatusLocked bool `json:"status_locked"`
|
StatusLocked bool `json:"status_locked" csv:"is_status_locked"`
|
||||||
RatingLocked bool `json:"rating_locked"`
|
RatingLocked bool `json:"rating_locked" csv:"is_rating_locked"`
|
||||||
Deleted bool `json:"deleted"`
|
Deleted bool `json:"deleted" csv:"is_deleted"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Preview struct {
|
type Preview struct {
|
||||||
Width int64 `json:"width"`
|
Width int64 `json:"width" csv:"-"`
|
||||||
Height int64 `json:"height"`
|
Height int64 `json:"height" csv:"-"`
|
||||||
URL string `json:"url"`
|
URL string `json:"url" csv:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Relationships struct {
|
type Relationships struct {
|
||||||
ParentID interface{} `json:"parent_id"`
|
ParentID PostID `json:"parent_id" csv:"parent_id"`
|
||||||
HasChildren bool `json:"has_children"`
|
HasChildren bool `json:"has_children" csv:"-"`
|
||||||
HasActiveChildren bool `json:"has_active_children"`
|
HasActiveChildren bool `json:"has_active_children" csv:"-"`
|
||||||
Children []interface{} `json:"children"`
|
Children []PostID `json:"children" csv:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Sample struct {
|
type Sample struct {
|
||||||
Has bool `json:"has"`
|
Has bool `json:"has" csv:"-"`
|
||||||
Height int64 `json:"height"`
|
Height int64 `json:"height" csv:"-"`
|
||||||
Width int64 `json:"width"`
|
Width int64 `json:"width" csv:"-"`
|
||||||
URL string `json:"url"`
|
URL string `json:"url" csv:"-"`
|
||||||
Alternates Alternates `json:"alternates"`
|
Alternates Alternates `json:"alternates" csv:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Alternates struct {
|
type Alternates struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
type Score struct {
|
type Score struct {
|
||||||
Up int64 `json:"up"`
|
Up int64 `json:"up" csv:"up_score"`
|
||||||
Down int64 `json:"down"`
|
Down int64 `json:"down" csv:"down_score"`
|
||||||
Total int64 `json:"total"`
|
Total int64 `json:"total" csv:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Tags struct {
|
type Tags struct {
|
||||||
General []string `json:"general"`
|
General []string `json:"general" csv:"-"`
|
||||||
Artist []string `json:"artist"`
|
Artist []string `json:"artist" csv:"-"`
|
||||||
Copyright []string `json:"copyright"`
|
Copyright []string `json:"copyright" csv:"-"`
|
||||||
Character []string `json:"character"`
|
Character []string `json:"character" csv:"-"`
|
||||||
Species []string `json:"species"`
|
Species []string `json:"species" csv:"-"`
|
||||||
Invalid []string `json:"invalid"`
|
Invalid []string `json:"invalid" csv:"-"`
|
||||||
Meta []string `json:"meta"`
|
Meta []string `json:"meta" csv:"-"`
|
||||||
Lore []string `json:"lore"`
|
Lore []string `json:"lore" csv:"-"`
|
||||||
}
|
}
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
package utils
|
package utils
|
||||||
|
|
||||||
// E621_MAX_POST_COUNT is the maximum allowable post count for E621.
|
// E621_MAX_POST_COUNT is the maximum allowable post count for e621.
|
||||||
const E621_MAX_POST_COUNT = 320
|
const E621_MAX_POST_COUNT = 320
|
||||||
|
@ -3,6 +3,12 @@ package utils
|
|||||||
import "fmt"
|
import "fmt"
|
||||||
|
|
||||||
// StatusCodesToError maps HTTP status codes to corresponding error types.
|
// StatusCodesToError maps HTTP status codes to corresponding error types.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - statusCode: The HTTP status code to be mapped to an error type.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - error: An error representing the mapped HTTP status code.
|
||||||
func StatusCodesToError(statusCode int) error {
|
func StatusCodesToError(statusCode int) error {
|
||||||
var err error
|
var err error
|
||||||
switch statusCode {
|
switch statusCode {
|
||||||
@ -31,6 +37,7 @@ func StatusCodesToError(statusCode int) error {
|
|||||||
// AccessDeniedError represents an "Access Denied" error.
|
// AccessDeniedError represents an "Access Denied" error.
|
||||||
type AccessDeniedError struct{}
|
type AccessDeniedError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for AccessDeniedError.
|
||||||
func (_ AccessDeniedError) Error() string {
|
func (_ AccessDeniedError) Error() string {
|
||||||
return "access denied"
|
return "access denied"
|
||||||
}
|
}
|
||||||
@ -38,6 +45,7 @@ func (_ AccessDeniedError) Error() string {
|
|||||||
// NotFoundError represents a "Not Found" error.
|
// NotFoundError represents a "Not Found" error.
|
||||||
type NotFoundError struct{}
|
type NotFoundError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for NotFoundError.
|
||||||
func (_ NotFoundError) Error() string {
|
func (_ NotFoundError) Error() string {
|
||||||
return "not found"
|
return "not found"
|
||||||
}
|
}
|
||||||
@ -45,6 +53,7 @@ func (_ NotFoundError) Error() string {
|
|||||||
// PreconditionFailedError represents a "Precondition Failed" error.
|
// PreconditionFailedError represents a "Precondition Failed" error.
|
||||||
type PreconditionFailedError struct{}
|
type PreconditionFailedError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for PreconditionFailedError.
|
||||||
func (_ PreconditionFailedError) Error() string {
|
func (_ PreconditionFailedError) Error() string {
|
||||||
return "precondition failed"
|
return "precondition failed"
|
||||||
}
|
}
|
||||||
@ -52,6 +61,7 @@ func (_ PreconditionFailedError) Error() string {
|
|||||||
// RateLimitReachedError represents a "Rate Limit Reached" error.
|
// RateLimitReachedError represents a "Rate Limit Reached" error.
|
||||||
type RateLimitReachedError struct{}
|
type RateLimitReachedError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for RateLimitReachedError.
|
||||||
func (_ RateLimitReachedError) Error() string {
|
func (_ RateLimitReachedError) Error() string {
|
||||||
return "rate limit reached"
|
return "rate limit reached"
|
||||||
}
|
}
|
||||||
@ -59,6 +69,7 @@ func (_ RateLimitReachedError) Error() string {
|
|||||||
// InvalidParametersError represents an "Invalid Parameters" error.
|
// InvalidParametersError represents an "Invalid Parameters" error.
|
||||||
type InvalidParametersError struct{}
|
type InvalidParametersError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for InvalidParametersError.
|
||||||
func (_ InvalidParametersError) Error() string {
|
func (_ InvalidParametersError) Error() string {
|
||||||
return "invalid parameters"
|
return "invalid parameters"
|
||||||
}
|
}
|
||||||
@ -66,6 +77,7 @@ func (_ InvalidParametersError) Error() string {
|
|||||||
// InternalServerError represents an "Internal Server Error" error.
|
// InternalServerError represents an "Internal Server Error" error.
|
||||||
type InternalServerError struct{}
|
type InternalServerError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for InternalServerError.
|
||||||
func (_ InternalServerError) Error() string {
|
func (_ InternalServerError) Error() string {
|
||||||
return "internal server error"
|
return "internal server error"
|
||||||
}
|
}
|
||||||
@ -73,6 +85,7 @@ func (_ InternalServerError) Error() string {
|
|||||||
// BadGatewayError represents a "Bad Gateway" error.
|
// BadGatewayError represents a "Bad Gateway" error.
|
||||||
type BadGatewayError struct{}
|
type BadGatewayError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for BadGatewayError.
|
||||||
func (_ BadGatewayError) Error() string {
|
func (_ BadGatewayError) Error() string {
|
||||||
return "bad gateway"
|
return "bad gateway"
|
||||||
}
|
}
|
||||||
@ -80,6 +93,7 @@ func (_ BadGatewayError) Error() string {
|
|||||||
// ServiceUnavailableError represents a "Service Unavailable" error.
|
// ServiceUnavailableError represents a "Service Unavailable" error.
|
||||||
type ServiceUnavailableError struct{}
|
type ServiceUnavailableError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for ServiceUnavailableError.
|
||||||
func (_ ServiceUnavailableError) Error() string {
|
func (_ ServiceUnavailableError) Error() string {
|
||||||
return "service unavailable"
|
return "service unavailable"
|
||||||
}
|
}
|
||||||
@ -87,6 +101,7 @@ func (_ ServiceUnavailableError) Error() string {
|
|||||||
// UnknownError represents an "Unknown" error.
|
// UnknownError represents an "Unknown" error.
|
||||||
type UnknownError struct{}
|
type UnknownError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for UnknownError.
|
||||||
func (_ UnknownError) Error() string {
|
func (_ UnknownError) Error() string {
|
||||||
return "unknown error"
|
return "unknown error"
|
||||||
}
|
}
|
||||||
@ -94,6 +109,7 @@ func (_ UnknownError) Error() string {
|
|||||||
// OriginConnectionTimeOutError represents an "Origin Connection Time-Out" error.
|
// OriginConnectionTimeOutError represents an "Origin Connection Time-Out" error.
|
||||||
type OriginConnectionTimeOutError struct{}
|
type OriginConnectionTimeOutError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for OriginConnectionTimeOutError.
|
||||||
func (_ OriginConnectionTimeOutError) Error() string {
|
func (_ OriginConnectionTimeOutError) Error() string {
|
||||||
return "origin connection time-out"
|
return "origin connection time-out"
|
||||||
}
|
}
|
||||||
@ -101,6 +117,7 @@ func (_ OriginConnectionTimeOutError) Error() string {
|
|||||||
// SSLHandshakeFailedError represents an "SSL Handshake Failed" error.
|
// SSLHandshakeFailedError represents an "SSL Handshake Failed" error.
|
||||||
type SSLHandshakeFailedError struct{}
|
type SSLHandshakeFailedError struct{}
|
||||||
|
|
||||||
|
// Error returns the error message for SSLHandshakeFailedError.
|
||||||
func (_ SSLHandshakeFailedError) Error() string {
|
func (_ SSLHandshakeFailedError) Error() string {
|
||||||
return "ssl handshake failed"
|
return "ssl handshake failed"
|
||||||
}
|
}
|
||||||
|
53
pkg/e621/utils/helper.go
Normal file
53
pkg/e621/utils/helper.go
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
// SliceFilter filters elements of a generic type from a slice based on a provided filter function.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - slice: The input slice containing elements of type T.
|
||||||
|
// - filter: A function that takes an element of type T and returns a boolean indicating whether to include the element in the result.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - ret: A slice containing elements that satisfy the filtering condition.
|
||||||
|
func SliceFilter[T any](slice []T, filter func(T) bool) (ret []T) {
|
||||||
|
for _, s := range slice {
|
||||||
|
if filter(s) {
|
||||||
|
ret = append(ret, s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadJsonTestData decodes JSON data from a file and populates a variable of a generic type T.
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - testDataPath: The file path to the JSON data.
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - T: A variable of type T populated with the decoded JSON data.
|
||||||
|
// - error: An error, if any, encountered during file opening, decoding, or closing.
|
||||||
|
func LoadJsonTestData[T any](testDataPath string) (T, error) {
|
||||||
|
// Create a variable to store the decoded JSON data
|
||||||
|
var jsonData T
|
||||||
|
|
||||||
|
// Open the JSON file
|
||||||
|
file, err := os.Open(testDataPath)
|
||||||
|
if err != nil {
|
||||||
|
return jsonData, err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
// Create a decoder
|
||||||
|
decoder := json.NewDecoder(file)
|
||||||
|
|
||||||
|
// Decode the JSON data into the struct
|
||||||
|
if err := decoder.Decode(&jsonData); err != nil {
|
||||||
|
return jsonData, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return jsonData, nil
|
||||||
|
}
|
@ -1,28 +0,0 @@
|
|||||||
package utils
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"os"
|
|
||||||
)
|
|
||||||
|
|
||||||
func LoadJsonTestData[T any](testDataPath string) (T, error) {
|
|
||||||
// Create a variable to store the decoded JSON data
|
|
||||||
var jsonData T
|
|
||||||
|
|
||||||
// Open the JSON file
|
|
||||||
file, err := os.Open(testDataPath)
|
|
||||||
if err != nil {
|
|
||||||
return jsonData, err
|
|
||||||
}
|
|
||||||
defer file.Close()
|
|
||||||
|
|
||||||
// Create a decoder
|
|
||||||
decoder := json.NewDecoder(file)
|
|
||||||
|
|
||||||
// Decode the JSON data into the struct
|
|
||||||
if err := decoder.Decode(&jsonData); err != nil {
|
|
||||||
return jsonData, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return jsonData, nil
|
|
||||||
}
|
|
Reference in New Issue
Block a user