From fd96a0b694c6a66010521ec19d79e7dcb62f1843 Mon Sep 17 00:00:00 2001
From: Tim Abbott <tabbott@zulip.com>
Date: Mon, 26 Aug 2013 15:43:29 -0400
Subject: [PATCH] zmirror: Add monitoring for personals mirrors.

(imported from commit acb7f2222076d2f3884a2e52c7032cb48fde1757)
---
 bots/zephyr_mirror_backend.py      | 18 ++++++++++++++++++
 bots/zmirror_private.conf.template |  2 +-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/bots/zephyr_mirror_backend.py b/bots/zephyr_mirror_backend.py
index 90a73e4..ace4c75 100755
--- a/bots/zephyr_mirror_backend.py
+++ b/bots/zephyr_mirror_backend.py
@@ -366,6 +366,12 @@ def process_notice(notice, log):
 
     zephyr_class = notice.cls.lower()
 
+    if zephyr_class == options.nagios_class:
+        # Mark that we got the message and proceed
+        with file(options.nagios_path, "w") as f:
+            f.write("0\n")
+        return
+
     if notice.recipient != "":
         is_personal = True
     # Drop messages not to the listed subscriptions
@@ -517,6 +523,8 @@ def zephyr_to_zulip(options):
             zephyr_subscribe_autoretry(("message", "*", "%me%"))
             if subscribed_to_mail_messages():
                 zephyr_subscribe_autoretry(("mail", "inbox", "%me%"))
+        if options.nagios_class:
+            zephyr_subscribe_autoretry((options.nagios_class, "*", "*"))
         if options.use_sessions:
             file(options.session_path, "w").write(zephyr._z.dump_session())
 
@@ -961,6 +969,12 @@ def parse_args():
     parser.add_option('--session-path',
                       default=None,
                       help=optparse.SUPPRESS_HELP)
+    parser.add_option('--nagios-class',
+                      default=None,
+                      help=optparse.SUPPRESS_HELP)
+    parser.add_option('--nagios-path',
+                      default=None,
+                      help=optparse.SUPPRESS_HELP)
     parser.add_option('--use-sessions',
                       default=False,
                       action='store_true',
@@ -1029,6 +1043,10 @@ or specify the --api-key-file option.""" % (options.api_key_file,))))
         # don't need to read it in
         os.environ["HUMBUG_API_KEY"] = api_key
 
+    if options.nagios_path is None and options.nagios_class is not None:
+        logger.error("\n" + "nagios_path is required with nagios_class\n")
+        sys.exit(1)
+
     zulip_account_email = options.user + "@mit.edu"
     import zulip
     zulip_client = zulip.Client(
diff --git a/bots/zmirror_private.conf.template b/bots/zmirror_private.conf.template
index e1bf5d5..a40084c 100644
--- a/bots/zmirror_private.conf.template
+++ b/bots/zmirror_private.conf.template
@@ -1,5 +1,5 @@
 [program:zmirror-USERNAME]
-command=python /home/humbug/humbug/bots/zephyr_mirror_backend.py --root-path=/home/humbug/humbug/bots --user=USERNAME --enable-log=/home/humbug/logs/mirror-log-%(program_name)s --use-sessions --session-path=/home/humbug/zephyr_sessions/%(program_name)s --api-key-file=/home/humbug/api-keys/%(program_name)s --ignore-expired-tickets
+command=python /home/humbug/humbug/bots/zephyr_mirror_backend.py --root-path=/home/humbug/humbug/bots --user=USERNAME --enable-log=/home/humbug/logs/mirror-log-%(program_name)s --use-sessions --session-path=/home/humbug/zephyr_sessions/%(program_name)s --api-key-file=/home/humbug/api-keys/%(program_name)s --ignore-expired-tickets --nagios-path=/home/humbug/mirror_status/%(program_name)s --nagios-class=zulip-mirror-nagios
 priority=200                   ; the relative start priority (default 999)
 autostart=true                 ; start at supervisord start (default: true)
 autorestart=true               ; whether/when to restart (default: unexpected)