mirror of
https://github.com/rust-lang/rustlings.git
synced 2024-11-08 09:09:17 +01:00
91 lines
2.7 KiB
Rust
91 lines
2.7 KiB
Rust
// Let's build a little machine in the form of a function. As input, we're going
|
|
// to give a list of strings and commands. These commands determine what action
|
|
// is going to be applied to the string. It can either be:
|
|
// - Uppercase the string
|
|
// - Trim the string
|
|
// - Append "bar" to the string a specified amount of times
|
|
//
|
|
// The exact form of this will be:
|
|
// - The input is going to be a vector of 2-length tuples,
|
|
// the first element is the string, the second one is the command.
|
|
// - The output element is going to be a vector of strings.
|
|
|
|
enum Command {
|
|
Uppercase,
|
|
Trim,
|
|
Append(usize),
|
|
}
|
|
|
|
mod my_module {
|
|
use super::Command;
|
|
|
|
// The solution with a loop. Check out `transformer_iter` for a version
|
|
// with iterators.
|
|
pub fn transformer(input: Vec<(String, Command)>) -> Vec<String> {
|
|
let mut output = Vec::new();
|
|
|
|
for (string, command) in input {
|
|
// Create the new string.
|
|
let new_string = match command {
|
|
Command::Uppercase => string.to_uppercase(),
|
|
Command::Trim => string.trim().to_string(),
|
|
Command::Append(n) => string + &"bar".repeat(n),
|
|
};
|
|
|
|
// Push the new string to the output vector.
|
|
output.push(new_string);
|
|
}
|
|
|
|
output
|
|
}
|
|
|
|
// Equivalent to `transform` but uses an iterator instead of a loop for
|
|
// comparison. Don't worry, we will practice iterators later ;)
|
|
pub fn transformer_iter(input: Vec<(String, Command)>) -> Vec<String> {
|
|
input
|
|
.into_iter()
|
|
.map(|(string, command)| match command {
|
|
Command::Uppercase => string.to_uppercase(),
|
|
Command::Trim => string.trim().to_string(),
|
|
Command::Append(n) => string + &"bar".repeat(n),
|
|
})
|
|
.collect()
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
// You can optionally experiment here.
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
// Import `transformer`.
|
|
use super::my_module::transformer;
|
|
|
|
use super::my_module::transformer_iter;
|
|
use super::Command;
|
|
|
|
#[test]
|
|
fn it_works() {
|
|
for transformer in [transformer, transformer_iter] {
|
|
let input = vec![
|
|
("hello".to_string(), Command::Uppercase),
|
|
(" all roads lead to rome! ".to_string(), Command::Trim),
|
|
("foo".to_string(), Command::Append(1)),
|
|
("bar".to_string(), Command::Append(5)),
|
|
];
|
|
let output = transformer(input);
|
|
|
|
assert_eq!(
|
|
output,
|
|
[
|
|
"HELLO",
|
|
"all roads lead to rome!",
|
|
"foobar",
|
|
"barbarbarbarbarbar",
|
|
]
|
|
);
|
|
}
|
|
}
|
|
}
|