+global\MigrateTranslatableTask::run()

Complexity

Conditions	18
Paths	222

Size

Total Lines

133

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

<?php
    public function run($request)
    {
        $ids = array();
        
        echo "#################################\n";
        echo "# Adding translation groups to existing records" . "\n";
        echo "#################################\n";
        
        $allSiteTreeIDs = DB::query('SELECT "ID" FROM "SiteTree"')->column();
        if ($allSiteTreeIDs) {
            foreach ($allSiteTreeIDs as $id) {
                $original = DataObject::get_by_id('SiteTree', $id);
                $existingGroupID = $original->getTranslationGroup();
                if (!$existingGroupID) {
                    $original->addTranslationGroup($original->ID);
                }
                $original->destroy();
                unset($original);
            }
        }
        
        DataObject::flush_and_destroy_cache();
        
        echo sprintf("Created translation groups for %d records\n", count($allSiteTreeIDs));

        foreach (array('Stage', 'Live') as $stage) {
            echo "\n\n#################################\n";
            echo "# Migrating stage $stage" . "\n";
            echo "#################################\n";
            
            $suffix = ($stage == 'Live') ? '_Live' : '';
        
            // First get all entries in SiteTree_lang
            // This should be all translated pages
            $trans = DB::query(sprintf('SELECT * FROM "_obsolete_SiteTree_lang%s"', $suffix));
        
            // Iterate over each translated pages
            foreach ($trans as $oldtrans) {
                $newLocale = i18n::get_locale_from_lang($oldtrans['Lang']);
                
                echo sprintf(
                    "Migrating from %s to %s translation of '%s' (#%d)\n",
                    $oldtrans['Lang'],
                    $newLocale,
                    Convert::raw2xml($oldtrans['Title']),
                    $oldtrans['OriginalLangID']
                );
            
                // Get the untranslated page

                $original = Versioned::get_one_by_stage(
                    $oldtrans['ClassName'],
                    $stage,
                    '"SiteTree"."ID" = ' .  $oldtrans['OriginalLangID']
                );
                
                if (!$original) {
                    echo sprintf("Couldn't find original for #%d", $oldtrans['OriginalLangID']);
                    continue;
                }
                
                // write locale to $original
                $original->Locale = i18n::get_locale_from_lang(Translatable::default_lang());

The method Translatable::default_lang() has been deprecated with message: 2.4 Use default_locale()

                $original->writeToStage($stage);
                
                // Clone the original, and set it up as a translation
                $existingTrans = $original->getTranslation($newLocale, $stage);
                
                if ($existingTrans) {
                    echo sprintf(
                        "Found existing new-style translation for #%d. Already merged? Skipping.\n",
                        $oldtrans['OriginalLangID']
                    );
                    continue;
                }
                
                // Doesn't work with stage/live split
                //$newtrans = $original->createTranslation($newLocale);

                $newtrans = $original->duplicate(false);
                $newtrans->OriginalID = $original->ID;
                // we have to "guess" a locale based on the language
                $newtrans->Locale = $newLocale;
                if ($stage == 'Live' && array_key_exists($original->ID, $ids)) {
                    $newtrans->ID = $ids[$original->ID];
                }
                
                // Look at each class in the ancestry, and see if there is a _lang table for it
                foreach (ClassInfo::ancestry($oldtrans['ClassName']) as $classname) {
                    $oldtransitem = false;
                
                    // If the class is SiteTree, we already have the DB record, 
                    // else check for the table and get the record
                    if ($classname == 'SiteTree') {
                        $oldtransitem = $oldtrans;
                    } elseif (in_array(strtolower($classname) . '_lang', DB::tableList())) {
                        $oldtransitem = DB::query(sprintf(
                            'SELECT * FROM "_obsolete_%s_lang%s" WHERE "OriginalLangID" = %d AND "Lang" = \'%s\'',
                            $classname,
                            $suffix,
                            $original->ID,
                            $oldtrans['Lang']
                        ))->first();
                    }
                
                    // Copy each translated field into the new translation
                    if ($oldtransitem) {
                        foreach ($oldtransitem as $key => $value) {
                            if (!in_array($key, array('ID', 'OriginalLangID'))) {
                                $newtrans->$key = $value;
                            }
                        }
                    }
                }

                // Write the new translation to the database
                $sitelang = Translatable::get_current_locale();
                Translatable::set_current_locale($newtrans->Locale);
                $newtrans->writeToStage($stage);
                Translatable::set_current_locale($sitelang);
                
                $newtrans->addTranslationGroup($original->getTranslationGroup(), true);

                
                if ($stage == 'Stage') {
                    $ids[$original->ID] = $newtrans->ID;
                }
            }
        }
        
        echo "\n\n#################################\n";
        echo "Done!\n";
    }
}