composer/src/Composer/Command/SearchCommand.php

<?php

/*
 * This file is part of Composer.
 *
 * (c) Nils Adermann <naderman@naderman.de>
 *     Jordi Boggiano <j.boggiano@seld.be>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Composer\Command;

use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Output\OutputInterface;
use Composer\Repository\CompositeRepository;
use Composer\Repository\PlatformRepository;
use Composer\Package\CompletePackageInterface;
use Composer\Package\AliasPackage;
use Composer\Factory;

/**
 * @author Robert Schönthal <seroscho@googlemail.com>
 */
class SearchCommand extends Command
{
    protected $matches;
    protected $lowMatches;
    protected $tokens;
    protected $output;

    protected function configure()
    {
        $this
            ->setName('search')
            ->setDescription('Search for packages')
            ->setDefinition(array(
                new InputArgument('tokens', InputArgument::IS_ARRAY | InputArgument::REQUIRED, 'tokens to search for'),
            ))
            ->setHelp(<<<EOT
The search command searches for packages by its name
<info>php composer.phar search symfony composer</info>

EOT
            )
        ;
    }

    protected function execute(InputInterface $input, OutputInterface $output)
    {
        // init repos
        $platformRepo = new PlatformRepository;
        if ($composer = $this->getComposer(false)) {
            $localRepo = $composer->getRepositoryManager()->getLocalRepository();
            $installedRepo = new CompositeRepository(array($localRepo, $platformRepo));
            $repos = new CompositeRepository(array_merge(array($installedRepo), $composer->getRepositoryManager()->getRepositories()));
        } else {
            $defaultRepos = Factory::createDefaultRepositories($this->getIO());
            $output->writeln('No composer.json found in the current directory, showing packages from ' . implode(', ', array_keys($defaultRepos)));
            $installedRepo = $platformRepo;
            $repos = new CompositeRepository(array_merge(array($installedRepo), $defaultRepos));
        }

        $time = microtime(true);

        $this->tokens = $input->getArgument('tokens');
        $this->output = $output;
        $repos->filterPackages(array($this, 'processPackage'), 'Composer\Package\CompletePackage');

        foreach ($this->lowMatches as $details) {
            $output->writeln($details['name'] . '<comment>:</comment> '. $details['description']);
        }

        var_dump((memory_get_peak_usage() / 1024 / 1024) . 'MB memory, '.round(microtime(true) - $time, 2) .'secs');
    }

    public function processPackage($package)
    {
        if ($package instanceof AliasPackage || isset($this->matches[$package->getName()])) {
            return;
        }

        foreach ($this->tokens as $token) {
            if (!$score = $this->matchPackage($package, $token)) {
                continue;
            }

            if (false !== ($pos = stripos($package->getName(), $token))) {
                $name = substr($package->getPrettyName(), 0, $pos)
                    . '<highlight>' . substr($package->getPrettyName(), $pos, strlen($token)) . '</highlight>'
                    . substr($package->getPrettyName(), $pos + strlen($token));
            } else {
                $name = $package->getPrettyName();
            }

            $description = strtok($package->getDescription(), "\r\n");
            if (false !== ($pos = stripos($description, $token))) {
                $description = substr($description, 0, $pos)
                    . '<highlight>' . substr($description, $pos, strlen($token)) . '</highlight>'
                    . substr($description, $pos + strlen($token));
            }

            if ($score >= 3) {
                $this->output->writeln($name . '<comment>:</comment> '. $description);
                $this->matches[$package->getName()] = true;
            } else {
                $this->lowMatches[$package->getName()] = array(
                    'name' => $name,
                    'description' => $description,
                );
            }

            return;
        }
    }

    /**
     * tries to find a token within the name/keywords/description
     *
     * @param  CompletePackageInterface $package
     * @param  string           $token
     * @return boolean
     */
    private function matchPackage(CompletePackageInterface $package, $token)
    {
        $score = 0;

        if (false !== stripos($package->getName(), $token)) {
            $score += 5;
        }

        if (false !== stripos(join(',', $package->getKeywords() ?: array()), $token)) {
            $score += 3;
        }

        if (false !== stripos($package->getDescription(), $token)) {
            $score += 1;
        }

        return $score;
    }
}