From a82e8bef03d595a07f64d86768f7de3098302772 Mon Sep 17 00:00:00 2001
From: Urban Guacamole <urban-guacamole@protonmail.com>
Date: Wed, 2 Sep 2020 16:20:59 +0200
Subject: [PATCH] Add support for quotes and exclusion for search

You can now use "" and - to require exact match or exclude some terms.
---
 api/main.go | 45 +++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 41 insertions(+), 4 deletions(-)

diff --git a/api/main.go b/api/main.go
index bec47b2..759f9cd 100644
--- a/api/main.go
+++ b/api/main.go
@@ -7,8 +7,9 @@ import (
 	"log"
 	"net/http"
 	"strings"
+	"sync"
 
-	_ "github.com/lib/pq"
+	"github.com/lib/pq"
 )
 
 type Results []Result
@@ -21,6 +22,12 @@ type Result struct {
 	L    int    `json:"l"`
 }
 
+type ingestRequest struct {
+	Ih     string
+	Name   string
+	Length int
+}
+
 func initDb() *sql.DB {
 	connStr := "user=nextgen dbname=nextgen host=/var/run/postgresql"
 	db, err := sql.Open("postgres", connStr)
@@ -41,6 +48,8 @@ func initDb() *sql.DB {
 func main() {
 	db := initDb()
 
+	var alreadyIngested sync.Map
+
 	http.HandleFunc("/api/telemetry", func(w http.ResponseWriter, r *http.Request) {
 		body, err := ioutil.ReadAll(r.Body)
 		if err != nil {
@@ -54,6 +63,30 @@ func main() {
 		w.WriteHeader(http.StatusOK)
 	})
 
+	// INSECURE! BE SURE TO WALL OFF THIS PROPERLY IN PROD
+	http.HandleFunc("/api/ingest", func(w http.ResponseWriter, r *http.Request) {
+		decoder := json.NewDecoder(r.Body)
+		var req ingestRequest
+		err := decoder.Decode(&req)
+		if err != nil {
+			w.WriteHeader(http.StatusBadRequest)
+			log.Println(err)
+			return
+		}
+
+		_, cached := alreadyIngested.Load(req.Ih)
+		if !cached {
+			_, err = db.Exec("INSERT INTO torrent (infohash, name, length) VALUES ($1, $2, $3)", req.Ih, req.Name, req.Length)
+			if err, ok := err.(*pq.Error); ok { //dark magic
+				if err.Code != "23505" {
+					log.Println(err)
+				}
+			}
+			alreadyIngested.Store(req.Ih, true)
+		}
+		w.WriteHeader(http.StatusOK)
+	})
+
 	http.HandleFunc("/api/search", func(w http.ResponseWriter, r *http.Request) {
 		q := r.URL.Query().Get("q")
 		if len(q) == 0 {
@@ -61,10 +94,14 @@ func main() {
 			w.WriteHeader(http.StatusBadRequest)
 			return
 		}
-		if strings.Contains(q, "preteen") {
-			w.WriteHeader(500)
+		pedoWords := []string{"pedo", "preteen", "14yo", "13yo", "12yo", "15yo", "16yo", "17yo", "11y", "underage", "11yo", "10yo", "10y", "9yo", "mafiasex", "pedofilia", "violacion"}
+		for _, word := range pedoWords {
+			if strings.Contains(q, word) {
+				w.WriteHeader(500)
+				return
+			}
 		}
-		rows, err := db.Query("select infohash, name, length, s, l from search where vect @@ plainto_tsquery($1) and copyrighted = 'f' limit 150", q)
+		rows, err := db.Query("select infohash, name, length, s, l from search where vect @@ websearch_to_tsquery($1) and copyrighted = 'f' limit 150", q)
 		if err != nil {
 			log.Print(err)
 			w.WriteHeader(http.StatusInternalServerError)