oddluck-limnoria-plugins/Serve/plugin.py

import sqlite3
import time
import random
from supybot.commands import *
import supybot.callbacks as callbacks

class Serve(callbacks.Plugin):
    def __init__(self, irc):
        self.__parent = super(Serve, self)
        self.__parent.__init__(irc)
        self.db = sqlite3.connect("/home/ubuntu/limnoria/plugins/Serve/servestats.db")
        self.init_db()

        # Settings for spam replies and date format
        self.settings = {
            "antispam": (1, 3),  # 1 trigger every 3 seconds
            "spamreplies": [
                "Hey hey there, don't you think it's going a bit too fast there? Only {since} sec, since your last ...",
                "I am busy ...",
                "Haven't you just had ?"
            ],
            "dateformat": "%H:%M:%S",  # Corresponds to 'h:i:s' in PHP
        }

        # Dictionary to track the last command time for each user
        self.last_command_time = {}

    def add_ordinal_number_suffix(self, num):
        if not (num % 100 in [11, 12, 13]):
            if num % 10 == 1:
                return f"{num}st"
            elif num % 10 == 2:
                return f"{num}nd"
            elif num % 10 == 3:
                return f"{num}rd"
        return f"{num}th"

    def init_db(self):
        # Create the table if it doesn't exist
        with self.db:
            self.db.execute('''CREATE TABLE IF NOT EXISTS servestats (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                nick TEXT NOT NULL,
                address TEXT NOT NULL,
                type TEXT NOT NULL,
                last REAL NOT NULL,
                today INTEGER NOT NULL,
                total INTEGER NOT NULL,
                channel TEXT NOT NULL,
                network TEXT NOT NULL
            )''')

    def start_daily_clear(self):
        """Start a timer to clear daily stats at midnight."""
        now = time.time()
        midnight = time.mktime(time.strptime(time.strftime('%Y-%m-%d') + ' 00:00:00', '%Y-%m-%d %H:%M:%S')) + 86400
        delay = midnight - now
        if delay < 0:
            delay += 86400  # Schedule for tomorrow if midnight has passed today
        threading.Timer(delay, self.timer_serve).start()

    def timer_serve(self):
        """Clear daily stats and set the timer for the next day."""
        # Clear daily stats
        with self.db:
            self.db.execute("UPDATE servestats SET today = 0")
        self.ircClass.privMsg("#bot", "Cleared serve_today")

        # Start the timer for the next day
        self.start_daily_clear()            

    def _get_stats(self, nick, drink_type, channel, network):
        today = int(time.strftime('%Y%m%d'))
        cursor = self.db.cursor()

        cursor.execute('''SELECT today, SUM(total), COUNT(*)
                        FROM servestats
                        WHERE nick = ? AND type = ? AND channel = ? AND network = ?''',
                    (nick, drink_type, channel, network))

        result = cursor.fetchone()
        if result:
            today_count, total_count, sumtotal = result
            total_count = total_count or 0  # Ensure total_count is not None
        else:
            today_count, total_count, sumtotal = 0, 0, 0

        return today_count, total_count, sumtotal

    def _update_stats(self, nick, address, drink_type, channel, network):
        today = int(time.strftime('%Y%m%d'))
        cursor = self.db.cursor()

        today_count, total_count, sumtotal = self._get_stats(nick, drink_type, channel, network)

        if sumtotal == 0:
            # This means it's the first time this user has requested this drink
            today_count = 1  # Set today's count to 1
            total_count = 1  # Set total count to 1
            cursor.execute('''INSERT INTO servestats (nick, address, type, last, today, total, channel, network)
                            VALUES (?, ?, ?, ?, ?, ?, ?, ?)''',
                        (nick, address, drink_type, time.time(), today_count, total_count, channel, network))
        else:
            if today_count == today:
                today_count += 1
            else:
                today_count = 1

            total_count += 1

            cursor.execute('''UPDATE servestats
                            SET last = ?, today = ?, total = ?
                            WHERE nick = ? AND type = ? AND channel = ? AND network = ?''',
                        (time.time(), today_count, total_count, nick, drink_type, channel, network))

        self.db.commit()
        return today_count, total_count, sumtotal + 1

    def _select_spam_reply(self, last_served_time):
        """Select a spam reply if the user requests drinks too quickly."""
        time_since_last = time.time() - last_served_time
        # Convert the time since last command to seconds
        formatted_time = int(time_since_last)  # Get the time in seconds
        reply = random.choice(self.settings["spamreplies"])
        return reply.format(since=formatted_time)

    def _is_spamming(self, nick):
        """Check if the user is spamming commands too quickly."""
        current_time = time.time()
        last_time = self.last_command_time.get(nick, 0)

        # Get the antispam configuration
        triggers, seconds = self.settings["antispam"]

        # Calculate the threshold in seconds
        threshold = seconds

        # Check the time difference based on the antispam settings
        if current_time - last_time < threshold:  # Allow one command every 'seconds'
            return True
        
        self.last_command_time[nick] = current_time  # Update the last command time
        return False
