package dao

import (
	"server/global"
	"time"
)

type Shortlist struct {
	global.GVA_MODEL
	ListType       string    `json:"list_type" gorm:"comment:名单类型"`
	VehicleId      int       `json:"vehicle_id" gorm:"comment:车辆ID"`
	Vehicle        Vehicle   `json:"vehicle" gorm:"foreignKey:VehicleId"`
	ExpirationTime time.Time `json:"expiration_time" gorm:"comment:过期时间;default:null"`
}

func (Shortlist) TableName() string {
	return "shortlist"
}

func QueryShortlistList(limit, offset int) (list []Shortlist, total int64, err error) {
	db := global.GVA_DB.Model(&Shortlist{})

	err = db.Count(&total).Error

	if err != nil {
		return list, total, err
	}

	db = db.Limit(limit).Offset(offset)

	err = db.Find(&list).Error
	return list, total, err
}

func QueryAllShortlists() (list []Shortlist, err error) {
	err = global.GVA_DB.Model(&Shortlist{}).Find(&list).Error

	return list, err
}

func CreateShortlist(shortlist Shortlist) error {
	return global.GVA_DB.Create(&shortlist).Error
}

func UpdateShortlist(shortlist *Shortlist) error {
	return global.GVA_DB.Where("id = ?", shortlist.ID).Updates(shortlist).Error
}

func DeleteShortlist(id int) error {
	return global.GVA_DB.Where("id = ?", id).Unscoped().Delete(&Shortlist{}).Error
}