♻️ refactor(cli): streamline message retrieval and parameter setting
- extract parameter setting logic into a separate function `set_parameters` - consolidate label adding, query setting, and max results setting into `set_parameters` - rename `run` to `get_messages` for clarity - add methods for retrieving labels, query, and max_results
This commit is contained in:
committed by
Jeremiah Russell
parent
fe4a85f5e3
commit
e488cd25cf
@@ -1,6 +1,8 @@
|
|||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use cull_gmail::{GmailClient, MessageList, Result};
|
use cull_gmail::{GmailClient, MessageList, Result};
|
||||||
|
|
||||||
|
use crate::message_trait::Message;
|
||||||
|
|
||||||
/// Command line options for the list subcommand
|
/// Command line options for the list subcommand
|
||||||
#[derive(Debug, Parser)]
|
#[derive(Debug, Parser)]
|
||||||
pub struct MessageCli {
|
pub struct MessageCli {
|
||||||
@@ -20,18 +22,20 @@ pub struct MessageCli {
|
|||||||
|
|
||||||
impl MessageCli {
|
impl MessageCli {
|
||||||
pub(crate) async fn run(&self, client: &mut GmailClient) -> Result<()> {
|
pub(crate) async fn run(&self, client: &mut GmailClient) -> Result<()> {
|
||||||
if !self.labels.is_empty() {
|
self.set_parameters(client)?;
|
||||||
client.add_labels(&self.labels).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(query) = self.query.as_ref() {
|
client.get_messages(self.pages).await
|
||||||
client.set_query(query)
|
}
|
||||||
}
|
|
||||||
|
|
||||||
log::trace!("Max results: `{}`", self.max_results);
|
pub(crate) fn labels(&self) -> &Vec<String> {
|
||||||
client.set_max_results(self.max_results);
|
&self.labels
|
||||||
log::debug!("List max results set to {}", client.max_results());
|
}
|
||||||
|
|
||||||
client.run(self.pages).await
|
pub(crate) fn query(&self) -> &Option<String> {
|
||||||
|
&self.query
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn max_results(&self) -> u32 {
|
||||||
|
self.max_results
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user