PubObjectsExportPlugin.inc.php

<?php
 
import('lib.pkp.classes.plugins.ImportExportPlugin');
 
// The statuses.
define('EXPORT_STATUS_ANY', '');
define('EXPORT_STATUS_NOT_DEPOSITED', 'notDeposited');
define('EXPORT_STATUS_MARKEDREGISTERED', 'markedRegistered');
define('EXPORT_STATUS_REGISTERED', 'registered');
 
// The actions.
define('EXPORT_ACTION_EXPORT', 'export');
define('EXPORT_ACTION_MARKREGISTERED', 'markRegistered');
define('EXPORT_ACTION_DEPOSIT', 'deposit');
 
// Configuration errors.
define('EXPORT_CONFIG_ERROR_SETTINGS', 0x02);
 
 
abstract class PubObjectsExportPlugin extends ImportExportPlugin {
   var $_cache;
 
   function getCache() {
      if (!is_a($this->_cache, 'PubObjectCache')) {
         // Instantiate the cache.
         import('classes.plugins.PubObjectCache');
         $this->_cache = new PubObjectCache();
      }
      return $this->_cache;
   }
 
   function register($category, $path, $mainContextId = null) {
      if (!parent::register($category, $path, $mainContextId)) return false;
 
      AppLocale::requireComponents(LOCALE_COMPONENT_APP_MANAGER);
      $this->addLocaleData();
      
      HookRegistry::register('AcronPlugin::parseCronTab', array($this, 'callbackParseCronTab'));
      foreach ($this->_getDAOs() as $dao) {
         if ($dao instanceof SchemaDAO) {
            HookRegistry::register('Schema::get::' . $dao->schemaName, array($this, 'addToSchema'));
         } else {
            HookRegistry::register(strtolower_codesafe(get_class($dao)) . '::getAdditionalFieldNames', array(&$this, 'getAdditionalFieldNames'));
         }
      }
      return true;
   }
 
   function manage($args, $request) {
      $user = $request->getUser();
      $router = $request->getRouter();
      $context = $router->getContext($request);
 
      $form = $this->_instantiateSettingsForm($context);
      $notificationManager = new NotificationManager();
      switch ($request->getUserVar('verb')) {
         case 'save':
            $form->readInputData();
            if ($form->validate()) {
               $form->execute();
               $notificationManager->createTrivialNotification($user->getId(), NOTIFICATION_TYPE_SUCCESS);
               return new JSONMessage(true);
            } else {
               return new JSONMessage(true, $form->fetch($request));
            }
         case 'index':
            $form->initData();
            return new JSONMessage(true, $form->fetch($request));
         case 'statusMessage':
            $statusMessage = $this->getStatusMessage($request);
            if ($statusMessage) {
               $templateMgr = TemplateManager::getManager($request);
               $templateMgr->assign(array(
                  'statusMessage' => htmlentities($statusMessage),
               ));
               return new JSONMessage(true, $templateMgr->fetch($this->getTemplateResource('statusMessage.tpl')));
            }
      }
      return parent::manage($args, $request);
   }
 
   function display($args, $request) {
      parent::display($args, $request);
 
      $context = $request->getContext();
      switch (array_shift($args)) {
         case 'index':
         case '':
            // Check for configuration errors:
            $configurationErrors = array();
            // missing plugin settings
            $form = $this->_instantiateSettingsForm($context);
            foreach($form->getFormFields() as $fieldName => $fieldType) {
               if ($form->isOptional($fieldName)) continue;
               $pluginSetting = $this->getSetting($context->getId(), $fieldName);
               if (empty($pluginSetting)) {
                  $configurationErrors[] = EXPORT_CONFIG_ERROR_SETTINGS;
                  break;
               }
            }
 
            // Add link actions
            $actions = $this->getExportActions($context);
            $actionNames = array_intersect_key($this->getExportActionNames(), array_flip($actions));
            import('lib.pkp.classes.linkAction.request.NullAction');
            $linkActions = array();
            foreach ($actionNames as $action => $actionName) {
               $linkActions[] = new LinkAction($action, new NullAction(), $actionName);
            }
            $templateMgr = TemplateManager::getManager($request);
            $templateMgr->assign(array(
               'plugin' => $this,
               'actionNames' => $actionNames,
               'configurationErrors' => $configurationErrors,
            ));
            break;
         case 'exportSubmissions':
         case 'exportRepresentations':
            $selectedSubmissions = (array) $request->getUserVar('selectedSubmissions');
            $selectedRepresentations = (array) $request->getUserVar('selectedRepresentations');
            $tab = (string) $request->getUserVar('tab');
            $noValidation = $request->getUserVar('validation') ? false : true;
 
            if (empty($selectedSubmissions) && empty($selectedRepresentations)) {
               fatalError(__('plugins.importexport.common.error.noObjectsSelected'));
            }
            if (!empty($selectedSubmissions)) {
               $objects = $this->getPublishedSubmissions($selectedSubmissions, $context);
               $filter = $this->getSubmissionFilter();
               $objectsFileNamePart = 'preprints';
            } elseif (!empty($selectedRepresentations)) {
               $objects = $this->getArticleGalleys($selectedRepresentations);
               $filter = $this->getRepresentationFilter();
               $objectsFileNamePart = 'galleys';
            }
 
            // Execute export action
            $this->executeExportAction($request, $objects, $filter, $tab, $objectsFileNamePart, $noValidation);
      }
   }
 
