nyaa-pantsu/model/torrent.go

package model

import (
	"github.com/ewhal/nyaa/config"
	"github.com/ewhal/nyaa/util"

	"fmt"
	"html/template"
	"strconv"
	"strings"
	"time"
)

type Feed struct {
	Id        int
	Name      string
	Hash      string
	Magnet    string
	Timestamp string
}

type Torrents struct {
	Id           uint      `gorm:"column:torrent_id;primary_key"`
	Name         string    `gorm:"column:torrent_name"`
	Hash         string    `gorm:"column:torrent_hash"`
	Category     int       `gorm:"column:category"`
	Sub_Category int       `gorm:"column:sub_category"`
	Status       int       `gorm:"column:status"`
	Date         time.Time `gorm:"column:date"`
	UploaderId   uint      `gorm:"column:uploader"`
	Downloads    int       `gorm:"column:downloads"`
	Stardom      int       `gorm:"column:stardom"`
	Filesize     int64     `gorm:"column:filesize"`
	Description  string    `gorm:"column:description"`
	WebsiteLink  string    `gorm:"column:website_link"`

	Uploader    *User        `gorm:"ForeignKey:UploaderId"`
	OldComments []OldComment `gorm:"ForeignKey:torrent_id"`
	Comments    []Comment    `gorm:"ForeignKey:torrent_id"`
}

/* We need JSON Object instead because of Magnet URL that is not in the database but generated dynamically */

type ApiResultJson struct {
	Torrents         []TorrentsJson `json:"torrents"`
	QueryRecordCount int            `json:"queryRecordCount"`
	TotalRecordCount int            `json:"totalRecordCount"`
}

type CommentsJson struct {
	Username string        `json:"username"`
	Content  template.HTML `json:"content"`
	Date     time.Time     `json:"date"`
}

type TorrentsJson struct {
	Id           string         `json:"id"`
	Name         string         `json:"name"`
	Status       int            `json:"status"`
	Hash         string         `json:"hash"`
	Date         string         `json:"date"`
	Filesize     string         `json:"filesize"`
	Description  template.HTML  `json:"description"`
	Comments     []CommentsJson `json:"comments"`
	Sub_Category string         `json:"sub_category"`
	Category     string         `json:"category"`
	Downloads    int            `json:"downloads"`
	UploaderId   uint           `json:"uploader_id"`
	UploaderName template.HTML  `json:"uploader_name"`
	WebsiteLink  template.URL   `json:"website_link"`
	Magnet       template.URL   `json:"magnet"`
	TorrentLink  template.URL   `json:"torrent"`
}

/* Model Conversion to Json */

func (t *Torrents) ToJson() TorrentsJson {
	magnet := util.InfoHashToMagnet(strings.TrimSpace(t.Hash), t.Name, config.Trackers...)
	commentsJson := make([]CommentsJson, 0, len(t.OldComments)+len(t.Comments))
	for _, c := range t.OldComments {
		commentsJson = append(commentsJson, CommentsJson{Username: c.Username, Content: template.HTML(c.Content), Date: c.Date})
	}
	for _, c := range t.Comments {
		commentsJson = append(commentsJson, CommentsJson{Username: c.User.Username, Content: util.MarkdownToHTML(c.Content), Date: c.CreatedAt})
	}
	uploader := ""
	if t.Uploader != nil {
		uploader = t.Uploader.Username
	}
	torrentlink := ""
	if t.Id <= config.LastOldTorrentId && len(config.TorrentCacheLink) > 0 {
		torrentlink = fmt.Sprintf(config.TorrentCacheLink, t.Hash)
	} else if t.Id > config.LastOldTorrentId && len(config.TorrentStorageLink) > 0 {
		torrentlink = fmt.Sprintf(config.TorrentStorageLink, t.Hash)
	}
	res := TorrentsJson{
		Id:           strconv.FormatUint(uint64(t.Id), 10),
		Name:         t.Name,
		Status:       t.Status,
		Hash:         t.Hash,
		Date:         t.Date.Format(time.RFC3339),
		Filesize:     util.FormatFilesize2(t.Filesize),
		Description:  util.MarkdownToHTML(t.Description),
		Comments:     commentsJson,
		Sub_Category: strconv.Itoa(t.Sub_Category),
		Category:     strconv.Itoa(t.Category),
		Downloads:    t.Downloads,
		UploaderId:   t.UploaderId,
		UploaderName: util.SafeText(uploader),
		WebsiteLink:  util.Safe(t.WebsiteLink),
		Magnet:       util.Safe(magnet),
		TorrentLink:  util.Safe(torrentlink)}

	return res
}

/* Complete the functions when necessary... */

// Map Torrents to TorrentsToJSON without reallocations
func TorrentsToJSON(t []Torrents) []TorrentsJson {
	json := make([]TorrentsJson, len(t))
	for i := range t {
		json[i] = t[i].ToJson()
	}
	return json
}