Reorganise CLI crate

crates/cli/src/commands/manage.rs

@@ -0,0 +1,78 @@
+// Copyright 2021, 2022 The Matrix.org Foundation C.I.C.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+use argon2::Argon2;
+use clap::Parser;
+use mas_config::DatabaseConfig;
+use mas_storage::user::{
+    lookup_user_by_username, lookup_user_email, mark_user_email_as_verified, register_user,
+};
+use tracing::{info, warn};
+
+#[derive(Parser, Debug)]
+pub(super) struct Options {
+    #[clap(subcommand)]
+    subcommand: Subcommand,
+}
+
+#[derive(Parser, Debug)]
+enum Subcommand {
+    /// Register a new user
+    Register { username: String, password: String },
+
+    /// List active users
+    Users,
+
+    /// Mark email address as verified
+    VerifyEmail { username: String, email: String },
+}
+
+impl Options {
+    pub async fn run(&self, root: &super::Options) -> anyhow::Result<()> {
+        use Subcommand as SC;
+        match &self.subcommand {
+            SC::Register { username, password } => {
+                let config: DatabaseConfig = root.load_config()?;
+                let pool = config.connect().await?;
+                let mut txn = pool.begin().await?;
+                let hasher = Argon2::default();
+
+                let user = register_user(&mut txn, hasher, username, password).await?;
+                txn.commit().await?;
+                info!(?user, "User registered");
+
+                Ok(())
+            }
+            SC::Users => {
+                warn!("Not implemented yet");
+
+                Ok(())
+            }
+            SC::VerifyEmail { username, email } => {
+                let config: DatabaseConfig = root.load_config()?;
+                let pool = config.connect().await?;
+                let mut txn = pool.begin().await?;
+
+                let user = lookup_user_by_username(&mut txn, username).await?;
+                let email = lookup_user_email(&mut txn, &user, email).await?;
+                let email = mark_user_email_as_verified(&mut txn, email).await?;
+
+                txn.commit().await?;
+                info!(?email, "Email marked as verified");
+
+                Ok(())
+            }
+        }
+    }
+}