--- /dev/null
+<?php
+
+/**
+ * @file
+ * Drush integration for media_entity.
+ */
+
+use Drush\Log\LogLevel;
+
+/**
+ * Implements drush_hook_COMMAND_validate().
+ */
+function drush_media_entity_updatedb_validate() {
+ // This hook exists because when running DB updates using drush,
+ // hook_requirements() is not enforced (see
+ // https://github.com/drush-ops/drush/pull/2708 for more info on that).
+ // Here we just re-evaluate all the checks from hook_requirements() and abort
+ // the "updatedb/updb" command by returning FALSE to this hook if any of them
+ // is not met.
+ drush_bootstrap_to_phase(DRUSH_BOOTSTRAP_DRUPAL_FULL);
+
+ // Normally users should remove Media Entity from the codebase after the
+ // upgrade. However, if they don't do so, this function will be called on
+ // subsequent DB upgrades, even if ME is not enabled. Don't proceed on those
+ // circumstances.
+ if (!\Drupal::moduleHandler()->moduleExists('media_entity')) {
+ return TRUE;
+ }
+
+ $checks = \Drupal::service('media_entity.cli')->validateDbUpdateRequirements();
+ if (empty($checks['errors'])) {
+ return TRUE;
+ }
+ else {
+ foreach ($checks['errors'] as $error_msg) {
+ // We can't use drush_log() inside this hook.
+ drush_print($error_msg);
+ }
+ return FALSE;
+ }
+}
+
+/**
+ * Implements hook_drush_command().
+ */
+function media_entity_drush_command() {
+
+ $items['media-entity-check-upgrade'] = [
+ 'description' => 'Check upgrade requirements for Media Entity into Media in core.',
+ 'aliases' => ['mecu'],
+ 'core' => ['8+'],
+ 'examples' => [
+ "drush mecu" => "Checks upgrade requirements for Media Entity while upgrading to Media in core.",
+ ],
+ ];
+
+ return $items;
+}
+
+/**
+ * Callback for drush commmand "media-entity-check-upgrade" (mecu).
+ */
+function drush_media_entity_check_upgrade() {
+ // This command is useless if the DB updates have already been run.
+ if (drupal_get_installed_schema_version('media_entity') >= 8004) {
+ drush_log(dt('Your site has already run the media_entity DB updates. If you believe this is not correct, you should consider rolling back your database to a previous backup and try again.'), LogLevel::WARNING);
+ return;
+ }
+
+ drush_bootstrap_to_phase(DRUSH_BOOTSTRAP_DRUPAL_FULL);
+ $checks = \Drupal::service('media_entity.cli')->validateDbUpdateRequirements();
+
+ if (empty($checks['errors'])) {
+ drush_log(sprintf("\033[1;32;40m\033[1m%s\033[0m", '✓') . ' ' . dt('SUCCESS: All upgrade requirements are met and you can proceed with the DB updates.'), LogLevel::OK);
+ }
+ else {
+ drush_log(sprintf("\033[31;40m\033[1m%s\033[0m", '✗') . ' ' . dt('ERROR: Your site did not pass all upgrade checks. You can find more information in the error messages below.'), LogLevel::ERROR);
+ }
+ foreach ($checks['passes'] as $pass_msg) {
+ drush_log($pass_msg, LogLevel::OK);
+ }
+ foreach ($checks['errors'] as $error_msg) {
+ drush_log($error_msg, LogLevel::ERROR);
+ }
+}