aboutsummaryrefslogtreecommitdiff
path: root/src/clients/cave/cmd_generate_metadata.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/clients/cave/cmd_generate_metadata.cc')
-rw-r--r--src/clients/cave/cmd_generate_metadata.cc163
1 files changed, 163 insertions, 0 deletions
diff --git a/src/clients/cave/cmd_generate_metadata.cc b/src/clients/cave/cmd_generate_metadata.cc
new file mode 100644
index 000000000..29be83c6f
--- /dev/null
+++ b/src/clients/cave/cmd_generate_metadata.cc
@@ -0,0 +1,163 @@
+/* vim: set sw=4 sts=4 et foldmethod=syntax : */
+
+/*
+ * Copyright (c) 2011 Ciaran McCreesh
+ *
+ * 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_generate_metadata.hh"
+#include "format_string.hh"
+#include <paludis/args/args.hh>
+#include <paludis/args/do_help.hh>
+#include <paludis/name.hh>
+#include <paludis/environment.hh>
+#include <paludis/package_database.hh>
+#include <paludis/repository.hh>
+#include <paludis/util/set.hh>
+#include <paludis/util/wrapped_forward_iterator.hh>
+#include <paludis/util/indirect_iterator-impl.hh>
+#include <paludis/util/simple_visitor_cast.hh>
+#include <paludis/util/make_null_shared_ptr.hh>
+#include <paludis/util/accept_visitor.hh>
+#include <paludis/generator.hh>
+#include <paludis/filtered_generator.hh>
+#include <paludis/filter.hh>
+#include <paludis/filter_handler.hh>
+#include <paludis/selection.hh>
+#include <paludis/user_dep_spec.hh>
+#include <paludis/package_id.hh>
+#include <paludis/mask.hh>
+#include <paludis/metadata_key.hh>
+#include <cstdlib>
+#include <iostream>
+#include <algorithm>
+
+#include "command_command_line.hh"
+
+using namespace paludis;
+using namespace cave;
+using std::cout;
+using std::endl;
+
+namespace
+{
+ struct GenerateMetadataCommandLine :
+ CaveCommandCommandLine
+ {
+ virtual std::string app_name() const
+ {
+ return "cave generate-metadata";
+ }
+
+ virtual std::string app_synopsis() const
+ {
+ return "Pregenerate metadata for a set of IDs.";
+ }
+
+ virtual std::string app_description() const
+ {
+ return "Pregenerates metadata for a set of IDs.";
+ }
+
+ args::ArgsGroup g_filters;
+ args::StringSetArg a_matching;
+
+ GenerateMetadataCommandLine() :
+ g_filters(main_options_section(), "Filters", "Filter the output. Each filter may be specified more than once."),
+ a_matching(&g_filters, "matching", 'm', "Consider only IDs matching this spec. Note that certain specs "
+ "may force metadata generation anyway, e.g. to see whether a slot matches.",
+ args::StringSetArg::StringSetArgOptions())
+ {
+ add_usage_line("[ --matching spec ]");
+ }
+ };
+
+ struct MetadataVisitor
+ {
+ void visit(const MetadataSectionKey & k) const
+ {
+ std::for_each(indirect_iterator(k.begin_metadata()), indirect_iterator(k.end_metadata()), accept_visitor(*this));
+ }
+
+ template <typename T_>
+ void visit(const T_ & k) const
+ {
+ auto PALUDIS_ATTRIBUTE((unused)) v(k.value());
+ }
+ };
+}
+
+int
+GenerateMetadataCommand::run(
+ const std::shared_ptr<Environment> & env,
+ const std::shared_ptr<const Sequence<std::string > > & args
+ )
+{
+ GenerateMetadataCommandLine cmdline;
+ cmdline.run(args, "CAVE", "CAVE_GENERATE_METADATA_OPTIONS", "CAVE_GENERATE_METADATA_CMDLINE");
+
+ if (cmdline.a_help.specified())
+ {
+ cout << cmdline;
+ return EXIT_SUCCESS;
+ }
+
+ if (cmdline.begin_parameters() != cmdline.end_parameters())
+ throw args::DoHelp("generate-metadata takes no parameters");
+
+ Generator g((generator::All()));
+ if (cmdline.a_matching.specified())
+ {
+ for (args::StringSetArg::ConstIterator m(cmdline.a_matching.begin_args()),
+ m_end(cmdline.a_matching.end_args()) ;
+ m != m_end ; ++m)
+ {
+ PackageDepSpec s(parse_user_package_dep_spec(*m, env.get(), { updso_allow_wildcards }));
+ g = g & generator::Matches(s, make_null_shared_ptr(), { });
+ }
+ }
+
+ const std::shared_ptr<const PackageIDSequence> ids((*env)[selection::AllVersionsSorted(g)]);
+ bool fail(false);
+ MetadataVisitor v;
+ for (PackageIDSequence::ConstIterator i(ids->begin()), i_end(ids->end()) ;
+ i != i_end ; ++i)
+ {
+ for (PackageID::MetadataConstIterator m((*i)->begin_metadata()), m_end((*i)->end_metadata()); m_end != m; ++m)
+ try
+ {
+ (*m)->accept(v);
+ }
+ catch (const InternalError &)
+ {
+ throw;
+ }
+ catch (const Exception & e)
+ {
+ std::cerr << "When processing '" << **i << "' got exception '" << e.message() << "' (" << e.what() << ")" << std::endl;
+ fail = true;
+ break;
+ }
+ }
+
+ return fail ? EXIT_FAILURE : EXIT_SUCCESS;
+}
+
+std::shared_ptr<args::ArgsHandler>
+GenerateMetadataCommand::make_doc_cmdline()
+{
+ return std::make_shared<GenerateMetadataCommandLine>();
+}
+