package main

import (
	"git.anthrove.art/anthrove/e621-sdk-go/pkg/e621/builder"
	"git.anthrove.art/anthrove/e621-sdk-go/pkg/e621/model"
	_ "github.com/joho/godotenv/autoload"
	"log"
	"net/http"
	"os"
)

func main() {
	// Define the request context with essential information.
	requestContext := model.RequestContext{
		Client:    http.Client{},
		Host:      "https://e621.net",
		UserAgent: "Go-e621-SDK (@username)",
		Username:  os.Getenv("API_USER"), // Replace with your username
		APIKey:    os.Getenv("API_KEY"),  // Replace with your API key
	}

	// Log: Getting a single tag.
	log.Println("Getting a single tag: ")

	// Specify the tag ID for retrieval.
	tagID := 1530 // Replace with the desired tag's ID.

	// Call the GetTag function to retrieve the specified tag.
	getTag := builder.NewGetTagBuilder(requestContext)
	tag, err := getTag.SetTagID(tagID).Execute()

	if err != nil {
		log.Println(err)
	} else {
		// Log the name of the retrieved tag.
		log.Println(tag.Name)
	}
	log.Println("----------")

	// Log: Getting a list of tags.
	log.Println("Getting a list of tags: ")

	// Call the GetTags function to retrieve a list of tags based on the query parameters.
	getTags := builder.NewGetTagsBuilder(requestContext)
	tagList, err := getTags.SetLimit(5).Artist(false).Execute()

	if err != nil {
		log.Println(err)
	} else {
		// Log the number of tags retrieved.
		log.Println(len(tagList))
	}
	log.Println("----------")

	// Log: Searching for tags containing "cat."
	log.Println("Searching for tags containing 'cat': ")

	// Call the GetTags function with the search query to retrieve matching tags.
	tagList, err = getTags.SetLimit(5).SearchName("cat*").Execute()
	if err != nil {
		log.Println(err)
	} else {
		// Log the retrieved tags.
		log.Println(tagList)
	}
	log.Println("----------")

	// Log: Searching for tags with the category "artist."
	log.Println("Searching for tags with the category 'artist': ")

	// Call the GetTags function with the category filter to retrieve artist tags.
	tagList, err = getTags.SetLimit(5).SearchCategory(model.Artist).Execute()

	if err != nil {
		log.Println(err)
	} else {
		// Log the retrieved artist tags.
		log.Println(tagList)
	}
	log.Println("----------")
}