silverstripe / deploynaut

		};

		// Restore database into target environment

		if(in_array($dataTransfer->Mode, array('all', 'db'))) {

			$log->write(sprintf('Restore of database to "%s" started', $name));

			$args = array('data_path' => $workingDir . DIRECTORY_SEPARATOR . 'database.sql');

			$command = $this->getCommand('data:pushdb', 'db', $environment, $args, $log);

			$command->run(function($type, $buffer) use($log) {

				$log->write($buffer);

			});

			if(!$command->isSuccessful()) {

				$cleanupFn();

				$log->write(sprintf('Restore of database to "%s" failed: %s', $name, $command->getErrorOutput()));

				$this->extend('dataTransferFailure', $environment, $log);

				throw new RuntimeException($command->getErrorOutput());

			}

			$log->write(sprintf('Restore of database to "%s" done', $name));

		}

		// Restore assets into target environment

		if(in_array($dataTransfer->Mode, array('all', 'assets'))) {

		}

		// Restore assets into target environment

		if(in_array($dataTransfer->Mode, array('all', 'assets'))) {

			$log->write(sprintf('Restore of assets to "%s" started', $name));

			$args = array('data_path' => $workingDir . DIRECTORY_SEPARATOR . 'assets');

			$command = $this->getCommand('data:pushassets', 'web', $environment, $args, $log);

			$command->run(function($type, $buffer) use($log) {

				$log->write($buffer);

			});

			if(!$command->isSuccessful()) {

				$cleanupFn();

				$log->write(sprintf('Restore of assets to "%s" failed: %s', $name, $command->getErrorOutput()));

				$this->extend('dataTransferFailure', $environment, $log);

				throw new RuntimeException($command->getErrorOutput());

			}

			$log->write(sprintf('Restore of assets to "%s" done', $name));

		}

		$log->write('Rebuilding and cleaning up');

		$cleanupFn();