package database

import (
	"github.com/gotify/server/model"
)

// GetApplicationByToken returns the application for the given token or nil.
func (d *GormDatabase) GetApplicationByToken(token string) *model.Application {
	app := new(model.Application)
	d.DB.Where("token = ?", token).Find(app)
	if app.Token == token {
		return app
	}
	return nil
}

// GetApplicationByID returns the application for the given id or nil.
func (d *GormDatabase) GetApplicationByID(id uint) *model.Application {
	app := new(model.Application)
	d.DB.Where("id = ?", id).Find(app)
	if app.ID == id {
		return app
	}
	return nil
}

// CreateApplication creates an application.
func (d *GormDatabase) CreateApplication(application *model.Application) error {
	return d.DB.Create(application).Error
}

// DeleteApplicationByID deletes an application by its id.
func (d *GormDatabase) DeleteApplicationByID(id uint) error {
	d.DeleteMessagesByApplication(id)
	return d.DB.Where("id = ?", id).Delete(&model.Application{}).Error
}

// GetApplicationsByUser returns all applications from a user.
func (d *GormDatabase) GetApplicationsByUser(userID uint) []*model.Application {
	var apps []*model.Application
	d.DB.Where("user_id = ?", userID).Find(&apps)
	return apps
}

// UpdateApplication updates an application.
func (d *GormDatabase) UpdateApplication(app *model.Application) error {
	return d.DB.Save(app).Error
}