From de713f552cc3559c449836eac73d44e33a7edca7 Mon Sep 17 00:00:00 2001
From: Tim Abbott <tabbott@humbughq.com>
Date: Tue, 5 Feb 2013 09:58:27 -0500
Subject: [PATCH] Add script to sync public streams.

(imported from commit 73fc73d585dada4a250b5861437b5a5a0e3765b9)
---
 bots/sync-public-streams | 66 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100755 bots/sync-public-streams

diff --git a/bots/sync-public-streams b/bots/sync-public-streams
new file mode 100755
index 0000000..044072a
--- /dev/null
+++ b/bots/sync-public-streams
@@ -0,0 +1,66 @@
+#!/usr/bin/python
+import sys
+from os import path
+import logging
+import time
+import simplejson
+import subprocess
+import unicodedata
+
+sys.path.append(path.join(path.dirname(__file__), '..', 'api'))
+import humbug
+
+humbug_client = humbug.Client()
+
+def fetch_public_streams():
+    public_streams = set()
+
+    try:
+        res = humbug_client.get_public_streams()
+        if res.get("result") == "success":
+            streams = res["streams"]
+        else:
+            logging.error("Error getting public streams:\n%s" % res)
+            return None
+    except Exception:
+        logging.exception("Error getting public streams:")
+        return None
+
+    for stream in streams:
+        # Zephyr class names are canonicalized by first applying NFKC
+        # normalization and then lower-casing server-side
+        canonical_cls = unicodedata.normalize("NFKC", stream).lower().encode("utf-8")
+        if canonical_cls in ['security', 'login', 'network', 'ops', 'user_locate',
+                             'mit',
+                             'hm_ctl', 'hm_stat', 'zephyr_admin', 'zephyr_ctl']:
+            # These zephyr classes cannot be subscribed to by us, due
+            # to MIT's Zephyr access control settings
+            continue
+
+        public_streams.add(canonical_cls)
+
+    return public_streams
+
+if __name__ == "__main__":
+    log_file = "/home/humbug/sync_public_streams.log"
+    logger = logging.getLogger(__name__)
+    log_format = "%(asctime)s: %(message)s"
+    logging.basicConfig(format=log_format)
+    formatter = logging.Formatter(log_format)
+    logger.setLevel(logging.DEBUG)
+    file_handler = logging.FileHandler(log_file)
+    file_handler.setFormatter(formatter)
+    logger.addHandler(file_handler)
+
+    while True:
+        time.sleep(15)
+        public_streams = fetch_public_streams()
+        if public_streams is None:
+            continue
+
+        f = file("/home/humbug/public_streams.tmp", "w")
+        f.write(simplejson.dumps(list(public_streams)) + "\n")
+        f.close()
+
+        subprocess.call(["mv", "/home/humbug/public_streams.tmp", "/home/humbug/public_streams"])
+