Maki/commands.py

# Maki
# ----
# Discord bot by MrDetonia
#
# Copyright 2017 Zac Herd
# Licensed under BSD 3-clause License, see LICENSE.md for more info



# IMPORTS
import asyncio
import os
import sys
import re
import requests
import random
import subprocess


# LOCAL IMPORTS
from common import *
from helpers import *
from secret import lfmkey, steamkey
import markov



# COMMAND IMPLEMENTATIONS
@asyncio.coroutine
def cmd_help(client, msg):
	yield from discord_send(client, msg, helptext)


@asyncio.coroutine
def cmd_info(client, msg):
	pyver = "{}.{}.{}".format(sys.version_info[0], sys.version_info[1], sys.version_info[2])
	appinfo = yield from client.application_info()
	response = "I am **{}**, a Discord bot by **{}** | `{}` | Python `{}` | discord.py `{}`".format(appinfo.name, appinfo.owner.name, version, pyver, discord.__version__)
	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_upskirt(client, msg):
	response = "No, don\'t look at my pantsu, baka! <https://gitla.in/MrDetonia/maki>"
	yield from discord_send(client, msg, response)


whoistring = "**{}#{}**: `{}`\n**Account Created:** `{}`"


@asyncio.coroutine
def cmd_whoami(client, msg):
	response = whoistring.format(msg.author.name, msg.author.discriminator, msg.author.id, strfromdt(msg.author.created_at))
	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_whois(client, msg):
	tmp = msg.content[7:]
	user = msg.server.get_member_named(tmp)

	if user == None:
		reponse = "I can't find `{}`".format(tmp)
	else:
		response = whoistring.format(user.name, user.discriminator, user.id, strfromdt(user.created_at))

	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_seen(client, msg):
	tmp = msg.content[6:]
	user = msg.server.get_member_named(tmp)

	if user == None:
		reponse = "I can't find `{}`".format(tmp)
	elif user.name == "Maki":
		reponse = "I'm right here!"
	else:
		target = msg.server.id + user.id
		if target in history and history[target][0] == msg.server.id:
			response = "**{}** was last seen saying the following at {}:\n{}".format(user.name, strfromdt(dtfromts(history[target][1])), history[target][2])
		else:
			response = "I haven't seen **{}** speak yet!".format(tmp)

	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_say(client, msg):
	print("IN CMD_SAY")
	response = msg.content[5:]
	yield from client.delete_message(msg)
	print("CALLING SEND")
	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_sayy(client, msg):
	response = " ".join(msg.content[6:])
	yield from client.delete_message(msg)
	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_markov(client, msg):
	yield from discord_typing(client, msg)

	tmp = msg.content[8:]
	target = ""

	if tmp == "Maki":
		response = "My markovs always say the same thing"
	else:
		if tmp == "":
			target = "{}-{}".format(msg.server.id, msg.author.id)
		else:
			try:
				target = "{}-{}".format(msg.server.id, msg.server.get_member_named(tmp).id)
			except AttributeError:
				reponse = "I can't find `{}`".format(tmp)

	if target != "":
		mfile = "./markovs/" + target
		if os.path.isfile(mfile):
			mc = markov.Markov(open(mfile))
			response = mc.generate_text(random.randint(20,40))
		else:
			response = "I haven't seen `{}` speak yet.".format(tmp)

	yield from discord_send(client, msg, response);


@asyncio.coroutine
def cmd_roll(client, msg):
	tmp = msg.content[6:]

	pattern = re.compile("^([0-9]+)d([0-9]+)$")

	if pattern.match(tmp):
		# extract numbers
		nums = [int(s) for s in re.findall(r"\d+", tmp)]

		# limit ranges
		nums[0] = clamp(nums[0], 1,100)
		nums[1] = clamp(nums[1], 1, 1000000)

		# roll and sum dice
		rollsum = 0
		for i in range(nums[0]):
			rollsum += random.randint(1, nums[1])

		response = "Using `{}d{}` you rolled: `{}`".format(nums[0], nums[1], rollsum)
	else:
		response = "Expected format: `<num>d<value>`"

	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_qr(client, msg):
	tmp = msg.content[4:]

	yield from discord_typing(client, msg)

	# generate qr code
	qr = subprocess.Popen("qrencode -t png -o -".split(), stdin=subprocess.PIPE, stdout=subprocess.PIPE)
	qr.stdin.write(tmp.encode("utf-8"))
	qr.stdin.close()
	out = subprocess.check_output("curl -F upload=@- https://w1r3.net".split(), stdin=qr.stdout)

	response = out.decode("utf-8").strip()

	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_np(client, msg):
	tmp = msg.content[4:]

	if tmp == "":
		response = lastfm_np(msg.author.name)
	else:
		response = lastfm_np(tmp)

	print("CALLING SEND")
	yield from discord_send(client, msg, response)


