import MySQLdb
import random

def GenerateEquipment(phenny, parameters, itemCode):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)
    intIteration=0
    itemName=strRaritySpacing='';

    
    itemCode = itemCode.split(':')

    #Generate Rarity
    itemGenRarity = CreateEquipmentRarity(phenny, parameters, itemCode[0])

    #Generate Item
    itemGenEquipment = CreateEquipmentName(phenny, parameters, itemCode[1])
    itemGenEquipment['FabledPrefix'] = ''

    #Generate Enchantment
    itemGenEnchantment = CreateEquipmentEnchantment(phenny, parameters, itemCode[2], int(itemGenRarity["ID"]))

    #Generate Element
    itemGenElement = CreateEquipmentElement(phenny, parameters, itemCode[3], int(itemGenRarity["ID"]))

    #Attempt creating an Artifact
    itemGenEquipment = CreateEquipmentArtifact(phenny, parameters, itemGenEquipment, itemCode[4])

    strRaritySpacing = "" if len(itemGenRarity['RarityName']) == 0 else " "

    

    itemGen={}
    itemGen['RarityColor'] = "\x03"+itemGenRarity['RarityColor']
    itemGen['EquipmentName'] = (itemGenEquipment['FabledPrefix'] +
                                itemGenRarity['RarityName'] +
                                strRaritySpacing + 
                                ("" if itemGenElement == {} else itemGenElement['EffectName'] + " ") +
                                itemGenEquipment['EquipmentName'] + " " +
                                ("" if itemGenEnchantment == {} else itemGenEnchantment['Name'] + " "))

    itemGen['EquipmentSlot'] = itemGenEquipment['EquipmentSlot']
    itemGen['EquipmentType'] = itemGenEquipment['EquipmentType']
    itemGen['HandsReq'] = itemGenEquipment['HandsReq']
    itemGen['Strength'] = itemGenEquipment['Strength']
    itemGen['Dexterity'] = itemGenEquipment['Dexterity']
    itemGen['Constitution'] = itemGenEquipment['Constitution']
    itemGen['Intelligence'] = itemGenEquipment['Intelligence']
    itemGen['Dodge'] = itemGenEquipment['Dodge']
    itemGen['Crit'] = itemGenEquipment['Crit']
    itemGen['Armor'] = itemGenEquipment['Armor']
    itemGen['Precision']  = itemGenEquipment['Precision']
    itemGen['Resistance'] = itemGenEquipment['Resistance']
    itemGen['Luck'] = itemGenEquipment['Luck']
    itemGen['Block'] = itemGenEquipment['Block']
    itemGen['Permanent'] = itemGenEquipment['Permanent']

    itemGen['Cost'] = (itemGenEquipment['Cost'] + 
                        (0 if itemGenElement == {} else itemGenElement['Cost']) +
                        (0 if itemGenEnchantment == {} else itemGenEnchantment['Cost']) +
                        (0 if itemGenRarity == {} else itemGenRarity['RarityCost']))
    
    return itemGen

def CreateEquipmentName(phenny, parameters, itemCode):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)
    itemGenEquipment = {}


    #Item by ID
    if int(itemCode) is not 0:
        c.execute("SELECT `EquipmentName`,`EquipmentSlot`,`EquipmentType`,`HandsReq`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`,`Crit`,`Armor`,`Precision`,`Resistance`,`Luck`,`Block` FROM `Items_Master` WHERE `ID` = " + str(itemCode))
        itemGenEquipment = c.fetchone()

    #Item by Random
    elif int(itemCode) is 0:
        #UPDATE: Should be able to generate an available weapon with just one query
        # Pick random item that is available to the player
        c.execute("SELECT `Items_PlayerPool`.`Nick`, `Items_PlayerPool`.`EquipmentName`, `Items_Master`.`Locked` FROM `Items_PlayerPool` INNER JOIN `Items_Master` ON `Items_PlayerPool`.`EquipmentName` = `Items_Master`.`EquipmentName` WHERE `Nick`=%s AND `Locked`=0 ORDER BY RAND() LIMIT 1",parameters['Nick'])
        sqlItemPlayer = c.fetchone()

        # Get information from randomly picked weapon
        c.execute("SELECT `EquipmentName`, `EquipmentSlot`, `EquipmentType`, `HandsReq`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor`, `Precision`, `Resistance`, `Luck`, `Block` FROM `Items_Master` WHERE `EquipmentName`=%s",sqlItemPlayer['EquipmentName'])
        itemGenEquipment = c.fetchone()

    itemGenEquipment['Permanent'] = 0
    itemGenEquipment['Cost'] = 100
    return itemGenEquipment

def CreateEquipmentElement(phenny, parameters, itemCode, RarityID):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)

    if random.randint(1, 3) is 1 or RarityID <= 3:
        itemGenElement = {}
    else:
        if int(itemCode) is not 0:
            c.execute("SELECT `EffectName`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor` FROM `GenElement` WHERE `ID` = %s",itemCode)
            itemGenElement = c.fetchone()
        elif int(itemCode) is 0:
            c.execute("SELECT `EffectName`, `Strength`, `Dexterity`, `Constitution`, `Intelligence`, `Dodge`, `Crit`, `Armor` FROM `GenElement` ORDER BY RAND() LIMIT 1")
            itemGenElement = c.fetchone()

        itemGenElement['Cost'] = 200
    return itemGenElement

