From c80e1dd5091c9a2c814954f689bbee68bc84c997 Mon Sep 17 00:00:00 2001
From: Xe Iaso <me@xeiaso.net>
Date: Mon, 14 Aug 2023 18:15:20 -0400
Subject: cmd/sanguisuga: add crc32 checking flag

Signed-off-by: Xe Iaso <me@xeiaso.net>
---
 cmd/sanguisuga/main.go | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

(limited to 'cmd')

diff --git a/cmd/sanguisuga/main.go b/cmd/sanguisuga/main.go
index 13d8690..8029e56 100644
--- a/cmd/sanguisuga/main.go
+++ b/cmd/sanguisuga/main.go
@@ -33,6 +33,8 @@ var (
 	tysonConfig  = flag.String("tyson-config", "./config.ts", "path to configuration secrets (TySON)")
 	externalSeed = flag.Bool("external-seed", false, "try to external seed?")
 
+	crcCheckCLI = flag.Bool("crc-check", false, "if true, check args[0] against hash args[1]")
+
 	annRegex = regexp.MustCompile(`^New Torrent Announcement: <([^>]*)>\s+Name:'(.*)' uploaded by '.*' ?(freeleech)?\s+-\s+https://\w+.\w+.\w+./\w+./([0-9]+)$`)
 
 	snatches = expvar.NewInt("gauge_sanguisuga_snatches")
@@ -71,6 +73,22 @@ func main() {
 	internal.HandleStartup()
 	hostinfo.SetApp("within.website/x/cmd/sanguisuga")
 
+	if *crcCheckCLI {
+		if flag.NArg() != 2 {
+			log.Fatalf("usage: %s <filename> <hash>", os.Args[0])
+		}
+
+		fname := flag.Arg(0)
+		hash := flag.Arg(1)
+
+		ok, err := crcCheck(fname, hash)
+		if err != nil {
+			log.Fatal(err)
+		}
+		log.Printf("hash status: %v", ok)
+		return
+	}
+
 	var c Config
 	if err := tyson.Unmarshal(*tysonConfig, &c); err != nil {
 		log.Fatalf("can't unmarshal config: %v", err)
@@ -292,8 +310,6 @@ func (s *Sanguisuga) HandleIRCMessage(ev *irc.Event) {
 			return
 		}
 
-		lg.Info("found anime, starting external seed hack")
-
 		go s.ExternalSeedAnime(ta, lg)
 	case "TV :: Episodes HD":
 		ti, err := parsetorrentname.Parse(ta.Name)
-- 
cgit v1.2.3