159 lines
4.0 KiB
Rust
159 lines
4.0 KiB
Rust
use clap::{App, Arg};
|
|
use std::fs;
|
|
use std::io;
|
|
|
|
/// Struct to store the world.
|
|
struct World {
|
|
x: u32,
|
|
y: u32,
|
|
map: Vec<char>,
|
|
}
|
|
|
|
/// Struct to store robot position.
|
|
struct Position {
|
|
x: u32,
|
|
y: u32,
|
|
}
|
|
|
|
/// A Robot *aka droid* is represented here.
|
|
/// Each robot must have a unique id.
|
|
struct Robot {
|
|
id: u32,
|
|
o: Orientation,
|
|
p: Position,
|
|
}
|
|
|
|
impl Robot {
|
|
/// Apply forward instruction to the robot.
|
|
fn move_forward(&mut self) {
|
|
match self.o {
|
|
Orientation::N => self.p.y -= 1,
|
|
Orientation::S => self.p.y += 1,
|
|
Orientation::E => self.p.x += 1,
|
|
Orientation::W => self.p.x -= 1,
|
|
}
|
|
}
|
|
/// Apply right instruction to the robot.
|
|
fn move_right(&mut self) {
|
|
match self.o {
|
|
Orientation::N => self.p.x += 1,
|
|
Orientation::S => self.p.x -= 1,
|
|
Orientation::E => self.p.y += 1,
|
|
Orientation::W => self.p.y -= 1,
|
|
}
|
|
}
|
|
/// Apply left instruction to the robot.
|
|
fn move_left(&mut self) {
|
|
match self.o {
|
|
Orientation::N => self.p.x -= 1,
|
|
Orientation::S => self.p.x += 1,
|
|
Orientation::E => self.p.y -= 1,
|
|
Orientation::W => self.p.y += 1,
|
|
}
|
|
}
|
|
/// Apply North orientation to the robot.
|
|
fn to_north(&mut self) {
|
|
self.o = Orientation::N
|
|
}
|
|
/// Apply South orientation to the robot.
|
|
fn to_south(&mut self) {
|
|
self.o = Orientation::S
|
|
}
|
|
/// Apply East orientation to the robot.
|
|
fn to_east(&mut self) {
|
|
self.o = Orientation::E
|
|
}
|
|
/// Apply West orientation to the robot.
|
|
fn to_west(&mut self) {
|
|
self.o = Orientation::W
|
|
}
|
|
}
|
|
|
|
/// Enum to store all possible orientations.
|
|
enum Orientation {
|
|
N,
|
|
E,
|
|
S,
|
|
W,
|
|
}
|
|
|
|
/// Enum to store all possible instructions.
|
|
enum Instruction {
|
|
L,
|
|
R,
|
|
F,
|
|
}
|
|
|
|
/// Parse char and return corresponding orientation.
|
|
fn parse_orientation(c: char) -> Result<Orientation, &'static str> {
|
|
match c {
|
|
'N' => Ok(Orientation::N),
|
|
'E' => Ok(Orientation::E),
|
|
'S' => Ok(Orientation::S),
|
|
'W' => Ok(Orientation::W),
|
|
_ => Err("Invalid character, does not match any orientations"),
|
|
}
|
|
}
|
|
|
|
/// Parse char and return corresponding instruction.
|
|
fn parse_instruction(c: char) -> Result<Instruction, &'static str> {
|
|
match c {
|
|
'L' => Ok(Instruction::L),
|
|
'R' => Ok(Instruction::R),
|
|
'F' => Ok(Instruction::F),
|
|
_ => Err("Invalid character, does not match any instructions"),
|
|
}
|
|
}
|
|
|
|
/// Retrieve the content of a file and return it as a string.
|
|
fn open_file(filename: &str) -> io::Result<String> {
|
|
let content = fs::read_to_string(filename)?;
|
|
Ok(content)
|
|
}
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let matches = App::new("DancingDroids")
|
|
.version("0.1.0")
|
|
.about("When droids dance togethers")
|
|
.arg(
|
|
Arg::with_name("file")
|
|
.short("f")
|
|
.long("file")
|
|
.takes_value(true)
|
|
.help("Configuration file"),
|
|
)
|
|
.get_matches();
|
|
|
|
let mut robot_pool: Vec<Robot> = Vec::new();
|
|
let raw_conf = open_file(matches.value_of("file").unwrap_or("two_robots.txt"))?;
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_parse_orientation() {
|
|
assert!(parse_orientation('N').is_ok());
|
|
assert!(parse_orientation('E').is_ok());
|
|
assert!(parse_orientation('S').is_ok());
|
|
assert!(parse_orientation('W').is_ok());
|
|
assert!(parse_orientation('Z').is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_instruction() {
|
|
assert!(parse_instruction('L').is_ok());
|
|
assert!(parse_instruction('R').is_ok());
|
|
assert!(parse_instruction('F').is_ok());
|
|
assert!(parse_instruction('Z').is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_open_file() {
|
|
assert!(open_file("two_robots.txt").is_ok());
|
|
assert!(open_file("test_unexisting_file.extension").is_err());
|
|
}
|
|
}
|