feat: add post seeder

Author: ainunns

migrations/json/posts.json

@@ -3,12 +3,14 @@
     "text": "Your heart is the size of an ocean. Go find yourself in its hidden depths."
   },
   {
-    "text": "The Bay of Bengal is hit frequently by cyclones. The months of November and May, in particular, are dangerous in this regard."
+    "text": "The Bay of Bengal is hit frequently by cyclones. The months of November and May, in particular, are dangerous in this regard.",
+    "parent_id": 1
   },
   {
     "text": "Thinking is the capital, Enterprise is the way, Hard Work is the solution."
   },
   {
-    "text": "Life is a gamble. You can get hurt, but people die in plane crashes, lose their arms and legs in car accidents; people die every day. Same with fighters: some die, some get hurt, some go on. You just don't let yourself believe it will happen to you."
+    "text": "Life is a gamble. You can get hurt, but people die in plane crashes, lose their arms and legs in car accidents; people die every day. Same with fighters: some die, some get hurt, some go on. You just don't let yourself believe it will happen to you.",
+    "parent_id": 1
   }
 ]

migrations/seeder.go

@@ -10,5 +10,9 @@ func Seeder(db *gorm.DB) error {
 		return err
 	}
 
+	if err := seeds.ListPostSeeder(db); err != nil {
+		return err
+	}
+
 	return nil
 }

migrations/seeds/post_seed.go

@@ -2,6 +2,7 @@ package seeds
 
 import (
 	"encoding/json"
+	"errors"
 	"io"
 	"os"
 
@@ -31,9 +32,42 @@ func ListPostSeeder(db *gorm.DB) error {
 
 	var user entity.User
 	err = db.Find(&user).Limit(2).Error
-	if err != nil {
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return err
+	}
+
+	var users []entity.User
+	err = db.Find(&users).Limit(2).Error
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
 		return err
 	}
 
+	postMap := make(map[int]*entity.Post)
+
+	for i, post := range listPost {
+		newPost := entity.Post{
+			Text:   post.Text,
+			UserID: users[i%len(users)].ID,
+		}
+
+		if err := db.Create(&newPost).Error; err != nil {
+			return err
+		}
+		postMap[i] = &newPost
+	}
+
+	for i, post := range listPost {
+		if post.ParentID != nil {
+			parentPost := postMap[int(*post.ParentID-1)]
+			if parentPost != nil {
+				currentPost := postMap[i]
+				currentPost.ParentID = &parentPost.ID
+				if err := db.Save(currentPost).Error; err != nil {
+					return err
+				}
+			}
+		}
+	}
+
 	return nil
 }