import argparse
import time
from parse import parse
import datetime
from enum import Enum


class Channel(Enum):
    WHISPER = 0
    GLOBAL = 1
    PARTY = 2
    LOCAL = 3 
    TRADE = 4 
    GUILD = 5


channel_mapping = {'#': Channel.GLOBAL,
                   '@': Channel.WHISPER,
                   '%': Channel.PARTY,
                   '': Channel.LOCAL,
                   '$': Channel.TRADE,
                   '&': Channel.GUILD}

class Message():
    def __init__(self, message: str, date: datetime.datetime, user: str, channel: Channel) -> None:
        self.message = message
        self.date = date
        self.user = user
        self.channel = channel
        
    def __str__(self) -> str:
        return f'{self.date} - {self.channel.name}: {self.user}: {self.message}'

def follow(thefile: str):
    thefile.seek(0, 2)
    while True:
        line = thefile.readline()
        if not line:
            time.sleep(0.1)
            continue
        yield line
        

def parse_string(text: str) -> Message:
    result = parse('{date} {time} {timemicro} {noidea} [{level} {client} {id}]{mess_type:1}{user}: {message}', text)
    if not result: 
        return None
    date = datetime.datetime.strptime(f'{result["date"]} {result["time"]}', '%Y/%m/%d %H:%M:%S')
    return Message(result['message'], date, result['user'], channel_mapping[result['mess_type']])


def setup_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(
        description='Poe Trader', epilog="And that's how you trade")

    parser.add_argument(
        '-l', '--logfile', help='Path of the logfile that should be used', default=r'D:\Poe\logs\Client.txt')

    return parser.parse_args()


if __name__ == "__main__":
    args = setup_args()
    logfile = open(args.logfile, 'r', encoding='utf8')
    loglines = follow(logfile)
    for line in loglines:
        message = parse_string(line)
        print(message)