180 lines
5.3 KiB
JavaScript
180 lines
5.3 KiB
JavaScript
const path = require( 'path' );
|
|
const fs = require( 'fs' );
|
|
const mkdirp = require( 'mkdirp' );
|
|
const request = require( 'request' );
|
|
const StreamZip = require( 'node-stream-zip' );
|
|
const { resolveLocalE2ePath } = require( './test-config' );
|
|
|
|
/**
|
|
* Upload a plugin zip from a remote location, such as a GitHub URL or other hosted location.
|
|
*
|
|
* @param {string} fileUrl The URL where the zip file is located.
|
|
* @param {string} authorizationToken Authorization token used to authenticate with the GitHub API if required.
|
|
* @return {string} The path where the zip file is located.
|
|
*/
|
|
const getRemotePluginZip = async ( fileUrl, authorizationToken = '' ) => {
|
|
const savePath = resolveLocalE2ePath( 'plugins' );
|
|
mkdirp.sync( savePath );
|
|
|
|
// Pull the version from the end of the URL
|
|
const fileName = fileUrl.split( '/' ).pop();
|
|
let filePath = path.join( savePath, fileName );
|
|
|
|
// First, download the zip file
|
|
await downloadZip( fileUrl, filePath, authorizationToken );
|
|
|
|
// Check for a nested zip and update the filepath
|
|
filePath = await checkNestedZip( filePath, savePath );
|
|
|
|
return filePath;
|
|
};
|
|
|
|
/**
|
|
* Get the latest release zip for a plugin from a GiHub repository.
|
|
*
|
|
* @param {string} repository The repository owner and name. For example: `woocommerce/woocommerce`.
|
|
* @param {string} authorizationToken Authorization token used to authenticate with the GitHub API if required.
|
|
* @param {boolean} getPrerelease Flag on whether to get a prelease or not.
|
|
* @param {number} perPage Limit of entries returned from the latest releases list, defaults to 3.
|
|
* @return {Promise<string>}} Returns the URL for the release zip file.
|
|
*/
|
|
const getLatestReleaseZipUrl = async (
|
|
repository,
|
|
authorizationToken = '',
|
|
getPrerelease = false,
|
|
perPage = 3
|
|
) => {
|
|
let requesturl;
|
|
|
|
if ( getPrerelease ) {
|
|
requesturl = `https://api.github.com/repos/${ repository }/releases?per_page=${ perPage }`;
|
|
} else {
|
|
requesturl = `https://api.github.com/repos/${ repository }/releases/latest`;
|
|
}
|
|
|
|
const options = {
|
|
url: requesturl,
|
|
method: 'GET',
|
|
json: true,
|
|
headers: { 'user-agent': 'node.js' },
|
|
};
|
|
|
|
// If provided with a token, use it for authorization
|
|
if ( authorizationToken ) {
|
|
options.headers.Authorization = `token ${ authorizationToken }`;
|
|
}
|
|
|
|
// Wrap in a promise to make the request async
|
|
return new Promise( function ( resolve, reject ) {
|
|
request.get( options, function ( err, resp, body ) {
|
|
if ( err ) {
|
|
reject( err );
|
|
} else if ( getPrerelease ) {
|
|
// Loop until we find the first pre-release, then return it.
|
|
body.forEach( ( release ) => {
|
|
if ( release.prerelease ) {
|
|
resolve( release.assets[ 0 ].browser_download_url );
|
|
}
|
|
} );
|
|
} else if ( authorizationToken ) {
|
|
// If it's a private repo, we need to download the archive this way
|
|
const tagName = body.tag_name;
|
|
resolve(
|
|
`https://github.com/${ repository }/archive/${ tagName }.zip`
|
|
);
|
|
} else {
|
|
resolve( body.assets[ 0 ].browser_download_url );
|
|
}
|
|
} );
|
|
} );
|
|
};
|
|
|
|
/**
|
|
* Check the zip file for any nested zips. If one is found, extract it.
|
|
*
|
|
* @param {string} zipFilePath The location of the zip file.
|
|
* @param {string} savePath The location where to save a nested zip if found.
|
|
* @return {string} The path where the zip file is located.
|
|
*/
|
|
const checkNestedZip = async ( zipFilePath, savePath ) => {
|
|
const zip = new StreamZip.async( { file: zipFilePath } );
|
|
const entries = await zip.entries();
|
|
|
|
for ( const entry of Object.values( entries ) ) {
|
|
if ( entry.name.match( /\.zip/ ) ) {
|
|
await zip.extract( null, savePath );
|
|
await zip.close();
|
|
return path.join( savePath, entry.name );
|
|
}
|
|
}
|
|
|
|
return zipFilePath;
|
|
};
|
|
|
|
/**
|
|
* Download the zip file from a remote location.
|
|
*
|
|
* @param {string} fileUrl The URL where the zip file is located.
|
|
* @param {string} downloadPath The location where to download the zip to.
|
|
* @param {string} authorizationToken Authorization token used to authenticate with the GitHub API if required.
|
|
* @return {Promise<void>}
|
|
*/
|
|
const downloadZip = async ( fileUrl, downloadPath, authorizationToken ) => {
|
|
const options = {
|
|
url: fileUrl,
|
|
method: 'GET',
|
|
encoding: null,
|
|
headers: { 'user-agent': 'node.js' },
|
|
};
|
|
|
|
// If provided with a token, use it for authorization
|
|
if ( authorizationToken ) {
|
|
options.headers.Authorization = `token ${ authorizationToken }`;
|
|
}
|
|
|
|
// Wrap in a promise to make the request async
|
|
return new Promise( function ( resolve, reject ) {
|
|
request
|
|
.get( options, function ( err, resp, body ) {
|
|
if ( err ) {
|
|
reject( err );
|
|
} else {
|
|
resolve( body );
|
|
}
|
|
} )
|
|
.pipe( fs.createWriteStream( downloadPath ) );
|
|
} );
|
|
};
|
|
|
|
/**
|
|
* Delete the downloaded plugin files.
|
|
*/
|
|
const deleteDownloadedPluginFiles = async () => {
|
|
const pluginSavePath = resolveLocalE2ePath( 'plugins' );
|
|
|
|
fs.readdir( pluginSavePath, ( err, contents ) => {
|
|
if ( err ) throw err;
|
|
|
|
for ( const content of contents ) {
|
|
const contentPath = path.join( pluginSavePath, content );
|
|
const stats = fs.lstatSync( contentPath );
|
|
|
|
if ( stats.isDirectory() ) {
|
|
fs.rmSync( contentPath, { recursive: true, force: true } );
|
|
} else {
|
|
fs.unlink( contentPath, ( error ) => {
|
|
if ( error ) throw error;
|
|
} );
|
|
}
|
|
}
|
|
} );
|
|
};
|
|
|
|
module.exports = {
|
|
getRemotePluginZip,
|
|
getLatestReleaseZipUrl,
|
|
checkNestedZip,
|
|
downloadZip,
|
|
deleteDownloadedPluginFiles,
|
|
};
|