public function testInterceptors()
{
$itc = new DummyInterceptor();
$interceptorsProvider = \Phake::mock('Modera\\FileRepositoryBundle\\Intercepting\\InterceptorsProviderInterface');
\Phake::when($interceptorsProvider)->getInterceptors($this->isInstanceOf(Repository::clazz()))->thenReturn([$itc]);
$container = \Phake::mock('Symfony\\Component\\DependencyInjection\\ContainerInterface');
\Phake::when($container)->get('modera_file_repository.intercepting.interceptors_provider')->thenReturn($interceptorsProvider);
$repository = new Repository('foo', array('filesystem' => 'foo'));
$repository->init($container);
$splFile = new \SplFileInfo(__FILE__);
$storedFile = \Phake::mock(StoredFile::clazz());
// ---
$this->assertEquals(0, count($itc->beforePutInvocations));
$repository->beforePut($splFile);
$this->assertEquals(1, count($itc->beforePutInvocations));
$this->assertSame($splFile, $itc->beforePutInvocations[0][0]);
$this->assertSame($repository, $itc->beforePutInvocations[0][1]);
// ---
$receivedInterceptor = null;
$repository->beforePut($splFile, function ($interceptor) use(&$receivedInterceptor) {
$receivedInterceptor = $interceptor;
return false;
});
$this->assertSame($itc, $receivedInterceptor);
$this->assertEquals(1, count($itc->beforePutInvocations));
// ---
$this->assertEquals(0, count($itc->onPutInvocations));
$repository->onPut($storedFile, $splFile);
$this->assertEquals(1, count($itc->onPutInvocations));
$this->assertSame($storedFile, $itc->onPutInvocations[0][0]);
$this->assertSame($splFile, $itc->onPutInvocations[0][1]);
$this->assertSame($repository, $itc->onPutInvocations[0][2]);
// ---
$receivedInterceptor = null;
$repository->onPut($storedFile, $splFile, function ($interceptor) use(&$receivedInterceptor) {
$receivedInterceptor = $interceptor;
return false;
});
$this->assertSame($itc, $receivedInterceptor);
$this->assertEquals(1, count($itc->onPutInvocations));
// ---
$this->assertEquals(0, count($itc->afterPutInvocations));
$repository->afterPut($storedFile, $splFile);
$this->assertEquals(1, count($itc->afterPutInvocations));
$this->assertSame($storedFile, $itc->afterPutInvocations[0][0]);
$this->assertSame($splFile, $itc->afterPutInvocations[0][1]);
$this->assertSame($repository, $itc->afterPutInvocations[0][2]);
$receivedInterceptor = null;
$repository->afterPut($storedFile, $splFile, function ($interceptor) use(&$receivedInterceptor) {
$receivedInterceptor = $interceptor;
return false;
});
$this->assertSame($itc, $receivedInterceptor);
$this->assertEquals(1, count($itc->afterPutInvocations));
}
/**
* {@inheritdoc}
*/
protected function execute(InputInterface $input, OutputInterface $output)
{
/* @var EntityManager $em */
$em = $this->getContainer()->get('doctrine.orm.entity_manager');
/* @var StoredFile $storedFile */
$storedFile = $em->find(StoredFile::clazz(), $input->getArgument('file_id'));
if (!$storedFile) {
throw new \RuntimeException('Unable to find a file with ID ' . $input->getArgument('file_id'));
}
$output->writeln(sprintf('Deleting file "%s" from repository "%s"', $storedFile->getFilename(), $storedFile->getRepository()->getName()));
$em->remove($storedFile);
$em->flush();
$output->writeln('<info>Done!</info>');
}
/**
* {@inheritdoc}
*/
protected function execute(InputInterface $input, OutputInterface $output)
{
/* @var EntityManager $em */
$em = $this->getContainer()->get('doctrine.orm.entity_manager');
/* @var StoredFile $storedFile */
$storedFile = $em->getRepository(StoredFile::clazz())->find($input->getArgument('file_id'));
if (!$storedFile) {
throw new \RuntimeException(sprintf('Unable to find a file with ID "%s".', $input->getArgument('file_id')));
}
$localPath = $input->getArgument('local_path');
$output->writeln('Downloading the file ...');
ob_start();
$result = file_put_contents($localPath, $storedFile->getContents());
$errorOutput = ob_get_clean();
if (false !== $result) {
$output->writeln(sprintf('<info>File from repository "%s" has been successfully downloaded and stored locally at %s</info>', $storedFile->getRepository()->getName(), $localPath));
} else {
$output->writeln('<error>Something went wrong, we were unable to save a file locally: </error>');
$output->writeln($errorOutput);
}
}
private function createStoredFile($storageKey, $content)
{
if ($storageKey) {
$parts = explode('/', $storageKey);
if (count($parts) > 1) {
$filename = $parts[count($parts) - 1];
} else {
$filename = 'foo.txt';
}
list($name, $extension) = explode('.', $filename);
$mimeType = array('txt' => 'text/plain');
$storedFile = \Phake::mock(StoredFile::clazz());
\Phake::when($storedFile)->getStorageKey()->thenReturn($parts[0]);
\Phake::when($storedFile)->getFilename()->thenReturn($filename);
\Phake::when($storedFile)->getMimeType()->thenReturn($mimeType[$extension]);
\Phake::when($storedFile)->getExtension()->thenReturn($extension);
\Phake::when($storedFile)->getCreatedAt()->thenReturn(new \DateTime());
\Phake::when($storedFile)->getContents()->thenReturn($content);
\Phake::when($storedFile)->getSize()->thenReturn(strlen($content));
return $storedFile;
}
return;
}
/**
* {@inheritdoc}
*/
public static function doTearDownAfterClass()
{
self::$st->dropSchema(array(self::$em->getClassMetadata(Repository::clazz()), self::$em->getClassMetadata(StoredFile::clazz())));
}
public function testHowWellItWorks()
{
/* @var FileRepository $fr */
$fr = self::$container->get('modera_file_repository.repository.file_repository');
$this->assertNull($fr->getRepository('dummy_repository'));
$repositoryConfig = array('storage_key_generator' => 'modera_file_repository.repository.uniqid_key_generator', 'filesystem' => 'dummy_tmp_fs');
$this->assertFalse($fr->repositoryExists('dummy_repository'));
$repository = $fr->createRepository('dummy_repository', $repositoryConfig, 'My dummy repository');
$this->assertTrue($fr->repositoryExists('dummy_repository'));
$this->assertInstanceOf(Repository::clazz(), $repository);
$this->assertNotNull($repository->getId());
$this->assertEquals('dummy_repository', $repository->getName());
$this->assertEquals('My dummy repository', $repository->getLabel());
$this->assertSame($repositoryConfig, $repository->getConfig());
$this->assertInstanceOf('Symfony\\Component\\DependencyInjection\\ContainerInterface', Toolkit::getPropertyValue($repository, 'container'));
// ---
$fileContents = 'foo contents';
$filePath = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'our-dummy-file.txt';
file_put_contents($filePath, $fileContents);
$file = new File($filePath);
$storedFile = $fr->put($repository->getName(), $file, array());
self::$em->clear();
// this way we will make sure that data is actually persisted in database
/* @var StoredFile $storedFile */
$storedFile = self::$em->find(StoredFile::clazz(), $storedFile->getId());
$this->assertInstanceOf(StoredFile::clazz(), $storedFile);
$this->assertNotNull($storedFile->getId());
$this->assertNotNull($storedFile->getStorageKey());
$this->assertEquals('our-dummy-file.txt', $storedFile->getFilename());
$this->assertNotNull($storedFile->getCreatedAt());
$this->assertEquals('txt', $storedFile->getExtension());
$this->assertEquals('text/plain', $storedFile->getMimeType());
$this->assertSame($repository->getId(), $storedFile->getRepository()->getId());
$this->assertEquals($fileContents, $storedFile->getContents());
$this->assertTrue('' != $storedFile->getChecksum());
$this->assertEquals($file->getSize(), $storedFile->getSize());
// ---
$storedFileData = array('id' => $storedFile->getId(), 'storageKey' => $storedFile->getStorageKey(), 'filename' => $storedFile->getFilename());
$fileContents = 'bar contents';
$filePath = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'bar-dummy-file.txt';
file_put_contents($filePath, $fileContents);
$file = new File($filePath);
$storedFile = $fr->put($repository->getName(), $file, array());
self::$em->clear();
// this way we will make sure that data is actually persisted in database
/* @var StoredFile $storedFile */
$storedFile = self::$em->find(StoredFile::clazz(), $storedFile->getId());
$this->assertNotEquals($storedFileData['id'], $storedFile->getId());
$this->assertNotEquals($storedFileData['storageKey'], $storedFile->getStorageKey());
$this->assertNotEquals($storedFileData['filename'], $storedFile->getFilename());
// ---
$repositoryConfig['overwrite_files'] = true;
$repository = $storedFile->getRepository();
$repository->setConfig($repositoryConfig);
self::$em->persist($repository);
self::$em->flush();
$storedFileData = array('id' => $storedFile->getId(), 'storageKey' => $storedFile->getStorageKey(), 'filename' => $storedFile->getFilename());
$storedFile = $fr->put($repository->getName(), $file, array());
self::$em->clear();
// this way we will make sure that data is actually persisted in database
/* @var StoredFile $storedFile */
$storedFile = self::$em->find(StoredFile::clazz(), $storedFile->getId());
$this->assertEquals($storedFileData['id'], $storedFile->getId());
$this->assertEquals($storedFileData['storageKey'], $storedFile->getStorageKey());
$this->assertEquals($storedFileData['filename'], $storedFile->getFilename());
// ---
$fs = $storedFile->getRepository()->getFilesystem();
$this->assertTrue($fs->has($storedFile->getStorageKey()));
self::$em->remove($storedFile);
self::$em->flush();
$this->assertFalse($fs->has($storedFile->getStorageKey()));
}
/**
* @throws \RuntimeException
*
* @param $repositoryName
* @param \SplFileInfo $file
* @param array $context
*
* @return \Modera\FileRepositoryBundle\Entity\StoredFile
*/
public function put($repositoryName, \SplFileInfo $file, array $context = array())
{
$repository = $this->getRepository($repositoryName);
if (!$repository) {
throw new \RuntimeException("Unable to find repository '{$repositoryName}'.");
}
$config = $repository->getConfig();
if (!$this->isInterceptorDisabled($context, 'before')) {
$repository->beforePut($file, $this->createInterceptorsFilter($context, 'before'));
}
$storedFile = null;
$overwrite = isset($config['overwrite_files']) ? $config['overwrite_files'] : false;
if ($overwrite) {
$filename = $file->getFilename();
if ($file instanceof UploadedFile) {
$filename = $file->getClientOriginalName();
}
/* @var StoredFile $storedFile */
$storedFile = $this->em->getRepository(StoredFile::clazz())->findOneBy(array('repository' => $repository->getId(), 'filename' => $filename));
if ($storedFile) {
$storedFile->setCreatedAt(new \DateTime('now'));
} else {
$overwrite = false;
}
}
if (!$storedFile) {
$storedFile = $repository->createFile($file, $context);
}
$contents = @file_get_contents($file->getPathname());
if (false === $contents) {
throw new \RuntimeException(sprintf('Unable to read contents of "%s" file!', $file->getPath()));
}
if (!$this->isInterceptorDisabled($context, 'put')) {
$repository->onPut($storedFile, $file, $this->createInterceptorsFilter($context, 'put'));
}
$storageKey = $storedFile->getStorageKey();
/* @var Filesystem $fs */
$fs = $repository->getFilesystem();
// physically stored file
$fs->write($storageKey, $contents, $overwrite);
try {
$this->em->persist($storedFile);
$this->em->flush($storedFile);
} catch (\Exception $e) {
if (!$storedFile->getId()) {
$fs->delete($storageKey);
}
throw $e;
}
if (!$this->isInterceptorDisabled($context, 'after')) {
$repository->afterPut($storedFile, $file, $this->createInterceptorsFilter($context, 'after'));
}
return $storedFile;
}
/**
* @param string $storageKey
*
* @return null|StoredFile
*/
protected function getFile($storageKey)
{
/* @var ObjectRepository $repository */
$repository = $this->getDoctrine()->getManager()->getRepository(StoredFile::clazz());
return $repository->findOneBy(array('storageKey' => $storageKey));
}