1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use crate::frame::frame_errors::ParseError;
use bytes::BufMut;
use std::convert::TryInto;

use crate::frame::request::{Request, RequestOpcode};
use crate::frame::Authenticator;

// Implements Authenticate Response
pub struct AuthResponse {
    pub username: Option<String>,
    pub password: Option<String>,
    pub authenticator: Authenticator,
}

impl Request for AuthResponse {
    const OPCODE: RequestOpcode = RequestOpcode::AuthResponse;

    fn serialize(&self, buf: &mut impl BufMut) -> Result<(), ParseError> {
        if self.username.is_none() || self.password.is_none() {
            return Err(ParseError::BadDataToSerialize(
                "Bad credentials: username or password missing. You can use SessionBuilder::user(\"user\", \"pass\") to provide credentials.".to_string(),
            ));
        }
        let username_as_bytes = self.username.as_ref().unwrap().as_bytes();
        let password_as_bytes = self.password.as_ref().unwrap().as_bytes();

        // The body of AuthResponse is a single [bytes] value (i32 length and then contents)
        let buf_size = 2 + username_as_bytes.len() + password_as_bytes.len();
        buf.put_i32(buf_size.try_into()?);

        buf.put_u8(0);
        buf.put_slice(username_as_bytes);
        buf.put_u8(0);
        buf.put_slice(password_as_bytes);
        Ok(())
    }
}