Merge branch 'develop'

Author: mbtamuli

src/Shell_Command.php

@@ -1,18 +1,19 @@
 <?php
 
+use EE\Utils;
+use EE\Model\Site;
+use function EE\Site\Utils\auto_site_name;
+
 /**
- * Executes wp-cli command on a site.
+ * Brings up a shell to run wp-cli, composer etc.
  *
  * ## EXAMPLES
  *
- *     # Create simple WordPress site
- *     $ ee wp test.local plugin list
+ *     # Open shell of example.com
+ *     $ ee shell example.com
  *
  * @package ee-cli
  */
-
-use EE\Utils;
-
 class Shell_Command extends EE_Command {
 
 	/**
@@ -24,28 +25,38 @@ class Shell_Command extends EE_Command {
 	 * : Name of website to run shell on.
 	 */
 	public function __invoke( $args ) {
+
 		EE\Utils\delem_log( 'ee shell start' );
-		$args = EE\Utils\set_site_arg( $args, 'shell' );
+		$args      = auto_site_name( $args, 'shell', '' );
 		$site_name = EE\Utils\remove_trailing_slash( $args[0] );
-		if ( EE::db()::site_in_db( $site_name ) ) {
-			$db_select = EE::db()::select( ['site_path'], ['sitename' => $site_name]);
-			$site_root    = $db_select[0]['site_path'];
-		} else {
-			EE::error( "Site $site_name does not exist." );
+
+		$site = Site::find( $site_name );
+
+		if ( ! $site || ! $site->site_enabled ) {
+			EE::error( "Site $site_name does not exist or is not enabled." );
 		}
-		chdir($site_root);
+
+		chdir( $site->site_fs_path );
+		$this->check_shell_available( 'php', $site );
 		$this->run( "docker-compose exec --user='www-data' php bash" );
 		EE\Utils\delem_log( 'ee shell end' );
 	}
 
+	/**
+	 * Run the command to open shell.
+	 *
+	 * @param string $cmd             Command to be executed to open shell.
+	 * @param null|array $descriptors File descriptors for proc.
+	 */
 	private function run( $cmd, $descriptors = null ) {
+
 		EE\Utils\check_proc_available( 'ee_shell' );
 		if ( ! $descriptors ) {
 			$descriptors = array( STDIN, STDOUT, STDERR );
 		}
-		
+
 		$final_cmd = EE\Utils\force_env_on_nix_systems( $cmd );
-		$proc = EE\Utils\proc_open_compat( $final_cmd, $descriptors, $pipes );
+		$proc      = EE\Utils\proc_open_compat( $final_cmd, $descriptors, $pipes );
 		if ( ! $proc ) {
 			exit( 1 );
 		}
@@ -55,4 +66,21 @@ private function run( $cmd, $descriptors = null ) {
 		}
 	}
 
+	/**
+	 * Function to check if container supporting shell is present in docker-compose.yml or not.
+	 *
+	 * @param string $shell_container Container to be checked.
+	 * @param Object $site            Contains relevant site info.
+	 */
+	private function check_shell_available( $shell_container, $site ) {
+
+		$launch   = EE::launch( 'docker-compose config --services' );
+		$services = explode( PHP_EOL, trim( $launch->stdout ) );
+		if ( in_array( $shell_container, $services, true ) ) {
+			return;
+		}
+		EE::debug( 'Site type: ' . $site->site_type );
+		EE::debug( 'Site command: ' . $site->app_sub_type );
+		EE::error( sprintf( '%s site does not have support to launch %s shell.', $site->site_url, $shell_container ) );
+	}
 }