idanoo/autobrr
1
0
Fork 0
mirror of https://github.com/idanoo/autobrr synced 2025-07-23 08:49:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(tools): Add a simple MockIndexer helper for IRC announcing and indexing (#555)

Browse source 
* Add a simple SelfIndexer for IRC announcing and indexing

* Rename to MockIndexer

* fix: close file after reading
This commit is contained in:
Viktor Elofsson 2022-12-03 16:55:36 +01:00 • committed by GitHub
parent ef088c27ad
commit 0f2ce26ba2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 242 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

59
test/mockindexer/main.go Normal file
View file

@ -0,0 +1,59 @@
package main
import (
"bufio"
"io"
"log"
"net/http"
"os"
"github.com/autobrr/autobrr/test/mockindexer/irc"
"github.com/go-chi/chi/v5"
"github.com/go-chi/chi/v5/middleware"
)
func main() {
s, err := irc.NewServer(&irc.ServerOptions{
BotName: "_AnnounceBot_",
Channel: "#announces",
})
if err != nil {
log.Fatalf("Err: %v", err)
}
go s.Run()
log.Print("autobrr MockIndexer running")
r := chi.NewRouter()
r.Use(middleware.Logger)
r.Get("/", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("<html><form method=\"POST\" action=\"/send\">Send an announce line to the channel<br><input style=\"width: 100%; margin-top: 5px; margin-bottom: 5px;\" name=\"line\" type=\"text\"><br><button type=\"submit\">Send to channel</button></form></html>"))
})
r.Get("/file/{fileId}/{fileName}", func(w http.ResponseWriter, r *http.Request) {
f, err := os.Open("files/" + chi.URLParam(r, "fileId") + ".torrent")
if err != nil {
log.Fatalf("Err: %v", err)
}
defer f.Close()
w.Header().Set("Content-Disposition", "attachment; filename="+chi.URLParam(r, "fileName"))
w.Header().Set("Content-Type", "application/x-bittorrent")
io.Copy(w, bufio.NewReader(f))
})
r.Post("/send", func(w http.ResponseWriter, r *http.Request) {
r.ParseForm()
line := r.Form.Get("line")
s.SendAll(line)
http.Redirect(w, r, "/", 302)
})
http.ListenAndServe("localhost:3999", r)
}