devices.go

package repo

import (
	"errors"
	"regexp"
	"time"

	apierrors "github.com/wearedevx/keystone/api/internal/errors"
	"github.com/wearedevx/keystone/api/pkg/models"
	"gorm.io/gorm"
)

func (r *Repo) GetDevice(device *models.Device) IRepo {
	if r.Err() != nil {
		return r
	}

	r.err = r.GetDb().
		Where(&device).
		First(device).
		Error

	return r
}

func (r *Repo) GetDeviceByUserID(userID uint, device *models.Device) IRepo {
	if r.Err() != nil {
		return r
	}

	r.err = r.GetDb().
		Joins("left join user_devices on user_devices.device_id = devices.id").
		Joins("left join users on users.id = user_devices.user_id").
		Where(&device).
		Where("users.id = ?", userID).
		First(device).
		Error

	return r
}

func (r *Repo) GetDevices(userID uint, devices *[]models.Device) IRepo {
	if r.Err() != nil {
		return r
	}

	r.err = r.GetDb().
		Joins("left join user_devices on user_devices.device_id = devices.id").
		Joins("left join users on users.id = user_devices.user_id").
		Where("users.id = ?", userID).
		Find(devices).
		Error

	return r
}
func (r *Repo) GetNewlyCreatedDevices(devices *[]models.Device) IRepo {
	if r.Err() != nil {
		return r
	}

	r.err = r.GetDb().
		Preload("Users").
		Joins("left join user_devices on user_devices.device_id = devices.id").
		Where("user_devices.newly_created = true").
		Find(devices).
		Error

	return r
}

func (r *Repo) UpdateDeviceLastUsedAt(deviceUID string) IRepo {
	if r.Err() != nil {
		return r
	}

	device := models.Device{
		UID: deviceUID,
	}

	if r.err = r.GetDevice(&device).Err(); r.err != nil {
		return r
	}

	device.LastUsedAt = time.Now()

	r.err = r.GetDb().Save(&device).Error

	return r
}

func (r *Repo) RevokeDevice(userID uint, deviceUID string) IRepo {
	if r.Err() != nil {
		return r
	}

	device := models.Device{}
	r.err = r.GetDb().
		Joins("left join user_devices on user_devices.device_id = devices.id").
		Joins("left join users on users.id = user_devices.user_id").
		Where("users.id = ? and devices.uid = ?", userID, deviceUID).
		First(&device).
		Error

	if r.err != nil {
		return r
	}

	// Remove message aimed for the device
	r.err = r.GetDb().
		Model(&models.Message{}).
		Where("recipient_device_id = ? and sender_device_id = ?", device.ID, device.ID).
		Delete(models.Message{}).Error

	if r.err != nil {
		return r
	}

	r.err = r.GetDb().
		Delete(&device).Error

	if r.err != nil {
		return r
	}

	return r
}

func (r *Repo) AddNewDevice(
	device models.Device,
	user models.User,
) IRepo {
	db := r.GetDb()

	// TODO: this should probably go to the controller
	matched, _ := regexp.MatchString(`^[a-zA-Z0-9\.\-\_]{1,}$`, device.Name)
	if !matched {
		r.err = apierrors.ErrorBadDeviceName()
		return r
	}

	if err := db.Where("uid = ?", device.UID).First(&device).Error; err != nil {
		if errors.Is(gorm.ErrRecordNotFound, err) {
			r.err = db.Create(&device).Error
		} else {
			r.err = err
		}
	}

	if r.err != nil {
		return r
	}

	userDevice := models.UserDevice{UserID: user.ID, DeviceID: device.ID, NewlyCreated: true}

	err := db.SetupJoinTable(&models.User{}, "Devices", &models.UserDevice{})
	if err != nil {
		r.err = err
		return r
	}

	r.err = db.Create(&userDevice).Error

	if r.err != nil {
		return r
	}

	return r
}

func (r *Repo) SetNewlyCreatedDevice(flag bool, deviceID uint, userID uint) IRepo {

	if err := r.GetDb().Model(&models.UserDevice{}).Where("user_id = ? and device_id = ?", userID, deviceID).Update("newly_created", flag).Error; err != nil {
		r.err = err
	}
	return r
}