You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
2.6 KiB
114 lines
2.6 KiB
use std::collections::HashSet;
|
|
use std::convert::TryFrom;
|
|
|
|
use rusqlite::params;
|
|
|
|
use crate::database::Db;
|
|
use crate::id::Id;
|
|
use crate::result::RudeResult;
|
|
use crate::try_log;
|
|
use crate::world::Zone;
|
|
|
|
impl Db {
|
|
/// Get root zone
|
|
pub fn root_zone(&self, zone: &Zone) -> RudeResult<Zone> {
|
|
let mut zone: Zone = zone.into();
|
|
|
|
// loop until find the root
|
|
loop {
|
|
zone = match self.load_zone(zone.id) {
|
|
Ok(Some(z)) => z,
|
|
Ok(None) => {
|
|
log::error!(
|
|
"{}({}) :: Can't find parent zone :: None value",
|
|
file!(),
|
|
line!()
|
|
);
|
|
return Err(Box::from("None value"));
|
|
}
|
|
Err(e) => return Err(e),
|
|
};
|
|
|
|
if zone.id == zone.parent {
|
|
return Ok(zone);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Save a zone to the database.
|
|
pub fn save_zone(&self, zone: &Zone) -> RudeResult<()> {
|
|
let mut statement = try_log!(
|
|
self.0.prepare(
|
|
"insert into zones (id, parent, name, users_visible) values (?, ?, ?, ?) on conflict(id) do update set id=?, parent=?, name=?, users_visible=?;"
|
|
),
|
|
"Unable to prepare sql statement"
|
|
);
|
|
|
|
let _ = try_log!(
|
|
statement.execute(params![
|
|
zone.id,
|
|
zone.parent,
|
|
zone.name,
|
|
zone.users_visible,
|
|
zone.id,
|
|
zone.parent,
|
|
zone.name,
|
|
zone.users_visible,
|
|
]),
|
|
"Unable to perform query"
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Load a zone from the database.
|
|
pub fn load_zone(&self, id: Id) -> RudeResult<Option<Zone>> {
|
|
let mut statement = try_log!(
|
|
self.0
|
|
.prepare("select id, parent, name, users_visible from zones where id = ?"),
|
|
"Unable to prepare sql statement"
|
|
);
|
|
|
|
let mut rows = try_log!(statement.query(params![id]), "Unable to perform query");
|
|
|
|
let mut zone = if let Some(row) = try_log!(rows.next(), "Unable to retrieve row") {
|
|
try_log!(Zone::try_from(row), "Unable to get Zone from Row")
|
|
} else {
|
|
return Ok(None);
|
|
};
|
|
|
|
zone.areas = {
|
|
let mut a = HashSet::new();
|
|
|
|
let mut statement = try_log!(
|
|
self.0.prepare("select id from zones where parent = ?;"),
|
|
"Unable to prepare sql statement"
|
|
);
|
|
|
|
let mut rows = try_log!(statement.query(params![id]), "Unable to perform query");
|
|
|
|
while let Some(row) = try_log!(rows.next(), "Unable to retrieve row") {
|
|
let new_id = try_log!(row.get(0), "Unable to retrieve field");
|
|
if new_id != id {
|
|
a.insert(new_id);
|
|
}
|
|
}
|
|
|
|
let mut statement = try_log!(
|
|
self.0.prepare("select id from rooms where zone = ?;"),
|
|
"Unable to prepare sql statement"
|
|
);
|
|
|
|
let mut rows = try_log!(statement.query(params![id]), "Unable to perform query");
|
|
|
|
while let Some(row) = try_log!(rows.next(), "Unable to retrieve row") {
|
|
a.insert(try_log!(row.get(0), "Unable to retrieve field"));
|
|
}
|
|
|
|
a
|
|
};
|
|
|
|
Ok(Some(zone))
|
|
}
|
|
}
|