Function main

backend/src/bin/api.rs:100–159

()

#[tokio::main]
async fn main() -> async_graphql::Result<()> {
    let config = Config::load()?;
    env_logger::init();

    log::info!("setting up database connection");
    let pool = db::db_connection(&config).await?;

    log::info!("reading data from {}", config.digraph_data_directory);
    let root = git::DataRoot::new(PathBuf::from(&config.digraph_data_directory));

    log::info!("running migrations");
    sqlx::migrate!("./db/migrations").run(&pool).await?;

    log::info!("loading graphql schema");
    let schema = Schema::build(QueryRoot, MutationRoot, EmptySubscription)
        .extension(extensions::Logger)
        .finish();

    log::info!("loading graphql schema");
    let redis = Arc::new(redis::Redis::new(config.digraph_redis_url.to_owned())?);

    log::info!("setting up app state");
    let state = digraph::graphql::State::new(
        pool,
        root,
        schema.clone(),
        config.digraph_server_secret,
        redis,
    );

    let socket = env::var("LISTEN_ADDR").unwrap_or_else(|_| "0.0.0.0:8080".to_owned());
    let listener = tokio::net::TcpListener::bind(socket).await.unwrap();

    let cors = CorsLayer::new()
        .allow_origin(AllowOrigin::predicate(
            |origin: &HeaderValue, _request_parts: &request::Parts| {
                let bytes = origin.as_bytes();
                bytes == b"http://localhost:3002"
                    || bytes == b"https://digraph.app"
                    || bytes.ends_with(b".digraph.app")
            },
        ))
        .allow_headers(vec![header::CONTENT_TYPE]);

    // For metrics, see https://github.com/oliverjumpertz/axum-graphql/blob/main/src/main.rs
    log::info!("starting server");
    let app = Router::new()
        .route("/health", get(health))
        .route("/graphql", post(graphql_handler))
        .layer(Extension(schema))
        .layer(cors)
        .with_state(state);

    axum::serve(listener, app)
        .with_graceful_shutdown(shutdown_signal())
        .await
        .unwrap();