 # This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

"""
This scripts pushes the passed in branch (argument) to the bot's
repository and creates a pull request for it.
"""

import os
import re
import subprocess
import sys
import taskcluster

from github import Github

# Make sure we have all needed arguments
if len(sys.argv) != 2:
	print "Usage", sys.argv[0], "BRANCH"
	exit(1)

# Get token for GitHub bot account from secrets service
secrets = taskcluster.Secrets({'baseUrl': 'http://taskcluster/secrets/v1'})
data = secrets.get('project/focus/github')
token = data['secret']['botAccountToken']

BRANCH = sys.argv[1]
OWNER = 'mozilla-mobile'
USER = 'MickeyMoz'
REPO = 'focus-android'
BASE = 'master'
HEAD = "MickeyMoz:%s" % BRANCH
URL = "https://%s:%s@github.com/%s/%s/" % (USER, token, USER, REPO)

# Check if there's already a pull request. If one is found then only update the existing one.
for request in pull_requests:
	if request.user.login == USER:
		print "There's already an unmerged pull request. Updating existing one."
		BRANCH=request.head.ref
		print subprocess.check_output(['git', 'checkout', '-b', BRANCH])
		print subprocess.check_output(['git', 'push', URL, BRANCH], '-f')
		exit(0)

# Push local state to branch
print subprocess.check_output(['git', 'push', URL, BRANCH])

# Read the log file and create the pull request body from it
log_path = os.path.join(os.path.dirname(__file__), '../../import-log.txt')
with open(log_path) as log_file:
	# Remove color codes from android2po output
	ansi_escape = re.compile(r'\x1b[^m]*m')
	log = ansi_escape.sub('', log_file.read())

body = """
Automated import %s

Log:
```
%s
```
""" % (BRANCH, log)

# Create pull request
github = Github(login_or_token=token)
repo = github.get_user(OWNER).get_repo(REPO)
pull_request = repo.create_pull(title='String import ' + BRANCH, body=body, base=BASE, head=HEAD)
print pull_request

# Add the [needs merge] label to the pull request
issue = repo.get_issue(pull_request.number)
issue.add_to_labels("needs merge")
print issue

@@ -944,16 +944,100 @@
  REPO)%0A%0A
+github = Github(login_or_token=token)%0Arepo = github.get_user(OWNER).get_repo(REPO)%0A%0A
 # Check 
@@ -1141,21 +1141,36 @@
  in 
-pull_requests
+repo.get_pulls(state='open')
 :%0A%09i
@@ -1931,91 +1931,8 @@
 est%0A
-github = Github(login_or_token=token)%0Arepo = github.get_user(OWNER).get_repo(REPO)%0A
 pull

Fix crash caused by wrong command order.