pacCoinDiscordTipBot.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import subprocess
import requests
import logging
import discord, asyncio
from discord.ext import commands
import re
import time as time_util
from datetime import datetime
from pytz import timezone

''' begin configuration section '''

BOTNAME  = '____CHANGEME____'
BOTUUID  = '____CHANGEME____'
BOTCHID  = '____CHANGEME____'
TOKEN    = '____CHANGEME____'
LOG_FILE = '____CHANGEME____'
APP_EXE  = '____CHANGEME____'

''' end configuration section '''



''' DO NOT EDIT BELLOW THS LINE '''

fh = logging.FileHandler(LOG_FILE)
fh.setLevel(logging.INFO)
ft = logging.Formatter('%(asctime)-15s - %(message)s')
fh.setFormatter(ft)

logger = logging.getLogger(BOTNAME)
logger.setLevel(logging.INFO)
logger.addHandler(fh)

bot = commands.Bot(command_prefix='!') 
bot.remove_command('help')


def main():
    bot.run(TOKEN)


@bot.command(pass_context=True)
async def help(ctx):
    text = ctx.message.content
    name = ctx.message.author.name
    uuid = ctx.message.author.id

    n="The following commands are at your disposal:"
    v="!info, !balance, !deposit, !withdraw, !tip, !rain, !price, and !time"
    msg = discord.Embed(color=0x00b3b3)
    msg.add_field(name=n, value=v, inline=False)

    await bot.say(embed=msg)


@bot.command(pass_context=True)
async def info(ctx):
    uuid = ctx.message.author.id
    name = ctx.message.author.name

    msg = \
    """
      ```
                  INFO SECTION
      ```
      commands like *!tip* & *!withdraw* have a specfic format,\
    use them like so:
     
     Tipping format: 
       `!tip @[user] [amount]        (without brackets)`
       
     Rain format: 
       `!rain [amount]               (without brackets)`
          
     Withdrawing format: 
       `!withdraw [address] [amount] (without brackets)`


        WHERE:
            `[address] = withdraw #$PAC address`
            `[user] = discord username`
            `[amount] = amount of #$PAC to utilise`

     *NOTE*:
      - don't deposit a significant amount of #$PAC through this #BOT
      - make sure that you enter a valid #$PAC address when you perform a withdraw
      - we are not responsible of your funds if something bad happen to this #BOT 
     ```
          USE THIS #BOT AT YOUR OWN RISK
     ```

    """

    embed = discord.Embed(color=0x00b3b3)
    embed.add_field(name="\a", value=msg, inline=False)

    await bot.say(embed=embed)


@bot.command(pass_context=True)
async def balance(ctx):
    user_name = ctx.message.author.name
    user_uuid = ctx.message.author.id

    if user_name is None:
        msg = "Invalid username!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if user_uuid is None:
        msg = "Invalid userid!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    cmd = [
        APP_EXE, 
        '-rpcuser=bitcoin',
        '-rpcpassword=local321', 
        'getbalance',
        str(user_uuid)
    ]
    ret = rpc_call(cmd)
    if ret is None:
        msg = 'rpc internal error!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    balance = float(ret)
    balance = str('{:,.8f}'.format(balance))

    msg = '@{0} your current balance is: {1} $PAC'.format(user_name, balance)
    embed = discord.Embed(color=discord.Color.green())
    embed.add_field(name="BALANCE", value=msg, inline=True)

    await bot.send_message(ctx.message.author, embed=embed)
    #await bot.say(embed=embed)


@bot.command(pass_context=True)
async def deposit(ctx):
    user_name = ctx.message.author.name
    user_uuid = ctx.message.author.id

    if user_name is None:
        msg = "Invalid username!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if user_uuid is None:
        msg = "Invalid userid!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    cmd = [
        APP_EXE, 
        '-rpcuser=bitcoin',
        '-rpcpassword=local321', 
        'getaccountaddress',
        str(user_uuid)
    ] 
    ret = rpc_call(cmd)
    if ret is None:
        msg = 'rpc internal error!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    msg = '@{} your depositing address is: `{}`'.format(user_name, ret)
    embed = discord.Embed(color=discord.Color.green())
    embed.add_field(name="DEPOSIT", value=msg, inline=True)

    await bot.send_message(ctx.message.author, embed=embed)
    # await bot.say(embed=embed)


