media/cast/test/utility/input_builder.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "media/cast/test/utility/input_builder.h"
   6
   7 #include <stdlib.h>
   8 #include <cstdio>
   9
  10 #include "base/command_line.h"
  11 #include "base/logging.h"
  12 #include "base/strings/string_number_conversions.h"
  13
  14 namespace media {
  15 namespace cast {
  16 namespace test {
  17
  18 static const char kEnablePromptsSwitch[] = "enable-prompts";
  19
  20 InputBuilder::InputBuilder(const std::string& title,
  21                            const std::string& default_value,
  22                            int low_range,
  23                            int high_range)
  24     : title_(title),
  25       default_value_(default_value),
  26       low_range_(low_range),
  27       high_range_(high_range) {}
  28
  29 InputBuilder::~InputBuilder() {}
  30
  31 std::string InputBuilder::GetStringInput() const {
  32   if (!base::CommandLine::ForCurrentProcess()->HasSwitch(kEnablePromptsSwitch))
  33     return default_value_;
  34
  35   printf("\n%s\n", title_.c_str());
  36   if (!default_value_.empty())
  37     printf("Hit enter for default (%s):\n", default_value_.c_str());
  38
  39   printf("# ");
  40   fflush(stdout);
  41   char raw_input[128];
  42   if (!fgets(raw_input, 128, stdin)) {
  43     NOTREACHED();
  44     return std::string();
  45   }
  46
  47   std::string input = raw_input;
  48   input = input.substr(0, input.size() - 1);  // Strip last \n.
  49   if (input.empty() && !default_value_.empty())
  50     return default_value_;
  51
  52   if (!ValidateInput(input)) {
  53     printf("Invalid input. Please try again.\n");
  54     return GetStringInput();
  55   }
  56   return input;
  57 }
  58
  59 int InputBuilder::GetIntInput() const {
  60   std::string string_input = GetStringInput();
  61   int int_value;
  62   CHECK(base::StringToInt(string_input, &int_value));
  63   return int_value;
  64 }
  65
  66 bool InputBuilder::ValidateInput(const std::string input) const {
  67   // Check for a valid range.
  68   if (low_range_ == INT_MIN && high_range_ == INT_MAX)
  69     return true;
  70   int value;
  71   if (!base::StringToInt(input, &value))
  72     return false;
  73   return value >= low_range_ && value <= high_range_;
  74 }
  75
  76 }  // namespace test
  77 }  // namespace cast
  78 }  // namespace media