@asyncio.coroutine
def cmd_steam(client, msg):
	tmp = msg.content[7:]

	if tmp == "":
		response = steamdata(msg.author.name)
	else:
		response = steamdata(tmp)

	yield from discord_send(client, msg, response)



# HELPER FUNCTIONS

# gets now playing information from last.fm
def lastfm_np(username):
	# sanitise username
	cleanusername = re.sub(r'[^a-zA-Z0-9_-]', '', username, 0)

	# fetch JSON from last.fm
	payload = {'format': 'json', 'method': 'user.getRecentTracks', 'user': cleanusername, 'limit': '1', 'api_key': lfmkey}
	r = requests.get("http://ws.audioscrobbler.com/2.0/", params=payload)

	# read json data
	np = r.json()

	# check we got a valid response
	if 'error' in np:
		return "I couldn't get last.fm data for `{}`".format(username)

	# get fields
	try:
		username = np['recenttracks']['@attr']['user']
		track = np['recenttracks']['track'][0]
		album = track['album']['#text']
		artist = track['artist']['#text']
		song = track['name']
		nowplaying = '@attr' in track
	except IndexError:
		return "It looks like `{}` hasn't played anything recently.".format(username)

	# grammar
	if album != "":
		albumtext = "` from the album `{}`".format(album)
	else:
		albumtext = "`"

	if nowplaying == True:
		nowplaying = " is listening"
	else:
		nowplaying = " last listened"

	# construct string
	return "{}{} to `{}` by `{}{}".format(username, nowplaying, song, artist, albumtext)


# gets general steam user info from a vanityurl name
def steamdata(vanityname):
	# sanitise username
	cleanvanityname = re.sub(r'[^a-zA-Z0-9_-]', '', vanityname, 0)

	resolveurl = 'http://api.steampowered.com/ISteamUser/ResolveVanityURL/v0001/?key='
	dataurl =  'http://api.steampowered.com/ISteamUser/GetPlayerSummaries/v0002/?key='

	# fetch json from steam
	try:
		idresponse = requests.get(resolveurl + steamkey + '&vanityurl=' + vanityname).json()['response']
	except:
		return "I can't connect to Steam"

	# check if user was found and extract steamid
	if idresponse['success'] is not 1:
		return "I couldn't find `{}`".format(vanityname)
	else:
		steamid = idresponse['steamid']

	# fetch steam user info
	try:
		dataresponse = requests.get(dataurl + steamkey + '&steamids=' + steamid).json()['response']['players'][0]
	except:
		return "Can't find info on `{}`".format(vanityname)

	personastates = ['Offline', 'Online', 'Busy', 'Away', 'Snoozed', 'Looking to trade', 'Looking to play']

	if 'personaname' in dataresponse: namestr = dataresponse['personaname']
	else: namestr = ''
	if 'personastate' in dataresponse: statestr = '`' + personastates[dataresponse['personastate']] + '`'
	else: statestr = ''
	if 'gameextrainfo' in dataresponse: gamestr = ' playing `' + dataresponse['gameextrainfo'] + '`'
	else: gamestr = ''

	responsetext = [(namestr + ' is ' + statestr + gamestr).replace('  ', ' ')]

	return '\n'.join(responsetext)



# COMMAND HANDLING
prefix = "."


commands = {
	"help": cmd_help,
	"info": cmd_info,
	"upskirt": cmd_upskirt,
	"whoami": cmd_whoami,
	"whois": cmd_whois,
	"seen": cmd_seen,
	"say": cmd_say,
	"sayy": cmd_sayy,
	"markov":	 cmd_markov,
	"roll": cmd_roll,
	"qr":	 cmd_qr,
	"np":	 cmd_np,
	"steam": cmd_steam,
}