   function executeExportAction($request, $objects, $filter, $tab, $objectsFileNamePart, $noValidation = null) {
      $context = $request->getContext();
      $path = array('plugin', $this->getName());
      if ($request->getUserVar(EXPORT_ACTION_EXPORT)) {
         assert($filter != null);
         // Get the XML
         $exportXml = $this->exportXML($objects, $filter, $context, $noValidation);
         import('lib.pkp.classes.file.FileManager');
         $fileManager = new FileManager();
         $exportFileName = $this->getExportFileName($this->getExportPath(), $objectsFileNamePart, $context, '.xml');
         $fileManager->writeFile($exportFileName, $exportXml);
         $fileManager->downloadByPath($exportFileName);
         $fileManager->deleteByPath($exportFileName);
      } elseif ($request->getUserVar(EXPORT_ACTION_DEPOSIT)) {
         assert($filter != null);
         // Get the XML
         $exportXml = $this->exportXML($objects, $filter, $context, $noValidation);
         // Write the XML to a file.
         // export file name example: crossref-20160723-160036-articles-1.xml
         import('lib.pkp.classes.file.FileManager');
         $fileManager = new FileManager();
         $exportFileName = $this->getExportFileName($this->getExportPath(), $objectsFileNamePart, $context, '.xml');
         $fileManager->writeFile($exportFileName, $exportXml);
         // Deposit the XML file.
         $result = $this->depositXML($objects, $context, $exportFileName);
         // send notifications
         if ($result === true) {
            $this->_sendNotification(
               $request->getUser(),
               $this->getDepositSuccessNotificationMessageKey(),
               NOTIFICATION_TYPE_SUCCESS
            );
         } else {
            if (is_array($result)) {
               foreach($result as $error) {
                  assert(is_array($error) && count($error) >= 1);
                  $this->_sendNotification(
                     $request->getUser(),
                     $error[0],
                     NOTIFICATION_TYPE_ERROR,
                     (isset($error[1]) ? $error[1] : null)
                  );
               }
            }
         }
         // Remove all temporary files.
         $fileManager->deleteByPath($exportFileName);
         // redirect back to the right tab
         $request->redirect(null, null, null, $path, null, $tab);
      } elseif ($request->getUserVar(EXPORT_ACTION_MARKREGISTERED)) {
         $this->markRegistered($context, $objects);
         // redirect back to the right tab
         $request->redirect(null, null, null, $path, null, $tab);
      } else {
         $dispatcher = $request->getDispatcher();
         $dispatcher->handle404();
      }
   }
 
   function getDepositSuccessNotificationMessageKey() {
      return 'plugins.importexport.common.register.success';
   }
 
   abstract function depositXML($objects, $context, $filename);
 
   function getStatusMessage($request) {
      return null;
   }
 
   function getSubmissionFilter() {
      return null;
   }
 
   function getRepresentationFilter() {
      return null;
   }
 
   function getStatusNames() {
      return array(
         EXPORT_STATUS_ANY => __('plugins.importexport.common.status.any'),
         EXPORT_STATUS_NOT_DEPOSITED => __('plugins.importexport.common.status.notDeposited'),
         EXPORT_STATUS_MARKEDREGISTERED => __('plugins.importexport.common.status.markedRegistered'),
         EXPORT_STATUS_REGISTERED => __('plugins.importexport.common.status.registered'),
      );
   }
 
   function getStatusActions($pubObject) {
      return array();
   }
 
   function getExportActions($context) {
      $actions = array(EXPORT_ACTION_EXPORT, EXPORT_ACTION_MARKREGISTERED);
      if ($this->getSetting($context->getId(), 'username') && $this->getSetting($context->getId(), 'password')) {
         array_unshift($actions, EXPORT_ACTION_DEPOSIT);
      }
      return $actions;
   }
 
   function getExportActionNames() {
      return array(
         EXPORT_ACTION_DEPOSIT => __('plugins.importexport.common.action.register'),
         EXPORT_ACTION_EXPORT => __('plugins.importexport.common.action.export'),
         EXPORT_ACTION_MARKREGISTERED => __('plugins.importexport.common.action.markRegistered'),
      );
   }
 
