713ab02450
* Added configor as a new library
Now config is a singleton. You only need to do config.Get() instead of doing config.Conf.
* Forgot godep save 🐤
* Fix accidental removal of }
415 lignes
12 Kio
Go
415 lignes
12 Kio
Go
package controllers
|
|
|
|
import (
|
|
"errors"
|
|
"html"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
"github.com/NyaaPantsu/nyaa/models"
|
|
"github.com/NyaaPantsu/nyaa/models/users"
|
|
"github.com/NyaaPantsu/nyaa/utils/categories"
|
|
"github.com/NyaaPantsu/nyaa/utils/feeds"
|
|
"github.com/NyaaPantsu/nyaa/utils/publicSettings"
|
|
"github.com/NyaaPantsu/nyaa/utils/search"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gorilla/feeds"
|
|
)
|
|
|
|
// RSSHandler : Controller for displaying rss feed, accepting common search arguments
|
|
func RSSHandler(c *gin.Context) {
|
|
// We only get the basic variable for rss based on search param
|
|
torrents, createdAsTime, title, err := getTorrentList(c)
|
|
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
feed := &nyaafeeds.RssFeed{
|
|
Title: title,
|
|
Link: config.WebAddress() + "/",
|
|
PubDate: createdAsTime.String(),
|
|
}
|
|
feed.Items = make([]*nyaafeeds.RssItem, len(torrents))
|
|
|
|
for i, torrent := range torrents {
|
|
torrentJSON := torrent.ToJSON()
|
|
feed.Items[i] = &nyaafeeds.RssItem{
|
|
Title: torrentJSON.Name,
|
|
Link: config.WebAddress() + "/download/" + torrentJSON.Hash,
|
|
Description: string(torrentJSON.Description),
|
|
PubDate: torrent.Date.Format(time.RFC822),
|
|
GUID: config.WebAddress() + "/view/" + strconv.FormatUint(uint64(torrentJSON.ID), 10),
|
|
Enclosure: &nyaafeeds.RssEnclosure{
|
|
URL: config.WebAddress() + "/download/" + strings.TrimSpace(torrentJSON.Hash),
|
|
Length: strconv.FormatUint(uint64(torrentJSON.Filesize), 10),
|
|
Type: "application/x-bittorrent",
|
|
},
|
|
}
|
|
}
|
|
// allow cross domain AJAX requests
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
rss, rssErr := feeds.ToXML(feed)
|
|
if rssErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, rssErr)
|
|
}
|
|
|
|
_, writeErr := c.Writer.Write([]byte(rss))
|
|
if writeErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, writeErr)
|
|
}
|
|
}
|
|
|
|
// RSSMagnetHandler : Controller for displaying rss feeds with magnet URL, accepting common search arguments
|
|
func RSSMagnetHandler(c *gin.Context) {
|
|
// We only get the basic variable for rss based on search param
|
|
torrents, createdAsTime, title, err := getTorrentList(c)
|
|
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
feed := &nyaafeeds.RssFeed{
|
|
Title: title,
|
|
Link: config.WebAddress() + "/",
|
|
PubDate: createdAsTime.String(),
|
|
}
|
|
feed.Items = make([]*nyaafeeds.RssItem, len(torrents))
|
|
|
|
for i, torrent := range torrents {
|
|
torrentJSON := torrent.ToJSON()
|
|
feed.Items[i] = &nyaafeeds.RssItem{
|
|
Title: torrentJSON.Name,
|
|
Link: &nyaafeeds.RssMagnetLink{Text: string(torrentJSON.Magnet)},
|
|
Description: string(torrentJSON.Description),
|
|
PubDate: torrent.Date.Format(time.RFC822),
|
|
GUID: config.WebAddress() + "/view/" + strconv.FormatUint(uint64(torrentJSON.ID), 10),
|
|
Enclosure: &nyaafeeds.RssEnclosure{
|
|
URL: config.WebAddress() + "/download/" + strings.TrimSpace(torrentJSON.Hash),
|
|
Length: strconv.FormatUint(uint64(torrentJSON.Filesize), 10),
|
|
Type: "application/x-bittorrent",
|
|
},
|
|
}
|
|
}
|
|
// allow cross domain AJAX requests
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
rss, rssErr := feeds.ToXML(feed)
|
|
if rssErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, rssErr)
|
|
}
|
|
|
|
_, writeErr := c.Writer.Write([]byte(rss))
|
|
if writeErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, writeErr)
|
|
}
|
|
}
|
|
|
|
// RSSEztvHandler : Controller for displaying rss feed, accepting common search arguments
|
|
func RSSEztvHandler(c *gin.Context) {
|
|
// We only get the basic variable for rss based on search param
|
|
torrents, createdAsTime, title, err := getTorrentList(c)
|
|
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
feed := &nyaafeeds.RssFeed{
|
|
Title: title,
|
|
Link: config.WebAddress() + "/",
|
|
PubDate: createdAsTime.String(),
|
|
}
|
|
feed.Items = make([]*nyaafeeds.RssItem, len(torrents))
|
|
|
|
for i, torrent := range torrents {
|
|
torrentJSON := torrent.ToJSON()
|
|
feed.Items[i] = &nyaafeeds.RssItem{
|
|
Title: torrentJSON.Name,
|
|
Link: config.WebAddress() + "/download/" + torrentJSON.Hash,
|
|
Category: &nyaafeeds.RssCategory{
|
|
Domain: config.WebAddress() + "/search?c=" + torrentJSON.Category + "_" + torrentJSON.SubCategory,
|
|
},
|
|
Description: string(torrentJSON.Description),
|
|
Comments: config.WebAddress() + "/view/" + strconv.FormatUint(uint64(torrentJSON.ID), 10),
|
|
PubDate: torrent.Date.Format(time.RFC822),
|
|
GUID: config.WebAddress() + "/view/" + strconv.FormatUint(uint64(torrentJSON.ID), 10),
|
|
Enclosure: &nyaafeeds.RssEnclosure{
|
|
URL: config.WebAddress() + "/download/" + strings.TrimSpace(torrentJSON.Hash),
|
|
Length: strconv.FormatUint(uint64(torrentJSON.Filesize), 10),
|
|
Type: "application/x-bittorrent",
|
|
},
|
|
Torrent: &nyaafeeds.RssTorrent{
|
|
Xmlns: "http://xmlns.ezrss.it/0.1/",
|
|
FileName: torrentJSON.Name + ".torrent",
|
|
ContentLength: strconv.FormatUint(uint64(torrentJSON.Filesize), 10),
|
|
InfoHash: torrentJSON.Hash,
|
|
MagnetURI: string(torrentJSON.Magnet),
|
|
},
|
|
}
|
|
}
|
|
// allow cross domain AJAX requests
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
rss, rssErr := feeds.ToXML(feed)
|
|
if rssErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, rssErr)
|
|
}
|
|
|
|
_, writeErr := c.Writer.Write([]byte(rss))
|
|
if writeErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, writeErr)
|
|
}
|
|
}
|
|
|
|
// RSSTorznabHandler : Controller for displaying rss feed, accepting common search arguments
|
|
func RSSTorznabHandler(c *gin.Context) {
|
|
t := c.Query("t")
|
|
rss := ""
|
|
title := "Nyaa Pantsu"
|
|
if config.IsSukebei() {
|
|
title = "Sukebei Pantsu"
|
|
}
|
|
if t == "caps" {
|
|
T := publicSettings.GetTfuncFromRequest(c)
|
|
cats := categories.GetSelect(true, true)
|
|
var categories []*nyaafeeds.RssCategoryTorznab
|
|
categories = append(categories, &nyaafeeds.RssCategoryTorznab{
|
|
ID: "5070",
|
|
Name: "Anime",
|
|
Description: "Anime",
|
|
})
|
|
|
|
last := 0
|
|
for _, v := range cats {
|
|
if len(v.ID) <= 2 {
|
|
categories = append(categories, &nyaafeeds.RssCategoryTorznab{
|
|
ID: nyaafeeds.ConvertFromCat(v.ID),
|
|
Name: string(T(v.Name)),
|
|
})
|
|
last++
|
|
} else {
|
|
categories[last].Subcat = append(categories[last].Subcat, &nyaafeeds.RssSubCat{
|
|
ID: nyaafeeds.ConvertFromCat(v.ID),
|
|
Name: string(T(v.Name)),
|
|
})
|
|
}
|
|
}
|
|
feed := &nyaafeeds.RssCaps{
|
|
Server: &nyaafeeds.RssServer{
|
|
Version: "1.0",
|
|
Title: title,
|
|
Strapline: "...",
|
|
Email: config.Get().Email.From,
|
|
URL: config.WebAddress(),
|
|
Image: config.WebAddress() + "/img/logo.png",
|
|
},
|
|
Limits: &nyaafeeds.RssLimits{
|
|
Max: "300",
|
|
Default: "50",
|
|
},
|
|
Registration: &nyaafeeds.RssRegistration{
|
|
Available: "yes",
|
|
Open: "yes",
|
|
},
|
|
Searching: &nyaafeeds.RssSearching{
|
|
Search: &nyaafeeds.RssSearch{
|
|
Available: "yes",
|
|
SupportedParams: "q",
|
|
},
|
|
TvSearch: &nyaafeeds.RssSearch{
|
|
Available: "no",
|
|
},
|
|
MovieSearch: &nyaafeeds.RssSearch{
|
|
Available: "no",
|
|
},
|
|
},
|
|
Categories: &nyaafeeds.RssCategories{
|
|
Category: categories,
|
|
},
|
|
}
|
|
var rssErr error
|
|
rss, rssErr = feeds.ToXML(feed)
|
|
if rssErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, rssErr)
|
|
}
|
|
} else {
|
|
// We only get the basic variable for rss based on search param
|
|
torrents, createdAsTime, title, err := getTorrentList(c)
|
|
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
feed := &nyaafeeds.RssFeed{
|
|
Title: title,
|
|
Xmlns: "http://torznab.com/schemas/2015/feed",
|
|
Link: config.WebAddress() + "/",
|
|
PubDate: createdAsTime.String(),
|
|
}
|
|
feed.Items = make([]*nyaafeeds.RssItem, len(torrents))
|
|
|
|
for i, torrent := range torrents {
|
|
|
|
torrentJSON := torrent.ToJSON()
|
|
filesNumber := ""
|
|
if len(torrentJSON.FileList) > 0 {
|
|
filesNumber = strconv.Itoa(len(torrentJSON.FileList))
|
|
}
|
|
seeders := ""
|
|
if torrentJSON.Seeders > 0 {
|
|
seeders = strconv.Itoa(int(torrentJSON.Seeders))
|
|
}
|
|
leechers := ""
|
|
if torrentJSON.Leechers > 0 {
|
|
leechers = strconv.Itoa(int(torrentJSON.Leechers))
|
|
}
|
|
feed.Items[i] = &nyaafeeds.RssItem{
|
|
Title: torrentJSON.Name,
|
|
Link: config.WebAddress() + "/download/" + torrentJSON.Hash,
|
|
Category: &nyaafeeds.RssCategory{
|
|
Domain: config.WebAddress() + "/search?c=" + torrentJSON.Category + "_" + torrentJSON.SubCategory,
|
|
},
|
|
Description: string(torrentJSON.Description),
|
|
Comments: config.WebAddress() + "/view/" + strconv.FormatUint(uint64(torrentJSON.ID), 10),
|
|
PubDate: torrent.Date.Format(time.RFC822),
|
|
GUID: config.WebAddress() + "/view/" + strconv.FormatUint(uint64(torrentJSON.ID), 10),
|
|
Enclosure: &nyaafeeds.RssEnclosure{
|
|
URL: config.WebAddress() + "/download/" + strings.TrimSpace(torrentJSON.Hash),
|
|
Length: strconv.FormatUint(uint64(torrentJSON.Filesize), 10),
|
|
Type: "application/x-bittorrent",
|
|
},
|
|
}
|
|
torznab := []*nyaafeeds.RssTorznab{}
|
|
if torrentJSON.Filesize > 0 {
|
|
torznab = append(torznab, &nyaafeeds.RssTorznab{
|
|
Name: "size",
|
|
Value: strconv.FormatUint(uint64(torrentJSON.Filesize), 10),
|
|
})
|
|
}
|
|
if filesNumber != "" {
|
|
torznab = append(torznab, &nyaafeeds.RssTorznab{
|
|
Name: "files",
|
|
Value: filesNumber,
|
|
})
|
|
}
|
|
torznab = append(torznab, &nyaafeeds.RssTorznab{
|
|
Name: "grabs",
|
|
Value: strconv.Itoa(int(torrentJSON.Completed)),
|
|
})
|
|
if seeders != "" {
|
|
torznab = append(torznab, &nyaafeeds.RssTorznab{
|
|
Name: "seeders",
|
|
Value: seeders,
|
|
})
|
|
}
|
|
if leechers != "" {
|
|
torznab = append(torznab, &nyaafeeds.RssTorznab{
|
|
Name: "leechers",
|
|
Value: leechers,
|
|
})
|
|
}
|
|
if torrentJSON.Hash != "" {
|
|
torznab = append(torznab, &nyaafeeds.RssTorznab{
|
|
Name: "infohash",
|
|
Value: torrentJSON.Hash,
|
|
})
|
|
}
|
|
if torrentJSON.Magnet != "" {
|
|
torznab = append(torznab, &nyaafeeds.RssTorznab{
|
|
Name: "magneturl",
|
|
Value: string(torrentJSON.Magnet),
|
|
})
|
|
}
|
|
if len(torznab) > 0 {
|
|
feed.Items[i].Torznab = torznab
|
|
}
|
|
}
|
|
var rssErr error
|
|
rss, rssErr = feeds.ToXML(feed)
|
|
if rssErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, rssErr)
|
|
}
|
|
}
|
|
// allow cross domain AJAX requests
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
|
|
_, writeErr := c.Writer.Write([]byte(rss))
|
|
if writeErr != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, writeErr)
|
|
}
|
|
}
|
|
|
|
func getTorrentList(c *gin.Context) (torrents []models.Torrent, createdAsTime time.Time, title string, err error) {
|
|
page := c.Param("page")
|
|
userID := c.Param("id")
|
|
cat := c.Query("cat")
|
|
offset := 0
|
|
if c.Query("offset") != "" {
|
|
offset, err = strconv.Atoi(html.EscapeString(c.Query("offset")))
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
createdAsTime = time.Now()
|
|
|
|
if len(torrents) > 0 {
|
|
createdAsTime = torrents[0].Date
|
|
}
|
|
|
|
title = "Nyaa Pantsu"
|
|
if config.IsSukebei() {
|
|
title = "Sukebei Pantsu"
|
|
}
|
|
|
|
pagenum := 1
|
|
if page == "" && offset > 0 { // first page for offset is 0
|
|
pagenum = offset + 1
|
|
} else if page != "" {
|
|
pagenum, err = strconv.Atoi(html.EscapeString(page))
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
if pagenum <= 0 {
|
|
err = errors.New("Page number is invalid")
|
|
return
|
|
}
|
|
|
|
if userID != "" {
|
|
userIDnum := 0
|
|
userIDnum, err = strconv.Atoi(html.EscapeString(userID))
|
|
// Should we have a feed for anonymous uploads?
|
|
if err != nil || userIDnum == 0 {
|
|
return
|
|
}
|
|
|
|
_, _, err = users.FindForAdmin(uint(userIDnum))
|
|
if err != nil {
|
|
return
|
|
}
|
|
// Set the user ID on the request, so that SearchByQuery finds it.
|
|
query := c.Request.URL.Query()
|
|
query.Set("userID", userID)
|
|
c.Request.URL.RawQuery = query.Encode()
|
|
}
|
|
|
|
if cat != "" {
|
|
query := c.Request.URL.Query()
|
|
catConv := nyaafeeds.ConvertToCat(cat)
|
|
if catConv == "" {
|
|
return
|
|
}
|
|
query.Set("c", catConv)
|
|
c.Request.URL.RawQuery = query.Encode()
|
|
}
|
|
|
|
_, torrents, err = search.ByQueryNoCount(c, pagenum)
|
|
|
|
return
|
|
}
|