birdseed 0.2.0

birdseed enables you to seed a libellis database with fake data, clear all tables, or rebuild all tables per the current embedded migrations
Documentation 
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use diesel::pg::PgConnection;
use diesel::prelude::*;

use std::error::Error;

use rayon::prelude::*;

use rand::Rng;

use indicatif::ProgressBar;

use crate::pg_pool::Pool;

use crate::models::survey::{NewSurvey, Survey, UpdateSurveyData};

/// Populates surveys table with row_count random surveys making sure each survey is being created
/// by an existing user.
pub fn populate(
    pool: &Pool,
    authors: &Vec<String>,
    categories: &Vec<String>,
    row_count: u32,
    bar: &ProgressBar,
) -> Result<Vec<i32>, Box<dyn Error>> {
    bar.set_message(&format!("Seeding {} surveys", row_count));

    let survey_ids: Vec<i32> = authors
        .par_iter()
        .map(|auth| {
            let pool = pool.clone();
            let conn = pool.get().unwrap();
            let mut rng = rand::thread_rng();

            let survey_title = format!("{}", fake!(Lorem.sentence(4, 8)));

            let cat = &categories[rng.gen_range(0, row_count-1) as usize];

            let survey = create(&conn, &auth, &survey_title, cat);
            bar.inc(1);

            survey.id
        })
        .collect();

    Ok(survey_ids)
}

/// Gets a single survey from the database by the given survey id.
pub fn get(conn: &PgConnection, survey_id: i32) -> Result<Survey, diesel::result::Error> {
    use crate::schema::surveys::dsl::*;

    surveys.find(survey_id).first(conn)
}

/// Gets all surveys filtered by a search parameter
pub fn get_all(
    conn: &PgConnection,
    search: Option<String>,
) -> Result<Vec<Survey>, diesel::result::Error> {
    use crate::schema::surveys::dsl::*;

    match search {
        Some(s) => {
            let fuzzy_s = format!("%{}%", s);
            surveys
                .filter(
                    author
                        .ilike(&fuzzy_s)
                        .or(title.ilike(&fuzzy_s).or(description.ilike(&fuzzy_s))),
                )
                .filter(published.eq(true))
                .get_results(conn)
        }
        None => surveys.filter(published.eq(true)).get_results(conn),
    }
}

/// Creates a single survey in the database for the given author (user).
pub fn create<'a>(
    conn: &PgConnection,
    auth: &'a str,
    survey_title: &'a str,
    cat: &'a str,
) -> Survey {
    use crate::schema::surveys;

    let new_survey = NewSurvey {
        author: auth,
        title: survey_title,
        published: true,
        category: cat,
    };

    diesel::insert_into(surveys::table)
        .values(&new_survey)
        .get_result(conn)
        .expect("Error saving new survey")
}

/// Updates a survey based on optional fields it receives from a patch route.
pub fn update(conn: &PgConnection, survey_id: i32, data: UpdateSurveyData) -> Survey {
    use crate::schema::surveys::dsl::*;

    diesel::update(surveys.filter(id.eq(survey_id)))
        .set(&data)
        .get_result(conn)
        .expect("Error updating survey.")
}

/// Deletes a survey based on a survey id.
pub fn delete(conn: &PgConnection, survey_id: i32) -> Result<(), Box<dyn Error>> {
    use crate::schema::surveys::dsl::*;

    diesel::delete(surveys.filter(id.eq(survey_id))).execute(conn)?;

    Ok(())
}