114 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			114 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| /**
 | |
|  * @license
 | |
|  * Copyright Google Inc. All Rights Reserved.
 | |
|  *
 | |
|  * Use of this source code is governed by an MIT-style license that can be
 | |
|  * found in the LICENSE file at https://angular.io/license
 | |
|  */
 | |
| 
 | |
| /**
 | |
|  * **Usage:**
 | |
|  * ```
 | |
|  * node rebase-pr <github-repository> <pull-request-number>
 | |
|  * ```
 | |
|  * **Example:**
 | |
|  * ```
 | |
|  * node rebase-pr angular/angular 123
 | |
|  * ```
 | |
|  *
 | |
|  * Rebases the current branch on top of the GitHub PR target branch.
 | |
|  *
 | |
|  * **Context:**
 | |
|  * Since a GitHub PR is not necessarily up to date with its target branch, it is useful to rebase
 | |
|  * prior to testing it on CI to ensure more up to date test results.
 | |
|  *
 | |
|  * **Implementation details:**
 | |
|  * This script obtains the base for a GitHub PR via the
 | |
|  * [GitHub PR API](https://developer.github.com/v3/pulls/#get-a-single-pull-request), then
 | |
|  * fetches that branch, and rebases the current branch on top of it.
 | |
|  *
 | |
|  * **NOTE:**
 | |
|  * This script cannot use external dependencies or be compiled because it needs to run before the
 | |
|  * environment is setup.
 | |
|  * Use only features supported by the NodeJS versions used in the environment.
 | |
|  */
 | |
| 
 | |
| // This script uses `console` to print messages to the user.
 | |
| // tslint:disable:no-console
 | |
| 
 | |
| // Imports
 | |
| const util = require('util');
 | |
| const https = require('https');
 | |
| const child_process = require('child_process');
 | |
| const exec = util.promisify(child_process.exec);
 | |
| 
 | |
| // CLI validation
 | |
| if (process.argv.length != 4) {
 | |
|   console.error(`This script requires the GitHub repository and PR number as arguments.`);
 | |
|   console.error(`Example: node tools/rebase-pr.js angular/angular 123`);
 | |
|   process.exitCode = 1;
 | |
|   return;
 | |
| }
 | |
| 
 | |
| // Run
 | |
| _main(...process.argv.slice(2)).catch(err => {
 | |
|   console.log('Failed to rebase on top of target branch.\n');
 | |
|   console.error(err);
 | |
|   process.exitCode = 1;
 | |
| });
 | |
| 
 | |
| // Helpers
 | |
| async function _main(repository, prNumber) {
 | |
|   console.log(`Determining target branch for PR ${prNumber} on ${repository}.`);
 | |
|   const targetBranch = await determineTargetBranch(repository, prNumber);
 | |
|   console.log(`Target branch is ${targetBranch}.`);
 | |
|   await exec(`git fetch origin ${targetBranch}`);
 | |
|   console.log(`Rebasing current branch on ${targetBranch}.`);
 | |
|   await exec(`git rebase origin/${targetBranch}`);
 | |
|   console.log('Rebase successful.');
 | |
| }
 | |
| 
 | |
| function determineTargetBranch(repository, prNumber) {
 | |
|   const pullsUrl = `https://api.github.com/repos/${repository}/pulls/${prNumber}`;
 | |
|   // GitHub requires a user agent: https://developer.github.com/v3/#user-agent-required
 | |
|   const options = {headers: {'User-Agent': repository}};
 | |
| 
 | |
|   return new Promise((resolve, reject) => {
 | |
|     https
 | |
|         .get(
 | |
|             pullsUrl, options,
 | |
|             (res) => {
 | |
|               const {statusCode} = res;
 | |
|               const contentType = res.headers['content-type'];
 | |
|               let rawData = '';
 | |
| 
 | |
|               res.on('data', (chunk) => { rawData += chunk; });
 | |
|               res.on('end', () => {
 | |
| 
 | |
|                 let error;
 | |
|                 if (statusCode !== 200) {
 | |
|                   error = new Error(
 | |
|                       `Request Failed.\nStatus Code: ${statusCode}.\nResponse: ${rawData}`);
 | |
|                 } else if (!/^application\/json/.test(contentType)) {
 | |
|                   error = new Error(
 | |
|                       'Invalid content-type.\n' +
 | |
|                       `Expected application/json but received ${contentType}`);
 | |
|                 }
 | |
| 
 | |
|                 if (error) {
 | |
|                   reject(error);
 | |
|                   return;
 | |
|                 }
 | |
| 
 | |
|                 try {
 | |
|                   const parsedData = JSON.parse(rawData);
 | |
|                   resolve(parsedData['base']['ref']);
 | |
|                 } catch (e) {
 | |
|                   reject(e);
 | |
|                 }
 | |
|               });
 | |
|             })
 | |
|         .on('error', (e) => { reject(e); });
 | |
|   });
 | |
| }
 |