Big monolithic catchup commit

src/commands.rs

@@ -1,15 +1,18 @@
-use matrix_sdk::{identifiers::UserId, Client, Session};
+use matrix_sdk::{
+    ruma::{api::client::r0::alias, RoomAliasId, RoomId, UserId},
+    Client, Session, SyncSettings,
+};
 use rpassword::prompt_password_stderr;
-use serde_lexpr;
 use std::convert::TryFrom;
-use std::fs::File;
 use std::io::{self, Write};
 
-static SESSION_FILE: &str = "session_info";
+use crate::util::{build_client_config, restore_session, save_session};
 
-pub async fn login(username: Option<&str>) -> Result<(), Box<dyn std::error::Error>> {
-    let stdin = io::stdin();
+type CommandResult = Result<(), Box<dyn std::error::Error>>;
 
+macro_rules! room_fmt {() => {"{:30.30} | {:30.30} | {}"};}
+
+pub async fn login(username: Option<&str>) -> CommandResult {
     let user_id = match username {
         Some(s) => UserId::try_from(s)?,
         None => {
@@ -18,6 +21,7 @@ pub async fn login(username: Option<&str>) -> Result<(), Box<dyn std::error::Err
             eprint!("Username: ");
             io::stderr().flush().unwrap();
 
+            let stdin = io::stdin();
             stdin.read_line(&mut s)?;
             UserId::try_from(s.trim_end())?
         }
@@ -25,17 +29,11 @@ pub async fn login(username: Option<&str>) -> Result<(), Box<dyn std::error::Err
 
     let password = prompt_password_stderr("Password: ")?;
 
-    let client = Client::new_from_user_id(user_id.clone()).await?;
-    println!(
-        "{} logged in? {}",
-        client.homeserver().await,
-        client.logged_in().await
-    );
-
+    let client =
+        Client::new_from_user_id_with_config(user_id.clone(), build_client_config()).await?;
     let response = client
         .login(user_id.localpart(), &password, None, Some("mxadm"))
         .await?;
-    println!("{:?}", response);
     println!(
         "{} logged in? {}",
         client.homeserver().await,
@@ -47,41 +45,56 @@ pub async fn login(username: Option<&str>) -> Result<(), Box<dyn std::error::Err
         user_id: response.user_id,
         device_id: response.device_id,
     };
-    println!("{:?}", session_info);
-
-    let mut session_file = File::create(SESSION_FILE)?;
-    serde_lexpr::to_writer(&mut session_file, &session_info)?;
-
+    save_session(session_info)?;
     Ok(())
 }
 
-pub async fn status() -> Result<(), Box<dyn std::error::Error>> {
-    let session_file = match File::open(SESSION_FILE) {
-        Ok(f) => f,
-        _ => {
-            println!("Not logged in: no session info file found");
-            return Ok(())
-        }
-    };
-
-    let session_info: Session = serde_lexpr::from_reader(session_file)?;
-    let client = Client::new_from_user_id(session_info.user_id.clone()).await?;
-    if let Err(e) = client.restore_login(session_info).await {
-        println!("Not logged in: unable to restore session; {}", e);
-        return Ok(())
-    }
+pub async fn status() -> CommandResult {
+    let client = restore_session().await?;
 
     // Need to make a query requiring a valid token to check session validity
-    if let Ok(_) = client.devices().await {
-        println!(
-            "Logged in as {} ({}) with device ID {}",
-            client.user_id().await.unwrap(),
-            client.display_name().await?.unwrap(),
-            client.device_id().await.unwrap()
-        );
-    } else {
-        println!("Not logged in");
+    match client.devices().await {
+        Ok(_) => {
+            println!(
+                "Logged in as {} ({}) with device ID {}",
+                client.user_id().await.unwrap(),
+                client.display_name().await?.unwrap(),
+                client.device_id().await.unwrap()
+            );
+        }
+        Err(e) => println!("Not logged in: {:?}", e),
     }
 
     Ok(())
 }
+
+pub async fn list_rooms() -> CommandResult {
+    let client = restore_session().await?;
+
+    println!("Syncing...");
+    // client.sync_once(SyncSettings::default()).await?;
+    println!(" ...done");
+
+    println!("Joined rooms:");
+    println!(room_fmt!(), "Name", "Main alias", "Room ID");
+    println!(room_fmt!(), "----", "----------", "-------");
+    for r in client.joined_rooms() {
+        println!(
+            room_fmt!(),
+            r.display_name().await?,
+            r.canonical_alias().map_or("".into(), |a| a.into_string()),
+            r.room_id().as_str(),
+        );
+    }
+
+    Ok(())
+}
+
+pub async fn add_alias(room_id: &str, alias: &str) -> CommandResult {
+    let room_id = RoomId::try_from(room_id)?;
+    let alias_id = RoomAliasId::try_from(alias)?;
+    let client = restore_session().await?;
+
+
+    todo!()
+}

src/main.rs

@@ -2,6 +2,7 @@
 extern crate clap;
 
 mod commands;
+mod util;
 
 use clap::{App, SubCommand};
 
@@ -23,15 +24,31 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         .subcommand(SubCommand::with_name("logout").about("ends the current session"))
         .subcommand(SubCommand::with_name("status").about("displays current session status"))
         .subcommand(SubCommand::with_name("list-rooms").about("lists rooms available to the user"))
+        .subcommand(
+            SubCommand::with_name("add-alias")
+                .about("adds an alias to a room")
+                .args_from_usage(
+                    "<room_id>  'The ID of the room to alias'
+                     <alias>    'The new alias to add'",
+                ),
+        )
         .get_matches();
 
     match matches.subcommand() {
-        ("login", Some(login_matches)) => commands::login(login_matches.value_of("user")).await?,
-        ("logout", Some(_)) => println!("logging out..."),
+        ("login", Some(submatches)) => commands::login(submatches.value_of("user")).await?,
         ("status", Some(_)) => commands::status().await?,
-        ("list-rooms", Some(_)) => println!("listing rooms..."),
-        ("", None) => println!("No subcommand given"),
-        _ => unreachable!(),
+        ("list-rooms", Some(_)) => commands::list_rooms().await?,
+        ("add-alias", Some(submatches)) => {
+            commands::add_alias(
+                submatches.value_of("room_id").unwrap(),
+                submatches.value_of("alias").unwrap(),
+            )
+            .await?
+        }
+        ("", None) => eprintln!("No subcommand given"),
+        (c, _) => {
+            todo!("Subcommand '{}' not implemented yet!", c);
+        }
     }
 
     Ok(())

src/util.rs

@@ -0,0 +1,26 @@
+use matrix_sdk::{Client, ClientConfig, Session};
+use serde_lexpr;
+use std::fs::File;
+
+static SESSION_FILE: &str = "session_info";
+
+pub fn build_client_config() -> ClientConfig {
+    ClientConfig::new().store_path("./store")
+}
+
+pub fn save_session(session: Session) -> Result<(), Box<dyn std::error::Error>> {
+    let mut session_file = File::create(SESSION_FILE)?;
+    serde_lexpr::to_writer(&mut session_file, &session)?;
+    Ok(())
+}
+
+pub async fn restore_session() -> Result<Client, Box<dyn std::error::Error>> {
+    let session_file = File::open(SESSION_FILE)?;
+    let session_info: Session = serde_lexpr::from_reader(session_file)?;
+
+    let client =
+        Client::new_from_user_id_with_config(session_info.user_id.clone(), build_client_config())
+            .await?;
+    client.restore_login(session_info).await?;
+    Ok(client)
+}