-
Notifications
You must be signed in to change notification settings - Fork 150
/
report.go
67 lines (57 loc) · 1.86 KB
/
report.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
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"`
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"`
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, 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(definitely bool) (int, error) {
if definitely {
return 0, ORM.Unscoped().Delete(report).Error
}
return http.StatusOK, nil
}