| 1 | // |
| 2 | // Copyright 2019 The Abseil Authors. |
| 3 | // |
| 4 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | // you may not use this file except in compliance with the License. |
| 6 | // You may obtain a copy of the License at |
| 7 | // |
| 8 | // https://www.apache.org/licenses/LICENSE-2.0 |
| 9 | // |
| 10 | // Unless required by applicable law or agreed to in writing, software |
| 11 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | // See the License for the specific language governing permissions and |
| 14 | // limitations under the License. |
| 15 | |
| 16 | #ifndef ABSL_FLAGS_INTERNAL_FLAG_H_ |
| 17 | #define ABSL_FLAGS_INTERNAL_FLAG_H_ |
| 18 | |
| 19 | #include "absl/flags/internal/commandlineflag.h" |
| 20 | #include "absl/flags/internal/registry.h" |
| 21 | |
| 22 | namespace absl { |
| 23 | namespace flags_internal { |
| 24 | |
| 25 | // This is "unspecified" implementation of absl::Flag<T> type. |
| 26 | template <typename T> |
| 27 | class Flag { |
| 28 | public: |
| 29 | constexpr Flag(const char* name, const flags_internal::HelpGenFunc help_gen, |
| 30 | const char* filename, |
| 31 | const flags_internal::FlagMarshallingOpFn marshalling_op, |
| 32 | const flags_internal::InitialValGenFunc initial_value_gen) |
| 33 | : internal(name, flags_internal::HelpText::FromFunctionPointer(help_gen), |
| 34 | filename, &flags_internal::FlagOps<T>, marshalling_op, |
| 35 | initial_value_gen, |
| 36 | /*retired_arg=*/false, /*def_arg=*/nullptr, |
| 37 | /*cur_arg=*/nullptr) {} |
| 38 | |
| 39 | // Not copyable/assignable. |
| 40 | Flag(const Flag<T>&) = delete; |
| 41 | Flag<T>& operator=(const Flag<T>&) = delete; |
| 42 | |
| 43 | absl::string_view Name() const { return internal.Name(); } |
| 44 | std::string Help() const { return internal.Help(); } |
| 45 | std::string Filename() const { return internal.Filename(); } |
| 46 | |
| 47 | absl::flags_internal::CommandLineFlag internal; |
| 48 | |
| 49 | void SetCallback(const flags_internal::FlagCallback mutation_callback) { |
| 50 | internal.SetCallback(mutation_callback); |
| 51 | } |
| 52 | |
| 53 | private: |
| 54 | // TODO(rogeeff): add these validations once UnparseFlag invocation is fixed |
| 55 | // for built-in types and when we cleanup existing code from operating on |
| 56 | // forward declared types. |
| 57 | // auto IsCopyConstructible(const T& v) -> decltype(T(v)); |
| 58 | // auto HasAbslParseFlag(absl::string_view in, T* dst, std::string* err) |
| 59 | // -> decltype(AbslParseFlag(in, dst, err)); |
| 60 | // auto HasAbslUnparseFlag(const T& v) -> decltype(AbslUnparseFlag(v)); |
| 61 | }; |
| 62 | |
| 63 | // This class facilitates Flag object registration and tail expression-based |
| 64 | // flag definition, for example: |
| 65 | // ABSL_FLAG(int, foo, 42, "Foo help").OnUpdate(NotifyFooWatcher); |
| 66 | template <typename T, bool do_register> |
| 67 | class FlagRegistrar { |
| 68 | public: |
| 69 | explicit FlagRegistrar(Flag<T>* flag) : flag_(flag) { |
| 70 | if (do_register) flags_internal::RegisterCommandLineFlag(&flag_->internal); |
| 71 | } |
| 72 | |
| 73 | FlagRegistrar& OnUpdate(flags_internal::FlagCallback cb) && { |
| 74 | flag_->SetCallback(cb); |
| 75 | return *this; |
| 76 | } |
| 77 | |
| 78 | // Make the registrar "die" gracefully as a bool on a line where registration |
| 79 | // happens. Registrar objects are intended to live only as temporary. |
| 80 | operator bool() const { return true; } // NOLINT |
| 81 | |
| 82 | private: |
| 83 | Flag<T>* flag_; // Flag being registered (not owned). |
| 84 | }; |
| 85 | |
| 86 | // This struct and corresponding overload to MakeDefaultValue are used to |
| 87 | // facilitate usage of {} as default value in ABSL_FLAG macro. |
| 88 | struct EmptyBraces {}; |
| 89 | |
| 90 | template <typename T> |
| 91 | T* MakeFromDefaultValue(T t) { |
| 92 | return new T(std::move(t)); |
| 93 | } |
| 94 | |
| 95 | template <typename T> |
| 96 | T* MakeFromDefaultValue(EmptyBraces) { |
| 97 | return new T; |
| 98 | } |
| 99 | |
| 100 | } // namespace flags_internal |
| 101 | } // namespace absl |
| 102 | |
| 103 | #endif // ABSL_FLAGS_INTERNAL_FLAG_H_ |
| 104 | |