<?php

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

namespace CloudObjects\PhpMAE;

use GuzzleHttp\Client;
use CloudObjects\SDK\ObjectRetriever;

class ObjectRetrieverPool {

    private $baseObjectRetriever;
    private $baseHostname;
    private $options;
    private $objectRetrievers = [];

    public function __construct(ObjectRetriever $baseObjectRetriever, $baseHostname, $options) {
        $this->baseObjectRetriever = $baseObjectRetriever;
        $this->baseHostname = $baseHostname;
        $this->objectRetrievers[$baseHostname] = $baseObjectRetriever;
        $this->options = $options;
    }

    public function getBaseObjectRetriever() {
        return $this->baseObjectRetriever;
    }

    public function getObjectRetriever($hostname) {			
        if (!isset($this->baseHostname))
            // Never create a new retriever when using developer credentials
            return $this->baseObjectRetriever;

        if (!isset($this->objectRetrievers[$hostname])) {
            if (substr($hostname, -7) == '.phpmae') {
                // Get an anonymous retriever
                $retriever = new ObjectRetriever(array_merge($this->options, [
                    'auth_ns' => '',
                    'auth_secret' => '',
                    'cache_prefix' => 'clobj:'.$hostname.':',
                    'logger' => $this->options['logger']->withName('CO-'.$hostname)
                ]));
            } else {
                // Get a retriever with namespace impersonation
                $config = $this->baseObjectRetriever->getClient()->getConfig();
                $config['headers']['C-Act-As'] = $hostname;

		    	$retriever = new ObjectRetriever(array_merge($this->options, [
                    'cache_prefix' => 'clobj:'.$hostname.':',
                    'logger' => $this->options['logger']->withName('CO-'.$hostname)
                ]));
                $retriever->setClient(new Client($config));
            }
			$this->objectRetrievers[$hostname] = $retriever;
        }

        return $this->objectRetrievers[$hostname];
    }
}