custom_flag2.cc

// Copyright (c) 2020 The Console Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <iostream>

#include "console/autocompletion.h"
#include "console/flag.h"
#include "console/stream.h"

struct OddNumber {
  uint16_t number;
};

namespace console {

template <>
class FlagValueTraits<OddNumber> {
 public:
  static bool ParseValue(absl::string_view input, OddNumber* value,
                         std::string* reason) {
    uint16_t n;
    if (!console::FlagValueTraits<uint16_t>::ParseValue(input, &n, reason)) {
      return false;
    }
    if (n % 2 == 0) {
      *reason = absl::StrFormat("%u is not a odd number", n);
      return false;
    }
    value->number = n;
    return true;
  }
};

}  // namespace console

int main(int argc, char** argv) {
#if defined(OS_WIN)
  console::Console::EnableAnsi(std::cout);
#endif

  OddNumber number;
  console::FlagParser flag_parser;
  flag_parser.set_program_name("custom_flag2");
  flag_parser.AddFlag<console::Flag<OddNumber>>(&number)
      .set_name("number")
      .set_required()
      .set_help("Please input only odd numbers!");

  // Uncomment if you want to generate json file used by console-autocomplete!
  // console::Autocompletion::WriteToJson(flag_parser, "custom_flag2.json");

  if (!flag_parser.Parse(argc, argv)) {
    {
      console::Stream stream(std::cerr);
      stream.Red();
      std::cerr << "[ERROR]: ";
    }
    std::cerr << flag_parser.error_message() << std::endl;
    return 1;
  }

  return 0;
}