"""
upload.py: upload the helper script to a wiki (requires mwclient and git)
(C) 2014 Theopolisme <theopolismewiki@gmail.com>

Usage
=====

Run from the main afch-rewrite directory:

>>> python scripts/upload.py [site] [root] [username] [password]

site: enwiki or testwiki

root: Base page name for the script, without any file extension.
      For example, if "MediaWiki:Gadget-afch" was specified the
      script can be loaded from `MediaWiki:Gadget-afch.js`.

username: username of account on site
password: password of account on site
"""
from __future__ import unicode_literals

import sys
import os
import git
import mwclient

wiki = sys.argv[1]

if wiki == 'enwiki':
	site = mwclient.Site('en.wikipedia.org')
elif wiki == 'testwiki':
	site = mwclient.Site('test.wikipedia.org')
else:
	print 'Error: unrecognized wiki "{}"'.format(wiki)
	sys.exit(0)

# Login with username and password
site.login(sys.argv[3], sys.argv[4])

# Base page name on-wiki
root = sys.argv[2]

# Get branch name and the current commit
repo = git.Repo(os.getcwd())
branch = repo.active_branch
sha1 = branch.commit.hexsha

# Prepend this to every page
header = '/* Uploaded from https://github.com/WPAFC/afch-rewrite, commit: {} ({}) */\n'.format(sha1, branch)

def uploadFile(pagename, content):
	page = site.Pages[pagename]

	# Add header and update static referencres to root directory
	content = header + content.decode('utf-8')
	content = content.replace('MediaWiki:Gadget-afch',root)

	def stripFirstLine(text):
		return '\n'.join(text.splitlines()[1:])

	# Only update the page if its contents have changed (excluding the header)
	if stripFirstLine(content) != stripFirstLine(page.edit()):
		print 'Uploading to {}'.format(pagename)
		page.save(content, 'Updating AFCH: {} @ {}'.format(branch, sha1[:6]))
	else:
		print 'Skipping {}, no changes made'.format(pagename)

def uploadSubscript(scriptName, content):
	uploadFile(root + '.js/' + scriptName + '.js', content)

def uploadDirectory(directory):
	files = os.listdir(directory)
	for script in files:
		# Skip hidden files
		if not script.startswith('.'):
			with open(directory + '/' + script, 'r') as f:
				content = f.read()
			uploadSubscript(os.path.splitext(script)[0], content)

# Upload afch.js
with open('src/afch.js', 'r') as f:
	uploadFile(root + '.js', f.read())

# Upload afch.css
with open('src/afch.css', 'r') as f:
	uploadFile(root + '.css', f.read())

# Now upload everything else: modules, templates, dependencies
uploadDirectory('src/modules')
uploadDirectory('src/templates')
uploadDirectory('dependencies')

print 'AFCH uploaded to {} successfully!'.format(wiki)