aboutsummaryrefslogtreecommitdiffstats
path: root/license-generator/src/lib.rs
blob: cc39cdb0c51ef32ebb4eee1c403db94086f3310c (plain)
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#[macro_use]
extern crate error_chain;
extern crate mysql;
extern crate rand;
extern crate sha1;

mod errors {
    use mysql;

    error_chain! {
        foreign_links {
            MySql(mysql::error::Error);
        }
    }
}

use errors::*;

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

impl<'a> Purchaser<'a> {
    fn new(name: &'a str, email: &'a str) -> Self {
        Purchaser {
            name: name,
            email: email,
            secret: None,
        }
    }

    fn with_secret(mut self, secret: String) -> Self {
        self.secret = Some(secret);
        self
    }

    fn generate_secret(&mut self) {
        let random: usize = rand::random();

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

        self.secret = Some(digest);
    }

    fn insert(&self, cx: &mut mysql::Conn) -> 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),
                )?;
            },
        }

        Ok(())
    }
}