2017-05-02 12:39:53 +02:00
package main
import (
"encoding/json"
2017-05-04 21:48:40 +02:00
"github.com/gorilla/feeds"
2017-05-02 12:39:53 +02:00
"github.com/gorilla/mux"
2017-05-04 05:28:42 +02:00
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/sqlite"
2017-05-02 12:39:53 +02:00
"html"
"html/template"
"log"
"net/http"
"strconv"
2017-05-03 06:59:27 +02:00
"strings"
2017-05-02 12:39:53 +02:00
"time"
)
2017-05-04 05:28:42 +02:00
var db * gorm . DB
2017-05-05 02:54:37 +02:00
var templates = template . Must ( template . ParseFiles ( "index.html" , "FAQ.html" , "view.html" ) )
2017-05-02 12:39:53 +02:00
var debugLogger * log . Logger
2017-05-03 09:33:39 +02:00
var trackers = "&tr=udp://zer0day.to:1337/announce&tr=udp://tracker.leechers-paradise.org:6969&tr=udp://explodie.org:6969&tr=udp://tracker.opentrackr.org:1337&tr=udp://tracker.coppersurfer.tk:6969"
2017-05-02 12:39:53 +02:00
2017-05-04 05:28:42 +02:00
func getDBHandle ( ) * gorm . DB {
dbInit , err := gorm . Open ( "sqlite3" , "./nyaa.db" )
2017-05-02 12:39:53 +02:00
2017-05-04 05:28:42 +02:00
// Migrate the schema of Torrents
2017-05-05 06:24:58 +02:00
dbInit . AutoMigrate ( & Torrents { } , & Categories { } , & Sub_Categories { } , & Statuses { } )
2017-05-02 12:39:53 +02:00
checkErr ( err )
2017-05-04 05:28:42 +02:00
return dbInit
2017-05-02 12:39:53 +02:00
}
func checkErr ( err error ) {
if err != nil {
debugLogger . Println ( " " + err . Error ( ) )
}
}
func apiHandler ( w http . ResponseWriter , r * http . Request ) {
vars := mux . Vars ( r )
page := vars [ "page" ]
pagenum , _ := strconv . Atoi ( html . EscapeString ( page ) )
2017-05-04 05:28:42 +02:00
b := CategoryJson { Torrents : [ ] TorrentsJson { } }
maxPerPage := 50
nbTorrents := 0
torrents := getAllTorrents ( maxPerPage , maxPerPage * ( pagenum - 1 ) )
for i , _ := range torrents {
nbTorrents ++
res := torrents [ i ] . toJson ( )
2017-05-02 12:39:53 +02:00
2017-05-04 05:28:42 +02:00
b . Torrents = append ( b . Torrents , res )
2017-05-02 12:39:53 +02:00
}
2017-05-04 05:28:42 +02:00
b . QueryRecordCount = maxPerPage
b . TotalRecordCount = nbTorrents
2017-05-02 12:39:53 +02:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2017-05-04 05:28:42 +02:00
err := json . NewEncoder ( w ) . Encode ( b )
2017-05-02 12:39:53 +02:00
if err != nil {
http . Error ( w , err . Error ( ) , http . StatusInternalServerError )
return
}
}
2017-05-04 17:16:20 +02:00
2017-05-05 02:54:37 +02:00
func apiViewHandler ( w http . ResponseWriter , r * http . Request ) {
2017-05-02 12:39:53 +02:00
vars := mux . Vars ( r )
id := vars [ "id" ]
2017-05-04 05:28:42 +02:00
b := CategoryJson { Torrents : [ ] TorrentsJson { } }
torrent , err := getTorrentById ( id )
res := torrent . toJson ( )
b . Torrents = append ( b . Torrents , res )
2017-05-02 12:39:53 +02:00
b . QueryRecordCount = 1
2017-05-04 05:28:42 +02:00
b . TotalRecordCount = 1
2017-05-02 12:39:53 +02:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
err = json . NewEncoder ( w ) . Encode ( b )
if err != nil {
http . Error ( w , err . Error ( ) , http . StatusInternalServerError )
return
}
}
func searchHandler ( w http . ResponseWriter , r * http . Request ) {
vars := mux . Vars ( r )
page := vars [ "page" ]
2017-05-03 19:45:18 +02:00
// db params url
2017-05-04 00:20:50 +02:00
maxPerPage , errConv := strconv . Atoi ( r . URL . Query ( ) . Get ( "max" ) )
if errConv != nil {
2017-05-03 19:45:18 +02:00
maxPerPage = 50 // default Value maxPerPage
2017-05-04 00:20:50 +02:00
}
2017-05-02 12:39:53 +02:00
pagenum , _ := strconv . Atoi ( html . EscapeString ( page ) )
2017-05-04 05:28:42 +02:00
searchQuery := r . URL . Query ( ) . Get ( "q" )
2017-05-03 06:59:27 +02:00
cat := r . URL . Query ( ) . Get ( "c" )
2017-05-04 14:01:07 +02:00
stat := r . URL . Query ( ) . Get ( "s" )
2017-05-05 06:24:58 +02:00
sort := r . URL . Query ( ) . Get ( "sort" )
order := r . URL . Query ( ) . Get ( "order" )
2017-05-04 08:53:21 +02:00
catsSplit := strings . Split ( cat , "_" )
2017-05-04 07:01:29 +02:00
// need this to prevent out of index panics
var searchCatId , searchSubCatId string
if len ( catsSplit ) == 2 {
2017-05-04 08:53:21 +02:00
2017-05-04 07:01:29 +02:00
searchCatId = html . EscapeString ( catsSplit [ 0 ] )
searchSubCatId = html . EscapeString ( catsSplit [ 1 ] )
}
2017-05-05 06:24:58 +02:00
if sort == "" {
sort = "torrent_id"
}
if order == "" {
order = "desc"
}
order_by := sort + " " + order
2017-05-03 19:45:18 +02:00
nbTorrents := 0
2017-05-04 05:28:42 +02:00
b := [ ] TorrentsJson { }
2017-05-04 00:54:07 +02:00
2017-05-05 06:24:58 +02:00
parameters := createWhereParams ( "torrent_name LIKE ? AND status_id LIKE ? AND category_id LIKE ? AND sub_category_id LIKE ?" ,
"%" + searchQuery + "%" , stat + "%" , searchCatId + "%" , searchSubCatId + "%" )
torrents := getTorrentsOrderBy ( & parameters , order_by , maxPerPage , maxPerPage * ( pagenum - 1 ) )
2017-05-04 05:38:11 +02:00
2017-05-04 05:28:42 +02:00
for i , _ := range torrents {
2017-05-03 19:45:18 +02:00
nbTorrents ++
2017-05-04 05:28:42 +02:00
res := torrents [ i ] . toJson ( )
b = append ( b , res )
2017-05-02 12:39:53 +02:00
}
2017-05-05 06:24:58 +02:00
htv := HomeTemplateVariables { b , getAllCategories ( false ) , searchQuery , stat , cat , sort , order , maxPerPage , nbTorrents }
2017-05-04 05:28:42 +02:00
err := templates . ExecuteTemplate ( w , "index.html" , htv )
2017-05-02 12:39:53 +02:00
if err != nil {
http . Error ( w , err . Error ( ) , http . StatusInternalServerError )
}
}
2017-05-03 09:33:39 +02:00
func safe ( s string ) template . URL {
return template . URL ( s )
}
2017-05-02 12:39:53 +02:00
2017-05-04 01:15:20 +02:00
func faqHandler ( w http . ResponseWriter , r * http . Request ) {
2017-05-04 04:29:22 +02:00
err := templates . ExecuteTemplate ( w , "FAQ.html" , "" )
2017-05-04 01:15:20 +02:00
if err != nil {
http . Error ( w , err . Error ( ) , http . StatusInternalServerError )
}
}
2017-05-04 21:48:40 +02:00
func rssHandler ( w http . ResponseWriter , r * http . Request ) {
//vars := mux.Vars(r)
//category := vars["c"]
// db params url
//maxPerPage := 50 // default Value maxPerPage
torrents := getFeeds ( )
created := time . Now ( ) . String ( )
2017-05-05 06:24:58 +02:00
if len ( torrents ) > 0 {
2017-05-04 21:48:40 +02:00
created = torrents [ 0 ] . Timestamp
}
created_as_time , err := time . Parse ( "2006-01-02 15:04:05" , created )
if err == nil {
2017-05-05 06:24:58 +02:00
2017-05-04 21:48:40 +02:00
}
feed := & feeds . Feed {
2017-05-05 06:24:58 +02:00
Title : "Nyaa Pantsu" ,
Link : & feeds . Link { Href : "https://nyaa.pantsu.cat/" } ,
Created : created_as_time ,
2017-05-04 21:48:40 +02:00
}
feed . Items = [ ] * feeds . Item { }
2017-05-05 06:24:58 +02:00
feed . Items = make ( [ ] * feeds . Item , len ( torrents ) )
2017-05-04 21:48:40 +02:00
for i , _ := range torrents {
timestamp_as_time , err := time . Parse ( "2006-01-02 15:04:05" , torrents [ i ] . Timestamp )
if err == nil {
2017-05-05 06:24:58 +02:00
feed . Items [ i ] = & feeds . Item {
2017-05-04 21:48:40 +02:00
// need a torrent view first
//Id: URL + torrents[i].Hash,
2017-05-05 06:24:58 +02:00
Title : torrents [ i ] . Name ,
Link : & feeds . Link { Href : string ( torrents [ i ] . Magnet ) } ,
2017-05-04 21:48:40 +02:00
Description : "" ,
2017-05-05 06:24:58 +02:00
Created : timestamp_as_time ,
Updated : timestamp_as_time ,
2017-05-04 21:48:40 +02:00
}
}
}
rss , err := feed . ToRss ( )
if err == nil {
2017-05-05 06:24:58 +02:00
w . Write ( [ ] byte ( rss ) )
2017-05-04 21:48:40 +02:00
} else {
http . Error ( w , err . Error ( ) , http . StatusInternalServerError )
}
}
2017-05-05 02:54:37 +02:00
func viewHandler ( w http . ResponseWriter , r * http . Request ) {
vars := mux . Vars ( r )
id := vars [ "id" ]
b := [ ] TorrentsJson { }
torrent , err := getTorrentById ( id )
res := torrent . toJson ( )
b = append ( b , res )
2017-05-05 03:34:49 +02:00
htv := HomeTemplateVariables { b , getAllCategories ( false ) , "" , "" , "_" , "" , "" , 1 , 1 }
2017-05-05 02:54:37 +02:00
err = templates . ExecuteTemplate ( w , "view.html" , htv )
if err != nil {
http . Error ( w , err . Error ( ) , http . StatusInternalServerError )
}
}
2017-05-04 21:48:40 +02:00
2017-05-02 12:39:53 +02:00
func rootHandler ( w http . ResponseWriter , r * http . Request ) {
vars := mux . Vars ( r )
page := vars [ "page" ]
2017-05-03 19:45:18 +02:00
2017-05-04 00:20:50 +02:00
// db params url
maxPerPage , errConv := strconv . Atoi ( r . URL . Query ( ) . Get ( "max" ) )
if errConv != nil {
2017-05-03 19:45:18 +02:00
maxPerPage = 50 // default Value maxPerPage
2017-05-04 00:20:50 +02:00
}
2017-05-03 19:45:18 +02:00
nbTorrents := 0
2017-05-02 12:39:53 +02:00
pagenum , _ := strconv . Atoi ( html . EscapeString ( page ) )
2017-05-04 05:28:42 +02:00
b := [ ] TorrentsJson { }
torrents := getAllTorrents ( maxPerPage , maxPerPage * ( pagenum - 1 ) )
for i , _ := range torrents {
2017-05-03 19:45:18 +02:00
nbTorrents ++
2017-05-04 05:28:42 +02:00
res := torrents [ i ] . toJson ( )
b = append ( b , res )
2017-05-02 12:39:53 +02:00
}
2017-05-04 05:28:42 +02:00
2017-05-05 06:30:30 +02:00
htv := HomeTemplateVariables { b , getAllCategories ( false ) , "" , "" , "_" , "torrent_id" , "desc" , maxPerPage , nbTorrents }
2017-05-04 05:28:42 +02:00
err := templates . ExecuteTemplate ( w , "index.html" , htv )
2017-05-02 12:39:53 +02:00
if err != nil {
http . Error ( w , err . Error ( ) , http . StatusInternalServerError )
}
}
func main ( ) {
2017-05-04 05:28:42 +02:00
db = getDBHandle ( )
2017-05-02 12:39:53 +02:00
router := mux . NewRouter ( )
2017-05-03 19:45:18 +02:00
cssHandler := http . FileServer ( http . Dir ( "./css/" ) )
jsHandler := http . FileServer ( http . Dir ( "./js/" ) )
2017-05-04 00:20:50 +02:00
http . Handle ( "/css/" , http . StripPrefix ( "/css/" , cssHandler ) )
http . Handle ( "/js/" , http . StripPrefix ( "/js/" , jsHandler ) )
2017-05-03 19:45:18 +02:00
2017-05-02 12:39:53 +02:00
// Routes,
router . HandleFunc ( "/" , rootHandler )
router . HandleFunc ( "/page/{page}" , rootHandler )
router . HandleFunc ( "/search" , searchHandler )
2017-05-02 13:07:04 +02:00
router . HandleFunc ( "/search/{page}" , searchHandler )
2017-05-02 12:39:53 +02:00
router . HandleFunc ( "/api/{page}" , apiHandler ) . Methods ( "GET" )
2017-05-05 02:54:37 +02:00
router . HandleFunc ( "/api/view/{id}" , apiViewHandler ) . Methods ( "GET" )
2017-05-04 01:15:20 +02:00
router . HandleFunc ( "/faq" , faqHandler )
2017-05-04 21:48:40 +02:00
router . HandleFunc ( "/feed.xml" , rssHandler )
2017-05-05 02:54:37 +02:00
router . HandleFunc ( "/view/{id}" , viewHandler )
2017-05-03 19:45:18 +02:00
http . Handle ( "/" , router )
2017-05-02 12:39:53 +02:00
// Set up server,
srv := & http . Server {
Addr : "localhost:9999" ,
WriteTimeout : 15 * time . Second ,
ReadTimeout : 15 * time . Second ,
}
err := srv . ListenAndServe ( )
checkErr ( err )
}