#!/bin/python
#
# Cryptonote tipbot - bet utils
# Copyright 2015 moneromooo
#
# The Cryptonote tipbot 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, or (at your option)
# any later version.
#

import random
import hashlib
import time
import datetime
from Crypto.Random.random import getrandbits
import tipbot.coinspecs as coinspecs
from tipbot.command_manager import *
from utils import *

def IsBanned(link):
  try:
    banned = redis_hget('banned',link.identity())
    if not banned:
      return False, None
    banned = float(banned)
    now = time.time()
    if banned < now:
      redis_hdel('banned',link.identity())
      return False, None
    return True, 'You are banned for %s' % TimeToString(banned-now)
  except Exception,e:
    log_error('Failed to check bannishment for %s: %s' % (link.identity(),str(e)))
    return False, None

def IsBetValid(link,amount,minbet,maxbet,potential_loss,max_loss,max_loss_ratio):
  banned,reason = IsBanned(link)
  if banned:
    return False, reason
  try:
    units = StringToUnits(amount)
    amount = float(amount)
  except Exception,e:
    return False, "Invalid amount"
  if units <= 0:
    return False, "Invalid amount"
  if maxbet != None and amount > maxbet:
    return False, "Max bet is %s" % AmountToString(maxbet * coinspecs.atomic_units)
  if minbet != None and amount < minbet:
    return False, "Min bet is %s" % AmountToString(minbet * coinspecs.atomic_units)

  enough, reason = IsPlayerBalanceAtLeast(link,units)
  if not enough:
    return False, reason

  if potential_loss > 0:
    if potential_loss > max_loss:
      return False, "Max potential loss is %s" % AmountToString(max_loss * coinspecs.atomic_units)
    try:
      house_balance = RetrieveHouseBalance()
    except Exception,e:
      log_error('Failed to get house balance: %s' % str(e))
      return False, "Failed to get house balance"
    max_floating_loss = max_loss_ratio * house_balance / coinspecs.atomic_units
    if potential_loss > max_floating_loss:
      return False, "Potential loss too large for house balance"

  return True, None

def IsPlayerBalanceAtLeast(link,units):
  try:
    account = GetAccount(link)
    balance = redis_hget("balances",account)
    if balance == None:
      balance = 0
    balance=long(balance)
    if units > balance:
      log_info ('%s does not have enough balance' % link.identity())
      return False, "You only have %s" % (AmountToString(balance))
  except Exception,e:
    log_error ('failed to query balance')
    return False, "Failed to query balance"
  return True, None

def SetServerSeed(link,game,seed):
  identity=link.identity()
  try:
    redis_hset('%s:serverseed' % game,identity,seed)
    log_info('%s\'s %s serverseed set' % (identity, game))
  except Exception,e:
    log_error('Failed to set %s server seed for %s: %s' % (game, identity, str(e)))
    raise

def GetServerSeed(link,game):
  EnsureServerSeed(link,game)
  identity=link.identity()
  try:
    return redis_hget('%s:serverseed' % game,identity)
  except Exception,e:
    log_error('Failed to get %s server seed for %s: %s' % (game, identity, str(e)))
    raise

def GenerateServerSeed(link,game):
  identity=link.identity()
  try:
    salt="kfn3kjg4nkngvekjvn3u4vgb:" + config.site_game_salt + ":" + game
    s=salt+":"+identity+":"+str(time.time())+":"+str(getrandbits(128))
    seed=hashlib.sha256(s).hexdigest()
    SetServerSeed(link,game,seed)
  except Exception,e:
    log_error('Failed to generate %s server seed for %s: %s' % (game,identity,str(e)))
    raise

def EnsureServerSeed(link,game):
  if not redis_hexists('%s:serverseed' % game,link.identity()):
    GenerateServerSeed(link,game)

def SetPlayerSeed(link,game,seed):
  identity=link.identity()
  try:
    redis_hset('%s:playerseed' % game,identity,seed)
    log_info('%s\'s %s playerseed set to %s' % (identity, game, seed))
  except Exception,e:
    log_error('Failed to set %s player seed for %s: %s' % (game, identity, str(e)))
    raise

def GetPlayerSeed(link,game):
  identity=link.identity()
  try:
    if not redis_hexists('%s:playerseed' % game,identity):
      return ""
    return redis_hget('%s:playerseed' % game,identity)
  except Exception,e:
    log_error('Failed to get %s player seed for %s: %s' % (game, identity, str(e)))
    raise