def CreateEquipmentRarity(phenny, parameters, itemCode):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)
    itemGenRarity = {}
    if int(itemCode) is not 0:
        c.execute("SELECT `ID`,`RarityName`,`RarityColor`,`RarityModifier`,`RarityChance`,`RarityCost` FROM `GenRarity` WHERE `ID` = " + str(itemCode))
        itemGenRarity = c.fetchone()
    elif int(itemCode) is 0:
        # Obtain sum of all rarities
        c.execute("SELECT SUM(`RarityChance`) AS RarityTotal FROM `GenRarity`")
        intRarityTotal = c.fetchone()['RarityTotal']

        # Variable declaration
        intIteration = 0
        intRandom = random.randint(1, intRarityTotal)

        while True:
            intIteration = intIteration + 1

            # Obtain Rarity in order of lowest to highest with each iteration
            c.execute("SELECT `ID`,`RarityName`,`RarityColor`,`RarityModifier`,`RarityChance`,`RarityCost` FROM `GenRarity` WHERE `ID` = " + str(intIteration))
            itemGenRarity = c.fetchone()

            # Subtract the current iterations rarity from the random roll
            # If intRandom becomes negative, our item rarity has been found. End the loop.
            intRandom = intRandom - itemGenRarity['RarityChance']
            if intRandom <= 0:
                break
    return itemGenRarity
    db.close()

def CreateEquipmentEnchantment(phenny, parameters, itemCode, RarityID):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)

    if random.randint(1, 3) is 1 or RarityID <= 2:
        itemGenEnchantment = {}
    else:
        if int(itemCode) is not 0:
            c.execute("SELECT `Name`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`, `Crit`, `Armor` FROM `GenEnchantment` WHERE `ID` = %s", str(itemCode))
            itemGenEnchantment = c.fetchone()
        elif int(itemCode) is 0:
            c.execute("SELECT `Name`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`, `Crit`, `Armor` FROM `GenEnchantment` ORDER BY RAND() LIMIT 1")
            itemGenEnchantment = c.fetchone()

        itemGenEnchantment['Cost'] = 200
    return itemGenEnchantment

def CreateEquipmentArtifact(phenny, parameters, itemGenEquipment, itemCode):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)

    # Gen for Artifact item:
    # 0 - Will attempt to add an artifact randomly and not exceed the standard limit
    if random.randint(1, 100) == 1 and int(itemCode) == 0:
        if itemGenEquipment['EquipmentType'] == 'Armor' or itemGenEquipment['EquipmentType'] == 'Weapon':
            c.execute("SELECT COUNT(`Permanent`) AS `Artifacts` FROM `Items_PlayerInventory` WHERE `Nick`=%s AND `Permanent`=1;",parameters['Nick'])
            sqlCountArtifacts = c.fetchone()
            if sqlCountArtifacts['Artifacts'] < 2:
                itemGenEquipment['FabledPrefix'] = 'Fabled '
                itemGenEquipment['Permanent'] = 1
    # 1 - Will not generate item as an artifact
    elif int(itemCode) == 1:
        itemGenEquipment['FabledPrefix'] = ''
        
    # 2 - Will force the item to become an artifact and exceed the limit of 2 artifacts
    elif int(itemCode) == 2:
        if itemGenEquipment['EquipmentType'] == 'Armor' or itemGenEquipment['EquipmentType'] == 'Weapon':
            c.execute("SELECT COUNT(`Permanent`) AS `Artifacts` FROM `Items_PlayerInventory` WHERE `Nick`=%s AND `Permanent`=1;",parameters['Nick'])
            sqlCountArtifacts = c.fetchone()
            itemGenEquipment['FabledPrefix'] = 'Fabled '
            itemGenEquipment['Permanent'] = 1
    return itemGenEquipment

def CreateItem(phenny, enemyStats, playerStats, parameters):
    import random
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)

    # Get random element
    c.execute("SELECT `LootName`,`ItemType`,`Potency` FROM `GenLoot` WHERE `Locked`=0 ORDER BY RAND() LIMIT 1")
    sqlRarity = c.fetchone()
    strLootName = sqlRarity['LootName']
    strItemType = sqlRarity['ItemType']
    intPotency = sqlRarity['Potency']

    if strItemType == 'Quantity':
        intPotency = enemyStats['GoldCarried'] + int(
            random.uniform(float(enemyStats['GoldCarried']) * -0.3, float(enemyStats['GoldCarried']) * 0.3))

    db.close()
    return strLootName, intPotency

def AddToShop(phenny, parameters, item):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)
    c.execute("""INSERT INTO `PlayerShop` (`Nick`,`Cost`,`Equipped`,`EquipmentColor`,`EquipmentName`,`EquipmentSlot`,`EquipmentType`,`Permanent`,`HandsReq`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`,`Crit`,`Armor`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s);""", (parameters['Nick'], item['Cost'], 0, item['RarityColor'], item['EquipmentName'], item['EquipmentSlot'], item['EquipmentType'],item['Permanent'], item['HandsReq'], item['Strength'], item['Dexterity'], item['Constitution'], item['Intelligence'], item['Dodge'], item['Crit'], item['Armor']));db.commit();
    return

def AddToInventory(phenny, parameters, item):
    db = MySQLdb.connect(host="localhost", user=phenny.config.mysql_username, passwd=phenny.config.mysql_password,db=parameters['LiveDB'])
    c = db.cursor(MySQLdb.cursors.DictCursor)
    c.execute("""INSERT INTO `Items_PlayerInventory` (`EquipmentColor`,`EquipmentName`,`EquipmentSlot`,`EquipmentType`,`Permanent`,`HandsReq`,`Equipped`,`Strength`,`Dexterity`,`Constitution`,`Intelligence`,`Dodge`,`Crit`,`Armor`,`Nick`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s);""", (item['RarityColor'], item['EquipmentName'], item['EquipmentSlot'], item['EquipmentType'],item['Permanent'], item['HandsReq'], 0, item['Strength'], item['Dexterity'], item['Constitution'], item['Intelligence'], item['Dodge'], item['Crit'], item['Armor'],parameters['Nick']));db.commit();
    return