import time, threading, pandas, numpy

from api.kraken import Kraken
from api.fake import Fake

if __name__ == "__main__":
    currency_buy = "XXBT"
    currency_sell = "ZUSD"
    currency_pair = "XXBTZUSD"

    k = Kraken()
    print("1")

    time.sleep(1)

    k.getPrices(currency_pair)
    print("3")

    time.sleep(1)

    while True:
        #print("Main loop")
        current_price = numpy.float64(k.krakenWrapper.get_ticker_information(currency_pair)["c"][0][0])
        new_row = pandas.DataFrame([[current_price]], index=[pandas.to_datetime(time.time(), unit = "s")])

        k.prices = pandas.concat([k.prices, pandas.DataFrame(new_row)], ignore_index=False)
        k.calculateEMA()

        # If we have crypto, check for sell behaviour, if we have fiat, check for buy behavior, otherwise, get really confused
        if k.balances["vol"][currency_buy] > 10**-5:
            # Sell shit
            # If short term change dips below long term change, sell
            if float(k.shortEMA.tail(1)[0]) < float(k.longEMA.tail(1)[0]) and abs(float(k.shortEMA.tail(1)[0])-float(k.longEMA.tail(1)[0]))/((float(k.shortEMA.tail(1)[0])+float(k.longEMA.tail(1)[0]))/2) > 0.001:
                print("Selling shit")
                k.sellOrder(currency_pair, currency_buy, currency_sell, k.balances["vol"][currency_buy])
                print(k.balances)

        elif k.balances["vol"][currency_sell] > 10**-2:
            # Buy shit
            # If short term goes above long term, buy
            if float(k.shortEMA.tail(1)[0]) > float(k.longEMA.tail(1)[0]) and abs(float(k.shortEMA.tail(1)[0])-float(k.longEMA.tail(1)[0]))/((float(k.shortEMA.tail(1)[0])+float(k.longEMA.tail(1)[0]))/2) > 0.001:
                print("Buying shit")
                k.buyOrder(currency_pair, currency_buy, currency_sell, k.balances["vol"][currency_sell] * 0.95 * (1/k.prices.tail(1)[0][0]))
                print(k.balances)

        else:
            raise Exception("There is no currency in this account.")

        time.sleep(3)