Compare commits
15 Commits
v0.2.1
...
602d562c66
Author | SHA1 | Date | |
---|---|---|---|
602d562c66 | |||
c386d85d90 | |||
214e1de972 | |||
001a0214d8 | |||
720ecac8f8 | |||
02837b00fe | |||
ffd977b6d5 | |||
208df18fe4 | |||
4b5d9e2cde | |||
0f8f580050 | |||
0df350dae6 | |||
52a89428f3 | |||
903577aba7 | |||
a7629a650a | |||
629e13710d |
3386
Cargo.lock
generated
3386
Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
23
Cargo.toml
23
Cargo.toml
@@ -1,13 +1,13 @@
|
||||
[package]
|
||||
name = "bebot"
|
||||
version = "0.2.1"
|
||||
version = "0.2.4"
|
||||
description = "Gitlab webhook bot that publishes events to Matrix"
|
||||
license = "GPL-3.0"
|
||||
license = "AGPL-3.0"
|
||||
authors = [
|
||||
"Brian J. Tarricone <brian@tarricone.org>",
|
||||
]
|
||||
homepage = "https://git.spurint.org/kelnos/bebot"
|
||||
repository = "https://git.spurint.org/kelnos/bebot"
|
||||
homepage = "https://git.spurint.org/brian/bebot"
|
||||
repository = "https://git.spurint.org/brian/bebot"
|
||||
edition = "2021"
|
||||
categories = [
|
||||
"development-tools",
|
||||
@@ -24,20 +24,21 @@ exclude = [
|
||||
|
||||
[dependencies]
|
||||
anyhow = "1"
|
||||
axum = "0.8.4"
|
||||
axum-extra = { version = "0.10.1", features = ["typed-header"] }
|
||||
chrono = { version = "0.4", features = ["serde"] }
|
||||
constant_time_eq = "0.3"
|
||||
constant_time_eq = "0.4"
|
||||
dateparser = "0.2"
|
||||
env_logger = "0.10"
|
||||
env_logger = "0.11"
|
||||
futures = "0.3"
|
||||
http = "0.2"
|
||||
http = "1.3"
|
||||
log = { version = "0.4", features = ["std"] }
|
||||
matrix-sdk = { version = "0.6", features = ["anyhow", "markdown", "rustls-tls"], default-features = false }
|
||||
quick-xml = { version = "0.30", features = ["serialize"] }
|
||||
matrix-sdk = { version = "0.13", features = ["anyhow", "markdown", "rustls-tls"], default-features = false }
|
||||
quick-xml = { version = "0.38", features = ["serialize"] }
|
||||
regex = "1"
|
||||
reqwest = { version = "0.11", default-features = false, features = ["tokio-rustls", "rustls-tls-native-roots", "gzip"] }
|
||||
reqwest = { version = "0.12", default-features = false, features = ["charset", "http2", "gzip", "rustls-tls-native-roots", "system-proxy"] }
|
||||
serde = { version = "1", features = ["derive"] }
|
||||
serde_json = "1"
|
||||
serde_regex = "1"
|
||||
serde_yaml = "0.9"
|
||||
tokio = { version = "1", default-features = false, features = ["rt-multi-thread", "macros", "time"] }
|
||||
warp = "0.3"
|
||||
|
@@ -1,4 +1,4 @@
|
||||
FROM rust:1.72-slim-bullseye AS builder
|
||||
FROM rust:1.88-slim-bookworm AS builder
|
||||
|
||||
WORKDIR /bebot-build
|
||||
|
||||
|
155
LICENSE
155
LICENSE
@@ -1,18 +1,5 @@
|
||||
bebot -- Gitlab webhook bot for Matrix
|
||||
Copyright (C) 2023 Brian Tarricone <brian@tarricone.org>
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
GNU GENERAL PUBLIC LICENSE
|
||||
Version 3, 29 June 2007
|
||||
GNU AFFERO GENERAL PUBLIC LICENSE
|
||||
Version 3, 19 November 2007
|
||||
|
||||
Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
|
||||
Everyone is permitted to copy and distribute verbatim copies
|
||||
@@ -20,17 +7,15 @@ GNU General Public License for more details.
|
||||
|
||||
Preamble
|
||||
|
||||
The GNU General Public License is a free, copyleft license for
|
||||
software and other kinds of works.
|
||||
The GNU Affero General Public License is a free, copyleft license for
|
||||
software and other kinds of works, specifically designed to ensure
|
||||
cooperation with the community in the case of network server software.
|
||||
|
||||
The licenses for most software and other practical works are designed
|
||||
to take away your freedom to share and change the works. By contrast,
|
||||
the GNU General Public License is intended to guarantee your freedom to
|
||||
our General Public Licenses are intended to guarantee your freedom to
|
||||
share and change all versions of a program--to make sure it remains free
|
||||
software for all its users. We, the Free Software Foundation, use the
|
||||
GNU General Public License for most of our software; it applies also to
|
||||
any other work released this way by its authors. You can apply it to
|
||||
your programs, too.
|
||||
software for all its users.
|
||||
|
||||
When we speak of free software, we are referring to freedom, not
|
||||
price. Our General Public Licenses are designed to make sure that you
|
||||
@@ -39,44 +24,34 @@ them if you wish), that you receive source code or can get it if you
|
||||
want it, that you can change the software or use pieces of it in new
|
||||
free programs, and that you know you can do these things.
|
||||
|
||||
To protect your rights, we need to prevent others from denying you
|
||||
these rights or asking you to surrender the rights. Therefore, you have
|
||||
certain responsibilities if you distribute copies of the software, or if
|
||||
you modify it: responsibilities to respect the freedom of others.
|
||||
Developers that use our General Public Licenses protect your rights
|
||||
with two steps: (1) assert copyright on the software, and (2) offer
|
||||
you this License which gives you legal permission to copy, distribute
|
||||
and/or modify the software.
|
||||
|
||||
For example, if you distribute copies of such a program, whether
|
||||
gratis or for a fee, you must pass on to the recipients the same
|
||||
freedoms that you received. You must make sure that they, too, receive
|
||||
or can get the source code. And you must show them these terms so they
|
||||
know their rights.
|
||||
A secondary benefit of defending all users' freedom is that
|
||||
improvements made in alternate versions of the program, if they
|
||||
receive widespread use, become available for other developers to
|
||||
incorporate. Many developers of free software are heartened and
|
||||
encouraged by the resulting cooperation. However, in the case of
|
||||
software used on network servers, this result may fail to come about.
|
||||
The GNU General Public License permits making a modified version and
|
||||
letting the public access it on a server without ever releasing its
|
||||
source code to the public.
|
||||
|
||||
Developers that use the GNU GPL protect your rights with two steps:
|
||||
(1) assert copyright on the software, and (2) offer you this License
|
||||
giving you legal permission to copy, distribute and/or modify it.
|
||||
The GNU Affero General Public License is designed specifically to
|
||||
ensure that, in such cases, the modified source code becomes available
|
||||
to the community. It requires the operator of a network server to
|
||||
provide the source code of the modified version running there to the
|
||||
users of that server. Therefore, public use of a modified version, on
|
||||
a publicly accessible server, gives the public access to the source
|
||||
code of the modified version.
|
||||
|
||||
For the developers' and authors' protection, the GPL clearly explains
|
||||
that there is no warranty for this free software. For both users' and
|
||||
authors' sake, the GPL requires that modified versions be marked as
|
||||
changed, so that their problems will not be attributed erroneously to
|
||||
authors of previous versions.
|
||||
|
||||
Some devices are designed to deny users access to install or run
|
||||
modified versions of the software inside them, although the manufacturer
|
||||
can do so. This is fundamentally incompatible with the aim of
|
||||
protecting users' freedom to change the software. The systematic
|
||||
pattern of such abuse occurs in the area of products for individuals to
|
||||
use, which is precisely where it is most unacceptable. Therefore, we
|
||||
have designed this version of the GPL to prohibit the practice for those
|
||||
products. If such problems arise substantially in other domains, we
|
||||
stand ready to extend this provision to those domains in future versions
|
||||
of the GPL, as needed to protect the freedom of users.
|
||||
|
||||
Finally, every program is threatened constantly by software patents.
|
||||
States should not allow patents to restrict development and use of
|
||||
software on general-purpose computers, but in those that do, we wish to
|
||||
avoid the special danger that patents applied to a free program could
|
||||
make it effectively proprietary. To prevent this, the GPL assures that
|
||||
patents cannot be used to render the program non-free.
|
||||
An older license, called the Affero General Public License and
|
||||
published by Affero, was designed to accomplish similar goals. This is
|
||||
a different license, not a version of the Affero GPL, but Affero has
|
||||
released a new version of the Affero GPL which permits relicensing under
|
||||
this license.
|
||||
|
||||
The precise terms and conditions for copying, distribution and
|
||||
modification follow.
|
||||
@@ -85,7 +60,7 @@ modification follow.
|
||||
|
||||
0. Definitions.
|
||||
|
||||
"This License" refers to version 3 of the GNU General Public License.
|
||||
"This License" refers to version 3 of the GNU Affero General Public License.
|
||||
|
||||
"Copyright" also means copyright-like laws that apply to other kinds of
|
||||
works, such as semiconductor masks.
|
||||
@@ -562,35 +537,45 @@ to collect a royalty for further conveying from those to whom you convey
|
||||
the Program, the only way you could satisfy both those terms and this
|
||||
License would be to refrain entirely from conveying the Program.
|
||||
|
||||
13. Use with the GNU Affero General Public License.
|
||||
13. Remote Network Interaction; Use with the GNU General Public License.
|
||||
|
||||
Notwithstanding any other provision of this License, if you modify the
|
||||
Program, your modified version must prominently offer all users
|
||||
interacting with it remotely through a computer network (if your version
|
||||
supports such interaction) an opportunity to receive the Corresponding
|
||||
Source of your version by providing access to the Corresponding Source
|
||||
from a network server at no charge, through some standard or customary
|
||||
means of facilitating copying of software. This Corresponding Source
|
||||
shall include the Corresponding Source for any work covered by version 3
|
||||
of the GNU General Public License that is incorporated pursuant to the
|
||||
following paragraph.
|
||||
|
||||
Notwithstanding any other provision of this License, you have
|
||||
permission to link or combine any covered work with a work licensed
|
||||
under version 3 of the GNU Affero General Public License into a single
|
||||
under version 3 of the GNU General Public License into a single
|
||||
combined work, and to convey the resulting work. The terms of this
|
||||
License will continue to apply to the part which is the covered work,
|
||||
but the special requirements of the GNU Affero General Public License,
|
||||
section 13, concerning interaction through a network will apply to the
|
||||
combination as such.
|
||||
but the work with which it is combined will remain governed by version
|
||||
3 of the GNU General Public License.
|
||||
|
||||
14. Revised Versions of this License.
|
||||
|
||||
The Free Software Foundation may publish revised and/or new versions of
|
||||
the GNU General Public License from time to time. Such new versions will
|
||||
be similar in spirit to the present version, but may differ in detail to
|
||||
the GNU Affero General Public License from time to time. Such new versions
|
||||
will be similar in spirit to the present version, but may differ in detail to
|
||||
address new problems or concerns.
|
||||
|
||||
Each version is given a distinguishing version number. If the
|
||||
Program specifies that a certain numbered version of the GNU General
|
||||
Program specifies that a certain numbered version of the GNU Affero General
|
||||
Public License "or any later version" applies to it, you have the
|
||||
option of following the terms and conditions either of that numbered
|
||||
version or of any later version published by the Free Software
|
||||
Foundation. If the Program does not specify a version number of the
|
||||
GNU General Public License, you may choose any version ever published
|
||||
GNU Affero General Public License, you may choose any version ever published
|
||||
by the Free Software Foundation.
|
||||
|
||||
If the Program specifies that a proxy can decide which future
|
||||
versions of the GNU General Public License can be used, that proxy's
|
||||
versions of the GNU Affero General Public License can be used, that proxy's
|
||||
public statement of acceptance of a version permanently authorizes you
|
||||
to choose that version for the Program.
|
||||
|
||||
@@ -648,41 +633,29 @@ the "copyright" line and a pointer to where the full notice is found.
|
||||
Copyright (C) <year> <name of author>
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
it under the terms of the GNU Affero General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
GNU Affero General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
You should have received a copy of the GNU Affero General Public License
|
||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
Also add information on how to contact you by electronic and paper mail.
|
||||
|
||||
If the program does terminal interaction, make it output a short
|
||||
notice like this when it starts in an interactive mode:
|
||||
|
||||
<program> Copyright (C) <year> <name of author>
|
||||
This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
|
||||
This is free software, and you are welcome to redistribute it
|
||||
under certain conditions; type `show c' for details.
|
||||
|
||||
The hypothetical commands `show w' and `show c' should show the appropriate
|
||||
parts of the General Public License. Of course, your program's commands
|
||||
might be different; for a GUI interface, you would use an "about box".
|
||||
If your software can interact with users remotely through a computer
|
||||
network, you should also make sure that it provides a way for users to
|
||||
get its source. For example, if your program is a web application, its
|
||||
interface could display a "Source" link that leads users to an archive
|
||||
of the code. There are many ways you could offer source, and different
|
||||
solutions will be better for different programs; see section 13 for the
|
||||
specific requirements.
|
||||
|
||||
You should also get your employer (if you work as a programmer) or school,
|
||||
if any, to sign a "copyright disclaimer" for the program, if necessary.
|
||||
For more information on this, and how to apply and follow the GNU GPL, see
|
||||
For more information on this, and how to apply and follow the GNU AGPL, see
|
||||
<https://www.gnu.org/licenses/>.
|
||||
|
||||
The GNU General Public License does not permit incorporating your program
|
||||
into proprietary programs. If your program is a subroutine library, you
|
||||
may consider it more useful to permit linking proprietary applications with
|
||||
the library. If this is what you want to do, use the GNU Lesser General
|
||||
Public License instead of this License. But first, please read
|
||||
<https://www.gnu.org/licenses/why-not-lgpl.html>.
|
||||
|
||||
|
37
README.md
37
README.md
@@ -11,11 +11,17 @@ Currently-supported Gitlab event types:
|
||||
* Merge request events
|
||||
* Pipeline events (only publishes on failure for now)
|
||||
|
||||
It can also watch for new messages in a mailing list (assuming that
|
||||
mailing list is tracked by [The Mail
|
||||
Archive](https://mail-archive.com/)), and publish notifications of new
|
||||
messages to Matrix.
|
||||
|
||||
## Building
|
||||
|
||||
Bebot is written in Rust, and requires a Rust toolchain in order to
|
||||
build. The usual `cargo build` or `cargo build --release` will do the
|
||||
trick.
|
||||
build. I'm actually not sure what Bebot's MSRV is, but as of this
|
||||
writing, 1.72 worked. The usual `cargo build` or `cargo build
|
||||
--release` will do the trick.
|
||||
|
||||
You can also build and install the latest released version of Bebot by
|
||||
running `cargo install bebot`.
|
||||
@@ -26,6 +32,8 @@ Bebot requires a configuration file in YAML format. See
|
||||
`sample-config.yaml` for all existing configuration options, as well as
|
||||
documentation on what each option does.
|
||||
|
||||
### Gitlab Hooks
|
||||
|
||||
When setting up the webhook in Gitlab, use the same `token` from the
|
||||
configuration file in the webhook's "Secret token" field. You should
|
||||
only select "Push events", "Tag push events", "Issues events", "Merge
|
||||
@@ -42,6 +50,18 @@ output to stdout a YAML snippet that goes under the `repo_configs`
|
||||
section of the configuration file. If you run the script with no
|
||||
arguments, it will print out usage details.
|
||||
|
||||
### `mail-archive.com`
|
||||
|
||||
If you want Bebot to publish a Matrix message when a new email hits one
|
||||
of your configured mailing lists, you need to provide a directory for
|
||||
Bebot to store state, so it can keep track of what emails it has already
|
||||
sent a Matrix message for. Otherwise, it will re-publish messages every
|
||||
time you restart Bebot.
|
||||
|
||||
Remember that if you are running Bebot in a Docker container, you'll
|
||||
need to mount a volume to store state so it persists across container
|
||||
restarts and upgrades.
|
||||
|
||||
## Running
|
||||
|
||||
After you've done all that, simply run Bebot:
|
||||
@@ -57,7 +77,18 @@ logging verbosity. (Try `debug`, `info`, `warn` `error`, or `off`.)
|
||||
|
||||
A `Dockerfile` is also provided. When running the container it builds,
|
||||
mount the configuration file so it appears inside the container as
|
||||
`/bebot/config/bebot.yaml`.
|
||||
`/bebot/config/bebot.yaml`. If you are publishing from
|
||||
`mail-archive.com`, also remember to mount a state storage directory or
|
||||
volume wherever you've specified in the configuration file.
|
||||
|
||||
Release images are [published to Docker
|
||||
Hub](https://hub.docker.com/r/kelnos/bebot).
|
||||
|
||||
## Contributing
|
||||
|
||||
I currently host Bebot on my [private Gitea
|
||||
server](https://git.spurint.org/brian/bebot). Since I don't want to
|
||||
deal with spam, I don't enable user registrations. If you'd like to
|
||||
submit issues and/or merge requests, please [message me on
|
||||
Matrix](https://matrix.to/#/@brian:tarricone.org) with your email
|
||||
address and preferred username, and I'll create an account for you.
|
||||
|
@@ -7,7 +7,7 @@ user_id: "@mybebot:example.com"
|
||||
# Password for Matrix user.
|
||||
password: "secret-matrix-account-password"
|
||||
# All Gitlab-specific settings are under here.
|
||||
gitlab:
|
||||
gitlab_webhook:
|
||||
# Optional prefix to serve the webhook path under (default is empty string).
|
||||
url_prefix: "/bebot"
|
||||
# Default Matrix room to publish Gitlab events to.
|
||||
@@ -58,3 +58,28 @@ gitlab:
|
||||
"gitlab.example.com/myuser/some-other-less-cool-app":
|
||||
token: "kljaslkdjaklsdjalksd"
|
||||
# This repo uses the default events and room.
|
||||
# The mail_archive configuration section allows you to set up bebot to publish
|
||||
# messages based on RSS feeds from mail-archive.com.
|
||||
mail_archive:
|
||||
# List of rooms that will be published to by default, unless overridden by
|
||||
# a per-list config.
|
||||
default_rooms:
|
||||
- "#some-room:example.com"
|
||||
- "#some-other-room:example.com"
|
||||
# How often bebot will fetch the RSS feed to check for updates, in seconds.
|
||||
update_interval: 60
|
||||
# A directory where bebot can store state, such as the data of the last
|
||||
# entry in the RSS feed it has seen.
|
||||
state_dir: "/var/lib/bebot/mail-archive-state"
|
||||
# A list of mailing lists.
|
||||
lists:
|
||||
# This is the list name as is displayed in mail-archive.com URLS.
|
||||
- name: "my-list@example.com"
|
||||
# Disable publishing a matrix message for replies sent to the list
|
||||
# (default true). This isn't perfect, and can only guess if a message
|
||||
# is a reply based on the subject line.
|
||||
publish_on_replies: false
|
||||
# An optional list of rooms to publish to. If not specified, the
|
||||
# default_rooms setting above will be used.
|
||||
rooms:
|
||||
- "#yet-some-other-room:example.com"
|
||||
|
@@ -1,18 +1,18 @@
|
||||
// bebot
|
||||
// Copyright (C) 2023 Brian Tarricone <brian@tarricone.org>
|
||||
// bebot -- a Gitlab -> Matrix event publisher
|
||||
// Copyright (C) 2023-2025 Brian Tarricone <brian@tarricone.org>
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
use std::{collections::HashMap, fs::File, io::BufReader, path::PathBuf};
|
||||
|
||||
@@ -65,6 +65,8 @@ pub struct GitlabWebhookConfig {
|
||||
#[derive(Clone, Deserialize)]
|
||||
pub struct MailListConfig {
|
||||
pub name: String,
|
||||
#[serde(default = "default_true")]
|
||||
pub publish_on_replies: bool,
|
||||
#[serde(default)]
|
||||
pub rooms: Vec<OwnedRoomOrAliasId>,
|
||||
}
|
||||
@@ -89,6 +91,10 @@ pub struct Config {
|
||||
pub mail_archive: Option<MailArchiveConfig>,
|
||||
}
|
||||
|
||||
fn default_true() -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn load_blocking(path: &String) -> anyhow::Result<Config> {
|
||||
let f = File::open(path)?;
|
||||
let r = BufReader::new(f);
|
||||
@@ -99,7 +105,7 @@ fn load_blocking(path: &String) -> anyhow::Result<Config> {
|
||||
pub async fn load<S: AsRef<str>>(path: S) -> anyhow::Result<Config> {
|
||||
let p = String::from(path.as_ref());
|
||||
let config = tokio::task::spawn_blocking(move || {
|
||||
load_blocking(&p).with_context(|| format!("Failed to load config from {}", p))
|
||||
load_blocking(&p).with_context(|| format!("Failed to load config from {p}"))
|
||||
})
|
||||
.await??;
|
||||
Ok(config)
|
||||
|
@@ -1,18 +1,20 @@
|
||||
// bebot
|
||||
// Copyright (C) 2023 Brian Tarricone <brian@tarricone.org>
|
||||
// bebot -- a Gitlab -> Matrix event publisher
|
||||
// Copyright (C) 2023-2025 Brian Tarricone <brian@tarricone.org>
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
#![allow(unused)]
|
||||
|
||||
use core::fmt;
|
||||
|
||||
@@ -281,7 +283,7 @@ impl GitlabEventExt for GitlabEvent {
|
||||
false
|
||||
}
|
||||
}
|
||||
GitlabEvent::TagPush { .. } => find_publish_event!(publish_events, PublishEvent::TagPush { .. }).is_some(),
|
||||
GitlabEvent::TagPush { .. } => find_publish_event!(publish_events, PublishEvent::TagPush).is_some(),
|
||||
GitlabEvent::Issue { object_attributes, .. } => {
|
||||
if let Some(PublishEvent::Issues { actions }) =
|
||||
find_publish_event!(publish_events, PublishEvent::Issues { .. })
|
||||
@@ -437,7 +439,7 @@ impl GitlabEventExt for GitlabEvent {
|
||||
.or(merge_request.as_ref().map(|mr| mr.title.clone()))
|
||||
.iter()
|
||||
.fold(format!("Pipeline **{}**", object_attributes.status), |accum, title| {
|
||||
format!("{}: {}", accum, title)
|
||||
format!("{accum}: {title}")
|
||||
});
|
||||
vec![markdown_link(&title, &object_attributes.url)]
|
||||
} else {
|
||||
@@ -451,7 +453,7 @@ impl GitlabEventExt for GitlabEvent {
|
||||
|
||||
#[inline]
|
||||
fn markdown_link(title: &String, url: &String) -> String {
|
||||
format!("[{}]({})", title, url)
|
||||
format!("[{title}]({url})")
|
||||
}
|
||||
|
||||
pub fn parse_ref(r#ref: &str) -> String {
|
||||
|
@@ -1,29 +1,30 @@
|
||||
// bebot
|
||||
// Copyright (C) 2023 Brian Tarricone <brian@tarricone.org>
|
||||
// bebot -- a Gitlab -> Matrix event publisher
|
||||
// Copyright (C) 2023-2025 Brian Tarricone <brian@tarricone.org>
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
use std::sync::Arc;
|
||||
|
||||
use axum::{extract::State, routing::post, Json, Router};
|
||||
use axum_extra::{headers, TypedHeader};
|
||||
use constant_time_eq::constant_time_eq;
|
||||
use http::StatusCode;
|
||||
use http::{HeaderName, HeaderValue, StatusCode};
|
||||
use matrix_sdk::{
|
||||
ruma::{events::room::message::RoomMessageEventContent, OwnedRoomOrAliasId},
|
||||
Client,
|
||||
};
|
||||
use tokio::sync::mpsc;
|
||||
use warp::{filters::BoxedFilter, reply::Reply, Filter};
|
||||
use tokio::sync::mpsc::{self, Sender};
|
||||
|
||||
use crate::{
|
||||
config::GitlabWebhookConfig,
|
||||
@@ -31,6 +32,42 @@ use crate::{
|
||||
matrix,
|
||||
};
|
||||
|
||||
static X_GITLAB_TOKEN: HeaderName = HeaderName::from_static("x-gitlab-token");
|
||||
|
||||
struct XGitlabToken(String);
|
||||
|
||||
impl headers::Header for XGitlabToken {
|
||||
fn name() -> &'static HeaderName {
|
||||
&X_GITLAB_TOKEN
|
||||
}
|
||||
|
||||
fn decode<'i, I>(values: &mut I) -> Result<Self, headers::Error>
|
||||
where
|
||||
Self: Sized,
|
||||
I: Iterator<Item = &'i http::HeaderValue>,
|
||||
{
|
||||
let value = values.next().ok_or_else(headers::Error::invalid)?;
|
||||
|
||||
if value.is_empty() {
|
||||
Err(headers::Error::invalid())
|
||||
} else {
|
||||
Ok(XGitlabToken(
|
||||
value.to_str().map_err(|_| headers::Error::invalid())?.to_string(),
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
fn encode<E: Extend<http::HeaderValue>>(&self, values: &mut E) {
|
||||
values.extend(std::iter::once(HeaderValue::from_str(self.0.as_str()).unwrap()));
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
struct WebhookState {
|
||||
config: Arc<GitlabWebhookConfig>,
|
||||
event_tx: Sender<(GitlabEvent, OwnedRoomOrAliasId)>,
|
||||
}
|
||||
|
||||
pub fn build_gitlab_messages(event: &GitlabEvent) -> Vec<String> {
|
||||
let project = event.project();
|
||||
let refname = event.r#ref().map(parse_ref);
|
||||
@@ -41,10 +78,7 @@ pub fn build_gitlab_messages(event: &GitlabEvent) -> Vec<String> {
|
||||
format!(
|
||||
"\\[{}\\] {}*{}* {}",
|
||||
project.path_with_namespace,
|
||||
refname
|
||||
.as_ref()
|
||||
.map(|rn| format!("`{}` ", rn))
|
||||
.unwrap_or_else(|| "".to_string()),
|
||||
refname.as_ref().map(|rn| format!("`{rn}` ")).unwrap_or_default(),
|
||||
event.user(),
|
||||
title,
|
||||
)
|
||||
@@ -59,87 +93,73 @@ pub async fn handle_gitlab_event(
|
||||
) -> anyhow::Result<()> {
|
||||
let room = matrix::ensure_room_joined(matrix_client, room_id).await?;
|
||||
for msg in build_gitlab_messages(&event) {
|
||||
debug!("Sending message to {}: {}", room_id, msg);
|
||||
debug!("Sending message to {room_id}: {msg}");
|
||||
let msg_content = RoomMessageEventContent::text_markdown(&msg);
|
||||
room.send(msg_content, None).await?;
|
||||
room.send(msg_content).await?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn build_route(config: GitlabWebhookConfig, matrix_client: Client) -> anyhow::Result<BoxedFilter<(impl Reply,)>> {
|
||||
async fn handle_hooks_gitlab(
|
||||
State(state): State<WebhookState>,
|
||||
TypedHeader(token): TypedHeader<XGitlabToken>,
|
||||
Json(event): Json<GitlabEvent>,
|
||||
) -> (StatusCode, &'static str) {
|
||||
match event {
|
||||
GitlabEvent::Other => (StatusCode::BAD_REQUEST, "Unsupported Gitlab event type"),
|
||||
_ => {
|
||||
let project = event.project();
|
||||
let config_key = project.web_url.replace("http://", "").replace("https://", "");
|
||||
if let Some(repo_config) = state.config.repo_configs.get(&config_key) {
|
||||
if !constant_time_eq(token.0.as_bytes(), repo_config.token.as_bytes()) {
|
||||
warn!("Invalid token for repo '{config_key}'");
|
||||
(StatusCode::FORBIDDEN, "Invalid token")
|
||||
} else {
|
||||
debug!("payload: {event:?}");
|
||||
if let Some(room) = &repo_config.room.as_ref().or(state.config.default_room.as_ref()) {
|
||||
let publish_events = repo_config
|
||||
.publish_events
|
||||
.as_ref()
|
||||
.or(state.config.default_publish_events.as_ref());
|
||||
if publish_events.map(|ecs| event.should_publish(ecs)).unwrap_or(true) {
|
||||
if let Err(err) = state.event_tx.send((event, (*room).clone())).await {
|
||||
warn!("Failed to enqueue payload: {err}");
|
||||
}
|
||||
}
|
||||
(StatusCode::OK, "OK")
|
||||
} else {
|
||||
info!("Channel not configured for repo '{config_key}'");
|
||||
(StatusCode::NOT_FOUND, "Matrix room not configured for repo")
|
||||
}
|
||||
}
|
||||
} else {
|
||||
info!("Repo '{config_key}' unconfigured");
|
||||
(StatusCode::NOT_FOUND, "Repo not configured")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn build_route(config: GitlabWebhookConfig, matrix_client: Client) -> Router {
|
||||
let (event_tx, mut event_rx) = mpsc::channel::<(GitlabEvent, OwnedRoomOrAliasId)>(100);
|
||||
tokio::spawn(async move {
|
||||
while let Some((event, room)) = event_rx.recv().await {
|
||||
if let Err(err) = handle_gitlab_event(event, &room, &matrix_client).await {
|
||||
warn!("Failed to handle payload: {}", err);
|
||||
warn!("Failed to handle payload: {err}");
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
let gitlab_root_path = if let Some(url_prefix) = config.url_prefix.as_ref() {
|
||||
url_prefix.split('/').fold(warp::any().boxed(), |last, segment| {
|
||||
if segment.is_empty() {
|
||||
last
|
||||
} else {
|
||||
last.and(warp::path(segment.to_string())).boxed()
|
||||
}
|
||||
})
|
||||
let path = if let Some(url_prefix) = &config.url_prefix {
|
||||
format!("{url_prefix}/hooks/gitlab")
|
||||
} else {
|
||||
warp::any().boxed()
|
||||
"/hooks/gitlab".to_owned()
|
||||
};
|
||||
|
||||
let config = Arc::new(config);
|
||||
let gitlab = gitlab_root_path
|
||||
.and(warp::path!("hooks" / "gitlab"))
|
||||
.and(warp::post())
|
||||
.and(warp::header::<String>("x-gitlab-token"))
|
||||
.and(warp::body::json())
|
||||
.then(move |token: String, event: GitlabEvent| {
|
||||
let event_tx = event_tx.clone();
|
||||
let config = Arc::clone(&config);
|
||||
let state = WebhookState {
|
||||
config: Arc::new(config),
|
||||
event_tx,
|
||||
};
|
||||
|
||||
async move {
|
||||
match event {
|
||||
GitlabEvent::Other => {
|
||||
warp::reply::with_status("Unsupported Gitlab event type", StatusCode::BAD_REQUEST)
|
||||
}
|
||||
_ => {
|
||||
let project = event.project();
|
||||
let config_key = project.web_url.replace("http://", "").replace("https://", "");
|
||||
if let Some(repo_config) = config.repo_configs.get(&config_key) {
|
||||
if !constant_time_eq(token.as_bytes(), repo_config.token.as_bytes()) {
|
||||
warn!("Invalid token for repo '{}'", config_key);
|
||||
warp::reply::with_status("Invalid token", StatusCode::FORBIDDEN)
|
||||
} else {
|
||||
debug!("payload: {:?}", event);
|
||||
if let Some(room) = &repo_config.room.as_ref().or(config.default_room.as_ref()) {
|
||||
let publish_events = repo_config
|
||||
.publish_events
|
||||
.as_ref()
|
||||
.or(config.default_publish_events.as_ref());
|
||||
if publish_events.map(|ecs| event.should_publish(ecs)).unwrap_or(true) {
|
||||
if let Err(err) = event_tx.send((event, (*room).clone())).await {
|
||||
warn!("Failed to enqueue payload: {}", err);
|
||||
}
|
||||
}
|
||||
warp::reply::with_status("OK", StatusCode::OK)
|
||||
} else {
|
||||
info!("Channel not configured for repo '{}'", config_key);
|
||||
warp::reply::with_status(
|
||||
"Matrix room not configured for repo",
|
||||
StatusCode::NOT_FOUND,
|
||||
)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
info!("Repo '{}' unconfigured", config_key);
|
||||
warp::reply::with_status("Repo not configured", StatusCode::NOT_FOUND)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.boxed();
|
||||
|
||||
Ok(gitlab)
|
||||
Router::new().route(&path, post(handle_hooks_gitlab)).with_state(state)
|
||||
}
|
||||
|
@@ -1,18 +1,18 @@
|
||||
// bebot
|
||||
// Copyright (C) 2023 Brian Tarricone <brian@tarricone.org>
|
||||
// bebot -- a Gitlab -> Matrix event publisher
|
||||
// Copyright (C) 2023-2025 Brian Tarricone <brian@tarricone.org>
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
use std::{
|
||||
fmt,
|
||||
@@ -105,20 +105,21 @@ async fn handle_list(
|
||||
http_client: &reqwest::Client,
|
||||
url: &String,
|
||||
matrix_client: &Client,
|
||||
publish_on_replies: bool,
|
||||
room_ids: &[OwnedRoomOrAliasId],
|
||||
) -> anyhow::Result<()> {
|
||||
let list_state = load_list_state(state_file).await?;
|
||||
|
||||
let rooms_f = room_ids.iter().map(|room_id| {
|
||||
matrix::ensure_room_joined(matrix_client, room_id)
|
||||
.map(move |res| res.with_context(|| format!("Failed to join Matrix room '{}'", room_id)))
|
||||
.map(move |res| res.with_context(|| format!("Failed to join Matrix room '{room_id}'")))
|
||||
});
|
||||
let rooms = join_all(rooms_f)
|
||||
.await
|
||||
.into_iter()
|
||||
.flat_map(|room_res| match room_res {
|
||||
Err(err) => {
|
||||
warn!("{:#}", err);
|
||||
warn!("{err:#}");
|
||||
vec![]
|
||||
}
|
||||
Ok(room) => vec![room],
|
||||
@@ -132,7 +133,7 @@ async fn handle_list(
|
||||
.get(url)
|
||||
.send()
|
||||
.await
|
||||
.with_context(|| format!("Failed to fetch mail RSS feed from '{}'", url))
|
||||
.with_context(|| format!("Failed to fetch mail RSS feed from '{url}'"))
|
||||
.and_then(|response| {
|
||||
if !response.status().is_success() {
|
||||
Err(anyhow!(
|
||||
@@ -147,10 +148,10 @@ async fn handle_list(
|
||||
let body = response
|
||||
.text()
|
||||
.await
|
||||
.with_context(|| format!("Failed to decode RSS response body for '{}'", url))?;
|
||||
.with_context(|| format!("Failed to decode RSS response body for '{url}'"))?;
|
||||
let mail_rss = tokio::task::spawn_blocking(move || quick_xml::de::from_str::<MailRss>(&body))
|
||||
.await?
|
||||
.with_context(|| format!("Failed to parse RSS feed for '{}'", url))?;
|
||||
.with_context(|| format!("Failed to parse RSS feed for '{url}'"))?;
|
||||
let items = mail_rss
|
||||
.channel
|
||||
.items
|
||||
@@ -161,11 +162,15 @@ async fn handle_list(
|
||||
|
||||
for room in rooms {
|
||||
for item in &items {
|
||||
let msg =
|
||||
RoomMessageEventContent::text_markdown(format!("\\[{}\\] [{}]({})", list.name, item.title, item.link));
|
||||
room.send(msg, None)
|
||||
.await
|
||||
.with_context(|| format!("Failed to send message to room '{}'", room.room_id()))?;
|
||||
if publish_on_replies || !item.title.starts_with("Re: ") {
|
||||
let msg = RoomMessageEventContent::text_markdown(format!(
|
||||
"\\[{}\\] [{}]({})",
|
||||
list.name, item.title, item.link
|
||||
));
|
||||
room.send(msg)
|
||||
.await
|
||||
.with_context(|| format!("Failed to send message to room '{}'", room.room_id()))?;
|
||||
}
|
||||
save_list_state(
|
||||
ListState {
|
||||
last_pub_date: item.pub_date.value,
|
||||
@@ -207,10 +212,18 @@ pub fn start_polling(config: MailArchiveConfig, matrix_client: Client) -> anyhow
|
||||
tokio::spawn(async move {
|
||||
if !room_ids.is_empty() {
|
||||
loop {
|
||||
if let Err(err) =
|
||||
handle_list(&list, &state_file, &http_client, &url, &matrix_client, &room_ids).await
|
||||
if let Err(err) = handle_list(
|
||||
&list,
|
||||
&state_file,
|
||||
&http_client,
|
||||
&url,
|
||||
&matrix_client,
|
||||
list.publish_on_replies,
|
||||
&room_ids,
|
||||
)
|
||||
.await
|
||||
{
|
||||
warn!("{:#}", err);
|
||||
warn!("{err:#}");
|
||||
}
|
||||
|
||||
sleep(update_interval).await;
|
||||
@@ -256,7 +269,7 @@ mod test {
|
||||
let f = File::open(format!("{}/test-data/maillist.xml", env!("CARGO_MANIFEST_DIR")))?;
|
||||
let r = BufReader::new(f);
|
||||
let mail_rss = quick_xml::de::from_reader::<_, MailRss>(r)?;
|
||||
println!("{:#?}", mail_rss);
|
||||
println!("{mail_rss:#?}");
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
40
src/main.rs
40
src/main.rs
@@ -1,18 +1,18 @@
|
||||
// bebot
|
||||
// Copyright (C) 2023 Brian Tarricone <brian@tarricone.org>
|
||||
// bebot -- a Gitlab -> Matrix event publisher
|
||||
// Copyright (C) 2023-2025 Brian Tarricone <brian@tarricone.org>
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
#[macro_use(anyhow)]
|
||||
extern crate anyhow;
|
||||
@@ -27,13 +27,15 @@ mod gitlab_webhook;
|
||||
mod mail_archive;
|
||||
mod matrix;
|
||||
|
||||
use std::{env, net::IpAddr, process::exit};
|
||||
use std::{env, process::exit};
|
||||
|
||||
use anyhow::Context;
|
||||
use futures::future::join_all;
|
||||
use warp::Filter;
|
||||
use tokio::net::TcpListener;
|
||||
|
||||
async fn run() -> anyhow::Result<()> {
|
||||
info!("{} v{} starting...", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
|
||||
|
||||
let config_path = env::args()
|
||||
.nth(1)
|
||||
.ok_or_else(|| anyhow!("Config file should be passed as only parameter"))?;
|
||||
@@ -48,18 +50,14 @@ async fn run() -> anyhow::Result<()> {
|
||||
};
|
||||
|
||||
if let Some(gitlab_webhook) = config.gitlab_webhook.take() {
|
||||
let gitlab = gitlab_webhook::build_route(gitlab_webhook, matrix_client.clone())?;
|
||||
let routes = gitlab.with(warp::log("bebot"));
|
||||
|
||||
let addr = config
|
||||
.bind_address
|
||||
.as_ref()
|
||||
.cloned()
|
||||
.unwrap_or_else(|| "127.0.0.1".to_string())
|
||||
.parse::<IpAddr>()
|
||||
.context("Failed to parse bind_address")?;
|
||||
let port = config.bind_port.unwrap_or(3000);
|
||||
warp::serve(routes).run((addr, port)).await;
|
||||
let gitlab = gitlab_webhook::build_route(gitlab_webhook, matrix_client.clone());
|
||||
let bind_addr = format!(
|
||||
"{}:{}",
|
||||
config.bind_address.as_deref().unwrap_or("127.0.0.1"),
|
||||
config.bind_port.unwrap_or(3000)
|
||||
);
|
||||
let listener = TcpListener::bind(bind_addr).await?;
|
||||
axum::serve(listener, gitlab).await?;
|
||||
}
|
||||
|
||||
join_all(handles).await;
|
||||
@@ -76,7 +74,7 @@ async fn main() {
|
||||
env_logger::init_from_env(lenv);
|
||||
|
||||
if let Err(err) = run().await {
|
||||
error!("{:#}", err);
|
||||
error!("{err:#}");
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
|
@@ -1,37 +1,34 @@
|
||||
// bebot
|
||||
// Copyright (C) 2023 Brian Tarricone <brian@tarricone.org>
|
||||
// bebot -- a Gitlab -> Matrix event publisher
|
||||
// Copyright (C) 2023-2025 Brian Tarricone <brian@tarricone.org>
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// it under the terms of the GNU Affero General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
// GNU Affero General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
// You should have received a copy of the GNU Affero General Public License
|
||||
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
use std::{fmt, process::exit, time::Duration};
|
||||
|
||||
use matrix_sdk::{
|
||||
config::SyncSettings,
|
||||
room::Joined,
|
||||
room::Room,
|
||||
ruma::{OwnedRoomOrAliasId, OwnedUserId, RoomOrAliasId, UserId},
|
||||
BaseRoom, Client,
|
||||
};
|
||||
use serde::de;
|
||||
use tokio::time::sleep;
|
||||
|
||||
use crate::config::Config;
|
||||
|
||||
async fn build_sync_settings(matrix_client: &Client) -> SyncSettings {
|
||||
let mut settings = SyncSettings::default().timeout(Duration::from_secs(30));
|
||||
if let Some(token) = matrix_client.sync_token().await {
|
||||
settings = settings.token(token);
|
||||
}
|
||||
settings
|
||||
async fn build_sync_settings() -> SyncSettings {
|
||||
SyncSettings::default().timeout(Duration::from_secs(30))
|
||||
}
|
||||
|
||||
pub async fn connect(config: &Config) -> anyhow::Result<Client> {
|
||||
@@ -41,21 +38,21 @@ pub async fn connect(config: &Config) -> anyhow::Result<Client> {
|
||||
.build()
|
||||
.await?;
|
||||
client
|
||||
.matrix_auth()
|
||||
.login_username(&config.user_id, &config.password)
|
||||
.initial_device_display_name("Bebot")
|
||||
.send()
|
||||
.await?;
|
||||
info!("Connected to matrix as {}; waiting for first sync", config.user_id);
|
||||
|
||||
let settings = build_sync_settings(&client).await;
|
||||
let settings = build_sync_settings().await;
|
||||
client.sync_once(settings).await?;
|
||||
info!("First matrix sync complete");
|
||||
|
||||
let sync_client = client.clone();
|
||||
tokio::spawn(async move {
|
||||
let settings = build_sync_settings(&sync_client).await;
|
||||
let settings = build_sync_settings().await;
|
||||
if let Err(err) = sync_client.sync(settings).await {
|
||||
error!("Matrix sync failed: {}", err);
|
||||
error!("Matrix sync failed: {err}");
|
||||
exit(1);
|
||||
}
|
||||
});
|
||||
@@ -63,7 +60,7 @@ pub async fn connect(config: &Config) -> anyhow::Result<Client> {
|
||||
Ok(client)
|
||||
}
|
||||
|
||||
pub async fn ensure_room_joined(matrix_client: &Client, room_id: &OwnedRoomOrAliasId) -> anyhow::Result<Joined> {
|
||||
pub async fn ensure_room_joined(matrix_client: &Client, room_id: &OwnedRoomOrAliasId) -> anyhow::Result<Room> {
|
||||
fn room_matches(a_room: &BaseRoom, our_room: &OwnedRoomOrAliasId) -> bool {
|
||||
let our_room_str = our_room.as_str();
|
||||
a_room.room_id().as_str() == our_room_str
|
||||
@@ -85,17 +82,26 @@ pub async fn ensure_room_joined(matrix_client: &Client, room_id: &OwnedRoomOrAli
|
||||
.iter()
|
||||
.find(|a_room| room_matches(a_room, room_id))
|
||||
{
|
||||
invited.accept_invitation().await?;
|
||||
info!("Accepting invitation to room {room_id}");
|
||||
invited.join().await?;
|
||||
} else {
|
||||
info!("Joining room {room_id}");
|
||||
matrix_client.join_room_by_id_or_alias(room_id, &[]).await?;
|
||||
}
|
||||
let settings = build_sync_settings(matrix_client).await;
|
||||
matrix_client.sync_once(settings).await?;
|
||||
room = matrix_client
|
||||
.joined_rooms()
|
||||
.iter()
|
||||
.find(|a_room| room_matches(a_room, room_id))
|
||||
.cloned();
|
||||
|
||||
for _ in 0..4 {
|
||||
let settings = build_sync_settings().await;
|
||||
matrix_client.sync_once(settings).await?;
|
||||
room = matrix_client
|
||||
.joined_rooms()
|
||||
.iter()
|
||||
.find(|a_room| room_matches(a_room, room_id))
|
||||
.cloned();
|
||||
if room.is_some() {
|
||||
break;
|
||||
}
|
||||
sleep(Duration::from_millis(500)).await;
|
||||
}
|
||||
}
|
||||
|
||||
room.ok_or_else(|| anyhow!("Unable to join room {}", room_id))
|
||||
|
Reference in New Issue
Block a user