ente/server/pkg/controller/commonbilling/controller.go

package commonbilling

import (
	"context"
	"github.com/ente-io/museum/pkg/repo"
	"github.com/ente-io/museum/pkg/repo/storagebonus"
	"github.com/ente-io/stacktrace"
	"github.com/sirupsen/logrus"
)

type Controller struct {
	StorageBonusRepo *storagebonus.Repository
	UserRepo         *repo.UserRepository
	UsageRepo        *repo.UsageRepository
}

func NewController(
	storageBonusRepo *storagebonus.Repository,
	userRepo *repo.UserRepository,
	usageRepo *repo.UsageRepository,
) *Controller {
	return &Controller{
		StorageBonusRepo: storageBonusRepo,
		UserRepo:         userRepo,
		UsageRepo:        usageRepo,
	}
}

func (c *Controller) CanDowngradeToGivenStorage(newStorage int64, userID int64) (bool, error) {
	adminID, adminErr := c.UserRepo.GetFamilyAdminID(userID)
	if adminErr != nil {
		return false, stacktrace.Propagate(adminErr, "")
	}

	if adminID == nil {
		bonusStorage, bonErr := c.StorageBonusRepo.GetActiveStorageBonuses(context.Background(), userID)
		if bonErr != nil {
			return false, stacktrace.Propagate(bonErr, "")
		}
		usage, err := c.UsageRepo.GetUsage(userID)
		if err != nil {
			return false, stacktrace.Propagate(err, "")
		}
		// newStore + addOnStorage + referralStorage should not be greater than usage.

		if usage > (newStorage + bonusStorage.GetUsableBonus(newStorage)) {
			logrus.Infof("user with %d usage and %d bonus, can not downgrade to %d", usage, bonusStorage.GetUsableBonus(newStorage), newStorage)
			return false, nil
		}
	} else {
		bonusStorage, bonErr := c.StorageBonusRepo.GetActiveStorageBonuses(context.Background(), *adminID)
		if bonErr != nil {
			return false, stacktrace.Propagate(bonErr, "")
		}
		usage, err := c.UsageRepo.StorageForFamilyAdmin(*adminID)
		if err != nil {
			return false, stacktrace.Propagate(err, "")
		}
		if usage > (newStorage + bonusStorage.GetUsableBonus(newStorage)) {
			logrus.Infof("user with %d usage and %d bonus, can not downgrade to %d", usage, bonusStorage.GetUsableBonus(newStorage), newStorage)
			return false, nil
		}
	}
	return true, nil
}