// Copyright (c) 2018  Teddy Wing
//
// This file is part of DomeKey Web.
//
// DomeKey Web is free software: you can redistribute it and/or modify it
// under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// DomeKey Web is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public
// License along with DomeKey Web. If not, see
// <https://www.gnu.org/licenses/>.

use chrono::Utc;
use mysql;
use rand::{self, Rng};
use sha1;

use errors::*;

pub struct Purchaser<'a> {
    pub name: &'a str,
    pub email: &'a str,
    pub secret: Option<String>,
}

impl<'a> Purchaser<'a> {
    pub fn new(name: &'a str, email: &'a str) -> Self {
        let mut purchaser = Purchaser {
            name: name,
            email: email,
            secret: None,
        };

        purchaser.generate_secret();

        purchaser
    }

    fn generate_secret(&mut self) {
        let timestamp = Utc::now().timestamp();

        let mut rng = rand::thread_rng();
        let random: usize = rng.gen_range(1_000_000_000, ::std::usize::MAX);

        let source = format!("{}{}{}{}", self.name, self.email, timestamp, random);
        let digest = sha1::Sha1::from(source).hexdigest();

        self.secret = Some(digest);
    }

    pub fn insert(&self, cx: &mut mysql::PooledConn) -> Result<()> {
        let mut tx = cx.start_transaction(
            false,  // consistent_snapshot
            None,   // isolation_level
            None,   // readonly
        )?;

        match self.secret {
            Some(ref s) => {
                tx.prep_exec("
                    INSERT INTO purchasers
                        (name, email, secret)
                        VALUES
                        (?, ?, ?)",
                    (self.name, self.email, s),
                )?;
            },
            None => {
                tx.prep_exec("
                    INSERT INTO purchasers
                        (name, email)
                        VALUES
                        (?, ?)",
                    (self.name, self.email),
                )?;
            },
        }

        tx.commit()?;

        Ok(())
    }
}