darkfi/bin/darkirc/script/bots/taubot.py

import argparse
import signal
import irc
import json
import sys

# parse arguments
parser = argparse.ArgumentParser(description='IRC bot to send a pipe to an IRC channel')
parser.add_argument('--server',default='127.0.0.1', help='IRC server')
parser.add_argument('--port', default=22024, help='port of the IRC server')
parser.add_argument('--nickname', default='tau-notifier', help='bot nickname in IRC')
parser.add_argument('--channel', default="#dev", action='append', help='channel to join')
parser.add_argument('--pipe', default="/tmp/tau_pipe" , help='pipe to read from')
parser.add_argument('--skip', default="prv", help='Project or Tags to skip notifications for')
parser.add_argument('--skip-workspace', default="prv-ws", help='Workspace to skip notifications for')
parser.add_argument('--alt-chan', default="#test", required='--skip' in sys.argv or '--skip-workspace' in sys.argv, help='Alternative channel to send notifications to when there are skipped tasks')

args = parser.parse_args()

channels = [args.channel, args.alt_chan] if args.alt_chan is not None else args.channel

ircc = irc.IRC()
ircc.connect(args.server, int(args.port), channels, args.nickname)

def signal_handler(sig, frame):
    print("Caught termination signal, cleaning up and exiting...")
    ircc.disconnect(args.server, args.port)
    print("Shut down successfully")
    exit(0)

signal.signal(signal.SIGINT, signal_handler)

while True:
    with open(args.pipe) as handle:
        while True:
            log_line = handle.readline()
            if not log_line:
                break
            print(log_line)
            print("======================================")
            task = json.loads(log_line)
            channel = args.channel

            for event in task['events']:
                cmd = event['action']
                if cmd == "add_task":
                    user = task['owner']
                    refid = task['ref_id'][:6]
                    title = task['title']
                    assigned = ", ".join(task['assign'])

                    project = task['project'] if task['project'] is not None else []
                    if args.skip in project or '+' + args.skip in task['tags'] or args.skip_workspace in task['workspace']:
                        channel = args.alt_chan

                    if len(assigned) > 0:
                        notification = f"{user} added task ({refid}): {title}. assigned to {assigned}"
                    else:
                        notification = f"{user} added task ({refid}): {title}"
                    # print(notification)
                    ircc.send(channel, notification)
                elif cmd == "state":
                    user = event['author']
                    state = event['content']
                    refid = task['ref_id'][:6]
                    title = task['title']

                    project = task['project'] if task['project'] is not None else []
                    if args.skip in project or '+' + args.skip in task['tags'] or args.skip_workspace in task['workspace']:
                        channel = args.alt_chan

                    if state == "start":
                        notification = f"{user} started task ({refid}): {title}"
                    elif state == "pause":
                        notification = f"{user} paused task ({refid}): {title}"
                    elif state == "stop":
                        notification = f"{user} stopped task ({refid}): {title}"
                    elif state == "cancel":
                        notification = f"{user} canceled task ({refid}): {title}"
                    # print(notification)
                    ircc.send(channel, notification)
                elif cmd == "comment":
                    user = event['author']
                    refid = task['ref_id'][:6]
                    title = task['title']
                    
                    project = task['project'] if task['project'] is not None else []
                    if args.skip in project or '+' + args.skip in task['tags'] or args.skip_workspace in task['workspace']:
                        channel = args.alt_chan

                    notification = f"{user} commented on task ({refid}): {title}"
                    # print(notification)
                    ircc.send(channel, notification)
                elif cmd == "assign":
                    user = event['author']
                    assignees = event['content']
                    refid = task['ref_id'][:6]
                    title = task['title']

                    project = task['project'] if task['project'] is not None else []
                    if args.skip in project or '+' + args.skip in task['tags'] or args.skip_workspace in task['workspace']:
                        channel = args.alt_chan
                    
                    removed = []
                    added = []
                    for assignee in assignees.split(', '):
                        if assignee.startswith("-"):
                            removed.append(assignee[1:])
                        else:
                            added.append(assignee)

                    if removed and added:
                        notification = f"{user} added {', '.join(added)} and removed {', '.join(removed)} from assign in task ({refid}): {title}"
                    if (not removed) and added:
                        notification = f"{user} added {', '.join(added)} to assign in task ({refid}): {title}"
                    if (not added) and removed:
                        notification = f"{user} removed {', '.join(removed)} from assign in task ({refid}): {title}"

                    # print(notification)
                    ircc.send(channel, notification)