Add most of deletion logic

Author: michplunkett

cmd/deleteRedditHistory/main.go

@@ -7,9 +7,11 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"log"
 	"os"
+	"time"
 
 	"github.com/vartanbeno/go-reddit/v2/reddit"
 )
@@ -21,6 +23,10 @@ type envVars struct {
 	userPassword string
 }
 
+const LIMIT = 100
+
+var THRESHOLD = time.Now().AddDate(-2, 0, 0)
+
 func arrayHasNoEmptyStrings(envVars []string) bool {
 	for _, value := range envVars {
 		if value == "" {
@@ -32,6 +38,8 @@ func arrayHasNoEmptyStrings(envVars []string) bool {
 }
 
 func main() {
+	ctx := context.Background()
+
 	e := envVars{
 		os.Getenv("REDDIT_APP_ID"),
 		os.Getenv("REDDIT_SECRET"),
@@ -44,9 +52,108 @@ func main() {
 	}
 
 	credentials := reddit.Credentials{ID: e.appID, Secret: e.appSecret, Username: e.userName, Password: e.userPassword}
-	_, clientErr := reddit.NewClient(credentials)
+	cli, clientErr := reddit.NewClient(credentials)
 
 	if clientErr != nil {
 		log.Fatal(clientErr)
 	}
+
+	// Get overview of user
+	commentService := *cli.Comment
+	postService := *cli.Post
+	userService := *cli.User
+
+	// Get all posts
+	fmt.Println("------- Pulling user posts. -------")
+
+	lastPostID := ""
+	postIds := make([]string, 0)
+	postOptions := &reddit.ListUserOverviewOptions{
+		ListOptions: reddit.ListOptions{
+			Limit: LIMIT,
+		},
+		Sort: "new",
+		Time: "all",
+	}
+	for {
+		if lastPostID != "" {
+			postOptions.ListOptions.After = lastPostID
+		}
+
+		fmt.Println("Pulling more posts.")
+		posts, _, err := userService.Posts(ctx, postOptions)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		if len(posts) == 0 {
+			break
+		}
+
+		for _, post := range posts {
+			if post.Created.Time.Before(THRESHOLD) ||
+				post.Created.Time.Equal(THRESHOLD) {
+				fmt.Println(post.Created)
+				postIds = append(postIds, post.FullID)
+			}
+		}
+
+		lastPostID = posts[len(posts)-1].FullID
+	}
+
+	// Delete all posts
+	fmt.Println("------- Deleting user posts. -------")
+	for _, pID := range postIds {
+		fmt.Println(pID)
+		//_, err := postService.Delete(ctx, pID)
+		//if err != nil {
+		//	log.Fatal(err)
+		//}
+	}
+
+	// Get all comments
+	fmt.Println("------- Pulling user comments. -------")
+	lastCommentID := ""
+	commentIds := make([]string, 0)
+	commentOptions := &reddit.ListUserOverviewOptions{
+		ListOptions: reddit.ListOptions{
+			Limit: LIMIT,
+		},
+		Sort: "new",
+		Time: "all",
+	}
+	for {
+		if lastCommentID != "" {
+			commentOptions.ListOptions.After = lastCommentID
+		}
+
+		fmt.Println("Pulling more comments.")
+		comments, _, err := userService.Comments(ctx, commentOptions)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		if len(comments) == 0 {
+			break
+		}
+
+		for _, c := range comments {
+			if c.Created.Time.Before(THRESHOLD) ||
+				c.Created.Time.Equal(THRESHOLD) {
+				commentIds = append(commentIds, c.FullID)
+			}
+		}
+
+		lastCommentID = comments[len(comments)-1].FullID
+	}
+
+	// Delete all comments
+	fmt.Println("------- Deleting user comments. -------")
+	for _, cID := range commentIds {
+		fmt.Println(cID)
+		//_, err := commentService.Delete(ctx, cID)
+		//if err != nil {
+		//	log.Fatal(err)
+		//}
+	}
 }