Backend: add Subscription CRUD

internal/services/membership_service.go

@@ -5,14 +5,17 @@ import (
 
 	"GoMembership/internal/models"
 	"GoMembership/internal/repositories"
+	"GoMembership/pkg/errors"
 )
 
 type MembershipServiceInterface interface {
 	RegisterMembership(membership *models.Membership) (uint, error)
 	FindMembershipByUserID(userID uint) (*models.Membership, error)
 	RegisterSubscription(subscription *models.SubscriptionModel) (uint, error)
-	GetMembershipModelNames() ([]string, error)
-	GetModelByName(modelname *string) (*models.SubscriptionModel, error)
+	UpdateSubscription(subscription *models.SubscriptionModel) (*models.SubscriptionModel, error)
+	DeleteSubscription(subscription *models.SubscriptionModel) error
+	GetSubscriptionModelNames() ([]string, error)
+	GetSubscriptionByName(modelname *string) (*models.SubscriptionModel, error)
 	GetSubscriptions(where map[string]interface{}) (*[]models.SubscriptionModel, error)
 }
 
@@ -26,6 +29,48 @@ func (service *MembershipService) RegisterMembership(membership *models.Membersh
 	return service.Repo.CreateMembership(membership)
 }
 
+func (service *MembershipService) UpdateSubscription(subscription *models.SubscriptionModel) (*models.SubscriptionModel, error) {
+
+	existingSubscription, err := repositories.GetSubscriptionByName(&subscription.Name)
+	if err != nil {
+		return nil, err
+	}
+	if existingSubscription == nil {
+		return nil, errors.ErrSubscriptionNotFound
+	}
+	if existingSubscription.MonthlyFee != subscription.MonthlyFee ||
+		existingSubscription.HourlyRate != subscription.HourlyRate ||
+		existingSubscription.Conditions != subscription.Conditions ||
+		existingSubscription.IncludedPerYear != subscription.IncludedPerYear ||
+		existingSubscription.IncludedPerMonth != subscription.IncludedPerMonth {
+		return nil, errors.ErrInvalidSubscriptionData
+	}
+	subscription.ID = existingSubscription.ID
+	return service.SubscriptionRepo.UpdateSubscription(subscription)
+
+}
+
+func (service *MembershipService) DeleteSubscription(subscription *models.SubscriptionModel) error {
+	exists, err := repositories.GetSubscriptionByName(&subscription.Name)
+	if err != nil {
+		return err
+	}
+	if exists == nil {
+		return errors.ErrNotFound
+	}
+
+	subscription.ID = exists.ID
+	usersInSubscription, err := repositories.GetUsersBySubscription(subscription.ID)
+
+	if err != nil {
+		return err
+	}
+	if len(*usersInSubscription) > 0 {
+		return errors.ErrSubscriptionInUse
+	}
+	return service.SubscriptionRepo.DeleteSubscription(subscription)
+}
+
 func (service *MembershipService) FindMembershipByUserID(userID uint) (*models.Membership, error) {
 	return service.Repo.FindMembershipByUserID(userID)
 }
@@ -35,12 +80,12 @@ func (service *MembershipService) RegisterSubscription(subscription *models.Subs
 	return service.SubscriptionRepo.CreateSubscriptionModel(subscription)
 }
 
-func (service *MembershipService) GetMembershipModelNames() ([]string, error) {
-	return service.SubscriptionRepo.GetMembershipModelNames()
+func (service *MembershipService) GetSubscriptionModelNames() ([]string, error) {
+	return service.SubscriptionRepo.GetSubscriptionModelNames()
 }
 
-func (service *MembershipService) GetModelByName(modelname *string) (*models.SubscriptionModel, error) {
-	return repositories.GetModelByName(modelname)
+func (service *MembershipService) GetSubscriptionByName(modelname *string) (*models.SubscriptionModel, error) {
+	return repositories.GetSubscriptionByName(modelname)
 }
 
 func (service *MembershipService) GetSubscriptions(where map[string]interface{}) (*[]models.SubscriptionModel, error) {

internal/services/user_service.go

@@ -59,7 +59,7 @@ func (service *UserService) UpdateUser(user *models.User) (*models.User, error)
 	}
 
 	// Validate subscription model
-	selectedModel, err := repositories.GetModelByName(&user.Membership.SubscriptionModel.Name)
+	selectedModel, err := repositories.GetSubscriptionByName(&user.Membership.SubscriptionModel.Name)
 	if err != nil {
 		return nil, errors.ErrSubscriptionNotFound
 	}