   abstract function getExportDeploymentClassName();
 
   function exportXML($objects, $filter, $context, $noValidation = null) {
      $filterDao = DAORegistry::getDAO('FilterDAO'); /* @var $filterDao FilterDAO */
      $exportFilters = $filterDao->getObjectsByGroup($filter);
      assert(count($exportFilters) == 1); // Assert only a single serialization filter
      $exportFilter = array_shift($exportFilters);
      $exportDeployment = $this->_instantiateExportDeployment($context);
      $exportFilter->setDeployment($exportDeployment);
      if ($noValidation) $exportFilter->setNoValidation($noValidation);
      libxml_use_internal_errors(true);
      $exportXml = $exportFilter->execute($objects, true);
      $xml = $exportXml->saveXml();
      $errors = array_filter(libxml_get_errors(), function($a) {
         return $a->level == LIBXML_ERR_ERROR || $a->level == LIBXML_ERR_FATAL;
      });
      if (!empty($errors)) {
         $this->displayXMLValidationErrors($errors, $xml);
      }
      return $xml;
   }
 
   function markRegistered($context, $objects) {
      foreach ($objects as $object) {
         $object->setData($this->getDepositStatusSettingName(), EXPORT_STATUS_MARKEDREGISTERED);
         $this->updateObject($object);
      }
   }
 
   protected function updateObject($object) {
      // Register a hook for the required additional
      // object fields. We do this on a temporary
      // basis as the hook adds a performance overhead
      // and the field will "stealthily" survive even
      // when the DAO does not know about it.
      $dao = $object->getDAO();
      $dao->updateObject($object);
   }
 
   function getAdditionalFieldNames($hookName, $args) {
      assert(count($args) == 2);
      $additionalFields =& $args[1];
      assert(is_array($additionalFields));
      foreach ($this->_getObjectAdditionalSettings() as $fieldName) {
         $additionalFields[] = $fieldName;
      }
      
      return false;
   }
 
   public function addToSchema($hookName, $params) {
      $schema =& $params[0];
      foreach ($this->_getObjectAdditionalSettings() as $fieldName) {
         $schema->properties->{$fieldName} = (object) [
            'type' => 'string',
            'apiSummary' => true,
            'validation' => ['nullable'],
         ];
      }
 
      return false;
   }
 
   protected function _getObjectAdditionalSettings() {
      return array($this->getDepositStatusSettingName());
   }
 
   function callbackParseCronTab($hookName, $args) {
      $taskFilesPath =& $args[0];
      $taskFilesPath[] = $this->getPluginPath() . DIRECTORY_SEPARATOR . 'scheduledTasks.xml';
      return false;
   }
 
   function getUnregisteredArticles($context) {
      // Retrieve all published submissions that have not yet been registered.
      $submissionDao = DAORegistry::getDAO('SubmissionDAO'); /* @var $submissionDao SubmissionDAO */
      $articles = $submissionDao->getExportable(
         $context->getId(),
         null,
         null,
         null,
         null,
         $this->getDepositStatusSettingName(),
         EXPORT_STATUS_NOT_DEPOSITED,
         null
      );
      return $articles->toArray();
   }
   function isTestMode($context) {
      return ($this->getSetting($context->getId(), 'testMode') == 1);
   }
 
   function getDepositStatusSettingName() {
      return $this->getPluginSettingsPrefix().'::status';
   }
 
 
 
   function usage($scriptName) {
      echo __(
         'plugins.importexport.' . $this->getPluginSettingsPrefix() . '.cliUsage',
         array(
            'scriptName' => $scriptName,
            'pluginName' => $this->getName()
         )
      ) . "\n";
   }
 