def GetServerSeedHash(link,game):
  seed = GetServerSeed(link,game)
  return hashlib.sha256(seed).hexdigest()

def RecordGameResult(link,game,win,lose,units):
  identity=link.identity()
  account=GetAccount(identity)
  try:
    ts=datetime.datetime.utcnow()
    tsh="%u" % (ts.hour)
    tsd="%u-%02u-%02u" % (ts.year,ts.month,ts.day)
    p = redis_pipeline()
    tname="%s:stats:"%game+identity
    rtname="%s:stats:reset:"%game+identity
    alltname="%s:stats:"%game
    nalltname="%s:stats:%s:"%(game,link.network.name)
    zhtname="%s:zstats:hourly:"%game
    zdtname="%s:zstats:daily:"%game
    nzhtname="%s:zstats:%s:hourly:"%(game,link.network.name)
    nzdtname="%s:zstats:%s:daily:"%(game,link.network.name)
    p.hincrby(tname,"bets",1)
    p.hincrby(rtname,"bets",1)
    p.hincrby(alltname,"bets",1)
    p.hincrby(nalltname,"bets",1)
    p.zincrby(zhtname+"bets",tsh,1)
    p.zincrby(zdtname+"bets",tsd,1)
    p.zincrby(nzhtname+"bets",tsh,1)
    p.zincrby(nzdtname+"bets",tsd,1)
    p.hincrby(tname,"wagered",units)
    p.hincrby(rtname,"wagered",units)
    p.hincrby(alltname,"wagered",units)
    p.hincrby(nalltname,"wagered",units)
    p.zincrby(zhtname+"wagered",tsh,units)
    p.zincrby(zdtname+"wagered",tsd,units)
    p.zincrby(nzhtname+"wagered",tsh,units)
    p.zincrby(nzdtname+"wagered",tsd,units)
    if win:
      p.hincrby("balances",account,units)
      p.hincrby(tname,"won",units)
      p.hincrby(rtname,"won",units)
      p.hincrby(alltname,"won",units)
      p.hincrby(nalltname,"won",units)
      p.zincrby(zhtname+"won",tsh,units)
      p.zincrby(zdtname+"won",tsd,units)
      p.zincrby(nzhtname+"won",tsh,units)
      p.zincrby(nzdtname+"won",tsd,units)
      p.hincrby(tname,"nwon",1)
      p.hincrby(rtname,"nwon",1)
      p.hincrby(alltname,"nwon",1)
      p.hincrby(nalltname,"nwon",1)
      p.zincrby(zhtname+"nwon",tsh,1)
      p.zincrby(zdtname+"nwon",tsd,1)
      p.zincrby(nzhtname+"nwon",tsh,1)
      p.zincrby(nzdtname+"nwon",tsd,1)
    if lose:
      p.hincrby("balances",account,-units)
      p.hincrby(tname,"lost",units)
      p.hincrby(rtname,"lost",units)
      p.hincrby(alltname,"lost",units)
      p.hincrby(nalltname,"lost",units)
      p.zincrby(zhtname+"lost",tsh,units)
      p.zincrby(zdtname+"lost",tsd,units)
      p.zincrby(nzhtname+"lost",tsh,units)
      p.zincrby(nzdtname+"lost",tsd,units)
      p.hincrby(tname,"nlost",1)
      p.hincrby(rtname,"nlost",1)
      p.hincrby(alltname,"nlost",1)
      p.hincrby(nalltname,"nlost",1)
      p.zincrby(zhtname+"nlost",tsh,1)
      p.zincrby(zdtname+"nlost",tsd,1)
      p.zincrby(nzhtname+"nlost",tsh,1)
      p.zincrby(nzdtname+"nlost",tsd,1)
    p.execute()
  except Exception,e:
    log_error('RecordGameResult: exception updating redis: %s' % str(e))
    raise

