HowToPatchTablesForI18nFromGeneratedSchema: doPatchDbI18nTask.class.php

<?php

class doPatchDbI18nTask extends sfBaseTask
{
  protected function configure()
  {
    $this->namespace        = 'project';
    $this->name             = 'patch-db-i18n';
    $this->briefDescription = 'Patch a database generated from DB to use i18n';
    $this->detailedDescription = <<<EOF
The [project:patch-db-i18n|INFO] task does things.
Call it with:

  [php symfony project:patch-db-i18n|INFO]
EOF;
  }

  protected function execute($arguments = array(), $options = array())
  {
    define('SF_ROOT_DIR',    sfConfig::get("sf_root_dir"));
    define('SF_ENVIRONMENT', 'dev');
    define('SF_DEBUG',       false);

    $suffix = isset($arguments[0]) ? $arguments[0]: "_i18n";
    $column = isset($arguments[1]) ? $arguments[1]: "culture";

    $this->logSection('patch-db-i18n', 'Parsing schema.yml');

    // check for `schema.yml`
    $schema_path = sfConfig::get('sf_config_dir') . DIRECTORY_SEPARATOR . 'schema.yml';

    if(!file_exists($schema_path))
      throw new Exception("You must first generate schema.yml from the database");

    $schema = sfYaml::load($schema_path);

    $this->logSection('patch-db-i18n', 'Finished parsing `schema.yml`');
    $this->logSection('patch-db-i18n', 'Scanning for I18n tables');

    $table_names = array_keys($schema['propel']);
    $pairs = array();

    // find parent/child pairs
    foreach($table_names as $name) 
    {
      if(in_array($name . $suffix, $table_names)) 
        $pairs[] = array("parent" => $name, "child" => $name . $suffix);
    }

    // list the tables
    $this->logSection('patch-db-i18n', 'Tables to patch:');
    foreach($pairs as $pair)
      $this->logSection('patch-db-i18n', '        - "' . $pair['parent'] . '" and "' . $pair['child']);

    $this->logSection('patch-db-i18n', 'Appending i18n parameters to tables');

    $parent = null;
    $child = null;

    // iterate through the matched tables and append necessary attributes
    foreach($pairs as $pair) 
    {
      $parent =& $schema['propel'][$pair["parent"]];
      $child =& $schema['propel'][$pair["child"]];

      // set parent as I18n enabled
      $parent['_attributes']['isI18n'] = 'true';
      $parent['_attributes']['i18nTable'] = $pair['child'];

      // set child table's culture column
      if(!isset($child[$column]))
        $child[$column] = array();

      $child[$column]['isCulture'] = 'true';

      /*
      $child[$column]['type'] = 'varchar';
      $child[$column]['size'] = 7;
      $child[$column]['required'] = 'true';
      $child[$column]['primaryKey'] = 'true';
      */
    }

    $this->logSection('patch-db-i18n', 'Finished appending, writing files');

    // back up the current schema.yml
    if(!copy($schema_path, $schema_path . '.bak'))
      throw new Exception("Could not back up the original `schema.yml`");
    else
      $this->logSection('patch-db-i18n', 'File+   ' . $schema_path . '.bak');

    if(file_put_contents($schema_path, sfYaml::dump($schema)) === false)
      throw new Exception("Could not write modified `schema.yml` file!");
    else
      $this->logSection('patch-db-i18n', 'File*   ' . $schema_path);

    $this->logSection('patch-db-i18n', 'Finished modifying schema.yml');
    $this->logSection('patch-db-i18n', 'Please run `propel-build-model` and `symfony cc` to generate the modified Propel classes');
  }
}