Your IP : 216.73.216.79
<?php
namespace Codeception\Lib\Driver;
use Codeception\Exception\ModuleConfigException;
use Codeception\Exception\ModuleException;
use MongoDB\Database;
class MongoDb
{
const DEFAULT_PORT = 27017;
private $legacy;
private $dbh;
private $dsn;
private $dbName;
private $host;
private $user;
private $password;
private $client;
private $quiet = '';
public static function connect($dsn, $user, $password)
{
throw new \Exception(__CLASS__ . '::connect() - hm, it looked like this method had become obsolete...');
}
/**
* Connect to the Mongo server using the MongoDB extension.
*/
protected function setupMongoDB($dsn, $options)
{
try {
$this->client = new \MongoDB\Client($dsn, $options);
$this->dbh = $this->client->selectDatabase($this->dbName);
} catch (\MongoDB\Driver\Exception $e) {
throw new ModuleException($this, sprintf('Failed to open Mongo connection: %s', $e->getMessage()));
}
}
/**
* Connect to the Mongo server using the legacy mongo extension.
*/
protected function setupMongo($dsn, $options)
{
try {
$this->client = new \MongoClient($dsn, $options);
$this->dbh = $this->client->selectDB($this->dbName);
} catch (\MongoConnectionException $e) {
throw new ModuleException($this, sprintf('Failed to open Mongo connection: %s', $e->getMessage()));
}
}
/**
* Clean up the Mongo database using the MongoDB extension.
*/
protected function cleanupMongoDB()
{
try {
$this->dbh->drop();
} catch (\MongoDB\Driver\Exception $e) {
throw new \Exception(sprintf('Failed to drop the DB: %s', $e->getMessage()));
}
}
/**
* Clean up the Mongo database using the legacy Mongo extension.
*/
protected function cleanupMongo()
{
try {
$list = $this->dbh->listCollections();
} catch (\MongoException $e) {
throw new \Exception(sprintf('Failed to list collections of the DB: %s', $e->getMessage()));
}
foreach ($list as $collection) {
try {
$collection->drop();
} catch (\MongoException $e) {
throw new \Exception(sprintf('Failed to drop collection: %s', $e->getMessage()));
}
}
}
/**
* $dsn has to contain db_name after the host. E.g. "mongodb://localhost:27017/mongo_test_db"
*
* @static
*
* @param $dsn
* @param $user
* @param $password
*
* @throws ModuleConfigException
* @throws \Exception
*/
public function __construct($dsn, $user, $password)
{
$this->legacy = extension_loaded('mongodb') === false &&
class_exists('\\MongoClient') &&
strpos(\MongoClient::VERSION, 'mongofill') === false;
/* defining DB name */
$this->dbName = preg_replace('/\?.*/', '', substr($dsn, strrpos($dsn, '/') + 1));
if (strlen($this->dbName) == 0) {
throw new ModuleConfigException($this, 'Please specify valid $dsn with DB name after the host:port');
}
/* defining host */
if (strpos($dsn, 'mongodb://') !== false) {
$this->host = str_replace('mongodb://', '', preg_replace('/\?.*/', '', $dsn));
} else {
$this->host = $dsn;
}
$this->host = rtrim(str_replace($this->dbName, '', $this->host), '/');
$options = [
'connect' => true
];
if ($user && $password) {
$options += [
'username' => $user,
'password' => $password
];
}
$this->{$this->legacy ? 'setupMongo' : 'setupMongoDB'}($dsn, $options);
$this->dsn = $dsn;
$this->user = $user;
$this->password = $password;
}
/**
* @static
*
* @param $dsn
* @param $user
* @param $password
*
* @return MongoDb
*/
public static function create($dsn, $user, $password)
{
return new MongoDb($dsn, $user, $password);
}
public function cleanup()
{
$this->{$this->legacy ? 'cleanupMongo' : 'cleanupMongoDB'}();
}
/**
* dump file has to be a javascript document where one can use all the mongo shell's commands
* just FYI: this file can be easily created be RockMongo's export button
*
* @param $dumpFile
*/
public function load($dumpFile)
{
$cmd = sprintf(
'mongo %s %s%s',
$this->host . '/' . $this->dbName,
$this->createUserPasswordCmdString(),
escapeshellarg($dumpFile)
);
shell_exec($cmd);
}
public function loadFromMongoDump($dumpFile)
{
list($host, $port) = $this->getHostPort();
$cmd = sprintf(
"mongorestore %s --host %s --port %s -d %s %s %s",
$this->quiet,
$host,
$port,
$this->dbName,
$this->createUserPasswordCmdString(),
escapeshellarg($dumpFile)
);
shell_exec($cmd);
}
public function loadFromTarGzMongoDump($dumpFile)
{
list($host, $port) = $this->getHostPort();
$getDirCmd = sprintf(
"tar -tf %s | awk 'BEGIN { FS = \"/\" } ; { print $1 }' | uniq",
escapeshellarg($dumpFile)
);
$dirCountCmd = $getDirCmd . ' | wc -l';
if (trim(shell_exec($dirCountCmd)) !== '1') {
throw new ModuleException(
$this,
'Archive MUST contain single directory with db dump'
);
}
$dirName = trim(shell_exec($getDirCmd));
$cmd = sprintf(
'tar -xzf %s && mongorestore %s --host %s --port %s -d %s %s %s && rm -r %s',
escapeshellarg($dumpFile),
$this->quiet,
$host,
$port,
$this->dbName,
$this->createUserPasswordCmdString(),
$dirName,
$dirName
);
shell_exec($cmd);
}
private function createUserPasswordCmdString()
{
if ($this->user && $this->password) {
return sprintf(
'--username %s --password %s ',
$this->user,
$this->password
);
}
return '';
}
public function getDbh()
{
return $this->dbh;
}
public function setDatabase($dbName)
{
$this->dbh = $this->client->{$this->legacy ? 'selectDB' : 'selectDatabase'}($dbName);
}
/**
* Determine if this driver is using the legacy extension or not.
*
* @return bool
*/
public function isLegacy()
{
return $this->legacy;
}
private function getHostPort()
{
$hostPort = explode(':', $this->host);
if (count($hostPort) === 2) {
return $hostPort;
}
if (count($hostPort) === 1) {
return [$hostPort[0], self::DEFAULT_PORT];
}
throw new ModuleException($this, '$dsn MUST be like (mongodb://)<host>:<port>/<db name>');
}
public function setQuiet($quiet)
{
$this->quiet = $quiet ? '--quiet' : '';
}
}