example/lowlevel/dmail.go

@@ -1,81 +1,65 @@
 package main
 
 import (
-	"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/endpoints"
-	"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
-	_ "github.com/joho/godotenv/autoload"
 	"log"
 	"net/http"
 	"os"
+
+	"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/endpoints"
+	"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
+	_ "github.com/joho/godotenv/autoload"
 )
 
 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
+		Username:  os.Getenv("API_USER"),
+		APIKey:    os.Getenv("API_KEY"),
 	}
 
-	// Log: Getting a single post.
-	log.Println("Getting single dm: ")
-
-	// Specify the post ID for retrieval.
-	dMailID := 2062973 // Replace with the desired post's ID.
-
-	// Call the GetPost function to retrieve the specified post.
-	dmail, err := endpoints.GetDmail(requestContext, dMailID)
-
+	log.Println("Getting a DMail by ID:")
+	dmail, err := endpoints.GetDmail(requestContext, 1)
 	if err != nil {
 		log.Println(err)
 	} else {
-		// Log the URL of the retrieved post.
 		log.Println(dmail.Body)
 	}
 	log.Println("----------")
 
-	// Log: Getting a list of posts.
-	log.Println("Getting a list of posts: ")
+	log.Println("Deleting a DMail by ID:")
+	err = endpoints.DeleteDmail(requestContext, 1)
+	if err != nil {
+		log.Println(err)
+	}
+	log.Println("----------")
 
-	// Define query parameters for retrieving a list of posts.
+	log.Println("Marking a DMail as read by ID:")
+	err = endpoints.MarkAsReadDmail(requestContext, 1)
+	if err != nil {
+		log.Println(err)
+	}
+	log.Println("----------")
+
+	log.Println("Getting all DMails:")
 	query := map[string]string{
-		"title_matches": "test",
+		"limit": "5",
 	}
-
-	// Call the GetPosts function to retrieve a list of posts based on the query parameters.
-	posts, err := endpoints.GetDmails(requestContext, query)
-
+	dmails, err := endpoints.GetAllDmails(requestContext, query)
 	if err != nil {
 		log.Println(err)
 	} else {
-		// Log the number of posts retrieved.
-		log.Println(posts[0].Body)
+		for _, dmail := range dmails {
+			log.Println(dmail.Body)
+		}
 	}
 	log.Println("----------")
 
-	log.Println("Mark all DMails as read: ")
-
-	// Call the ReadAllDmails function to set all DMails as read.
-	err = endpoints.ReadAllDmails(requestContext)
-
+	log.Println("Marking all DMails as read:")
+	err = endpoints.MarkAsReadAllDmails(requestContext)
 	if err != nil {
 		log.Println(err)
-	} else {
-		log.Println("read all posts")
-	}
-	log.Println("----------")
-
-	log.Println("Delete all DMails as read: ")
-
-	// Call the ReadAllDmails function to set all DMails as read.
-	err = endpoints.DeleateDmail(requestContext, dMailID)
-
-	if err != nil {
-		log.Println(err)
-	} else {
-		log.Println("DMail deleted")
 	}
 	log.Println("----------")
 }