|
- # Maki
- # ----
- # Discord bot by MrDetonia
- #
- # Copyright 2018 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://github.com/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):
- response = msg.content[5:]
- yield from client.delete_message(msg)
- 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,
- }
|