<?php

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

namespace Packagist\WebBundle\Model;

use Packagist\WebBundle\Entity\Package;
use Packagist\WebBundle\Entity\Version;
use Predis\Client;

/**
 * Manages the download counts for packages.
 */
class DownloadManager
{

    protected $redis;

    public function __construct(Client $redis)
    {
        $this->redis = $redis;
    }

    /**
     * Gets the total, monthly, and daily download counts for a package.
     *
     * @param \Packagist\WebBundle\Entity\Package|int $package
     * @return array
     */
    public function getDownloads($package)
    {
        if ($package instanceof Package) {
            $package = $package->getId();
        }

        $counts = $this->redis->mget(
            'dl:' . $package,
            'dl:' . $package.':'.date('Ym'),
            'dl:' . $package.':'.date('Ymd')
        );

        return array(
            'total' => (int) $counts[0] ?: 0,
            'monthly' => (int) $counts[1] ?: 0,
            'daily' => (int)$counts[2] ?: 0,
        );
    }

    /**
     * Gets total download counts for multiple package IDs.
     *
     * @param array $packageIds
     * @return array a map of package ID to download count
     */
    public function getPackagesDownloads(array $packageIds)
    {
        $keys = array();

        foreach ($packageIds as $id) {
            $keys[$id] = 'dl:'.$id;
        }

        $res = array_map('intval', $this->redis->mget(array_values($keys)));
        return array_combine(array_keys($keys), $res);
    }

    /**
     * Tracks a new download by updating the relevant keys.
     *
     * @param \Packagist\WebBundle\Entity\Package|int $package
     * @param \Packagist\WebBundle\Entity\Version|int $version
     */
    public function addDownload($package,  $version)
    {
        $redis = $this->redis;

        if ($package instanceof Package) {
            $package = $package->getId();
        }

        if ($version instanceof Version) {
            $version = $version->getId();
        }

        $redis->incr('downloads');

        $redis->incr('dl:'.$package);
        $redis->incr('dl:'.$package.':'.date('Ym'));
        $redis->incr('dl:'.$package.':'.date('Ymd'));

        $redis->incr('dl:'.$package.'-'.$version);
        $redis->incr('dl:'.$package.'-'.$version.':'.date('Ym'));
        $redis->incr('dl:'.$package.'-'.$version.':'.date('Ymd'));
    }

}