namespace Wallabag\ImportBundle\Import;
use Wallabag\CoreBundle\Entity\Entry;
-use Wallabag\UserBundle\Entity\User;
abstract class WallabagImport extends AbstractImport
{
- protected $user;
- protected $skippedEntries = 0;
- protected $importedEntries = 0;
protected $filepath;
- protected $markAsRead;
// untitled in all languages from v1
protected $untitled = [
'Untitled',
'',
];
- /**
- * 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}
*/
return false;
}
+ if ($this->producer) {
+ $this->parseEntriesForProducer($data);
+
+ return true;
+ }
+
$this->parseEntries($data);
return true;
}
/**
- * Set whether articles must be all marked as read.
- *
- * @param bool $markAsRead
+ * {@inheritdoc}
*/
- public function setMarkAsRead($markAsRead)
+ public function parseEntry(array $importedEntry)
{
- $this->markAsRead = $markAsRead;
+ $existingEntry = $this->em
+ ->getRepository('WallabagCoreBundle:Entry')
+ ->findByUrlAndUserId($importedEntry['url'], $this->user->getId());
- return $this;
- }
+ if (false !== $existingEntry) {
+ ++$this->skippedEntries;
- /**
- * Parse and insert all given entries.
- *
- * @param $entries
- */
- protected function parseEntries($entries)
- {
- $i = 1;
+ return;
+ }
+
+ $data = $this->prepareEntry($importedEntry);
- foreach ($entries as $importedEntry) {
- $existingEntry = $this->em
- ->getRepository('WallabagCoreBundle:Entry')
- ->findByUrlAndUserId($importedEntry['url'], $this->user->getId());
+ $entry = $this->fetchContent(
+ new Entry($this->user),
+ $importedEntry['url'],
+ $data
+ );
- if (false !== $existingEntry) {
- ++$this->skippedEntries;
- continue;
- }
+ // jump to next entry in case of problem while getting content
+ if (false === $entry) {
+ ++$this->skippedEntries;
- $data = $this->prepareEntry($importedEntry, $this->markAsRead);
+ return;
+ }
- $entry = $this->fetchContent(
- new Entry($this->user),
- $importedEntry['url'],
- $data
+ if (array_key_exists('tags', $data)) {
+ $this->contentProxy->assignTagsToEntry(
+ $entry,
+ $data['tags']
);
+ }
- // jump to next entry in case of problem while getting content
- if (false === $entry) {
- ++$this->skippedEntries;
- continue;
- }
-
- if (array_key_exists('tags', $data)) {
- $this->contentProxy->assignTagsToEntry(
- $entry,
- $data['tags']
- );
- }
-
- if (isset($importedEntry['preview_picture'])) {
- $entry->setPreviewPicture($importedEntry['preview_picture']);
- }
-
- $entry->setArchived($data['is_archived']);
- $entry->setStarred($data['is_starred']);
-
- $this->em->persist($entry);
- ++$this->importedEntries;
-
- // flush every 20 entries
- if (($i % 20) === 0) {
- $this->em->flush();
- $this->em->clear($entry);
- }
- ++$i;
+ if (isset($importedEntry['preview_picture'])) {
+ $entry->setPreviewPicture($importedEntry['preview_picture']);
}
- $this->em->flush();
+ $entry->setArchived($data['is_archived']);
+ $entry->setStarred($data['is_starred']);
+
+ $this->em->persist($entry);
+ ++$this->importedEntries;
+
+ return $entry;
}
/**
* This should return a cleaned array for a given entry to be given to `updateEntry`.
*
- * @param array $entry Data from the imported file
- * @param bool $markAsRead Should we mark as read content?
+ * @param array $entry Data from the imported file
*
* @return array
*/
- abstract protected function prepareEntry($entry = [], $markAsRead = false);
+ abstract protected function prepareEntry($entry = []);
}