lethdat
dfd999e058
- Introduced a new test file for setting up an in-memory SQLite database for testing purposes. - Added helper functions for seeding test data, including users, plans, subscriptions, and wallet transactions. - Implemented usage helpers to load user video counts and storage usage. - Created user payload struct and functions to build user payloads with preferences and wallet balance. - Refactored gRPC server setup to include new services and handlers. - Updated proto files to simplify service definitions by removing redundant service prefixes.
64 lines
1.8 KiB
Go
64 lines
1.8 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"gorm.io/gorm"
|
|
"stream.api/internal/database/model"
|
|
"stream.api/pkg/logger"
|
|
)
|
|
|
|
type updatePreferencesInput struct {
|
|
EmailNotifications *bool
|
|
PushNotifications *bool
|
|
MarketingNotifications *bool
|
|
TelegramNotifications *bool
|
|
Language *string
|
|
Locale *string
|
|
}
|
|
|
|
func loadUserPreferences(ctx context.Context, db *gorm.DB, userID string) (*model.UserPreference, error) {
|
|
return model.FindOrCreateUserPreference(ctx, db, userID)
|
|
}
|
|
|
|
func updateUserPreferences(ctx context.Context, db *gorm.DB, l logger.Logger, userID string, req updatePreferencesInput) (*model.UserPreference, error) {
|
|
pref, err := model.FindOrCreateUserPreference(ctx, db, userID)
|
|
if err != nil {
|
|
l.Error("Failed to load preferences", "error", err)
|
|
return nil, err
|
|
}
|
|
|
|
if req.EmailNotifications != nil {
|
|
pref.EmailNotifications = model.BoolPtr(*req.EmailNotifications)
|
|
}
|
|
if req.PushNotifications != nil {
|
|
pref.PushNotifications = model.BoolPtr(*req.PushNotifications)
|
|
}
|
|
if req.MarketingNotifications != nil {
|
|
pref.MarketingNotifications = *req.MarketingNotifications
|
|
}
|
|
if req.TelegramNotifications != nil {
|
|
pref.TelegramNotifications = *req.TelegramNotifications
|
|
}
|
|
if req.Language != nil {
|
|
pref.Language = model.StringPtr(strings.TrimSpace(*req.Language))
|
|
}
|
|
if req.Locale != nil {
|
|
pref.Locale = model.StringPtr(strings.TrimSpace(*req.Locale))
|
|
}
|
|
if strings.TrimSpace(model.StringValue(pref.Language)) == "" {
|
|
pref.Language = model.StringPtr("en")
|
|
}
|
|
if strings.TrimSpace(model.StringValue(pref.Locale)) == "" {
|
|
pref.Locale = model.StringPtr(model.StringValue(pref.Language))
|
|
}
|
|
|
|
if err := db.WithContext(ctx).Save(pref).Error; err != nil {
|
|
l.Error("Failed to save preferences", "error", err)
|
|
return nil, err
|
|
}
|
|
|
|
return pref, nil
|
|
}
|