<?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 IRI\Bundle\WikiTagBundle\Command;

use Doctrine\ORM\Query;

use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;

use Symfony\Component\Console\Input\InputArgument;

use Symfony\Component\Console\Input\InputInterface;

use Symfony\Component\Console\Input\InputOption;

use Symfony\Component\Console\Output\OutputInterface;

class SyncDocumentsCommand extends ContainerAwareCommand

{

    protected function configure()

    {

        parent::configure();

        $this

            ->setName('wikitag:sync-doc')

            ->setDescription('Synchronize and index document class')

            ->addOption('missing', 'm', InputOption::VALUE_NONE, "process missing")

            ->addOption('tags', 't', InputOption::VALUE_NONE, "update tags")

            ->addOption('clear', 'c', InputOption::VALUE_NONE, "clear all docs");

    }

    protected function execute(InputInterface $input, OutputInterface $output)

    {

        $class = $this->getContainer()->getParameter('wiki_tag.document_class');

        $clear = $input->getOption('clear');

        $missing = $input->getOption('missing');

        $doctrine = $this->getContainer()->get('doctrine');

        $docrep = $doctrine->getRepository('WikiTagBundle:Document');

        $rep = $doctrine->getRepository($class);

        if(is_null($rep)) {

            //TODO : translate

            $output->writeln("$class does not have a repository : exiting.");

            return ;

        }

        if($input->getOption('tags'))

        {

            if(!$missing)

            {

                $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc from WikiTagBundle:Document doc");

                $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT COUNT(doc.id) from WikiTagBundle:Document doc");

            }

            else

            {

                $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc from WikiTagBundle:Document doc WHERE doc.tagsStr IS NULL");

                $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT COUNT(doc.id) from WikiTagBundle:Document doc WHERE doc.tagsStr IS NULL");

            }

            $total = $doccountquery->getSingleScalarResult();

            $done = 0;

            $iterable = $docquery->iterate();

            $todetach = array();

            while (($row = $iterable->next()) !== false) {

                $done++;

                $memory = ((($done%10)==0)?" - mem: ".strval(memory_get_usage(true)):"");

                $doc = $row[0];

                $todetach[] = $doc;

                $output->writeln("Process doc id ".$doc->getId()." $done/$total ".strval(intval(floatval($done)/floatval($total)*100.0))."%$memory");

                $docrep->updateTagsStr($doc);

                if($done%10 == 0)

                {

                    $doctrine->getEntityManager()->flush();

                    foreach($todetach as $obj)

                    {

                        $doctrine->getEntityManager()->detach($obj);

                    }

                    $todetach = array();

                }

            }

            $doctrine->getEntityManager()->flush();

            return;

        }

        //TODO : check class to implement DocumentInterface

        //TODO : write progress

        if($missing)

        {

            $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc FROM $class doc WHERE doc.id not in (SELECT wtdoc FROM WikiTagBundle:Document wtdoc)");

            $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT count(doc.id) FROM $class doc WHERE doc.id not in (SELECT wtdoc FROM WikiTagBundle:Document wtdoc)");

            //$doclist = $doctrine->getEntityManager()->createQuery("SELECT doc FROM $class doc WHERE doc.id not in (SELECT wtdoc FROM WikiTagBundle:Document wtdoc)")->getResult();

        }

        else

        {

            $docquery = $doctrine->getEntityManager()->createQuery("SELECT doc FROM $class doc");

            $doccountquery = $doctrine->getEntityManager()->createQuery("SELECT count(doc.id) FROM $class doc");

            //$doclist = $rep->findAll();

        }

        $total = $doccountquery->getSingleScalarResult();

        $done = 0;

        $iterable = $docquery->iterate();

        while (($row = $iterable->next()) !== false) {

            $done++;

            $doc = $row[0];

            $memory = ((($done%10)==0)?" - mem: ".strval(memory_get_usage(true)):"");

            $output->writeln("Process doc id ".$doc->getId()." $done/$total ".strval(intval(floatval($done)/floatval($total)*100.0))."%$memory");

            $docrep->writeDocument($doc, $this->getContainer()->getParameter('wiki_tag.document_id_column'), $this->getContainer()->getParameter('wiki_tag.fields'));

            if($done%10 == 0)

            {

                $doctrine->getEntityManager()->flush();

                $doctrine->getEntityManager()->clear();

            }

        }

        $doctrine->getEntityManager()->flush();

        if($clear) {

            $req = $doctrine->getEntityManager()->createQuery("DELETE WikiTagBundle:Document wtdoc WHERE wtdoc.externalId NOT IN (SELECT doc FROM $class doc)");

            $req->getResult();

            $doctrine->getEntityManager()->flush();

        }

        $output->writeln(strval(count($doclist)) ." documents imported.");

    }

}