@bot.command(pass_context=True)
async def tip(ctx):
    user_name = ctx.message.author.name
    user_uuid = ctx.message.author.id

    if user_name is None:
        msg = "Invalid username!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if user_uuid is None:
        msg = "Invalid userid!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    
    message = ctx.message.content.split(' ')
    if len(message) != 3:
        msg = "Please use !tip <username> <amount>!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    if not isValidUsername(message[1]):
        msg = "Please input a valid username (ex: @JonDoe01#0964)!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if not isValidAmount(message[2]):
        msg = "Please input a valid amount (ex: 1000)!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    amount = float(message[2])
    target = message[1]
    uuid = list(filter(str.isdigit, target))
    target_uuid = str(''.join(uuid))

    if amount > 100000 or amount < 1:
        msg = "Please send value between 1 and 100,000 $PAC!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if target_uuid == user_uuid:
        msg = "You can't tip yourself silly!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if target_uuid == BOTUUID:
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value="HODL.", inline=True)
        await bot.say(embed=embed)
        return False
    if not target_uuid:
        msg = '@{} has no activity in this chat!'.format(target)
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    cmd = [
        APP_EXE, 
        '-rpcuser=bitcoin',
        '-rpcpassword=local321', 
        'getbalance',
        str(user_uuid)
    ]
    ret = rpc_call(cmd)
    if ret is None:
        msg = 'rpc internal error!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    balance = float(ret)

    if balance < amount:
        msg = '@{0} you have insufficent funds.'.format(user_name)
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    cmd = [
        APP_EXE,
        '-rpcuser=bitcoin',
        '-rpcpassword=local321',
        'move',
        str(user_uuid),
        str(target_uuid),
        str(amount),
    ]
    tx = rpc_call(cmd)
    if tx is None:
        msg = 'rpc internal error!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    msg = '@{0} tipped {1} of {2} $PAC'.format(user_name, target, amount)
    embed = discord.Embed(color=discord.Color.green())
    embed.add_field(name="TIP", value=msg, inline=True)

    await bot.say(embed=embed)

    
@bot.command(pass_context=True)
async def rain(ctx):
    user_name = ctx.message.author.name
    user_uuid = ctx.message.author.id

    if user_name is None:
        msg = "Invalid username!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if user_uuid is None:
        msg = "Invalid userid!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    
    message = ctx.message.content.split(' ')
    if len(message) != 2:
        msg = "Please use !rain <amount>!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if not isValidAmount(message[1]):
        msg = "Please input a valid amount (ex: 1000)!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    amount = float(message[1])

    if amount > 100000 or amount < 1:
        msg = "Please send value between 1 and 100,000 $PAC!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    cmd = [
        APP_EXE, 
        '-rpcuser=bitcoin',
        '-rpcpassword=local321', 
        'getbalance',
        str(user_uuid)
    ]
    ret = rpc_call(cmd)
    if ret is None:
        msg = 'rpc internal error!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    balance = float(ret)

    if balance < amount:
        msg = '@{0} you have insufficent funds.'.format(user_name)
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    users_online = {}
    for server in bot.servers:
      for u in server.members:
        if str(u.status) is 'online':
          users_online[u.id] = u.name

    online  = len(users_online)
    pamount = '{:.8f}'.format(float(amount/online))

    for key in sorted(users_online):
      time_util.sleep(0.1)
      target_uuid = key
      target_name = users_online[target_uuid]
      cmd = [
        APP_EXE,
        '-rpcuser=bitcoin',
        '-rpcpassword=local321',
        'move',
        str(user_uuid),
        str(target_uuid),
        str(pamount),
      ]
      tx = rpc_call(cmd)
      if tx is None:
          msg = "failed to #tip @{}!".format(target_name)
          embed = discord.Embed(color=discord.Color.red())
          embed.add_field(name="ERROR", value=msg, inline=True)
          await bot.say(embed=embed)
          return False

    sub_list = list(users_online.values())
    user_list = ",".join(sub_list[0:2])
    other_list = online - 50;
    if online > 50:
        msg = "{} invoked rain spell with {} $PAC over #{} 50 users ({}) and {} other people"\
              .format(user_name, pamount, online, user_list, other_list)
    else:
       msg = "{} invoked rain spell with {} $PAC over #{} users ({})"\
             .format(user_name, pamount, online, user_list)
    embed = discord.Embed(color=discord.Color.green())
    embed.add_field(name="RAIN", value=msg, inline=True)

    await bot.say(embed=embed)


