Add comments to submissions

lib/database/database.go

@@ -31,7 +31,9 @@ type DB interface {
 	AddSubmission(submission Submission) (id int, err error)
 	UpdateSubmission(id int, status string, book *Book) error
 	UpdateSubmissionByBook(bookID string, status string, book *Book) error
+	UpdateSubmissionComment(submissionID, bookID, comment string) error
 	GetSubmission(submissionID string) (submission []Submission, err error)
+	GetComment(bookID string) (string, error)
 }
 
 const (

lib/database/ro.go

@@ -97,10 +97,18 @@ func (db *roDB) UpdateSubmission(id int, status string, book *Book) error {
 	return errors.New("RO database")
 }
 
+func (db *roDB) UpdateSubmissionComment(submissionID, bookdID, comment string) error {
+	return errors.New("RO database")
+}
+
 func (db *roDB) UpdateSubmissionByBook(bookID string, status string, book *Book) error {
 	return errors.New("RO database")
 }
 
+func (db *roDB) GetComment(bookID string) (string, error) {
+	return db.db.GetComment(bookID)
+}
+
 func (db *roDB) GetSubmission(submissionID string) (submission []Submission, err error) {
 	return db.db.GetSubmission(submissionID)
 }

lib/database/submissions.go

@@ -5,6 +5,7 @@ type Submission struct {
 	SubmissionID string `sql:"type:varchar(16)"`
 	Filename     string
 	Status       string
+	Comment      string
 	BookID       string `sql:"type:varchar(16),unique"`
 	Book         *Book
 }
@@ -32,6 +33,26 @@ func (db *pgDB) UpdateSubmissionByBook(bookID string, status string, book *Book)
 	return err
 }
 
+func (db *pgDB) UpdateSubmissionComment(submissionID, bookID, comment string) error {
+	_, err := db.sql.Model(&Submission{}).
+		Set("comment = ?", comment).
+		Where("submission_id = ?", submissionID).
+		Where("book_id = ?", bookID).
+		Update()
+	return err
+}
+
+func (db *pgDB) GetComment(bookID string) (string, error) {
+	var submission Submission
+	err := db.sql.Model(&submission).
+		Where("book_id = ?", bookID).
+		Select()
+	if err != nil {
+		return "", err
+	}
+	return submission.Comment, nil
+}
+
 func (db *pgDB) GetSubmission(submissionID string) (submission []Submission, err error) {
 	err = db.sql.Model(&submission).
 		Column("Book").