digest_registry.hh

/* vim: set sw=4 sts=4 et foldmethod=syntax : */
 
/*
 * Copyright (c) 2011, 2012 David Leverton
 *
 * This file is part of the Paludis package manager. Paludis is free software;
 * you can redistribute it and/or modify it under the terms of the GNU General
 * Public License version 2, as published by the Free Software Foundation.
 *
 * Paludis is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
 * Place, Suite 330, Boston, MA  02111-1307  USA
 */
 
#ifndef PALUDIS_GUARD_PALUDIS_UTIL_DIGEST_REGISTRY_HH
#define PALUDIS_GUARD_PALUDIS_UTIL_DIGEST_REGISTRY_HH 1
 
#include <iosfwd>
#include <string>
#include <paludis/util/attributes.hh>
#include <paludis/util/pimp.hh>
#include <paludis/util/singleton.hh>
#include <paludis/util/wrapped_forward_iterator.hh>
#include <functional>
#include <utility>
 
namespace paludis
{
    class DigestRegistry;
 
    extern template class Pimp<DigestRegistry>;
    extern template class PALUDIS_VISIBLE Singleton<DigestRegistry>;
 
    class PALUDIS_VISIBLE DigestRegistry :
        public Singleton<DigestRegistry>
    {
        friend class Singleton<DigestRegistry>;
 
        public:
            typedef std::function<std::string (std::istream &)> Function;
 
            Function get(const std::string & algo) const;
 
            struct AlgorithmsConstIteratorTag;
            typedef WrappedForwardIterator<AlgorithmsConstIteratorTag, const std::pair<const std::string, Function> > AlgorithmsConstIterator;
 
            AlgorithmsConstIterator begin_algorithms() const PALUDIS_ATTRIBUTE((warn_unused_result));
            AlgorithmsConstIterator end_algorithms() const PALUDIS_ATTRIBUTE((warn_unused_result));
 
            template <typename T_>
            class Registration
            {
                public:
                    Registration(const std::string & algo)
                    {
                        get_instance()->register_function(algo, do_digest<T_>);
                    }
            };
 
        private:
            DigestRegistry();
            ~DigestRegistry();
 
            Pimp<DigestRegistry> _imp;
 
            void register_function(const std::string & algo, const Function & func);
 
            template <typename T_>
            static std::string
            do_digest(std::istream & stream)
            {
                T_ digest(stream);
                return digest.hexsum();
            }
    };
 
    extern template class PALUDIS_VISIBLE WrappedForwardIterator<
            DigestRegistry::AlgorithmsConstIteratorTag,
            const std::pair<const std::string, DigestRegistry::Function> >;
}
 
#endif