internet-delay-chat/clients/srirce/main.py

#    srIRCeBot, a Simple IRC RElay Bot
#
#    Copyright (C) 2022 Ferass EL HAFIDI <vitali64pmemail@protonmail.com>
#    Copyright (C) 2022 Andrew Yu <andrew@andrewyu.org>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <https://www.gnu.org/licenses/>.


import miniirc
import miniirc_idc
from miniirc_extras import *
from time import sleep

# Configuration

server          = [ ['irc.andrewyu.org', 'IRC'], ['andrewyu.org', 'IDC'] ]
relayedChannels = [ '#IDC'                     , '#hackers@andrewyu.org' ]
nick            = 'idcbot'
debug           = True

# IRC
irc = miniirc.IRC(server[0][0], 6697, nick, relayedChannels[0], ns_identity=None, debug=False)
# IDC
idc = miniirc_idc.IDC(server[1][0], 6835, nick, relayedChannels[1], ns_identity=["idcbot", ""], debug=debug, ssl=True)

# Relay functions

def relay_msgs(chat, hostmask, args, which):
    #sleep(0.5) # Sleep for 0.5 seconds
    whichchan = 0 if which == 1 else 1
    # Send in the irc side what the user sent in the idc side and vice-versa
    print("NEW MESSAGE")
    chat.send('PRIVMSG', relayedChannels[whichchan], 
                "<\x03%s%s\x0300> %s" 
                % (len(hostmask[0]), hostmask[0], args[-1]))

def relay_nick(chat, hostmask, args, which):
    #sleep(0.5)
    whichchan = 0 if which == 1 else 1
    # Send in the idc side the nick change in the irc side
    chat.send('PRIVMSG', relayedChannels[whichchan], 
                "\x03%s%s\x0300 is now known as \x0302%s" 
                % (len(hostmask[0]), hostmask[0], args[-1]))

def relay_joins(chat, hostmask, args, which):
    #sleep(0.5)
    whichchan = 0 if which == 0 else 1
    # Send in the idc side joins in the irc side and vice-versa
    chat.send('PRIVMSG', relayedChannels[whichchan], 
                "\x03%s%s\x03 has joined \x02%s" 
                % (len(hostmask[0]), hostmask[0], args[-1]))

def relay_mode(chat, hostmask, args, which):
    #sleep(0.5)
    whichchan = 0 if which == 1 else 1
    chat.send('PRIVMSG', relayedChannels[whichchan], 
                "mode/%s: [%s %s] by %s" 
                % (server[which][1], args[0], args[1], args[2], hostmask[0]))

def relay_quits(chat, hostmask, args, which):
    #sleep(0.5)
    whichchan = 0 if which == 1 else 1
    # Send in the idc side quits in the irc side and vice-versa
    chat.send('PRIVMSG', relayedChannels[whichchan], 
                "\x03%s%s\x03 has quit %s" 
                % (len(hostmask[0]), hostmask[0], args[0]))

def relay_kicks(chat, hostmask, args, which):
    #sleep(0.5)
    whichchan = 0 if which == 1 else 1
    # Send in the idc side kicks in the irc side and vice-versa
    chat.send('PRIVMSG', relayedChannels[whichchan], 
                "%s was kicked from %s by %s" 
                % (server[which][1], args[1], args[0], hostmask[0]))

# Functions that run the relay functions, depending on the server where the message has been sent
@irc.Handler('NICK', colon=False)
def handle_nicks(chat, hostmask, args):
    if chat == idc:
        relay_nick(irc, hostmask, args, 0)
    elif chat == irc:
        relay_nick(idc, hostmask, args, 1)


@irc.Handler('KICK', colon=False)
def handle_kicks(chat, hostmask, args):
    if chat == irc:
        relay_kicks(idc, hostmask, args, 0)
    elif chat == idc:
        relay_kicks(irc, hostmask, args, 1)


@miniirc.Handler('JOIN', colon=False)
def handle_joins(chat, hostmask, args):
    if chat == irc:
        relay_joins(idc, hostmask, args, 0)
    elif chat == idc:
        relay_joins(irc, hostmask, args, 1)

@miniirc.Handler('PART', colon=False)
def handle_quits(chat, hostmask, args):
    if chat == irc:
        relay_quits(irc, hostmask, args, 0)
    elif chat == idc:
        relay_quits(idc, hostmask, args, 1)


@miniirc.Handler('PRIVMSG', colon=False)
def handle_privmsgs(chat, hostmask, args):
    words = args[-1].split(" ")
    w = [x.lower() for x in words]
    channel = args[0]
    command = words[0].lower()
    if chat == irc:
        print("IRC->IDC")
        relay_msgs(idc, hostmask, args, 0)
    else:
        relay_msgs(irc, hostmask, args, 1)
        print("IDC->IRC")

@miniirc.Handler('CHANMSG', colon=False)
def handle_privmsgs(chat, hostmask, args):
    words = args[-1].split(" ")
    w = [x.lower() for x in words]
    channel = args[0]
    command = words[0].lower()
    if chat == irc:
        print("IRC->IDC")
        relay_msgs(idc, hostmask, args, 0)
    else:
        relay_msgs(irc, hostmask, args, 1)
        print("IDC->IRC")



@miniirc.Handler('MODE', colon=False)
def handle_mode(chat, hostmask, args):
    if chat == irc:
        relay_mode(irc, hostmask, args, 0)
    elif chat == idc:
        relay_mode(idc, hostmask, args, 1)