+1

.gitignore

reviewed

··· 1 1 + bsky-repost-likes

+70

go.mod

reviewed

··· 1 1 + module bsky-repost-likes 2 2 + 3 3 + go 1.23.9 4 4 + 5 5 + require ( 6 6 + github.com/bluesky-social/indigo v0.0.0-20250606055443-008e4ed915ad 7 7 + github.com/bluesky-social/jetstream v0.0.0-20250414024304-d17bd81a945e 8 8 + github.com/gorilla/mux v1.8.1 9 9 + github.com/gorilla/websocket v1.5.3 10 10 + ) 11 11 + 12 12 + require ( 13 13 + github.com/beorn7/perks v1.0.1 // indirect 14 14 + github.com/carlmjohnson/versioninfo v0.22.5 // indirect 15 15 + github.com/cespare/xxhash/v2 v2.3.0 // indirect 16 16 + github.com/felixge/httpsnoop v1.0.4 // indirect 17 17 + github.com/go-logr/logr v1.4.1 // indirect 18 18 + github.com/go-logr/stdr v1.2.2 // indirect 19 19 + github.com/goccy/go-json v0.10.2 // indirect 20 20 + github.com/gogo/protobuf v1.3.2 // indirect 21 21 + github.com/google/uuid v1.6.0 // indirect 22 22 + github.com/hashicorp/go-cleanhttp v0.5.2 // indirect 23 23 + github.com/hashicorp/go-retryablehttp v0.7.5 // indirect 24 24 + github.com/hashicorp/golang-lru v1.0.2 // indirect 25 25 + github.com/ipfs/bbloom v0.0.4 // indirect 26 26 + github.com/ipfs/go-block-format v0.2.0 // indirect 27 27 + github.com/ipfs/go-cid v0.4.1 // indirect 28 28 + github.com/ipfs/go-datastore v0.6.0 // indirect 29 29 + github.com/ipfs/go-ipfs-blockstore v1.3.1 // indirect 30 30 + github.com/ipfs/go-ipfs-ds-help v1.1.1 // indirect 31 31 + github.com/ipfs/go-ipfs-util v0.0.3 // indirect 32 32 + github.com/ipfs/go-ipld-cbor v0.1.0 // indirect 33 33 + github.com/ipfs/go-ipld-format v0.6.0 // indirect 34 34 + github.com/ipfs/go-log v1.0.5 // indirect 35 35 + github.com/ipfs/go-log/v2 v2.5.1 // indirect 36 36 + github.com/ipfs/go-metrics-interface v0.0.1 // indirect 37 37 + github.com/jbenet/goprocess v0.1.4 // indirect 38 38 + github.com/klauspost/compress v1.17.9 // indirect 39 39 + github.com/klauspost/cpuid/v2 v2.2.7 // indirect 40 40 + github.com/mattn/go-isatty v0.0.20 // indirect 41 41 + github.com/minio/sha256-simd v1.0.1 // indirect 42 42 + github.com/mr-tron/base58 v1.2.0 // indirect 43 43 + github.com/multiformats/go-base32 v0.1.0 // indirect 44 44 + github.com/multiformats/go-base36 v0.2.0 // indirect 45 45 + github.com/multiformats/go-multibase v0.2.0 // indirect 46 46 + github.com/multiformats/go-multihash v0.2.3 // indirect 47 47 + github.com/multiformats/go-varint v0.0.7 // indirect 48 48 + github.com/opentracing/opentracing-go v1.2.0 // indirect 49 49 + github.com/polydawn/refmt v0.89.1-0.20221221234430-40501e09de1f // indirect 50 50 + github.com/prometheus/client_golang v1.19.1 // indirect 51 51 + github.com/prometheus/client_model v0.6.1 // indirect 52 52 + github.com/prometheus/common v0.54.0 // indirect 53 53 + github.com/prometheus/procfs v0.15.1 // indirect 54 54 + github.com/spaolacci/murmur3 v1.1.0 // indirect 55 55 + github.com/whyrusleeping/cbor-gen v0.2.1-0.20241030202151-b7a6831be65e // indirect 56 56 + go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.46.1 // indirect 57 57 + go.opentelemetry.io/otel v1.21.0 // indirect 58 58 + go.opentelemetry.io/otel/metric v1.21.0 // indirect 59 59 + go.opentelemetry.io/otel/trace v1.21.0 // indirect 60 60 + go.uber.org/atomic v1.11.0 // indirect 61 61 + go.uber.org/multierr v1.11.0 // indirect 62 62 + go.uber.org/zap v1.26.0 // indirect 63 63 + golang.org/x/crypto v0.22.0 // indirect 64 64 + golang.org/x/sys v0.22.0 // indirect 65 65 + golang.org/x/xerrors v0.0.0-20231012003039-104605ab7028 // indirect 66 66 + google.golang.org/protobuf v1.34.2 // indirect 67 67 + lukechampine.com/blake3 v1.2.1 // indirect 68 68 + ) 69 69 + 70 70 + replace github.com/bluesky-social/jetstream => github.com/caseyho/jetstream v0.0.0-20250310034359-bee7b7fc4d0f

+217

go.sum

reviewed

··· 1 1 + github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= 2 2 + github.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA= 3 3 + github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM= 4 4 + github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw= 5 5 + github.com/bluesky-social/indigo v0.0.0-20250606055443-008e4ed915ad h1:VQnna39DdE6i2s0rm3ZG1GfucQOWJdYuyACzwxNQ+T4= 6 6 + github.com/bluesky-social/indigo v0.0.0-20250606055443-008e4ed915ad/go.mod h1:ovyxp8AMO1Hoe838vMJUbqHTZaAR8ABM3g3TXu+A5Ng= 7 7 + github.com/bluesky-social/jetstream v0.0.0-20250414024304-d17bd81a945e h1:P/O6TDHs53gwgV845uDHI+Nri889ixksRrh4bCkCdxo= 8 8 + github.com/bluesky-social/jetstream v0.0.0-20250414024304-d17bd81a945e/go.mod h1:WiYEeyJSdUwqoaZ71KJSpTblemUCpwJfh5oVXplK6T4= 9 9 + github.com/carlmjohnson/versioninfo v0.22.5 h1:O00sjOLUAFxYQjlN/bzYTuZiS0y6fWDQjMRvwtKgwwc= 10 10 + github.com/carlmjohnson/versioninfo v0.22.5/go.mod h1:QT9mph3wcVfISUKd0i9sZfVrPviHuSF+cUtLjm2WSf8= 11 11 + github.com/caseyho/jetstream v0.0.0-20250310034359-bee7b7fc4d0f h1:Ed2fW2wWekh6HK2S7vaHnlSlSJTT7iNdxcgdPYFaBrg= 12 12 + github.com/caseyho/jetstream v0.0.0-20250310034359-bee7b7fc4d0f/go.mod h1:WiYEeyJSdUwqoaZ71KJSpTblemUCpwJfh5oVXplK6T4= 13 13 + github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs= 14 14 + github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= 15 15 + github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU= 16 16 + github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= 17 17 + github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= 18 18 + github.com/felixge/httpsnoop v1.0.4 h1:NFTV2Zj1bL4mc9sqWACXbQFVBBg2W3GPvqp8/ESS2Wg= 19 19 + github.com/felixge/httpsnoop v1.0.4/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U= 20 20 + github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= 21 21 + github.com/go-logr/logr v1.4.1 h1:pKouT5E8xu9zeFC39JXRDukb6JFQPXM5p5I91188VAQ= 22 22 + github.com/go-logr/logr v1.4.1/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY= 23 23 + github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag= 24 24 + github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE= 25 25 + github.com/go-yaml/yaml v2.1.0+incompatible/go.mod h1:w2MrLa16VYP0jy6N7M5kHaCkaLENm+P+Tv+MfurjSw0= 26 26 + github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU= 27 27 + github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= 28 28 + github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q= 29 29 + github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q= 30 30 + github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI= 31 31 + github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= 32 32 + github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= 33 33 + github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= 34 34 + github.com/gorilla/mux v1.8.1 h1:TuBL49tXwgrFYWhqrNgrUNEY92u81SPhu7sTdzQEiWY= 35 35 + github.com/gorilla/mux v1.8.1/go.mod h1:AKf9I4AEqPTmMytcMc0KkNouC66V3BtZ4qD5fmWSiMQ= 36 36 + github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg= 37 37 + github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= 38 38 + github.com/hashicorp/go-cleanhttp v0.5.2 h1:035FKYIWjmULyFRBKPs8TBQoi0x6d9G4xc9neXJWAZQ= 39 39 + github.com/hashicorp/go-cleanhttp v0.5.2/go.mod h1:kO/YDlP8L1346E6Sodw+PrpBSV4/SoxCXGY6BqNFT48= 40 40 + github.com/hashicorp/go-hclog v0.9.2/go.mod h1:5CU+agLiy3J7N7QjHK5d05KxGsuXiQLrjA0H7acj2lQ= 41 41 + github.com/hashicorp/go-retryablehttp v0.7.5 h1:bJj+Pj19UZMIweq/iie+1u5YCdGrnxCT9yvm0e+Nd5M= 42 42 + github.com/hashicorp/go-retryablehttp v0.7.5/go.mod h1:Jy/gPYAdjqffZ/yFGCFV2doI5wjtH1ewM9u8iYVjtX8= 43 43 + github.com/hashicorp/golang-lru v1.0.2 h1:dV3g9Z/unq5DpblPpw+Oqcv4dU/1omnb4Ok8iPY6p1c= 44 44 + github.com/hashicorp/golang-lru v1.0.2/go.mod h1:iADmTwqILo4mZ8BN3D2Q6+9jd8WM5uGBxy+E8yxSoD4= 45 45 + github.com/ipfs/bbloom v0.0.4 h1:Gi+8EGJ2y5qiD5FbsbpX/TMNcJw8gSqr7eyjHa4Fhvs= 46 46 + github.com/ipfs/bbloom v0.0.4/go.mod h1:cS9YprKXpoZ9lT0n/Mw/a6/aFV6DTjTLYHeA+gyqMG0= 47 47 + github.com/ipfs/go-block-format v0.2.0 h1:ZqrkxBA2ICbDRbK8KJs/u0O3dlp6gmAuuXUJNiW1Ycs= 48 48 + github.com/ipfs/go-block-format v0.2.0/go.mod h1:+jpL11nFx5A/SPpsoBn6Bzkra/zaArfSmsknbPMYgzM= 49 49 + github.com/ipfs/go-cid v0.4.1 h1:A/T3qGvxi4kpKWWcPC/PgbvDA2bjVLO7n4UeVwnbs/s= 50 50 + github.com/ipfs/go-cid v0.4.1/go.mod h1:uQHwDeX4c6CtyrFwdqyhpNcxVewur1M7l7fNU7LKwZk= 51 51 + github.com/ipfs/go-datastore v0.6.0 h1:JKyz+Gvz1QEZw0LsX1IBn+JFCJQH4SJVFtM4uWU0Myk= 52 52 + github.com/ipfs/go-datastore v0.6.0/go.mod h1:rt5M3nNbSO/8q1t4LNkLyUwRs8HupMeN/8O4Vn9YAT8= 53 53 + github.com/ipfs/go-ipfs-blockstore v1.3.1 h1:cEI9ci7V0sRNivqaOr0elDsamxXFxJMMMy7PTTDQNsQ= 54 54 + github.com/ipfs/go-ipfs-blockstore v1.3.1/go.mod h1:KgtZyc9fq+P2xJUiCAzbRdhhqJHvsw8u2Dlqy2MyRTE= 55 55 + github.com/ipfs/go-ipfs-ds-help v1.1.1 h1:B5UJOH52IbcfS56+Ul+sv8jnIV10lbjLF5eOO0C66Nw= 56 56 + github.com/ipfs/go-ipfs-ds-help v1.1.1/go.mod h1:75vrVCkSdSFidJscs8n4W+77AtTpCIAdDGAwjitJMIo= 57 57 + github.com/ipfs/go-ipfs-util v0.0.3 h1:2RFdGez6bu2ZlZdI+rWfIdbQb1KudQp3VGwPtdNCmE0= 58 58 + github.com/ipfs/go-ipfs-util v0.0.3/go.mod h1:LHzG1a0Ig4G+iZ26UUOMjHd+lfM84LZCrn17xAKWBvs= 59 59 + github.com/ipfs/go-ipld-cbor v0.1.0 h1:dx0nS0kILVivGhfWuB6dUpMa/LAwElHPw1yOGYopoYs= 60 60 + github.com/ipfs/go-ipld-cbor v0.1.0/go.mod h1:U2aYlmVrJr2wsUBU67K4KgepApSZddGRDWBYR0H4sCk= 61 61 + github.com/ipfs/go-ipld-format v0.6.0 h1:VEJlA2kQ3LqFSIm5Vu6eIlSxD/Ze90xtc4Meten1F5U= 62 62 + github.com/ipfs/go-ipld-format v0.6.0/go.mod h1:g4QVMTn3marU3qXchwjpKPKgJv+zF+OlaKMyhJ4LHPg= 63 63 + github.com/ipfs/go-log v1.0.5 h1:2dOuUCB1Z7uoczMWgAyDck5JLb72zHzrMnGnCNNbvY8= 64 64 + github.com/ipfs/go-log v1.0.5/go.mod h1:j0b8ZoR+7+R99LD9jZ6+AJsrzkPbSXbZfGakb5JPtIo= 65 65 + github.com/ipfs/go-log/v2 v2.1.3/go.mod h1:/8d0SH3Su5Ooc31QlL1WysJhvyOTDCjcCZ9Axpmri6g= 66 66 + github.com/ipfs/go-log/v2 v2.5.1 h1:1XdUzF7048prq4aBjDQQ4SL5RxftpRGdXhNRwKSAlcY= 67 67 + github.com/ipfs/go-log/v2 v2.5.1/go.mod h1:prSpmC1Gpllc9UYWxDiZDreBYw7zp4Iqp1kOLU9U5UI= 68 68 + github.com/ipfs/go-metrics-interface v0.0.1 h1:j+cpbjYvu4R8zbleSs36gvB7jR+wsL2fGD6n0jO4kdg= 69 69 + github.com/ipfs/go-metrics-interface v0.0.1/go.mod h1:6s6euYU4zowdslK0GKHmqaIZ3j/b/tL7HTWtJ4VPgWY= 70 70 + github.com/jbenet/go-cienv v0.1.0/go.mod h1:TqNnHUmJgXau0nCzC7kXWeotg3J9W34CUv5Djy1+FlA= 71 71 + github.com/jbenet/goprocess v0.1.4 h1:DRGOFReOMqqDNXwW70QkacFW0YN9QnwLV0Vqk+3oU0o= 72 72 + github.com/jbenet/goprocess v0.1.4/go.mod h1:5yspPrukOVuOLORacaBi858NqyClJPQxYZlqdZVfqY4= 73 73 + github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= 74 74 + github.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8= 75 75 + github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= 76 76 + github.com/klauspost/compress v1.17.9 h1:6KIumPrER1LHsvBVuDa0r5xaG0Es51mhhB9BQB2qeMA= 77 77 + github.com/klauspost/compress v1.17.9/go.mod h1:Di0epgTjJY877eYKx5yC51cX2A2Vl2ibi7bDH9ttBbw= 78 78 + github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM= 79 79 + github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws= 80 80 + github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= 81 81 + github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= 82 82 + github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= 83 83 + github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94= 84 84 + github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= 85 85 + github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= 86 86 + github.com/minio/sha256-simd v1.0.1 h1:6kaan5IFmwTNynnKKpDHe6FWHohJOHhCPchzK49dzMM= 87 87 + github.com/minio/sha256-simd v1.0.1/go.mod h1:Pz6AKMiUdngCLpeTL/RJY1M9rUuPMYujV5xJjtbRSN8= 88 88 + github.com/mr-tron/base58 v1.2.0 h1:T/HDJBh4ZCPbU39/+c3rRvE0uKBQlU27+QI8LJ4t64o= 89 89 + github.com/mr-tron/base58 v1.2.0/go.mod h1:BinMc/sQntlIE1frQmRFPUoPA1Zkr8VRgBdjWI2mNwc= 90 90 + github.com/multiformats/go-base32 v0.1.0 h1:pVx9xoSPqEIQG8o+UbAe7DNi51oej1NtK+aGkbLYxPE= 91 91 + github.com/multiformats/go-base32 v0.1.0/go.mod h1:Kj3tFY6zNr+ABYMqeUNeGvkIC/UYgtWibDcT0rExnbI= 92 92 + github.com/multiformats/go-base36 v0.2.0 h1:lFsAbNOGeKtuKozrtBsAkSVhv1p9D0/qedU9rQyccr0= 93 93 + github.com/multiformats/go-base36 v0.2.0/go.mod h1:qvnKE++v+2MWCfePClUEjE78Z7P2a1UV0xHgWc0hkp4= 94 94 + github.com/multiformats/go-multibase v0.2.0 h1:isdYCVLvksgWlMW9OZRYJEa9pZETFivncJHmHnnd87g= 95 95 + github.com/multiformats/go-multibase v0.2.0/go.mod h1:bFBZX4lKCA/2lyOFSAoKH5SS6oPyjtnzK/XTFDPkNuk= 96 96 + github.com/multiformats/go-multihash v0.2.3 h1:7Lyc8XfX/IY2jWb/gI7JP+o7JEq9hOa7BFvVU9RSh+U= 97 97 + github.com/multiformats/go-multihash v0.2.3/go.mod h1:dXgKXCXjBzdscBLk9JkjINiEsCKRVch90MdaGiKsvSM= 98 98 + github.com/multiformats/go-varint v0.0.7 h1:sWSGR+f/eu5ABZA2ZpYKBILXTTs9JWpdEM/nEGOHFS8= 99 99 + github.com/multiformats/go-varint v0.0.7/go.mod h1:r8PUYw/fD/SjBCiKOoDlGF6QawOELpZAu9eioSos/OU= 100 100 + github.com/opentracing/opentracing-go v1.2.0 h1:uEJPy/1a5RIPAJ0Ov+OIO8OxWu77jEv+1B0VhjKrZUs= 101 101 + github.com/opentracing/opentracing-go v1.2.0/go.mod h1:GxEUsuufX4nBwe+T+Wl9TAgYrxe9dPLANfrWvHYVTgc= 102 102 + github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= 103 103 + github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= 104 104 + github.com/polydawn/refmt v0.89.1-0.20221221234430-40501e09de1f h1:VXTQfuJj9vKR4TCkEuWIckKvdHFeJH/huIFJ9/cXOB0= 105 105 + github.com/polydawn/refmt v0.89.1-0.20221221234430-40501e09de1f/go.mod h1:/zvteZs/GwLtCgZ4BL6CBsk9IKIlexP43ObX9AxTqTw= 106 106 + github.com/prometheus/client_golang v1.19.1 h1:wZWJDwK+NameRJuPGDhlnFgx8e8HN3XHQeLaYJFJBOE= 107 107 + github.com/prometheus/client_golang v1.19.1/go.mod h1:mP78NwGzrVks5S2H6ab8+ZZGJLZUq1hoULYBAYBw1Ho= 108 108 + github.com/prometheus/client_model v0.6.1 h1:ZKSh/rekM+n3CeS952MLRAdFwIKqeY8b62p8ais2e9E= 109 109 + github.com/prometheus/client_model v0.6.1/go.mod h1:OrxVMOVHjw3lKMa8+x6HeMGkHMQyHDk9E3jmP2AmGiY= 110 110 + github.com/prometheus/common v0.54.0 h1:ZlZy0BgJhTwVZUn7dLOkwCZHUkrAqd3WYtcFCWnM1D8= 111 111 + github.com/prometheus/common v0.54.0/go.mod h1:/TQgMJP5CuVYveyT7n/0Ix8yLNNXy9yRSkhnLTHPDIQ= 112 112 + github.com/prometheus/procfs v0.15.1 h1:YagwOFzUgYfKKHX6Dr+sHT7km/hxC76UB0learggepc= 113 113 + github.com/prometheus/procfs v0.15.1/go.mod h1:fB45yRUv8NstnjriLhBQLuOUt+WW4BsoGhij/e3PBqk= 114 114 + github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= 115 115 + github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= 116 116 + github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc= 117 117 + github.com/smartystreets/assertions v1.2.0/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo= 118 118 + github.com/smartystreets/goconvey v1.7.2/go.mod h1:Vw0tHAZW6lzCRk3xgdin6fKYcG+G3Pg9vgXWeJpQFMM= 119 119 + github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI= 120 120 + github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA= 121 121 + github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= 122 122 + github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= 123 123 + github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= 124 124 + github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4= 125 125 + github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= 126 126 + github.com/urfave/cli v1.22.10/go.mod h1:Gos4lmkARVdJ6EkW0WaNv/tZAAMe9V7XWyB60NtXRu0= 127 127 + github.com/warpfork/go-wish v0.0.0-20220906213052-39a1cc7a02d0/go.mod h1:x6AKhvSSexNrVSrViXSHUEbICjmGXhtgABaHIySUSGw= 128 128 + github.com/whyrusleeping/cbor-gen v0.2.1-0.20241030202151-b7a6831be65e h1:28X54ciEwwUxyHn9yrZfl5ojgF4CBNLWX7LR0rvBkf4= 129 129 + github.com/whyrusleeping/cbor-gen v0.2.1-0.20241030202151-b7a6831be65e/go.mod h1:pM99HXyEbSQHcosHc0iW7YFmwnscr+t9Te4ibko05so= 130 130 + github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= 131 131 + github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= 132 132 + github.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k= 133 133 + go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.46.1 h1:aFJWCqJMNjENlcleuuOkGAPH82y0yULBScfXcIEdS24= 134 134 + go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.46.1/go.mod h1:sEGXWArGqc3tVa+ekntsN65DmVbVeW+7lTKTjZF3/Fo= 135 135 + go.opentelemetry.io/otel v1.21.0 h1:hzLeKBZEL7Okw2mGzZ0cc4k/A7Fta0uoPgaJCr8fsFc= 136 136 + go.opentelemetry.io/otel v1.21.0/go.mod h1:QZzNPQPm1zLX4gZK4cMi+71eaorMSGT3A4znnUvNNEo= 137 137 + go.opentelemetry.io/otel/metric v1.21.0 h1:tlYWfeo+Bocx5kLEloTjbcDwBuELRrIFxwdQ36PlJu4= 138 138 + go.opentelemetry.io/otel/metric v1.21.0/go.mod h1:o1p3CA8nNHW8j5yuQLdc1eeqEaPfzug24uvsyIEJRWM= 139 139 + go.opentelemetry.io/otel/trace v1.21.0 h1:WD9i5gzvoUPuXIXH24ZNBudiarZDKuekPqi/E8fpfLc= 140 140 + go.opentelemetry.io/otel/trace v1.21.0/go.mod h1:LGbsEB0f9LGjN+OZaQQ26sohbOmiMR+BaslueVtS/qQ= 141 141 + go.uber.org/atomic v1.6.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ= 142 142 + go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc= 143 143 + go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE= 144 144 + go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0= 145 145 + go.uber.org/goleak v1.1.11-0.20210813005559-691160354723/go.mod h1:cwTWslyiVhfpKIDGSZEM2HlOvcqm+tG4zioyIeLoqMQ= 146 146 + go.uber.org/multierr v1.5.0/go.mod h1:FeouvMocqHpRaaGuG9EjoKcStLC43Zu/fmqdUMPcKYU= 147 147 + go.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU= 148 148 + go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0= 149 149 + go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y= 150 150 + go.uber.org/tools v0.0.0-20190618225709-2cfd321de3ee/go.mod h1:vJERXedbb3MVM5f9Ejo0C68/HhF8uaILCdgjnY+goOA= 151 151 + go.uber.org/zap v1.16.0/go.mod h1:MA8QOfq0BHJwdXa996Y4dYkAqRKB8/1K1QMMZVaNZjQ= 152 152 + go.uber.org/zap v1.19.1/go.mod h1:j3DNczoxDZroyBnOT1L/Q79cfUMGZxlv/9dzN7SM1rI= 153 153 + go.uber.org/zap v1.26.0 h1:sI7k6L95XOKS281NhVKOFCUNIvv9e0w4BF8N3u+tCRo= 154 154 + go.uber.org/zap v1.26.0/go.mod h1:dtElttAiwGvoJ/vj4IwHBS/gXsEu/pZ50mUIRWuG0so= 155 155 + golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= 156 156 + golang.org/x/crypto v0.0.0-20190510104115-cbcb75029529/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= 157 157 + golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= 158 158 + golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= 159 159 + golang.org/x/crypto v0.22.0 h1:g1v0xeRhjcugydODzvb3mEM9SQ0HGp9s/nh3COQ/C30= 160 160 + golang.org/x/crypto v0.22.0/go.mod h1:vr6Su+7cTlO45qkww3VDJlzDn0ctJvRgYbC2NvXHt+M= 161 161 + golang.org/x/lint v0.0.0-20190930215403-16217165b5de/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= 162 162 + golang.org/x/mod v0.0.0-20190513183733-4bf6d317e70e/go.mod h1:mXi4GBBbnImb6dmsKGUJ2LatrhH/nqhxcFungHvyanc= 163 163 + golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= 164 164 + golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= 165 165 + golang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= 166 166 + golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= 167 167 + golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= 168 168 + golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= 169 169 + golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= 170 170 + golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= 171 171 + golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM= 172 172 + golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= 173 173 + golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= 174 174 + golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= 175 175 + golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= 176 176 + golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= 177 177 + golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= 178 178 + golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= 179 179 + golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= 180 180 + golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= 181 181 + golang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= 182 182 + golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= 183 183 + golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= 184 184 + golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= 185 185 + golang.org/x/sys v0.22.0 h1:RI27ohtqKCnwULzJLqkv897zojh5/DwS/ENaMzUOaWI= 186 186 + golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= 187 187 + golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= 188 188 + golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= 189 189 + golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= 190 190 + golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= 191 191 + golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= 192 192 + golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= 193 193 + golang.org/x/tools v0.0.0-20190621195816-6e04913cbbac/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc= 194 194 + golang.org/x/tools v0.0.0-20191029041327-9cc4af7d6b2c/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= 195 195 + golang.org/x/tools v0.0.0-20191029190741-b9c20aec41a5/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= 196 196 + golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= 197 197 + golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE= 198 198 + golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA= 199 199 + golang.org/x/tools v0.1.5/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk= 200 200 + golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= 201 201 + golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= 202 202 + golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= 203 203 + golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= 204 204 + golang.org/x/xerrors v0.0.0-20231012003039-104605ab7028 h1:+cNy6SZtPcJQH3LJVLOSmiC7MMxXNOb3PU/VUEz+EhU= 205 205 + golang.org/x/xerrors v0.0.0-20231012003039-104605ab7028/go.mod h1:NDW/Ps6MPRej6fsCIbMTohpP40sJ/P/vI1MoTEGwX90= 206 206 + google.golang.org/protobuf v1.34.2 h1:6xV6lTsCfpGD21XK49h7MhtcApnLqkfYgPcdHftf6hg= 207 207 + google.golang.org/protobuf v1.34.2/go.mod h1:qYOHts0dSfpeUzUFpOMr/WGzszTmLH+DiWniOlNbLDw= 208 208 + gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= 209 209 + gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= 210 210 + gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI= 211 211 + gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= 212 212 + gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= 213 213 + gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= 214 214 + gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= 215 215 + honnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg= 216 216 + lukechampine.com/blake3 v1.2.1 h1:YuqqRuaqsGV71BV/nm9xlI0MKUv4QC54jQnBChWbGnI= 217 217 + lukechampine.com/blake3 v1.2.1/go.mod h1:0OFRp7fBtAylGVCO40o87sbupkyIGgbpv1+M1k1LM6k=

+384

main.go

reviewed

··· 1 1 + package main 2 2 + 3 3 + import ( 4 4 + "context" 5 5 + "encoding/json" 6 6 + "fmt" 7 7 + "log" 8 8 + "log/slog" 9 9 + "net/http" 10 10 + "sync" 11 11 + "time" 12 12 + 13 13 + "github.com/bluesky-social/indigo/api/atproto" 14 14 + "github.com/bluesky-social/indigo/api/bsky" 15 15 + "github.com/bluesky-social/indigo/xrpc" 16 16 + "github.com/bluesky-social/jetstream/pkg/client" 17 17 + "github.com/bluesky-social/jetstream/pkg/client/schedulers/sequential" 18 18 + "github.com/bluesky-social/jetstream/pkg/models" 19 19 + "github.com/gorilla/mux" 20 20 + "github.com/gorilla/websocket" 21 21 + ) 22 22 + 23 23 + type Set[T comparable] map[T]struct{} 24 24 + 25 25 + // Data structures 26 26 + type SubscriberData struct { 27 27 + DID string 28 28 + Conn *websocket.Conn 29 29 + Follows Set[string] 30 30 + Reposts Set[string] 31 31 + } 32 32 + 33 33 + type NotificationMessage struct { 34 34 + Liked bool `json:"liked"` 35 35 + ByDid string `json:"did"` 36 36 + RepostURI string `json:"repost_uri"` 37 37 + } 38 38 + 39 39 + // Global state 40 40 + var ( 41 41 + subscribers = make(map[string]*SubscriberData) 42 42 + subscribersMux sync.RWMutex 43 43 + 44 44 + likeStream *client.Client 45 45 + subscriberStream *client.Client 46 46 + 47 47 + xrpcClient *xrpc.Client 48 48 + 49 49 + upgrader = websocket.Upgrader{ 50 50 + CheckOrigin: func(r *http.Request) bool { 51 51 + return true 52 52 + }, 53 53 + } 54 54 + 55 55 + logger *slog.Logger 56 56 + ) 57 57 + 58 58 + func main() { 59 59 + logger = slog.Default() 60 60 + 61 61 + xrpcClient = &xrpc.Client{ 62 62 + Client: &http.Client{ 63 63 + Timeout: 30 * time.Second, 64 64 + }, 65 65 + Host: "https://bsky.social", 66 66 + } 67 67 + 68 68 + if err := initializeJetstreams(); err != nil { 69 69 + log.Fatalf("cannot start jetstream: %s", err) 70 70 + } 71 71 + 72 72 + r := mux.NewRouter() 73 73 + r.HandleFunc("/subscribe/{did}", handleSubscribe).Methods("GET") 74 74 + 75 75 + log.Println("Server starting on :8080") 76 76 + if err := http.ListenAndServe(":8080", r); err != nil { 77 77 + log.Fatalf("error while serving: %s", err) 78 78 + } 79 79 + } 80 80 + 81 81 + func handleSubscribe(w http.ResponseWriter, r *http.Request) { 82 82 + vars := mux.Vars(r) 83 83 + did := vars["did"] 84 84 + 85 85 + conn, err := upgrader.Upgrade(w, r, nil) 86 86 + if err != nil { 87 87 + logger.Error("WebSocket upgrade failed", "error", err) 88 88 + return 89 89 + } 90 90 + defer conn.Close() 91 91 + 92 92 + logger.Info("New subscriber", "did", did) 93 93 + 94 94 + follows, err := fetchFollows(r.Context(), did) 95 95 + if err != nil { 96 96 + logger.Error("Error fetching follows", "did", did, "error", err) 97 97 + return 98 98 + } 99 99 + 100 100 + reposts, err := fetchReposts(r.Context(), did) 101 101 + if err != nil { 102 102 + logger.Error("Error fetching reposts", "did", did, "error", err) 103 103 + return 104 104 + } 105 105 + 106 106 + // Store subscriber data 107 107 + subscriber := &SubscriberData{ 108 108 + DID: did, 109 109 + Conn: conn, 110 110 + Follows: follows, 111 111 + Reposts: reposts, 112 112 + } 113 113 + 114 114 + subscribersMux.Lock() 115 115 + subscribers[did] = subscriber 116 116 + subscribersMux.Unlock() 117 117 + updateSubscriberStreamOpts() 118 118 + // delete subscriber after we are done 119 119 + defer func() { 120 120 + subscribersMux.Lock() 121 121 + delete(subscribers, did) 122 122 + subscribersMux.Unlock() 123 123 + updateSubscriberStreamOpts() 124 124 + }() 125 125 + 126 126 + for { 127 127 + _, _, err := conn.ReadMessage() 128 128 + if err != nil { 129 129 + logger.Info("WebSocket connection closed", "did", did, "error", err) 130 130 + break 131 131 + } 132 132 + } 133 133 + } 134 134 + 135 135 + func fetchReposts(ctx context.Context, did string) (Set[string], error) { 136 136 + all := make(Set[string]) 137 137 + cursor := "" 138 138 + 139 139 + for { 140 140 + out, err := atproto.RepoListRecords(ctx, &xrpc.Client{}, "app.bsky.feed.repost", cursor, 100, did, false) 141 141 + if err != nil { 142 142 + return nil, err 143 143 + } 144 144 + 145 145 + for _, record := range out.Records { 146 146 + all[record.Uri] = struct{}{} 147 147 + } 148 148 + 149 149 + if out.Cursor == nil || *out.Cursor == "" { 150 150 + break 151 151 + } 152 152 + cursor = *out.Cursor 153 153 + } 154 154 + 155 155 + return all, nil 156 156 + } 157 157 + 158 158 + func fetchFollows(ctx context.Context, did string) (Set[string], error) { 159 159 + all := make(Set[string]) 160 160 + cursor := "" 161 161 + 162 162 + for { 163 163 + out, err := bsky.GraphGetFollows(ctx, &xrpc.Client{}, did, cursor, 100) 164 164 + if err != nil { 165 165 + return nil, err 166 166 + } 167 167 + 168 168 + for _, record := range out.Follows { 169 169 + all[record.Did] = struct{}{} 170 170 + } 171 171 + 172 172 + if out.Cursor == nil || *out.Cursor == "" { 173 173 + break 174 174 + } 175 175 + cursor = *out.Cursor 176 176 + } 177 177 + 178 178 + return all, nil 179 179 + } 180 180 + 181 181 + func initializeJetstreams() error { 182 182 + if err := startLikeClient(); err != nil { 183 183 + return fmt.Errorf("like stream: %w", err) 184 184 + } 185 185 + if err := startSubscriberClient(); err != nil { 186 186 + return fmt.Errorf("subscriber stream: %w", err) 187 187 + } 188 188 + return nil 189 189 + } 190 190 + 191 191 + func getLikeStreamOpts() models.SubscriberOptionsUpdatePayload { 192 192 + return models.SubscriberOptionsUpdatePayload{ 193 193 + WantedCollections: []string{"app.bsky.feed.like"}, 194 194 + WantedDIDs: getFollowsDids(), 195 195 + } 196 196 + } 197 197 + 198 198 + func getSubscriberStreamOpts() models.SubscriberOptionsUpdatePayload { 199 199 + return models.SubscriberOptionsUpdatePayload{ 200 200 + WantedCollections: []string{"app.bsky.feed.repost", "app.bsky.graph.follow"}, 201 201 + WantedDIDs: getSubscriberDids(), 202 202 + } 203 203 + } 204 204 + 205 205 + func updateLikeStreamOpts() { 206 206 + err := likeStream.SendOptionsUpdate(getLikeStreamOpts()) 207 207 + if err != nil { 208 208 + // reinit like stream 209 209 + } 210 210 + } 211 211 + 212 212 + func updateSubscriberStreamOpts() { 213 213 + err := subscriberStream.SendOptionsUpdate(getSubscriberStreamOpts()) 214 214 + if err != nil { 215 215 + // reinit subscriber stream 216 216 + } 217 217 + } 218 218 + 219 219 + func startLikeClient() error { 220 220 + opts := getLikeStreamOpts() 221 221 + if len(opts.WantedDIDs) == 0 { 222 222 + return nil // No follows to track 223 223 + } 224 224 + 225 225 + handler := &likeHandler{} 226 226 + var err error 227 227 + likeStream, err = startJetstreamClient("like_tracker", opts, handler.HandleEvent) 228 228 + if err != nil { 229 229 + return err 230 230 + } 231 231 + 232 232 + return nil 233 233 + } 234 234 + 235 235 + func startSubscriberClient() error { 236 236 + opts := getSubscriberStreamOpts() 237 237 + if len(opts.WantedDIDs) == 0 { 238 238 + return nil // No subscribers to track 239 239 + } 240 240 + 241 241 + handler := &subscriberHandler{} 242 242 + var err error 243 243 + subscriberStream, err = startJetstreamClient("subscriber", opts, handler.HandleEvent) 244 244 + if err != nil { 245 245 + return err 246 246 + } 247 247 + 248 248 + return nil 249 249 + } 250 250 + 251 251 + func startJetstreamClient(name string, opts models.SubscriberOptionsUpdatePayload, handleEvent func(context.Context, *models.Event) error) (*client.Client, error) { 252 252 + ctx := context.Background() 253 253 + 254 254 + config := client.DefaultClientConfig() 255 255 + config.WebsocketURL = "wss://jetstream.atproto.tools/subscribe" 256 256 + config.Compress = true 257 257 + config.WantedCollections = opts.WantedCollections 258 258 + config.WantedDids = opts.WantedDIDs 259 259 + 260 260 + scheduler := sequential.NewScheduler(name, logger, handleEvent) 261 261 + 262 262 + c, err := client.NewClient(config, logger, scheduler) 263 263 + if err != nil { 264 264 + logger.Error("Failed to create client", "name", name, "error", err) 265 265 + return nil, err 266 266 + } 267 267 + 268 268 + cursor := time.Now().UnixMicro() 269 269 + 270 270 + logger.Info("Starting client", "name", name, "collections", opts.WantedCollections, "wanted_dids", len(opts.WantedDIDs)) 271 271 + if err := c.ConnectAndRead(ctx, &cursor); err != nil { 272 272 + logger.Error("Client failed", "name", name, "error", err) 273 273 + return nil, err 274 274 + } 275 275 + 276 276 + return c, nil 277 277 + } 278 278 + 279 279 + func getFollowsDids() []string { 280 280 + subscribersMux.RLock() 281 281 + defer subscribersMux.RUnlock() 282 282 + 283 283 + var dids []string 284 284 + for _, subscriber := range subscribers { 285 285 + for follow, _ := range subscriber.Follows { 286 286 + dids = append(dids, follow) 287 287 + } 288 288 + } 289 289 + 290 290 + return dids 291 291 + } 292 292 + 293 293 + func getSubscriberDids() []string { 294 294 + subscribersMux.RLock() 295 295 + defer subscribersMux.RUnlock() 296 296 + 297 297 + var dids []string 298 298 + for did := range subscribers { 299 299 + dids = append(dids, did) 300 300 + } 301 301 + 302 302 + return dids 303 303 + } 304 304 + 305 305 + type likeHandler struct{} 306 306 + 307 307 + func (h *likeHandler) HandleEvent(ctx context.Context, event *models.Event) error { 308 308 + var like bsky.FeedLike 309 309 + if err := json.Unmarshal(event.Commit.Record, &like); err != nil { 310 310 + logger.Error("Failed to unmarshal like", "error", err) 311 311 + return nil 312 312 + } 313 313 + 314 314 + subscribersMux.RLock() 315 315 + defer subscribersMux.RUnlock() 316 316 + 317 317 + for _, subscriber := range subscribers { 318 318 + for repostURI, _ := range subscriber.Reposts { 319 319 + // (un)liked a post the subscriber reposted 320 320 + if like.Subject.Uri == repostURI { 321 321 + notification := NotificationMessage{ 322 322 + Liked: event.Commit.Operation != models.CommitOperationDelete, 323 323 + ByDid: event.Did, 324 324 + RepostURI: repostURI, 325 325 + } 326 326 + 327 327 + if err := subscriber.Conn.WriteJSON(notification); err != nil { 328 328 + logger.Error("Failed to send notification", "subscriber", subscriber.DID, "error", err) 329 329 + } 330 330 + } 331 331 + } 332 332 + } 333 333 + 334 334 + return nil 335 335 + } 336 336 + 337 337 + type subscriberHandler struct{} 338 338 + 339 339 + func (h *subscriberHandler) HandleEvent(ctx context.Context, event *models.Event) error { 340 340 + switch event.Commit.Collection { 341 341 + case "app.bsky.feed.repost": 342 342 + modifySubscribersWithEvent( 343 343 + event, 344 344 + func(s *SubscriberData, r bsky.FeedRepost) { delete(s.Reposts, r.Subject.Uri) }, 345 345 + func(s *SubscriberData, r bsky.FeedRepost) { 346 346 + s.Reposts[r.Subject.Uri] = struct{}{} 347 347 + }, 348 348 + ) 349 349 + case "app.bsky.graph.follow": 350 350 + modifySubscribersWithEvent( 351 351 + event, 352 352 + func(s *SubscriberData, r bsky.GraphFollow) { delete(s.Follows, r.Subject) }, 353 353 + func(s *SubscriberData, r bsky.GraphFollow) { 354 354 + s.Follows[r.Subject] = struct{}{} 355 355 + }, 356 356 + ) 357 357 + updateLikeStreamOpts() 358 358 + } 359 359 + 360 360 + return nil 361 361 + } 362 362 + 363 363 + type ModifyFunc[v any] func(*SubscriberData, v) 364 364 + 365 365 + func modifySubscribersWithEvent[v any](event *models.Event, onDelete ModifyFunc[v], onUpdate ModifyFunc[v]) error { 366 366 + var data v 367 367 + if err := json.Unmarshal(event.Commit.Record, &data); err != nil { 368 368 + logger.Error("Failed to unmarshal repost", "error", err) 369 369 + return nil 370 370 + } 371 371 + 372 372 + subscribersMux.Lock() 373 373 + defer subscribersMux.Unlock() 374 374 + 375 375 + if subscriber, exists := subscribers[event.Did]; exists { 376 376 + if event.Commit.Operation == models.CommitOperationDelete { 377 377 + onDelete(subscriber, data) 378 378 + } else { 379 379 + onUpdate(subscriber, data) 380 380 + } 381 381 + } 382 382 + 383 383 + return nil 384 384 + }