Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Cette révision appartient à :
ayame-git 2017-05-03 10:15:09 +03:00
révision 345ead44ba

43
main.go
Voir le fichier

@ -10,6 +10,7 @@ import (
"log"
"net/http"
"net/url"
"os"
"strconv"
"strings"
"time"
@ -18,7 +19,6 @@ import (
var dbHandle *sql.DB
var templates = template.Must(template.ParseFiles("index.html"))
var debugLogger *log.Logger
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"
type Record struct {
Category string `json: "category"`
@ -28,10 +28,11 @@ type Record struct {
}
type Records struct {
Id string `json: "id"`
Name string `json: "name"`
Hash string `json: "hash"`
Magnet template.URL `json: "magnet"`
Id string `json: "id"`
Name string `json: "name"`
Status int `json: "status"`
Hash string `json: "hash"`
Magnet string `json: "magnet"`
}
func getDBHandle() *sql.DB {
@ -43,6 +44,7 @@ func getDBHandle() *sql.DB {
func checkErr(err error) {
if err != nil {
debugLogger.Println(" " + err.Error())
os.Exit(1)
}
}
@ -55,14 +57,15 @@ func apiHandler(w http.ResponseWriter, r *http.Request) {
rows, err := dbHandle.Query("select torrent_id, torrent_name, status_id, torrent_hash from torrents ORDER BY torrent_id DESC LIMIT 50 offset ?", 50*pagenum-1)
for rows.Next() {
var id, name, hash, magnet string
rows.Scan(&id, &name, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + trackers
var status int
rows.Scan(&id, &name, &status, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + "&tr=udp://tracker.openbittorrent.com"
res := Records{
Id: id,
Name: name,
Status: status,
Hash: hash,
Magnet: safe(magnet)}
Magnet: magnet}
b.Records = append(b.Records, res)
@ -86,14 +89,15 @@ func singleapiHandler(w http.ResponseWriter, r *http.Request) {
rows, err := dbHandle.Query("select torrent_id, torrent_name, status_id, torrent_hash from torrents where torrent_id = ? ORDER BY torrent_id DESC", html.EscapeString(id))
for rows.Next() {
var id, name, hash, magnet string
rows.Scan(&id, &name, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + trackers
var status int
rows.Scan(&id, &name, &status, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + "&tr=udp://tracker.openbittorrent.com"
res := Records{
Id: id,
Name: name,
Status: status,
Hash: hash,
Magnet: safe(magnet)}
Magnet: magnet}
b.Records = append(b.Records, res)
@ -125,14 +129,15 @@ func searchHandler(w http.ResponseWriter, r *http.Request) {
"%"+html.EscapeString(param1)+"%", html.EscapeString(param2)+"%", html.EscapeString(param3)+"%", 50*pagenum-1)
for rows.Next() {
var id, name, hash, magnet string
rows.Scan(&id, &name, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + trackers
var status int
rows.Scan(&id, &name, &status, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + "&tr=udp://tracker.openbittorrent.com"
res := Records{
Id: id,
Name: name,
Status: status,
Hash: hash,
Magnet: safe(magnet)}
Magnet: magnet}
b.Records = append(b.Records, res)
@ -144,9 +149,6 @@ func searchHandler(w http.ResponseWriter, r *http.Request) {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
}
func safe(s string) template.URL {
return template.URL(s)
}
func rootHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
@ -156,14 +158,15 @@ func rootHandler(w http.ResponseWriter, r *http.Request) {
rows, err := dbHandle.Query("select torrent_id, torrent_name, status_id, torrent_hash from torrents ORDER BY torrent_id DESC LIMIT 50 offset ?", 50*pagenum-1)
for rows.Next() {
var id, name, hash, magnet string
rows.Scan(&id, &name, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + trackers
var status int
rows.Scan(&id, &name, &status, &hash)
magnet = "magnet:?xt=urn:btih:" + hash + "&dn=" + url.QueryEscape(name) + "&tr=udp://tracker.openbittorrent.com"
res := Records{
Id: id,
Name: name,
Status: status,
Hash: hash,
Magnet: safe(magnet)}
Magnet: magnet}
b.Records = append(b.Records, res)