#!/usr/bin/env python3
# Copyright (C) 2020 Alibek Omarov
# This program 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 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.


import requests
import json
import sys
import os
from optparse import OptionParser

DISCORD_API_GUILD = 'https://discordapp.com/api/v6/guilds/%s'
DISCORD_CDN_EMOJI = 'https://cdn.discordapp.com/emojis/%s.%s'

def die(msg):
	print(msg)
	sys.exit(1)

def discord_api_get(url, headers):
	print('API GET %s' % url)
	r = requests.get(url, headers = headers)
	return json.loads(r.text)

def discord_cdn_get(url, out, force):
	if os.path.exists(out) and not force:
		print('Skipping %s, already exists' % out)
		return
	
	print('CDN GET %s -> %s' % (url, out))
	
	r = requests.get(url, allow_redirects=True)
	open(out, 'wb').write(r.content)

def main(token, guild, force = False, out = None, ua = None):
	if out is None:
		out = guild
	
	try:
		os.mkdir(out)
	except FileExistsError:
		if not force:
			die('Directory %s already exists' % out)
	
	headers = { 'Authorization': token }
	
	if ua != None:
		headers['User-Agent'] = ua
	
	guildinfo = discord_api_get(DISCORD_API_GUILD % guild, headers)
	
	emojilist = guildinfo['emojis']
		
	packjson = {
		'files': {},
		'pack': {
			'description': 'Emojis from Discord server %s' % guildinfo['name'],
			'license': 'Unknown', # did you asked author permission? I didn't. :(
			'homepage': 'https://example.org', # fill with invite if you want to
			'share-files': True # I don't know what it means
		}
	}
	
	for emoji in emojilist:
		animated = emoji['animated']
		id = emoji['id']
		name = emoji['name']
		
		ext = 'gif' if animated else 'png'
		
		discord_cdn_get(DISCORD_CDN_EMOJI % (id, ext), os.path.join(out, name + '.' + ext), force)
		packjson['files'][name] = name + '.' + ext
	
	# always write pack.json for consistency
	print('Writing pack.json...')
	open(os.path.join(out, 'pack.json'), 'w').write(json.dumps(packjson))
	
	print('Done!')
	
	return 0

if __name__ == '__main__':
	import config # to be filled by user

	parser = OptionParser()
	parser.add_option('-g', '--guild', action='store', dest='guild',
		help='server to scrap emojis from')
	parser.add_option('-o', '--out', action='store', dest='out',
		help='result emojipack name')
	parser.add_option('-f', '--force', action='store_true', dest='force',
		help='ignore existing files')
	
	(options, args) = parser.parse_args()
	
	if options.guild is None:
		die('Enter guild ID')

	if len(config.AUTHORIZATION_TOKEN) == 0:
		die('Enter authorization token in config.py')
	
	sys.exit(main(
		config.AUTHORIZATION_TOKEN,
		options.guild,
		options.force,
		options.out, config.USER_AGENT))