user.go

package infrastructure

import (
	"errors"

	"github.com/scarlet0725/prism-api/infrastructure/repository"
	"github.com/scarlet0725/prism-api/model"
	"github.com/scarlet0725/prism-api/schema"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

type userRepository struct {
	db *gorm.DB
}

func NewUserRepository(db *gorm.DB) repository.User {
	return &userRepository{
		db: db,
	}
}

func (u *userRepository) GetUser(id string) (*model.User, error) {
	var user model.User
	err := u.db.Preload(clause.Associations).Where("user_id = ?", id).First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

func (u *userRepository) CreateUser(user *model.User) (*model.User, error) {
	var schema schema.User
	schema.User = *user

	result := u.db.Create(&schema)

	err := result.Error

	if err != nil {
		return nil, err
	}

	return user, nil
}

func (u *userRepository) UpdateUser(user *model.User) (*model.User, error) {
	var schema schema.User
	schema.User = *user
	err := u.db.Save(user).Error
	if err != nil {
		return nil, err
	}
	return &schema.User, nil
}

func (u *userRepository) DeleteUser(user *model.User) error {
	return u.db.Delete(user).Error
}

func (u *userRepository) GetUserByAPIKey(apiKey string) (*model.User, error) {
	var user model.User
	err := u.db.Preload(clause.Associations).Where("api_key = ?", apiKey).First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

func (u *userRepository) GetUserCalendarByUserID(id int) (*model.ExternalCalendar, error) {
	var calendar model.ExternalCalendar
	err := u.db.Where("user_id = ?", id).First(&calendar).Error
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, repository.ErrNotFound
		}
		return nil, err
	}
	return &calendar, nil
}

func (u *userRepository) GetGoogleCalendarConfig(id int) (*model.GoogleCalenderConfig, error) {
	var config *model.GoogleCalenderConfig

	err := u.db.Table("google_oauth_tokens").Select("*").Joins("INNER JOIN external_calendars ON google_oauth_tokens.user_id = external_calendars.user_id").Where("google_oauth_tokens.user_id = ? AND external_calendars.user_id = ?", id, id).Take(&config).Error

	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, repository.ErrNotFound
		}
		return nil, err
	}

	return config, nil

}

func (u *userRepository) GetGoogleOAuthToken(id int) (*model.GoogleOAuthToken, error) {
	token := &model.GoogleOAuthToken{}

	err := u.db.Table("google_oauth_tokens").Select("*").Where("user_id = ?", id).Take(&token).Error

	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, repository.ErrNotFound
		}
		return nil, err
	}

	return token, nil
}

func (u *userRepository) SaveExternalCalendar(cal *model.ExternalCalendar) (*model.ExternalCalendar, error) {
	err := u.db.Create(cal).Error
	if err != nil {
		return nil, err
	}
	return cal, nil
}

func (u *userRepository) AddRegistrationEvent(user *model.User, event *model.Event) error {
	err := u.db.Model(user).Association("Events").Append(event)
	return err
}

func (u *userRepository) RemoveRegistrationEvent(user *model.User, event *model.Event) error {
	err := u.db.Model(user).Association("Events").Delete(event)
	return err
}

func (u *userRepository) GetUserByUsername(username string) (*model.User, error) {
	var user model.User
	err := u.db.Preload(clause.Associations).Where("username = ?", username).First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}