Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/models/tag.go

116 lignes
2,7 Kio
Go

package models
import (
"encoding/json"
"errors"
"net/http"
"github.com/NyaaPantsu/nyaa/utils/log"
"github.com/fatih/structs"
)
// Tag model for a torrent vote system
type Tag struct {
TorrentID uint `gorm:"column:torrent_id" json:"torrent_id"`
UserID uint `gorm:"column:user_id" json:"user_id"`
Tag string `gorm:"column:tag" json:"tag"`
Type string `gorm:"column:type" json:"type"`
Weight float64 `gorm:"column:weight" json:"weight"`
Accepted bool `gorm:"column:accepted" json:"accepted"`
Total float64 `gorm:"-" json:"total"`
}
// Update a tag
func (ta *Tag) Update() (int, error) {
if ORM.Model(ta).UpdateColumn(ta.toMap()).Error != nil {
return http.StatusInternalServerError, errors.New("Tag was not updated")
}
return http.StatusOK, nil
}
// Delete : delete a tag based on id
func (ta *Tag) Delete() (int, error) {
if ORM.Where("tag = ? AND type = ? AND torrent_id = ? AND user_id = ?", ta.Tag, ta.Type, ta.TorrentID, ta.UserID).Delete(ta).Error != nil {
return http.StatusInternalServerError, errors.New("tag_not_deleted")
}
return http.StatusOK, nil
}
// toMap : convert the model to a map of interface
func (ta *Tag) toMap() map[string]interface{} {
return structs.Map(ta)
}
type Tags []Tag
// Contains check if the tag map has the same tag in it (tag value + tag type)
func (ts Tags) Contains(tag Tag) bool {
for _, ta := range ts {
if ta.Tag == tag.Tag && ta.Type == tag.Type {
return true
}
}
return false
}
// HasType check if the tag map has the same tag type in it and returns its index
func (ts Tags) HasType(tagtype string) int {
for i, ta := range ts {
if ta.Type == tagtype {
return i
}
}
return -1
}
// DeleteType remove all tags from the map that have the same tag type in it
func (ts *Tags) DeleteType(tagtype string) {
var newTs Tags
for _, ta := range *ts {
if ta.Type == tagtype {
continue
}
newTs = append(newTs, ta)
}
ts = &newTs
}
// HasAccepted check if a tag has been accepted in the tags map
func (ts Tags) HasAccepted() bool {
for _, tag := range ts {
if tag.Accepted {
return true
}
}
return false
}
// Replace a tag in map of tags
func (ts Tags) Replace(index int, tag *Tag) {
if index >= 0 && index < len(ts) {
ts[index].Delete()
ts[index] = *tag
ts[index].Update()
}
}
// ToJSON convert tags map to a json map and can exclud non accepted tags
func (ts Tags) ToJSON(onlyAccepted bool) string {
var toParse Tags
if onlyAccepted {
for _, tag := range ts {
if tag.Accepted {
toParse = append(toParse, tag)
}
}
} else {
toParse = ts
}
b, err := json.Marshal(toParse)
if err != nil {
log.Infof("Couldn't parse to json the tags %v", toParse)
return ""
}
return string(b)
}