Subversion-Projekte lars-tiefland.php_share

<?php

/*

 * This file is part of the symfony package.

 * (c) 2004-2006 Fabien Potencier <fabien.potencier@symfony-project.com>

 *

 * For the full copyright and license information, please view the LICENSE

 * file that was distributed with this source code.

 */

require_once(dirname(__FILE__).'/sfPropelBaseTask.class.php');

/**

 * Inserts SQL for current model.

 *

 * @package    symfony

 * @subpackage propel

 * @author     Fabien Potencier <fabien.potencier@symfony-project.com>

 * @version    SVN: $Id: sfPropelInsertSqlTask.class.php 23922 2009-11-14 14:58:38Z fabien $

 */

class sfPropelInsertSqlTask extends sfPropelBaseTask

{

  /**

   * @see sfTask

   */

  protected function configure()

  {

    $this->addOptions(array(

      new sfCommandOption('application', null, sfCommandOption::PARAMETER_OPTIONAL, 'The application name', true),

      new sfCommandOption('env', null, sfCommandOption::PARAMETER_REQUIRED, 'The environment', 'cli'),

      new sfCommandOption('connection', null, sfCommandOption::PARAMETER_REQUIRED, 'The connection name', null),

      new sfCommandOption('no-confirmation', null, sfCommandOption::PARAMETER_NONE, 'Do not ask for confirmation'),

      new sfCommandOption('phing-arg', null, sfCommandOption::PARAMETER_REQUIRED | sfCommandOption::IS_ARRAY, 'Arbitrary phing argument'),

    ));

    $this->namespace = 'propel';

    $this->name = 'insert-sql';

    $this->briefDescription = 'Inserts SQL for current model';

    $this->detailedDescription = <<<EOF

The [propel:insert-sql|INFO] task creates database tables:

  [./symfony propel:insert-sql|INFO]

The task connects to the database and executes all SQL statements

found in [config/sql/*schema.sql|COMMENT] files.

Before execution, the task will ask you to confirm the execution

as it deletes all data in your database.

To bypass the confirmation, you can pass the [--no-confirmation|COMMENT]

option:

  [./symfony propel:insert-sql --no-confirmation|INFO]

The task read the database configuration from `databases.yml`.

You can use a specific application/environment by passing

an [--application|INFO] or [--env|INFO] option.

You can also use the [--connection|INFO] option if you want to

only load SQL statements for a given connection.

EOF;

  }

  /**

   * @see sfTask

   */

  protected function execute($arguments = array(), $options = array())

  {

    $this->schemaToXML(self::DO_NOT_CHECK_SCHEMA, 'generated-');

    $this->copyXmlSchemaFromPlugins('generated-');

    $databaseManager = new sfDatabaseManager($this->configuration);

    $properties = $this->getProperties(sfConfig::get('sf_data_dir').'/sql/sqldb.map');

    $sqls = array();

    foreach ($properties as $file => $connection)

    {

      if (null !== $options['connection'] && $options['connection'] != $connection)

      {

        continue;

      }

      if (!isset($sqls[$connection]))

      {

        $sqls[$connection] = array();

      }

      $sqls[$connection][] = $file;

    }

    if (

      !$options['no-confirmation']

      &&

      !$this->askConfirmation(array(

          'WARNING: The data in the database'.(count($sqls) > 1 ? 's' : '').' related to the connection name'.(count($sqls) > 1 ? 's' : ''),

          sprintf('         %s will be removed.', implode(', ', array_keys($sqls))),

          '',

          'Are you sure you want to proceed? (y/N)',

        ), 'QUESTION_LARGE', false)

    )

    {

      $this->logSection('propel', 'Task aborted.');

      return 1;

    }

    $this->tmpDir = sys_get_temp_dir().'/propel_insert_sql_'.rand(11111, 99999);

    register_shutdown_function(array($this, 'removeTmpDir'));

    mkdir($this->tmpDir, 0777, true);

    foreach ($sqls as $connection => $files)

    {

      $dir = $this->tmpDir.'/'.$connection;

      mkdir($dir, 0777, true);

      $content = '';

      foreach ($files as $file)

      {

        $content .= "$file=$connection\n";

        copy(sfConfig::get('sf_data_dir').'/sql/'.$file, $dir.'/'.$file);

      }

      file_put_contents($dir.'/sqldb.map', $content);

      $properties = $this->getPhingPropertiesForConnection($databaseManager, $connection);

      $properties['propel.sql.dir'] = $dir;

      $ret = $this->callPhing('insert-sql', self::CHECK_SCHEMA, $properties);

    }

    $this->removeTmpDir();

    $this->cleanup();

    return !$ret;

  }

  public function removeTmpDir()

  {

    if (!is_dir($this->tmpDir))

    {

      return;

    }

    sfToolkit::clearDirectory($this->tmpDir);

    rmdir($this->tmpDir);

  }

}