def ShowGameStats(link,sidentity,title,game):
  tname="%s:stats:"%game+sidentity
  try:
    bets=redis_hget(tname,"bets")
    wagered=redis_hget(tname,"wagered")
    won=redis_hget(tname,"won")
    lost=redis_hget(tname,"lost")
    nwon=redis_hget(tname,"nwon")
    nlost=redis_hget(tname,"nlost")
  except Exception,e:
    log_error('Failed to retrieve %s stats for %s: %s' % (game, title, str(e)))
    link.send("An error occured")
    return
  if not bets:
    link.send("No %s stats available for %s" % (game,title))
    return

  bets = long(bets)
  wagered = long(wagered)
  won = long(won or 0)
  lost = long(lost or 0)
  nwon = long(nwon or 0)
  nlost = long(nlost or 0)

  if bets==0:
    link.send("No %s stats available for %s" % (game,title))
    return

  swagered = AmountToString(wagered)
  savg = AmountToString(wagered / bets)
  swon = AmountToString(won)
  slost = AmountToString(lost)
  if won >= lost:
    sov = "+" + AmountToString(won-lost)
  else:
    sov = "-" + AmountToString(lost-won)
  link.send("%s: %d games %d won, %d lost, %s wagered (average %s per game), %s won, %s lost, overall %s" % (title, bets, nwon, nlost, swagered, savg, swon, slost, sov))

def ResetGameStats(link,sidentity,game):
  identity=IdentityFromString(link,sidentity)
  try:
    p = redis_pipeline()
    tname="%s:stats:reset:"%game+sidentity
    bets=p.hset(tname,"bets",0)
    wagered=p.hset(tname,"wagered",0)
    won=p.hset(tname,"won",0)
    lost=p.hset(tname,"lost",0)
    nwon=p.hset(tname,"nwon",0)
    nlost=p.hset(tname,"nlost",0)
    p.execute()
    link.send("%s stats reset for %s" % (game,sidentity))
  except Exception,e:
    log_error('Error resetting %s stats for %s: %s' % (game,sidentity,str(e)))
    raise

def RetrieveHouseBalance(force_refresh=False):
  balance, unlocked_balance = RetrieveTipbotBalance(force_refresh)
  house_balance = unlocked_balance

  user_balances=0
  accounts = redis_hgetall("balances")
  for account in accounts:
    ab = long(accounts[account])
    house_balance = house_balance - ab
    user_balances+=ab

  earmarked_balances=0
  earmarked = redis_hgetall("earmarked")
  for e in earmarked:
    eb = long(earmarked[e])
    house_balance = house_balance - eb
    earmarked_balances+=eb

  rbal=long(redis_get('reserve_balance') or 0)
  if rbal:
    house_balance = house_balance - rbal

  if house_balance < 0:
    raise RuntimeError('Negative house balance')
    return
  log_info('RetrieveHouseBalance: unlocked %s, users %s, earmarked %s, reserve %s, house %s' % (AmountToString(unlocked_balance), AmountToString(user_balances), AmountToString(earmarked_balances), AmountToString(rbal), AmountToString(house_balance)))
  return house_balance

def GetHouseBalance(link,cmd):
  try:
    balance = RetrieveHouseBalance()
    personal_balance=0
    seen_accounts=[]
    for network in networks:
      identity=network.name+':'+config.tipbot_name
      account=redis_hget('accounts',identity)
      if not account in seen_accounts:
        personal_balance += long(redis_hget('balances',account) or 0)
        seen_accounts.append(account)
  except Exception,e:
    log_error('Failed to retrieve house balance: %s' % str(e))
    link.send('An error occured')
    return
  link.send('House balance: %s, %s personal balance: %s' % (AmountToString(balance), config.tipbot_name, AmountToString(personal_balance)))

def ReserveBalance(link,cmd):
  rbal=GetParam(cmd,1)
  if rbal:
    try:
      rbal=StringToUnits(cmd[1])
      if rbal < 0:
        raise RuntimeError('negative balance')
    except Exception,e:
      log_error('SetReserveBalance: invalid balance: %s' % str(e))
      link.send("Invalid balance")
      return

  try:
    current_rbal=long(redis_get('reserve_balance') or 0)
  except Exception,e:
    log_error('Failed to get current reserve balance: %s' % str(e))
    link.send("Failed to get current reserve balance")
    return
  if rbal == None:
    link.send("Reserve balance is %s" % AmountToString(current_rbal))
    return

  try:
    house_balance = RetrieveHouseBalance()
  except Exception,e:
    log_error('Failed to get house balance: %s' % str(e))
    link.send("Failed to get house balance")
    return
  if rbal > house_balance + current_rbal:
    log_error('Cannot set reserve balance higher than max house balance')
    link.send('Cannot set reserve balance higher than max house balance')
    return
  try:
    redis_set('reserve_balance',rbal)
  except Exception,e:
    log_error('Failed to set reserve balance: %s' % str(e))
    link.send("Failed to set reserve balance")
    return
  link.send("Reserve balance set")

