mirror of
https://github.com/zed-industries/zed.git
synced 2024-09-18 18:08:07 +03:00
Add bootstrap script, avoid hard-coding zed team members
This commit is contained in:
parent
dfeb5e010e
commit
776095caf0
@ -18,9 +18,3 @@ CREATE TABLE IF NOT EXISTS "signups" (
|
|||||||
"email_address" VARCHAR,
|
"email_address" VARCHAR,
|
||||||
"about" TEXT
|
"about" TEXT
|
||||||
);
|
);
|
||||||
|
|
||||||
INSERT INTO users (github_login, admin)
|
|
||||||
VALUES
|
|
||||||
('nathansobo', true),
|
|
||||||
('maxbrunsfeld', true),
|
|
||||||
('as-cii', true);
|
|
||||||
|
@ -1,8 +1,7 @@
|
|||||||
use clap::Parser;
|
|
||||||
use collab::{Error, Result};
|
use collab::{Error, Result};
|
||||||
use db::{Db, PostgresDb, UserId};
|
use db::{Db, PostgresDb, UserId};
|
||||||
use rand::prelude::*;
|
use rand::prelude::*;
|
||||||
use serde::Deserialize;
|
use serde::{de::DeserializeOwned, Deserialize};
|
||||||
use std::fmt::Write;
|
use std::fmt::Write;
|
||||||
use time::{Duration, OffsetDateTime};
|
use time::{Duration, OffsetDateTime};
|
||||||
|
|
||||||
@ -10,62 +9,52 @@ use time::{Duration, OffsetDateTime};
|
|||||||
#[path = "../db.rs"]
|
#[path = "../db.rs"]
|
||||||
mod db;
|
mod db;
|
||||||
|
|
||||||
#[derive(Parser)]
|
|
||||||
struct Args {
|
|
||||||
/// Seed users from GitHub.
|
|
||||||
#[clap(short, long)]
|
|
||||||
github_users: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
struct GitHubUser {
|
struct GitHubUser {
|
||||||
id: usize,
|
id: usize,
|
||||||
login: String,
|
login: String,
|
||||||
|
email: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
let args = Args::parse();
|
|
||||||
let mut rng = StdRng::from_entropy();
|
let mut rng = StdRng::from_entropy();
|
||||||
let database_url = std::env::var("DATABASE_URL").expect("missing DATABASE_URL env var");
|
let database_url = std::env::var("DATABASE_URL").expect("missing DATABASE_URL env var");
|
||||||
let db = PostgresDb::new(&database_url, 5)
|
let db = PostgresDb::new(&database_url, 5)
|
||||||
.await
|
.await
|
||||||
.expect("failed to connect to postgres database");
|
.expect("failed to connect to postgres database");
|
||||||
|
let github_token = std::env::var("GITHUB_TOKEN").expect("missing GITHUB_TOKEN env var");
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
|
||||||
let mut zed_users = vec![
|
let current_user =
|
||||||
("nathansobo".to_string(), Some("nathan@zed.dev")),
|
fetch_github::<GitHubUser>(&client, &github_token, "https://api.github.com/user").await;
|
||||||
("maxbrunsfeld".to_string(), Some("max@zed.dev")),
|
let staff_users = fetch_github::<Vec<GitHubUser>>(
|
||||||
("as-cii".to_string(), Some("antonio@zed.dev")),
|
&client,
|
||||||
("iamnbutler".to_string(), Some("nate@zed.dev")),
|
&github_token,
|
||||||
("gibusu".to_string(), Some("greg@zed.dev")),
|
"https://api.github.com/orgs/zed-industries/teams/staff/members",
|
||||||
("Kethku".to_string(), Some("keith@zed.dev")),
|
)
|
||||||
];
|
.await;
|
||||||
|
|
||||||
if args.github_users {
|
let mut zed_users = Vec::new();
|
||||||
let github_token = std::env::var("GITHUB_TOKEN").expect("missing GITHUB_TOKEN env var");
|
zed_users.push((current_user, true));
|
||||||
let client = reqwest::Client::new();
|
zed_users.extend(staff_users.into_iter().map(|user| (user, true)));
|
||||||
|
|
||||||
|
let user_count = db
|
||||||
|
.get_all_users(0, 200)
|
||||||
|
.await
|
||||||
|
.expect("failed to load users from db")
|
||||||
|
.len();
|
||||||
|
if user_count < 100 {
|
||||||
let mut last_user_id = None;
|
let mut last_user_id = None;
|
||||||
for page in 0..20 {
|
for _ in 0..10 {
|
||||||
println!("Downloading users from GitHub, page {}", page);
|
|
||||||
let mut uri = "https://api.github.com/users?per_page=100".to_string();
|
let mut uri = "https://api.github.com/users?per_page=100".to_string();
|
||||||
if let Some(last_user_id) = last_user_id {
|
if let Some(last_user_id) = last_user_id {
|
||||||
write!(&mut uri, "&since={}", last_user_id).unwrap();
|
write!(&mut uri, "&since={}", last_user_id).unwrap();
|
||||||
}
|
}
|
||||||
let response = client
|
let users = fetch_github::<Vec<GitHubUser>>(&client, &github_token, &uri).await;
|
||||||
.get(uri)
|
|
||||||
.bearer_auth(&github_token)
|
|
||||||
.header("user-agent", "zed")
|
|
||||||
.send()
|
|
||||||
.await
|
|
||||||
.expect("failed to fetch github users");
|
|
||||||
let users = response
|
|
||||||
.json::<Vec<GitHubUser>>()
|
|
||||||
.await
|
|
||||||
.expect("failed to deserialize github user");
|
|
||||||
zed_users.extend(users.iter().map(|user| (user.login.clone(), None)));
|
|
||||||
|
|
||||||
if let Some(last_user) = users.last() {
|
if let Some(last_user) = users.last() {
|
||||||
last_user_id = Some(last_user.id);
|
last_user_id = Some(last_user.id);
|
||||||
|
zed_users.extend(users.into_iter().map(|user| (user, false)));
|
||||||
} else {
|
} else {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
@ -73,16 +62,16 @@ async fn main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
let mut zed_user_ids = Vec::<UserId>::new();
|
let mut zed_user_ids = Vec::<UserId>::new();
|
||||||
for (zed_user, email) in zed_users {
|
for (github_user, admin) in zed_users {
|
||||||
if let Some(user) = db
|
if let Some(user) = db
|
||||||
.get_user_by_github_login(&zed_user)
|
.get_user_by_github_login(&github_user.login)
|
||||||
.await
|
.await
|
||||||
.expect("failed to fetch user")
|
.expect("failed to fetch user")
|
||||||
{
|
{
|
||||||
zed_user_ids.push(user.id);
|
zed_user_ids.push(user.id);
|
||||||
} else {
|
} else {
|
||||||
zed_user_ids.push(
|
zed_user_ids.push(
|
||||||
db.create_user(&zed_user, email, true)
|
db.create_user(&github_user.login, github_user.email.as_deref(), admin)
|
||||||
.await
|
.await
|
||||||
.expect("failed to insert user"),
|
.expect("failed to insert user"),
|
||||||
);
|
);
|
||||||
@ -140,3 +129,21 @@ async fn main() {
|
|||||||
.expect("failed to insert channel membership");
|
.expect("failed to insert channel membership");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn fetch_github<T: DeserializeOwned>(
|
||||||
|
client: &reqwest::Client,
|
||||||
|
access_token: &str,
|
||||||
|
url: &str,
|
||||||
|
) -> T {
|
||||||
|
let response = client
|
||||||
|
.get(url)
|
||||||
|
.bearer_auth(&access_token)
|
||||||
|
.header("user-agent", "zed")
|
||||||
|
.send()
|
||||||
|
.await
|
||||||
|
.expect(&format!("failed to fetch '{}'", url));
|
||||||
|
response
|
||||||
|
.json()
|
||||||
|
.await
|
||||||
|
.expect(&format!("failed to deserialize github user from '{}'", url))
|
||||||
|
}
|
||||||
|
13
script/bootstrap
Executable file
13
script/bootstrap
Executable file
@ -0,0 +1,13 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
echo "installing foreman..."
|
||||||
|
which foreman > /dev/null || brew install foreman
|
||||||
|
|
||||||
|
echo "creating database..."
|
||||||
|
script/sqlx database create
|
||||||
|
|
||||||
|
echo "migrating database..."
|
||||||
|
script/sqlx migrate run
|
||||||
|
|
||||||
|
echo "seeding database..."
|
||||||
|
script/seed-db
|
@ -4,6 +4,6 @@ set -e
|
|||||||
cd crates/collab
|
cd crates/collab
|
||||||
|
|
||||||
# Export contents of .env.toml
|
# Export contents of .env.toml
|
||||||
eval "$(cargo run --bin dotenv)"
|
eval "$(cargo run --quiet --bin dotenv)"
|
||||||
|
|
||||||
cargo run --package=collab --features seed-support --bin seed -- $@
|
cargo run --quiet --package=collab --features seed-support --bin seed -- $@
|
||||||
|
@ -8,7 +8,7 @@ set -e
|
|||||||
cd crates/collab
|
cd crates/collab
|
||||||
|
|
||||||
# Export contents of .env.toml
|
# Export contents of .env.toml
|
||||||
eval "$(cargo run --bin dotenv)"
|
eval "$(cargo run --quiet --bin dotenv)"
|
||||||
|
|
||||||
# Run sqlx command
|
# Run sqlx command
|
||||||
sqlx $@
|
sqlx $@
|
||||||
|
Loading…
Reference in New Issue
Block a user