update comments

[erebus.git] / modules / steam.py

# Erebus IRC bot - Author: Erebus Team
# vim: fileencoding=utf-8
# Steam API commands (!steamnp)

# module info
modinfo = {
	'author': 'Erebus Team',
	'license': 'public domain',
	'compatible': [0], # compatible module API versions
	'depends': [], # other modules required to work properly?
	'softdeps': ['help'], # modules which are preferred but not required
}

# preamble
import modlib
lib = modlib.modlib(__name__)
modstart = lib.modstart
modstop = lib.modstop

# module code
import json
import sys
if sys.version_info.major < 3:
	from urllib import urlopen, quote, HTTPError
else:
	from urllib.request import urlopen
	from urllib.parse import quote
	from urllib.error import HTTPError
_lookup_states = {
	0: 'Offline',
	1: 'Online',
	2: 'Busy',
	3: 'Away',
	4: 'Snooze',
	5: 'Looking to trade',
	6: 'Looking to play'
}

def _key():
	return lib.parent.cfg.get('steam', 'apikey')

def _resolve_urlname(urlname):
	if urlname.isascii() and urlname.isdecimal():
		return urlname
	url = "https://api.steampowered.com/ISteamUser/ResolveVanityURL/v0001/?key=%s&vanityurl=%s" % (_key(), urlname)
	try:
		data = json.load(urlopen(url))
		if lib.parent.cfg.getboolean('debug', 'steam'):
			lib.parent.log('*', "?", repr(data))
		return data['response']['steamid']
	except HTTPError as e:
		raise Exception("(In looking up SteamID:) No such user (or some other error: %s %s) " % (e.code, e.msg))
	except json.JSONDecodeError as e:
		raise Exception("(In looking up SteamID:) Couldn't decode response: %s %d" % (e.msg, e.pos))
	except KeyError as e:
		raise Exception("(In looking up SteamID:) Couldn't read response: %r" % (e))
	except Exception as e:
		raise Exception("(In looking up SteamID:) Some other error occurred: %s %r" % (type(e).__name__, e))



@lib.hook(needchan=False, wantchan=True)
@lib.help('<custom URL|SteamID>', 'looks up what a user is playing on Steam')
@lib.argsEQ(1)
def steamnp(bot, user, chan, realtarget, *args):
	steamid = _resolve_urlname(args[0])

	url = "https://api.steampowered.com/ISteamUser/GetPlayerSummaries/v0002/?key=%s&steamids=%s" % (_key(), steamid)

	try:
		data = json.load(urlopen(url))
	except HTTPError as e:
		raise Exception("No such user (or some other error: %s %s) " % (e.code, e.msg))
	except json.JSONDecodeError as e:
		raise Exception("Couldn't decode response: %s %d" % (e.msg, e.pos))
	except KeyError as e:
		raise Exception("Couldn't read response: %r" % (e))
	except Exception as e:
		raise Exception("Some other error occurred: %s %r" % (type(e).__name__, e))

	if lib.parent.cfg.getboolean('debug', 'steam'):
		lib.parent.log('*', "?", repr(data))

	player = data['response']['players'][0]

	profileurl = player['profileurl']
	stateint = player['personastate']
	state = _lookup_states[stateint]
	displayname = player['personaname']
	if 'gameextrainfo' in player:
		gamename = player['gameextrainfo']
	else:
		gamename = None

	if gamename is None:
		return "%s (%s) is not playing a game right now. %s" % (displayname, state, profileurl)
	else:
		return "%s (%s) is playing \037%s\037. %s" % (displayname, state, gamename, profileurl)