Hunt The Wumpus/Rust

From Rosetta Code
Hunt The Wumpus/Rust is part of Hunt_The_Wumpus. You may find other members of Hunt_The_Wumpus at Category:Hunt_The_Wumpus.

Rust version of Hunt The Wumpus .

Code

//! *Hunt the Wumpus* reimplementation in Rust.

use rand;
use rand::prelude::SliceRandom;
use std::io;
use std::io::Write;
use std::process::exit;

/// Help message.
const HELP: &str = "\
Welcome to \"Hunt the Wumpus\"

The wumpus lives in a cave of 20 rooms. Each room has 3
tunnels to other rooms. (The tunnels form a dodecahedron:
http://en.wikipedia.org/wiki/dodecahedron)

Hazards:

 Bottomless pits: Two rooms have bottomless pits in them. If you go
   there, you fall into the pit (& lose)!

 Super bats: Two other rooms have super bats. If you go
   there, a bat grabs you and takes you to some other room
   at random (which may be troublesome).

Wumpus:

   The wumpus is not bothered by hazards. (He has sucker
   feet and is too big for a bat to lift.)  Usually he is
   asleep. Two things wake him up: your shooting an arrow,
   or your entering his room.  If the wumpus wakes, he moves
   one room or stays still.  After that, if he is where you
   are, he eats you up and you lose!

You:

   Each turn you may move or shoot a crooked arrow.

   Moving: You can move one room (through one tunnel).

   Arrows: You have 5 arrows. You lose when you run out.
      You can only shoot to nearby rooms. If the arrow hits
      the wumpus, you win.

Warnings:

   When you are one room away from a wumpus or hazard, the
   computer says:

   Wumpus:  \"You smell something terrible nearby.\"
   Bat:  \"You hear a rustling.\"
   Pit:  \"You feel a cold wind blowing from a nearby cavern.\"
";

/// The maze is an dodecahedron.
const MAZE_ROOMS: usize = 20;
const ROOM_NEIGHBORS: usize = 3;

/// Number of bats.
const BATS: usize = 2;
/// Number of pits.
const PITS: usize = 2;
/// Initial number of arrows.
const ARROWS: usize = 5;

/// Fractional chance of waking the Wumpus on entry to its room.
const WAKE_WUMPUS_PROB: f32 = 0.75;

type RoomNum = usize;

/// Description of the current player state.
struct Player {
    /// Player location.
    room: RoomNum,
    /// Remaining number of arrows.
    arrows: usize,
}

impl Player {
    /// Make a new player starting in the given room.
    fn new(room: RoomNum) -> Self {
        Player {
            arrows: ARROWS,
            room,
        }
    }
}

/// Dangerous things that can be in a room.
#[derive(PartialEq)]
enum Danger {
    Wumpus,
    Bat,
    Pit,
}

/// Room description.
#[derive(Default)]
struct Room {
    id: RoomNum,
    /// The indices of neighboring rooms.
    neighbours: [Option<RoomNum>; ROOM_NEIGHBORS],
    /// Possible danger in the room.
    dangers: Vec<Danger>,
}

impl Room {
    fn new(id: RoomNum) -> Self {
        let default_room = Room::default();

        Room { id, ..default_room }
    }

    fn neighbour_ids(&self) -> Vec<RoomNum> {
        self.neighbours.iter().cloned().filter_map(|n| n).collect()
    }
}

/// The Maze,
struct Maze {
    /// Room list.
    rooms: Vec<Room>,
}

impl Maze {
    // List of adjacencies used to wire up the dodecahedron.
    // https://stackoverflow.com/a/44096541/364875
    const ADJS: [[usize; 3]; 20] = [
        [1, 4, 7],
        [0, 2, 9],
        [1, 3, 11],
        [2, 4, 13],
        [0, 3, 5],
        [4, 6, 14],
        [5, 7, 16],
        [0, 6, 8],
        [7, 9, 17],
        [1, 8, 10],
        [9, 11, 18],
        [2, 10, 12],
        [11, 13, 19],
        [3, 12, 14],
        [5, 13, 15],
        [14, 16, 19],
        [6, 15, 17],
        [8, 16, 18],
        [10, 17, 19],
        [12, 15, 18],
    ];

    // Builds a vector of rooms comprising a dodecahedron.
    fn new() -> Self {
        let mut rooms: Vec<Room> = (0..MAZE_ROOMS)
            .map(|idx| Room::new(idx as RoomNum))
            .collect();

        for (i, room) in rooms.iter_mut().enumerate() {
            for (j, nb) in room.neighbours.iter_mut().enumerate() {
                *nb = Some(Maze::ADJS[i][j]);
            }
        }

        let mut maze = Maze { rooms };

        // place the wumpus, pits and bats in empty rooms
        let empty_room = maze.rnd_empty_room();
        maze.rooms[empty_room].dangers.push(Danger::Wumpus);

        for _ in 0..PITS {
            let empty_room = maze.rnd_empty_room();
            maze.rooms[empty_room].dangers.push(Danger::Pit);
        }

        for _ in 0..BATS {
            let empty_room = maze.rnd_empty_room();
            maze.rooms[empty_room].dangers.push(Danger::Bat);
        }

        maze
    }

    /// Return a randomly-selected empty room.
    fn rnd_empty_room(&mut self) -> RoomNum {
        let empty_rooms: Vec<_> = self.rooms.iter().filter(|n| n.dangers.is_empty()).collect();

        empty_rooms.choose(&mut rand::thread_rng()).unwrap().id
    }

    /// Retrun the id of a random empty neighbour if any
    fn rnd_empty_neighbour(&mut self, room: RoomNum) -> Option<RoomNum> {
        let neighbour_ids = self.rooms[room].neighbour_ids();

        let empty_neighbours: Vec<_> = neighbour_ids
            .iter()
            .filter(|&n| self.rooms[*n].dangers.is_empty())
            .collect();

        if empty_neighbours.is_empty() {
            return None;
        }

        let empty_neighbour = empty_neighbours.choose(&mut rand::thread_rng()).unwrap();

        Some(**empty_neighbour)
    }

    /// Current room description string.
    fn describe_room(&self, room: RoomNum) -> String {
        let mut description = format!("You are in room #{}", room);

        if self.is_danger_nearby(room, Danger::Pit) {
            description.push_str("\nYou feel a cold wind blowing from a nearby cavern.");
        }
        if self.is_danger_nearby(room, Danger::Bat) {
            description.push_str("\nYou hear a rustling.");
        }
        if self.is_danger_nearby(room, Danger::Wumpus) {
            description.push_str("\nYou smell something terrible nearby.");
        }

        description.push_str(&format!(
            "\nExits go to: {}",
            self.rooms[room]
                .neighbours
                .iter()
                .map(|n| n.unwrap().to_string())
                .collect::<Vec<String>>()
                .join(", ")
        ));

        description
    }

    /// Adjacent room contains a non-wumpus danger.
    fn is_danger_nearby(&self, room: RoomNum, danger: Danger) -> bool {
        self.rooms[room]
            .neighbours
            .iter()
            .any(|n| self.rooms[n.unwrap()].dangers.contains(&danger))
    }

    /// Index of neighboring room given by user `destination`, else an error message.
    fn parse_room(&self, destination: &str, current_room: RoomNum) -> Result<RoomNum, ()> {
        let destination: Result<RoomNum, _> = destination.parse();

        // check that the given destination is both a number an the number of a linked room
        if let Ok(room) = destination {
            if self.rooms[current_room].neighbour_ids().contains(&room) {
                return Ok(room);
            }
        }

        Err(())
    }
}

/// Current game state.
enum Status {
    Normal,
    Quitting,
    Moving,
    Shooting,
}

fn main() {
    let mut maze = Maze::new();
    let mut player = Player::new(maze.rnd_empty_room());
    let mut status = Status::Normal;

    let describe = |maze: &Maze, player: &Player| {
        println!("{}", maze.describe_room(player.room));
        println!("What do you want to do? (m)ove or (s)hoot?");
    };

    let prompt = || {
        print!("> ");
        io::stdout().flush().expect("Error flushing");
    };

    describe(&maze, &player);
    prompt();

    // main loop
    loop {
        let mut input = String::new();
        io::stdin()
            .read_line(&mut input)
            .expect("Cannot read from stdin");
        let input: &str = &input.trim().to_lowercase();

        match status {
            Status::Quitting => match input {
                "y" => {
                    println!("Goodbye, braveheart!");
                    exit(0);
                }
                "n" => {
                    println!("Good. the Wumpus is looking for you!");
                    status = Status::Normal;
                }
                _ => println!("That doesn't make any sense"),
            },
            Status::Moving => {
                if let Ok(room) = maze.parse_room(input, player.room) {
                    if maze.rooms[room].dangers.contains(&Danger::Wumpus) {
                        println!("The wumpus ate you up!\nGAME OVER");
                        exit(0);
                    } else if maze.rooms[room].dangers.contains(&Danger::Pit) {
                        println!("You fall into a bottomless pit!\nGAME OVER");
                        exit(0);
                    } else if maze.rooms[room].dangers.contains(&Danger::Bat) {
                        println!("The bats whisk you away!");
                        player.room = maze.rnd_empty_room();
                    } else {
                        player.room = room;
                    }

                    status = Status::Normal;
                    describe(&maze, &player);
                } else {
                    println!(
                        "There are no tunnels from here to that room. Where do you wanto do go?"
                    );
                }
            }
            Status::Shooting => {
                if let Ok(room) = maze.parse_room(input, player.room) {
                    if maze.rooms[room].dangers.contains(&Danger::Wumpus) {
                        println!("YOU KILLED THE WUMPUS! GOOD JOB, BUDDY!!!");
                        exit(0);
                    } else {
                        // 75% chances of waking up the wumpus that would go into another room
                        if rand::random::<f32>() < WAKE_WUMPUS_PROB {
                            let wumpus_room = maze
                                .rooms
                                .iter()
                                .find(|r| r.dangers.contains(&Danger::Wumpus))
                                .unwrap()
                                .id;

                            if let Some(new_wumpus_room) = maze.rnd_empty_neighbour(wumpus_room) {
                                if new_wumpus_room == player.room {
                                    println!("You woke up the wumpus and he ate you!\nGAME OVER");
                                    exit(1);
                                }

                                maze.rooms[wumpus_room]
                                    .dangers
                                    .retain(|d| d != &Danger::Wumpus);
                                maze.rooms[new_wumpus_room].dangers.push(Danger::Wumpus);
                                println!("You heard a rumbling in a nearby cavern.");
                            }
                        }

                        player.arrows -= 1;
                        if player.arrows == 0 {
                            println!("You ran out of arrows.\nGAME OVER");
                            exit(1);
                        }

                        status = Status::Normal;
                    }
                } else {
                    println!(
                        "There are no tunnels from here to that room. Where do you wanto do shoot?"
                    );
                }
            }
            _ => match input {
                "h" => println!("{}", HELP),
                "q" => {
                    println!("Are you so easily scared? [y/n]");
                    status = Status::Quitting;
                }
                "m" => {
                    println!("Where?");
                    status = Status::Moving;
                }
                "s" => {
                    println!("Where?");
                    status = Status::Shooting;
                }
                _ => println!("That doesn't make any sense"),
            },
        }

        prompt();
    }
}

#[test]
fn test_maze_connected() {
    use std::collections::HashSet;
    let maze = Maze::new();
    let n = maze.rooms.len();

    fn exists_path(i: RoomNum, j: RoomNum, vis: &mut HashSet<RoomNum>, maze: &Maze) -> bool {
        if i == j {
            return true;
        }
        vis.insert(i);
        maze.rooms[i].neighbours.iter().any(|neighbour| {
            // Check that all rooms have three neighbors.
            let k = neighbour.unwrap();
            !vis.contains(&k) && exists_path(k, j, vis, maze)
        })
    }
    for i in 0..n {
        for j in 0..n {
            assert!(exists_path(i, j, &mut HashSet::new(), &maze));
        }
    }
}