wubloader/zulip_bots/zulip_bots/pubbot.py

import json
import logging
import os
import socket
import time

from .config import get_config
from .zulip import Client

import requests
session = requests.Session()

def stream():
	tt = 0
	tr = 0
	while True:
		channels = ["total:RZZQRDQNLNLW"]
		channels = ",".join(channels)
		resp = session.get(f"https://ps8.pndsn.com/v2/subscribe/sub-cbd7f5f5-1d3f-11e2-ac11-877a976e347c/{channels}/0",
			params={
				"tt": tt,
				"tr": tr,
			},
		)
		resp.raise_for_status()
		data = resp.json()
		for msg in data.get("m", []):
			yield msg
		tt = data["t"]["t"]
		tr = data["t"]["r"]


def get_prize(prize_id):
	resp = session.get("https://desertbus.org/wapi/prize/{}".format(prize_id))
	resp.raise_for_status()
	return resp.json()["prize"]


giveaway_cache = [None, None]
def get_giveaway():
	REFRESH_RISING = 30
	REFRESH_FALLING = 300
	t, g = giveaway_cache
	now = time.time()
	timeout = REFRESH_RISING if g is None else REFRESH_FALLING
	if t is None or now - t > timeout:
		resp = session.get("https://desertbus.org/wapi/currentGiveaway")
		resp.raise_for_status()
		g = resp.json()["giveaway"]
		giveaway_cache[0] = t
		giveaway_cache[1] = g
	return g


def main(conf_file, message_log_file, name=socket.gethostname()):
	"""Config:
		zulip_url
		zulip_email
		zulip_api_key
	"""
	logging.basicConfig(level="INFO")

	config = get_config(conf_file)
	client = Client(config["zulip_url"], config["zulip_email"], config["zulip_api_key"])

	message_log = open(message_log_file, "a")
	def write_log(log):
		message_log.write(json.dumps(log) + '\n')

	write_log({
		"type": "startup",
		"host": name,
		"pid": os.getpid(),
		"time": time.time(),
	})

	prizes = {}
	total = None
	for msg in stream():
		log = {
			"type": "unknown",
			"host": name,
			"pid": os.getpid(),
			"time": time.time(),
			"message": msg,
		}

		try:
			try:
				message_time = float(msg["p"]["t"]) / 10000000
			except (KeyError, ValueError):
				message_time = None

			log["message_time"] = message_time

			if msg["c"].startswith("total"):
				log["type"] == "total"
				increase = None if total is None else msg["d"] - total
				log["increase"] = increase
				increase_str = "" if increase is None else " (+${:.2f})".format(msg["d"] - total)
				giveaway = None
				entries_str = ""
				if increase is not None and giveaway is not None:
					amount = giveaway["amount"]
					if (increase + 0.005) % amount < 0.01:
						entries = int((increase + 0.005) / amount)
						log["giveaway_amount"] = amount
						log["giveaway_entries"] = entries
						entries_str = " ({} entries of ${:.2f})".format(entries, amount)
				logging.info("New donation total: {}{}{}".format(msg["d"], increase_str, entries_str))
				client.send_to_stream("bot-spam", "Donation Firehose", "Donation total is now ${:.2f}{}{}".format(msg["d"], increase_str, entries_str))
				if increase is not None and increase >= 500:
					client.send_to_stream("bot-spam", "Notable Donations", "Large donation of ${:.2f} (total ${:.2f}){}".format(increase, msg['d'], entries_str))
				total = msg["d"]

			elif msg["c"] == "db_vue" and msg["d"].get("channel").startswith("prize:"):
				log["type"] = "prize"
				d = msg["d"]
				prize_id = msg["d"]["channel"].split(":")[1]
				data = d["data"]
				logging.info(f"Prize update for {prize_id}: {data}")
				if prize_id not in prizes:
					prizes[prize_id] = get_prize(prize_id)
				log["prize_id"] = prize_id
				if "bidder" in data and "bid" in data:
					log["bidder"] = data["bidder"]
					log["bid"] = data["bid"]
					client.send_to_stream("bot-spam", "Bids", "At <time:{time}>, {bidder} bid ${bid:.2f} for [{title}](https://desertbus.org/prize/{prize_id})".format(
						time = message_time,
						bidder = data["bidder"],
						bid = data["bid"],
						title = prizes[prize_id]["title"],
						prize_id = prize_id,
					))

			else:
				logging.warning("Unknown message: {}".format(msg))
		except Exception:
			logging.exception(f"Failed to handle message {msg}")
			log["type"] = "error"

		write_log(log)


if __name__ == '__main__':
	import argh
	argh.dispatch_command(main)