use mysql::prelude::Queryable;
use mysql::*;

pub struct QueryResult {}

mod query_constructor;
mod query_join;

pub struct Fudge {
    pool: Pool,
}

impl Fudge {
    /// Initialise a new `Fudge`
    /// ```
    /// use fudge::Fudge;
    /// let fudge_db = Fudge::new("user", "pass", "database_1", "localhost", 3306 /*port*/);
    /// ```
    pub fn new(username: &str, password: &str, database: &str, host: &str, port: usize) -> Fudge {
        let url = format!(
            "mysql://{}:{}@{}:{}/{}",
            username, password, host, port, database
        );
        println!("{}", url);
        let pool = Pool::new(url).unwrap();
        return Fudge { pool };
    }

    /// send a raw SQL query to the Database
    /// ```
    /// # use fudge::Fudge;
    /// # let fudge_db = Fudge::new("user", "pass", "database_1", "localhost", 3306 /*port*/);
    /// let query_result = fudge_db.query_raw("SELECT column_name FROM table_name")?;
    /// ```
    pub fn query_raw(&self, query: &str) -> Result<Statement> {
        let mut connection: Conn = self.pool.get_conn()?.unwrap();
        let mut transaction: Transaction = connection.start_transaction(TxOpts::default())?;
        let result = transaction.prep(query);
        return if let Err(_err) = result {
            transaction.rollback()?;
            println!(
                "\x1b[31mError with Query:\nConnection ID: {}\x1b[0m\n{:?}",
                connection.connection_id(),
                _err
            );
            Err(_err)
        } else {
            let qr = result.unwrap();
            transaction.commit()?;
            println!("{:#?}", qr);
            Ok(qr)
        };
    }
}

#[cfg(test)]
mod test;