package download

import "gorm.io/gorm"

type Repository struct {
	db *gorm.DB
}

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

func (r *Repository) GetLatest() (*Info, error) {
	var info Info
	if err := r.db.Last(&info).Error; err != nil {
		return nil, err
	}
	return &info, nil
}

func (r *Repository) Save(info *Info) error {
	if info.ID == 0 {
		return r.db.Create(info).Error
	}
	return r.db.Save(info).Error
}