added tdd for comment repo, and comment repo

repos/comment_repo.go

@@ -0,0 +1,80 @@
+package repos
+
+import (
+	"errors"
+
+	"gitlab.com/rawleyifowler/site-rework/models"
+	"gitlab.com/rawleyifowler/site-rework/utils"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+)
+
+type CommentRepo struct {
+	DB          *gorm.DB
+	Initialized bool
+}
+
+func NewCommentRepo(dsnPath string) (*CommentRepo, error) {
+	c := CommentRepo{}
+	err := c.Initialize(dsnPath)
+	if err != nil {
+		return &c, err
+	}
+	return &c, nil
+}
+
+func (c *CommentRepo) Initialize(path string) error {
+	dsn := utils.LoadDSN(path)
+	var err error
+	c.DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
+	if err != nil {
+		return err
+	}
+	c.Initialized = true
+	return nil
+}
+
+func (c *CommentRepo) GetCommentById(id uint) (*models.Comment, error) {
+	if id == 0 {
+		return nil, nil
+	}
+	var comm models.Comment
+	c.DB.Where(&models.Comment{Id: id}).First(&comm)
+	return &comm, nil
+}
+
+func (c *CommentRepo) CreateComment(comm *models.Comment) error {
+	if comm == nil {
+		return errors.New("nil comment")
+	}
+	err := c.DB.Create(comm).Error
+	if err != nil {
+		return err
+	}
+	// nil return means no errors, yay!
+	return nil
+}
+
+func (c *CommentRepo) GetCommentsByAssociatedPost(title string) (*[]models.Comment, error) {
+	if len(title) < 1 {
+		return nil, errors.New("empty associated post title")
+	}
+	var comms []models.Comment
+	err := c.DB.Where(&models.Comment{AssociatedPost: title}).Scan(&comms).Error
+	if err != nil {
+		return nil, err
+	}
+	return &comms, nil
+}
+
+func (c *CommentRepo) GetCommentsByAuthor(auth string) (*[]models.Comment, error) {
+	if len(auth) < 1 {
+		return nil, errors.New("empty author name")
+	}
+	var comms []models.Comment
+	err := c.DB.Where(&models.Comment{Author: auth}).Scan(&comms).Error
+	if err != nil {
+		return nil, err
+	}
+	return &comms, nil
+}