View source
<?php
namespace Drupal\search_api_synonym\Import;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\Entity\EntityRepositoryInterface;
use Drupal\Core\Extension\ModuleHandlerInterface;
use Drupal\Core\Datetime\DrupalDateTime;
use Drupal\Core\Database\Connection;
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
use Drupal\Core\StringTranslation\TranslationInterface;
use Drupal\search_api_synonym\Entity\Synonym;
use Drupal\Core\Messenger\MessengerInterface;
use Symfony\Component\DependencyInjection\ContainerInterface;
class Importer {
protected $entityManager;
protected $entityRepository;
protected $moduleHandler;
protected $connection;
public function __construct() {
$this->entityManager = \Drupal::service('entity_type.manager');
$this->entityRepository = \Drupal::service('entity_type.repository');
$this->moduleHandler = \Drupal::service('module_handler');
$this->connection = \Drupal::service('database');
}
public function execute(array $items, array $settings) {
$items = $this
->prepare($items, $settings);
$results = $this
->createSynonyms($items, $settings);
return $results;
}
private function prepare(array $items, array $settings) {
$prepared = [];
foreach ($items as $item) {
if ($settings['synonym_type'] != 'mixed') {
$type = $settings['synonym_type'];
}
else {
$type = !empty($item['type']) ? $item['type'] : 'empty';
}
$prepared[$type][$item['word']][] = $item['synonym'];
}
return $prepared;
}
public function createSynonyms(array $items, array $settings) {
$context = [];
$operations = [];
foreach ($items as $type => $item) {
if ($type == 'empty') {
$context['results']['errors'][] = [
'word' => key($item),
'synonyms' => current($item),
];
continue;
}
foreach ($item as $word => $synonyms) {
$operations[] = [
'\\Drupal\\search_api_synonym\\Import\\Importer::createSynonym',
[
$word,
$synonyms,
$type,
$settings,
],
];
}
}
$batch = [
'title' => t('Import synonyms...'),
'operations' => $operations,
'finished' => '\\Drupal\\search_api_synonym\\Import\\Importer::createSynonymBatchFinishedCallback',
];
batch_set($batch);
return isset($context['results']) ? $context['results'] : NULL;
}
public static function createSynonym($word, array $synonyms, $type, array $settings, array &$context) {
$request_time = \Drupal::time()
->getRequestTime();
$sid = Importer::lookUpSynonym($word, $type, $settings['langcode']);
$synonyms = array_map('trim', $synonyms);
if ($sid) {
$entity = Synonym::load($sid);
if ($settings['update_existing'] == 'merge') {
$existing = $entity
->getSynonyms();
$existing = array_map('trim', explode(',', $existing));
$synonyms = array_unique(array_merge($existing, $synonyms));
}
$synonyms_str = implode(',', $synonyms);
$entity
->setSynonyms($synonyms_str);
}
else {
$entity = Synonym::create([
'langcode' => $settings['langcode'],
]);
$uid = \Drupal::currentUser()
->id();
$entity
->setOwnerId($uid);
$entity
->setCreatedTime($request_time);
$entity
->setType($type);
$entity
->setWord($word);
$synonyms_str = implode(',', $synonyms);
$entity
->setSynonyms($synonyms_str);
}
$entity
->setChangedTime($request_time);
$entity
->setActive($settings['status']);
$entity
->save();
if ($sid = $entity
->id()) {
$context['results']['success'][] = $sid;
}
else {
$context['results']['errors'][] = [
'word' => $word,
'synonyms' => $synonyms,
];
}
}
public static function createSynonymBatchFinishedCallback($success, $result, $operations, $elapsed) {
if ($success) {
if (!empty($result['success'])) {
$count = count($result['success']);
$message = \Drupal::translation()
->formatPlural($count, '@count synonym was successfully imported.', '@count synonyms was successfully imported.', [
'@count' => $count,
]);
\Drupal::service('messenger')
->addMessage($message);
}
}
}
public static function lookUpSynonym($word, $type, $langcode) {
$query = \Drupal::database()
->select('search_api_synonym', 's');
$query
->fields('s', [
'sid',
]);
$query
->condition('s.type', $type);
$query
->condition('s.word', $word, 'LIKE');
$query
->condition('s.langcode', $langcode);
$query
->range(0, 1);
return (int) $query
->execute()
->fetchField(0);
}
}