package chain

import "gorm.io/gorm"

type Repository struct {
	db *gorm.DB
}

func NewRepository(db *gorm.DB) *Repository {
	return &Repository{db: db}
}

func (r *Repository) Create(w *UserWallet) error {
	return r.db.Create(w).Error
}

func (r *Repository) FindByUserID(userID uint) (*UserWallet, error) {
	var w UserWallet
	if err := r.db.Where("user_id = ?", userID).First(&w).Error; err != nil {
		return nil, err
	}
	return &w, nil
}

func (r *Repository) FindByPubKeyHex(pubKeyHex string) (*UserWallet, error) {
	var w UserWallet
	if err := r.db.Where("pub_key_hex = ?", pubKeyHex).First(&w).Error; err != nil {
		return nil, err
	}
	return &w, nil
}