feat: automatically archive envelopes when category is archived (#759)

With this change, envelopes are archived automatically if their parent
category is archived.

Author: morremeyer

pkg/models/category.go

@@ -48,3 +48,27 @@ func (c *Category) links(tx *gorm.DB) {
 	c.Links.Self = fmt.Sprintf("%s/v1/categories/%s", url, c.ID)
 	c.Links.Envelopes = fmt.Sprintf("%s/v1/envelopes?category=%s", url, c.ID)
 }
+
+// BeforeUpdate archives all envelopes when the category is archived.
+func (c *Category) BeforeUpdate(tx *gorm.DB) (err error) {
+	if tx.Statement.Changed("Hidden") && !c.Hidden {
+		var envelopes []Envelope
+		err = tx.Model(&Envelope{EnvelopeCreate: EnvelopeCreate{
+			CategoryID: c.ID,
+		}}).
+			Find(&envelopes).Error
+		if err != nil {
+			return
+		}
+
+		for _, e := range envelopes {
+			e.Hidden = true
+			err = tx.Model(&e).Updates(&e).Error
+			if err != nil {
+				return
+			}
+		}
+	}
+
+	return nil
+}

pkg/models/category_test.go

@@ -0,0 +1,27 @@
+package models_test
+
+import (
+	"github.com/envelope-zero/backend/v3/pkg/models"
+	"github.com/stretchr/testify/assert"
+)
+
+func (suite *TestSuiteStandard) TestCategoryArchiveArchivesEnvelopes() {
+	category := suite.createTestCategory(models.CategoryCreate{
+		BudgetID: suite.createTestBudget(models.BudgetCreate{}).ID,
+	})
+
+	envelope := suite.createTestEnvelope(models.EnvelopeCreate{
+		CategoryID: category.ID,
+		Hidden:     false,
+	})
+	assert.False(suite.T(), envelope.Hidden, "Envelope archived on creation, it should not be")
+
+	// Archive the category
+	err := suite.db.Model(&category).Select("Hidden").Updates(models.Category{CategoryCreate: models.CategoryCreate{Hidden: true}}).Error
+	assert.Nil(suite.T(), err)
+
+	// Verify that the envelope is not archived
+	err = suite.db.First(&envelope, envelope.ID).Error
+	assert.Nil(suite.T(), err)
+	assert.True(suite.T(), envelope.Hidden, "Envelope was not archived together with category")
+}