@bot.command(pass_context=True)
async def withdraw(ctx):
    user_name = ctx.message.author.name
    user_uuid = ctx.message.author.id

    if user_name is None:
        msg = "Invalid username!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if user_uuid is None:
        msg = "Invalid userid!"
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    message = ctx.message.content.split(' ')
    if len(message) != 3:
        msg = 'Please use !withdraw <address> <amount>!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    if not isValidAddress(message[1]):
        msg = 'Please input a valid $PAC address!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False
    if not isValidAmount(message[2]):
        msg = 'Please input a valid amount (ex: 1000)!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    amount = float(message[2])
    address = message[1]

    cmd = [
        APP_EXE, 
        '-rpcuser=bitcoin',
        '-rpcpassword=local321', 
        'getbalance' ,
        str(user_uuid)
    ]
    ret = rpc_call(cmd)
    if ret is None:
        msg = 'rpc internal error!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    balance = float(ret)
    if balance < amount+1:
        msg = '@{0} you have insufficent funds.'.format(user_name)
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    cmd = [
        APP_EXE,
        '-rpcuser=bitcoin',
        '-rpcpassword=local321',
        'sendfrom',
        str(user_uuid),
        str(address),
        str(amount),
    ]
    tx = rpc_call(cmd)
    if tx is None:
        msg = 'rpc internal error!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    msg = '@{0} has successfully withdrew {2} $PAC to address: {1}'\
              .format(user_name, address, amount)
    embed = discord.Embed(color=discord.Color.green())
    embed.add_field(name="WITHDRAW", value=msg, inline=True)

    await bot.say(embed=embed)


def yobit(market):
    getmarket = 'https://yobit.net/api/3/ticker/pac_'+market.lower()
    response  = requests.get(getmarket)
    json_data = response.json()

    if not json_data or not json_data.get('pac_btc') or json_data.get('error'):
      return {}

    json_data = json_data.get('pac_btc')
    message = {}

    message['*LastPrice*'] = '{:,.8f}'.format(float(json_data['last']))
    message['*AskPrice*'] = '{:,.8f}'.format(float(json_data['sell']))
    message['*BidPrice*'] = '{:,.8f}'.format(float(json_data['buy']))
    message['*Volume*'] = '{:,.8f}'.format(float(json_data['vol_cur']))
    message['*Volume*'] = '{:,.8f}'.format(float(json_data['vol']))

    return message


def cryptopia(market):
    getmarket = 'https://www.cryptopia.co.nz/api/GetMarket/$PAC_'+market
    response  = requests.get(getmarket)
    json_data = response.json()

    if not json_data or json_data.get('error'):
      return {}
    if not json_data['Data']: 
      return {}

    message = {}

    message['*LastPrice*'] = '{:,.8f}'.format(json_data['Data']['LastPrice'])
    message['*AskPrice*'] = '{:,.8f}'.format(json_data['Data']['AskPrice'])
    message['*BidPrice*'] = '{:,.8f}'.format(json_data['Data']['BidPrice'])
    message['*Volume*']   = '{:,.8f}'.format(json_data['Data']['BaseVolume'])

    return message

