1use db::{Db, UserId};
 2use rand::prelude::*;
 3use tide::log;
 4use time::{Duration, OffsetDateTime};
 5
 6#[allow(unused)]
 7#[path = "../db.rs"]
 8mod db;
 9#[path = "../env.rs"]
10mod env;
11
12#[async_std::main]
13async fn main() {
14    if let Err(error) = env::load_dotenv() {
15        log::error!(
16            "error loading .env.toml (this is expected in production): {}",
17            error
18        );
19    }
20
21    let mut rng = StdRng::from_entropy();
22    let database_url = std::env::var("DATABASE_URL").expect("missing DATABASE_URL env var");
23    let db = Db::new(&database_url, 5)
24        .await
25        .expect("failed to connect to postgres database");
26
27    let zed_users = ["nathansobo", "maxbrunsfeld", "as-cii", "iamnbutler"];
28    let mut zed_user_ids = Vec::<UserId>::new();
29    for zed_user in zed_users {
30        if let Some(user) = db
31            .get_user_by_github_login(zed_user)
32            .await
33            .expect("failed to fetch user")
34        {
35            zed_user_ids.push(user.id);
36        } else {
37            zed_user_ids.push(
38                db.create_user(zed_user, true)
39                    .await
40                    .expect("failed to insert user"),
41            );
42        }
43    }
44
45    let zed_org_id = if let Some(org) = db
46        .find_org_by_slug("zed")
47        .await
48        .expect("failed to fetch org")
49    {
50        org.id
51    } else {
52        db.create_org("Zed", "zed")
53            .await
54            .expect("failed to insert org")
55    };
56
57    let general_channel_id = if let Some(channel) = db
58        .get_org_channels(zed_org_id)
59        .await
60        .expect("failed to fetch channels")
61        .iter()
62        .find(|c| c.name == "General")
63    {
64        channel.id
65    } else {
66        let channel_id = db
67            .create_org_channel(zed_org_id, "General")
68            .await
69            .expect("failed to insert channel");
70
71        let now = OffsetDateTime::now_utc();
72        let max_seconds = Duration::days(100).as_seconds_f64();
73        let mut timestamps = (0..1000)
74            .map(|_| now - Duration::seconds_f64(rng.gen_range(0_f64..=max_seconds)))
75            .collect::<Vec<_>>();
76        timestamps.sort();
77        for timestamp in timestamps {
78            let sender_id = *zed_user_ids.choose(&mut rng).unwrap();
79            let body = lipsum::lipsum_words(rng.gen_range(1..=50));
80            db.create_channel_message(channel_id, sender_id, &body, timestamp, rng.gen())
81                .await
82                .expect("failed to insert message");
83        }
84        channel_id
85    };
86
87    for user_id in zed_user_ids {
88        db.add_org_member(zed_org_id, user_id, true)
89            .await
90            .expect("failed to insert org membership");
91        db.add_channel_member(general_channel_id, user_id, true)
92            .await
93            .expect("failed to insert channel membership");
94    }
95}