From b8804973ca1de9a12fdce68cb2aca748a550fc0f Mon Sep 17 00:00:00 2001
From: Adam <adam111316@users.noreply.github.com>
Date: Mon, 29 Dec 2014 12:22:34 +0800
Subject: [PATCH] Add pull request checkout option to General Config/Advanced
 Settings

---
 CHANGES.md                                    |  1 +
 .../interfaces/default/config_general.tmpl    | 18 +++++++++++++
 gui/slick/js/config.js                        |  4 +++
 sickbeard/gh_api.py                           | 25 ++++++++++++++++++-
 sickbeard/versionChecker.py                   | 23 ++++++++++++++++-
 sickbeard/webserve.py                         | 11 ++++++++
 6 files changed, 80 insertions(+), 2 deletions(-)

diff --git a/CHANGES.md b/CHANGES.md
index 09c4b64a..4ec76cfd 100644
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -15,6 +15,7 @@
 * Add anime unit test cases (port from lad1337/sickbeard)
 * Fix normal tv show regex (port from midgetspy/sickbeard)
 * Fix anime regex (port from lad1337/sickbeard)
+* Add pull request checkout option to General Config/Advanced Settings
 
 [develop changelog]
 
diff --git a/gui/slick/interfaces/default/config_general.tmpl b/gui/slick/interfaces/default/config_general.tmpl
index 2817d0e0..410410c1 100644
--- a/gui/slick/interfaces/default/config_general.tmpl
+++ b/gui/slick/interfaces/default/config_general.tmpl
@@ -461,6 +461,24 @@
 							</label>
 						</div>
 
+						#set pulls = sickbeard.versionCheckScheduler.action.list_remote_pulls()
+						#if len(pulls) > 0 and $sickbeard.BRANCH != 'master':
+							<div class="field-pair">
+								<label>
+									<span class="component-title">Pull request:</span>
+									<span class="component-desc">
+										<select id="pullRequestVersion" class="form-control form-control-inline input-sm pull-left">
+										#for $cur_branch in $pulls:
+											<option value="$cur_branch.fetch_name()" #if $cur_branch == $sickbeard.BRANCH then 'selected="selected"' else ''#>$cur_branch</option>
+										#end for
+										</select>
+										<input class="btn btn-inline" style="margin-left: 6px;" type="button" id="pullRequestCheckout" value="Checkout Pull Request">
+										<div class="clear-left"><p>select pull request to test (restart required)</p></div>
+									</span>
+								</label>
+							</div>
+						#end if
+
 						<div class="field-pair">
 							<label for="git_remote">
 								<span class="component-title">Git remote for branch</span>
diff --git a/gui/slick/js/config.js b/gui/slick/js/config.js
index 41046e69..06a2682b 100644
--- a/gui/slick/js/config.js
+++ b/gui/slick/js/config.js
@@ -113,6 +113,10 @@ $(document).ready(function(){
 	$('#branchCheckout').click(function(){
 		window.location.href = sbRoot + '/home/branchCheckout?branch=' + $('#branchVersion').val();
 	});
+
+	$('#pullRequestCheckout').click(function(){
+		window.location.href = sbRoot + '/home/pullRequestCheckout?branch=' + $('#pullRequestVersion').val();
+	});
 	
 });
 
diff --git a/sickbeard/gh_api.py b/sickbeard/gh_api.py
index ef97bf44..89ed8b48 100644
--- a/sickbeard/gh_api.py
+++ b/sickbeard/gh_api.py
@@ -92,4 +92,27 @@ class GitHub(object):
         access_API = self._access_API(
             ['repos', self.github_repo_user, self.github_repo, 'branches'],
             params={'per_page': 100})
-        return access_API
\ No newline at end of file
+        return access_API
+
+    def pull_requests(self):
+        access_API = self._access_API(
+            ['repos', self.github_repo_user, self.github_repo, 'pulls'],
+            params={'per_page': 100})
+        pull = []
+        for x in access_API:
+            try:
+                pull.append(PullRequest(x['head']['ref'], x['number']))
+            except:
+                continue
+        return pull
+
+class PullRequest(object):
+    def __init__(self, ref, number):
+        self.ref = ref
+        self.number = number
+
+    def __repr__(self):
+        return '%s: %s' % (self.number, self.ref)
+
+    def fetch_name(self):
+        return 'pull/%s/head:pull/%s/%s' % (self.number, self.number, self.ref)
\ No newline at end of file
diff --git a/sickbeard/versionChecker.py b/sickbeard/versionChecker.py
index 4c6d5a32..3840026c 100644
--- a/sickbeard/versionChecker.py
+++ b/sickbeard/versionChecker.py
@@ -115,9 +115,15 @@ class CheckVersion():
         if self.updater.need_update():
             return self.updater.update()
 
+    def fetch(self, pull_request):
+        return self.updater.fetch(pull_request)
+
     def list_remote_branches(self):
         return self.updater.list_remote_branches()
 
+    def list_remote_pulls(self):
+        return self.updater.list_remote_pulls()
+
     def get_branch(self):
         return self.updater.branch
 
@@ -400,6 +406,17 @@ class GitUpdateManager(UpdateManager):
             return re.findall('\S+\Wrefs/heads/(.*)', branches)
         return []
 
+    def list_remote_pulls(self):
+        gh = github.GitHub(self.github_repo_user, self.github_repo, self.branch)
+        return gh.pull_requests()
+
+    def fetch(self, pull_request):
+        output, err, exit_status = self._run_git(self._git_path, 'fetch -f %s %s' % (sickbeard.GIT_REMOTE, pull_request))  # @UnusedVariable
+        if exit_status == 0:
+            return True
+        return False
+
+
 
 class SourceUpdateManager(UpdateManager):
     def __init__(self):
@@ -596,4 +613,8 @@ class SourceUpdateManager(UpdateManager):
 
     def list_remote_branches(self):
         gh = github.GitHub(self.github_repo_user, self.github_repo, self.branch)
-        return [x['name'] for x in gh.branches() if x and 'name' in x]
\ No newline at end of file
+        return [x['name'] for x in gh.branches() if x and 'name' in x]
+
+    def list_remote_pulls(self):
+        # we don't care about testers that don't use git
+        return []
\ No newline at end of file
diff --git a/sickbeard/webserve.py b/sickbeard/webserve.py
index e51cded0..619a1761 100644
--- a/sickbeard/webserve.py
+++ b/sickbeard/webserve.py
@@ -3628,6 +3628,17 @@ class Home(MainHandler):
         ui.notifications.message('Checking out branch: ', branch)
         return self.update(sickbeard.PID)
 
+    def pullRequestCheckout(self, branch):
+        pull_request = branch
+        branch = branch.split(':')[1]
+        fetched = sickbeard.versionCheckScheduler.action.fetch(pull_request)
+        if fetched:
+            sickbeard.BRANCH = branch
+            ui.notifications.message('Checking out branch: ', branch)
+            return self.update(sickbeard.PID)
+        else:
+            return redirect('/home/')
+
     def displayShow(self, show=None):
 
         if show is None: