61 lines
1.9 KiB
Rust
61 lines
1.9 KiB
Rust
|
|
||
|
|
use boitalettres::proto::{Request, Response};
|
||
|
|
use crate::login::ArcLoginProvider;
|
||
|
|
use crate::imap::Context;
|
||
|
|
|
||
|
|
//--- dispatching
|
||
|
|
|
||
|
|
pub async fn dispatch(ctx: Context) -> Result<Response> {
|
||
|
|
match ctx.req.body {
|
||
|
|
CommandBody::Capability => anonymous::capability(ctx).await,
|
||
|
|
CommandBody::Login { username, password } => anonymous::login(ctx, username, password).await,
|
||
|
|
_ => Status::no(Some(ctx.req.tag.clone()), None, "This command is not available in the ANONYMOUS state.")
|
||
|
|
.map(|s| vec![ImapRes::Status(s)])
|
||
|
|
.map_err(Error::msg),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//--- Command controllers
|
||
|
|
|
||
|
|
pub async fn capability(ctx: Context) -> Result<Response> {
|
||
|
|
let capabilities = vec![Capability::Imap4Rev1, Capability::Idle];
|
||
|
|
let res = vec![
|
||
|
|
ImapRes::Data(Data::Capability(capabilities)),
|
||
|
|
ImapRes::Status(
|
||
|
|
Status::ok(Some(ctx.req.tag.clone()), None, "Server capabilities")
|
||
|
|
.map_err(Error::msg)?,
|
||
|
|
),
|
||
|
|
];
|
||
|
|
Ok(res)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn login(ctx: Context, username: AString, password: AString) -> Result<Response> {
|
||
|
|
let (u, p) = (String::try_from(username)?, String::try_from(password)?);
|
||
|
|
tracing::info!(user = %u, "command.login");
|
||
|
|
|
||
|
|
let creds = match ctx.login_provider.login(&u, &p).await {
|
||
|
|
Err(e) => {
|
||
|
|
tracing::debug!(error=%e, "authentication failed");
|
||
|
|
return Ok(vec![ImapRes::Status(
|
||
|
|
Status::no(Some(ctx.req.tag.clone()), None, "Authentication failed")
|
||
|
|
.map_err(Error::msg)?,
|
||
|
|
)]);
|
||
|
|
}
|
||
|
|
Ok(c) => c,
|
||
|
|
};
|
||
|
|
|
||
|
|
let user = flow::User {
|
||
|
|
creds,
|
||
|
|
name: u.clone(),
|
||
|
|
};
|
||
|
|
ctx.state.authenticate(user)?;
|
||
|
|
|
||
|
|
tracing::info!(username=%u, "connected");
|
||
|
|
Ok(vec![
|
||
|
|
//@FIXME we could send a capability status here too
|
||
|
|
ImapRes::Status(
|
||
|
|
Status::ok(Some(ctx.req.tag.clone()), None, "completed").map_err(Error::msg)?,
|
||
|
|
),
|
||
|
|
])
|
||
|
|
}
|