Merge pull request #25 from amfoss/develop

release v1.2.0

Author: ivinjabraham

.gitignore

@@ -10,3 +10,4 @@ target/
 *.pdb
 
 .env
+amd.log

Cargo.lock

@@ -16,3 +16,4 @@ tracing = "0.1.37"
 dotenv = "0.15.0"
 serenity = { version = "0.12.4", features = ["chrono"] }
 poise = "0.6.1"
+tracing-subscriber = { version = "0.3.19", features = ["env-filter"] }

Cargo.toml

@@ -15,15 +15,49 @@ GNU General Public License for more details.
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
+use tracing::{info, trace};
+use tracing_subscriber::EnvFilter;
+
 use crate::{Context, Data, Error};
 
 #[poise::command(prefix_command)]
 async fn amdctl(ctx: Context<'_>) -> Result<(), Error> {
+    trace!("Running amdctl command");
     ctx.say("amD is up and running.").await?;
     Ok(())
 }
 
+#[poise::command(prefix_command, owners_only)]
+async fn set_log_level(ctx: Context<'_>, level: String) -> Result<(), Error> {
+    trace!("Running set_log_level command");
+    let data = ctx.data();
+    let reload_handle = data.log_reload_handle.write().await;
+
+    let new_filter = match level.to_lowercase().as_str() {
+        "trace" => "trace",
+        "debug" => "debug",
+        "info" => "info",
+        "warn" => "warn",
+        "error" => "error",
+        _ => {
+            ctx.say("Invalid log level! Use: trace, debug, info, warn, error")
+                .await?;
+            return Ok(());
+        }
+    };
+
+    if reload_handle.reload(EnvFilter::new(new_filter)).is_ok() {
+        ctx.say(format!("Log level changed to **{}**", new_filter))
+            .await?;
+        info!("Log level changed to {}", new_filter);
+    } else {
+        ctx.say("Failed to update log level.").await?;
+    }
+
+    Ok(())
+}
+
 /// Returns a vector containg [Poise Commands][`poise::Command`]
 pub fn get_commands() -> Vec<poise::Command<Data, Error>> {
-    vec![amdctl()]
+    vec![amdctl(), set_log_level()]
 }

src/commands.rs

@@ -15,8 +15,10 @@ GNU General Public License for more details.
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
-use crate::graphql::models::{Member, Streak};
 use anyhow::{anyhow, Context};
+use tracing::debug;
+
+use crate::graphql::models::{Member, Streak};
 
 pub async fn fetch_members() -> anyhow::Result<Vec<Member>> {
     let request_url = std::env::var("ROOT_URL").context("ROOT_URL not found in ENV")?;
@@ -35,6 +37,7 @@ pub async fn fetch_members() -> anyhow::Result<Vec<Member>> {
         }
     }"#;
 
+    debug!("Sending query {}", query);
     let response = client
         .post(request_url)
         .json(&serde_json::json!({"query": query}))
@@ -54,6 +57,7 @@ pub async fn fetch_members() -> anyhow::Result<Vec<Member>> {
         .await
         .context("Failed to serialize response")?;
 
+    debug!("Response: {}", response_json);
     let members = response_json
         .get("data")
         .and_then(|data| data.get("members"))
@@ -85,6 +89,7 @@ pub async fn increment_streak(member: &mut Member) -> anyhow::Result<()> {
         member.member_id
     );
 
+    debug!("Sending mutation {}", mutation);
     let response = client
         .post(request_url)
         .json(&serde_json::json!({"query": mutation}))
@@ -98,6 +103,7 @@ pub async fn increment_streak(member: &mut Member) -> anyhow::Result<()> {
             response.status()
         ));
     }
+    debug!("Response: {:?}", response.text().await);
 
     // Handle the streak vector
     if member.streak.is_empty() {
@@ -134,6 +140,7 @@ pub async fn reset_streak(member: &mut Member) -> anyhow::Result<()> {
         member.member_id
     );
 
+    debug!("Sending mutation {}", mutation);
     let response = client
         .post(&request_url)
         .json(&serde_json::json!({ "query": mutation }))
@@ -152,6 +159,7 @@ pub async fn reset_streak(member: &mut Member) -> anyhow::Result<()> {
         .json()
         .await
         .context("Failed to parse response JSON")?;
+    debug!("Response: {}", response_json);
 
     if let Some(data) = response_json
         .get("data")