refactoring

go-backend/internal/models/drivers_licence.go

@@ -1,7 +1,11 @@
 package models
 
 import (
+	"GoMembership/pkg/logger"
+	"fmt"
 	"time"
+
+	"gorm.io/gorm"
 )
 
 type Licence struct {
@@ -9,15 +13,54 @@ type Licence struct {
 	UserID         uint `json:"user_id"`
 	CreatedAt      time.Time
 	UpdatedAt      time.Time
-	Status         int8       `json:"status" binding:"omitempty,number"`
-	Number         string     `json:"number" binding:"omitempty,safe_content"`
-	IssuedDate     time.Time  `json:"issued_date" binding:"omitempty"`
-	ExpirationDate time.Time  `json:"expiration_date" binding:"omitempty"`
-	IssuingCountry string     `json:"country" binding:"safe_content"`
-	Categories     []Category `json:"categories" gorm:"many2many:licence_2_categories"`
+	Status         int8        `json:"status" binding:"omitempty,number"`
+	Number         string      `json:"number" binding:"omitempty,safe_content"`
+	IssuedDate     time.Time   `json:"issued_date" binding:"omitempty"`
+	ExpirationDate time.Time   `json:"expiration_date" binding:"omitempty"`
+	IssuingCountry string      `json:"country" binding:"safe_content"`
+	Categories     []*Category `json:"categories" gorm:"many2many:licence_2_categories"`
 }
 
-type Category struct {
-	ID   uint   `json:"id" gorm:"primaryKey"`
-	Name string `json:"category" binding:"safe_content"`
+func (l *Licence) BeforeSafe(tx *gorm.DB) error {
+	if err := tx.Model(l).Association("Categories").Replace(l.Categories); err != nil {
+		return fmt.Errorf("failed to link categories: %w", err)
+	}
+
+	return nil
+}
+
+func (l *Licence) Create(db *gorm.DB) error {
+	if err := db.Omit("Categories").Create(l).Error; err != nil {
+		return err
+	}
+
+	if err := db.Model(&l).Association("Categories").Replace(l.Categories); err != nil {
+		return err
+	}
+
+	logger.Info.Printf("Licence created: %#v", l)
+	return db.Preload("Categories").First(l, l.ID).Error // Refresh the object with Categories
+}
+
+func (l *Licence) Update(db *gorm.DB) error {
+
+	return db.Transaction(func(tx *gorm.DB) error {
+		// Check if the user exists in the database
+		var existingLicence Licence
+
+		logger.Info.Printf("updating Licence: %#v", l)
+		if err := tx.First(&existingLicence, l.ID).Error; err != nil {
+			return err
+		}
+
+		if err := tx.Model(&existingLicence).Updates(l).Error; err != nil {
+			return err
+		}
+		return tx.First(l, l.ID).Error
+	})
+
+}
+
+func (l *Licence) Delete(db *gorm.DB) error {
+	return db.Delete(&l).Error
 }