monorepo/kifu/core/src/database.rs

use std::{ffi::OsStr, io::Read, os::unix::ffi::OsStrExt, path::PathBuf};

use go_sgf::{parse_sgf, GameTree};
use thiserror::Error;

#[derive(Error, Debug)]
pub enum Error {
    #[error("Database permission denied")]
    PermissionDenied,
    #[error("An IO error occurred: {0}")]
    IOError(std::io::Error),
}

impl From<std::io::Error> for Error {
    fn from(err: std::io::Error) -> Self {
        Error::IOError(err)
    }
}

pub struct Database {
    path: PathBuf,
    games: Vec<GameTree>,
}

impl Database {
    pub fn open_path(path: PathBuf) -> Result<Database, Error> {
        let mut games: Vec<GameTree> = Vec::new();

        let extension = PathBuf::from("sgf").into_os_string();

        let path_iter = std::fs::read_dir(path.clone())?;
        for entry in path_iter {
            match entry {
                Ok(entry) => {
                    if entry.path().extension() == Some(&extension) {
                        let mut buffer = String::new();
                        std::fs::File::open(entry.path())
                            .unwrap()
                            .read_to_string(&mut buffer)
                            .unwrap();
                        let sgf = parse_sgf(&buffer).unwrap();
                        games.extend(sgf);
                    }
                }
                Err(err) => println!("failed entry: {:?}", err),
            }
        }

        Ok(Database { path, games })
    }

    pub fn all_games(&self) -> impl Iterator<Item = &GameTree> {
        self.games.iter()
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use cool_asserts::assert_matches;
    use go_sgf::{Date, GameType};

    #[test]
    fn it_reads_empty_database() {
        let db = Database::open_path(PathBuf::from("fixtures/empty_database/"))
            .expect("database to open");
        assert_eq!(db.all_games().count(), 0);
    }

    #[test]
    fn it_reads_five_games_from_database() {
        let db =
            Database::open_path(PathBuf::from("fixtures/five_games/")).expect("database to open");
        assert_eq!(db.all_games().count(), 5);
        for game in db.all_games() {
            assert_eq!(game.game_type, GameType::Go);
        }

        assert_matches!(db.all_games().find(|g| g.info.black_player == Some("Steve".to_owned())),
        Some(game) => {
            assert_eq!(game.info.black_player, Some("Steve".to_owned()));
            assert_eq!(game.info.white_player, Some("Savanni".to_owned()));
            assert_eq!(game.info.date, vec![Date::Date(chrono::NaiveDate::from_ymd_opt(2023, 4, 19).unwrap())]);
            assert_eq!(game.info.komi, Some(6.5));
        }
        );
    }
}