package db

import (
	"gorm.io/gorm"
)

type Entry struct {
	gorm.Model
	Title        string
	Link         string
	Format       string
	OutputFolder string
}

type EntryModel struct {
	DB *gorm.DB
}

func (m *EntryModel) Find(id string) Entry {
	var entry Entry
	where := "id = " + id
	m.DB.Where(where).FirstOrInit(&entry)
	return entry
}

func (m *EntryModel) All() []Entry {
	var entries []Entry
	m.DB.Find(&entries)
	return entries
}

func (m *EntryModel) Create(entry *Entry) {
	m.DB.Create(entry)
}

func (m *EntryModel) Update(entry Entry) {
	m.DB.Save(&entry)
}

func (m *EntryModel) Delete(id string) {
	m.DB.Delete(&Entry{}, id)
}