1 files changed, 129 insertions, 0 deletions
diff --git a/src/clients/cave/cmd_print_checksum.cc b/src/clients/cave/cmd_print_checksum.cc
new file mode 100644
index 000000000..7210c678d
--- /dev/null
+++ b/src/clients/cave/cmd_print_checksum.cc
@@ -0,0 +1,129 @@
+/* vim: set sw=4 sts=4 et foldmethod=syntax : */
+
+/*
+ * Copyright (c) 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
+ */
+
+#include "cmd_print_checksum.hh"
+#include "command_command_line.hh"
+
+#include <paludis/util/digest_registry.hh>
+#include <paludis/util/fs_path.hh>
+#include <paludis/util/safe_ifstream.hh>
+#include <paludis/util/upper_lower.hh>
+
+#include <paludis/args/do_help.hh>
+
+#include <cstdlib>
+#include <iostream>
+#include <sstream>
+
+using namespace paludis;
+using namespace paludis::cave;
+
+namespace
+{
+    struct PrintChecksumCommandLine :
+        CaveCommandCommandLine
+    {
+        args::ArgsGroup g_input;
+        args::SwitchArg a_stdin;
+        args::StringArg a_file;
+        args::StringArg a_text;
+
+        PrintChecksumCommandLine() :
+            g_input(main_options_section(), "Input Options", "Input Options"),
+            a_stdin(&g_input, "stdin", 's', "Calculate the checksum of standard input.", false),
+            a_file(&g_input, "file", 'f', "Calculate the checksum of the contents of the specified file."),
+            a_text(&g_input, "text", 't', "Calculate the checksum of the specified text.")
+        {
+            add_usage_line("ALGORITHM --stdin");
+            add_usage_line("ALGORITHM --file FILENAME");
+            add_usage_line("ALGORITHM --text TEXT");
+        }
+
+        virtual std::string app_name() const
+        {
+            return "cave print-checksum";
+        }
+
+        virtual std::string app_synopsis() const
+        {
+            return "prints cryptographic checksums";
+        }
+
+        virtual std::string app_description() const
+        {
+            return "Prints the checksum of the specified input using the specified algorithm. No formatting "
+                "is used, making the output suitable for parsing by scripts.";
+        }
+    };
+}
+
+int
+PrintChecksumCommand::run(const std::shared_ptr<Environment> &,
+    const std::shared_ptr<const Sequence<std::string> > & args)
+{
+    PrintChecksumCommandLine cmdline;
+    cmdline.run(args, "CAVE", "CAVE_PRINT_CHECKSUM_OPTIONS", "CAVE_PRINT_CHECKSUM_CMDLINE");
+
+    if (cmdline.a_help.specified())
+    {
+        std::cout << cmdline;
+        return EXIT_SUCCESS;
+    }
+
+    if (std::distance(cmdline.begin_parameters(), cmdline.end_parameters()) != 1)
+        throw args::DoHelp("print-checksum takes exactly one parameter");
+
+    std::string algo(toupper(*cmdline.begin_parameters()));
+    DigestRegistry::Function f(DigestRegistry::get_instance()->get(algo));
+    if (! f)
+        throw args::DoHelp("algorithm '" + algo + "' is not supported");
+
+    if (cmdline.a_stdin.specified() + cmdline.a_file.specified() + cmdline.a_text.specified() != 1)
+        throw args::DoHelp("print-checksum requires exactly one of --stdin, --file or --text");
+
+    std::string result;
+    if (cmdline.a_stdin.specified())
+        result = f(std::cin);
+    else if (cmdline.a_file.specified())
+    {
+        SafeIFStream s(FSPath(cmdline.a_file.argument()));
+        result = f(s);
+    }
+    else if (cmdline.a_text.specified())
+    {
+        std::istringstream s(cmdline.a_text.argument());
+        result = f(s);
+    }
+
+    std::cout << result << std::endl;
+    return EXIT_SUCCESS;
+}
+
+std::shared_ptr<args::ArgsHandler>
+PrintChecksumCommand::make_doc_cmdline()
+{
+    return std::make_shared<PrintChecksumCommandLine>();
+}
+
+CommandImportance
+PrintChecksumCommand::importance() const
+{
+    return ci_scripting;
+}
+