3 files changed, 51 insertions(+), 1 deletions(-)

A bin/rmrepo.tcl
M main.tcl
M src/jetstream.tcl

@@ 0,0 1,46 @@
+#!/bin/env tclsh
+
+if {$argc < 2} { return -code error "Usage: bin/rmrepo.tcl DB ?-pds PDS? ?-feed FEED? REPO ?REPO ...?\n-feed may be passed repeatedly." }
+
+cd [file dirname [file dirname [dict get [info frame [info frame]] file]]]
+
+package require sqlite3
+package require json
+package require http
+package require tls
+::http::register https 443 ::tls::socket
+
+set repos [lassign $argv db]
+sqlite3 db $db
+
+set pds "bsky.social"
+if {[lindex $repos 0] == "-pds"} {
+  set repos [lassign $repos _ pds]
+}
+while {[lindex $repos 0] == "-feed"} {
+  set repos [lassign $repos _ feed]
+  lappend feeds $feed
+}
+
+if ![info exists feeds] {
+  set feeds [db eval {SELECT name FROM sqlite_master WHERE type = 'table' AND sql LIKE '%uri TEXT%'}]
+}
+
+foreach repo $repos {
+  switch -regexp -- $repo {
+    {^did:plc:.+} {}
+    default {
+      set endpoint "https://$pds/xrpc/com.atproto.repo.describeRepo"
+
+      set http_state [http::geturl "$endpoint?repo=$repo"]
+      if {[::http::ncode $http_state] != 200} { return -code error "couldn't resolve handle $repo!\n[::http::data $http_state]" }
+      set repo [dict get [::json::json2dict [::http::data $http_state]] did]
+    }
+  }
+
+  db eval {INSERT OR IGNORE INTO blocked_repos VALUES (:repo);}
+  set repo_pat "%${repo}%"
+  foreach table $feeds {
+    db eval "DELETE FROM $table WHERE uri LIKE :repo_pat;"
+  }
+}

@@ 26,7 26,8 @@ ${log}::info "tclfeed v0.0.1"
 ### DB STUFF ###
 sqlite3 db [::ini::value $config database path]
 
-foreach algo $::algos::list {
+db eval {CREATE TABLE IF NOT EXISTS blocked_repos (repo TEXT NOT NULL PRIMARY KEY);}
+foreach algo [concat $::algos::list] {
   db eval {SELECT count(*) as n FROM sqlite_master WHERE name=$algo;} values {
     if ![set values(n)] {
       ${log}::info "Creating table \"$algo\"!"

@@ 22,6 22,9 @@ namespace eval ::jetstream {
           set data [json::json2dict $data]
           if {[dict get $data kind] != "commit"} return
 
+          set did [dict get $data did]
+          if [db eval {SELECT count(*) FROM blocked_repos WHERE repo = :did}] return
+
           set uri at://[dict get $data did]/[dict get $data commit collection]/[dict get $data commit rkey]
 
           switch -- [dict get $data commit operation] {