Moved logic inside stop command and added flag

Author: nightnei

cli/commands/site/stop-all.ts

@@ -1,6 +1,12 @@
-import { __ } from '@wordpress/i18n';
+import { __, _n, sprintf } from '@wordpress/i18n';
 import { SiteCommandLoggerAction as LoggerAction } from 'common/logger-actions';
-import { clearSiteLatestCliPid, getSiteByFolder, updateSiteAutoStart } from 'cli/lib/appdata';
+import {
+	clearSiteLatestCliPid,
+	getSiteByFolder,
+	readAppdata,
+	updateSiteAutoStart,
+	type SiteData,
+} from 'cli/lib/appdata';
 import { connect, disconnect } from 'cli/lib/pm2-manager';
 import { stopProxyIfNoSitesNeedIt } from 'cli/lib/site-utils';
 import { isServerRunning, stopWordPressServer } from 'cli/lib/wordpress-server-manager';
@@ -9,6 +15,20 @@ import { StudioArgv } from 'cli/types';
 
 const logger = new Logger< LoggerAction >();
 
+const filterRunningSites = async ( sites: SiteData[] ): Promise< SiteData[] > => {
+	const runningSites = [];
+
+	for ( const site of sites ) {
+		const runningProcess = await isServerRunning( site.id );
+
+		if ( runningProcess ) {
+			runningSites.push( site );
+		}
+	}
+
+	return runningSites;
+};
+
 export async function runCommand( siteFolder: string, autoStart: boolean ): Promise< void > {
 	try {
 		const site = await getSiteByFolder( siteFolder );
@@ -36,26 +56,122 @@ export async function runCommand( siteFolder: string, autoStart: boolean ): Prom
 	}
 }
 
+export async function runCommandAll( autoStart: boolean ): Promise< void > {
+	try {
+		const appdata = await readAppdata();
+		const allSites = appdata.sites;
+
+		if ( ! allSites.length ) {
+			logger.reportSuccess( __( 'No sites found' ) );
+			return;
+		}
+
+		await connect();
+
+		const runningSites = await filterRunningSites( allSites );
+
+		if ( ! runningSites.length ) {
+			logger.reportSuccess( __( 'No sites are currently running' ) );
+			return;
+		}
+
+		const stoppedSiteIds: string[] = [];
+
+		logger.reportStart(
+			LoggerAction.STOP_ALL_SITES,
+			sprintf(
+				__( 'Stopping all WordPress sites... (%d/%d)' ),
+				stoppedSiteIds.length,
+				runningSites.length
+			)
+		);
+
+		for ( const site of runningSites ) {
+			try {
+				logger.reportProgress(
+					sprintf(
+						__( 'Stopping site "%s" (%d/%d)…' ),
+						site.name,
+						stoppedSiteIds.length + 1,
+						runningSites.length
+					)
+				);
+				await stopWordPressServer( site.id );
+				await clearSiteLatestCliPid( site.id );
+				await updateSiteAutoStart( site.id, autoStart );
+
+				stoppedSiteIds.push( site.id );
+			} catch ( error ) {
+				logger.reportError(
+					new LoggerError( sprintf( __( 'Failed to stop site %s' ), site.name ) )
+				);
+			}
+		}
+
+		try {
+			await stopProxyIfNoSitesNeedIt( stoppedSiteIds, logger );
+		} catch ( error ) {
+			throw new LoggerError( __( 'Failed to stop proxy server' ), error );
+		}
+
+		if ( stoppedSiteIds.length === runningSites.length ) {
+			logger.reportSuccess(
+				sprintf(
+					_n(
+						'Successfully stopped %d site',
+						'Successfully stopped %d sites',
+						runningSites.length
+					),
+					runningSites.length
+				)
+			);
+		} else if ( stoppedSiteIds.length === 0 ) {
+			throw new LoggerError(
+				sprintf( __( 'Failed to stop all (%d) sites' ), runningSites.length )
+			);
+		} else {
+			throw new LoggerError(
+				sprintf( __( 'Stopped %d sites out of %d' ), stoppedSiteIds.length, runningSites.length )
+			);
+		}
+	} finally {
+		disconnect();
+	}
+}
+
 export const registerCommand = ( yargs: StudioArgv ) => {
 	return yargs.command( {
 		command: 'stop',
-		describe: __( 'Stop local site' ),
+		describe: __( 'Stop local site(s)' ),
 		builder: ( yargs ) => {
-			return yargs.option( 'auto-start', {
-				type: 'boolean',
-				describe: __( 'Set auto-start flag for the site' ),
-				default: false,
-				hidden: true,
-			} );
+			return yargs
+				.option( 'all', {
+					type: 'boolean',
+					describe: __( 'Stop all local sites' ),
+					default: false,
+				} )
+				.option( 'auto-start', {
+					type: 'boolean',
+					describe: __( 'Set auto-start flag for the site(s)' ),
+					default: false,
+					hidden: true,
+				} );
 		},
 		handler: async ( argv ) => {
 			try {
-				await runCommand( argv.path, argv.autoStart );
+				if ( argv.all ) {
+					await runCommandAll( argv.autoStart );
+				} else {
+					await runCommand( argv.path, argv.autoStart );
+				}
 			} catch ( error ) {
 				if ( error instanceof LoggerError ) {
 					logger.reportError( error );
 				} else {
-					const loggerError = new LoggerError( __( 'Failed to stop site' ), error );
+					const loggerError = new LoggerError(
+						argv.all ? __( 'Failed to stop sites' ) : __( 'Failed to stop site' ),
+						error
+					);
 					logger.reportError( loggerError );
 				}
 			}