Add seed target for inserting seed data

Max Brunsfeld and Nathan Sobo created 4 years ago

Co-Authored-By: Nathan Sobo <[e-mail]>

Change summary

server/migrations/20210805175147_create_chat_tables.sql |  28 --
server/src/bin/seed.rs                                  | 107 +++++++++++
2 files changed, 115 insertions(+), 20 deletions(-)

Detailed changes

server/migrations/20210805175147_create_chat_tables.sql

@@ -4,6 +4,8 @@ CREATE TABLE IF NOT EXISTS "orgs" (
     "slug" VARCHAR NOT NULL
 );
 
+CREATE UNIQUE INDEX "index_orgs_slug" ON "orgs" ("slug");
+
 CREATE TABLE IF NOT EXISTS "org_memberships" (
     "id" SERIAL PRIMARY KEY,
     "org_id" INTEGER REFERENCES orgs (id) NOT NULL,
@@ -11,8 +13,8 @@ CREATE TABLE IF NOT EXISTS "org_memberships" (
     "admin" BOOLEAN NOT NULL
 );
 
-CREATE UNIQUE INDEX "index_org_memberships_user_id" ON "org_memberships" ("user_id");
-CREATE UNIQUE INDEX "index_org_memberships_org_id" ON "org_memberships" ("org_id");
+CREATE INDEX "index_org_memberships_user_id" ON "org_memberships" ("user_id");
+CREATE UNIQUE INDEX "index_org_memberships_org_id_and_user_id" ON "org_memberships" ("org_id", "user_id");
 
 CREATE TABLE IF NOT EXISTS "channels" (
     "id" SERIAL PRIMARY KEY,
@@ -21,7 +23,7 @@ CREATE TABLE IF NOT EXISTS "channels" (
     "name" VARCHAR NOT NULL
 );
 
-CREATE UNIQUE INDEX "index_channels_owner" ON "channels" ("owner_is_user", "owner_id");
+CREATE UNIQUE INDEX "index_channels_owner_and_name" ON "channels" ("owner_is_user", "owner_id", "name");
 
 CREATE TABLE IF NOT EXISTS "channel_memberships" (
     "id" SERIAL PRIMARY KEY,
@@ -30,8 +32,8 @@ CREATE TABLE IF NOT EXISTS "channel_memberships" (
     "admin" BOOLEAN NOT NULL
 );
 
-CREATE UNIQUE INDEX "index_channel_memberships_user_id" ON "channel_memberships" ("user_id");
-CREATE UNIQUE INDEX "index_channel_memberships_channel_id" ON "channel_memberships" ("channel_id");
+CREATE INDEX "index_channel_memberships_user_id" ON "channel_memberships" ("user_id");
+CREATE UNIQUE INDEX "index_channel_memberships_channel_id_and_user_id" ON "channel_memberships" ("channel_id", "user_id");
 
 CREATE TABLE IF NOT EXISTS "channel_messages" (
     "id" SERIAL PRIMARY KEY,
@@ -41,18 +43,4 @@ CREATE TABLE IF NOT EXISTS "channel_messages" (
     "sent_at" TIMESTAMP
 );
 
-CREATE UNIQUE INDEX "index_channel_messages_channel_id" ON "channel_messages" ("channel_id");
-
-INSERT INTO users (github_login, admin) VALUES ('iamnbutler', true);
-
-DO $$ 
-DECLARE
-    zed_org_id INTEGER;
-    max_id INTEGER;
-    nathan_id INTEGER;
-    antonio_id INTEGER;
-    nate_id INTEGER;
-BEGIN 
-    INSERT INTO "orgs" (name, slug) VALUES ('Zed', 'zed') RETURNING id into zed_org_id;
-END $$;
-
+CREATE INDEX "index_channel_messages_channel_id" ON "channel_messages" ("channel_id");

server/src/bin/seed.rs

@@ -0,0 +1,107 @@
+use sqlx::postgres::PgPoolOptions;
+use tide::log;
+
+#[path = "../env.rs"]
+mod env;
+
+#[async_std::main]
+async fn main() {
+    if let Err(error) = env::load_dotenv() {
+        log::error!(
+            "error loading .env.toml (this is expected in production): {}",
+            error
+        );
+    }
+
+    let database_url = std::env::var("DATABASE_URL").expect("missing DATABASE_URL env var");
+    let db = PgPoolOptions::new()
+        .max_connections(5)
+        .connect(&database_url)
+        .await
+        .expect("failed to connect to postgres database");
+
+    let zed_users = ["nathansobo", "maxbrunsfeld", "as-cii", "iamnbutler"];
+    let mut zed_user_ids = Vec::<i32>::new();
+    for zed_user in zed_users {
+        zed_user_ids.push(
+            sqlx::query_scalar(
+                r#"
+                INSERT INTO users
+                    (github_login, admin)
+                VALUES
+                    ($1, true)
+                ON CONFLICT (github_login) DO UPDATE SET
+                    github_login=EXCLUDED.github_login
+                RETURNING id
+                "#,
+            )
+            .bind(zed_user)
+            .fetch_one(&db)
+            .await
+            .expect("failed to insert user"),
+        )
+    }
+
+    let zed_org_id: i32 = sqlx::query_scalar(
+        r#"
+        INSERT INTO orgs
+            (name, slug)
+        VALUES
+            ('Zed', 'zed')
+        ON CONFLICT (slug) DO UPDATE SET
+            slug=EXCLUDED.slug
+        RETURNING id
+        "#,
+    )
+    .fetch_one(&db)
+    .await
+    .expect("failed to insert org");
+
+    let general_channel_id: i32 = sqlx::query_scalar(
+        r#"
+        INSERT INTO channels
+            (owner_is_user, owner_id, name)
+        VALUES
+            (false, $1, 'General')
+        ON CONFLICT (owner_is_user, owner_id, name) DO UPDATE SET
+            name=EXCLUDED.name
+        RETURNING id
+        "#,
+    )
+    .bind(zed_org_id)
+    .fetch_one(&db)
+    .await
+    .expect("failed to insert channel");
+
+    for user_id in zed_user_ids {
+        sqlx::query(
+            r#"
+            INSERT INTO org_memberships
+                (org_id, user_id, admin)
+            VALUES
+                ($1, $2, true)
+            ON CONFLICT DO NOTHING
+            "#,
+        )
+        .bind(zed_org_id)
+        .bind(user_id)
+        .execute(&db)
+        .await
+        .expect("failed to insert org membership");
+
+        sqlx::query(
+            r#"
+            INSERT INTO channel_memberships
+                (channel_id, user_id, admin)
+            VALUES
+                ($1, $2, true)
+            ON CONFLICT DO NOTHING
+            "#,
+        )
+        .bind(general_channel_id)
+        .bind(user_id)
+        .execute(&db)
+        .await
+        .expect("failed to insert channel membership");
+    }
+}