# BAR

    @wrap([optional('channel')])
    def bar(self, irc, msg, args, channel):
        """+bar - Show all available drink commands."""
        # List of available drink commands
        available_drinks = [
            "+cola",     # Cola
            "+beer",     # Beer
            "+coffee",   # Coffee
            "+cap",      # Cappuccino
            "+whiskey",  # Whiskey
            "+wine",     # Wine
            "+mix",      # Mix
            "+pipe",     # Pipe
            "+coke",     # Coke
            "+pussy",    # Pussy (from your existing code)
            "+ice",      # Ice cream
            # Add other drink commands here as needed
        ]

        # Format the response
        response = "Available drinks: " + ", ".join(available_drinks)
        
        # Send the response
        irc.reply(response)    

# DRINKS
    @wrap([optional('channel')])
    def cola(self, irc, msg, args, channel):
        """+cola - Serve a cola."""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return        

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "cola", channel, network)

        responses = [
            f"Serves icecold cola ({today_count}/{total_count}/{sumtotal})",
            f"Serves cola that has been laying in a pile of shit ~45c ({today_count}/{total_count}/{sumtotal})",
            f"Serves cola that's been standing close to a box of dry ice ~1.3c ({today_count}/{total_count}/{sumtotal})",
            f"Serves a warm flat cola that no one wants ~25c ({today_count}/{total_count}/{sumtotal})",
            f"Serves cola fresh from the fridge, chilled to perfection ~5c ({today_count}/{total_count}/{sumtotal})",
            f"Serves cola that tastes slightly metallic after standing in a can too long ~18c ({today_count}/{total_count}/{sumtotal})"
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)

    @wrap([optional('channel')])
    def beer(self, irc, msg, args, channel):
        """+beer - Serve a beer."""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return        

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "beer", channel, network)

        responses = [
            f"Serves icecold beer ({today_count}/{total_count}/{sumtotal})",
            f"Serves a beer for {nick} that has been laying in a pile of shit ~45c ({today_count}/{total_count}/{sumtotal})",
            f"Serves a warn beer for {nick} ({today_count}/{total_count}/{sumtotal})",
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)        

    @wrap([optional('channel')])
    def coffee(self, irc, msg, args, channel):
        """+coffee - Make a coffee."""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return           

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "coffee", channel, network)

        # Add ordinal suffix to sumtotal
        ordinal_suffix = self.add_ordinal_number_suffix(sumtotal)

        responses = [
            f"Making a cup of coffee for {nick}, {today_count} made today out of {total_count} ordered, making it the {ordinal_suffix} time I made coffee."
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)

    @wrap([optional('channel')])
    def cap(self, irc, msg, args, channel):
        """+cap - Make a Cappuccino."""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "cap", channel, network)

        # Add ordinal suffix to sumtotal
        ordinal_suffix = self.add_ordinal_number_suffix(sumtotal)

        responses = [
            f"Making a nice Cappuccino for {nick}, {today_count} made today out of {total_count} ordered, making it the {ordinal_suffix} time I made Cappuccino."
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)        

    @wrap([optional('channel')])
    def whiskey(self, irc, msg, args, channel):
        """+whiskey - Serve a whiskey."""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "whiskey", channel, network)

        responses = [
            f"serves whiskey on the rocks ({total_count})",
            f"found some weird looking bottle in corner, might hit gold cheers ({total_count})",
            f"cola and bad whiskey for you ({total_count})",
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response) 

    @wrap([optional('channel')])
    def wine(self, irc, msg, args, channel):
        """+wine - Serve a wine."""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "wine", channel, network)

        responses = [
            f"pours up some fine stuff from the basement ({total_count})",
            f"here you are, found something out back ({total_count})",
            f"lucky you we just got one of this left enjoy ({total_count})",
            f"so you're hit hard, where you want it ?, don't cry",
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)
# END

