| /* |
| * Copyright (C) 2018 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <cstdlib> // EXIT_{FAILURE,SUCCESS} |
| #include <functional> |
| #include <iostream> |
| #include <map> |
| #include <memory> |
| #include <ostream> |
| #include <string> |
| #include <vector> |
| |
| #include "Commands.h" |
| #include "idmap2/CommandLineOptions.h" |
| #include "idmap2/Result.h" |
| #include "idmap2/SysTrace.h" |
| |
| using android::idmap2::CommandLineOptions; |
| using android::idmap2::Result; |
| using android::idmap2::Unit; |
| |
| using NameToFunctionMap = |
| std::map<std::string, std::function<Result<Unit>(const std::vector<std::string>&)>>; |
| |
| namespace { |
| |
| void PrintUsage(const NameToFunctionMap& commands, std::ostream& out) { |
| out << "usage: idmap2 ["; |
| for (auto iter = commands.cbegin(); iter != commands.cend(); iter++) { |
| if (iter != commands.cbegin()) { |
| out << "|"; |
| } |
| out << iter->first; |
| } |
| out << "]" << std::endl; |
| } |
| |
| } // namespace |
| |
| int main(int argc, char** argv) { |
| SYSTRACE << "main"; |
| const NameToFunctionMap commands = { |
| {"create", Create}, |
| {"create-multiple", CreateMultiple}, |
| {"dump", Dump}, |
| {"lookup", Lookup}, |
| }; |
| if (argc <= 1) { |
| PrintUsage(commands, std::cerr); |
| return EXIT_FAILURE; |
| } |
| const std::unique_ptr<std::vector<std::string>> args = |
| CommandLineOptions::ConvertArgvToVector(argc - 1, const_cast<const char**>(argv + 1)); |
| if (!args) { |
| std::cerr << "error: failed to parse command line options" << std::endl; |
| return EXIT_FAILURE; |
| } |
| const auto iter = commands.find(argv[1]); |
| if (iter == commands.end()) { |
| std::cerr << argv[1] << ": command not found" << std::endl; |
| PrintUsage(commands, std::cerr); |
| return EXIT_FAILURE; |
| } |
| const auto result = iter->second(*args); |
| if (!result) { |
| std::cerr << "error: " << result.GetErrorMessage() << std::endl; |
| return EXIT_FAILURE; |
| } |
| return EXIT_SUCCESS; |
| } |