wip: invoice reference generation

internal/models/invoice.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/google/uuid"
 
+	"github.com/hazemKrimi/crimson-vault/internal/lib"
 	"github.com/hazemKrimi/crimson-vault/internal/types"
 )
 
@@ -30,7 +31,7 @@ func (db *DB) CreateItem(userId, invoiceId uuid.UUID, body types.CreateItemReque
 	return item, nil
 }
 
-func (db *DB) CreateInvoice(userId uuid.UUID, body types.CreateInvoiceRequestBody) (types.Invoice, error) {
+func (db *DB) CreateInvoice(userId uuid.UUID, body types.CreateInvoiceRequestBody, issuedInvoicesThisYear uint32) (types.Invoice, error) {
 	dueAt, err := time.Parse("2006-01-02T15:04:05Z", body.DueAt)
 
 	if err != nil {
@@ -45,6 +46,7 @@ func (db *DB) CreateInvoice(userId uuid.UUID, body types.CreateInvoiceRequestBod
 		Currency: body.Currency,
 		VAT:      body.VAT,
 		Status:   types.Draft.String(),
+		Reference: lib.GenerateInvoiceReference(issuedInvoicesThisYear + 1),
 	}
 
 	result := db.instance.Create(&invoice)

internal/models/user.go

@@ -21,6 +21,7 @@ func (db *DB) CreateUser(body types.CreateUserRequestBody) (types.User, error) {
 		Country:    body.Country,
 		Phone:      body.Phone,
 		Email:      body.Email,
+		IssuedInvoicesThisYear: 0,
 	}
 
 	result := db.instance.Create(&user)
@@ -99,6 +100,24 @@ func (db *DB) UpdateUser(id uuid.UUID, body types.UpdateUserRequestBody, user *t
 	return nil
 }
 
+func (db *DB) UpdateUserIssuesInvoicesThisYear(id uuid.UUID, issuedInvoicesThisYear uint32, user *types.User) error {
+	result := db.instance.Where("id = ?", id).First(user, id)
+
+	if result.Error != nil {
+		return result.Error
+	}
+
+	result = db.instance.Model(user).Updates(types.User{
+		IssuedInvoicesThisYear: issuedInvoicesThisYear,
+	})
+
+	if result.Error != nil {
+		return result.Error
+	}
+
+	return nil
+}
+
 func (db *DB) UpdateUserSecurityCredentials(id uuid.UUID, body types.UpdateUserSecurityCredentialsBody, user *types.User) error {
 	result := db.instance.Where("id = ?", id).First(user, id)