1 files changed, 73 insertions, 0 deletions
diff --git a/paludis/util/digest_registry.hh b/paludis/util/digest_registry.hh
new file mode 100644
index 000000000..cfa86ceea
--- /dev/null
+++ b/paludis/util/digest_registry.hh
@@ -0,0 +1,73 @@
+/* vim: set sw=4 sts=4 et foldmethod=syntax : */
+
+/*
+ * Copyright (c) 2011 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 <functional>
+
+namespace paludis
+{
+    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;
+
+            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 Pimp<DigestRegistry>;
+    extern template class Singleton<DigestRegistry>;
+}
+
+#endif