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
InputBuilder(const std::string & title,const std::string & default_value,int low_range,int high_range)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
~InputBuilder()29 InputBuilder::~InputBuilder() {}
30
GetStringInput() const31 std::string InputBuilder::GetStringInput() const {
32 if (!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
GetIntInput() const59 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
ValidateInput(const std::string input) const66 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
79