# DougDoug Note: 

# This is the code that connects to Twitch and checks for new messages.

# You should not need to modify anything in this file, just use as is.

# Original code by Wituz, updated by DDarknut

import sys

import socket

import re

import random

import time

MAX_TIME_TO_WAIT_FOR_LOGIN = 3

class Twitch:

    re_prog = None

    sock = None

    partial = b''

    login_ok = False

    channel = ''

    login_timestamp = 0

    def twitch_connect(self, channel):

        if self.sock: self.sock.close()

        self.sock = None

        self.partial = b''

        self.login_ok = False

        self.channel = channel

        # Compile regular expression

        self.re_prog = re.compile(b'^(?::(?:([^ !\r\n]+)![^ \r\n]*|[^ \r\n]*) )?([^ \r\n]+)(?: ([^:\r\n]*))?(?: :([^\r\n]*))?\r\n', re.MULTILINE)

        # Create socket

        print('Connecting to Twitch...')

        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        # Attempt to connect socket

        self.sock.connect(('irc.chat.twitch.tv', 6667))

        # Log in anonymously

        user = 'justinfan%i' % random.randint(10000, 99999)

        print('Connected to Twitch. Logging in anonymously...')

        self.sock.send(('PASS asdf\r\nNICK %s\r\n' % user).encode())

        self.sock.settimeout(1.0/60.0)

        self.login_timestamp = time.time()

    # Attempt to reconnect after a delay

    def reconnect(self, delay):

        time.sleep(delay)

        self.twitch_connect(self.channel)

    # Returns a list of irc messages received

    def receive_and_parse_data(self):

        buffer = b''

        while True:

            received = b''

            try:

                received = self.sock.recv(4096)

            except socket.timeout:

                break

            # except OSError as e:

            #     if e.winerror == 10035:

            #         # This "error" is expected -- we receive it if timeout is set to zero, and there is no data to read on the socket.

            #         break

            except Exception as e:

                print('Unexpected connection error. Reconnecting in a second...', e)

                self.reconnect(1)

                return []

            if not received:

                print('Connection closed by Twitch. Reconnecting in 5 seconds...')

                self.reconnect(5)

                return []

            buffer += received

        if buffer:

            # Prepend unparsed data from previous iterations

            if self.partial:

                buffer = self.partial + buffer

                self.partial = []

            # Parse irc messages

            res = []

            matches = list(self.re_prog.finditer(buffer))

            for match in matches:

                res.append({

                    'name':     (match.group(1) or b'').decode(errors='replace'),

                    'command':  (match.group(2) or b'').decode(errors='replace'),

                    'params':   list(map(lambda p: p.decode(errors='replace'), (match.group(3) or b'').split(b' '))),

                    'trailing': (match.group(4) or b'').decode(errors='replace'),

                })

            # Save any data we couldn't parse for the next iteration

            if not matches:

                self.partial += buffer

            else:

                end = matches[-1].end()

                if end < len(buffer):

                    self.partial = buffer[end:]

​​