Repair channels if necessary

internal/database/database.go

@@ -79,6 +79,8 @@ func (d *Database) Close() {
 
 // Populate fills the database with initial information like the admin user.
 func (d *Database) Populate(name, password, matrixID string) error {
+	log.Print("Populating database.")
+
 	var user model.User
 
 	query := d.gormdb.Where("name = ?", name).First(&user)
@@ -86,14 +88,53 @@ func (d *Database) Populate(name, password, matrixID string) error {
 	if errors.Is(query.Error, gorm.ErrRecordNotFound) {
 		user, err := model.NewUser(d.credentialsManager, name, password, true, matrixID)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 
 		if err := d.gormdb.Create(&user).Error; err != nil {
 			return errors.New("user cannot be created")
 		}
 	} else {
-		log.Printf("Admin user %s already exists.\n", name)
+		log.Printf("Priviledged user %s already exists.", name)
+	}
+
+	return nil
+}
+
+// RepairChannels resets channels that have been modified by a user.
+func (d *Database) RepairChannels(dp Dispatcher) error {
+	log.Print("Repairing application channels.")
+
+	users, err := d.GetUsers()
+	if err != nil {
+		return err
+	}
+
+	for _, user := range users {
+		applications, err := d.GetApplications(&user)
+		if err != nil {
+			return err
+		}
+
+		for _, application := range applications {
+			if err := dp.UpdateApplication(&application); err != nil {
+				return err
+			}
+
+			orphan, err := dp.IsOrphan(&application, &user)
+			if err != nil {
+				return err
+			}
+
+			if orphan {
+				log.Printf("Found orphan channel for application %s (ID %d)", application.Name, application.ID)
+
+				if err = dp.RepairApplication(&application, &user); err != nil {
+					log.Printf("Unable to repair application %s (ID %d).", application.Name, application.ID)
+					log.Println(err)
+				}
+			}
+		}
 	}
 
 	return nil