Add type column in migrations table

Author: sagarnasit

migrations/.gitkeep

@@ -0,0 +1,64 @@
+<?php
+
+namespace EE\Migration;
+
+use EE;
+use EE\Migration\Base;
+
+class InsertDockerImagesVersion extends Base {
+
+	private static $pdo;
+
+	public function __construct() {
+
+		try {
+			self::$pdo = new \PDO( 'sqlite:' . DB );
+			self::$pdo->setAttribute( \PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION );
+		} catch ( \PDOException $exception ) {
+			EE::error( $exception->getMessage() );
+		}
+
+	}
+
+	/**
+	 * Execute create table query for site and sitemeta table.
+	 *
+	 * @throws EE\ExitException
+	 */
+	public function up() {
+
+		EE::log( 'Checking/Pulling required images' );
+		$images = EE\Utils\get_image_versions();
+
+		$query = '';
+		foreach ( $images as $image => $tag ) {
+			EE::debug( "Checking/Pulling docker image $image:$tag" );
+			if ( ! \EE::exec( "docker pull ${image}:${tag}" ) ) {
+				throw new \Exception( "Unable to pull ${image}:${tag}. Please check logs for more details." );
+			}
+			$query .= "INSERT INTO options VALUES( '${image}', '${tag}' );";
+		}
+
+		try {
+			self::$pdo->exec( $query );
+		} catch ( PDOException $exception ) {
+			EE::error( 'Encountered Error while inserting in "options" table: ' . $exception->getMessage(), false );
+		}
+	}
+
+	/**
+	 * Execute drop table query for site and sitemeta table.
+	 *
+	 * @throws EE\ExitException
+	 */
+	public function down() {
+
+		$query = "DELETE FROM options WHERE key LIKE 'easyengine/%';";
+
+		try {
+			self::$pdo->exec( $query );
+		} catch ( PDOException $exception ) {
+			EE::error( 'Encountered Error while deleting from "options" table: ' . $exception->getMessage(), false );
+		}
+	}
+}

migrations/db/20181016052850_easyengine_insert_docker_images_version.php

@@ -40,7 +40,7 @@ public static function execute_migrations() {
 	 * @return array of available migrations
 	 */
 	private static function get_all_migrations() {
-		$migrations = [];
+		$migrations    = [];
 		$packages_path = scandir( EE_VENDOR_DIR . '/easyengine' );
 
 		// get migrations from packages.
@@ -50,25 +50,25 @@ private static function get_all_migrations() {
 					continue;
 				}
 
-				$migration_path = EE_VENDOR_DIR . '/easyengine/' . $package . '/migrations';
+				$migration_path = EE_VENDOR_DIR . '/easyengine/' . $package . '/migrations/db';
 				if ( is_dir( $migration_path ) ) {
 					$files = scandir( $migration_path );
 					if ( \EE\Utils\inside_phar() ) {
 						$migrations[] = $files;
-					} else{
-						$migrations[] = array_slice( $files,2);
+					} else {
+						$migrations[] = array_slice( $files, 2 );
 					}
 				}
 			}
 		}
 
 		// get migrations from core.
 		if ( is_dir( EE_ROOT . '/migrations' ) ) {
-			$files = scandir( EE_ROOT . '/migrations' );
+			$files = scandir( EE_ROOT . '/migrations/db' );
 			if ( \EE\Utils\inside_phar() ) {
 				$migrations[] = $files;
-			} else{
-				$migrations[] = array_slice( $files,2);
+			} else {
+				$migrations[] = array_slice( $files, 2 );
 			}
 		}
 
@@ -121,6 +121,7 @@ private static function execute_migration_stack( $migrations ) {
 
 			Migration::create( [
 				'migration' => $migrations[0],
+				'type'      => 'db',
 				'timestamp' => date( 'Y-m-d H:i:s' ),
 			] );
 
@@ -167,7 +168,7 @@ private static function get_migrations_to_execute( $migrations ) {
 	 * @return array
 	 */
 	private static function get_migrations_from_db() {
-		return array_column( Migration::all(), 'migration' );
+		return array_column( Migration::where( 'type', 'db' ), 'migration' );
 	}
 
 	/**
@@ -184,9 +185,9 @@ private static function get_migration_path( $migration_name ) {
 			return '';
 		}
 		if ( 'easyengine' === $matches[1] ) {
-			return EE_ROOT . "/migrations/$migration_name";
+			return EE_ROOT . "/migrations/db/$migration_name";
 		} else {
-			return EE_ROOT . "/vendor/easyengine/$matches[1]/migrations/$migration_name";
+			return EE_ROOT . "/vendor/easyengine/$matches[1]/migrations/db/$migration_name";
 		}
 
 	}

php/EE/Migration/Executor.php

@@ -60,6 +60,7 @@ private static function create_required_tables() {
 		$query = 'CREATE TABLE migrations (
 			id INTEGER,
 			migration VARCHAR,
+			type VARCHAR,
 			timestamp DATETIME,
 			PRIMARY KEY (id)
 		);';