67 lignes
1.9 KiB
Go
67 lignes
1.9 KiB
Go
package models
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
)
|
|
|
|
// TorrentReport model
|
|
// User can be null (anonymous reports)
|
|
// FIXME can't preload field Torrents for models.TorrentReport
|
|
type TorrentReport struct {
|
|
ID uint `gorm:"column:torrent_report_id;primary_key"`
|
|
Description string `gorm:"column:type"`
|
|
Message string `gorm:"column:message"`
|
|
TorrentID uint `gorm:"column:torrent_id"`
|
|
UserID uint `gorm:"column:user_id"`
|
|
|
|
CreatedAt time.Time `gorm:"column:created_at"`
|
|
|
|
Torrent *Torrent `gorm:"AssociationForeignKey:TorrentID;ForeignKey:torrent_id"`
|
|
User *User `gorm:"AssociationForeignKey:UserID;ForeignKey:user_id"`
|
|
}
|
|
|
|
// TableName : Return the name of torrent report table
|
|
func (report TorrentReport) TableName() string {
|
|
return config.Get().Models.ReportsTableName
|
|
}
|
|
|
|
// TorrentReportJSON : Json struct of torrent report model
|
|
type TorrentReportJSON struct {
|
|
ID uint `json:"id"`
|
|
Description string `json:"description"`
|
|
Message string `json:"message"`
|
|
Torrent TorrentJSON `json:"torrent"`
|
|
User UserJSON `json:"user"`
|
|
}
|
|
|
|
// ToJSON : conversion to json of a torrent report
|
|
func (report *TorrentReport) ToJSON() TorrentReportJSON {
|
|
t := TorrentJSON{}
|
|
if report.Torrent != nil { // FIXME: report.Torrent should never be nil
|
|
t = report.Torrent.ToJSON()
|
|
}
|
|
u := UserJSON{}
|
|
if report.User != nil {
|
|
u = report.User.ToJSON()
|
|
}
|
|
json := TorrentReportJSON{report.ID, report.Description, report.Message, t, u}
|
|
return json
|
|
}
|
|
|
|
// TorrentReportsToJSON : Conversion of multiple reports to json
|
|
func TorrentReportsToJSON(reports []TorrentReport) []TorrentReportJSON {
|
|
json := make([]TorrentReportJSON, len(reports))
|
|
for i := range reports {
|
|
json[i] = reports[i].ToJSON()
|
|
}
|
|
return json
|
|
}
|
|
|
|
// Delete : Delete torrent report
|
|
func (report *TorrentReport) Delete() (int, error) {
|
|
return http.StatusOK, ORM.Unscoped().Delete(report).Error
|
|
}
|