Initial commit: Open sourcing all of the Maple Open Technologies code.

cloud/maplefile-backend/internal/usecase/storageusageevent/get_events.go

@@ -0,0 +1,159 @@
+// monorepo/cloud/maplefile-backend/internal/maplefile/usecase/storageusageevent/get_events.go
+package storageusageevent
+
+import (
+	"context"
+	"time"
+
+	"go.uber.org/zap"
+
+	"github.com/gocql/gocql"
+	"codeberg.org/mapleopentech/monorepo/cloud/maplefile-backend/config"
+	"codeberg.org/mapleopentech/monorepo/cloud/maplefile-backend/internal/domain/storageusageevent"
+	"codeberg.org/mapleopentech/monorepo/cloud/maplefile-backend/pkg/httperror"
+)
+
+// GetStorageUsageEventsRequest contains the filtering parameters
+type GetStorageUsageEventsRequest struct {
+	UserID      gocql.UUID  `json:"user_id"`
+	TrendPeriod string      `json:"trend_period"` // "7days", "monthly", "yearly"
+	Year        *int        `json:"year,omitempty"`
+	Month       *time.Month `json:"month,omitempty"`
+	Days        *int        `json:"days,omitempty"` // For custom day ranges
+}
+
+// GetStorageUsageEventsResponse contains the filtered events
+type GetStorageUsageEventsResponse struct {
+	UserID      gocql.UUID                             `json:"user_id"`
+	TrendPeriod string                                 `json:"trend_period"`
+	StartDate   time.Time                              `json:"start_date"`
+	EndDate     time.Time                              `json:"end_date"`
+	Events      []*storageusageevent.StorageUsageEvent `json:"events"`
+	EventCount  int                                    `json:"event_count"`
+}
+
+type GetStorageUsageEventsUseCase interface {
+	Execute(ctx context.Context, req *GetStorageUsageEventsRequest) (*GetStorageUsageEventsResponse, error)
+}
+
+type getStorageUsageEventsUseCaseImpl struct {
+	config *config.Configuration
+	logger *zap.Logger
+	repo   storageusageevent.StorageUsageEventRepository
+}
+
+func NewGetStorageUsageEventsUseCase(
+	config *config.Configuration,
+	logger *zap.Logger,
+	repo storageusageevent.StorageUsageEventRepository,
+) GetStorageUsageEventsUseCase {
+	logger = logger.Named("GetStorageUsageEventsUseCase")
+	return &getStorageUsageEventsUseCaseImpl{config, logger, repo}
+}
+
+func (uc *getStorageUsageEventsUseCaseImpl) Execute(ctx context.Context, req *GetStorageUsageEventsRequest) (*GetStorageUsageEventsResponse, error) {
+	//
+	// STEP 1: Validation.
+	//
+
+	e := make(map[string]string)
+	if req == nil {
+		e["request"] = "Request is required"
+	} else {
+		if req.UserID.String() == "" {
+			e["user_id"] = "User ID is required"
+		}
+		if req.TrendPeriod == "" {
+			e["trend_period"] = "Trend period is required"
+		} else if req.TrendPeriod != "7days" && req.TrendPeriod != "monthly" && req.TrendPeriod != "yearly" && req.TrendPeriod != "custom" {
+			e["trend_period"] = "Trend period must be one of: 7days, monthly, yearly, custom"
+		}
+
+		// Validate period-specific parameters
+		switch req.TrendPeriod {
+		case "monthly":
+			if req.Year == nil {
+				e["year"] = "Year is required for monthly trend"
+			}
+			if req.Month == nil {
+				e["month"] = "Month is required for monthly trend"
+			}
+		case "yearly":
+			if req.Year == nil {
+				e["year"] = "Year is required for yearly trend"
+			}
+		case "custom":
+			if req.Days == nil || *req.Days <= 0 {
+				e["days"] = "Days must be greater than 0 for custom trend"
+			}
+		}
+	}
+	if len(e) != 0 {
+		uc.logger.Warn("Failed validating get storage usage events",
+			zap.Any("error", e))
+		return nil, httperror.NewForBadRequest(&e)
+	}
+
+	//
+	// STEP 2: Get events based on trend period.
+	//
+
+	var events []*storageusageevent.StorageUsageEvent
+	var err error
+	var startDate, endDate time.Time
+
+	switch req.TrendPeriod {
+	case "7days":
+		events, err = uc.repo.GetLast7DaysEvents(ctx, req.UserID)
+		endDate = time.Now().Truncate(24 * time.Hour)
+		startDate = endDate.Add(-6 * 24 * time.Hour)
+
+	case "monthly":
+		events, err = uc.repo.GetMonthlyEvents(ctx, req.UserID, *req.Year, *req.Month)
+		startDate = time.Date(*req.Year, *req.Month, 1, 0, 0, 0, 0, time.UTC)
+		endDate = startDate.AddDate(0, 1, -1) // Last day of the month
+
+	case "yearly":
+		events, err = uc.repo.GetYearlyEvents(ctx, req.UserID, *req.Year)
+		startDate = time.Date(*req.Year, 1, 1, 0, 0, 0, 0, time.UTC)
+		endDate = time.Date(*req.Year, 12, 31, 0, 0, 0, 0, time.UTC)
+
+	case "custom":
+		events, err = uc.repo.GetLastNDaysEvents(ctx, req.UserID, *req.Days)
+		endDate = time.Now().Truncate(24 * time.Hour)
+		startDate = endDate.Add(-time.Duration(*req.Days-1) * 24 * time.Hour)
+
+	default:
+		return nil, httperror.NewForBadRequestWithSingleField("trend_period", "Invalid trend period")
+	}
+
+	if err != nil {
+		uc.logger.Error("Failed to get storage usage events",
+			zap.String("user_id", req.UserID.String()),
+			zap.String("trend_period", req.TrendPeriod),
+			zap.Error(err))
+		return nil, err
+	}
+
+	//
+	// STEP 3: Build response.
+	//
+
+	response := &GetStorageUsageEventsResponse{
+		UserID:      req.UserID,
+		TrendPeriod: req.TrendPeriod,
+		StartDate:   startDate,
+		EndDate:     endDate,
+		Events:      events,
+		EventCount:  len(events),
+	}
+
+	uc.logger.Debug("Successfully retrieved storage usage events",
+		zap.String("user_id", req.UserID.String()),
+		zap.String("trend_period", req.TrendPeriod),
+		zap.Int("event_count", len(events)),
+		zap.Time("start_date", startDate),
+		zap.Time("end_date", endDate))
+
+	return response, nil
+}