bool CommandGetParents::setup(const std::vector<std::string>& arguments) {
po::options_description opts_cmd{"COMMAND OPTIONS"};
opts_cmd.add_options()
("default-type", po::value<std::string>()->default_value("node"), "Default item type")
("id-file,i", po::value<std::vector<std::string>>(), "Read OSM IDs from text file")
("id-osm-file,I", po::value<std::vector<std::string>>(), "Read OSM IDs from OSM file")
("add-self,s", "Add objects with specified IDs themselves")
("verbose-ids", "Print all requested IDs")
;
po::options_description opts_common{add_common_options()};
po::options_description opts_input{add_single_input_options()};
po::options_description opts_output{add_output_options()};
po::options_description hidden;
hidden.add_options()
("input-filename", po::value<std::string>(), "OSM input file")
("ids", po::value<std::vector<std::string>>(), "OSM IDs")
;
po::options_description desc;
desc.add(opts_cmd).add(opts_common).add(opts_input).add(opts_output);
po::options_description parsed_options;
parsed_options.add(desc).add(hidden);
po::positional_options_description positional;
positional.add("input-filename", 1);
positional.add("ids", -1);
po::variables_map vm;
po::store(po::command_line_parser(arguments).options(parsed_options).positional(positional).run(), vm);
po::notify(vm);
setup_common(vm, desc);
setup_progress(vm);
setup_input_file(vm);
setup_output_file(vm);
if (vm.count("add-self")) {
m_add_self = true;
}
if (vm.count("default-type")) {
m_default_item_type = parse_item_type(vm["default-type"].as<std::string>());
}
if (vm.count("verbose-ids")) {
m_vout.verbose(true);
m_verbose_ids = true;
}
if (vm.count("id-file")) {
for (const std::string& filename : vm["id-file"].as<std::vector<std::string>>()) {
if (filename == "-") {
if (m_input_filename.empty() || m_input_filename == "-") {
throw argument_error{"Can not read OSM input and IDs both from STDIN."};
}
m_vout << "Reading IDs from STDIN...\n";
read_id_file(std::cin, m_ids, m_default_item_type);
} else {
std::ifstream id_file{filename};
if (!id_file.is_open()) {
throw argument_error{"Could not open file '" + filename + "'"};
}
m_vout << "Reading ID file...\n";
read_id_file(id_file, m_ids, m_default_item_type);
}
}
}
if (vm.count("id-osm-file")) {
for (const std::string& filename : vm["id-osm-file"].as<std::vector<std::string>>()) {
m_vout << "Reading OSM ID file...\n";
read_id_osm_file(filename, m_ids);
}
}
if (vm.count("ids")) {
std::string sids;
for (const auto& s : vm["ids"].as<std::vector<std::string>>()) {
sids += s + " ";
}
for (const auto& s : osmium::split_string(sids, "\t ;,/|", true)) {
parse_and_add_id(s, m_ids, m_default_item_type);
}
}
if (no_ids(m_ids)) {
throw argument_error{"Please specify IDs to look for on command line or with option --id-file/-i or --id-osm-file/-I."};
}
return true;
}
int
main (int argc, char **argv)
{
set_program_name (argv[0]);
idh.idh_file_name = 0;
#if ENABLE_NLS
/* Set locale according to user's wishes. */
setlocale (LC_ALL, "");
/* Tell program which translations to use and where to find. */
bindtextdomain (PACKAGE, LOCALEDIR);
textdomain (PACKAGE);
#endif
atexit (close_stdout);
for (;;)
{
int optc = getopt_long (argc, argv, "f:S:",
long_options, (int *) 0);
if (optc < 0)
break;
switch (optc)
{
case 0:
break;
case 'f':
idh.idh_file_name = optarg;
break;
case 'S':
separator_style = parse_separator_style (optarg);
break;
default:
usage ();
}
}
if (show_version)
{
printf ("%s - %s\n", program_name, PACKAGE_VERSION);
exit (EXIT_SUCCESS);
}
if (show_help)
help_me ();
if (separator_style == ss_contextual)
{
if (isatty (STDOUT_FILENO))
separator_style = DEFAULT_SEPARATOR_STYLE;
else
separator_style = ss_newline;
}
argc -= optind;
argv += optind;
if (argc == 0)
{
static char star[] = "*";
static char *starp = star;
argc = 1;
argv = &starp;
}
/* Look for the ID database up the tree */
idh.idh_file_name = locate_id_file_name (idh.idh_file_name);
if (idh.idh_file_name == 0)
error (EXIT_FAILURE, errno, _("can't locate `ID'"));
init_idh_obstacks (&idh);
init_idh_tables (&idh);
cw_dlink = get_current_dir_link ();
{
struct file_link **members = read_id_file (idh.idh_file_name, &idh);
struct file_link **members_N = &members[idh.idh_files];
struct file_link **flinkv_0 = xmalloc (sizeof(struct file_link *) * (idh.idh_files + 1));
struct file_link **flinkv = flinkv_0;
char **patv_0 = xmalloc (sizeof(char *) * (argc * 2));
char **patv_N;
char **patv = patv_0;
char *file_name = alloca (PATH_MAX);
for ( ; argc; argc--, argv++)
{
char *arg = *argv;
*patv++ = arg;
if (*arg != '*' && *arg != '/')
{
char *pat = xmalloc (strlen (arg) + 2);
sprintf (pat, "*/%s", arg);
*patv++ = pat;
}
}
patv_N = patv;
for ( ; members < members_N; members++)
{
maybe_relative_file_name (file_name, *members, cw_dlink);
for (patv = patv_0; patv < patv_N; patv++)
{
if (fnmatch (*patv, file_name, MAYBE_FNM_CASEFOLD) == 0)
{
*flinkv++ = *members;
break;
}
}
}
*flinkv = 0;
print_filenames (flinkv_0, separator_style);
}
return 0;
}