   function executeCLI($scriptName, &$args) {
      AppLocale::requireComponents(LOCALE_COMPONENT_APP_MANAGER);
 
      $command = array_shift($args);
      if (!in_array($command, array('export', 'register'))) {
         $this->usage($scriptName);
         return;
      }
 
      $outputFile = $command == 'export' ? array_shift($args) : null;
      $contextPath = array_shift($args);
      $objectType = array_shift($args);
 
      $contextDao = DAORegistry::getDAO('JournalDAO');
      $context = $contextDao->getByPath($contextPath);
      if (!$context) {
         if ($contextPath != '') {
            echo __('plugins.importexport.common.cliError') . "\n";
            echo __('plugins.importexport.common.error.unknownJournal', array('journalPath' => $contextPath)) . "\n\n";
         }
         $this->usage($scriptName);
         return;
      }
 
      if ($outputFile) {
         if ($this->isRelativePath($outputFile)) {
            $outputFile = PWD . '/' . $outputFile;
         }
         $outputDir = dirname($outputFile);
         if (!is_writable($outputDir) || (file_exists($outputFile) && !is_writable($outputFile))) {
            echo __('plugins.importexport.common.cliError') . "\n";
            echo __('plugins.importexport.common.export.error.outputFileNotWritable', array('param' => $outputFile)) . "\n\n";
            $this->usage($scriptName);
            return;
         }
      }
 
      switch ($objectType) {
         case 'articles':
            $objects = $this->getPublishedSubmissions($args, $context);
            $filter = $this->getSubmissionFilter();
            $objectsFileNamePart = 'preprints';
            break;
         case 'galleys':
            $objects = $this->getArticleGalleys($args);
            $filter = $this->getRepresentationFilter();
            $objectsFileNamePart = 'galleys';
            break;
         default:
            $this->usage($scriptName);
            return;
 
      }
      if (empty($objects)) {
         echo __('plugins.importexport.common.cliError') . "\n";
         echo __('plugins.importexport.common.error.unknownObjects') . "\n\n";
         $this->usage($scriptName);
         return;
      }
      if (!$filter) {
         $this->usage($scriptName);
         return;
      }
 
      $this->executeCLICommand($scriptName, $command, $context, $outputFile, $objects, $filter, $objectsFileNamePart);
      return;
   }
 
   function executeCLICommand($scriptName, $command, $context, $outputFile, $objects, $filter, $objectsFileNamePart) {
      $exportXml = $this->exportXML($objects, $filter, $context);
      if ($command == 'export' && $outputFile) file_put_contents($outputFile, $exportXml);
 
      if ($command == 'register') {
         import('lib.pkp.classes.file.FileManager');
         $fileManager = new FileManager();
         $exportFileName = $this->getExportFileName($this->getExportPath(), $objectsFileNamePart, $context, '.xml');
         $fileManager->writeFile($exportFileName, $exportXml);
         $result = $this->depositXML($objects, $context, $exportFileName);
         if ($result === true) {
            echo __('plugins.importexport.common.register.success') . "\n";
         } else {
            echo __('plugins.importexport.common.cliError') . "\n";
            if (is_array($result)) {
               foreach($result as $error) {
                  assert(is_array($error) && count($error) >= 1);
                  $errorMessage = __($error[0], array('param' => (isset($error[1]) ? $error[1] : null)));
                  echo "*** $errorMessage\n";
               }
               echo "\n";
            } else {
               echo __('plugins.importexport.common.register.error.mdsError', array('param' => ' - ')) . "\n\n";
            }
            $this->usage($scriptName);
         }
         $fileManager->deleteByPath($exportFileName);
      }
   }
 
   function getPublishedSubmissions($submissionIds, $context) {
      $submissions = array_map(function($submissionId) {
         return Services::get('submission')->get($submissionId);
      }, $submissionIds);
      return array_filter($submissions, function($submission) {
         return $submission->getData('status') === STATUS_PUBLISHED;
      });
   }
 
   function getArticleGalleys($galleyIds) {
      $galleys = array();
      $articleGalleyDao = DAORegistry::getDAO('ArticleGalleyDAO');
      foreach ($galleyIds as $galleyId) {
         $articleGalley = $articleGalleyDao->getById($galleyId);
         if ($articleGalley) $galleys[] = $articleGalley;
      }
      return $galleys;
   }
 
   function _sendNotification($user, $message, $notificationType, $param = null) {
      static $notificationManager = null;
      if (is_null($notificationManager)) {
         import('classes.notification.NotificationManager');
         $notificationManager = new NotificationManager();
      }
      if (!is_null($param)) {
         $params = array('param' => $param);
      } else {
         $params = null;
      }
      $notificationManager->createTrivialNotification(
            $user->getId(),
            $notificationType,
            array('contents' => __($message, $params))
            );
   }
 
   function _instantiateExportDeployment($context) {
      $exportDeploymentClassName = $this->getExportDeploymentClassName();
      $this->import($exportDeploymentClassName);
      $exportDeployment = new $exportDeploymentClassName($context, $this);
      return $exportDeployment;
   }
 
   function _instantiateSettingsForm($context) {
      $settingsFormClassName = $this->getSettingsFormClassName();
      $this->import('classes.form.' . $settingsFormClassName);
      $settingsForm = new $settingsFormClassName($this, $context->getId());
      return $settingsForm;
   }
 
   protected function _getDAOs() {
      return array(
         DAORegistry::getDAO('PublicationDAO'),
         DAORegistry::getDAO('SubmissionDAO'),
         Application::getRepresentationDAO(),
         DAORegistry::getDAO('SubmissionFileDAO'),
      );
   }
}