def tradesatoshi(market):
    getmarket = 'https://tradesatoshi.com/api/public/GetTicker?market=$PAC_'+market
    response  = requests.get(getmarket)
    json_data = response.json()

    if not json_data or not json_data.get('result') or json_data.get('error'):
      return {}

    json_data = json_data.get('result')
    message = {}

    message['*LastPrice*'] = '{:,.8f}'.format(float(json_data['last']))
    message['*AskPrice*'] = '{:,.8f}'.format(float(json_data['ask']))
    message['*BidPrice*'] = '{:,.8f}'.format(float(json_data['bid']))

    return message

@bot.command(pass_context=True)
async def price(ctx):
    markets = ('BTC', 'LTC', 'BCH', 'DOGE', 'USDT')

    message = ctx.message.content.split(' ')
    if len(message) != 2:
        msg = 'Please use !price [btc/ltc/bch/doge/usdt]!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    market = message[1].upper()
    if market not in markets:
        msg = 'This market is not available!'
        embed = discord.Embed(color=discord.Color.red())
        embed.add_field(name="ERROR", value=msg, inline=True)
        await bot.say(embed=embed)
        return False

    message = {
      "yobit": yobit(market),
      "cryptopia": cryptopia(market),
      "tradesatoshi": tradesatoshi(market)
    }

    msg = json.dumps(message, indent=4, sort_keys=True)

    embed = discord.Embed(color=0x00b3b3)
    embed.add_field(name="PRICE", value=msg, inline=True)

    await bot.say(embed=embed)



@bot.command(pass_context=True)
async def time(ctx):

    msg = datetime.utcnow().strftime("%a %b %d %H:%M:%S %Y") + " UTC\n"
    msg += datetime.now(timezone('EST')).strftime("%a %b %d %H:%M:%S %Y %Z")

    embed = discord.Embed(color=0x00b3b3)
    embed.add_field(name="TIME", value=msg, inline=True)

    await bot.say(embed=embed)


@bot.command(pass_context=True)
async def mcap(ctx):
    user_name = ctx.message.author.name
    user_uuid = ctx.message.author.id

    await bot.say('work in progress!')


@bot.command(pass_context=True)
async def moon(ctx):
    user_name = ctx.message.author.name
    user_uuid = ctx.message.author.id

    await bot.say('Moon mission inbound!')


@bot.command(pass_context=True)
async def hi(ctx):
    name = ctx.message.author.name
    uuid = ctx.message.author.id
    
    msg = 'Hello @{}[{}], how are you doing today?'.format(name, uuid)

    await bot.say(msg)


@bot.event
async def on_ready():
    print('Bot is ready for use!')


@bot.event
async def on_message(message, user: discord.Member = None):
    if message is None: return

    if user is None:
      user = message.author

    msg  = message.content
    uuid = user.id
    user = user.name

    logger.info("@{} [#{}]: {}".format(user, uuid, msg))

    cuid = str(message.channel.id)
    if cuid != BOTCHID and message.content.startswith('!'):
      logger.info("wrong #BOTCHID [@{}]".format(BOTCHID))
      await bot.delete_message(message)
    else:
      await bot.process_commands(message)



def rpc_call(cmd):
    try:
        result = subprocess.run(cmd,
            stdout=subprocess.PIPE)
    except:
        return
    ret = result.stdout.strip()
    ret = ret.decode('utf-8')
    if len(ret) == 0:
        return
    else:
        return ret


def isValidAddress(param):
    if len(param) < 30:
        return False
    elif len(param) > 35:
        return False
    elif not param.isalnum():
        return False
    elif not param[0] == 'P':
        return False
    else:
        return True


def isValidUsername(user):
    if re.match('^<@\!?[0-9]+>$', user):
        return True
    else:
        return False


def isValidAmount(amount):
    try:
        float(amount)
    except ValueError:
    	return False
    else:
        return True


if __name__ == '__main__':
    main()