Initial commit

4 files changed, 149 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..6911dba
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+env.sh
+rawlog
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..1e1c450
--- /dev/null
+++ b/README.md
@@ -0,0 +1,18 @@
+# Discord bot that notifies you when your user count in a guild is prime
+
+Also this was written in bash
+
+## Installation
+
+Required software:
+
+ - bash 5
+ - jq 1.6
+ - curl 8.2.1
+ - websocat
+ - paste
+
+Copy `env-example.sh` to `env.sh` and enter your variables.
+
+Run `./bot.sh`
+
diff --git a/bot.sh b/bot.sh
new file mode 100755
index 0000000..f135bd4
--- /dev/null
+++ b/bot.sh
@@ -0,0 +1,122 @@
+#!/usr/bin/env bash
+
+
+
+source env.sh
+is_prime() {
+    if (( $1 < 3 )); then
+        return 0
+    fi
+    s=$(seq 2 $(($1-1)))
+    while IFS= read -r i; do
+        if (( $1 % $i == 0 )); then
+            return 1
+        fi
+    done<<<"$s"
+    return 0
+}
+
+rm -fr wsr wsw o w seq g tty
+mkfifo wsr
+trap 'rm -fr wsr wsw w o g seq tty' EXIT
+mkfifo wsw
+mkdir o
+mkdir g
+mkdir w
+websocat wss://gateway.discord.gg <wsw >wsr &
+ln -s $(tty) tty
+heartbeat() {
+    sendmsg '{"op":1,"d":'$(cat seq)'}'
+}
+dbg() {
+    echo "$@" >tty
+}
+sendmsg() {
+    fn=$(mktemp -p w)
+    echo "$@" > $fn
+    ln -s "$(readlink -f $fn)" o
+}
+
+(
+dbg subshell started
+while [[ -d o ]]; do
+    for m in $(find $(readlink -f o) -type l); do
+        dbg Sending message $m
+        cat $m
+        dbg "$(cat $m)"
+        rm $m
+    done
+done > wsw)&
+
+(
+while [[ -d o ]]; do
+    dbg handshaking
+    heartbeat 
+    sleep 10
+done
+)&
+
+
+test_guild() {
+    dbg Testing guild $1
+    nonce=$(mktemp -p g -d)
+    sendmsg '{"op":8,"d":{"guild_id":"'$1'", "query":"", "limit":0,"nonce":"'$nonce'"}}'
+}
+
+
+dispatch() {
+    event=$(echo "$1" | jq -r .t)
+    dbg Dispatching event $event
+    if [[ $event = GUILD_MEMBER_ADD ]] || [[ $event = GUILD_MEMBER_REMOVE ]] ; then
+        guildid=$(echo "$1" | jq -r .d.guild_id)
+        dbg Someone joined $guildid
+        test_guild $guildid
+    elif [[ $event = MESSAGE_CREATE ]]; then
+        content="$(echo "$1"|jq -r .d.content)"
+        dbg Received message
+        case "$content" in 
+            gtest*)
+                guildid=$(echo $content | tr -d '\n gtes')
+                test_guild $guildid
+                ;;
+        esac
+    elif [[ $event = GUILD_MEMBERS_CHUNK ]]; then
+        chunk_size=$(echo "$1" | jq -r '[.d.members[]]|length')
+        nonce=$(echo "$1" | jq -r '.d.nonce')
+        chunk_index=$(echo "$1"| jq -r '.d.chunk_index')
+        chunk_count=$(echo "$1"| jq -r '.d.chunk_count')
+        echo $chunk_size > $nonce/$chunk_index
+        total_received=$(find $nonce -type f| wc -l)
+        dbg "Received chunk $chunk_index/$chunk_count (total $total_received) of size $chunk_size for $nonce"
+        if [[ $total_received -eq $chunk_count ]]; then
+            total_member_count=$(cat $nonce/* | paste -sd+ - | bc)
+            dbg "we are done for now. total members: $total_member_count"
+            if is_prime $total_member_count||true; then
+                dbg "prime number woohoooo sending message to $ANNOUNCEMENT"
+                curl -X POST -H "Authorization: Bot $TOKEN" https://discord.com/api/v10/channels/$ANNOUNCEMENT/messages -F payload_json='{"content": "WOOOO BABY A PRIME NUMBER!! WE ARE NOW '$total_member_count' MEMBERS!!!! WOOO THATS WHAT IVE BEEN WAITING FOR"}' >/dev/null 2>/dev/null
+            fi
+        fi
+
+    fi
+}
+
+echo null > seq
+while true; do
+    dbg Trying to read line
+    if read -r line ; then 
+        echo $line | jq 'if .s then .s else '"$(cat seq)"' end'>seq
+        op=$(echo $line | jq .op)
+        echo $line >> rawlog
+        if [[ $op -eq 10 ]]; then
+            sendmsg '{"op":2,"d":{"token":"'$TOKEN'", "properties":{"os":"linux","browser":"bash","device":"bash"},"presence":{"status":"online","afk":false, "activities":[{"type":0,"name":"Being coded in Bash"}]}, "intents":33287}}' # change intents back to 7
+        elif [[ $op -eq 0 ]]; then
+            dispatch "$line"
+        elif [[ $op -eq 1 ]] || [[ $op -eq 10 ]]; then
+            dbg Received heartbeat
+            dbg New sequence number = $(cat seq)
+        else 
+            dbg Received message of type $op
+        fi
+    fi
+done <wsr
+
diff --git a/env-example.sh b/env-example.sh
new file mode 100644
index 0000000..49833e9
--- /dev/null
+++ b/env-example.sh
@@ -0,0 +1,7 @@
+# Copy to env.sh
+
+# Discord Bot Token
+export TOKEN=yourtoken
+# Announcement channel id
+export ANNOUNCEMENT=455292999455211530
+