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\Tools\Utils;
-
-class WallabagV1Import implements ImportInterface
+class WallabagV1Import extends WallabagImport
{
- private $user;
- private $em;
- private $logger;
- private $skippedEntries = 0;
- private $importedEntries = 0;
- private $filepath;
+ protected $fetchingErrorMessage;
+ protected $fetchingErrorMessageTitle;
- public function __construct(EntityManager $em)
+ public function __construct($em, $contentProxy, $tagsAssigner, $eventDispatcher, $fetchingErrorMessageTitle, $fetchingErrorMessage)
{
- $this->em = $em;
- $this->logger = new NullLogger();
- }
+ $this->fetchingErrorMessageTitle = $fetchingErrorMessageTitle;
+ $this->fetchingErrorMessage = $fetchingErrorMessage;
- public function setLogger(LoggerInterface $logger)
- {
- $this->logger = $logger;
+ parent::__construct($em, $contentProxy, $tagsAssigner, $eventDispatcher);
}
/**
- * 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
+ * {@inheritdoc}
*/
- public function setUser(User $user)
+ public function getName()
{
- $this->user = $user;
-
- return $this;
+ return 'wallabag v1';
}
/**
* {@inheritdoc}
*/
- public function getName()
+ public function getUrl()
{
- return 'Wallabag v1';
+ return 'import_wallabag_v1';
}
/**
*/
public function getDescription()
{
- return 'This importer will import all your wallabag v1 articles.';
+ return 'import.wallabag_v1.description';
}
/**
* {@inheritdoc}
*/
- public function import()
+ protected function prepareEntry($entry = [])
{
- if (!$this->user) {
- $this->logger->error('WallabagV1Import: user is not defined');
+ $data = [
+ 'title' => $entry['title'],
+ 'html' => $entry['content'],
+ 'url' => $entry['url'],
+ 'is_archived' => $entry['is_read'] || $this->markAsRead,
+ 'is_starred' => $entry['is_fav'],
+ 'tags' => '',
+ 'created_at' => '',
+ ];
- return false;
+ // In case of a bad fetch in v1, replace title and content with v2 error strings
+ // If fetching fails again, they will get this instead of the v1 strings
+ if (\in_array($entry['title'], $this->untitled, true)) {
+ $data['title'] = $this->fetchingErrorMessageTitle;
+ $data['html'] = $this->fetchingErrorMessage;
}
- if (!file_exists($this->filepath) || !is_readable($this->filepath)) {
- $this->logger->error('WallabagV1Import: unable to read file', array('filepath' => $this->filepath));
-
- return false;
+ if (array_key_exists('tags', $entry) && '' !== $entry['tags']) {
+ $data['tags'] = $entry['tags'];
}
- $this->parseEntries(json_decode(file_get_contents($this->filepath), true));
-
- return true;
+ return $data;
}
/**
* {@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)
+ protected function setEntryAsRead(array $importedEntry)
{
- $this->filepath = $filepath;
-
- return $this;
- }
-
- /**
- * @param $entries
- */
- private function parseEntries($entries)
- {
- foreach ($entries as $importedEntry) {
- $existingEntry = $this->em
- ->getRepository('WallabagCoreBundle:Entry')
- ->existByUrlAndUserId($importedEntry['url'], $this->user->getId());
-
- if (false !== $existingEntry) {
- ++$this->skippedEntries;
- continue;
- }
-
- // @see ContentProxy->updateEntry
- $entry = new Entry($this->user);
- $entry->setUrl($importedEntry['url']);
- $entry->setTitle($importedEntry['title']);
- $entry->setArchived($importedEntry['is_read']);
- $entry->setStarred($importedEntry['is_fav']);
- $entry->setContent($importedEntry['content']);
- $entry->setReadingTime(Utils::getReadingTime($importedEntry['content']));
- $entry->setDomainName(parse_url($importedEntry['url'], PHP_URL_HOST));
-
- $this->em->persist($entry);
- ++$this->importedEntries;
- }
+ $importedEntry['is_read'] = 1;
- $this->em->flush();
+ return $importedEntry;
}
}