basebot.py

#!/usr/bin/env python3

# -*- coding: utf-8 -*-

import threading
import socket
import time
import os
from . import botthread
from . import irccommand
from .listeners import (echolistener, argecholistener, pinglistener,
                        selfnicklistener, nickfaillistener, versionlistener)

from functions import ansicodes

BASELOGDIR = os.path.abspath("../logs")

class BaseBot(botthread.BotThread):

    def __init__(self, server, port, master):

        super().__init__()

        self.server    = server
        self.port      = port
        self.socket    = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.name      = "tichybot"
        self.uName     = "tichybot"
        self.rName     = "Tichy Bot"
        self.essential = [
                          pinglistener.PingListener(),
                          nickfaillistener.NickFailListener(),
                          selfnicklistener.SelfNickListener(),
                          versionlistener.VersionListener(),
                         ]

        self.listeners = [
                          #echolistener.EchoListener(),
                          argecholistener.ArgEchoListener()
                         ]

        self.currentData = ""

        self.master = master

    #######
    ###
    ##  MAIN
    ###
    #######


    def run(self):
        self.connect()

        while not self.stopped:
            data = self.receiveData()

            self.addToData(data)
            newLines = self.getNewLines()

            for listener in self.essential + self.listeners:
                listener.process(newLines, self)

            if data.startswith("ERROR :Closing link:"):   # Connection ded :(
                self.remove()
                return

        self.quit("Quitting")

    #######
    ###
    ##  CONNECTING
    ###
    #######


    def connect(self):
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.connect( (self.server, self.port) )
        self.socket.setblocking(False)

        nickSend = irccommand.IRCCommand("NICK", [], "")
        userSend = irccommand.IRCCommand("USER", [self.uName, self.uName, "*"], self.rName)

        self.sendCommand(userSend)

        self.receiveData()

        nickSend = irccommand.IRCCommand("NICK", [self.name], "")
        self.sendCommand(nickSend)

    #######
    ###
    ##  DATA
    ###
    #######

    def addToData(self, newData):
        self.currentData += newData

    def getNewLines(self):
        ret = self.currentData.rpartition("\n")

        self.currentData = ret[2]

        return ret[0]


    #######
    ###
    ##  SOCKET
    ###
    #######


    def receiveData(self, buf=2**16, timeout=0):

        try:
            ret = self.socket.recv(buf, timeout)

        except socket.error:
            ret = b""

        ret = ret.decode()
        ret = ret.replace("\r\n", "\n")
        ret2 = ""

        for char in ret:
            if ord(char) <= 127:
                ret2 += char
            else:
                ret2 += "?"

        return ret2


    def sendCommand(self, commandObj):

        for listener in self.listeners:
            listener.processAction(commandObj, self)

        data = str(commandObj) + "\n"
        sData = bytes(data, encoding="utf-8")

        self.socket.send(sData)

    #######
    ###
    ##  MESSAGING
    ###
    #######


    def sendMessage(self, data):

        if not data:
            return

        dataSplit = data.partition(":")
        dataList  = [i for i in dataSplit[0].split() if i]

        try:
            command = dataList[0]
            args    = dataList[1:]
            message = dataSplit[2]
        except:
            self.log("Invalid command '{}'".format(data) )
        else:
            ircCommand = irccommand.IRCCommand(command, args, message)
            self.sendCommand(ircCommand)


    #######
    ###
    ##  LOGGING
    ###
    #######

    def writeToLog(self, line):
        cTime      = time.gmtime()
        cDate = time.strftime("%Y-%m-%d", cTime)

        logdir = BASELOGDIR
        olddir = os.getcwd()

        try:
            os.makedirs(logdir)
        except OSError:
            pass

        os.chdir(logdir)

        line = ansicodes.stripCodes(line)

        log = open(cDate + ".txt", "a")
        log.write(line + "\n")
        log.close()

        os.chdir(olddir)



    def log(self, line):
        cTime      = time.gmtime()
        cTimestamp = time.strftime("<%H:%M>", cTime)

        newline = "{} {}".format(cTimestamp, line)

        self.writeToLog(newline)
        print(newline)


    #######
    ###
    ##  EXITING
    ###
    #######


    def quit(self, reason):
        quitCommand = irccommand.IRCCommand("QUIT", [], reason)
        self.sendCommand(quitCommand)

    def remove(self):
        cTime      = time.gmtime()
        cTimestamp = time.strftime("<%H:%M>", cTime)

        self.log("!!! Exiting")
        del self