bendns' repos / panel
html terminal
Diffstat (limited to 'src/bot/maps.rs')
-rw-r--r--src/bot/maps.rs41
1 files changed, 41 insertions, 0 deletions
diff --git a/src/bot/maps.rs b/src/bot/maps.rs
new file mode 100644
index 0000000..ba330d7
--- /dev/null
+++ b/src/bot/maps.rs
@@ -0,0 +1,41 @@
+use super::{get_nextblock, strip_colors, Context};
+use crate::send;
+use futures_util::StreamExt;
+use tokio::sync::broadcast;
+use tokio::sync::OnceCell;
+
+pub struct Maps;
+impl Maps {
+ pub async fn find(map: &str, stdin: &broadcast::Sender<String>) -> usize {
+ Self::get_all(stdin)
+ .await
+ .iter()
+ .position(|r| r == map)
+ .unwrap()
+ }
+
+ pub async fn get_all(stdin: &broadcast::Sender<String>) -> &Vec<String> {
+ static MAPS: OnceCell<Vec<String>> = OnceCell::const_new();
+ MAPS.get_or_init(|| async move {
+ send!(stdin, "maps").unwrap();
+ let res = get_nextblock().await;
+ let mut vec = vec![];
+ for line in res.lines() {
+ if let Some((_, name)) = line.split_once(':') {
+ vec.push(strip_colors(name));
+ }
+ }
+ vec
+ })
+ .await
+ }
+}
+
+pub async fn autocomplete<'a>(
+ ctx: Context<'a>,
+ partial: &'a str,
+) -> impl futures::Stream<Item = String> + 'a {
+ futures::stream::iter(Maps::get_all(&ctx.data().stdin).await)
+ .filter(move |name| futures::future::ready(name.starts_with(partial)))
+ .map(|name| name.to_string())
+}