def Ban(link,cmd):
  t = 3600
  try:
    sidentity = GetParam(cmd,1)
    if sidentity:
      sidentity=IdentityFromString(link,sidentity)
      if sidentity!=link.identity() and not IsAdmin(link):
        log_error('%s is not admin, cannot ban %s' % (link.identity(),sidentity))
        link.send('Access denied')
        return
    else:
      sidentity=link.identity()

    banned = redis_hget('banned',sidentity)
    now=time.time()
    if banned and float(banned) > now+t:
      link.send('%s is already banned for %s' % (NickFromIdentity(sidentity), TimeToString(banned-now)))
    else:
      redis_hset('banned',sidentity,now+t)
      link.send('%s is banned for %s' % (NickFromIdentity(sidentity), TimeToString(t)))
  except Exception,e:
    log_error('Failed to ban %s: %s' % (sidentity,str(e)))
    link.send('An error occured')
    return

def Unban(link,cmd):
  try:
    sidentity=GetParam(cmd,1)
    if not sidentity:
      sidentity=link.identity()
    sidentity=IdentityFromString(link,sidentity)
    redis_hdel('banned',sidentity)
    link.send('%s was unbanned' % (NickFromIdentity(sidentity)))
  except Exception,e:
    log_error('Failed to unban %s: %s' % (sidentity,str(e)))
    link.send('An error occured')
    return

def Report(link,cmd):
  GetHouseBalance(link,cmd)
  games=[]
  try:
    keys=redis_keys('*:zstats:daily:*')
    for key in keys:
      game=key.split(':')[0]
      if game not in games:
        games.append(game)
  except Exception,e:
    log_error('Failed to enumerate games: %s' % str(e))
    link.send('Failed to enumerate games')
    return
  now=datetime.datetime.utcnow()
  for game in games:
    try:
      ShowGameStats(link,'',game,game)
      period={1:'yesterday',7:'last week',30:'last month'}
      zdtname="%s:zstats:daily:"%game
      bets=0
      wagered=0
      won=0
      lost=0
      for days in range(1,31):
        ts=now-datetime.timedelta(days=days)
        tsd="%u-%02u-%02u" % (ts.year,ts.month,ts.day)
        bets+=long(redis_zscore(zdtname+"bets",tsd) or 0)
        wagered+=long(redis_zscore(zdtname+"wagered",tsd) or 0)
        won+=long(redis_zscore(zdtname+"won",tsd) or 0)
        lost+=long(redis_zscore(zdtname+"lost",tsd) or 0)
        if days in period.keys():
          if won>lost:
            wonlost='lost'
            balance_change=AmountToString(won-lost)
          else:
            wonlost='won'
            balance_change=AmountToString(lost-won)
          link.send('%s, %s: %d bets, %s wagered, house %s %s' % (game,period[days],bets,AmountToString(wagered),wonlost,balance_change))
    except Exception,e:
      log_error('Failed to generate report for %s: %s' % (game,str(e)))
      link.send('Failed to generate report for %s' % game)

RegisterCommand({
  'module': 'betting',
  'name': 'reserve_balance',
  'parms': '[<amount>]',
  'function': ReserveBalance,
  'admin': True,
  'help': "Set or get reserve balance (not part of the house balance)"
})
RegisterCommand({
  'module': 'betting',
  'name': 'house_balance',
  'function': GetHouseBalance,
  'admin': True,
  'registered': True,
  'help': "get the house balance"
})
RegisterCommand({
  'module': 'betting',
  'name': 'ban',
  'function': Ban,
  'registered': True,
  'help': "ban yourself from playing for an hour"
})
RegisterCommand({
  'module': 'betting',
  'name': 'unban',
  'function': Unban,
  'admin': True,
  'help': "unban someone from playing"
})
RegisterCommand({
  'module': 'betting',
  'name': 'report',
  'function': Report,
  'admin': True,
  'help': "Betting report"
})