namespace Wallabag\ImportBundle\Import;
-use Psr\Log\LoggerInterface;
-use Psr\Log\NullLogger;
-use Doctrine\ORM\EntityManager;
-use Wallabag\CoreBundle\Entity\Entry;
-use Wallabag\UserBundle\Entity\User;
-use Wallabag\CoreBundle\Helper\ContentProxy;
-
-class WallabagV1Import implements ImportInterface
+class WallabagV1Import extends WallabagImport
{
- protected $user;
- protected $em;
- protected $logger;
- protected $contentProxy;
- protected $skippedEntries = 0;
- protected $importedEntries = 0;
- protected $filepath;
- protected $markAsRead;
-
- public function __construct(EntityManager $em, ContentProxy $contentProxy)
- {
- $this->em = $em;
- $this->logger = new NullLogger();
- $this->contentProxy = $contentProxy;
- }
-
- public function setLogger(LoggerInterface $logger)
- {
- $this->logger = $logger;
- }
-
- /**
- * We define the user in a custom call because on the import command there is no logged in user.
- * So we can't retrieve user from the `security.token_storage` service.
- *
- * @param User $user
- */
- public function setUser(User $user)
- {
- $this->user = $user;
-
- return $this;
- }
-
/**
* {@inheritdoc}
*/
/**
* {@inheritdoc}
*/
- public function import()
+ protected function prepareEntry($entry = [])
{
- if (!$this->user) {
- $this->logger->error('WallabagImport: user is not defined');
-
- return false;
- }
-
- if (!file_exists($this->filepath) || !is_readable($this->filepath)) {
- $this->logger->error('WallabagImport: unable to read file', array('filepath' => $this->filepath));
+ $data = [
+ 'title' => $entry['title'],
+ 'html' => $entry['content'],
+ 'url' => $entry['url'],
+ 'content_type' => '',
+ 'language' => '',
+ 'is_archived' => $entry['is_read'] || $this->markAsRead,
+ 'is_starred' => $entry['is_fav'],
+ 'tags' => '',
+ ];
- return false;
+ // force content to be refreshed in case on bad fetch in the v1 installation
+ if (in_array($entry['title'], $this->untitled)) {
+ $data['title'] = '';
+ $data['html'] = '';
}
- $data = json_decode(file_get_contents($this->filepath), true);
-
- if (empty($data)) {
- return false;
+ if (array_key_exists('tags', $entry) && $entry['tags'] != '') {
+ $data['tags'] = $entry['tags'];
}
- $this->parseEntries($data);
-
- return true;
- }
-
- /**
- * {@inheritdoc}
- */
- public function getSummary()
- {
- return [
- 'skipped' => $this->skippedEntries,
- 'imported' => $this->importedEntries,
- ];
- }
-
- /**
- * Set file path to the json file.
- *
- * @param string $filepath
- */
- public function setFilepath($filepath)
- {
- $this->filepath = $filepath;
-
- return $this;
- }
-
- /**
- * Set whether articles must be all marked as read.
- *
- * @param bool $markAsRead
- */
- public function setMarkAsRead($markAsRead)
- {
- $this->markAsRead = $markAsRead;
-
- return $this;
+ return $data;
}
- /**
- * @param $entries
- */
- protected function parseEntries($entries)
+ protected function parseEntriesForProducer($entries)
{
- $i = 1;
-
- //Untitled in all languages from v1. This should never have been translated
- $untitled = array('Untitled', 'Sans titre', 'podle nadpisu', 'Sin título', 'با عنوان', 'per titolo', 'Sem título', 'Без названия', 'po naslovu', 'Без назви', 'No title found', '');
-
foreach ($entries as $importedEntry) {
- $existingEntry = $this->em
- ->getRepository('WallabagCoreBundle:Entry')
- ->findByUrlAndUserId($importedEntry['url'], $this->user->getId());
-
- if (false !== $existingEntry) {
- ++$this->skippedEntries;
- continue;
- }
-
- $data = [
- 'title' => $importedEntry['title'],
- 'html' => $importedEntry['content'],
- 'url' => $importedEntry['url'],
- 'content_type' => '',
- 'language' => '',
- ];
+ // set userId for the producer (it won't know which user is connected)
+ $importedEntry['userId'] = $this->user->getId();
- // force content to be refreshed in case on bad fetch in the v1 installation
- if (in_array($importedEntry['title'], $untitled)) {
- $data = [];
+ if ($this->markAsRead) {
+ $importedEntry['is_read'] = 1;
}
- $entry = $this->contentProxy->updateEntry(
- new Entry($this->user),
- $importedEntry['url'],
- $data
- );
-
- if (array_key_exists('tags', $importedEntry) && $importedEntry['tags'] != '') {
- $this->contentProxy->assignTagsToEntry(
- $entry,
- $importedEntry['tags']
- );
- }
-
- $entry->setArchived($importedEntry['is_read'] || $this->markAsRead);
- $entry->setStarred($importedEntry['is_fav']);
-
- $this->em->persist($entry);
++$this->importedEntries;
- // flush every 20 entries
- if (($i % 20) === 0) {
- $this->em->flush();
- }
- ++$i;
+ $this->producer->publish(json_encode($importedEntry));
}
-
- $this->em->flush();
}
}