src/media/cast/test/utility/input_helper.cc

   1 // Copyright 2013 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_helper.h"
   6
   7 #include <stdlib.h>
   8 #include <cstdio>
   9
  10 #include "base/logging.h"
  11 #include "base/strings/string_number_conversions.h"
  12
  13 namespace media {
  14 namespace cast {
  15 namespace test {
  16
  17 InputBuilder::InputBuilder(const std::string& title,
  18                            const std::string& default_value,
  19                            int low_range,
  20                            int high_range)
  21     : title_(title),
  22       default_value_(default_value),
  23       low_range_(low_range),
  24       high_range_(high_range) {}
  25
  26 InputBuilder::~InputBuilder() {}
  27
  28 std::string InputBuilder::GetStringInput() const {
  29   printf("\n%s\n", title_.c_str());
  30   if (!default_value_.empty())
  31     printf("Hit enter for default (%s):\n", default_value_.c_str());
  32
  33   printf("# ");
  34   fflush(stdout);
  35   char raw_input[128];
  36   if (!fgets(raw_input, 128, stdin)) {
  37     NOTREACHED();
  38     return std::string();
  39   }
  40
  41   std::string input = raw_input;
  42   input = input.substr(0, input.size() - 1);  // Strip last \n.
  43   if (input.empty() && !default_value_.empty())
  44     return default_value_;
  45
  46   if (!ValidateInput(input)) {
  47     printf("Invalid input. Please try again.\n");
  48     return GetStringInput();
  49   }
  50   return input;
  51 }
  52
  53 int InputBuilder::GetIntInput() const {
  54   std::string string_input = GetStringInput();
  55   int int_value;
  56   CHECK(base::StringToInt(string_input, &int_value));
  57   return int_value;
  58 }
  59
  60 bool InputBuilder::ValidateInput(const std::string input) const {
  61   // Check for a valid range.
  62   if (low_range_ == INT_MIN && high_range_ == INT_MAX) return true;
  63   int value;
  64   if (!base::StringToInt(input, &value))
  65     return false;
  66   return value >= low_range_ && value <= high_range_;
  67 }
  68
  69 }  // namespace test
  70 }  // namespace cast
  71 }  // namespace media