# EATING

    @wrap([optional('channel')])
    def ice(self, irc, msg, args, channel):
        """+ice - Serve some icecreams"""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "ice", channel, network)

        responses = [
            f"here {nick}... one ball for you only ({today_count}/{total_count}/{sumtotal})",
            f"finds a biig icecream for {nick} eat and you get for free ($50 to use toilet) ({today_count}/{total_count}/{sumtotal})",
            f"dusts off something that look like icecream from the corner of fridge, here you go {nick} ({today_count}/{total_count}/{sumtotal})",
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)   
# END        

# MIX/HASH/ETC
    @wrap([optional('channel')])
    def mix(self, irc, msg, args, channel):
        """+mix - You need something to get high with."""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "mix", channel, network)

        # Fetch the list of users in the channel as a list
        channel_users = list(irc.state.channels[channel].users) if channel in irc.state.channels else []

        # Define the list of nicknames to exclude
        excluded_nicks = ['klapvogn', 'chat']  # Add any additional nicknames here

        # Exclude specified nicknames from the list of users
        potential_names = [user for user in channel_users if user not in excluded_nicks]

        # Select a random name from the channel users
        if potential_names:
            random_name = random.choice(potential_names)
        else:
            random_name = 'your mom'  # Fallback if no other users are present

        responses = [
            f"grinding up some weed for a mix ({total_count})",
            f"grabs some of the good stuff for a mix ({total_count})",
            f"sneaks into {random_name}'s stash and steals for a mix, here you go ({total_count})",
            f"Gges strain hunting in India for some good shit for your mix ({total_count})",
            f"goes strain hunting in Morocco for some good shit for your mix ({total_count})",
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)

    @wrap([optional('channel')])
    def pipe(self, irc, msg, args, channel):
        """+pussy - you need some pussy"""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "pipe", channel, network)

        responses = [
            f"goes strain hunting in morocco for some good shit for your pipe ({total_count})",
            f"saw some shit in corner, fills a pipe ({total_count})",
            f"skunky just arrieved peace all over ({total_count})",
        ]

    @wrap([optional('channel')])
    def coke(self, irc, msg, args, channel):
        """+pussy - you need some pussy"""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "coke", channel, network)

        responses = [
            f"Are you stupid? We don't do shit like this... ^_^ ({total_count})"

        ]        

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)        
# END

    @wrap([optional('channel')])
    def pussy(self, irc, msg, args, channel):
        """+pussy - you need some pussy"""
        nick = msg.nick
        address = msg.prefix
        network = irc.network

        # Check if the user is spamming commands
        if self._is_spamming(nick):
            last_served_time = self.last_command_time[nick]
            response = self._select_spam_reply(last_served_time)
            irc.reply(response)
            return            

        # Update stats
        today_count, total_count, sumtotal = self._update_stats(nick, address, "pussy", channel, network)

        responses = [
            f"slaps {nick} in face with a smelly pussy ({total_count})",
            f"sends some pussy {nick}'s way .. ({total_count})",
            f"not enough money to suply you aswell ... ({total_count})",
            f"if you have the cash {nick} I can pull down my undies for you ({total_count})",
        ]

        # Select a random response
        response = random.choice(responses)
        irc.reply(response)     

Class = Serve