You've already forked authentication-service
mirror of
https://github.com/matrix-org/matrix-authentication-service.git
synced 2025-11-23 11:02:35 +03:00
70 lines
2.3 KiB
Rust
70 lines
2.3 KiB
Rust
// Copyright 2023 The Matrix.org Foundation C.I.C.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use clap::Parser;
|
|
use mas_config::RootConfig;
|
|
use mas_handlers::HttpClientFactory;
|
|
use mas_router::UrlBuilder;
|
|
use mas_tasks::HomeserverConnection;
|
|
use rand::{
|
|
distributions::{Alphanumeric, DistString},
|
|
thread_rng,
|
|
};
|
|
use tracing::{info, info_span};
|
|
|
|
use crate::util::{database_from_config, mailer_from_config, templates_from_config};
|
|
|
|
#[derive(Parser, Debug, Default)]
|
|
pub(super) struct Options {}
|
|
|
|
impl Options {
|
|
pub async fn run(&self, root: &super::Options) -> anyhow::Result<()> {
|
|
let span = info_span!("cli.worker.init").entered();
|
|
let config: RootConfig = root.load_config()?;
|
|
|
|
// Connect to the database
|
|
info!("Connecting to the database");
|
|
let pool = database_from_config(&config.database).await?;
|
|
|
|
let url_builder = UrlBuilder::new(config.http.public_base.clone());
|
|
|
|
// Load and compile the templates
|
|
let templates = templates_from_config(&config.templates, &url_builder).await?;
|
|
|
|
let mailer = mailer_from_config(&config.email, &templates).await?;
|
|
mailer.test_connection().await?;
|
|
|
|
let http_client_factory = HttpClientFactory::new(50);
|
|
let conn = HomeserverConnection::new(
|
|
config.matrix.homeserver.clone(),
|
|
config.matrix.endpoint.clone(),
|
|
config.matrix.secret.clone(),
|
|
);
|
|
|
|
drop(config);
|
|
|
|
#[allow(clippy::disallowed_methods)]
|
|
let mut rng = thread_rng();
|
|
let worker_name = Alphanumeric.sample_string(&mut rng, 10);
|
|
|
|
info!(worker_name, "Starting task scheduler");
|
|
let monitor = mas_tasks::init(&worker_name, &pool, &mailer, conn, &http_client_factory);
|
|
|
|
span.exit();
|
|
|
|
monitor.run().await?;
|
|
Ok(())
|
|
}
|
|
}
|