#!/usr/bin/env python3
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

import socket
import random
import time
import shelve
import _pickle

class Dealer: 
    def __init__(self, name):
        self.name = name
        f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
        f["name"] = self.name
        if not "inventory" in f:
            f["inventory"] = {}
            f["inventory"]["money"] = 500
            f["inventory"]["meth"] = 10
            f["inventory"]["heroin"] = 10
            f["inventory"]["cocaine"] = 10
            f.sync()
            self.inventory = {}
            self.inventory["money"] = 500
            self.inventory["meth"] = 10
            self.inventory["heroin"] = 10
            self.inventory["cocaine"] = 10
        else:
            self.inventory = f["inventory"]
        f.close()


    def sell(self, drug, price, amount):
        f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
        if (int(f["inventory"][drug] >= 1)):
            f["inventory"][drug] -= int(amount)
            f["inventory"]["money"] += (int(amount) * int(price))
            drug_respond = str(drug)
            drug_respond_amount = str(f["inventory"][drug])
            drug_respond_cash = str(f["inventory"]["money"])
            sendmsg(channel, ("You have " + drug_respond_amount + " " + str(drug) + " left, and $" + drug_respond_cash + " left"))
            f.sync()
            f.close()
        else:
            sendmsg(channel, "Not enough drugs")

    def buy(self, drug, price, amount):
        f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
        if (int((f["inventory"]["money"])) >= (int(amount) * int(price))):
            f["inventory"][drug] += int(amount)
            f["inventory"]["money"] -= (int(amount) * int(price))
            drug_respond = str(drug)
            drug_respond_amount = str(f["inventory"][drug])
            drug_respond_cash = str(f["inventory"]["money"])
            sendmsg(channel, ("You have " + drug_respond_amount + " " + str(drug) + " left, and $" + drug_respond_cash + " left"))
            f.sync()
            f.close()
        else:
            sendmsg(channel, "Not enough cash")

        
        
    def check_amount(self, drug):
        f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
        if drug is "heroin":
            x = f["inventory"]['heroin']
            return(x)
            f.sync()
            f.close()
        if drug is "cocaine":
            return(self.inventory['cocaine'])
            f.sync()
            f.close()
        if drug is "meth":
            return(self.inventory['meth'])
            f.sync()
            f.close()
        if drug is "money":
            return(self.inventory['money'])
            f.sync()
            f.close()
        else:
            return("Drug not specified")
            f.close()


#******** IRC SHIT *********

server = "irc.freenode.net"
channel = "#lainchan"
botnick = "drug-bot"
password = ""
#all bytes must be converted to UTF-8
def joinchan(chan):
    ircsock.send(bytes("JOIN "+ chan +"\n", "UTF-8"))

ircsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ircsock.connect((server, 6667))
ircsock.send(bytes("USER "+ botnick +" "+ botnick +" "+ botnick +" :connected\n", "UTF-8"))
ircsock.send(bytes("NICK "+ botnick +"\n", "UTF-8"))
ircsock.send(bytes("NICKSERV :IDENTIFY %s\r\n" % password, "UTF-8"))
time.sleep(3)
joinchan(channel)
readbuffer = ""
while 1:
    drug_price = random.randrange(5,15,1)
    def answer_query():
        return(random.choice(responses))
    msg = answer_query()
    def sendmsg(chan , msg):
        ircsock.send(bytes("PRIVMSG "+ chan +" :"+ msg +"\n", "UTF-8"))
    ircmsg = ircsock.recv(2048)
    ircmsg = ircmsg.strip(bytes('\n\r', "UTF-8")) 
    print(ircmsg) 

    readbuffer = readbuffer+ircsock.recv(1024).decode("UTF-8")
    temp = str.split(readbuffer, "\n")
    print(readbuffer)
    readbuffer=temp.pop( )

    for line in temp:
        line = str.rstrip(line)
        line = str.split(line)
        try:
            if(line[0] == "PING"):
                ircsock.send(bytes("QUOTE :PONG %s\r\n" % line[7], "UTF-8"))
            if(line[3] == ":.sell"):
                sender = ""
                for char in line[0]:
                    if(char == "!"):
                        break
                    if(char != ":"):
                        sender += char
                current_player = Dealer(sender)
                drug_query = line[5]
                drug_amount = line[4]
                current_player.sell(drug_query, drug_price, drug_amount)
            if(line[3] == ":.buy"):
                sender = ""
                for char in line[0]:
                    if(char == "!"):
                        break
                    if(char != ":"):
                        sender += char
                current_player = Dealer(sender)
                drug_query = line[5]
                drug_amount = line[4]
                current_player.buy(drug_query, drug_price, drug_amount)
            if(line[3] == ":.check"):
                sender = ""
                for char in line[0]:
                    if(char == "!"):
                        break
                    if(char != ":"):
                        sender += char
                current_player = Dealer(sender)
                requested_item = line[4]
                #this is the part i'm testing to try and get working
                if requested_item:
                    if(requested_item == 'heroin'):
                        sendmsg(channel, (sender + "you have " + str(current_player.check_amount('heroin')) + " heroin left" +"\n"))
                    elif(requested_item == 'meth'):
                        sendmsg(channel, (sender + ", you have " + str(current_player.check_amount('meth')) + " meth left" +"\n"))
                    elif(requested_item == 'money'):
                        sendmsg(channel, (sender + ", you have " + str(current_player.check_amount('money')) + " money left" +"\n" ))
                    elif(requested_item == 'cocaine'):
                        sendmsg(channel, (sender + ", you have " + str(current_player.check_amount('cocaine')) + " cocaine left" +"\n"))
                    else:
                        sendmsg(channel, ("Invalid option. Options for .check are heroin, meth, cocaine, or money"))
        except:
            continue