extern crate imap;
extern crate openssl;
extern crate base64;

use openssl::ssl::{SslConnectorBuilder, SslMethod};
use base64::encode;
use imap::client::Client;
use imap::authenticator::Authenticator;

struct GmailOAuth2 {
    user: String,
    access_token: String,
}

impl Authenticator for GmailOAuth2 {
    #[allow(unused_variables)]
    fn process(&self, data: String) -> String {
        encode(
            format!(
                "user={}\x01auth=Bearer {}\x01\x01",
                self.user,
                self.access_token
            ).as_bytes(),
        )
    }
}

fn main() {
    let gmail_auth = GmailOAuth2 {
        user: String::from("sombody@gmail.com"),
        access_token: String::from("<access_token>"),
    };
    let domain = "imap.gmail.com";
    let port = 993;
    let socket_addr = (domain, port);
    let ssl_connector = SslConnectorBuilder::new(SslMethod::tls()).unwrap().build();
    let mut imap_socket = Client::secure_connect(socket_addr, domain, ssl_connector).unwrap();

    imap_socket.authenticate("XOAUTH2", gmail_auth).unwrap();

    match imap_socket.select("INBOX") {
        Ok(mailbox) => println!("{}", mailbox),
        Err(e) => println!("Error selecting INBOX: {}", e),
    };

    match imap_socket.fetch("2", "body[text]") {
        Ok(lines) => {
            for line in lines.iter() {
                print!("{}", line);
            }
        }
        Err(e) => println!("Error Fetching email 2: {}", e),
    };

    imap_socket.logout().unwrap();
}