<?php
/*
 * This file is part of the WikiTagBundle package.
 *
 * (c) IRI <http://www.iri.centrepompidou.fr/>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Application\Migrations;

use Doctrine\DBAL\Migrations\OutputWriter;

use Doctrine\DBAL\Migrations\AbstractMigration;
use Doctrine\DBAL\Schema\Schema;
use Doctrine\Common\EventManager;
use Doctrine\ORM\EntityManager;
use IRI\Bundle\WikiTagBundle\Utils\WikiTagUtils;
use IRI\Bundle\WikiTagBundle\Entity\Tag;


/**
 * Auto-generated Migration: Please modify to your need!
 */
class Version20111125175224 extends AbstractMigration
{
    
    private function showProgress($current, $total, $label, $width)
    {
        $percent = (floatval($current)/floatval($total)) * 100.0;
        $marks = intval(floor(floatval($width) * ($percent / 100.0) ));
        $spaces = $width - $marks;
    
        $status_bar="\r[";
        $status_bar.=str_repeat("=", $marks);
        if($marks<$width){
            $status_bar.=">";
            $status_bar.=str_repeat(" ", $spaces);
        } else {
            $status_bar.="=";
        }
    
        $disp=str_pad(number_format($percent, 0),3, " ", STR_PAD_LEFT);
    
        $label = str_pad(substr($label,0,50), 50, " ");
        $current_str = str_pad($current, strlen("$total"), " ", STR_PAD_LEFT);
    
        $status_bar.="] $disp%  $current_str/$total : $label";
    
        $this->write("$status_bar  ");
    
        if($current == $total) {
            $this->write("\n");
        }
    }
    
    public function up(Schema $schema)
    {
        // this up() migration is autogenerated, please modify it to your needs
        $this->abortIf($this->connection->getDatabasePlatform()->getName() != "mysql");
        
        $this->addSql("ALTER TABLE wikitag_document ADD CONSTRAINT FK_256601059F75D7B0 FOREIGN KEY (external_id) REFERENCES Document(id)");
        $this->addSql("ALTER TABLE wikitag_document_tag CHANGE wikipedia_revision_id wikipedia_revision_id BIGINT DEFAULT NULL");
        $this->addSql("ALTER TABLE wikitag_tag ADD alternative_label VARCHAR(1024) DEFAULT NULL, ADD wikipedia_alternative_url VARCHAR(2048) DEFAULT NULL, ADD alternative_wikipedia_pageid BIGINT DEFAULT NULL, CHANGE normalized_pageid wikipedia_pageid BIGINT DEFAULT NULL");
    }
    
    public function postUp(Schema $schema)
    {
        $doctrine = $GLOBALS["application"]->getKernel()->getContainer()->get('doctrine');
        $em = $doctrine->getEntityManager();
        
        $this->skipIf($this->version->isMigrated() !== true, 'postUp can only apply if migration completes.');
    
        //select all tags with redirect and alternative label null
        $query = $em->createQuery("SELECT t FROM WikiTagBundle:Tag t WHERE t.urlStatus = :url_status AND t.alternativeLabel IS NULL");
        $query->setParameter("url_status", Tag::$TAG_URL_STATUS_DICT['redirection']);

        $querycount = $em->createQuery("SELECT count(t) FROM WikiTagBundle:Tag t WHERE t.urlStatus = :url_status AND t.alternativeLabel IS NULL");
        $querycount->setParameter("url_status", Tag::$TAG_URL_STATUS_DICT['redirection']);
        
        
        $total = $querycount->getSingleScalarResult();
        $iterableResult = $query->iterate();
        $i = 0;
        $em->beginTransaction();
        foreach($iterableResult as $row) {
            $i++;
            $tag = $row[0];
    
            $memory = ((($i%10)==0)?" - mem: ".strval(memory_get_usage(true)):"");
            $this->showProgress($i, $total, "tag : ".$tag->getLabel()."$memory", 50);
            //process $tag
            $wk_info = WikiTagUtils::getWikipediaInfo($tag->getNormalizedLabel());
    
            $tag->setWikipediaInfo($wk_info);
            
            $em->persist($tag);
                            
            if (($i % 100) == 0) {
                $em->flush();
                $em->commit();
                $em->clear(); // Detaches all objects from Doctrine!
                $em->beginTransaction();
            }
        }
        
        $em->flush(); // Executes all updates.
        $em->commit();
        $em->clear(); // Detaches all objects from Doctrine!
    
    }
    
    public function down(Schema $schema)
    {
        // this down() migration is autogenerated, please modify it to your needs
        $this->abortIf($this->connection->getDatabasePlatform()->getName() != "mysql");
        
        $this->addSql("ALTER TABLE wikitag_document DROP FOREIGN KEY FK_256601059F75D7B0");
        $this->addSql("ALTER TABLE wikitag_document_tag CHANGE wikipedia_revision_id wikipedia_revision_id BIGINT NOT NULL");
        $this->addSql("ALTER TABLE wikitag_tag DROP alternative_label, DROP wikipedia_alternative_url, DROP alternative_wikipedia_pageid, CHANGE wikipedia_pageid normalized_pageid BIGINT DEFAULT NULL");
    }
    
    
}
