asg-server/internal/sms-gateway/modules/devices/service.go

package devices

import (
	"context"
	"crypto/sha256"
	"encoding/hex"
	"fmt"
	"time"

	"github.com/android-sms-gateway/server/internal/sms-gateway/models"
	"github.com/android-sms-gateway/server/internal/sms-gateway/modules/db"
	"github.com/capcom6/go-helpers/cache"
	"go.uber.org/fx"
	"go.uber.org/zap"
)

type ServiceParams struct {
	fx.In

	Config Config

	Devices *repository

	IDGen db.IDGen

	Logger *zap.Logger
}

type Service struct {
	config Config

	devices     *repository
	tokensCache *cache.Cache[models.Device]

	idGen db.IDGen

	logger *zap.Logger
}

func (s *Service) Insert(userID string, device *models.Device) error {
	device.ID = s.idGen()
	device.AuthToken = s.idGen()
	device.UserID = userID

	return s.devices.Insert(device)
}

// Select returns a list of devices for a specific user that match the provided filters.
func (s *Service) Select(userID string, filter ...SelectFilter) ([]models.Device, error) {
	filter = append(filter, WithUserID(userID))

	return s.devices.Select(filter...)
}

// Exists checks if there exists a device that matches the provided filters.
//
// If the device does not exist, it returns false and nil error. If there is an
// error during the query, it returns false and the error. Otherwise, it returns
// true and nil error.
func (s *Service) Exists(userID string, filter ...SelectFilter) (bool, error) {
	filter = append(filter, WithUserID(userID))

	return s.devices.Exists(filter...)
}

// Get returns a single device based on the provided filters for a specific user.
// It ensures that the filter includes the user's ID. If no device matches the
// criteria, it returns ErrNotFound. If more than one device matches, it returns
// ErrMoreThanOne.
func (s *Service) Get(userID string, filter ...SelectFilter) (models.Device, error) {
	filter = append(filter, WithUserID(userID))

	return s.devices.Get(filter...)
}

// GetByToken returns a device by token.
//
// This method is used to retrieve a device by its auth token. If the device
// does not exist, it returns ErrNotFound.
func (s *Service) GetByToken(token string) (models.Device, error) {
	hash := sha256.Sum256([]byte(token))
	cacheKey := hex.EncodeToString(hash[:])

	device, err := s.tokensCache.Get(cacheKey)
	if err != nil {
		device, err = s.devices.Get(WithToken(token))
		if err != nil {
			return device, fmt.Errorf("can't get device: %w", err)
		}

		if err := s.tokensCache.Set(cacheKey, device); err != nil {
			s.logger.Error("can't cache device", zap.Error(err))
		}
	}

	return device, nil
}

func (s *Service) UpdatePushToken(deviceId string, token string) error {
	return s.devices.UpdatePushToken(deviceId, token)
}

func (s *Service) SetLastSeen(ctx context.Context, batch map[string]time.Time) error {
	if len(batch) == 0 {
		return nil
	}

	for deviceId, lastSeen := range batch {
		if ctx.Err() != nil {
			break
		}

		if err := s.devices.SetLastSeen(ctx, deviceId, lastSeen); err != nil {
			s.logger.Error("can't set last seen", zap.String("device_id", deviceId), zap.Error(err))
		}
	}

	return ctx.Err()
}

// Remove removes devices for a specific user that match the provided filters.
// It ensures that the filter includes the user's ID.
func (s *Service) Remove(userID string, filter ...SelectFilter) error {
	filter = append(filter, WithUserID(userID))

	device, err := s.Get(userID, filter...)
	if err != nil {
		return err
	}

	if err := s.tokensCache.Delete(device.AuthToken); err != nil {
		s.logger.Error("can't invalidate token cache", zap.Error(err))
	}

	return s.devices.Remove(filter...)
}

func (s *Service) Clean(ctx context.Context) error {
	n, err := s.devices.removeUnused(ctx, time.Now().Add(-s.config.UnusedLifetime))

	s.logger.Info("Cleaned unused devices", zap.Int64("count", n))
	return err
}

func NewService(params ServiceParams) *Service {
	return &Service{
		config:      params.Config,
		devices:     params.Devices,
		tokensCache: cache.New[models.Device](cache.Config{TTL: 10 * time.Minute}),
		idGen:       params.IDGen,
		logger:      params.Logger.Named("service"),
	}
}