66 Commits

Author SHA1 Message Date
Ulyssa
89bb107c87 Release v0.0.8 (fix cargo publish issues) (#134) 2023-07-07 23:21:47 -07:00
Ulyssa
ca4c0034d9 Release v0.0.8 (#134) 2023-07-07 22:46:08 -07:00
Ulyssa
bb30cecc63 Handle sync failure after successful password entry (#133) 2023-07-07 22:35:33 -07:00
Ulyssa
7b050f82aa Indicate when there are new messages below scrollback viewport (#131) 2023-07-07 22:16:57 -07:00
Ulyssa
b1ccec6732 Code blocks get rendered without line breaks (#122) 2023-07-07 21:46:13 -07:00
Ulyssa
6e8e12b579 Need fallback behaviour when dirs::download_dir returns None (#118) 2023-07-07 20:35:01 -07:00
Ulyssa
3da9835a17 Profile session token should only be readable by the user (#130) 2023-07-07 20:34:52 -07:00
Ulyssa
64891ec68f Support hiding server part of username in message scrollback (#71) 2023-07-06 23:15:58 -07:00
Ulyssa
61aba80be1 Reduce number of Tokio workers (#129) 2023-07-05 15:25:42 -07:00
Ulyssa
8d4539831f Remove trailing newlines in body (#125) 2023-06-30 21:18:08 -07:00
Ulyssa
7c39e88ba2 Restore opened tabs and windows upon restart (#72) 2023-06-28 23:42:31 -07:00
satoqz
758397eb5a Fix Nix flake build on Darwin (#117) 2023-06-22 23:05:50 -07:00
u2on
1a0af6df37 Link to AUR pkg in README (#121) 2023-06-22 23:01:07 -07:00
Ulyssa
885b56038f Use terminal window focus to determine when a message has actually been seen (#94) 2023-06-14 22:42:23 -07:00
Benjamin Große
430c601ff2 Support configuring which program :open runs (#95) 2023-06-14 21:36:23 -07:00
Moritz Poldrack
0ddefcd7b3 Add manual pages (#88) 2023-06-14 21:14:23 -07:00
Ulyssa
2a573b6056 Show Git SHA information when printing version information (#120) 2023-06-14 20:28:01 -07:00
mikoto
a020b860dd Indicate number of members in room (#110) 2023-06-14 19:42:53 -07:00
jasalltime
6c031f589e Mention Minimum Supported Rust Version in README (#115) 2023-06-14 19:28:23 -07:00
Ulyssa
b0256d7120 Replace GitHub actions using deprecated features (#114) 2023-05-28 21:46:43 -07:00
Ulyssa
0f870367b3 Show errors fetching space hierarchy when list is empty (#113) 2023-05-28 13:16:37 -07:00
Ulyssa
8d22b83d85 Support sending and completing Emoji shortcodes in the message bar (#100) 2023-05-24 21:14:13 -07:00
Pavlo Rudy
529073f4d4 Upload artifacts built in GitHub Actions (#105) 2023-05-22 17:20:17 -07:00
Ulyssa
17c87a617e Cache build directory in GitHub Actions (#107) 2023-05-19 18:25:09 -07:00
Benjamin Große
2899d4f45a Support uploading image attachments from clipboard (#36) 2023-05-19 17:38:23 -07:00
Ulyssa
ad8b4a60d2 ChatStore::set_receipts locks up app for bad connections (#99) 2023-05-12 17:42:25 -07:00
Ulyssa
4935899aed Indicate when you're editing a message (#75) 2023-05-01 22:14:08 -07:00
Ulyssa
cc1d2f3bf8 Gracefully handle verification events that are unknown locally (#90) 2023-05-01 21:33:12 -07:00
Ulyssa
5df9fe7960 Tab completion panics for unrecognized commands (#81) 2023-05-01 21:14:19 -07:00
Ulyssa
a5c25f2487 Support leaving rooms (#45) 2023-04-28 16:52:33 -07:00
Benjamin Große
50023bad40 Append suffix to download filenames to avoid overwrites (#35) 2023-04-28 15:56:14 -07:00
Moritz Poldrack
b6a318dfe3 Fix error message for undefined download directory (#87) 2023-04-25 13:57:03 -07:00
Ulyssa
ad3b40d538 Interpret newlines as line breaks when converting Markdown to HTML (#74) 2023-04-06 16:10:48 -07:00
Ulyssa
953be6a195 Add FUNDING.yml to project (#77) 2023-03-31 18:38:13 -07:00
Benjamin Große
463d46b8ab Add Nix flake (#73) 2023-03-31 11:43:22 -07:00
Ulyssa
274234ce4c Update locked Cargo dependencies (#70) 2023-03-23 13:39:57 -07:00
Ulyssa
a2590b6bbb Release v0.0.7 (#68) 2023-03-22 21:28:34 -07:00
Ulyssa
725ebb9fd6 Redacted messages should have their HTML removed (#67) 2023-03-22 21:25:37 -07:00
jahway603
ca395097e1 Update README.md to include Arch Linux package (#66) 2023-03-22 20:13:25 -07:00
Ulyssa
e98d58a8cc Emote messages should always show sender (#65) 2023-03-21 14:02:42 -07:00
Ulyssa
e6cdd02f22 HTML self-closing tags are getting parsed incorrectly (#63) 2023-03-20 17:53:55 -07:00
Ulyssa
0bc4ff07b0 Lazy load room state events on initial sync (#62) 2023-03-20 16:17:59 -07:00
Ulyssa
14fe916d94 Allow log level to be configured (#58) 2023-03-13 16:43:08 -07:00
Ulyssa
db35581d07 Indicate when an encrypted room event has been redacted (#59) 2023-03-13 16:43:04 -07:00
Ulyssa
7c1c62897a Show events that couldn't be decrypted (#57) 2023-03-13 15:18:53 -07:00
Ulyssa
61897ea6f2 Fetch scrollback history independently of main loop (#39) 2023-03-13 10:46:26 -07:00
Ulyssa
6a0722795a Fix empty message check when sending (#56) 2023-03-13 09:26:49 -07:00
Ulyssa
f3bbc6ad9f Support configuring client request timeout (#54) 2023-03-12 15:43:13 -07:00
Ulyssa
2dd8c0fddf Link to iamb space in README (#55) 2023-03-10 18:08:42 -08:00
Pavlo Rudy
a786369b14 Create release profile with LTO (#52) 2023-03-10 16:41:32 -08:00
pin
066f60ad32 Add NetBSD install instructions (#51) 2023-03-09 09:27:40 -08:00
Ulyssa
10b142c071 Release v0.0.6 (#48) 2023-03-05 13:28:08 -08:00
Ulyssa
ac6ff63d25 Avoid breaking up words during wrapping when possible (#47) 2023-03-05 12:59:34 -08:00
Ulyssa
54a0e76823 Edited messages need to have their HTML reprocessed (#46) 2023-03-05 12:48:31 -08:00
Ulyssa
93eff79f79 Support creating new rooms and spaces (#40) 2023-03-04 12:23:17 -08:00
Ulyssa
11625262f1 Direct message rooms should be encrypted from creation (#29) 2023-03-03 16:37:11 -08:00
Ulyssa
0ed1d53946 Support completing commands, usernames, and room names (#44) 2023-03-01 18:46:33 -08:00
Ulyssa
e3be8c16cb Release v0.0.5 (#38) 2023-02-09 23:22:19 -08:00
Ulyssa
4c5c57e26c Window keybindings should be mapped in Visual mode (#37) 2023-02-09 23:05:02 -08:00
Benjamin Große
8eef8787cc fix: attachment download flags + exists check (#34)
Fix files never downloading (unless it has been downloaded in the past
and using `!` force flag).

The logic should be:

* If file does not exist, or `!` force flag used, then download it
* Else if neither `!` or `:open` flag used, then error out

and then return downloaded-message or open-and-message.

I.e. `:open` should still open the file if it has already been
downloaded. Otherwise the only way to open it is to use `!` and
re-download it.
2023-02-09 22:31:01 -08:00
Ulyssa
c9c547acc1 Support sending and displaying message reactions (#2) 2023-02-09 17:53:33 -08:00
Ulyssa
3629f15e0d Fix newer Clippy warnings for 1.67.0 (#33) 2023-01-30 13:51:32 -08:00
Ulyssa
fd72cf5c4e Update CI workflow to reduce warnings (#32) 2023-01-30 13:24:35 -08:00
Benjamin Große
1d93461183 Add :open attachments command (#31)
Fixes #27
2023-01-30 13:14:11 -08:00
Ulyssa
a1574c6b8d Show current date and local time for messages (#30) 2023-01-29 18:07:00 -08:00
Ulyssa
e8205df21d Support bracketed paste (#28) 2023-01-28 18:01:17 -08:00
28 changed files with 5529 additions and 1330 deletions

1
.github/FUNDING.yml vendored Normal file
View File

@@ -0,0 +1 @@
ko_fi: ulyssam

View File

@@ -9,25 +9,6 @@ on:
name: CI name: CI
jobs: jobs:
clippy_check:
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v1
with:
submodules: true
- name: Install Rust
uses: actions-rs/toolchain@v1
with:
toolchain: stable
components: clippy
override: true
- name: Check Clippy
uses: actions-rs/clippy-check@v1
with:
token: ${{ secrets.GITHUB_TOKEN }}
toolchain: stable
args:
test: test:
strategy: strategy:
matrix: matrix:
@@ -35,26 +16,41 @@ jobs:
runs-on: ${{ matrix.platform }} runs-on: ${{ matrix.platform }}
steps: steps:
- name: Checkout code - name: Checkout code
uses: actions/checkout@v2 uses: actions/checkout@v3
with: with:
submodules: true submodules: true
- name: Install Rust - name: Install Rust (1.66 w/ clippy)
uses: actions-rs/toolchain@v1 uses: dtolnay/rust-toolchain@1.66
with: with:
toolchain: nightly components: clippy
override: true - name: Install Rust (nightly w/ rustfmt)
components: rustfmt, clippy run: rustup toolchain install nightly --component rustfmt
- name: Cache cargo registry - name: Cache cargo registry
uses: actions/cache@v1 uses: actions/cache@v3
with: with:
path: ~/.cargo/registry path: ~/.cargo/registry
key: ${{ runner.os }}-cargo-registry-${{ hashFiles('**/Cargo.lock') }} key: ${{ runner.os }}-cargo-registry-${{ hashFiles('**/Cargo.lock') }}
- name: Cache cargo build
uses: actions/cache@v3
with:
path: target
key: ${{ runner.os }}-cargo-build-${{ hashFiles('**/Cargo.lock') }}
- name: Check formatting - name: Check formatting
uses: actions-rs/cargo@v1 run: cargo +nightly fmt --all -- --check
- name: Check Clippy
if: matrix.platform == 'ubuntu-latest'
uses: giraffate/clippy-action@v1
with: with:
command: fmt github_token: ${{ secrets.GITHUB_TOKEN }}
args: --all -- --check reporter: 'github-check'
- name: Run tests - name: Run tests
uses: actions-rs/cargo@v1 run: cargo test
- name: Build artifacts
run: cargo build --release
- name: Upload artifacts
uses: actions/upload-artifact@master
with: with:
command: test name: iamb-${{ matrix.platform }}
path: |
./target/release/iamb
./target/release/iamb.exe

2
.gitignore vendored
View File

@@ -1,2 +1,4 @@
/target /target
/result
/TODO /TODO
/docs/iamb.[15]

2052
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@@ -1,6 +1,6 @@
[package] [package]
name = "iamb" name = "iamb"
version = "0.0.4" version = "0.0.8"
edition = "2018" edition = "2018"
authors = ["Ulyssa <git@ulyssa.dev>"] authors = ["Ulyssa <git@ulyssa.dev>"]
repository = "https://github.com/ulyssa/iamb" repository = "https://github.com/ulyssa/iamb"
@@ -12,17 +12,34 @@ exclude = [".github", "CONTRIBUTING.md"]
keywords = ["matrix", "chat", "tui", "vim"] keywords = ["matrix", "chat", "tui", "vim"]
categories = ["command-line-utilities"] categories = ["command-line-utilities"]
rust-version = "1.66" rust-version = "1.66"
build = "build.rs"
[build-dependencies]
mandown = "0.1.3"
[build-dependencies.vergen]
version = "8"
default-features = false
features = ["build", "git", "gitcl",]
[dependencies] [dependencies]
arboard = "3.2.0"
bitflags = "1.3.2"
chrono = "0.4" chrono = "0.4"
clap = {version = "4.0", features = ["derive"]} clap = {version = "4.0", features = ["derive"]}
comrak = {version = "0.18.0", features = ["shortcodes"]}
css-color-parser = "0.1.2" css-color-parser = "0.1.2"
dirs = "4.0.0" dirs = "4.0.0"
emojis = "~0.5.2"
futures = "0.3"
gethostname = "0.4.1" gethostname = "0.4.1"
html5ever = "0.26.0" html5ever = "0.26.0"
image = "0.24.5"
libc = "0.2"
markup5ever_rcdom = "0.2.0" markup5ever_rcdom = "0.2.0"
mime = "^0.3.16" mime = "^0.3.16"
mime_guess = "^2.0.4" mime_guess = "^2.0.4"
open = "3.2.0"
regex = "^1.5" regex = "^1.5"
rpassword = "^7.2" rpassword = "^7.2"
serde = "^1.0" serde = "^1.0"
@@ -36,12 +53,12 @@ unicode-width = "0.1.10"
url = {version = "^2.2.2", features = ["serde"]} url = {version = "^2.2.2", features = ["serde"]}
[dependencies.modalkit] [dependencies.modalkit]
version = "0.0.10" version = "0.0.16"
[dependencies.matrix-sdk] [dependencies.matrix-sdk]
version = "0.6" version = "0.6"
default-features = false default-features = false
features = ["e2e-encryption", "markdown", "sled", "rustls-tls"] features = ["e2e-encryption", "sled", "rustls-tls"]
[dependencies.tokio] [dependencies.tokio]
version = "1.24.1" version = "1.24.1"
@@ -49,3 +66,8 @@ features = ["macros", "net", "rt-multi-thread", "sync", "time"]
[dev-dependencies] [dev-dependencies]
lazy_static = "1.4.0" lazy_static = "1.4.0"
pretty_assertions = "1.4.0"
[profile.release]
lto = true
incremental = false

View File

@@ -2,6 +2,7 @@
[![Build Status](https://github.com/ulyssa/iamb/workflows/CI/badge.svg)](https://github.com/ulyssa/iamb/actions?query=workflow%3ACI+) [![Build Status](https://github.com/ulyssa/iamb/workflows/CI/badge.svg)](https://github.com/ulyssa/iamb/actions?query=workflow%3ACI+)
[![License: Apache 2.0](https://img.shields.io/crates/l/iamb.svg?logo=apache)](https://crates.io/crates/iamb) [![License: Apache 2.0](https://img.shields.io/crates/l/iamb.svg?logo=apache)](https://crates.io/crates/iamb)
[![#iamb:0x.badd.cafe](https://img.shields.io/badge/matrix-%23iamb:0x.badd.cafe-blue)](https://matrix.to/#/#iamb:0x.badd.cafe)
[![Latest Version](https://img.shields.io/crates/v/iamb.svg?logo=rust)](https://crates.io/crates/iamb) [![Latest Version](https://img.shields.io/crates/v/iamb.svg?logo=rust)](https://crates.io/crates/iamb)
## About ## About
@@ -11,6 +12,8 @@
This project is a work-in-progress, and there's still a lot to be implemented, This project is a work-in-progress, and there's still a lot to be implemented,
but much of the basic client functionality is already present. but much of the basic client functionality is already present.
![Example Usage](https://iamb.chat/static/images/iamb-demo.gif)
## Documentation ## Documentation
You can find documentation for installing, configuring, and using iamb on its You can find documentation for installing, configuring, and using iamb on its
@@ -18,12 +21,35 @@ website, [iamb.chat].
## Installation ## Installation
Install Rust and Cargo, and then run: Install Rust (1.66.0 or above) and Cargo, and then run:
``` ```
cargo install --locked iamb cargo install --locked iamb
``` ```
### NetBSD
On NetBSD a package is available from the official repositories. To install it simply run:
```
pkgin install iamb
```
### Arch Linux
On Arch Linux a [package](https://aur.archlinux.org/packages/iamb-git) is available in the
Arch User Repositories (AUR). To install it simply run with your favorite AUR helper:
```
paru iamb-git
```
### Nix / NixOS (flake)
```
nix profile install "github:ulyssa/iamb"
```
## Configuration ## Configuration
You can create a basic configuration in `$CONFIG_DIR/iamb/config.json` that looks like: You can create a basic configuration in `$CONFIG_DIR/iamb/config.json` that looks like:
@@ -73,9 +99,9 @@ two other TUI clients and Element Web:
| Multiple Matrix Accounts | ✔️ | ❌ | ✔️ | ❌ | | Multiple Matrix Accounts | ✔️ | ❌ | ✔️ | ❌ |
| New user registration | ❌ | ❌ | ❌ | ✔️ | | New user registration | ❌ | ❌ | ❌ | ✔️ |
| VOIP | ❌ | ❌ | ❌ | ✔️ | | VOIP | ❌ | ❌ | ❌ | ✔️ |
| Reactions | ❌ ([#2]) | ✔️ | ❌ | ✔️ | | Reactions | ✔️ | ✔️ | ❌ | ✔️ |
| Message editing | ✔️ | ✔️ | ❌ | ✔️ | | Message editing | ✔️ | ✔️ | ❌ | ✔️ |
| Room upgrades | ❌ | ✔️ | ❌ | ✔️ | | Room upgrades | ❌ ([#41]) | ✔️ | ❌ | ✔️ |
| Localisations | ❌ | 1 | ❌ | 44 | | Localisations | ❌ | 1 | ❌ | 44 |
| SSO Support | ❌ | ✔️ | ✔️ | ✔️ | | SSO Support | ❌ | ✔️ | ✔️ | ✔️ |
@@ -88,18 +114,7 @@ iamb is released under the [Apache License, Version 2.0].
[iamb.chat]: https://iamb.chat [iamb.chat]: https://iamb.chat
[gomuks]: https://github.com/tulir/gomuks [gomuks]: https://github.com/tulir/gomuks
[weechat-matrix]: https://github.com/poljar/weechat-matrix [weechat-matrix]: https://github.com/poljar/weechat-matrix
[#2]: https://github.com/ulyssa/iamb/issues/2
[#3]: https://github.com/ulyssa/iamb/issues/3
[#4]: https://github.com/ulyssa/iamb/issues/4
[#5]: https://github.com/ulyssa/iamb/issues/5
[#6]: https://github.com/ulyssa/iamb/issues/6
[#7]: https://github.com/ulyssa/iamb/issues/7
[#8]: https://github.com/ulyssa/iamb/issues/8 [#8]: https://github.com/ulyssa/iamb/issues/8
[#9]: https://github.com/ulyssa/iamb/issues/9
[#10]: https://github.com/ulyssa/iamb/issues/10
[#11]: https://github.com/ulyssa/iamb/issues/11
[#12]: https://github.com/ulyssa/iamb/issues/12
[#13]: https://github.com/ulyssa/iamb/issues/13
[#14]: https://github.com/ulyssa/iamb/issues/14 [#14]: https://github.com/ulyssa/iamb/issues/14
[#15]: https://github.com/ulyssa/iamb/issues/15
[#16]: https://github.com/ulyssa/iamb/issues/16 [#16]: https://github.com/ulyssa/iamb/issues/16
[#41]: https://github.com/ulyssa/iamb/issues/41

29
build.rs Normal file
View File

@@ -0,0 +1,29 @@
use std::error::Error;
use std::fs;
use std::iter::FromIterator;
use std::path::PathBuf;
use mandown::convert;
use vergen::EmitBuilder;
const IAMB_1_MD: &str = include_str!("docs/iamb.1.md");
const IAMB_5_MD: &str = include_str!("docs/iamb.5.md");
fn main() -> Result<(), Box<dyn Error>> {
EmitBuilder::builder().git_sha(true).emit()?;
// Build the manual pages.
println!("cargo:rerun-if-changed=docs/iamb.1.md");
println!("cargo:rerun-if-changed=docs/iamb.5.md");
let iamb_1 = convert(IAMB_1_MD, "IAMB", 1);
let iamb_5 = convert(IAMB_5_MD, "IAMB", 5);
let out_dir = std::env::var("OUT_DIR");
let out_dir = out_dir.as_deref().unwrap_or("docs");
fs::write(PathBuf::from_iter([out_dir, "iamb.1"]), iamb_1.as_bytes())?;
fs::write(PathBuf::from_iter([out_dir, "iamb.5"]), iamb_5.as_bytes())?;
Ok(())
}

32
docs/example_config.json Normal file
View File

@@ -0,0 +1,32 @@
{
"default_profile": "default",
"profiles": {
"default": {
"user_id": "",
"url": "https://matrix.org",
"settings": {},
"dirs": {}
}
},
"settings": {
"log_level": "warn",
"reaction_display": true,
"reaction_shortcode_display": false,
"read_receipt_send": true,
"read_receipt_display": true,
"request_timeout": 10000,
"typing_notice_send": true,
"typing_notice_display": true,
"users": {
"@user:matrix.org": {
"name": "John Doe",
"color": "magenta"
}
},
"default_room": "#iamb-users:0x.badd.cafe"
},
"dirs": {
"cache": "~/.cache/iamb/",
"logs": "~/.local/share/iamb/logs/",
"downloads": "~/Downloads/"
}

29
docs/iamb.1.md Normal file
View File

@@ -0,0 +1,29 @@
# NAME
iamb a terminal-based client for Matrix for the Vim addict
# SYNOPSIS
**iamb** [**--profile** _profile_] [**--config-directory** _directory_] [**--help** | **--version**]
# OPTIONS
These options are primitives at the top-level of the file.
**--profile**, **-P**
> The profile to start with. Overrides **default_profile** from **iamb(5)**.
**--config-directory**, **-C**
> Path to the directory the configuration file is located in.
**--help**, **-h**
> Show a short help text and quit.
**--version**, **-V**
> Show the iamb version and quit.
# SEE ALSO
**iamb(5)**
Full documentation is available online at \<https://iamb.chat\>

134
docs/iamb.5.md Normal file
View File

@@ -0,0 +1,134 @@
# NAME
config.json configuration file for iamb
# SYNOPSIS
Configuration must be placed under _~/.config/iamb/_ and is named config.json.
Example configuration usually comes bundled with your installation and can
typically be found in _/usr/share/iamb_.
As implied by the filename, the configuration is formatted in JSON. It's
structure and fields are described below.
# BASIC SETTINGS
These options are primitives at the top-level of the file.
**default_profile** (type: string)
> The default profile to connect to, unless overwritten by a commandline
> switch. It has to be defined in the *PROFILES* section.
# PROFILES
These options are configured as a map under the profiles name.
**user_id** (type: string)
> The user ID to use when connecting to the server. For example "user" for
> "@user:matrix.org".
**url** (type: string)
> The URL of the users server. For example "https://matrix.org" for
> "@user:matrix.org".
**settings** (type: settings object)
> Overwrite general settings for this account. The fields are identical to
> those in *TUNABLES*.
**layout** (type: startup layout object)
> Overwrite general settings for this account. The fields are identical to
> those in *STARTUP LAYOUT*.
**dirs** (type: XDG overrides object)
> Overwrite general settings for this account. The fields are identical to
> those in *DIRECTORIES*.
# TUNABLES
These options are configured as a map under the *settings* key and can be
overridden as described in *PROFILES*.
**log_level** (type: string)
> Specifies the lowest log level that should be shown. Possible values
> are: _trace_, _debug_, _info_, _warn_, and _error_.
**reaction_display** (type: boolean)
> Defines whether or not reactions should be shown.
**reaction_shortcode_display** (type: boolean)
> Defines whether or not reactions should be shown as their respective
> shortcode.
**read_receipt_send** (type: boolean)
> Defines whether or not read confirmations are sent.
**read_receipt_display** (type: boolean)
> Defines whether or not read confirmations are displayed.
**request_timeout** (type: uint64)
> Defines the maximum time per request in seconds.
**typing_notice_send** (type: boolean)
> Defines whether or not the typing state is sent.
**typing_notice_display** (type: boolean)
> Defines whether or not the typing state is displayed.
**user** (type: map)
> Overrides values for the specified user. See *USER OVERRIDES* for
> details on the format.
**default_room** (type: string)
> The room to show by default instead of a welcome-screen.
## USER OVERRIDES
Overrides are mapped onto matrix User IDs such as _@user:matrix.org_ and are
maps containing the following key value pairs.
**name** (type: string)
> Change the display name of the user.
**color** (type: string)
> Change the color the user is shown as. Possible values are: _black_,
> _blue_, _cyan_, _dark-gray_, _gray_, _green_, _light-blue_,
> _light-cyan_, _light-green_, _light-magenta_, _light-red_,
> _light-yellow_, _magenta_, _none_, _red_, _white_, _yellow_
# STARTUP LAYOUT
Specifies what initial set of tabs and windows to show when starting the
client. Configured as an object under the key *layout*.
**style** (type: string)
> Specifies what window layout to load when starting. Valid values are
> _restore_ to restore the layout from the last time the client was exited,
> _new_ to open a single window (uses the value of _default\_room_ if set), or
> _config_ to open the layout described under _tabs_.
**tabs** (type: array of window objects)
> If **style** is set to _config_, then this value will be used to open a set
> of tabs and windows at startup. Each object can contain either a **window**
> key specifying a username, room identifier or room alias to show, or a
> **split** key specifying an array of window objects.
# DIRECTORIES
Specifies the directories to save data in. Configured as a map under the key
*dirs*.
**cache** (type: string)
> Specifies where to store assets and temporary data in.
**logs** (type: string)
> Specifies where to store log files.
**downloads** (type: string)
> Specifies where to store downloaded files.
# SEE ALSO
*iamb(1)*
Full documentation is available online at \<https://iamb.chat\>

94
flake.lock generated Normal file
View File

@@ -0,0 +1,94 @@
{
"nodes": {
"flake-utils": {
"locked": {
"lastModified": 1678901627,
"narHash": "sha256-U02riOqrKKzwjsxc/400XnElV+UtPUQWpANPlyazjH0=",
"owner": "numtide",
"repo": "flake-utils",
"rev": "93a2b84fc4b70d9e089d029deacc3583435c2ed6",
"type": "github"
},
"original": {
"owner": "numtide",
"repo": "flake-utils",
"type": "github"
}
},
"flake-utils_2": {
"locked": {
"lastModified": 1659877975,
"narHash": "sha256-zllb8aq3YO3h8B/U0/J1WBgAL8EX5yWf5pMj3G0NAmc=",
"owner": "numtide",
"repo": "flake-utils",
"rev": "c0e246b9b83f637f4681389ecabcb2681b4f3af0",
"type": "github"
},
"original": {
"owner": "numtide",
"repo": "flake-utils",
"type": "github"
}
},
"nixpkgs": {
"locked": {
"lastModified": 1679437018,
"narHash": "sha256-vOuiDPLHSEo/7NkiWtxpHpHgoXoNmrm+wkXZ6a072Fc=",
"owner": "nixos",
"repo": "nixpkgs",
"rev": "19cf008bb18e47b6e3b4e16e32a9a4bdd4b45f7e",
"type": "github"
},
"original": {
"owner": "nixos",
"ref": "nixos-unstable",
"repo": "nixpkgs",
"type": "github"
}
},
"nixpkgs_2": {
"locked": {
"lastModified": 1665296151,
"narHash": "sha256-uOB0oxqxN9K7XGF1hcnY+PQnlQJ+3bP2vCn/+Ru/bbc=",
"owner": "NixOS",
"repo": "nixpkgs",
"rev": "14ccaaedd95a488dd7ae142757884d8e125b3363",
"type": "github"
},
"original": {
"owner": "NixOS",
"ref": "nixpkgs-unstable",
"repo": "nixpkgs",
"type": "github"
}
},
"root": {
"inputs": {
"flake-utils": "flake-utils",
"nixpkgs": "nixpkgs",
"rust-overlay": "rust-overlay"
}
},
"rust-overlay": {
"inputs": {
"flake-utils": "flake-utils_2",
"nixpkgs": "nixpkgs_2"
},
"locked": {
"lastModified": 1679624450,
"narHash": "sha256-wiDqUaklmc31E1+wz5sv52sMcWvZKsL1FBeGJCxz628=",
"owner": "oxalica",
"repo": "rust-overlay",
"rev": "afbdcf305fd6f05f708fe76d52f24d37d066c251",
"type": "github"
},
"original": {
"owner": "oxalica",
"repo": "rust-overlay",
"type": "github"
}
}
},
"root": "root",
"version": 7
}

40
flake.nix Normal file
View File

@@ -0,0 +1,40 @@
{
description = "iamb";
nixConfig.bash-prompt = "\[nix-develop\]$ ";
inputs = {
nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
flake-utils.url = "github:numtide/flake-utils";
rust-overlay.url = "github:oxalica/rust-overlay";
};
outputs = { self, nixpkgs, flake-utils, rust-overlay, ... }:
flake-utils.lib.eachDefaultSystem (system:
let
# We only need the nightly overlay in the devShell because .rs files are formatted with nightly.
overlays = [ (import rust-overlay) ];
pkgs = import nixpkgs { inherit system overlays; };
rustNightly = pkgs.rust-bin.nightly."2023-03-17".default;
in
with pkgs;
{
packages.default = pkgs.rustPlatform.buildRustPackage {
pname = "iamb";
version = "0.0.7";
src = ./.;
cargoLock.lockFile = ./Cargo.lock;
nativeBuildInputs = [ pkgs.pkgconfig ];
buildInputs = [ pkgs.openssl ] ++ pkgs.lib.optionals pkgs.stdenv.isDarwin
(with pkgs.darwin.apple_sdk.frameworks; [ AppKit Security ]);
};
devShell = mkShell {
buildInputs = [
(rustNightly.override { extensions = [ "rust-src" ]; })
pkg-config
cargo-tarpaulin
rust-analyzer
rustfmt
];
};
});
}

File diff suppressed because it is too large Load Diff

View File

@@ -4,12 +4,16 @@ use matrix_sdk::ruma::{events::tag::TagName, OwnedUserId};
use modalkit::{ use modalkit::{
editing::base::OpenTarget, editing::base::OpenTarget,
env::vim::command::{CommandContext, CommandDescription}, env::vim::command::{CommandContext, CommandDescription, OptionType},
input::commands::{CommandError, CommandResult, CommandStep}, input::commands::{CommandError, CommandResult, CommandStep},
input::InputContext, input::InputContext,
}; };
use crate::base::{ use crate::base::{
CreateRoomFlags,
CreateRoomType,
DownloadFlags,
HomeserverAction,
IambAction, IambAction,
IambId, IambId,
MessageAction, MessageAction,
@@ -39,7 +43,7 @@ fn tag_name(name: String) -> Result<TagName, CommandError> {
if let Ok(tag) = name.parse() { if let Ok(tag) = name.parse() {
TagName::User(tag) TagName::User(tag)
} else { } else {
let msg = format!("Invalid user tag name: {}", name); let msg = format!("Invalid user tag name: {name}");
return Err(CommandError::Error(msg)); return Err(CommandError::Error(msg));
} }
@@ -157,13 +161,24 @@ fn iamb_members(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
return Ok(step); return Ok(step);
} }
fn iamb_leave(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
if !desc.arg.text.is_empty() {
return Result::Err(CommandError::InvalidArgument);
}
let leave = IambAction::Room(RoomAction::Leave(desc.bang));
let step = CommandStep::Continue(leave.into(), ctx.context.take());
return Ok(step);
}
fn iamb_cancel(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult { fn iamb_cancel(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
if !desc.arg.text.is_empty() { if !desc.arg.text.is_empty() {
return Result::Err(CommandError::InvalidArgument); return Result::Err(CommandError::InvalidArgument);
} }
let ract = IambAction::from(MessageAction::Cancel); let mact = IambAction::from(MessageAction::Cancel(desc.bang));
let step = CommandStep::Continue(ract.into(), ctx.context.take()); let step = CommandStep::Continue(mact.into(), ctx.context.take());
return Ok(step); return Ok(step);
} }
@@ -173,8 +188,55 @@ fn iamb_edit(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
return Result::Err(CommandError::InvalidArgument); return Result::Err(CommandError::InvalidArgument);
} }
let ract = IambAction::from(MessageAction::Edit); let mact = IambAction::from(MessageAction::Edit);
let step = CommandStep::Continue(ract.into(), ctx.context.take()); let step = CommandStep::Continue(mact.into(), ctx.context.take());
return Ok(step);
}
fn iamb_react(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
let args = desc.arg.strings()?;
if args.len() != 1 {
return Result::Err(CommandError::InvalidArgument);
}
let k = args[0].as_str();
if let Some(emoji) = emojis::get(k).or_else(|| emojis::get_by_shortcode(k)) {
let mact = IambAction::from(MessageAction::React(emoji.to_string()));
let step = CommandStep::Continue(mact.into(), ctx.context.take());
return Ok(step);
} else {
let msg = format!("Invalid Emoji or shortcode: {k}");
return Result::Err(CommandError::Error(msg));
}
}
fn iamb_unreact(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
let mut args = desc.arg.strings()?;
if args.len() > 1 {
return Result::Err(CommandError::InvalidArgument);
}
let mact = if let Some(k) = args.pop() {
let k = k.as_str();
if let Some(emoji) = emojis::get(k).or_else(|| emojis::get_by_shortcode(k)) {
IambAction::from(MessageAction::Unreact(Some(emoji.to_string())))
} else {
let msg = format!("Invalid Emoji or shortcode: {k}");
return Result::Err(CommandError::Error(msg));
}
} else {
IambAction::from(MessageAction::Unreact(None))
};
let step = CommandStep::Continue(mact.into(), ctx.context.take());
return Ok(step); return Ok(step);
} }
@@ -186,7 +248,8 @@ fn iamb_redact(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
return Result::Err(CommandError::InvalidArgument); return Result::Err(CommandError::InvalidArgument);
} }
let ract = IambAction::from(MessageAction::Redact(args.into_iter().next())); let reason = args.into_iter().next();
let ract = IambAction::from(MessageAction::Redact(reason, desc.bang));
let step = CommandStep::Continue(ract.into(), ctx.context.take()); let step = CommandStep::Continue(ract.into(), ctx.context.take());
return Ok(step); return Ok(step);
@@ -249,6 +312,53 @@ fn iamb_join(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
return Ok(step); return Ok(step);
} }
fn iamb_create(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
let args = desc.arg.options()?;
let mut flags = CreateRoomFlags::NONE;
let mut alias = None;
let mut ct = CreateRoomType::Room;
for arg in args {
match arg {
OptionType::Flag(name, Some(arg)) => {
match name.as_str() {
"alias" => {
if alias.is_some() {
let msg = "Multiple ++alias arguments are not allowed";
let err = CommandError::Error(msg.into());
return Err(err);
} else {
alias = Some(arg);
}
},
_ => return Err(CommandError::InvalidArgument),
}
},
OptionType::Flag(name, None) => {
match name.as_str() {
"public" => flags |= CreateRoomFlags::PUBLIC,
"space" => ct = CreateRoomType::Space,
"enc" | "encrypted" => flags |= CreateRoomFlags::ENCRYPTED,
_ => return Err(CommandError::InvalidArgument),
}
},
OptionType::Positional(_) => {
let msg = ":create doesn't take any positional arguments";
let err = CommandError::Error(msg.into());
return Err(err);
},
}
}
let hact = HomeserverAction::CreateRoom(alias, ct, flags);
let iact = IambAction::from(hact);
let step = CommandStep::Continue(iact.into(), ctx.context.take());
return Ok(step);
}
fn iamb_room(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult { fn iamb_room(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
let mut args = desc.arg.strings()?; let mut args = desc.arg.strings()?;
@@ -317,7 +427,29 @@ fn iamb_download(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult
return Result::Err(CommandError::InvalidArgument); return Result::Err(CommandError::InvalidArgument);
} }
let mact = MessageAction::Download(args.pop(), desc.bang); let mut flags = DownloadFlags::NONE;
if desc.bang {
flags |= DownloadFlags::FORCE;
};
let mact = MessageAction::Download(args.pop(), flags);
let iact = IambAction::from(mact);
let step = CommandStep::Continue(iact.into(), ctx.context.take());
return Ok(step);
}
fn iamb_open(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult {
let mut args = desc.arg.strings()?;
if args.len() > 1 {
return Result::Err(CommandError::InvalidArgument);
}
let mut flags = DownloadFlags::OPEN;
if desc.bang {
flags |= DownloadFlags::FORCE;
};
let mact = MessageAction::Download(args.pop(), flags);
let iact = IambAction::from(mact); let iact = IambAction::from(mact);
let step = CommandStep::Continue(iact.into(), ctx.context.take()); let step = CommandStep::Continue(iact.into(), ctx.context.take());
@@ -325,21 +457,86 @@ fn iamb_download(desc: CommandDescription, ctx: &mut ProgContext) -> ProgResult
} }
fn add_iamb_commands(cmds: &mut ProgramCommands) { fn add_iamb_commands(cmds: &mut ProgramCommands) {
cmds.add_command(ProgramCommand { names: vec!["cancel".into()], f: iamb_cancel }); cmds.add_command(ProgramCommand {
cmds.add_command(ProgramCommand { names: vec!["dms".into()], f: iamb_dms }); name: "cancel".into(),
cmds.add_command(ProgramCommand { names: vec!["download".into()], f: iamb_download }); aliases: vec![],
cmds.add_command(ProgramCommand { names: vec!["edit".into()], f: iamb_edit }); f: iamb_cancel,
cmds.add_command(ProgramCommand { names: vec!["invite".into()], f: iamb_invite }); });
cmds.add_command(ProgramCommand { names: vec!["join".into()], f: iamb_join }); cmds.add_command(ProgramCommand {
cmds.add_command(ProgramCommand { names: vec!["members".into()], f: iamb_members }); name: "create".into(),
cmds.add_command(ProgramCommand { names: vec!["redact".into()], f: iamb_redact }); aliases: vec![],
cmds.add_command(ProgramCommand { names: vec!["reply".into()], f: iamb_reply }); f: iamb_create,
cmds.add_command(ProgramCommand { names: vec!["rooms".into()], f: iamb_rooms }); });
cmds.add_command(ProgramCommand { names: vec!["room".into()], f: iamb_room }); cmds.add_command(ProgramCommand { name: "dms".into(), aliases: vec![], f: iamb_dms });
cmds.add_command(ProgramCommand { names: vec!["spaces".into()], f: iamb_spaces }); cmds.add_command(ProgramCommand {
cmds.add_command(ProgramCommand { names: vec!["upload".into()], f: iamb_upload }); name: "download".into(),
cmds.add_command(ProgramCommand { names: vec!["verify".into()], f: iamb_verify }); aliases: vec![],
cmds.add_command(ProgramCommand { names: vec!["welcome".into()], f: iamb_welcome }); f: iamb_download,
});
cmds.add_command(ProgramCommand { name: "open".into(), aliases: vec![], f: iamb_open });
cmds.add_command(ProgramCommand { name: "edit".into(), aliases: vec![], f: iamb_edit });
cmds.add_command(ProgramCommand {
name: "invite".into(),
aliases: vec![],
f: iamb_invite,
});
cmds.add_command(ProgramCommand { name: "join".into(), aliases: vec![], f: iamb_join });
cmds.add_command(ProgramCommand {
name: "leave".into(),
aliases: vec![],
f: iamb_leave,
});
cmds.add_command(ProgramCommand {
name: "members".into(),
aliases: vec![],
f: iamb_members,
});
cmds.add_command(ProgramCommand {
name: "react".into(),
aliases: vec![],
f: iamb_react,
});
cmds.add_command(ProgramCommand {
name: "redact".into(),
aliases: vec![],
f: iamb_redact,
});
cmds.add_command(ProgramCommand {
name: "reply".into(),
aliases: vec![],
f: iamb_reply,
});
cmds.add_command(ProgramCommand {
name: "rooms".into(),
aliases: vec![],
f: iamb_rooms,
});
cmds.add_command(ProgramCommand { name: "room".into(), aliases: vec![], f: iamb_room });
cmds.add_command(ProgramCommand {
name: "spaces".into(),
aliases: vec![],
f: iamb_spaces,
});
cmds.add_command(ProgramCommand {
name: "unreact".into(),
aliases: vec![],
f: iamb_unreact,
});
cmds.add_command(ProgramCommand {
name: "upload".into(),
aliases: vec![],
f: iamb_upload,
});
cmds.add_command(ProgramCommand {
name: "verify".into(),
aliases: vec![],
f: iamb_verify,
});
cmds.add_command(ProgramCommand {
name: "welcome".into(),
aliases: vec![],
f: iamb_welcome,
});
} }
pub fn setup_commands() -> ProgramCommands { pub fn setup_commands() -> ProgramCommands {
@@ -690,15 +887,19 @@ mod tests {
let ctx = ProgramContext::default(); let ctx = ProgramContext::default();
let res = cmds.input_cmd("redact", ctx.clone()).unwrap(); let res = cmds.input_cmd("redact", ctx.clone()).unwrap();
let act = IambAction::Message(MessageAction::Redact(None)); let act = IambAction::Message(MessageAction::Redact(None, false));
assert_eq!(res, vec![(act.into(), ctx.clone())]);
let res = cmds.input_cmd("redact!", ctx.clone()).unwrap();
let act = IambAction::Message(MessageAction::Redact(None, true));
assert_eq!(res, vec![(act.into(), ctx.clone())]); assert_eq!(res, vec![(act.into(), ctx.clone())]);
let res = cmds.input_cmd("redact Removed", ctx.clone()).unwrap(); let res = cmds.input_cmd("redact Removed", ctx.clone()).unwrap();
let act = IambAction::Message(MessageAction::Redact(Some("Removed".into()))); let act = IambAction::Message(MessageAction::Redact(Some("Removed".into()), false));
assert_eq!(res, vec![(act.into(), ctx.clone())]); assert_eq!(res, vec![(act.into(), ctx.clone())]);
let res = cmds.input_cmd("redact \"Removed\"", ctx.clone()).unwrap(); let res = cmds.input_cmd("redact \"Removed\"", ctx.clone()).unwrap();
let act = IambAction::Message(MessageAction::Redact(Some("Removed".into()))); let act = IambAction::Message(MessageAction::Redact(Some("Removed".into()), false));
assert_eq!(res, vec![(act.into(), ctx.clone())]); assert_eq!(res, vec![(act.into(), ctx.clone())]);
let res = cmds.input_cmd("redact Removed Removed", ctx.clone()); let res = cmds.input_cmd("redact Removed Removed", ctx.clone());

View File

@@ -9,8 +9,9 @@ use std::path::{Path, PathBuf};
use std::process; use std::process;
use clap::Parser; use clap::Parser;
use matrix_sdk::ruma::{OwnedUserId, UserId}; use matrix_sdk::ruma::{OwnedRoomAliasId, OwnedRoomId, OwnedUserId, UserId};
use serde::{de::Error as SerdeError, de::Visitor, Deserialize, Deserializer}; use serde::{de::Error as SerdeError, de::Visitor, Deserialize, Deserializer};
use tracing::Level;
use url::Url; use url::Url;
use modalkit::tui::{ use modalkit::tui::{
@@ -18,6 +19,8 @@ use modalkit::tui::{
text::Span, text::Span,
}; };
use super::base::{IambId, RoomInfo};
macro_rules! usage { macro_rules! usage {
( $($args: tt)* ) => { ( $($args: tt)* ) => {
println!($($args)*); println!($($args)*);
@@ -25,6 +28,8 @@ macro_rules! usage {
} }
} }
const DEFAULT_REQ_TIMEOUT: u64 = 120;
const COLORS: [Color; 13] = [ const COLORS: [Color; 13] = [
Color::Blue, Color::Blue,
Color::Cyan, Color::Cyan,
@@ -86,8 +91,13 @@ fn validate_profile_names(names: &HashMap<String, ProfileConfig>) {
} }
} }
const VERSION: &str = match option_env!("VERGEN_GIT_SHA") {
None => env!("CARGO_PKG_VERSION"),
Some(_) => concat!(env!("CARGO_PKG_VERSION"), " (", env!("VERGEN_GIT_SHA"), ")"),
};
#[derive(Parser)] #[derive(Parser)]
#[clap(version, about, long_about = None)] #[clap(version = VERSION, about, long_about = None)]
#[clap(propagate_version = true)] #[clap(propagate_version = true)]
pub struct Iamb { pub struct Iamb {
#[clap(short = 'P', long, value_parser)] #[clap(short = 'P', long, value_parser)]
@@ -106,6 +116,47 @@ pub enum ConfigError {
Invalid(#[from] serde_json::Error), Invalid(#[from] serde_json::Error),
} }
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct LogLevel(pub Level);
pub struct LogLevelVisitor;
impl From<LogLevel> for Level {
fn from(level: LogLevel) -> Level {
level.0
}
}
impl<'de> Visitor<'de> for LogLevelVisitor {
type Value = LogLevel;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
formatter.write_str("a valid log level (e.g. \"warn\" or \"debug\")")
}
fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
where
E: SerdeError,
{
match value {
"info" => Ok(LogLevel(Level::INFO)),
"debug" => Ok(LogLevel(Level::DEBUG)),
"warn" => Ok(LogLevel(Level::WARN)),
"error" => Ok(LogLevel(Level::ERROR)),
"trace" => Ok(LogLevel(Level::TRACE)),
_ => Err(E::custom("Could not parse log level")),
}
}
}
impl<'de> Deserialize<'de> for LogLevel {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_str(LogLevelVisitor)
}
}
#[derive(Clone, Debug, Eq, PartialEq)] #[derive(Clone, Debug, Eq, PartialEq)]
pub struct UserColor(pub Color); pub struct UserColor(pub Color);
pub struct UserColorVisitor; pub struct UserColorVisitor;
@@ -176,46 +227,90 @@ fn merge_users(a: Option<UserOverrides>, b: Option<UserOverrides>) -> Option<Use
} }
} }
#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq)]
#[serde(rename_all = "lowercase")]
pub enum UserDisplayStyle {
// The Matrix username for the sender (e.g., "@user:example.com").
#[default]
Username,
// The localpart of the Matrix username (e.g., "@user").
LocalPart,
// The display name for the Matrix user, calculated according to the rules from the spec.
//
// This is usually something like "Ada Lovelace" if the user has configured a display name, but
// it can wind up being the Matrix username if there are display name collisions in the room,
// in order to avoid any confusion.
DisplayName,
}
#[derive(Clone)] #[derive(Clone)]
pub struct TunableValues { pub struct TunableValues {
pub log_level: Level,
pub reaction_display: bool,
pub reaction_shortcode_display: bool,
pub read_receipt_send: bool, pub read_receipt_send: bool,
pub read_receipt_display: bool, pub read_receipt_display: bool,
pub request_timeout: u64,
pub typing_notice_send: bool, pub typing_notice_send: bool,
pub typing_notice_display: bool, pub typing_notice_display: bool,
pub users: UserOverrides, pub users: UserOverrides,
pub username_display: UserDisplayStyle,
pub default_room: Option<String>, pub default_room: Option<String>,
pub open_command: Option<Vec<String>>,
} }
#[derive(Clone, Default, Deserialize)] #[derive(Clone, Default, Deserialize)]
pub struct Tunables { pub struct Tunables {
pub log_level: Option<LogLevel>,
pub reaction_display: Option<bool>,
pub reaction_shortcode_display: Option<bool>,
pub read_receipt_send: Option<bool>, pub read_receipt_send: Option<bool>,
pub read_receipt_display: Option<bool>, pub read_receipt_display: Option<bool>,
pub request_timeout: Option<u64>,
pub typing_notice_send: Option<bool>, pub typing_notice_send: Option<bool>,
pub typing_notice_display: Option<bool>, pub typing_notice_display: Option<bool>,
pub users: Option<UserOverrides>, pub users: Option<UserOverrides>,
pub username_display: Option<UserDisplayStyle>,
pub default_room: Option<String>, pub default_room: Option<String>,
pub open_command: Option<Vec<String>>,
} }
impl Tunables { impl Tunables {
fn merge(self, other: Self) -> Self { fn merge(self, other: Self) -> Self {
Tunables { Tunables {
log_level: self.log_level.or(other.log_level),
reaction_display: self.reaction_display.or(other.reaction_display),
reaction_shortcode_display: self
.reaction_shortcode_display
.or(other.reaction_shortcode_display),
read_receipt_send: self.read_receipt_send.or(other.read_receipt_send), read_receipt_send: self.read_receipt_send.or(other.read_receipt_send),
read_receipt_display: self.read_receipt_display.or(other.read_receipt_display), read_receipt_display: self.read_receipt_display.or(other.read_receipt_display),
request_timeout: self.request_timeout.or(other.request_timeout),
typing_notice_send: self.typing_notice_send.or(other.typing_notice_send), typing_notice_send: self.typing_notice_send.or(other.typing_notice_send),
typing_notice_display: self.typing_notice_display.or(other.typing_notice_display), typing_notice_display: self.typing_notice_display.or(other.typing_notice_display),
users: merge_users(self.users, other.users), users: merge_users(self.users, other.users),
username_display: self.username_display.or(other.username_display),
default_room: self.default_room.or(other.default_room), default_room: self.default_room.or(other.default_room),
open_command: self.open_command.or(other.open_command),
} }
} }
fn values(self) -> TunableValues { fn values(self) -> TunableValues {
TunableValues { TunableValues {
log_level: self.log_level.map(Level::from).unwrap_or(Level::INFO),
reaction_display: self.reaction_display.unwrap_or(true),
reaction_shortcode_display: self.reaction_shortcode_display.unwrap_or(false),
read_receipt_send: self.read_receipt_send.unwrap_or(true), read_receipt_send: self.read_receipt_send.unwrap_or(true),
read_receipt_display: self.read_receipt_display.unwrap_or(true), read_receipt_display: self.read_receipt_display.unwrap_or(true),
request_timeout: self.request_timeout.unwrap_or(DEFAULT_REQ_TIMEOUT),
typing_notice_send: self.typing_notice_send.unwrap_or(true), typing_notice_send: self.typing_notice_send.unwrap_or(true),
typing_notice_display: self.typing_notice_display.unwrap_or(true), typing_notice_display: self.typing_notice_display.unwrap_or(true),
users: self.users.unwrap_or_default(), users: self.users.unwrap_or_default(),
username_display: self.username_display.unwrap_or_default(),
default_room: self.default_room, default_room: self.default_room,
open_command: self.open_command,
} }
} }
} }
@@ -224,7 +319,7 @@ impl Tunables {
pub struct DirectoryValues { pub struct DirectoryValues {
pub cache: PathBuf, pub cache: PathBuf,
pub logs: PathBuf, pub logs: PathBuf,
pub downloads: PathBuf, pub downloads: Option<PathBuf>,
} }
#[derive(Clone, Default, Deserialize)] #[derive(Clone, Default, Deserialize)]
@@ -259,21 +354,49 @@ impl Directories {
dir dir
}); });
let downloads = self let downloads = self.downloads.or_else(dirs::download_dir);
.downloads
.or_else(dirs::download_dir)
.expect("no dirs.download value configured!");
DirectoryValues { cache, logs, downloads } DirectoryValues { cache, logs, downloads }
} }
} }
#[derive(Clone, Debug, Deserialize, Eq, PartialEq)]
#[serde(untagged)]
pub enum WindowPath {
AliasId(OwnedRoomAliasId),
RoomId(OwnedRoomId),
UserId(OwnedUserId),
Window(IambId),
}
#[derive(Clone, Debug, Deserialize, Eq, PartialEq)]
#[serde(untagged, deny_unknown_fields)]
pub enum WindowLayout {
Window { window: WindowPath },
Split { split: Vec<WindowLayout> },
}
#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq)]
#[serde(rename_all = "lowercase", tag = "style")]
pub enum Layout {
/// Restore the layout from the previous session.
#[default]
Restore,
/// Open a single window using the `default_room` value.
New,
/// Open the window layouts described under `tabs`.
Config { tabs: Vec<WindowLayout> },
}
#[derive(Clone, Deserialize)] #[derive(Clone, Deserialize)]
pub struct ProfileConfig { pub struct ProfileConfig {
pub user_id: OwnedUserId, pub user_id: OwnedUserId,
pub url: Url, pub url: Url,
pub settings: Option<Tunables>, pub settings: Option<Tunables>,
pub dirs: Option<Directories>, pub dirs: Option<Directories>,
pub layout: Option<Layout>,
} }
#[derive(Clone, Deserialize)] #[derive(Clone, Deserialize)]
@@ -282,6 +405,7 @@ pub struct IambConfig {
pub default_profile: Option<String>, pub default_profile: Option<String>,
pub settings: Option<Tunables>, pub settings: Option<Tunables>,
pub dirs: Option<Directories>, pub dirs: Option<Directories>,
pub layout: Option<Layout>,
} }
impl IambConfig { impl IambConfig {
@@ -305,11 +429,13 @@ impl IambConfig {
#[derive(Clone)] #[derive(Clone)]
pub struct ApplicationSettings { pub struct ApplicationSettings {
pub matrix_dir: PathBuf, pub matrix_dir: PathBuf,
pub layout_json: PathBuf,
pub session_json: PathBuf, pub session_json: PathBuf,
pub profile_name: String, pub profile_name: String,
pub profile: ProfileConfig, pub profile: ProfileConfig,
pub tunables: TunableValues, pub tunables: TunableValues,
pub dirs: DirectoryValues, pub dirs: DirectoryValues,
pub layout: Layout,
} }
impl ApplicationSettings { impl ApplicationSettings {
@@ -330,6 +456,7 @@ impl ApplicationSettings {
default_profile, default_profile,
dirs, dirs,
settings: global, settings: global,
layout,
} = IambConfig::load(config_json.as_path())?; } = IambConfig::load(config_json.as_path())?;
validate_profile_names(&profiles); validate_profile_names(&profiles);
@@ -353,10 +480,17 @@ impl ApplicationSettings {
); );
}; };
let layout = profile.layout.take().or(layout).unwrap_or_default();
let tunables = global.unwrap_or_default(); let tunables = global.unwrap_or_default();
let tunables = profile.settings.take().unwrap_or_default().merge(tunables); let tunables = profile.settings.take().unwrap_or_default().merge(tunables);
let tunables = tunables.values(); let tunables = tunables.values();
let dirs = dirs.unwrap_or_default();
let dirs = profile.dirs.take().unwrap_or_default().merge(dirs);
let dirs = dirs.values();
// Set up paths that live inside the profile's data directory.
let mut profile_dir = config_dir.clone(); let mut profile_dir = config_dir.clone();
profile_dir.push("profiles"); profile_dir.push("profiles");
profile_dir.push(profile_name.as_str()); profile_dir.push(profile_name.as_str());
@@ -367,17 +501,23 @@ impl ApplicationSettings {
let mut session_json = profile_dir; let mut session_json = profile_dir;
session_json.push("session.json"); session_json.push("session.json");
let dirs = dirs.unwrap_or_default(); // Set up paths that live inside the profile's cache directory.
let dirs = profile.dirs.take().unwrap_or_default().merge(dirs); let mut cache_dir = dirs.cache.clone();
let dirs = dirs.values(); cache_dir.push("profiles");
cache_dir.push(profile_name.as_str());
let mut layout_json = cache_dir.clone();
layout_json.push("layout.json");
let settings = ApplicationSettings { let settings = ApplicationSettings {
matrix_dir, matrix_dir,
layout_json,
session_json, session_json,
profile_name, profile_name,
profile, profile,
tunables, tunables,
dirs, dirs,
layout,
}; };
Ok(settings) Ok(settings)
@@ -404,18 +544,45 @@ impl ApplicationSettings {
Span::styled(String::from(c), style) Span::styled(String::from(c), style)
} }
pub fn get_user_span<'a>(&self, user_id: &'a UserId) -> Span<'a> { pub fn get_user_overrides(
let (color, name) = self &self,
.tunables user_id: &UserId,
) -> (Option<Color>, Option<Cow<'static, str>>) {
self.tunables
.users .users
.get(user_id) .get(user_id)
.map(|user| (user.color.as_ref().map(|c| c.0), user.name.clone().map(Cow::Owned))) .map(|user| (user.color.as_ref().map(|c| c.0), user.name.clone().map(Cow::Owned)))
.unwrap_or_default(); .unwrap_or_default()
}
let user_id = user_id.as_str(); pub fn get_user_style(&self, user_id: &UserId) -> Style {
let color = color.unwrap_or_else(|| user_color(user_id)); let color = self
.tunables
.users
.get(user_id)
.and_then(|user| user.color.as_ref().map(|c| c.0))
.unwrap_or_else(|| user_color(user_id.as_str()));
user_style_from_color(color)
}
pub fn get_user_span<'a>(&self, user_id: &'a UserId, info: &'a RoomInfo) -> Span<'a> {
let (color, name) = self.get_user_overrides(user_id);
let color = color.unwrap_or_else(|| user_color(user_id.as_str()));
let style = user_style_from_color(color); let style = user_style_from_color(color);
let name = name.unwrap_or(Cow::Borrowed(user_id)); let name = match (name, &self.tunables.username_display) {
(Some(name), _) => name,
(None, UserDisplayStyle::Username) => Cow::Borrowed(user_id.as_str()),
(None, UserDisplayStyle::LocalPart) => Cow::Borrowed(user_id.localpart()),
(None, UserDisplayStyle::DisplayName) => {
if let Some(display) = info.display_names.get(user_id) {
Cow::Borrowed(display.as_str())
} else {
Cow::Borrowed(user_id.as_str())
}
},
};
Span::styled(name, style) Span::styled(name, style)
} }
@@ -425,6 +592,7 @@ impl ApplicationSettings {
mod tests { mod tests {
use super::*; use super::*;
use matrix_sdk::ruma::user_id; use matrix_sdk::ruma::user_id;
use std::convert::TryFrom;
#[test] #[test]
fn test_profile_name_invalid() { fn test_profile_name_invalid() {
@@ -518,4 +686,75 @@ mod tests {
})]; })];
assert_eq!(res.users, Some(users.into_iter().collect())); assert_eq!(res.users, Some(users.into_iter().collect()));
} }
#[test]
fn test_parse_tunables_username_display() {
let res: Tunables = serde_json::from_str("{\"username_display\": \"username\"}").unwrap();
assert_eq!(res.username_display, Some(UserDisplayStyle::Username));
let res: Tunables = serde_json::from_str("{\"username_display\": \"localpart\"}").unwrap();
assert_eq!(res.username_display, Some(UserDisplayStyle::LocalPart));
let res: Tunables =
serde_json::from_str("{\"username_display\": \"displayname\"}").unwrap();
assert_eq!(res.username_display, Some(UserDisplayStyle::DisplayName));
}
#[test]
fn test_parse_layout() {
let user = WindowPath::UserId(user_id!("@user:example.com").to_owned());
let alias = WindowPath::AliasId(OwnedRoomAliasId::try_from("#room:example.com").unwrap());
let room = WindowPath::RoomId(OwnedRoomId::try_from("!room:example.com").unwrap());
let dms = WindowPath::Window(IambId::DirectList);
let welcome = WindowPath::Window(IambId::Welcome);
let res: Layout = serde_json::from_str("{\"style\": \"restore\"}").unwrap();
assert_eq!(res, Layout::Restore);
let res: Layout = serde_json::from_str("{\"style\": \"new\"}").unwrap();
assert_eq!(res, Layout::New);
let res: Layout = serde_json::from_str(
"{\"style\": \"config\", \"tabs\": [{\"window\":\"@user:example.com\"}]}",
)
.unwrap();
assert_eq!(res, Layout::Config {
tabs: vec![WindowLayout::Window { window: user.clone() }]
});
let res: Layout = serde_json::from_str(
"{\
\"style\": \"config\",\
\"tabs\": [\
{\"split\":[\
{\"window\":\"@user:example.com\"},\
{\"window\":\"#room:example.com\"}\
]},\
{\"split\":[\
{\"window\":\"!room:example.com\"},\
{\"split\":[\
{\"window\":\"iamb://dms\"},\
{\"window\":\"iamb://welcome\"}\
]}\
]}\
]}",
)
.unwrap();
let split1 = WindowLayout::Split {
split: vec![
WindowLayout::Window { window: user.clone() },
WindowLayout::Window { window: alias },
],
};
let split2 = WindowLayout::Split {
split: vec![WindowLayout::Window { window: dms }, WindowLayout::Window {
window: welcome,
}],
};
let split3 = WindowLayout::Split {
split: vec![WindowLayout::Window { window: room }, split2],
};
let tabs = vec![split1, split3];
assert_eq!(res, Layout::Config { tabs });
}
} }

View File

@@ -1,32 +1,21 @@
use modalkit::{ use modalkit::{
editing::action::WindowAction, editing::action::WindowAction,
editing::base::WordStyle,
env::vim::keybindings::{InputStep, VimBindings}, env::vim::keybindings::{InputStep, VimBindings},
env::vim::VimMode, env::vim::VimMode,
input::bindings::{EdgeEvent, EdgeRepeat, InputBindings}, input::bindings::{EdgeEvent, EdgeRepeat, InputBindings},
input::key::TerminalKey, input::key::TerminalKey,
}; };
use crate::base::{IambAction, Keybindings}; use crate::base::{IambAction, IambInfo, Keybindings, MATRIX_ID_WORD};
/// Find the boundaries for a Matrix username, room alias, or room ID. type IambStep = InputStep<IambInfo>;
///
/// Technically "[" and "]" should be here since IPv6 addresses are allowed
/// in the server name, but in practice that should be uncommon, and people
/// can just use `gf` and friends in Visual mode instead.
fn is_mxid_char(c: char) -> bool {
return c >= 'a' && c <= 'z' ||
c >= 'A' && c <= 'Z' ||
c >= '0' && c <= '9' ||
":-./@_#!".contains(c);
}
pub fn setup_keybindings() -> Keybindings { pub fn setup_keybindings() -> Keybindings {
let mut ism = Keybindings::empty(); let mut ism = Keybindings::empty();
let vim = VimBindings::default() let vim = VimBindings::default()
.submit_on_enter() .submit_on_enter()
.cursor_open(WordStyle::CharSet(is_mxid_char)); .cursor_open(MATRIX_ID_WORD.clone());
vim.setup(&mut ism); vim.setup(&mut ism);
@@ -44,19 +33,27 @@ pub fn setup_keybindings() -> Keybindings {
(EdgeRepeat::Once, ctrl_w.clone()), (EdgeRepeat::Once, ctrl_w.clone()),
(EdgeRepeat::Once, ctrl_z), (EdgeRepeat::Once, ctrl_z),
]; ];
let zoom = InputStep::new().actions(vec![WindowAction::ZoomToggle.into()]); let zoom = IambStep::new()
.actions(vec![WindowAction::ZoomToggle.into()])
.goto(VimMode::Normal);
ism.add_mapping(VimMode::Normal, &cwz, &zoom); ism.add_mapping(VimMode::Normal, &cwz, &zoom);
ism.add_mapping(VimMode::Visual, &cwz, &zoom);
ism.add_mapping(VimMode::Normal, &cwcz, &zoom); ism.add_mapping(VimMode::Normal, &cwcz, &zoom);
ism.add_mapping(VimMode::Visual, &cwcz, &zoom);
let cwm = vec![ let cwm = vec![
(EdgeRepeat::Once, ctrl_w.clone()), (EdgeRepeat::Once, ctrl_w.clone()),
(EdgeRepeat::Once, key_m_lc), (EdgeRepeat::Once, key_m_lc),
]; ];
let cwcm = vec![(EdgeRepeat::Once, ctrl_w), (EdgeRepeat::Once, ctrl_m)]; let cwcm = vec![(EdgeRepeat::Once, ctrl_w), (EdgeRepeat::Once, ctrl_m)];
let stoggle = InputStep::new().actions(vec![IambAction::ToggleScrollbackFocus.into()]); let stoggle = IambStep::new()
.actions(vec![IambAction::ToggleScrollbackFocus.into()])
.goto(VimMode::Normal);
ism.add_mapping(VimMode::Normal, &cwm, &stoggle); ism.add_mapping(VimMode::Normal, &cwm, &stoggle);
ism.add_mapping(VimMode::Visual, &cwm, &stoggle);
ism.add_mapping(VimMode::Normal, &cwcm, &stoggle); ism.add_mapping(VimMode::Normal, &cwcm, &stoggle);
ism.add_mapping(VimMode::Visual, &cwcm, &stoggle);
return ism; return ism;
} }

View File

@@ -6,7 +6,7 @@ use std::collections::VecDeque;
use std::convert::TryFrom; use std::convert::TryFrom;
use std::fmt::Display; use std::fmt::Display;
use std::fs::{create_dir_all, File}; use std::fs::{create_dir_all, File};
use std::io::{stdout, BufReader, Stdout}; use std::io::{stdout, BufReader, BufWriter, Stdout};
use std::ops::DerefMut; use std::ops::DerefMut;
use std::process; use std::process;
use std::sync::atomic::{AtomicUsize, Ordering}; use std::sync::atomic::{AtomicUsize, Ordering};
@@ -15,15 +15,28 @@ use std::time::Duration;
use clap::Parser; use clap::Parser;
use tokio::sync::Mutex as AsyncMutex; use tokio::sync::Mutex as AsyncMutex;
use tracing::{self, Level};
use tracing_subscriber::FmtSubscriber; use tracing_subscriber::FmtSubscriber;
use matrix_sdk::ruma::OwnedUserId; use matrix_sdk::{
config::SyncSettings,
ruma::{
api::client::filter::{FilterDefinition, LazyLoadOptions, RoomEventFilter, RoomFilter},
OwnedUserId,
},
};
use modalkit::crossterm::{ use modalkit::crossterm::{
self, self,
cursor::Show as CursorShow, cursor::Show as CursorShow,
event::{poll, read, Event}, event::{
poll,
read,
DisableBracketedPaste,
DisableFocusChange,
EnableBracketedPaste,
EnableFocusChange,
Event,
},
execute, execute,
terminal::{EnterAlternateScreen, LeaveAlternateScreen, SetTitle}, terminal::{EnterAlternateScreen, LeaveAlternateScreen, SetTitle},
}; };
@@ -53,20 +66,19 @@ use crate::{
base::{ base::{
AsyncProgramStore, AsyncProgramStore,
ChatStore, ChatStore,
HomeserverAction,
IambAction, IambAction,
IambBufferId,
IambError, IambError,
IambId, IambId,
IambInfo, IambInfo,
IambResult, IambResult,
ProgramAction, ProgramAction,
ProgramCommands,
ProgramContext, ProgramContext,
ProgramStore, ProgramStore,
}, },
config::{ApplicationSettings, Iamb}, config::{ApplicationSettings, Iamb},
windows::IambWindow, windows::IambWindow,
worker::{ClientWorker, LoginStyle, Requester}, worker::{create_room, ClientWorker, LoginStyle, Requester},
}; };
use modalkit::{ use modalkit::{
@@ -77,45 +89,141 @@ use modalkit::{
EditError, EditError,
EditInfo, EditInfo,
Editable, Editable,
EditorAction,
InfoMessage,
InsertTextAction,
Jumpable, Jumpable,
Promptable, Promptable,
Scrollable, Scrollable,
TabAction,
TabContainer, TabContainer,
TabCount, TabCount,
UIError,
WindowAction, WindowAction,
WindowContainer, WindowContainer,
}, },
base::{OpenTarget, RepeatType}, base::{MoveDir1D, OpenTarget, RepeatType},
context::Resolve, context::Resolve,
key::KeyManager, key::KeyManager,
store::Store, store::Store,
}, },
input::{bindings::BindingMachine, key::TerminalKey}, input::{bindings::BindingMachine, dialog::Pager, key::TerminalKey},
widgets::{ widgets::{
cmdbar::CommandBarState, cmdbar::CommandBarState,
screen::{Screen, ScreenState}, screen::{FocusList, Screen, ScreenState, TabLayoutDescription},
windows::WindowLayoutDescription,
TerminalCursor, TerminalCursor,
TerminalExtOps, TerminalExtOps,
Window, Window,
}, },
}; };
const MIN_MSG_LOAD: u32 = 50; fn config_tab_to_desc(
layout: config::WindowLayout,
store: &mut ProgramStore,
) -> IambResult<WindowLayoutDescription<IambInfo>> {
let desc = match layout {
config::WindowLayout::Window { window } => {
let ChatStore { names, worker, .. } = &mut store.application;
fn msg_load_req(area: Rect) -> u32 { let window = match window {
let n = area.height as u32; config::WindowPath::UserId(user_id) => {
let name = user_id.to_string();
let room_id = worker.join_room(name.clone())?;
names.insert(name, room_id.clone());
IambId::Room(room_id)
},
config::WindowPath::RoomId(room_id) => IambId::Room(room_id),
config::WindowPath::AliasId(alias) => {
let name = alias.to_string();
let room_id = worker.join_room(name.clone())?;
names.insert(name, room_id.clone());
IambId::Room(room_id)
},
config::WindowPath::Window(id) => id,
};
n.max(MIN_MSG_LOAD) WindowLayoutDescription::Window { window, length: None }
},
config::WindowLayout::Split { split } => {
let children = split
.into_iter()
.map(|child| config_tab_to_desc(child, store))
.collect::<IambResult<Vec<_>>>()?;
WindowLayoutDescription::Split { children, length: None }
},
};
Ok(desc)
}
fn setup_screen(
settings: ApplicationSettings,
store: &mut ProgramStore,
) -> IambResult<ScreenState<IambWindow, IambInfo>> {
let cmd = CommandBarState::new(store);
let dims = crossterm::terminal::size()?;
let area = Rect::new(0, 0, dims.0, dims.1);
match settings.layout {
config::Layout::Restore => {
if let Ok(layout) = std::fs::read(&settings.layout_json) {
let tabs: TabLayoutDescription<IambInfo> =
serde_json::from_slice(&layout).map_err(IambError::from)?;
let tabs = tabs.to_layout(area.into(), store)?;
return Ok(ScreenState::from_list(tabs, cmd));
}
},
config::Layout::New => {},
config::Layout::Config { tabs } => {
let mut list = FocusList::default();
for tab in tabs.into_iter() {
let tab = config_tab_to_desc(tab, store)?;
let tab = tab.to_layout(area.into(), store)?;
list.push(tab);
}
return Ok(ScreenState::from_list(list, cmd));
},
}
let win = settings
.tunables
.default_room
.and_then(|room| IambWindow::find(room, store).ok())
.or_else(|| IambWindow::open(IambId::Welcome, store).ok())
.unwrap();
return Ok(ScreenState::new(win, cmd));
} }
struct Application { struct Application {
store: AsyncProgramStore, /// Terminal backend.
worker: Requester,
terminal: Terminal<CrosstermBackend<Stdout>>, terminal: Terminal<CrosstermBackend<Stdout>>,
bindings: KeyManager<TerminalKey, ProgramAction, RepeatType, ProgramContext>,
actstack: VecDeque<(ProgramAction, ProgramContext)>, /// State for the Matrix client, editing, etc.
cmds: ProgramCommands, store: AsyncProgramStore,
/// UI state (open tabs, command bar, etc.) to use when rendering.
screen: ScreenState<IambWindow, IambInfo>, screen: ScreenState<IambWindow, IambInfo>,
/// Handle to communicate synchronously with the Matrix worker task.
worker: Requester,
/// Mapped keybindings.
bindings: KeyManager<TerminalKey, ProgramAction, RepeatType, ProgramContext>,
/// Pending actions to run.
actstack: VecDeque<(ProgramAction, ProgramContext)>,
/// Whether or not the terminal is currently focused.
focused: bool,
/// The tab layout before the last executed [TabAction].
last_layout: Option<TabLayoutDescription<IambInfo>>,
} }
impl Application { impl Application {
@@ -126,6 +234,8 @@ impl Application {
let mut stdout = stdout(); let mut stdout = stdout();
crossterm::terminal::enable_raw_mode()?; crossterm::terminal::enable_raw_mode()?;
crossterm::execute!(stdout, EnterAlternateScreen)?; crossterm::execute!(stdout, EnterAlternateScreen)?;
crossterm::execute!(stdout, EnableBracketedPaste)?;
crossterm::execute!(stdout, EnableFocusChange)?;
let title = format!("iamb ({})", settings.profile.user_id); let title = format!("iamb ({})", settings.profile.user_id);
crossterm::execute!(stdout, SetTitle(title))?; crossterm::execute!(stdout, SetTitle(title))?;
@@ -135,19 +245,9 @@ impl Application {
let bindings = crate::keybindings::setup_keybindings(); let bindings = crate::keybindings::setup_keybindings();
let bindings = KeyManager::new(bindings); let bindings = KeyManager::new(bindings);
let cmds = crate::commands::setup_commands();
let mut locked = store.lock().await; let mut locked = store.lock().await;
let screen = setup_screen(settings, locked.deref_mut())?;
let win = settings
.tunables
.default_room
.and_then(|room| IambWindow::find(room, locked.deref_mut()).ok())
.or_else(|| IambWindow::open(IambId::Welcome, locked.deref_mut()).ok())
.unwrap();
let cmd = CommandBarState::new(IambBufferId::Command, locked.deref_mut());
let screen = ScreenState::new(win, cmd);
let worker = locked.application.worker.clone(); let worker = locked.application.worker.clone();
drop(locked); drop(locked);
@@ -160,14 +260,15 @@ impl Application {
terminal, terminal,
bindings, bindings,
actstack, actstack,
cmds,
screen, screen,
focused: true,
last_layout: None,
}) })
} }
fn redraw(&mut self, full: bool, store: &mut ProgramStore) -> Result<(), std::io::Error> { fn redraw(&mut self, full: bool, store: &mut ProgramStore) -> Result<(), std::io::Error> {
let modestr = self.bindings.showmode(); let bindings = &mut self.bindings;
let cursor = self.bindings.get_cursor_indicator(); let focused = self.focused;
let sstate = &mut self.screen; let sstate = &mut self.screen;
let term = &mut self.terminal; let term = &mut self.terminal;
@@ -178,9 +279,24 @@ impl Application {
term.draw(|f| { term.draw(|f| {
let area = f.size(); let area = f.size();
let screen = Screen::new(store).showmode(modestr).borders(true); let modestr = bindings.show_mode();
let cursor = bindings.get_cursor_indicator();
let dialogstr = bindings.show_dialog(area.height as usize, area.width as usize);
// Don't show terminal cursor when we show a dialog.
let hide_cursor = !dialogstr.is_empty();
let screen = Screen::new(store)
.show_dialog(dialogstr)
.show_mode(modestr)
.borders(true)
.focus(focused);
f.render_stateful_widget(screen, area, sstate); f.render_stateful_widget(screen, area, sstate);
if hide_cursor {
return;
}
if let Some((cx, cy)) = sstate.get_term_cursor() { if let Some((cx, cy)) = sstate.get_term_cursor() {
if let Some(c) = cursor { if let Some(c) = cursor {
let style = Style::default().fg(Color::Green); let style = Style::default().fg(Color::Green);
@@ -191,8 +307,6 @@ impl Application {
} }
f.set_cursor(cx, cy); f.set_cursor(cx, cy);
} }
store.application.load_older(msg_load_req(area));
})?; })?;
Ok(()) Ok(())
@@ -212,14 +326,31 @@ impl Application {
Event::Mouse(_) => { Event::Mouse(_) => {
// Do nothing for now. // Do nothing for now.
}, },
Event::FocusGained | Event::FocusLost => { Event::FocusGained => {
// Do nothing for now. self.focused = true;
},
Event::FocusLost => {
self.focused = false;
}, },
Event::Resize(_, _) => { Event::Resize(_, _) => {
// We'll redraw for the new size next time step() is called. // We'll redraw for the new size next time step() is called.
}, },
Event::Paste(_) => { Event::Paste(s) => {
// Do nothing for now. let act = InsertTextAction::Transcribe(s, MoveDir1D::Previous, 1.into());
let act = EditorAction::from(act);
let ctx = ProgramContext::default();
let mut store = self.store.lock().await;
match self.screen.editor_command(&act, &ctx, store.deref_mut()) {
Ok(None) => {},
Ok(Some(info)) => {
drop(store);
self.handle_info(info);
},
Err(e) => {
self.screen.push_error(e);
},
}
}, },
} }
} }
@@ -278,8 +409,7 @@ impl Application {
Action::CommandBar(act) => self.screen.command_bar(&act, &ctx)?, Action::CommandBar(act) => self.screen.command_bar(&act, &ctx)?,
Action::Macro(act) => self.bindings.macro_command(&act, &ctx, store)?, Action::Macro(act) => self.bindings.macro_command(&act, &ctx, store)?,
Action::Scroll(style) => self.screen.scroll(&style, &ctx, store)?, Action::Scroll(style) => self.screen.scroll(&style, &ctx, store)?,
Action::Suspend => self.terminal.program_suspend()?, Action::ShowInfoMessage(info) => Some(info),
Action::Tab(cmd) => self.screen.tab_command(&cmd, &ctx, store)?,
Action::Window(cmd) => self.screen.window_command(&cmd, &ctx, store)?, Action::Window(cmd) => self.screen.window_command(&cmd, &ctx, store)?,
Action::Jump(l, dir, count) => { Action::Jump(l, dir, count) => {
@@ -288,8 +418,20 @@ impl Application {
None None
}, },
Action::Suspend => {
self.terminal.program_suspend()?;
None
},
// UI actions. // UI actions.
Action::Tab(cmd) => {
if let TabAction::Close(_, _) = &cmd {
self.last_layout = self.screen.as_description().into();
}
self.screen.tab_command(&cmd, &ctx, store)?
},
Action::RedrawScreen => { Action::RedrawScreen => {
self.screen.clear_message(); self.screen.clear_message();
self.redraw(true, store)?; self.redraw(true, store)?;
@@ -305,7 +447,7 @@ impl Application {
None None
}, },
Action::Command(act) => { Action::Command(act) => {
let acts = self.cmds.command(&act, &ctx)?; let acts = store.application.cmds.command(&act, &ctx)?;
self.action_prepend(acts); self.action_prepend(acts);
None None
@@ -344,6 +486,12 @@ impl Application {
None None
}, },
IambAction::Homeserver(act) => {
let acts = self.homeserver_command(act, ctx, store).await?;
self.action_prepend(acts);
None
},
IambAction::Message(act) => { IambAction::Message(act) => {
self.screen.current_window_mut()?.message_command(act, ctx, store).await? self.screen.current_window_mut()?.message_command(act, ctx, store).await?
}, },
@@ -376,6 +524,37 @@ impl Application {
Ok(info) Ok(info)
} }
async fn homeserver_command(
&mut self,
action: HomeserverAction,
ctx: ProgramContext,
store: &mut ProgramStore,
) -> IambResult<Vec<(Action<IambInfo>, ProgramContext)>> {
match action {
HomeserverAction::CreateRoom(alias, vis, flags) => {
let client = &store.application.worker.client;
let room_id = create_room(client, alias.as_deref(), vis, flags).await?;
let room = IambId::Room(room_id);
let target = OpenTarget::Application(room);
let action = WindowAction::Switch(target);
Ok(vec![(action.into(), ctx)])
},
}
}
fn handle_info(&mut self, info: InfoMessage) {
match info {
InfoMessage::Message(info) => {
self.screen.push_info(info);
},
InfoMessage::Pager(text) => {
let pager = Box::new(Pager::new(text, vec![]));
self.bindings.run_dialog(pager);
},
}
}
pub async fn run(&mut self) -> Result<(), std::io::Error> { pub async fn run(&mut self) -> Result<(), std::io::Error> {
self.terminal.clear()?; self.terminal.clear()?;
@@ -396,11 +575,18 @@ impl Application {
continue; continue;
}, },
Ok(Some(info)) => { Ok(Some(info)) => {
self.screen.push_info(info); self.handle_info(info);
// Continue processing; we'll redraw later. // Continue processing; we'll redraw later.
continue; continue;
}, },
Err(
UIError::NeedConfirm(dialog) |
UIError::EditingFailure(EditError::NeedConfirm(dialog)),
) => {
self.bindings.run_dialog(dialog);
continue;
},
Err(e) => { Err(e) => {
self.screen.push_error(e); self.screen.push_error(e);
@@ -412,6 +598,19 @@ impl Application {
} }
} }
if let Some(ref layout) = self.last_layout {
let locked = self.store.lock().await;
let path = locked.application.settings.layout_json.as_path();
path.parent().map(create_dir_all).transpose()?;
let file = File::create(path)?;
let writer = BufWriter::new(file);
if let Err(e) = serde_json::to_writer(writer, layout) {
tracing::error!("Failed to save window layout while exiting: {}", e);
}
}
crossterm::terminal::disable_raw_mode()?; crossterm::terminal::disable_raw_mode()?;
execute!(self.terminal.backend_mut(), LeaveAlternateScreen)?; execute!(self.terminal.backend_mut(), LeaveAlternateScreen)?;
self.terminal.show_cursor()?; self.terminal.show_cursor()?;
@@ -439,23 +638,37 @@ async fn login(worker: Requester, settings: &ApplicationSettings) -> IambResult<
match worker.login(LoginStyle::Password(password)) { match worker.login(LoginStyle::Password(password)) {
Ok(info) => { Ok(info) => {
if let Some(msg) = info { if let Some(msg) = info {
println!("{}", msg); println!("{msg}");
} }
break; break;
}, },
Err(err) => { Err(err) => {
println!("Failed to login: {}", err); println!("Failed to login: {err}");
continue; continue;
}, },
} }
} }
// Perform an initial, lazily-loaded sync.
let mut room = RoomEventFilter::default();
room.lazy_load_options = LazyLoadOptions::Enabled { include_redundant_members: false };
let mut room_ev = RoomFilter::default();
room_ev.state = room;
let mut filter = FilterDefinition::default();
filter.room = room_ev;
let settings = SyncSettings::new().filter(filter.into());
worker.client.sync_once(settings).await.map_err(IambError::from)?;
Ok(()) Ok(())
} }
fn print_exit<T: Display, N>(v: T) -> N { fn print_exit<T: Display, N>(v: T) -> N {
println!("{}", v); println!("{v}");
process::exit(2); process::exit(2);
} }
@@ -469,12 +682,18 @@ async fn run(settings: ApplicationSettings) -> IambResult<()> {
login(worker, &settings).await.unwrap_or_else(print_exit); login(worker, &settings).await.unwrap_or_else(print_exit);
fn restore_tty() {
let _ = crossterm::terminal::disable_raw_mode();
let _ = crossterm::execute!(stdout(), DisableBracketedPaste);
let _ = crossterm::execute!(stdout(), DisableFocusChange);
let _ = crossterm::execute!(stdout(), LeaveAlternateScreen);
let _ = crossterm::execute!(stdout(), CursorShow);
}
// Make sure panics clean up the terminal properly. // Make sure panics clean up the terminal properly.
let orig_hook = std::panic::take_hook(); let orig_hook = std::panic::take_hook();
std::panic::set_hook(Box::new(move |panic_info| { std::panic::set_hook(Box::new(move |panic_info| {
let _ = crossterm::terminal::disable_raw_mode(); restore_tty();
let _ = crossterm::execute!(stdout(), LeaveAlternateScreen);
let _ = crossterm::execute!(stdout(), CursorShow);
orig_hook(panic_info); orig_hook(panic_info);
process::exit(1); process::exit(1);
})); }));
@@ -483,6 +702,7 @@ async fn run(settings: ApplicationSettings) -> IambResult<()> {
// We can now run the application. // We can now run the application.
application.run().await?; application.run().await?;
restore_tty();
Ok(()) Ok(())
} }
@@ -494,6 +714,12 @@ fn main() -> IambResult<()> {
// Load configuration and set up the Matrix SDK. // Load configuration and set up the Matrix SDK.
let settings = ApplicationSettings::load(iamb).unwrap_or_else(print_exit); let settings = ApplicationSettings::load(iamb).unwrap_or_else(print_exit);
// Set umask on Unix platforms so that tokens, keys, etc. are only readable by the user.
#[cfg(unix)]
unsafe {
libc::umask(0o077);
};
// Set up the tracing subscriber so we can log client messages. // Set up the tracing subscriber so we can log client messages.
let log_prefix = format!("iamb-log-{}", settings.profile_name); let log_prefix = format!("iamb-log-{}", settings.profile_name);
let log_dir = settings.dirs.logs.as_path(); let log_dir = settings.dirs.logs.as_path();
@@ -506,16 +732,17 @@ fn main() -> IambResult<()> {
let subscriber = FmtSubscriber::builder() let subscriber = FmtSubscriber::builder()
.with_writer(appender) .with_writer(appender)
.with_max_level(Level::TRACE) .with_max_level(settings.tunables.log_level)
.finish(); .finish();
tracing::subscriber::set_global_default(subscriber).expect("setting default subscriber failed"); tracing::subscriber::set_global_default(subscriber).expect("setting default subscriber failed");
let rt = tokio::runtime::Builder::new_multi_thread() let rt = tokio::runtime::Builder::new_multi_thread()
.enable_all() .enable_all()
.worker_threads(2)
.thread_name_fn(|| { .thread_name_fn(|| {
static ATOMIC_ID: AtomicUsize = AtomicUsize::new(0); static ATOMIC_ID: AtomicUsize = AtomicUsize::new(0);
let id = ATOMIC_ID.fetch_add(1, Ordering::SeqCst); let id = ATOMIC_ID.fetch_add(1, Ordering::SeqCst);
format!("iamb-worker-{}", id) format!("iamb-worker-{id}")
}) })
.build() .build()
.unwrap(); .unwrap();

View File

@@ -6,7 +6,7 @@
//! The Matrix specification recommends limiting rendered tags and attributes to a safe subset of //! The Matrix specification recommends limiting rendered tags and attributes to a safe subset of
//! HTML. You can read more in section 11.2.1.1, "m.room.message msgtypes": //! HTML. You can read more in section 11.2.1.1, "m.room.message msgtypes":
//! //!
//! https://spec.matrix.org/unstable/client-server-api/#mroommessage-msgtypes //! <https://spec.matrix.org/unstable/client-server-api/#mroommessage-msgtypes>
//! //!
//! This isn't as important for iamb, since it isn't a browser environment, but we do still map //! This isn't as important for iamb, since it isn't a browser environment, but we do still map
//! input onto an enum of the safe list of tags to keep it easy to understand and process. //! input onto an enum of the safe list of tags to keep it easy to understand and process.
@@ -130,7 +130,6 @@ impl Table {
let cell_min = cell_total / columns; let cell_min = cell_total / columns;
let mut cell_slop = cell_total - cell_min * columns; let mut cell_slop = cell_total - cell_min * columns;
let cell_widths = (0..columns) let cell_widths = (0..columns)
.into_iter()
.map(|_| { .map(|_| {
let slopped = cell_slop.min(1); let slopped = cell_slop.min(1);
cell_slop -= slopped; cell_slop -= slopped;
@@ -238,6 +237,7 @@ pub enum StyleTreeNode {
Image(Option<String>), Image(Option<String>),
List(StyleTreeChildren, ListStyle), List(StyleTreeChildren, ListStyle),
Paragraph(Box<StyleTreeNode>), Paragraph(Box<StyleTreeNode>),
Pre(Box<StyleTreeNode>),
Reply(Box<StyleTreeNode>), Reply(Box<StyleTreeNode>),
Ruler, Ruler,
Style(Box<StyleTreeNode>, Style), Style(Box<StyleTreeNode>, Style),
@@ -271,10 +271,12 @@ impl StyleTreeNode {
}, },
StyleTreeNode::Header(child, level) => { StyleTreeNode::Header(child, level) => {
let style = style.add_modifier(StyleModifier::BOLD); let style = style.add_modifier(StyleModifier::BOLD);
let mut hashes = "#".repeat(*level);
hashes.push(' ');
printer.push_str(hashes, style); for _ in 0..*level {
printer.push_str("#", style);
}
printer.push_str(" ", style);
child.print(printer, style); child.print(printer, style);
}, },
StyleTreeNode::Image(None) => {}, StyleTreeNode::Image(None) => {},
@@ -310,6 +312,39 @@ impl StyleTreeNode {
child.print(printer, style); child.print(printer, style);
printer.commit(); printer.commit();
}, },
StyleTreeNode::Pre(child) => {
let mut subp = printer.sub(2).literal(true);
let subw = subp.width();
child.print(&mut subp, style);
printer.commit();
printer.push_line(
vec![
Span::styled(line::TOP_LEFT, style),
Span::styled(line::HORIZONTAL.repeat(subw), style),
Span::styled(line::TOP_RIGHT, style),
]
.into(),
);
for mut line in subp.finish() {
line.0.insert(0, Span::styled(line::VERTICAL, style));
line.0.push(Span::styled(line::VERTICAL, style));
printer.push_line(line);
}
printer.push_line(
vec![
Span::styled(line::BOTTOM_LEFT, style),
Span::styled(line::HORIZONTAL.repeat(subw), style),
Span::styled(line::BOTTOM_RIGHT, style),
]
.into(),
);
printer.commit();
},
StyleTreeNode::Reply(child) => { StyleTreeNode::Reply(child) => {
if printer.hide_reply() { if printer.hide_reply() {
return; return;
@@ -320,7 +355,9 @@ impl StyleTreeNode {
printer.commit(); printer.commit();
}, },
StyleTreeNode::Ruler => { StyleTreeNode::Ruler => {
printer.push_str(line::HORIZONTAL.repeat(width), style); for _ in 0..width {
printer.push_str(line::HORIZONTAL, style);
}
}, },
StyleTreeNode::Table(table) => { StyleTreeNode::Table(table) => {
let text = table.to_text(width, style); let text = table.to_text(width, style);
@@ -582,6 +619,7 @@ fn h2t(hdl: &Handle) -> StyleTreeChildren {
// Other text blocks. // Other text blocks.
"blockquote" => StyleTreeNode::Blockquote(c2t(&node.children.borrow())), "blockquote" => StyleTreeNode::Blockquote(c2t(&node.children.borrow())),
"div" | "p" => StyleTreeNode::Paragraph(c2t(&node.children.borrow())), "div" | "p" => StyleTreeNode::Paragraph(c2t(&node.children.borrow())),
"pre" => StyleTreeNode::Pre(c2t(&node.children.borrow())),
// No children. // No children.
"hr" => StyleTreeNode::Ruler, "hr" => StyleTreeNode::Ruler,
@@ -590,7 +628,7 @@ fn h2t(hdl: &Handle) -> StyleTreeChildren {
"img" => StyleTreeNode::Image(attrs_to_alt(&attrs.borrow())), "img" => StyleTreeNode::Image(attrs_to_alt(&attrs.borrow())),
// These don't render in any special way. // These don't render in any special way.
"a" | "details" | "html" | "pre" | "summary" | "sub" | "sup" => { "a" | "details" | "html" | "summary" | "sub" | "sup" => {
*c2t(&node.children.borrow()) *c2t(&node.children.borrow())
}, },
@@ -615,7 +653,7 @@ pub fn parse_matrix_html(s: &str) -> StyleTree {
let dom = parse_fragment( let dom = parse_fragment(
RcDom::default(), RcDom::default(),
ParseOpts::default(), ParseOpts::default(),
QualName::new(None, ns!(), local_name!("div")), QualName::new(None, ns!(html), local_name!("body")),
vec![], vec![],
) )
.one(StrTendril::from(s)); .one(StrTendril::from(s));
@@ -627,6 +665,7 @@ pub fn parse_matrix_html(s: &str) -> StyleTree {
pub mod tests { pub mod tests {
use super::*; use super::*;
use crate::util::space_span; use crate::util::space_span;
use pretty_assertions::assert_eq;
#[test] #[test]
fn test_header() { fn test_header() {
@@ -637,7 +676,10 @@ pub mod tests {
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("#", bold), Span::styled("#", bold),
Span::styled("Header 1", bold), Span::styled(" ", bold),
Span::styled("Header", bold),
Span::styled(" ", bold),
Span::styled("1", bold),
space_span(10, Style::default()) space_span(10, Style::default())
])]); ])]);
@@ -645,8 +687,12 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("## ", bold), Span::styled("#", bold),
Span::styled("Header 2", bold), Span::styled("#", bold),
Span::styled(" ", bold),
Span::styled("Header", bold),
Span::styled(" ", bold),
Span::styled("2", bold),
space_span(9, Style::default()) space_span(9, Style::default())
])]); ])]);
@@ -654,8 +700,13 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("### ", bold), Span::styled("#", bold),
Span::styled("Header 3", bold), Span::styled("#", bold),
Span::styled("#", bold),
Span::styled(" ", bold),
Span::styled("Header", bold),
Span::styled(" ", bold),
Span::styled("3", bold),
space_span(8, Style::default()) space_span(8, Style::default())
])]); ])]);
@@ -663,8 +714,14 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("#### ", bold), Span::styled("#", bold),
Span::styled("Header 4", bold), Span::styled("#", bold),
Span::styled("#", bold),
Span::styled("#", bold),
Span::styled(" ", bold),
Span::styled("Header", bold),
Span::styled(" ", bold),
Span::styled("4", bold),
space_span(7, Style::default()) space_span(7, Style::default())
])]); ])]);
@@ -672,8 +729,15 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("##### ", bold), Span::styled("#", bold),
Span::styled("Header 5", bold), Span::styled("#", bold),
Span::styled("#", bold),
Span::styled("#", bold),
Span::styled("#", bold),
Span::styled(" ", bold),
Span::styled("Header", bold),
Span::styled(" ", bold),
Span::styled("5", bold),
space_span(6, Style::default()) space_span(6, Style::default())
])]); ])]);
@@ -681,8 +745,16 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("###### ", bold), Span::styled("#", bold),
Span::styled("Header 6", bold), Span::styled("#", bold),
Span::styled("#", bold),
Span::styled("#", bold),
Span::styled("#", bold),
Span::styled("#", bold),
Span::styled(" ", bold),
Span::styled("Header", bold),
Span::styled(" ", bold),
Span::styled("6", bold),
space_span(5, Style::default()) space_span(5, Style::default())
])]); ])]);
} }
@@ -700,7 +772,8 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Bold!", bold), Span::styled("Bold", bold),
Span::styled("!", bold),
space_span(15, def) space_span(15, def)
])]); ])]);
@@ -708,7 +781,8 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Bold!", bold), Span::styled("Bold", bold),
Span::styled("!", bold),
space_span(15, def) space_span(15, def)
])]); ])]);
@@ -716,7 +790,8 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Italic!", italic), Span::styled("Italic", italic),
Span::styled("!", italic),
space_span(13, def) space_span(13, def)
])]); ])]);
@@ -724,7 +799,8 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Italic!", italic), Span::styled("Italic", italic),
Span::styled("!", italic),
space_span(13, def) space_span(13, def)
])]); ])]);
@@ -732,7 +808,8 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Strikethrough!", strike), Span::styled("Strikethrough", strike),
Span::styled("!", strike),
space_span(6, def) space_span(6, def)
])]); ])]);
@@ -740,7 +817,8 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Strikethrough!", strike), Span::styled("Strikethrough", strike),
Span::styled("!", strike),
space_span(6, def) space_span(6, def)
])]); ])]);
@@ -748,19 +826,28 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![ assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Underline!", underl), Span::styled("Underline", underl),
Span::styled("!", underl),
space_span(10, def) space_span(10, def)
])]); ])]);
let s = "<font color=\"#ff0000\">Red!</u>"; let s = "<font color=\"#ff0000\">Red!</u>";
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![Span::styled("Red!", red), space_span(16, def)])]); assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Red", red),
Span::styled("!", red),
space_span(16, def)
])]);
let s = "<font color=\"red\">Red!</u>"; let s = "<font color=\"red\">Red!</u>";
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(20, Style::default(), false); let text = tree.to_text(20, Style::default(), false);
assert_eq!(text.lines, vec![Spans(vec![Span::styled("Red!", red), space_span(16, def)])]); assert_eq!(text.lines, vec![Spans(vec![
Span::styled("Red", red),
Span::styled("!", red),
space_span(16, def)
])]);
} }
#[test] #[test]
@@ -769,13 +856,25 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(10, Style::default(), false); let text = tree.to_text(10, Style::default(), false);
assert_eq!(text.lines.len(), 7); assert_eq!(text.lines.len(), 7);
assert_eq!(text.lines[0], Spans(vec![Span::raw("Hello worl")])); assert_eq!(
assert_eq!(text.lines[1], Spans(vec![Span::raw("d!"), Span::raw(" ")])); text.lines[0],
Spans(vec![Span::raw("Hello"), Span::raw(" "), Span::raw(" ")])
);
assert_eq!(
text.lines[1],
Spans(vec![Span::raw("world"), Span::raw("!"), Span::raw(" ")])
);
assert_eq!(text.lines[2], Spans(vec![Span::raw(" ")])); assert_eq!(text.lines[2], Spans(vec![Span::raw(" ")]));
assert_eq!(text.lines[3], Spans(vec![Span::raw("Content"), Span::raw(" ")])); assert_eq!(text.lines[3], Spans(vec![Span::raw("Content"), Span::raw(" ")]));
assert_eq!(text.lines[4], Spans(vec![Span::raw(" ")])); assert_eq!(text.lines[4], Spans(vec![Span::raw(" ")]));
assert_eq!(text.lines[5], Spans(vec![Span::raw("Goodbye wo")])); assert_eq!(
assert_eq!(text.lines[6], Spans(vec![Span::raw("rld!"), Span::raw(" ")])); text.lines[5],
Spans(vec![Span::raw("Goodbye"), Span::raw(" "), Span::raw(" ")])
);
assert_eq!(
text.lines[6],
Spans(vec![Span::raw("world"), Span::raw("!"), Span::raw(" ")])
);
} }
#[test] #[test]
@@ -784,8 +883,14 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(10, Style::default(), false); let text = tree.to_text(10, Style::default(), false);
assert_eq!(text.lines.len(), 2); assert_eq!(text.lines.len(), 2);
assert_eq!(text.lines[0], Spans(vec![Span::raw(" "), Span::raw("Hello ")])); assert_eq!(
assert_eq!(text.lines[1], Spans(vec![Span::raw(" "), Span::raw("world!")])); text.lines[0],
Spans(vec![Span::raw(" "), Span::raw("Hello"), Span::raw(" ")])
);
assert_eq!(
text.lines[1],
Spans(vec![Span::raw(" "), Span::raw("world"), Span::raw("!")])
);
} }
#[test] #[test]
@@ -794,12 +899,60 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(8, Style::default(), false); let text = tree.to_text(8, Style::default(), false);
assert_eq!(text.lines.len(), 6); assert_eq!(text.lines.len(), 6);
assert_eq!(text.lines[0], Spans(vec![Span::raw("- "), Span::raw("List I")])); assert_eq!(
assert_eq!(text.lines[1], Spans(vec![Span::raw(" "), Span::raw("tem 1"), Span::raw(" ")])); text.lines[0],
assert_eq!(text.lines[2], Spans(vec![Span::raw("- "), Span::raw("List I")])); Spans(vec![
assert_eq!(text.lines[3], Spans(vec![Span::raw(" "), Span::raw("tem 2"), Span::raw(" ")])); Span::raw("- "),
assert_eq!(text.lines[4], Spans(vec![Span::raw("- "), Span::raw("List I")])); Span::raw("List"),
assert_eq!(text.lines[5], Spans(vec![Span::raw(" "), Span::raw("tem 3"), Span::raw(" ")])); Span::raw(" "),
Span::raw(" ")
])
);
assert_eq!(
text.lines[1],
Spans(vec![
Span::raw(" "),
Span::raw("Item"),
Span::raw(" "),
Span::raw("1")
])
);
assert_eq!(
text.lines[2],
Spans(vec![
Span::raw("- "),
Span::raw("List"),
Span::raw(" "),
Span::raw(" ")
])
);
assert_eq!(
text.lines[3],
Spans(vec![
Span::raw(" "),
Span::raw("Item"),
Span::raw(" "),
Span::raw("2")
])
);
assert_eq!(
text.lines[4],
Spans(vec![
Span::raw("- "),
Span::raw("List"),
Span::raw(" "),
Span::raw(" ")
])
);
assert_eq!(
text.lines[5],
Spans(vec![
Span::raw(" "),
Span::raw("Item"),
Span::raw(" "),
Span::raw("3")
])
);
} }
#[test] #[test]
@@ -808,20 +961,59 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(9, Style::default(), false); let text = tree.to_text(9, Style::default(), false);
assert_eq!(text.lines.len(), 6); assert_eq!(text.lines.len(), 6);
assert_eq!(text.lines[0], Spans(vec![Span::raw("1. "), Span::raw("List I")])); assert_eq!(
text.lines[0],
Spans(vec![
Span::raw("1. "),
Span::raw("List"),
Span::raw(" "),
Span::raw(" ")
])
);
assert_eq!( assert_eq!(
text.lines[1], text.lines[1],
Spans(vec![Span::raw(" "), Span::raw("tem 1"), Span::raw(" ")]) Spans(vec![
Span::raw(" "),
Span::raw("Item"),
Span::raw(" "),
Span::raw("1")
])
);
assert_eq!(
text.lines[2],
Spans(vec![
Span::raw("2. "),
Span::raw("List"),
Span::raw(" "),
Span::raw(" ")
])
); );
assert_eq!(text.lines[2], Spans(vec![Span::raw("2. "), Span::raw("List I")]));
assert_eq!( assert_eq!(
text.lines[3], text.lines[3],
Spans(vec![Span::raw(" "), Span::raw("tem 2"), Span::raw(" ")]) Spans(vec![
Span::raw(" "),
Span::raw("Item"),
Span::raw(" "),
Span::raw("2")
])
);
assert_eq!(
text.lines[4],
Spans(vec![
Span::raw("3. "),
Span::raw("List"),
Span::raw(" "),
Span::raw(" ")
])
); );
assert_eq!(text.lines[4], Spans(vec![Span::raw("3. "), Span::raw("List I")]));
assert_eq!( assert_eq!(
text.lines[5], text.lines[5],
Spans(vec![Span::raw(" "), Span::raw("tem 3"), Span::raw(" ")]) Spans(vec![
Span::raw(" "),
Span::raw("Item"),
Span::raw(" "),
Span::raw("3")
])
); );
} }
@@ -854,9 +1046,13 @@ pub mod tests {
]); ]);
assert_eq!(text.lines[2].0, vec![ assert_eq!(text.lines[2].0, vec![
Span::raw(""), Span::raw(""),
Span::styled("mn 1", bold), Span::styled("mn", bold),
Span::styled(" ", bold),
Span::styled("1", bold),
Span::raw(""), Span::raw(""),
Span::styled("mn 2", bold), Span::styled("mn", bold),
Span::styled(" ", bold),
Span::styled("2", bold),
Span::raw(""), Span::raw(""),
Span::styled("umn", bold), Span::styled("umn", bold),
Span::raw("") Span::raw("")
@@ -928,15 +1124,161 @@ pub mod tests {
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(10, Style::default(), false); let text = tree.to_text(10, Style::default(), false);
assert_eq!(text.lines.len(), 4); assert_eq!(text.lines.len(), 4);
assert_eq!(text.lines[0], Spans(vec![Span::raw("This was r")])); assert_eq!(
assert_eq!(text.lines[1], Spans(vec![Span::raw("eplied to"), Span::raw(" ")])); text.lines[0],
assert_eq!(text.lines[2], Spans(vec![Span::raw("This is th")])); Spans(vec![
assert_eq!(text.lines[3], Spans(vec![Span::raw("e reply"), Span::raw(" ")])); Span::raw("This"),
Span::raw(" "),
Span::raw("was"),
Span::raw(" "),
Span::raw(" ")
])
);
assert_eq!(
text.lines[1],
Spans(vec![Span::raw("replied"), Span::raw(" "), Span::raw("to")])
);
assert_eq!(
text.lines[2],
Spans(vec![
Span::raw("This"),
Span::raw(" "),
Span::raw("is"),
Span::raw(" "),
Span::raw(" ")
])
);
assert_eq!(
text.lines[3],
Spans(vec![
Span::raw("the"),
Span::raw(" "),
Span::raw("reply"),
Span::raw(" ")
])
);
let tree = parse_matrix_html(s); let tree = parse_matrix_html(s);
let text = tree.to_text(10, Style::default(), true); let text = tree.to_text(10, Style::default(), true);
assert_eq!(text.lines.len(), 2); assert_eq!(text.lines.len(), 2);
assert_eq!(text.lines[0], Spans(vec![Span::raw("This is th")])); assert_eq!(
assert_eq!(text.lines[1], Spans(vec![Span::raw("e reply"), Span::raw(" ")])); text.lines[0],
Spans(vec![
Span::raw("This"),
Span::raw(" "),
Span::raw("is"),
Span::raw(" "),
Span::raw(" ")
])
);
assert_eq!(
text.lines[1],
Spans(vec![
Span::raw("the"),
Span::raw(" "),
Span::raw("reply"),
Span::raw(" ")
])
);
}
#[test]
fn test_self_closing() {
let s = "Hello<br>World<br>Goodbye";
let tree = parse_matrix_html(s);
let text = tree.to_text(7, Style::default(), true);
assert_eq!(text.lines.len(), 3);
assert_eq!(text.lines[0], Spans(vec![Span::raw("Hello"), Span::raw(" "),]));
assert_eq!(text.lines[1], Spans(vec![Span::raw("World"), Span::raw(" "),]));
assert_eq!(text.lines[2], Spans(vec![Span::raw("Goodbye")]),);
}
#[test]
fn test_embedded_newline() {
let s = "<p>Hello\nWorld</p>";
let tree = parse_matrix_html(s);
let text = tree.to_text(15, Style::default(), true);
assert_eq!(text.lines.len(), 1);
assert_eq!(
text.lines[0],
Spans(vec![
Span::raw("Hello"),
Span::raw(" "),
Span::raw("World"),
Span::raw(" ")
])
);
}
#[test]
fn test_pre_tag() {
let s = concat!(
"<pre><code class=\"language-rust\">",
"fn hello() -&gt; usize {\n",
" return 5;\n",
"}\n",
"</code></pre>\n"
);
let tree = parse_matrix_html(s);
let text = tree.to_text(25, Style::default(), true);
assert_eq!(text.lines.len(), 5);
assert_eq!(
text.lines[0],
Spans(vec![
Span::raw(line::TOP_LEFT),
Span::raw(line::HORIZONTAL.repeat(23)),
Span::raw(line::TOP_RIGHT)
])
);
assert_eq!(
text.lines[1],
Spans(vec![
Span::raw(line::VERTICAL),
Span::raw("fn"),
Span::raw(" "),
Span::raw("hello"),
Span::raw("("),
Span::raw(")"),
Span::raw(" "),
Span::raw("-"),
Span::raw(">"),
Span::raw(" "),
Span::raw("usize"),
Span::raw(" "),
Span::raw("{"),
Span::raw(" "),
Span::raw(line::VERTICAL)
])
);
assert_eq!(
text.lines[2],
Spans(vec![
Span::raw(line::VERTICAL),
Span::raw(" "),
Span::raw("return"),
Span::raw(" "),
Span::raw("5"),
Span::raw(";"),
Span::raw(" "),
Span::raw(line::VERTICAL)
])
);
assert_eq!(
text.lines[3],
Spans(vec![
Span::raw(line::VERTICAL),
Span::raw("}"),
Span::raw(" ".repeat(22)),
Span::raw(line::VERTICAL)
])
);
assert_eq!(
text.lines[4],
Spans(vec![
Span::raw(line::BOTTOM_LEFT),
Span::raw(line::HORIZONTAL.repeat(23)),
Span::raw(line::BOTTOM_RIGHT)
])
);
} }
} }

View File

@@ -6,12 +6,18 @@ use std::convert::TryFrom;
use std::hash::{Hash, Hasher}; use std::hash::{Hash, Hasher};
use std::slice::Iter; use std::slice::Iter;
use chrono::{DateTime, NaiveDateTime, Utc}; use chrono::{DateTime, Local as LocalTz, NaiveDateTime, TimeZone};
use comrak::{markdown_to_html, ComrakOptions};
use unicode_width::UnicodeWidthStr; use unicode_width::UnicodeWidthStr;
use matrix_sdk::ruma::{ use matrix_sdk::ruma::{
events::{ events::{
room::{ room::{
encrypted::{
OriginalRoomEncryptedEvent,
RedactedRoomEncryptedEvent,
RoomEncryptedEvent,
},
message::{ message::{
FormattedBody, FormattedBody,
MessageFormat, MessageFormat,
@@ -21,10 +27,13 @@ use matrix_sdk::ruma::{
Relation, Relation,
RoomMessageEvent, RoomMessageEvent,
RoomMessageEventContent, RoomMessageEventContent,
TextMessageEventContent,
}, },
redaction::SyncRoomRedactionEvent, redaction::SyncRoomRedactionEvent,
}, },
AnyMessageLikeEvent,
Redact, Redact,
RedactedUnsigned,
}, },
EventId, EventId,
MilliSecondsSinceUnixEpoch, MilliSecondsSinceUnixEpoch,
@@ -52,7 +61,7 @@ use crate::{
mod html; mod html;
mod printer; mod printer;
pub type MessageFetchResult = IambResult<(Option<String>, Vec<RoomMessageEvent>)>; pub type MessageFetchResult = IambResult<(Option<String>, Vec<AnyMessageLikeEvent>)>;
pub type MessageKey = (MessageTimeStamp, OwnedEventId); pub type MessageKey = (MessageTimeStamp, OwnedEventId);
pub type Messages = BTreeMap<MessageKey, Message>; pub type Messages = BTreeMap<MessageKey, Message>;
@@ -68,6 +77,13 @@ const fn span_static(s: &'static str) -> Span<'static> {
} }
} }
const BOLD_STYLE: Style = Style {
fg: None,
bg: None,
add_modifier: StyleModifier::BOLD,
sub_modifier: StyleModifier::empty(),
};
const USER_GUTTER: usize = 30; const USER_GUTTER: usize = 30;
const TIME_GUTTER: usize = 12; const TIME_GUTTER: usize = 12;
const READ_GUTTER: usize = 5; const READ_GUTTER: usize = 5;
@@ -79,6 +95,28 @@ const USER_GUTTER_EMPTY_SPAN: Span<'static> = span_static(USER_GUTTER_EMPTY);
const TIME_GUTTER_EMPTY: &str = " "; const TIME_GUTTER_EMPTY: &str = " ";
const TIME_GUTTER_EMPTY_SPAN: Span<'static> = span_static(TIME_GUTTER_EMPTY); const TIME_GUTTER_EMPTY_SPAN: Span<'static> = span_static(TIME_GUTTER_EMPTY);
fn text_to_message_content(input: String) -> TextMessageEventContent {
let mut options = ComrakOptions::default();
options.extension.shortcodes = true;
options.render.hardbreaks = true;
let html = markdown_to_html(input.as_str(), &options);
TextMessageEventContent::html(input, html)
}
pub fn text_to_message(input: String) -> RoomMessageEventContent {
let msg = MessageType::Text(text_to_message_content(input));
RoomMessageEventContent::new(msg)
}
#[inline]
fn millis_to_datetime(ms: UInt) -> DateTime<LocalTz> {
let time = i64::from(ms) / 1000;
let time = NaiveDateTime::from_timestamp_opt(time, 0).unwrap_or_default();
LocalTz.from_utc_datetime(&time)
}
#[derive(thiserror::Error, Debug)] #[derive(thiserror::Error, Debug)]
pub enum TimeStampIntError { pub enum TimeStampIntError {
#[error("Integer conversion error: {0}")] #[error("Integer conversion error: {0}")]
@@ -95,14 +133,31 @@ pub enum MessageTimeStamp {
} }
impl MessageTimeStamp { impl MessageTimeStamp {
fn show(&self) -> Option<Span> { fn as_datetime(&self) -> DateTime<LocalTz> {
match self { match self {
MessageTimeStamp::OriginServer(ts) => { MessageTimeStamp::OriginServer(ms) => millis_to_datetime(*ms),
let time = i64::from(*ts) / 1000; MessageTimeStamp::LocalEcho => LocalTz::now(),
let time = NaiveDateTime::from_timestamp_opt(time, 0)?; }
let time = DateTime::<Utc>::from_utc(time, Utc); }
let time = time.format("%T");
let time = format!(" [{}]", time); fn same_day(&self, other: &Self) -> bool {
let dt1 = self.as_datetime();
let dt2 = other.as_datetime();
dt1.date_naive() == dt2.date_naive()
}
fn show_date(&self) -> Option<Span> {
let time = self.as_datetime().format("%A, %B %d %Y").to_string();
Span::styled(time, BOLD_STYLE).into()
}
fn show_time(&self) -> Option<Span> {
match self {
MessageTimeStamp::OriginServer(ms) => {
let time = millis_to_datetime(*ms).format("%T");
let time = format!(" [{time}]");
Span::raw(time).into() Span::raw(time).into()
}, },
@@ -139,6 +194,12 @@ impl PartialOrd for MessageTimeStamp {
} }
} }
impl From<UInt> for MessageTimeStamp {
fn from(millis: UInt) -> Self {
MessageTimeStamp::OriginServer(millis)
}
}
impl From<MilliSecondsSinceUnixEpoch> for MessageTimeStamp { impl From<MilliSecondsSinceUnixEpoch> for MessageTimeStamp {
fn from(millis: MilliSecondsSinceUnixEpoch) -> Self { fn from(millis: MilliSecondsSinceUnixEpoch) -> Self {
MessageTimeStamp::OriginServer(millis.0) MessageTimeStamp::OriginServer(millis.0)
@@ -168,7 +229,7 @@ impl TryFrom<usize> for MessageTimeStamp {
let n = u64::try_from(u)?; let n = u64::try_from(u)?;
let n = UInt::try_from(n).map_err(TimeStampIntError::UIntError)?; let n = UInt::try_from(n).map_err(TimeStampIntError::UIntError)?;
Ok(MessageTimeStamp::OriginServer(n)) Ok(MessageTimeStamp::from(n))
} }
} }
} }
@@ -279,6 +340,8 @@ impl PartialOrd for MessageCursor {
#[derive(Clone)] #[derive(Clone)]
pub enum MessageEvent { pub enum MessageEvent {
EncryptedOriginal(Box<OriginalRoomEncryptedEvent>),
EncryptedRedacted(Box<RedactedRoomEncryptedEvent>),
Original(Box<OriginalRoomMessageEvent>), Original(Box<OriginalRoomMessageEvent>),
Redacted(Box<RedactedRoomMessageEvent>), Redacted(Box<RedactedRoomMessageEvent>),
Local(OwnedEventId, Box<RoomMessageEventContent>), Local(OwnedEventId, Box<RoomMessageEventContent>),
@@ -287,35 +350,45 @@ pub enum MessageEvent {
impl MessageEvent { impl MessageEvent {
pub fn event_id(&self) -> &EventId { pub fn event_id(&self) -> &EventId {
match self { match self {
MessageEvent::EncryptedOriginal(ev) => ev.event_id.as_ref(),
MessageEvent::EncryptedRedacted(ev) => ev.event_id.as_ref(),
MessageEvent::Original(ev) => ev.event_id.as_ref(), MessageEvent::Original(ev) => ev.event_id.as_ref(),
MessageEvent::Redacted(ev) => ev.event_id.as_ref(), MessageEvent::Redacted(ev) => ev.event_id.as_ref(),
MessageEvent::Local(event_id, _) => event_id.as_ref(), MessageEvent::Local(event_id, _) => event_id.as_ref(),
} }
} }
pub fn content(&self) -> Option<&RoomMessageEventContent> {
match self {
MessageEvent::EncryptedOriginal(_) => None,
MessageEvent::Original(ev) => Some(&ev.content),
MessageEvent::EncryptedRedacted(_) => None,
MessageEvent::Redacted(_) => None,
MessageEvent::Local(_, content) => Some(content),
}
}
pub fn is_emote(&self) -> bool {
matches!(
self.content(),
Some(RoomMessageEventContent { msgtype: MessageType::Emote(_), .. })
)
}
pub fn body(&self) -> Cow<'_, str> { pub fn body(&self) -> Cow<'_, str> {
match self { match self {
MessageEvent::EncryptedOriginal(_) => "[Unable to decrypt message]".into(),
MessageEvent::Original(ev) => body_cow_content(&ev.content), MessageEvent::Original(ev) => body_cow_content(&ev.content),
MessageEvent::Redacted(ev) => { MessageEvent::EncryptedRedacted(ev) => body_cow_reason(&ev.unsigned),
let reason = ev MessageEvent::Redacted(ev) => body_cow_reason(&ev.unsigned),
.unsigned
.redacted_because
.as_ref()
.and_then(|e| e.as_original())
.and_then(|r| r.content.reason.as_ref());
if let Some(r) = reason {
Cow::Owned(format!("[Redacted: {:?}]", r))
} else {
Cow::Borrowed("[Redacted]")
}
},
MessageEvent::Local(_, content) => body_cow_content(content), MessageEvent::Local(_, content) => body_cow_content(content),
} }
} }
pub fn html(&self) -> Option<StyleTree> { pub fn html(&self) -> Option<StyleTree> {
let content = match self { let content = match self {
MessageEvent::EncryptedOriginal(_) => return None,
MessageEvent::EncryptedRedacted(_) => return None,
MessageEvent::Original(ev) => &ev.content, MessageEvent::Original(ev) => &ev.content,
MessageEvent::Redacted(_) => return None, MessageEvent::Redacted(_) => return None,
MessageEvent::Local(_, content) => content, MessageEvent::Local(_, content) => content,
@@ -332,8 +405,10 @@ impl MessageEvent {
} }
} }
pub fn redact(&mut self, redaction: SyncRoomRedactionEvent, version: &RoomVersionId) { fn redact(&mut self, redaction: SyncRoomRedactionEvent, version: &RoomVersionId) {
match self { match self {
MessageEvent::EncryptedOriginal(_) => return,
MessageEvent::EncryptedRedacted(_) => return,
MessageEvent::Redacted(_) => return, MessageEvent::Redacted(_) => return,
MessageEvent::Local(_, _) => return, MessageEvent::Local(_, _) => return,
MessageEvent::Original(ev) => { MessageEvent::Original(ev) => {
@@ -372,6 +447,20 @@ fn body_cow_content(content: &RoomMessageEventContent) -> Cow<'_, str> {
Cow::Borrowed(s) Cow::Borrowed(s)
} }
fn body_cow_reason(unsigned: &RedactedUnsigned) -> Cow<'_, str> {
let reason = unsigned
.redacted_because
.as_ref()
.and_then(|e| e.as_original())
.and_then(|r| r.content.reason.as_ref());
if let Some(r) = reason {
Cow::Owned(format!("[Redacted: {r:?}]"))
} else {
Cow::Borrowed("[Redacted]")
}
}
enum MessageColumns { enum MessageColumns {
/// Four columns: sender, message, timestamp, read receipts. /// Four columns: sender, message, timestamp, read receipts.
Four, Four,
@@ -388,10 +477,26 @@ enum MessageColumns {
struct MessageFormatter<'a> { struct MessageFormatter<'a> {
settings: &'a ApplicationSettings, settings: &'a ApplicationSettings,
/// How many columns to print.
cols: MessageColumns, cols: MessageColumns,
/// The full, original width.
orig: usize,
/// The width that the message contents need to fill.
fill: usize, fill: usize,
/// The formatted Span for the message sender.
user: Option<Span<'a>>, user: Option<Span<'a>>,
/// The time the message was sent.
time: Option<Span<'a>>, time: Option<Span<'a>>,
/// The date the message was sent.
date: Option<Span<'a>>,
/// Iterator over the users who have read up to this message.
read: Iter<'a, OwnedUserId>, read: Iter<'a, OwnedUserId>,
} }
@@ -402,6 +507,15 @@ impl<'a> MessageFormatter<'a> {
#[inline] #[inline]
fn push_spans(&mut self, spans: Spans<'a>, style: Style, text: &mut Text<'a>) { fn push_spans(&mut self, spans: Spans<'a>, style: Style, text: &mut Text<'a>) {
if let Some(date) = self.date.take() {
let len = date.content.as_ref().len();
let padding = self.orig.saturating_sub(len);
let leading = space_span(padding / 2, Style::default());
let trailing = space_span(padding.saturating_sub(padding / 2), Style::default());
text.lines.push(Spans(vec![leading, date, trailing]));
}
match self.cols { match self.cols {
MessageColumns::Four => { MessageColumns::Four => {
let settings = self.settings; let settings = self.settings;
@@ -484,6 +598,8 @@ impl Message {
pub fn reply_to(&self) -> Option<OwnedEventId> { pub fn reply_to(&self) -> Option<OwnedEventId> {
let content = match &self.event { let content = match &self.event {
MessageEvent::EncryptedOriginal(_) => return None,
MessageEvent::EncryptedRedacted(_) => return None,
MessageEvent::Local(_, content) => content, MessageEvent::Local(_, content) => content,
MessageEvent::Original(ev) => &ev.content, MessageEvent::Original(ev) => &ev.content,
MessageEvent::Redacted(_) => return None, MessageEvent::Redacted(_) => return None,
@@ -517,43 +633,49 @@ impl Message {
info: &'a RoomInfo, info: &'a RoomInfo,
settings: &'a ApplicationSettings, settings: &'a ApplicationSettings,
) -> MessageFormatter<'a> { ) -> MessageFormatter<'a> {
let orig = width;
let date = match &prev {
Some(prev) if prev.timestamp.same_day(&self.timestamp) => None,
_ => self.timestamp.show_date(),
};
if USER_GUTTER + TIME_GUTTER + READ_GUTTER + MIN_MSG_LEN <= width && if USER_GUTTER + TIME_GUTTER + READ_GUTTER + MIN_MSG_LEN <= width &&
settings.tunables.read_receipt_display settings.tunables.read_receipt_display
{ {
let cols = MessageColumns::Four; let cols = MessageColumns::Four;
let fill = width - USER_GUTTER - TIME_GUTTER - READ_GUTTER; let fill = width - USER_GUTTER - TIME_GUTTER - READ_GUTTER;
let user = self.show_sender(prev, true, settings); let user = self.show_sender(prev, true, info, settings);
let time = self.timestamp.show(); let time = self.timestamp.show_time();
let read = match info.receipts.get(self.event.event_id()) { let read = match info.receipts.get(self.event.event_id()) {
Some(read) => read.iter(), Some(read) => read.iter(),
None => [].iter(), None => [].iter(),
}; };
MessageFormatter { settings, cols, fill, user, time, read } MessageFormatter { settings, cols, orig, fill, user, date, time, read }
} else if USER_GUTTER + TIME_GUTTER + MIN_MSG_LEN <= width { } else if USER_GUTTER + TIME_GUTTER + MIN_MSG_LEN <= width {
let cols = MessageColumns::Three; let cols = MessageColumns::Three;
let fill = width - USER_GUTTER - TIME_GUTTER; let fill = width - USER_GUTTER - TIME_GUTTER;
let user = self.show_sender(prev, true, settings); let user = self.show_sender(prev, true, info, settings);
let time = self.timestamp.show(); let time = self.timestamp.show_time();
let read = [].iter(); let read = [].iter();
MessageFormatter { settings, cols, fill, user, time, read } MessageFormatter { settings, cols, orig, fill, user, date, time, read }
} else if USER_GUTTER + MIN_MSG_LEN <= width { } else if USER_GUTTER + MIN_MSG_LEN <= width {
let cols = MessageColumns::Two; let cols = MessageColumns::Two;
let fill = width - USER_GUTTER; let fill = width - USER_GUTTER;
let user = self.show_sender(prev, true, settings); let user = self.show_sender(prev, true, info, settings);
let time = None; let time = None;
let read = [].iter(); let read = [].iter();
MessageFormatter { settings, cols, fill, user, time, read } MessageFormatter { settings, cols, orig, fill, user, date, time, read }
} else { } else {
let cols = MessageColumns::One; let cols = MessageColumns::One;
let fill = width.saturating_sub(2); let fill = width.saturating_sub(2);
let user = self.show_sender(prev, false, settings); let user = self.show_sender(prev, false, info, settings);
let time = None; let time = None;
let read = [].iter(); let read = [].iter();
MessageFormatter { settings, cols, fill, user, time, read } MessageFormatter { settings, cols, orig, fill, user, date, time, read }
} }
} }
@@ -578,7 +700,7 @@ impl Message {
if let Some(r) = &reply { if let Some(r) = &reply {
let w = width.saturating_sub(2); let w = width.saturating_sub(2);
let mut replied = r.show_msg(w, style, true); let mut replied = r.show_msg(w, style, true);
let mut sender = r.sender_span(settings); let mut sender = r.sender_span(info, settings);
let sender_width = UnicodeWidthStr::width(sender.content.as_ref()); let sender_width = UnicodeWidthStr::width(sender.content.as_ref());
let trailing = w.saturating_sub(sender_width + 1); let trailing = w.saturating_sub(sender_width + 1);
@@ -613,6 +735,47 @@ impl Message {
fmt.push_spans(space_span(width, style).into(), style, &mut text); fmt.push_spans(space_span(width, style).into(), style, &mut text);
} }
if settings.tunables.reaction_display {
let mut emojis = printer::TextPrinter::new(width, style, false);
let mut reactions = 0;
for (key, count) in info.get_reactions(self.event.event_id()).into_iter() {
if reactions != 0 {
emojis.push_str(" ", style);
}
let name = if settings.tunables.reaction_shortcode_display {
if let Some(emoji) = emojis::get(key) {
if let Some(short) = emoji.shortcode() {
short
} else {
// No ASCII shortcode name to show.
continue;
}
} else if key.chars().all(|c| c.is_ascii_alphanumeric()) {
key
} else {
// Not an Emoji or a printable ASCII string.
continue;
}
} else {
key
};
emojis.push_str("[", style);
emojis.push_str(name, style);
emojis.push_str(" ", style);
emojis.push_span_nobreak(Span::styled(count.to_string(), style));
emojis.push_str("]", style);
reactions += 1;
}
if reactions > 0 {
fmt.push_text(emojis.finish(), style, &mut text);
}
}
return text; return text;
} }
@@ -630,23 +793,31 @@ impl Message {
} }
} }
fn sender_span(&self, settings: &ApplicationSettings) -> Span { fn sender_span<'a>(
settings.get_user_span(self.sender.as_ref()) &'a self,
info: &'a RoomInfo,
settings: &'a ApplicationSettings,
) -> Span<'a> {
settings.get_user_span(self.sender.as_ref(), info)
} }
fn show_sender( fn show_sender<'a>(
&self, &'a self,
prev: Option<&Message>, prev: Option<&Message>,
align_right: bool, align_right: bool,
settings: &ApplicationSettings, info: &'a RoomInfo,
) -> Option<Span> { settings: &'a ApplicationSettings,
let user = if matches!(prev, Some(prev) if self.sender == prev.sender) { ) -> Option<Span<'a>> {
if let Some(prev) = prev {
if self.sender == prev.sender &&
self.timestamp.same_day(&prev.timestamp) &&
!self.event.is_emote()
{
return None; return None;
} else { }
self.sender_span(settings) }
};
let Span { content, style } = user; let Span { content, style } = self.sender_span(info, settings);
let stop = content.len().min(28); let stop = content.len().min(28);
let s = &content[..stop]; let s = &content[..stop];
@@ -658,6 +829,24 @@ impl Message {
Span::styled(sender, style).into() Span::styled(sender, style).into()
} }
pub fn redact(&mut self, redaction: SyncRoomRedactionEvent, version: &RoomVersionId) {
self.event.redact(redaction, version);
self.html = None;
}
}
impl From<RoomEncryptedEvent> for Message {
fn from(event: RoomEncryptedEvent) -> Self {
let timestamp = event.origin_server_ts().into();
let user_id = event.sender().to_owned();
let content = match event {
RoomEncryptedEvent::Original(ev) => MessageEvent::EncryptedOriginal(ev.into()),
RoomEncryptedEvent::Redacted(ev) => MessageEvent::EncryptedRedacted(ev.into()),
};
Message::new(content, user_id, timestamp)
}
} }
impl From<OriginalRoomMessageEvent> for Message { impl From<OriginalRoomMessageEvent> for Message {
@@ -804,4 +993,53 @@ pub mod tests {
// MessageCursor::latest() should point at the most recent message after conversion. // MessageCursor::latest() should point at the most recent message after conversion.
assert_eq!(identity(&mc6), mc1); assert_eq!(identity(&mc6), mc1);
} }
#[test]
fn test_markdown_message() {
let input = "**bold**\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(content.formatted.unwrap().body, "<p><strong>bold</strong></p>\n");
let input = "*emphasis*\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(content.formatted.unwrap().body, "<p><em>emphasis</em></p>\n");
let input = "`code`\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(content.formatted.unwrap().body, "<p><code>code</code></p>\n");
let input = "```rust\nconst A: usize = 1;\n```\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(
content.formatted.unwrap().body,
"<pre><code class=\"language-rust\">const A: usize = 1;\n</code></pre>\n"
);
let input = ":heart:\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(content.formatted.unwrap().body, "<p>\u{2764}\u{FE0F}</p>\n");
let input = "para 1\n\npara 2\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(content.formatted.unwrap().body, "<p>para 1</p>\n<p>para 2</p>\n");
let input = "line 1\nline 2\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(content.formatted.unwrap().body, "<p>line 1<br />\nline 2</p>\n");
let input = "# Heading\n## Subheading\n\ntext\n";
let content = text_to_message_content(input.into());
assert_eq!(content.body, input);
assert_eq!(
content.formatted.unwrap().body,
"<h1>Heading</h1>\n<h2>Subheading</h2>\n<p>text</p>\n"
);
}
} }

View File

@@ -3,6 +3,7 @@ use std::borrow::Cow;
use modalkit::tui::layout::Alignment; use modalkit::tui::layout::Alignment;
use modalkit::tui::style::Style; use modalkit::tui::style::Style;
use modalkit::tui::text::{Span, Spans, Text}; use modalkit::tui::text::{Span, Spans, Text};
use unicode_segmentation::UnicodeSegmentation;
use unicode_width::UnicodeWidthStr; use unicode_width::UnicodeWidthStr;
use crate::util::{space_span, take_width}; use crate::util::{space_span, take_width};
@@ -16,6 +17,7 @@ pub struct TextPrinter<'a> {
alignment: Alignment, alignment: Alignment,
curr_spans: Vec<Span<'a>>, curr_spans: Vec<Span<'a>>,
curr_width: usize, curr_width: usize,
literal: bool,
} }
impl<'a> TextPrinter<'a> { impl<'a> TextPrinter<'a> {
@@ -29,6 +31,7 @@ impl<'a> TextPrinter<'a> {
alignment: Alignment::Left, alignment: Alignment::Left,
curr_spans: vec![], curr_spans: vec![],
curr_width: 0, curr_width: 0,
literal: false,
} }
} }
@@ -37,6 +40,11 @@ impl<'a> TextPrinter<'a> {
self self
} }
pub fn literal(mut self, literal: bool) -> Self {
self.literal = literal;
self
}
pub fn hide_reply(&self) -> bool { pub fn hide_reply(&self) -> bool {
self.hide_reply self.hide_reply
} }
@@ -55,6 +63,7 @@ impl<'a> TextPrinter<'a> {
alignment: self.alignment, alignment: self.alignment,
curr_spans: vec![], curr_spans: vec![],
curr_width: 0, curr_width: 0,
literal: self.literal,
} }
} }
@@ -107,7 +116,7 @@ impl<'a> TextPrinter<'a> {
self.push(); self.push();
} }
pub fn push_str<T>(&mut self, s: T, style: Style) fn push_str_wrapped<T>(&mut self, s: T, style: Style)
where where
T: Into<Cow<'a, str>>, T: Into<Cow<'a, str>>,
{ {
@@ -140,6 +149,69 @@ impl<'a> TextPrinter<'a> {
} }
} }
pub fn push_span_nobreak(&mut self, span: Span<'a>) {
let sw = UnicodeWidthStr::width(span.content.as_ref());
if self.curr_width + sw > self.width {
// Span doesn't fit on this line, so start a new one.
self.commit();
}
self.curr_spans.push(span);
self.curr_width += sw;
}
pub fn push_str(&mut self, s: &'a str, style: Style) {
let style = self.base_style.patch(style);
if self.width == 0 {
return;
}
for mut word in UnicodeSegmentation::split_word_bounds(s) {
if let "\n" | "\r\n" = word {
if self.literal {
self.commit();
continue;
}
// Render embedded newlines as spaces.
word = " ";
}
if !self.literal && self.curr_width == 0 && word.chars().all(char::is_whitespace) {
// Drop leading whitespace.
continue;
}
let sw = UnicodeWidthStr::width(word);
if sw > self.width {
self.push_str_wrapped(word, style);
continue;
}
if self.curr_width + sw > self.width {
// Word doesn't fit on this line, so start a new one.
self.commit();
if !self.literal && word.chars().all(char::is_whitespace) {
// Drop leading whitespace.
continue;
}
}
let span = Span::styled(word, style);
self.curr_spans.push(span);
self.curr_width += sw;
}
if self.curr_width == self.width {
// If the last bit fills the full line, start a new one.
self.push();
}
}
pub fn push_line(&mut self, spans: Spans<'a>) { pub fn push_line(&mut self, spans: Spans<'a>) {
self.commit(); self.commit();
self.text.lines.push(spans); self.text.lines.push(spans);

View File

@@ -17,10 +17,11 @@ use matrix_sdk::ruma::{
use lazy_static::lazy_static; use lazy_static::lazy_static;
use modalkit::tui::style::{Color, Style}; use modalkit::tui::style::{Color, Style};
use tokio::sync::mpsc::unbounded_channel; use tokio::sync::mpsc::unbounded_channel;
use tracing::Level;
use url::Url; use url::Url;
use crate::{ use crate::{
base::{ChatStore, ProgramStore, RoomFetchStatus, RoomInfo}, base::{ChatStore, EventLocation, ProgramStore, RoomFetchStatus, RoomInfo},
config::{ config::{
user_color, user_color,
user_style_from_color, user_style_from_color,
@@ -29,6 +30,7 @@ use crate::{
ProfileConfig, ProfileConfig,
TunableValues, TunableValues,
UserColor, UserColor,
UserDisplayStyle,
UserDisplayTunables, UserDisplayTunables,
}, },
message::{ message::{
@@ -41,6 +43,8 @@ use crate::{
worker::Requester, worker::Requester,
}; };
const TEST_ROOM1_ALIAS: &str = "#room1:example.com";
lazy_static! { lazy_static! {
pub static ref TEST_ROOM1_ID: OwnedRoomId = RoomId::new(server_name!("example.com")).to_owned(); pub static ref TEST_ROOM1_ID: OwnedRoomId = RoomId::new(server_name!("example.com")).to_owned();
pub static ref TEST_USER1: OwnedUserId = user_id!("@user1:example.com").to_owned(); pub static ref TEST_USER1: OwnedUserId = user_id!("@user1:example.com").to_owned();
@@ -117,14 +121,14 @@ pub fn mock_message5() -> Message {
mock_room1_message(content, TEST_USER2.clone(), MSG4_KEY.clone()) mock_room1_message(content, TEST_USER2.clone(), MSG4_KEY.clone())
} }
pub fn mock_keys() -> HashMap<OwnedEventId, MessageKey> { pub fn mock_keys() -> HashMap<OwnedEventId, EventLocation> {
let mut keys = HashMap::new(); let mut keys = HashMap::new();
keys.insert(MSG1_EVID.clone(), MSG1_KEY.clone()); keys.insert(MSG1_EVID.clone(), EventLocation::Message(MSG1_KEY.clone()));
keys.insert(MSG2_EVID.clone(), MSG2_KEY.clone()); keys.insert(MSG2_EVID.clone(), EventLocation::Message(MSG2_KEY.clone()));
keys.insert(MSG3_EVID.clone(), MSG3_KEY.clone()); keys.insert(MSG3_EVID.clone(), EventLocation::Message(MSG3_KEY.clone()));
keys.insert(MSG4_EVID.clone(), MSG4_KEY.clone()); keys.insert(MSG4_EVID.clone(), EventLocation::Message(MSG4_KEY.clone()));
keys.insert(MSG5_EVID.clone(), MSG5_KEY.clone()); keys.insert(MSG5_EVID.clone(), EventLocation::Message(MSG5_KEY.clone()));
keys keys
} }
@@ -151,10 +155,13 @@ pub fn mock_room() -> RoomInfo {
receipts: HashMap::new(), receipts: HashMap::new(),
read_till: None, read_till: None,
reactions: HashMap::new(),
fetching: false,
fetch_id: RoomFetchStatus::NotStarted, fetch_id: RoomFetchStatus::NotStarted,
fetch_last: None, fetch_last: None,
users_typing: None, users_typing: None,
display_names: HashMap::new(),
} }
} }
@@ -162,15 +169,19 @@ pub fn mock_dirs() -> DirectoryValues {
DirectoryValues { DirectoryValues {
cache: PathBuf::new(), cache: PathBuf::new(),
logs: PathBuf::new(), logs: PathBuf::new(),
downloads: PathBuf::new(), downloads: None,
} }
} }
pub fn mock_tunables() -> TunableValues { pub fn mock_tunables() -> TunableValues {
TunableValues { TunableValues {
default_room: None, default_room: None,
log_level: Level::INFO,
reaction_display: true,
reaction_shortcode_display: false,
read_receipt_send: true, read_receipt_send: true,
read_receipt_display: true, read_receipt_display: true,
request_timeout: 120,
typing_notice_send: true, typing_notice_send: true,
typing_notice_display: true, typing_notice_display: true,
users: vec![(TEST_USER5.clone(), UserDisplayTunables { users: vec![(TEST_USER5.clone(), UserDisplayTunables {
@@ -179,22 +190,28 @@ pub fn mock_tunables() -> TunableValues {
})] })]
.into_iter() .into_iter()
.collect::<HashMap<_, _>>(), .collect::<HashMap<_, _>>(),
open_command: None,
username_display: UserDisplayStyle::Username,
} }
} }
pub fn mock_settings() -> ApplicationSettings { pub fn mock_settings() -> ApplicationSettings {
ApplicationSettings { ApplicationSettings {
matrix_dir: PathBuf::new(), matrix_dir: PathBuf::new(),
layout_json: PathBuf::new(),
session_json: PathBuf::new(), session_json: PathBuf::new(),
profile_name: "test".into(), profile_name: "test".into(),
profile: ProfileConfig { profile: ProfileConfig {
user_id: user_id!("@user:example.com").to_owned(), user_id: user_id!("@user:example.com").to_owned(),
url: Url::parse("https://example.com").unwrap(), url: Url::parse("https://example.com").unwrap(),
settings: None, settings: None,
dirs: None, dirs: None,
layout: None,
}, },
tunables: mock_tunables(), tunables: mock_tunables(),
dirs: mock_dirs(), dirs: mock_dirs(),
layout: Default::default(),
} }
} }
@@ -205,10 +222,19 @@ pub async fn mock_store() -> ProgramStore {
let worker = Requester { tx, client }; let worker = Requester { tx, client };
let mut store = ChatStore::new(worker, mock_settings()); let mut store = ChatStore::new(worker, mock_settings());
// Add presence information.
store.presences.get_or_default(TEST_USER1.clone());
store.presences.get_or_default(TEST_USER2.clone());
store.presences.get_or_default(TEST_USER3.clone());
store.presences.get_or_default(TEST_USER4.clone());
store.presences.get_or_default(TEST_USER5.clone());
let room_id = TEST_ROOM1_ID.clone(); let room_id = TEST_ROOM1_ID.clone();
let info = mock_room(); let info = mock_room();
store.rooms.insert(room_id, info); store.rooms.insert(room_id.clone(), info);
store.names.insert(TEST_ROOM1_ALIAS.to_string(), room_id);
ProgramStore::new(store) ProgramStore::new(store)
} }

View File

@@ -1,5 +1,7 @@
use std::cmp::{Ord, Ordering, PartialOrd}; use std::cmp::{Ord, Ordering, PartialOrd};
use std::collections::hash_map::Entry; use std::ops::Deref;
use std::sync::Arc;
use std::time::{Duration, Instant};
use matrix_sdk::{ use matrix_sdk::{
encryption::verification::{format_emojis, SasVerification}, encryption::verification::{format_emojis, SasVerification},
@@ -10,7 +12,6 @@ use matrix_sdk::{
OwnedRoomId, OwnedRoomId,
RoomId, RoomId,
}, },
DisplayName,
}; };
use modalkit::tui::{ use modalkit::tui::{
@@ -45,7 +46,9 @@ use modalkit::{
ScrollStyle, ScrollStyle,
ViewportContext, ViewportContext,
WordStyle, WordStyle,
WriteFlags,
}, },
completion::CompletionList,
}, },
widgets::{ widgets::{
list::{List, ListCursor, ListItem, ListState}, list::{List, ListCursor, ListItem, ListState},
@@ -76,6 +79,10 @@ use self::{room::RoomState, welcome::WelcomeState};
pub mod room; pub mod room;
pub mod welcome; pub mod welcome;
type MatrixRoomInfo = Arc<(MatrixRoom, Option<Tags>)>;
const MEMBER_FETCH_DEBOUNCE: Duration = Duration::from_secs(5);
#[inline] #[inline]
fn bold_style() -> Style { fn bold_style() -> Style {
Style::default().add_modifier(StyleModifier::BOLD) Style::default().add_modifier(StyleModifier::BOLD)
@@ -168,7 +175,7 @@ fn append_tags<'a>(tags: &'a Tags, spans: &mut Vec<Span<'a>>, style: Style) {
spans.push(Span::styled("User Tag: ", style)); spans.push(Span::styled("User Tag: ", style));
spans.push(Span::styled(tag.as_ref(), style)); spans.push(Span::styled(tag.as_ref(), style));
}, },
tag => spans.push(Span::styled(format!("{:?}", tag), style)), tag => spans.push(Span::styled(format!("{tag:?}"), style)),
} }
} }
@@ -195,7 +202,7 @@ fn room_prompt(
Err(err) Err(err)
}, },
PromptAction::Recall(_, _) => { PromptAction::Recall(..) => {
let msg = "Cannot recall history inside a list"; let msg = "Cannot recall history inside a list";
let err = EditError::Failure(msg.into()); let err = EditError::Failure(msg.into());
@@ -210,7 +217,7 @@ macro_rules! delegate {
match $s { match $s {
IambWindow::Room($id) => $e, IambWindow::Room($id) => $e,
IambWindow::DirectList($id) => $e, IambWindow::DirectList($id) => $e,
IambWindow::MemberList($id, _) => $e, IambWindow::MemberList($id, _, _) => $e,
IambWindow::RoomList($id) => $e, IambWindow::RoomList($id) => $e,
IambWindow::SpaceList($id) => $e, IambWindow::SpaceList($id) => $e,
IambWindow::VerifyList($id) => $e, IambWindow::VerifyList($id) => $e,
@@ -221,7 +228,7 @@ macro_rules! delegate {
pub enum IambWindow { pub enum IambWindow {
DirectList(DirectListState), DirectList(DirectListState),
MemberList(MemberListState, OwnedRoomId), MemberList(MemberListState, OwnedRoomId, Option<Instant>),
Room(RoomState), Room(RoomState),
VerifyList(VerifyListState), VerifyList(VerifyListState),
RoomList(RoomListState), RoomList(RoomListState),
@@ -376,10 +383,13 @@ impl WindowOps<IambInfo> for IambWindow {
match self { match self {
IambWindow::Room(state) => state.draw(area, buf, focused, store), IambWindow::Room(state) => state.draw(area, buf, focused, store),
IambWindow::DirectList(state) => { IambWindow::DirectList(state) => {
let dms = store.application.worker.direct_messages(); let mut items = store
let mut items = dms .application
.sync_info
.dms
.clone()
.into_iter() .into_iter()
.map(|(id, name, tags)| DirectItem::new(id, name, tags, store)) .map(|room_info| DirectItem::new(room_info, store))
.collect::<Vec<_>>(); .collect::<Vec<_>>();
items.sort(); items.sort();
@@ -391,10 +401,18 @@ impl WindowOps<IambInfo> for IambWindow {
.focus(focused) .focus(focused)
.render(area, buf, state); .render(area, buf, state);
}, },
IambWindow::MemberList(state, room_id) => { IambWindow::MemberList(state, room_id, last_fetch) => {
let need_fetch = match last_fetch {
Some(i) => i.elapsed() >= MEMBER_FETCH_DEBOUNCE,
None => true,
};
if need_fetch {
if let Ok(mems) = store.application.worker.members(room_id.clone()) { if let Ok(mems) = store.application.worker.members(room_id.clone()) {
let items = mems.into_iter().map(MemberItem::new); let items = mems.into_iter().map(|m| MemberItem::new(m, room_id.clone()));
state.set(items.collect()); state.set(items.collect());
*last_fetch = Some(Instant::now());
}
} }
List::new(store) List::new(store)
@@ -404,10 +422,13 @@ impl WindowOps<IambInfo> for IambWindow {
.render(area, buf, state); .render(area, buf, state);
}, },
IambWindow::RoomList(state) => { IambWindow::RoomList(state) => {
let joined = store.application.worker.active_rooms(); let mut items = store
let mut items = joined .application
.sync_info
.rooms
.clone()
.into_iter() .into_iter()
.map(|(room, name, tags)| RoomItem::new(room, name, tags, store)) .map(|room_info| RoomItem::new(room_info, store))
.collect::<Vec<_>>(); .collect::<Vec<_>>();
items.sort(); items.sort();
@@ -420,9 +441,13 @@ impl WindowOps<IambInfo> for IambWindow {
.render(area, buf, state); .render(area, buf, state);
}, },
IambWindow::SpaceList(state) => { IambWindow::SpaceList(state) => {
let spaces = store.application.worker.spaces(); let items = store
let items = .application
spaces.into_iter().map(|(room, name)| SpaceItem::new(room, name, store)); .sync_info
.spaces
.clone()
.into_iter()
.map(|room| SpaceItem::new(room, store));
state.set(items.collect()); state.set(items.collect());
state.draw(area, buf, focused, store); state.draw(area, buf, focused, store);
@@ -455,8 +480,8 @@ impl WindowOps<IambInfo> for IambWindow {
match self { match self {
IambWindow::Room(w) => w.dup(store).into(), IambWindow::Room(w) => w.dup(store).into(),
IambWindow::DirectList(w) => w.dup(store).into(), IambWindow::DirectList(w) => w.dup(store).into(),
IambWindow::MemberList(w, room_id) => { IambWindow::MemberList(w, room_id, last_fetch) => {
IambWindow::MemberList(w.dup(store), room_id.clone()) IambWindow::MemberList(w.dup(store), room_id.clone(), *last_fetch)
}, },
IambWindow::RoomList(w) => w.dup(store).into(), IambWindow::RoomList(w) => w.dup(store).into(),
IambWindow::SpaceList(w) => w.dup(store).into(), IambWindow::SpaceList(w) => w.dup(store).into(),
@@ -469,6 +494,19 @@ impl WindowOps<IambInfo> for IambWindow {
delegate!(self, w => w.close(flags, store)) delegate!(self, w => w.close(flags, store))
} }
fn write(
&mut self,
path: Option<&str>,
flags: WriteFlags,
store: &mut ProgramStore,
) -> IambResult<EditInfo> {
delegate!(self, w => w.write(path, flags, store))
}
fn get_completions(&self) -> Option<CompletionList> {
delegate!(self, w => w.get_completions())
}
fn get_cursor_word(&self, style: &WordStyle) -> Option<String> { fn get_cursor_word(&self, style: &WordStyle) -> Option<String> {
delegate!(self, w => w.get_cursor_word(style)) delegate!(self, w => w.get_cursor_word(style))
} }
@@ -483,7 +521,7 @@ impl Window<IambInfo> for IambWindow {
match self { match self {
IambWindow::Room(room) => IambId::Room(room.id().to_owned()), IambWindow::Room(room) => IambId::Room(room.id().to_owned()),
IambWindow::DirectList(_) => IambId::DirectList, IambWindow::DirectList(_) => IambId::DirectList,
IambWindow::MemberList(_, room_id) => IambId::MemberList(room_id.clone()), IambWindow::MemberList(_, room_id, _) => IambId::MemberList(room_id.clone()),
IambWindow::RoomList(_) => IambId::RoomList, IambWindow::RoomList(_) => IambId::RoomList,
IambWindow::SpaceList(_) => IambId::SpaceList, IambWindow::SpaceList(_) => IambId::SpaceList,
IambWindow::VerifyList(_) => IambId::VerifyList, IambWindow::VerifyList(_) => IambId::VerifyList,
@@ -504,10 +542,15 @@ impl Window<IambInfo> for IambWindow {
Spans::from(title) Spans::from(title)
}, },
IambWindow::MemberList(_, room_id) => { IambWindow::MemberList(state, room_id, _) => {
let title = store.application.get_room_title(room_id.as_ref()); let title = store.application.get_room_title(room_id.as_ref());
let n = state.len();
Spans(vec![bold_span("Room Members: "), title.into()]) let v = vec![
bold_span("Room Members "),
Span::styled(format!("({n}): "), bold_style()),
title.into(),
];
Spans(v)
}, },
} }
} }
@@ -521,10 +564,15 @@ impl Window<IambInfo> for IambWindow {
IambWindow::Welcome(_) => bold_spans("Welcome to iamb"), IambWindow::Welcome(_) => bold_spans("Welcome to iamb"),
IambWindow::Room(w) => w.get_title(store), IambWindow::Room(w) => w.get_title(store),
IambWindow::MemberList(_, room_id) => { IambWindow::MemberList(state, room_id, _) => {
let title = store.application.get_room_title(room_id.as_ref()); let title = store.application.get_room_title(room_id.as_ref());
let n = state.len();
Spans(vec![bold_span("Room Members: "), title.into()]) let v = vec![
bold_span("Room Members "),
Span::styled(format!("({n}): "), bold_style()),
title.into(),
];
Spans(v)
}, },
} }
} }
@@ -545,7 +593,7 @@ impl Window<IambInfo> for IambWindow {
IambId::MemberList(room_id) => { IambId::MemberList(room_id) => {
let id = IambBufferId::MemberList(room_id.clone()); let id = IambBufferId::MemberList(room_id.clone());
let list = MemberListState::new(id, vec![]); let list = MemberListState::new(id, vec![]);
let win = IambWindow::MemberList(list, room_id); let win = IambWindow::MemberList(list, room_id, None);
return Ok(win); return Ok(win);
}, },
@@ -575,26 +623,23 @@ impl Window<IambInfo> for IambWindow {
fn find(name: String, store: &mut ProgramStore) -> IambResult<Self> { fn find(name: String, store: &mut ProgramStore) -> IambResult<Self> {
let ChatStore { names, worker, .. } = &mut store.application; let ChatStore { names, worker, .. } = &mut store.application;
match names.entry(name) { if let Some(room) = names.get_mut(&name) {
Entry::Vacant(v) => { let id = IambId::Room(room.clone());
let room_id = worker.join_room(v.key().to_string())?;
v.insert(room_id.clone()); IambWindow::open(id, store)
} else {
let room_id = worker.join_room(name.clone())?;
names.insert(name, room_id.clone());
let (room, name, tags) = store.application.worker.get_room(room_id)?; let (room, name, tags) = store.application.worker.get_room(room_id)?;
let room = RoomState::new(room, name, tags, store); let room = RoomState::new(room, name, tags, store);
Ok(room.into()) Ok(room.into())
},
Entry::Occupied(o) => {
let id = IambId::Room(o.get().clone());
IambWindow::open(id, store)
},
} }
} }
fn posn(index: usize, _: &mut ProgramStore) -> IambResult<Self> { fn posn(index: usize, _: &mut ProgramStore) -> IambResult<Self> {
let msg = format!("Cannot find indexed buffer (index = {})", index); let msg = format!("Cannot find indexed buffer (index = {index})");
let err = UIError::Unimplemented(msg); let err = UIError::Unimplemented(msg);
Err(err) Err(err)
@@ -607,31 +652,45 @@ impl Window<IambInfo> for IambWindow {
#[derive(Clone)] #[derive(Clone)]
pub struct RoomItem { pub struct RoomItem {
room: MatrixRoom, room_info: MatrixRoomInfo,
tags: Option<Tags>,
name: String, name: String,
} }
impl RoomItem { impl RoomItem {
fn new( fn new(room_info: MatrixRoomInfo, store: &mut ProgramStore) -> Self {
room: MatrixRoom, let room = &room_info.deref().0;
name: DisplayName, let room_id = room.room_id();
tags: Option<Tags>,
store: &mut ProgramStore,
) -> Self {
let name = name.to_string();
let info = store.application.get_room_info(room.room_id().to_owned()); let info = store.application.get_room_info(room_id.to_owned());
info.name = name.clone().into(); let name = info.name.clone().unwrap_or_default();
info.tags = tags.clone(); info.tags = room_info.deref().1.clone();
RoomItem { room, tags, name } if let Some(alias) = room.canonical_alias() {
store.application.names.insert(alias.to_string(), room_id.to_owned());
}
RoomItem { room_info, name }
}
#[inline]
fn room(&self) -> &MatrixRoom {
&self.room_info.deref().0
}
#[inline]
fn room_id(&self) -> &RoomId {
self.room().room_id()
}
#[inline]
fn tags(&self) -> &Option<Tags> {
&self.room_info.deref().1
} }
} }
impl PartialEq for RoomItem { impl PartialEq for RoomItem {
fn eq(&self, other: &Self) -> bool { fn eq(&self, other: &Self) -> bool {
self.room.room_id() == other.room.room_id() self.room_id() == other.room_id()
} }
} }
@@ -639,7 +698,7 @@ impl Eq for RoomItem {}
impl Ord for RoomItem { impl Ord for RoomItem {
fn cmp(&self, other: &Self) -> Ordering { fn cmp(&self, other: &Self) -> Ordering {
tag_cmp(&self.tags, &other.tags).then_with(|| room_cmp(&self.room, &other.room)) tag_cmp(self.tags(), other.tags()).then_with(|| room_cmp(self.room(), other.room()))
} }
} }
@@ -657,7 +716,7 @@ impl ToString for RoomItem {
impl ListItem<IambInfo> for RoomItem { impl ListItem<IambInfo> for RoomItem {
fn show(&self, selected: bool, _: &ViewportContext<ListCursor>, _: &mut ProgramStore) -> Text { fn show(&self, selected: bool, _: &ViewportContext<ListCursor>, _: &mut ProgramStore) -> Text {
if let Some(tags) = &self.tags { if let Some(tags) = &self.tags() {
let style = selected_style(selected); let style = selected_style(selected);
let mut spans = vec![Span::styled(self.name.as_str(), style)]; let mut spans = vec![Span::styled(self.name.as_str(), style)];
@@ -670,7 +729,7 @@ impl ListItem<IambInfo> for RoomItem {
} }
fn get_word(&self) -> Option<String> { fn get_word(&self) -> Option<String> {
self.room.room_id().to_string().into() self.room_id().to_string().into()
} }
} }
@@ -681,29 +740,37 @@ impl Promptable<ProgramContext, ProgramStore, IambInfo> for RoomItem {
ctx: &ProgramContext, ctx: &ProgramContext,
_: &mut ProgramStore, _: &mut ProgramStore,
) -> EditResult<Vec<(ProgramAction, ProgramContext)>, IambInfo> { ) -> EditResult<Vec<(ProgramAction, ProgramContext)>, IambInfo> {
room_prompt(self.room.room_id(), act, ctx) room_prompt(self.room_id(), act, ctx)
} }
} }
#[derive(Clone)] #[derive(Clone)]
pub struct DirectItem { pub struct DirectItem {
room: MatrixRoom, room_info: MatrixRoomInfo,
tags: Option<Tags>,
name: String, name: String,
} }
impl DirectItem { impl DirectItem {
fn new( fn new(room_info: MatrixRoomInfo, store: &mut ProgramStore) -> Self {
room: MatrixRoom, let room_id = room_info.deref().0.room_id().to_owned();
name: DisplayName, let name = store.application.get_room_info(room_id).name.clone().unwrap_or_default();
tags: Option<Tags>,
store: &mut ProgramStore,
) -> Self {
let name = name.to_string();
store.application.set_room_name(room.room_id(), name.as_str()); DirectItem { room_info, name }
}
DirectItem { room, tags, name } #[inline]
fn room(&self) -> &MatrixRoom {
&self.room_info.deref().0
}
#[inline]
fn room_id(&self) -> &RoomId {
self.room().room_id()
}
#[inline]
fn tags(&self) -> &Option<Tags> {
&self.room_info.deref().1
} }
} }
@@ -715,7 +782,7 @@ impl ToString for DirectItem {
impl ListItem<IambInfo> for DirectItem { impl ListItem<IambInfo> for DirectItem {
fn show(&self, selected: bool, _: &ViewportContext<ListCursor>, _: &mut ProgramStore) -> Text { fn show(&self, selected: bool, _: &ViewportContext<ListCursor>, _: &mut ProgramStore) -> Text {
if let Some(tags) = &self.tags { if let Some(tags) = &self.tags() {
let style = selected_style(selected); let style = selected_style(selected);
let mut spans = vec![Span::styled(self.name.as_str(), style)]; let mut spans = vec![Span::styled(self.name.as_str(), style)];
@@ -728,13 +795,13 @@ impl ListItem<IambInfo> for DirectItem {
} }
fn get_word(&self) -> Option<String> { fn get_word(&self) -> Option<String> {
self.room.room_id().to_string().into() self.room_id().to_string().into()
} }
} }
impl PartialEq for DirectItem { impl PartialEq for DirectItem {
fn eq(&self, other: &Self) -> bool { fn eq(&self, other: &Self) -> bool {
self.room.room_id() == other.room.room_id() self.room_id() == other.room_id()
} }
} }
@@ -742,7 +809,7 @@ impl Eq for DirectItem {}
impl Ord for DirectItem { impl Ord for DirectItem {
fn cmp(&self, other: &Self) -> Ordering { fn cmp(&self, other: &Self) -> Ordering {
tag_cmp(&self.tags, &other.tags).then_with(|| room_cmp(&self.room, &other.room)) tag_cmp(self.tags(), other.tags()).then_with(|| room_cmp(self.room(), other.room()))
} }
} }
@@ -759,7 +826,7 @@ impl Promptable<ProgramContext, ProgramStore, IambInfo> for DirectItem {
ctx: &ProgramContext, ctx: &ProgramContext,
_: &mut ProgramStore, _: &mut ProgramStore,
) -> EditResult<Vec<(ProgramAction, ProgramContext)>, IambInfo> { ) -> EditResult<Vec<(ProgramAction, ProgramContext)>, IambInfo> {
room_prompt(self.room.room_id(), act, ctx) room_prompt(self.room_id(), act, ctx)
} }
} }
@@ -770,10 +837,18 @@ pub struct SpaceItem {
} }
impl SpaceItem { impl SpaceItem {
fn new(room: MatrixRoom, name: DisplayName, store: &mut ProgramStore) -> Self { fn new(room: MatrixRoom, store: &mut ProgramStore) -> Self {
let name = name.to_string(); let room_id = room.room_id();
let name = store
.application
.get_room_info(room_id.to_owned())
.name
.clone()
.unwrap_or_default();
store.application.set_room_name(room.room_id(), name.as_str()); if let Some(alias) = room.canonical_alias() {
store.application.names.insert(alias.to_string(), room_id.to_owned());
}
SpaceItem { room, name } SpaceItem { room, name }
} }
@@ -852,7 +927,7 @@ impl VerifyItem {
let device = self.sasv1.other_device(); let device = self.sasv1.other_device();
if let Some(display_name) = device.display_name() { if let Some(display_name) = device.display_name() {
format!("Device verification with {} ({})", display_name, state) format!("Device verification with {display_name} ({state})")
} else { } else {
format!("Device verification with device {} ({})", device.device_id(), state) format!("Device verification with device {} ({})", device.device_id(), state)
} }
@@ -958,7 +1033,7 @@ impl ListItem<IambInfo> for VerifyItem {
lines.push(Spans::from("")); lines.push(Spans::from(""));
for line in format_emojis(emoji).lines() { for line in format_emojis(emoji).lines() {
lines.push(Spans::from(format!(" {}", line))); lines.push(Spans::from(format!(" {line}")));
} }
lines.push(Spans::from("")); lines.push(Spans::from(""));
@@ -1011,7 +1086,7 @@ impl Promptable<ProgramContext, ProgramStore, IambInfo> for VerifyItem {
Err(err) Err(err)
}, },
PromptAction::Recall(_, _) => { PromptAction::Recall(..) => {
let msg = "Cannot recall history inside a list"; let msg = "Cannot recall history inside a list";
let err = EditError::Failure(msg.into()); let err = EditError::Failure(msg.into());
@@ -1025,11 +1100,12 @@ impl Promptable<ProgramContext, ProgramStore, IambInfo> for VerifyItem {
#[derive(Clone)] #[derive(Clone)]
pub struct MemberItem { pub struct MemberItem {
member: RoomMember, member: RoomMember,
room_id: OwnedRoomId,
} }
impl MemberItem { impl MemberItem {
fn new(member: RoomMember) -> Self { fn new(member: RoomMember, room_id: OwnedRoomId) -> Self {
Self { member } Self { member, room_id }
} }
} }
@@ -1046,12 +1122,32 @@ impl ListItem<IambInfo> for MemberItem {
_: &ViewportContext<ListCursor>, _: &ViewportContext<ListCursor>,
store: &mut ProgramStore, store: &mut ProgramStore,
) -> Text { ) -> Text {
let mut user = store.application.settings.get_user_span(self.member.user_id()); let info = store.application.rooms.get_or_default(self.room_id.clone());
let user_id = self.member.user_id();
let (color, name) = store.application.settings.get_user_overrides(self.member.user_id());
let color = color.unwrap_or_else(|| super::config::user_color(user_id.as_str()));
let mut style = super::config::user_style_from_color(color);
if selected { if selected {
user.style = user.style.add_modifier(StyleModifier::REVERSED); style = style.add_modifier(StyleModifier::REVERSED);
} }
let mut spans = vec![];
let mut parens = false;
if let Some(name) = name {
spans.push(Span::styled(name, style));
parens = true;
} else if let Some(display) = info.display_names.get(user_id) {
spans.push(Span::styled(display.clone(), style));
parens = true;
}
spans.extend(parens.then_some(Span::styled(" (", style)));
spans.push(Span::styled(user_id.as_str(), style));
spans.extend(parens.then_some(Span::styled(")", style)));
let state = match self.member.membership() { let state = match self.member.membership() {
MembershipState::Ban => Span::raw(" (banned)").into(), MembershipState::Ban => Span::raw(" (banned)").into(),
MembershipState::Invite => Span::raw(" (invited)").into(), MembershipState::Invite => Span::raw(" (invited)").into(),
@@ -1061,11 +1157,9 @@ impl ListItem<IambInfo> for MemberItem {
_ => None, _ => None,
}; };
if let Some(state) = state { spans.extend(state);
Spans(vec![user, state]).into()
} else { return Spans(spans).into();
user.into()
}
} }
fn get_word(&self) -> Option<String> { fn get_word(&self) -> Option<String> {
@@ -1088,7 +1182,7 @@ impl Promptable<ProgramContext, ProgramStore, IambInfo> for MemberItem {
Err(err) Err(err)
}, },
PromptAction::Recall(_, _) => { PromptAction::Recall(..) => {
let msg = "Cannot recall history inside a list"; let msg = "Cannot recall history inside a list";
let err = EditError::Failure(msg.into()); let err = EditError::Failure(msg.into());

View File

@@ -1,14 +1,19 @@
use std::borrow::Cow; use std::borrow::Cow;
use std::ffi::OsStr; use std::ffi::{OsStr, OsString};
use std::fs; use std::fs;
use std::ops::Deref; use std::ops::Deref;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use modalkit::editing::store::RegisterError;
use std::process::Command;
use tokio;
use matrix_sdk::{ use matrix_sdk::{
attachment::AttachmentConfig, attachment::AttachmentConfig,
media::{MediaFormat, MediaRequest}, media::{MediaFormat, MediaRequest},
room::Room as MatrixRoom, room::{Joined, Room as MatrixRoom},
ruma::{ ruma::{
events::reaction::{ReactionEventContent, Relation as Reaction},
events::room::message::{ events::room::message::{
MessageType, MessageType,
OriginalRoomMessageEvent, OriginalRoomMessageEvent,
@@ -17,12 +22,14 @@ use matrix_sdk::{
RoomMessageEventContent, RoomMessageEventContent,
TextMessageEventContent, TextMessageEventContent,
}, },
EventId,
OwnedRoomId, OwnedRoomId,
RoomId, RoomId,
}, },
}; };
use modalkit::{ use modalkit::{
input::dialog::PromptYesNo,
tui::{ tui::{
buffer::Buffer, buffer::Buffer,
layout::Rect, layout::Rect,
@@ -36,6 +43,7 @@ use modalkit::{
use modalkit::editing::{ use modalkit::editing::{
action::{ action::{
Action,
EditError, EditError,
EditInfo, EditInfo,
EditResult, EditResult,
@@ -48,13 +56,15 @@ use modalkit::editing::{
Scrollable, Scrollable,
UIError, UIError,
}, },
base::{CloseFlags, Count, MoveDir1D, PositionList, ScrollStyle, WordStyle}, base::{CloseFlags, Count, MoveDir1D, PositionList, ScrollStyle, WordStyle, WriteFlags},
completion::CompletionList,
context::Resolve, context::Resolve,
history::{self, HistoryList}, history::{self, HistoryList},
rope::EditRope, rope::EditRope,
}; };
use crate::base::{ use crate::base::{
DownloadFlags,
IambAction, IambAction,
IambBufferId, IambBufferId,
IambError, IambError,
@@ -69,7 +79,8 @@ use crate::base::{
SendAction, SendAction,
}; };
use crate::message::{Message, MessageEvent, MessageKey, MessageTimeStamp}; use crate::message::{text_to_message, Message, MessageEvent, MessageKey, MessageTimeStamp};
use crate::worker::Requester;
use super::scrollback::{Scrollback, ScrollbackState}; use super::scrollback::{Scrollback, ScrollbackState};
@@ -112,6 +123,10 @@ impl ChatState {
} }
} }
fn get_joined(&self, worker: &Requester) -> Result<Joined, IambError> {
worker.client.get_joined_room(self.id()).ok_or(IambError::NotJoined)
}
fn get_reply_to<'a>(&self, info: &'a RoomInfo) -> Option<&'a OriginalRoomMessageEvent> { fn get_reply_to<'a>(&self, info: &'a RoomInfo) -> Option<&'a OriginalRoomMessageEvent> {
let key = self.reply_to.as_ref()?; let key = self.reply_to.as_ref()?;
let msg = info.messages.get(key)?; let msg = info.messages.get(key)?;
@@ -144,74 +159,76 @@ impl ChatState {
let client = &store.application.worker.client; let client = &store.application.worker.client;
let settings = &store.application.settings; let settings = &store.application.settings;
let info = store.application.rooms.entry(self.room_id.clone()).or_default(); let info = store.application.rooms.get_or_default(self.room_id.clone());
let msg = self.scrollback.get_mut(info).ok_or(IambError::NoSelectedMessage)?; let msg = self
.scrollback
.get_mut(&mut info.messages)
.ok_or(IambError::NoSelectedMessage)?;
match act { match act {
MessageAction::Cancel => { MessageAction::Cancel(skip_confirm) => {
self.reply_to = None; self.reply_to = None;
self.editing = None; self.editing = None;
Ok(None) if skip_confirm {
return Ok(None);
}
let msg = "Would you like to clear the message bar?";
let act = PromptAction::Abort(false);
let prompt = PromptYesNo::new(msg, vec![Action::from(act)]);
let prompt = Box::new(prompt);
Err(UIError::NeedConfirm(prompt))
}, },
MessageAction::Download(filename, force) => { MessageAction::Download(filename, flags) => {
if let MessageEvent::Original(ev) = &msg.event { if let MessageEvent::Original(ev) = &msg.event {
let media = client.media(); let media = client.media();
let mut filename = match filename { let mut filename = match (filename, &settings.dirs.downloads) {
Some(f) => PathBuf::from(f), (Some(f), _) => PathBuf::from(f),
None => settings.dirs.downloads.clone(), (None, Some(downloads)) => downloads.clone(),
(None, None) => return Err(IambError::NoDownloadDir.into()),
}; };
let source = match &ev.content.msgtype { let (source, msg_filename) = match &ev.content.msgtype {
MessageType::Audio(c) => { MessageType::Audio(c) => (c.source.clone(), c.body.as_str()),
if filename.is_dir() {
filename.push(c.body.as_str());
}
c.source.clone()
},
MessageType::File(c) => { MessageType::File(c) => {
if filename.is_dir() { (c.source.clone(), c.filename.as_deref().unwrap_or(c.body.as_str()))
if let Some(name) = &c.filename {
filename.push(name);
} else {
filename.push(c.body.as_str());
}
}
c.source.clone()
},
MessageType::Image(c) => {
if filename.is_dir() {
filename.push(c.body.as_str());
}
c.source.clone()
},
MessageType::Video(c) => {
if filename.is_dir() {
filename.push(c.body.as_str());
}
c.source.clone()
}, },
MessageType::Image(c) => (c.source.clone(), c.body.as_str()),
MessageType::Video(c) => (c.source.clone(), c.body.as_str()),
_ => { _ => {
return Err(IambError::NoAttachment.into()); return Err(IambError::NoAttachment.into());
}, },
}; };
if !force && filename.exists() { if filename.is_dir() {
let msg = format!( filename.push(msg_filename);
"The file {} already exists; use :download! to overwrite it.",
filename.display()
);
let err = UIError::Failure(msg);
return Err(err);
} }
if filename.exists() && !flags.contains(DownloadFlags::FORCE) {
// Find an incrementally suffixed filename, e.g. image-2.jpg -> image-3.jpg
if let Some(stem) = filename.file_stem().and_then(OsStr::to_str) {
let ext = filename.extension();
let mut filename_incr = filename.clone();
for n in 1..=1000 {
if let Some(ext) = ext.and_then(OsStr::to_str) {
filename_incr.set_file_name(format!("{}-{}.{}", stem, n, ext));
} else {
filename_incr.set_file_name(format!("{}-{}", stem, n));
}
if !filename_incr.exists() {
filename = filename_incr;
break;
}
}
}
}
if !filename.exists() || flags.contains(DownloadFlags::FORCE) {
let req = MediaRequest { source, format: MediaFormat::File }; let req = MediaRequest { source, format: MediaFormat::File };
let bytes = let bytes =
@@ -220,11 +237,38 @@ impl ChatState {
fs::write(filename.as_path(), bytes.as_slice())?; fs::write(filename.as_path(), bytes.as_slice())?;
msg.downloaded = true; msg.downloaded = true;
} else if !flags.contains(DownloadFlags::OPEN) {
let msg = format!(
"The file {} already exists; add ! to end of command to overwrite it.",
filename.display()
);
let err = UIError::Failure(msg);
let info = InfoMessage::from(format!( return Err(err);
}
let info = if flags.contains(DownloadFlags::OPEN) {
let target = filename.clone().into_os_string();
match open_command(
store.application.settings.tunables.open_command.as_ref(),
target,
) {
Ok(_) => {
InfoMessage::from(format!(
"Attachment downloaded to {} and opened",
filename.display()
))
},
Err(err) => {
return Err(err);
},
}
} else {
InfoMessage::from(format!(
"Attachment downloaded to {}", "Attachment downloaded to {}",
filename.display() filename.display()
)); ))
};
return Ok(info.into()); return Ok(info.into());
} }
@@ -261,24 +305,51 @@ impl ChatState {
}; };
self.tbox.set_text(text); self.tbox.set_text(text);
self.reply_to = msg.reply_to().and_then(|id| info.get_message_key(&id)).cloned();
self.editing = self.scrollback.get_key(info); self.editing = self.scrollback.get_key(info);
self.focus = RoomFocus::MessageBar; self.focus = RoomFocus::MessageBar;
Ok(None) Ok(None)
}, },
MessageAction::Redact(reason) => { MessageAction::React(emoji) => {
let room = store let room = self.get_joined(&store.application.worker)?;
.application
.worker
.client
.get_joined_room(self.id())
.ok_or(IambError::NotJoined)?;
let event_id = match &msg.event { let event_id = match &msg.event {
MessageEvent::EncryptedOriginal(ev) => ev.event_id.clone(),
MessageEvent::EncryptedRedacted(ev) => ev.event_id.clone(),
MessageEvent::Original(ev) => ev.event_id.clone(), MessageEvent::Original(ev) => ev.event_id.clone(),
MessageEvent::Local(event_id, _) => event_id.clone(), MessageEvent::Local(event_id, _) => event_id.clone(),
MessageEvent::Redacted(_) => { MessageEvent::Redacted(_) => {
let msg = ""; let msg = "Cannot react to a redacted message";
let err = UIError::Failure(msg.into());
return Err(err);
},
};
let reaction = Reaction::new(event_id, emoji);
let msg = ReactionEventContent::new(reaction);
let _ = room.send(msg, None).await.map_err(IambError::from)?;
Ok(None)
},
MessageAction::Redact(reason, skip_confirm) => {
if !skip_confirm {
let msg = "Are you sure you want to redact this message?";
let act = IambAction::Message(MessageAction::Redact(reason, true));
let prompt = PromptYesNo::new(msg, vec![Action::from(act)]);
let prompt = Box::new(prompt);
return Err(UIError::NeedConfirm(prompt));
}
let room = self.get_joined(&store.application.worker)?;
let event_id = match &msg.event {
MessageEvent::EncryptedOriginal(ev) => ev.event_id.clone(),
MessageEvent::EncryptedRedacted(ev) => ev.event_id.clone(),
MessageEvent::Original(ev) => ev.event_id.clone(),
MessageEvent::Local(event_id, _) => event_id.clone(),
MessageEvent::Redacted(_) => {
let msg = "Cannot redact already redacted message";
let err = UIError::Failure(msg.into()); let err = UIError::Failure(msg.into());
return Err(err); return Err(err);
@@ -295,6 +366,48 @@ impl ChatState {
self.reply_to = self.scrollback.get_key(info); self.reply_to = self.scrollback.get_key(info);
self.focus = RoomFocus::MessageBar; self.focus = RoomFocus::MessageBar;
Ok(None)
},
MessageAction::Unreact(emoji) => {
let room = self.get_joined(&store.application.worker)?;
let event_id: &EventId = match &msg.event {
MessageEvent::EncryptedOriginal(ev) => ev.event_id.as_ref(),
MessageEvent::EncryptedRedacted(ev) => ev.event_id.as_ref(),
MessageEvent::Original(ev) => ev.event_id.as_ref(),
MessageEvent::Local(event_id, _) => event_id.as_ref(),
MessageEvent::Redacted(_) => {
let msg = "Cannot unreact to a redacted message";
let err = UIError::Failure(msg.into());
return Err(err);
},
};
let reactions = match info.reactions.get(event_id) {
Some(r) => r,
None => return Ok(None),
};
let reactions = reactions.iter().filter_map(|(event_id, (reaction, user_id))| {
if user_id != &settings.profile.user_id {
return None;
}
if let Some(emoji) = &emoji {
if emoji == reaction {
return Some(event_id);
} else {
return None;
}
} else {
return Some(event_id);
}
});
for reaction in reactions {
let _ = room.redact(reaction, None, None).await.map_err(IambError::from)?;
}
Ok(None) Ok(None)
}, },
} }
@@ -312,21 +425,18 @@ impl ChatState {
.client .client
.get_joined_room(self.id()) .get_joined_room(self.id())
.ok_or(IambError::NotJoined)?; .ok_or(IambError::NotJoined)?;
let info = store.application.rooms.entry(self.id().to_owned()).or_default(); let info = store.application.rooms.get_or_default(self.id().to_owned());
let mut show_echo = true; let mut show_echo = true;
let (event_id, msg) = match act { let (event_id, msg) = match act {
SendAction::Submit => { SendAction::Submit => {
let msg = self.tbox.get_text(); let msg = self.tbox.get();
if msg.is_empty() { if msg.is_blank() {
return Ok(None); return Ok(None);
} }
let msg = TextMessageEventContent::markdown(msg); let mut msg = text_to_message(msg.trim_end().to_string());
let msg = MessageType::Text(msg);
let mut msg = RoomMessageEventContent::new(msg);
if let Some((_, event_id)) = &self.editing { if let Some((_, event_id)) = &self.editing {
msg.relates_to = Some(Relation::Replacement(Replacement::new( msg.relates_to = Some(Relation::Replacement(Replacement::new(
@@ -367,7 +477,37 @@ impl ChatState {
.map_err(IambError::from)?; .map_err(IambError::from)?;
// Mock up the local echo message for the scrollback. // Mock up the local echo message for the scrollback.
let msg = TextMessageEventContent::plain(format!("[Attached File: {}]", name)); let msg = TextMessageEventContent::plain(format!("[Attached File: {name}]"));
let msg = MessageType::Text(msg);
let msg = RoomMessageEventContent::new(msg);
(resp.event_id, msg)
},
SendAction::UploadImage(width, height, bytes) => {
// Convert to png because arboard does not give us the mime type.
let bytes =
image::ImageBuffer::from_raw(width as _, height as _, bytes.into_owned())
.ok_or(IambError::Clipboard)
.and_then(|imagebuf| {
let dynimage = image::DynamicImage::ImageRgba8(imagebuf);
let bytes = Vec::<u8>::new();
let mut buff = std::io::Cursor::new(bytes);
dynimage.write_to(&mut buff, image::ImageOutputFormat::Png)?;
Ok(buff.into_inner())
})
.map_err(IambError::from)?;
let mime = mime::IMAGE_PNG;
let name = "Clipboard.png";
let config = AttachmentConfig::new();
let resp = room
.send_attachment(name.as_ref(), &mime, bytes.as_ref(), config)
.await
.map_err(IambError::from)?;
// Mock up the local echo message for the scrollback.
let msg = TextMessageEventContent::plain(format!("[Attached File: {name}]"));
let msg = MessageType::Text(msg); let msg = MessageType::Text(msg);
let msg = RoomMessageEventContent::new(msg); let msg = RoomMessageEventContent::new(msg);
@@ -473,6 +613,21 @@ impl WindowOps<IambInfo> for ChatState {
true true
} }
fn write(
&mut self,
_: Option<&str>,
_: WriteFlags,
_: &mut ProgramStore,
) -> IambResult<EditInfo> {
// XXX: what's the right writing behaviour for a room?
// Should write send a message?
Ok(None)
}
fn get_completions(&self) -> Option<CompletionList> {
delegate!(self, w => w.get_completions())
}
fn get_cursor_word(&self, style: &WordStyle) -> Option<String> { fn get_cursor_word(&self, style: &WordStyle) -> Option<String> {
delegate!(self, w => w.get_cursor_word(style)) delegate!(self, w => w.get_cursor_word(style))
} }
@@ -502,6 +657,15 @@ impl Editable<ProgramContext, ProgramStore, IambInfo> for ChatState {
// Run command again. // Run command again.
delegate!(self, w => w.editor_command(act, ctx, store)) delegate!(self, w => w.editor_command(act, ctx, store))
}, },
Err(EditError::Register(RegisterError::ClipboardImage(data))) => {
let msg = "Do you really want to upload the image from your system clipboard?";
let send =
IambAction::Send(SendAction::UploadImage(data.width, data.height, data.bytes));
let prompt = PromptYesNo::new(msg, vec![Action::from(send)]);
let prompt = Box::new(prompt);
Err(EditError::NeedConfirm(prompt))
},
res @ Err(_) => res, res @ Err(_) => res,
} }
} }
@@ -578,13 +742,14 @@ impl PromptActions<ProgramContext, ProgramStore, IambInfo> for ChatState {
&mut self, &mut self,
dir: &MoveDir1D, dir: &MoveDir1D,
count: &Count, count: &Count,
prefixed: bool,
ctx: &ProgramContext, ctx: &ProgramContext,
_: &mut ProgramStore, _: &mut ProgramStore,
) -> EditResult<Vec<(ProgramAction, ProgramContext)>, IambInfo> { ) -> EditResult<Vec<(ProgramAction, ProgramContext)>, IambInfo> {
let count = ctx.resolve(count); let count = ctx.resolve(count);
let rope = self.tbox.get(); let rope = self.tbox.get();
let text = self.sent.recall(&rope, &mut self.sent_scrollback, *dir, count); let text = self.sent.recall(&rope, &mut self.sent_scrollback, *dir, prefixed, count);
if let Some(text) = text { if let Some(text) = text {
self.tbox.set_text(text); self.tbox.set_text(text);
@@ -608,7 +773,9 @@ impl Promptable<ProgramContext, ProgramStore, IambInfo> for ChatState {
match act { match act {
PromptAction::Submit => self.submit(ctx, store), PromptAction::Submit => self.submit(ctx, store),
PromptAction::Abort(empty) => self.abort(*empty, ctx, store), PromptAction::Abort(empty) => self.abort(*empty, ctx, store),
PromptAction::Recall(dir, count) => self.recall(dir, count, ctx, store), PromptAction::Recall(dir, count, prefixed) => {
self.recall(dir, count, *prefixed, ctx, store)
},
_ => Err(EditError::Unimplemented("unknown prompt action".to_string())), _ => Err(EditError::Unimplemented("unknown prompt action".to_string())),
} }
} }
@@ -634,10 +801,33 @@ impl<'a> StatefulWidget for Chat<'a> {
type State = ChatState; type State = ChatState;
fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State) { fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {
// Determine whether we have a description to show for the message bar.
let desc_spans = match (&state.editing, &state.reply_to) {
(None, None) => None,
(Some(_), None) => Some(Spans::from("Editing message")),
(editing, Some(_)) => {
state.reply_to.as_ref().and_then(|k| {
let room = self.store.application.rooms.get(state.id())?;
let msg = room.messages.get(k)?;
let user =
self.store.application.settings.get_user_span(msg.sender.as_ref(), room);
let prefix = if editing.is_some() {
Span::from("Editing reply to ")
} else {
Span::from("Replying to ")
};
let spans = Spans(vec![prefix, user]);
spans.into()
})
},
};
// Determine the region to show each UI element.
let lines = state.tbox.has_lines(5).max(1) as u16; let lines = state.tbox.has_lines(5).max(1) as u16;
let drawh = area.height; let drawh = area.height;
let texth = lines.min(drawh).clamp(1, 5); let texth = lines.min(drawh).clamp(1, 5);
let desch = if state.reply_to.is_some() { let desch = if desc_spans.is_some() {
drawh.saturating_sub(texth).min(1) drawh.saturating_sub(texth).min(1)
} else { } else {
0 0
@@ -648,25 +838,7 @@ impl<'a> StatefulWidget for Chat<'a> {
let descarea = Rect::new(area.x, scrollarea.y + scrollh, area.width, desch); let descarea = Rect::new(area.x, scrollarea.y + scrollh, area.width, desch);
let textarea = Rect::new(area.x, descarea.y + desch, area.width, texth); let textarea = Rect::new(area.x, descarea.y + desch, area.width, texth);
let scrollback_focused = state.focus.is_scrollback() && self.focused; // Render the message bar and any description for it.
let scrollback = Scrollback::new(self.store).focus(scrollback_focused);
scrollback.render(scrollarea, buf, &mut state.scrollback);
let desc_spans = match (&state.editing, &state.reply_to) {
(None, None) => None,
(Some(_), _) => Some(Spans::from("Editing message")),
(_, Some(_)) => {
state.reply_to.as_ref().and_then(|k| {
let room = self.store.application.rooms.get(state.id())?;
let msg = room.messages.get(k)?;
let user = self.store.application.settings.get_user_span(msg.sender.as_ref());
let spans = Spans(vec![Span::from("Replying to "), user]);
spans.into()
})
},
};
if let Some(desc_spans) = desc_spans { if let Some(desc_spans) = desc_spans {
Paragraph::new(desc_spans).render(descarea, buf); Paragraph::new(desc_spans).render(descarea, buf);
} }
@@ -675,5 +847,35 @@ impl<'a> StatefulWidget for Chat<'a> {
let tbox = TextBox::new().prompt(prompt); let tbox = TextBox::new().prompt(prompt);
tbox.render(textarea, buf, &mut state.tbox); tbox.render(textarea, buf, &mut state.tbox);
// Render the message scrollback.
let scrollback_focused = state.focus.is_scrollback() && self.focused;
let scrollback = Scrollback::new(self.store)
.focus(scrollback_focused)
.room_focus(self.focused);
scrollback.render(scrollarea, buf, &mut state.scrollback);
} }
} }
fn open_command(open_command: Option<&Vec<String>>, target: OsString) -> IambResult<()> {
if let Some(mut cmd) = open_command.and_then(cmd) {
cmd.arg(target);
cmd.spawn()?;
return Ok(());
} else {
// open::that may not return until the spawned program closes.
tokio::task::spawn_blocking(move || {
return open::that(target);
});
return Ok(());
}
}
fn cmd(open_command: &Vec<String>) -> Option<Command> {
if let [program, args @ ..] = open_command.as_slice() {
let mut cmd = Command::new(program);
cmd.args(args);
return Some(cmd);
}
None
}

View File

@@ -40,12 +40,16 @@ use modalkit::{
PositionList, PositionList,
ScrollStyle, ScrollStyle,
WordStyle, WordStyle,
WriteFlags,
}, },
editing::completion::CompletionList,
input::dialog::PromptYesNo,
input::InputContext, input::InputContext,
widgets::{TermOffset, TerminalCursor, WindowOps}, widgets::{TermOffset, TerminalCursor, WindowOps},
}; };
use crate::base::{ use crate::base::{
IambAction,
IambError, IambError,
IambId, IambId,
IambInfo, IambInfo,
@@ -132,14 +136,12 @@ impl RoomState {
None => format!("{:?}", store.application.get_room_title(self.id())), None => format!("{:?}", store.application.get_room_title(self.id())),
}; };
let mut invited = vec![Span::from(format!( let mut invited = vec![Span::from(format!("You have been invited to join {name}"))];
"You have been invited to join {}",
name
))];
if let Ok(Some(inviter)) = &inviter { if let Ok(Some(inviter)) = &inviter {
let info = store.application.rooms.get_or_default(self.id().to_owned());
invited.push(Span::from(" by ")); invited.push(Span::from(" by "));
invited.push(store.application.settings.get_user_span(inviter.user_id())); invited.push(store.application.settings.get_user_span(inviter.user_id(), info));
} }
let l1 = Spans(invited); let l1 = Spans(invited);
@@ -186,8 +188,16 @@ impl RoomState {
match act { match act {
RoomAction::InviteAccept => { RoomAction::InviteAccept => {
if let Some(room) = store.application.worker.client.get_invited_room(self.id()) { if let Some(room) = store.application.worker.client.get_invited_room(self.id()) {
let details = room.invite_details().await.map_err(IambError::from)?;
let details = details.invitee.event().original_content();
let is_direct = details.and_then(|ev| ev.is_direct).unwrap_or_default();
room.accept_invitation().await.map_err(IambError::from)?; room.accept_invitation().await.map_err(IambError::from)?;
if is_direct {
room.set_is_direct(true).await.map_err(IambError::from)?;
}
Ok(vec![]) Ok(vec![])
} else { } else {
Err(IambError::NotInvited.into()) Err(IambError::NotInvited.into())
@@ -211,6 +221,24 @@ impl RoomState {
Err(IambError::NotJoined.into()) Err(IambError::NotJoined.into())
} }
}, },
RoomAction::Leave(skip_confirm) => {
if let Some(room) = store.application.worker.client.get_joined_room(self.id()) {
if skip_confirm {
room.leave().await.map_err(IambError::from)?;
Ok(vec![])
} else {
let msg = "Do you really want to leave this room?";
let leave = IambAction::Room(RoomAction::Leave(true));
let prompt = PromptYesNo::new(msg, vec![Action::from(leave)]);
let prompt = Box::new(prompt);
Err(UIError::NeedConfirm(prompt))
}
} else {
Err(IambError::NotJoined.into())
}
},
RoomAction::Members(mut cmd) => { RoomAction::Members(mut cmd) => {
let width = Count::Exact(30); let width = Count::Exact(30);
let act = let act =
@@ -386,10 +414,30 @@ impl WindowOps<IambInfo> for RoomState {
} }
} }
fn close(&mut self, _: CloseFlags, _: &mut ProgramStore) -> bool { fn close(&mut self, flags: CloseFlags, store: &mut ProgramStore) -> bool {
// XXX: what's the right closing behaviour for a room? match self {
// Should write send a message? RoomState::Chat(chat) => chat.close(flags, store),
true RoomState::Space(space) => space.close(flags, store),
}
}
fn write(
&mut self,
path: Option<&str>,
flags: WriteFlags,
store: &mut ProgramStore,
) -> IambResult<EditInfo> {
match self {
RoomState::Chat(chat) => chat.write(path, flags, store),
RoomState::Space(space) => space.write(path, flags, store),
}
}
fn get_completions(&self) -> Option<CompletionList> {
match self {
RoomState::Chat(chat) => chat.get_completions(),
RoomState::Space(space) => space.get_completions(),
}
} }
fn get_cursor_word(&self, style: &WordStyle) -> Option<String> { fn get_cursor_word(&self, style: &WordStyle) -> Option<String> {

View File

@@ -4,7 +4,13 @@ use regex::Regex;
use matrix_sdk::ruma::OwnedRoomId; use matrix_sdk::ruma::OwnedRoomId;
use modalkit::tui::{buffer::Buffer, layout::Rect, widgets::StatefulWidget}; use modalkit::tui::{
buffer::Buffer,
layout::{Alignment, Rect},
style::{Modifier as StyleModifier, Style},
text::{Span, Spans},
widgets::{Paragraph, StatefulWidget, Widget},
};
use modalkit::widgets::{ScrollActions, TerminalCursor, WindowOps}; use modalkit::widgets::{ScrollActions, TerminalCursor, WindowOps};
use modalkit::editing::{ use modalkit::editing::{
@@ -32,6 +38,9 @@ use modalkit::editing::{
base::{ base::{
Axis, Axis,
CloseFlags, CloseFlags,
CompletionDisplay,
CompletionSelection,
CompletionType,
Count, Count,
EditRange, EditRange,
EditTarget, EditTarget,
@@ -51,7 +60,9 @@ use modalkit::editing::{
TargetShape, TargetShape,
ViewportContext, ViewportContext,
WordStyle, WordStyle,
WriteFlags,
}, },
completion::CompletionList,
context::{EditContext, Resolve}, context::{EditContext, Resolve},
cursor::{CursorGroup, CursorState}, cursor::{CursorGroup, CursorState},
history::HistoryList, history::HistoryList,
@@ -60,9 +71,9 @@ use modalkit::editing::{
}; };
use crate::{ use crate::{
base::{IambBufferId, IambInfo, ProgramContext, ProgramStore, RoomFocus, RoomInfo}, base::{IambBufferId, IambInfo, IambResult, ProgramContext, ProgramStore, RoomFocus, RoomInfo},
config::ApplicationSettings, config::ApplicationSettings,
message::{Message, MessageCursor, MessageKey}, message::{Message, MessageCursor, MessageKey, Messages},
}; };
fn nth_key_before(pos: MessageKey, n: usize, info: &RoomInfo) -> MessageKey { fn nth_key_before(pos: MessageKey, n: usize, info: &RoomInfo) -> MessageKey {
@@ -103,6 +114,10 @@ fn nth_after(pos: MessageKey, n: usize, info: &RoomInfo) -> MessageCursor {
nth_key_after(pos, n, info).into() nth_key_after(pos, n, info).into()
} }
fn prevmsg<'a>(key: &MessageKey, info: &'a RoomInfo) -> Option<&'a Message> {
info.messages.range(..key).next_back().map(|(_, v)| v)
}
pub struct ScrollbackState { pub struct ScrollbackState {
/// The room identifier. /// The room identifier.
room_id: OwnedRoomId, room_id: OwnedRoomId,
@@ -160,11 +175,11 @@ impl ScrollbackState {
.or_else(|| info.messages.last_key_value().map(|kv| kv.0.clone())) .or_else(|| info.messages.last_key_value().map(|kv| kv.0.clone()))
} }
pub fn get_mut<'a>(&mut self, info: &'a mut RoomInfo) -> Option<&'a mut Message> { pub fn get_mut<'a>(&mut self, messages: &'a mut Messages) -> Option<&'a mut Message> {
if let Some(k) = &self.cursor.timestamp { if let Some(k) = &self.cursor.timestamp {
info.messages.get_mut(k) messages.get_mut(k)
} else { } else {
info.messages.last_entry().map(|o| o.into_mut()) messages.last_entry().map(|o| o.into_mut())
} }
} }
@@ -214,7 +229,8 @@ impl ScrollbackState {
for (key, item) in info.messages.range(..=&idx).rev() { for (key, item) in info.messages.range(..=&idx).rev() {
let sel = selidx == key; let sel = selidx == key;
let len = item.show(None, sel, &self.viewctx, info, settings).lines.len(); let prev = prevmsg(key, info);
let len = item.show(prev, sel, &self.viewctx, info, settings).lines.len();
if key == &idx { if key == &idx {
lines += len / 2; lines += len / 2;
@@ -236,7 +252,8 @@ impl ScrollbackState {
for (key, item) in info.messages.range(..=&idx).rev() { for (key, item) in info.messages.range(..=&idx).rev() {
let sel = key == selidx; let sel = key == selidx;
let len = item.show(None, sel, &self.viewctx, info, settings).lines.len(); let prev = prevmsg(key, info);
let len = item.show(prev, sel, &self.viewctx, info, settings).lines.len();
lines += len; lines += len;
@@ -269,6 +286,7 @@ impl ScrollbackState {
let mut lines = 0; let mut lines = 0;
let cursor_key = self.cursor.timestamp.as_ref().unwrap_or(last_key); let cursor_key = self.cursor.timestamp.as_ref().unwrap_or(last_key);
let mut prev = prevmsg(cursor_key, info);
for (idx, item) in info.messages.range(corner_key.clone()..) { for (idx, item) in info.messages.range(corner_key.clone()..) {
if idx == cursor_key { if idx == cursor_key {
@@ -276,13 +294,15 @@ impl ScrollbackState {
break; break;
} }
lines += item.show(None, false, &self.viewctx, info, settings).height().max(1); lines += item.show(prev, false, &self.viewctx, info, settings).height().max(1);
if lines >= self.viewctx.get_height() { if lines >= self.viewctx.get_height() {
// We've reached the end of the viewport; move cursor into it. // We've reached the end of the viewport; move cursor into it.
self.cursor = idx.clone().into(); self.cursor = idx.clone().into();
break; break;
} }
prev = Some(item);
} }
} }
@@ -506,6 +526,23 @@ impl WindowOps<IambInfo> for ScrollbackState {
true true
} }
fn write(
&mut self,
_: Option<&str>,
flags: WriteFlags,
_: &mut ProgramStore,
) -> IambResult<EditInfo> {
if flags.contains(WriteFlags::FORCE) {
Ok(None)
} else {
Err(EditError::ReadOnly.into())
}
}
fn get_completions(&self) -> Option<CompletionList> {
None
}
fn get_cursor_word(&self, _: &WordStyle) -> Option<String> { fn get_cursor_word(&self, _: &WordStyle) -> Option<String> {
None None
} }
@@ -523,7 +560,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
ctx: &ProgramContext, ctx: &ProgramContext,
store: &mut ProgramStore, store: &mut ProgramStore,
) -> EditResult<EditInfo, IambInfo> { ) -> EditResult<EditInfo, IambInfo> {
let info = store.application.rooms.entry(self.room_id.clone()).or_default(); let info = store.application.rooms.get_or_default(self.room_id.clone());
let key = if let Some(k) = self.cursor.to_key(info) { let key = if let Some(k) = self.cursor.to_key(info) {
k.clone() k.clone()
} else { } else {
@@ -582,7 +619,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
let needle = match ctx.get_search_regex() { let needle = match ctx.get_search_regex() {
Some(re) => re, Some(re) => re,
None => { None => {
let lsearch = store.registers.get(&Register::LastSearch); let lsearch = store.registers.get(&Register::LastSearch)?;
let lsearch = lsearch.value.to_string(); let lsearch = lsearch.value.to_string();
Regex::new(lsearch.as_ref())? Regex::new(lsearch.as_ref())?
@@ -606,7 +643,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
}, },
_ => { _ => {
let msg = format!("Unknown editing target: {:?}", motion); let msg = format!("Unknown editing target: {motion:?}");
let err = EditError::Unimplemented(msg); let err = EditError::Unimplemented(msg);
return Err(err); return Err(err);
@@ -668,7 +705,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
let needle = match ctx.get_search_regex() { let needle = match ctx.get_search_regex() {
Some(re) => re, Some(re) => re,
None => { None => {
let lsearch = store.registers.get(&Register::LastSearch); let lsearch = store.registers.get(&Register::LastSearch)?;
let lsearch = lsearch.value.to_string(); let lsearch = lsearch.value.to_string();
Regex::new(lsearch.as_ref())? Regex::new(lsearch.as_ref())?
@@ -693,7 +730,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
}, },
_ => { _ => {
let msg = format!("Unknown motion: {:?}", motion); let msg = format!("Unknown motion: {motion:?}");
let err = EditError::Unimplemented(msg); let err = EditError::Unimplemented(msg);
return Err(err); return Err(err);
@@ -716,7 +753,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
flags |= RegisterPutFlags::APPEND; flags |= RegisterPutFlags::APPEND;
} }
store.registers.put(&register, cell, flags); store.registers.put(&register, cell, flags)?;
} }
return Ok(None); return Ok(None);
@@ -724,7 +761,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
// Everything else is a modifying action. // Everything else is a modifying action.
EditAction::ChangeCase(_) => Err(EditError::ReadOnly), EditAction::ChangeCase(_) => Err(EditError::ReadOnly),
EditAction::ChangeNumber(_) => Err(EditError::ReadOnly), EditAction::ChangeNumber(_, _) => Err(EditError::ReadOnly),
EditAction::Delete => Err(EditError::ReadOnly), EditAction::Delete => Err(EditError::ReadOnly),
EditAction::Format => Err(EditError::ReadOnly), EditAction::Format => Err(EditError::ReadOnly),
EditAction::Indent(_) => Err(EditError::ReadOnly), EditAction::Indent(_) => Err(EditError::ReadOnly),
@@ -753,6 +790,17 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
} }
} }
fn complete(
&mut self,
_: &CompletionType,
_: &CompletionSelection,
_: &CompletionDisplay,
_: &ProgramContext,
_: &mut ProgramStore,
) -> EditResult<EditInfo, IambInfo> {
Err(EditError::ReadOnly)
}
fn insert_text( fn insert_text(
&mut self, &mut self,
_: &InsertTextAction, _: &InsertTextAction,
@@ -781,7 +829,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
HistoryAction::Checkpoint => Ok(None), HistoryAction::Checkpoint => Ok(None),
HistoryAction::Undo(_) => Err(EditError::Failure("Nothing to undo".into())), HistoryAction::Undo(_) => Err(EditError::Failure("Nothing to undo".into())),
HistoryAction::Redo(_) => Err(EditError::Failure("Nothing to redo".into())), HistoryAction::Redo(_) => Err(EditError::Failure("Nothing to redo".into())),
_ => Err(EditError::Unimplemented(format!("Unknown action: {:?}", act))), _ => Err(EditError::Unimplemented(format!("Unknown action: {act:?}"))),
} }
} }
@@ -838,7 +886,7 @@ impl EditorActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
Ok(None) Ok(None)
}, },
_ => Err(EditError::Unimplemented(format!("Unknown action: {:?}", act))), _ => Err(EditError::Unimplemented(format!("Unknown action: {act:?}"))),
} }
} }
} }
@@ -858,14 +906,14 @@ impl Editable<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
EditorAction::Mark(name) => self.mark(ctx.resolve(name), ctx, store), EditorAction::Mark(name) => self.mark(ctx.resolve(name), ctx, store),
EditorAction::Selection(act) => self.selection_command(act, ctx, store), EditorAction::Selection(act) => self.selection_command(act, ctx, store),
EditorAction::Complete(_, _) => { EditorAction::Complete(_, _, _) => {
let msg = ""; let msg = "Nothing to complete in message scrollback";
let err = EditError::Unimplemented(msg.into()); let err = EditError::Failure(msg.into());
Err(err) Err(err)
}, },
_ => Err(EditError::Unimplemented(format!("Unknown action: {:?}", act))), _ => Err(EditError::Unimplemented(format!("Unknown action: {act:?}"))),
} }
} }
} }
@@ -964,7 +1012,7 @@ impl Promptable<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
return Err(err); return Err(err);
}, },
_ => { _ => {
let msg = format!("Messages scrollback doesn't support {:?}", act); let msg = format!("Messages scrollback doesn't support {act:?}");
let err = EditError::Unimplemented(msg); let err = EditError::Unimplemented(msg);
return Err(err); return Err(err);
@@ -982,7 +1030,7 @@ impl ScrollActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
ctx: &ProgramContext, ctx: &ProgramContext,
store: &mut ProgramStore, store: &mut ProgramStore,
) -> EditResult<EditInfo, IambInfo> { ) -> EditResult<EditInfo, IambInfo> {
let info = store.application.rooms.entry(self.room_id.clone()).or_default(); let info = store.application.rooms.get_or_default(self.room_id.clone());
let settings = &store.application.settings; let settings = &store.application.settings;
let mut corner = self.viewctx.corner.clone(); let mut corner = self.viewctx.corner.clone();
@@ -1009,7 +1057,8 @@ impl ScrollActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
for (key, item) in info.messages.range(..=&corner_key).rev() { for (key, item) in info.messages.range(..=&corner_key).rev() {
let sel = key == cursor_key; let sel = key == cursor_key;
let txt = item.show(None, sel, &self.viewctx, info, settings); let prev = prevmsg(key, info);
let txt = item.show(prev, sel, &self.viewctx, info, settings);
let len = txt.height().max(1); let len = txt.height().max(1);
let max = len.saturating_sub(1); let max = len.saturating_sub(1);
@@ -1033,12 +1082,16 @@ impl ScrollActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
} }
}, },
MoveDir2D::Down => { MoveDir2D::Down => {
let mut prev = prevmsg(&corner_key, info);
for (key, item) in info.messages.range(&corner_key..) { for (key, item) in info.messages.range(&corner_key..) {
let sel = key == cursor_key; let sel = key == cursor_key;
let txt = item.show(None, sel, &self.viewctx, info, settings); let txt = item.show(prev, sel, &self.viewctx, info, settings);
let len = txt.height().max(1); let len = txt.height().max(1);
let max = len.saturating_sub(1); let max = len.saturating_sub(1);
prev = Some(item);
if key != &corner_key { if key != &corner_key {
corner.text_row = 0; corner.text_row = 0;
} }
@@ -1091,7 +1144,7 @@ impl ScrollActions<ProgramContext, ProgramStore, IambInfo> for ScrollbackState {
Err(err) Err(err)
}, },
Axis::Vertical => { Axis::Vertical => {
let info = store.application.rooms.entry(self.room_id.clone()).or_default(); let info = store.application.rooms.get_or_default(self.room_id.clone());
let settings = &store.application.settings; let settings = &store.application.settings;
if let Some(key) = self.cursor.to_key(info).cloned() { if let Some(key) = self.cursor.to_key(info).cloned() {
@@ -1158,14 +1211,43 @@ impl TerminalCursor for ScrollbackState {
} }
} }
fn render_jump_to_recent(area: Rect, buf: &mut Buffer, focused: bool) -> Rect {
if area.height <= 5 || area.width <= 20 {
return area;
}
let top = Rect::new(area.x, area.y, area.width, area.height - 1);
let bar = Rect::new(area.x, area.y + top.height, area.width, 1);
let msg = vec![
Span::raw("Use "),
Span::styled("G", Style::default().add_modifier(StyleModifier::BOLD)),
Span::raw(if focused { "" } else { " in scrollback" }),
Span::raw(" to jump to latest message"),
];
Paragraph::new(Spans::from(msg))
.alignment(Alignment::Center)
.render(bar, buf);
return top;
}
pub struct Scrollback<'a> { pub struct Scrollback<'a> {
room_focused: bool,
focused: bool, focused: bool,
store: &'a mut ProgramStore, store: &'a mut ProgramStore,
} }
impl<'a> Scrollback<'a> { impl<'a> Scrollback<'a> {
pub fn new(store: &'a mut ProgramStore) -> Self { pub fn new(store: &'a mut ProgramStore) -> Self {
Scrollback { focused: false, store } Scrollback { room_focused: false, focused: false, store }
}
/// Indicate whether the room window is currently focused, regardless of whether the scrollback
/// also is.
pub fn room_focus(mut self, focused: bool) -> Self {
self.room_focused = focused;
self
} }
/// Indicate whether the scrollback is currently focused. /// Indicate whether the scrollback is currently focused.
@@ -1179,9 +1261,13 @@ impl<'a> StatefulWidget for Scrollback<'a> {
type State = ScrollbackState; type State = ScrollbackState;
fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State) { fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {
let info = self.store.application.rooms.entry(state.room_id.clone()).or_default(); let info = self.store.application.rooms.get_or_default(state.room_id.clone());
let settings = &self.store.application.settings; let settings = &self.store.application.settings;
let area = info.render_typing(area, buf, &self.store.application.settings); let area = if state.cursor.timestamp.is_some() {
render_jump_to_recent(area, buf, self.focused)
} else {
info.render_typing(area, buf, &self.store.application.settings)
};
state.set_term_info(area); state.set_term_info(area);
@@ -1214,7 +1300,7 @@ impl<'a> StatefulWidget for Scrollback<'a> {
let full = std::mem::take(&mut state.show_full_on_redraw) || cursor.timestamp.is_none(); let full = std::mem::take(&mut state.show_full_on_redraw) || cursor.timestamp.is_none();
let mut lines = vec![]; let mut lines = vec![];
let mut sawit = false; let mut sawit = false;
let mut prev = None; let mut prev = prevmsg(&corner_key, info);
for (key, item) in info.messages.range(&corner_key..) { for (key, item) in info.messages.range(&corner_key..) {
let sel = key == cursor_key; let sel = key == cursor_key;
@@ -1260,7 +1346,10 @@ impl<'a> StatefulWidget for Scrollback<'a> {
y += 1; y += 1;
} }
if settings.tunables.read_receipt_send && state.cursor.timestamp.is_none() { if self.room_focused &&
settings.tunables.read_receipt_send &&
state.cursor.timestamp.is_none()
{
// If the cursor is at the last message, then update the read marker. // If the cursor is at the last message, then update the read marker.
info.read_till = info.messages.last_key_value().map(|(k, _)| k.1.clone()); info.read_till = info.messages.last_key_value().map(|(k, _)| k.1.clone());
} }
@@ -1373,10 +1462,11 @@ mod tests {
assert_eq!(scrollback.viewctx.dimensions, (0, 0)); assert_eq!(scrollback.viewctx.dimensions, (0, 0));
assert_eq!(scrollback.viewctx.corner, MessageCursor::latest()); assert_eq!(scrollback.viewctx.corner, MessageCursor::latest());
// Set a terminal width of 60, and height of 3, rendering in scrollback as: // Set a terminal width of 60, and height of 4, rendering in scrollback as:
// //
// |------------------------------------------------------------| // |------------------------------------------------------------|
// MSG2: | @user2:example.com helium | // MSG2: | Wednesday, December 31 1969 |
// | @user2:example.com helium |
// MSG3: | @user2:example.com this | // MSG3: | @user2:example.com this |
// | is | // | is |
// | a | // | a |
@@ -1384,14 +1474,15 @@ mod tests {
// | message | // | message |
// MSG4: | @user1:example.com help | // MSG4: | @user1:example.com help |
// MSG5: | @user2:example.com character | // MSG5: | @user2:example.com character |
// MSG1: | @user1:example.com writhe | // MSG1: | XXXday, Month NN 20XX |
// | @user1:example.com writhe |
// |------------------------------------------------------------| // |------------------------------------------------------------|
let area = Rect::new(0, 0, 60, 3); let area = Rect::new(0, 0, 60, 4);
let mut buffer = Buffer::empty(area); let mut buffer = Buffer::empty(area);
scrollback.draw(area, &mut buffer, true, &mut store); scrollback.draw(area, &mut buffer, true, &mut store);
assert_eq!(scrollback.cursor, MessageCursor::latest()); assert_eq!(scrollback.cursor, MessageCursor::latest());
assert_eq!(scrollback.viewctx.dimensions, (60, 3)); assert_eq!(scrollback.viewctx.dimensions, (60, 4));
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG4_KEY.clone(), 0)); assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG4_KEY.clone(), 0));
// Scroll up a line at a time until we hit the first message. // Scroll up a line at a time until we hit the first message.
@@ -1420,6 +1511,11 @@ mod tests {
.unwrap(); .unwrap();
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG3_KEY.clone(), 0)); assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG3_KEY.clone(), 0));
scrollback
.dirscroll(prev, ScrollSize::Cell, &1.into(), &ctx, &mut store)
.unwrap();
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG2_KEY.clone(), 1));
scrollback scrollback
.dirscroll(prev, ScrollSize::Cell, &1.into(), &ctx, &mut store) .dirscroll(prev, ScrollSize::Cell, &1.into(), &ctx, &mut store)
.unwrap(); .unwrap();
@@ -1432,6 +1528,11 @@ mod tests {
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG2_KEY.clone(), 0)); assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG2_KEY.clone(), 0));
// Now scroll back down one line at a time. // Now scroll back down one line at a time.
scrollback
.dirscroll(next, ScrollSize::Cell, &1.into(), &ctx, &mut store)
.unwrap();
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG2_KEY.clone(), 1));
scrollback scrollback
.dirscroll(next, ScrollSize::Cell, &1.into(), &ctx, &mut store) .dirscroll(next, ScrollSize::Cell, &1.into(), &ctx, &mut store)
.unwrap(); .unwrap();
@@ -1472,19 +1573,24 @@ mod tests {
.unwrap(); .unwrap();
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG1_KEY.clone(), 0)); assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG1_KEY.clone(), 0));
scrollback
.dirscroll(next, ScrollSize::Cell, &1.into(), &ctx, &mut store)
.unwrap();
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG1_KEY.clone(), 1));
// Cannot scroll down any further. // Cannot scroll down any further.
scrollback scrollback
.dirscroll(next, ScrollSize::Cell, &1.into(), &ctx, &mut store) .dirscroll(next, ScrollSize::Cell, &1.into(), &ctx, &mut store)
.unwrap(); .unwrap();
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG1_KEY.clone(), 0)); assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG1_KEY.clone(), 1));
// Scroll up two Pages (six lines). // Scroll up two Pages (eight lines).
scrollback scrollback
.dirscroll(prev, ScrollSize::Page, &2.into(), &ctx, &mut store) .dirscroll(prev, ScrollSize::Page, &2.into(), &ctx, &mut store)
.unwrap(); .unwrap();
assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG3_KEY.clone(), 1)); assert_eq!(scrollback.viewctx.corner, MessageCursor::new(MSG3_KEY.clone(), 0));
// Scroll down two HalfPages (three lines). // Scroll down two HalfPages (four lines).
scrollback scrollback
.dirscroll(next, ScrollSize::HalfPage, &2.into(), &ctx, &mut store) .dirscroll(next, ScrollSize::HalfPage, &2.into(), &ctx, &mut store)
.unwrap(); .unwrap();
@@ -1503,7 +1609,8 @@ mod tests {
// Set a terminal width of 60, and height of 3, rendering in scrollback as: // Set a terminal width of 60, and height of 3, rendering in scrollback as:
// //
// |------------------------------------------------------------| // |------------------------------------------------------------|
// MSG2: | @user2:example.com helium | // MSG2: | Wednesday, December 31 1969 |
// | @user2:example.com helium |
// MSG3: | @user2:example.com this | // MSG3: | @user2:example.com this |
// | is | // | is |
// | a | // | a |
@@ -1511,7 +1618,8 @@ mod tests {
// | message | // | message |
// MSG4: | @user1:example.com help | // MSG4: | @user1:example.com help |
// MSG5: | @user2:example.com character | // MSG5: | @user2:example.com character |
// MSG1: | @user1:example.com writhe | // MSG1: | XXXday, Month NN 20XX |
// | @user1:example.com writhe |
// |------------------------------------------------------------| // |------------------------------------------------------------|
let area = Rect::new(0, 0, 60, 3); let area = Rect::new(0, 0, 60, 3);
let mut buffer = Buffer::empty(area); let mut buffer = Buffer::empty(area);

View File

@@ -1,11 +1,18 @@
use std::ops::{Deref, DerefMut}; use std::ops::{Deref, DerefMut};
use std::time::{Duration, Instant};
use matrix_sdk::{ use matrix_sdk::{
room::Room as MatrixRoom, room::Room as MatrixRoom,
ruma::{OwnedRoomId, RoomId}, ruma::{OwnedRoomId, RoomId},
}; };
use modalkit::tui::{buffer::Buffer, layout::Rect, widgets::StatefulWidget}; use modalkit::tui::{
buffer::Buffer,
layout::Rect,
style::{Color, Style},
text::{Span, Spans, Text},
widgets::StatefulWidget,
};
use modalkit::{ use modalkit::{
widgets::list::{List, ListState}, widgets::list::{List, ListState},
@@ -16,10 +23,13 @@ use crate::base::{IambBufferId, IambInfo, ProgramStore, RoomFocus};
use crate::windows::RoomItem; use crate::windows::RoomItem;
const SPACE_HIERARCHY_DEBOUNCE: Duration = Duration::from_secs(5);
pub struct SpaceState { pub struct SpaceState {
room_id: OwnedRoomId, room_id: OwnedRoomId,
room: MatrixRoom, room: MatrixRoom,
list: ListState<RoomItem, IambInfo>, list: ListState<RoomItem, IambInfo>,
last_fetch: Option<Instant>,
} }
impl SpaceState { impl SpaceState {
@@ -27,8 +37,9 @@ impl SpaceState {
let room_id = room.room_id().to_owned(); let room_id = room.room_id().to_owned();
let content = IambBufferId::Room(room_id.clone(), RoomFocus::Scrollback); let content = IambBufferId::Room(room_id.clone(), RoomFocus::Scrollback);
let list = ListState::new(content, vec![]); let list = ListState::new(content, vec![]);
let last_fetch = None;
SpaceState { room_id, room, list } SpaceState { room_id, room, list, last_fetch }
} }
pub fn refresh_room(&mut self, store: &mut ProgramStore) { pub fn refresh_room(&mut self, store: &mut ProgramStore) {
@@ -50,6 +61,7 @@ impl SpaceState {
room_id: self.room_id.clone(), room_id: self.room_id.clone(),
room: self.room.clone(), room: self.room.clone(),
list: self.list.dup(store), list: self.list.dup(store),
last_fetch: self.last_fetch,
} }
} }
} }
@@ -94,20 +106,26 @@ impl<'a> StatefulWidget for Space<'a> {
type State = SpaceState; type State = SpaceState;
fn render(self, area: Rect, buffer: &mut Buffer, state: &mut Self::State) { fn render(self, area: Rect, buffer: &mut Buffer, state: &mut Self::State) {
let members = let mut empty_message = None;
if let Ok(m) = self.store.application.worker.space_members(state.room_id.clone()) { let need_fetch = match state.last_fetch {
m Some(i) => i.elapsed() >= SPACE_HIERARCHY_DEBOUNCE,
} else { None => true,
return;
}; };
if need_fetch {
let res = self.store.application.worker.space_members(state.room_id.clone());
match res {
Ok(members) => {
let items = members let items = members
.into_iter() .into_iter()
.filter_map(|id| { .filter_map(|id| {
let (room, name, tags) = self.store.application.worker.get_room(id.clone()).ok()?; let (room, _, tags) =
self.store.application.worker.get_room(id.clone()).ok()?;
let room_info = std::sync::Arc::new((room, tags));
if id != state.room_id { if id != state.room_id {
Some(RoomItem::new(room, name, tags, self.store)) Some(RoomItem::new(room_info, self.store))
} else { } else {
None None
} }
@@ -115,9 +133,25 @@ impl<'a> StatefulWidget for Space<'a> {
.collect(); .collect();
state.list.set(items); state.list.set(items);
state.last_fetch = Some(Instant::now());
},
Err(e) => {
let lines = vec![
Spans::from("Unable to fetch space room hierarchy:"),
Span::styled(e.to_string(), Style::default().fg(Color::Red)).into(),
];
List::new(self.store) empty_message = Text { lines }.into();
.focus(self.focused) },
.render(area, buffer, &mut state.list) }
}
let mut list = List::new(self.store).focus(self.focused);
if let Some(text) = empty_message {
list = list.empty_message(text);
}
list.render(area, buffer, &mut state.list)
} }
} }

View File

@@ -8,9 +8,11 @@ use modalkit::{
widgets::{TermOffset, TerminalCursor}, widgets::{TermOffset, TerminalCursor},
}; };
use modalkit::editing::base::{CloseFlags, WordStyle}; use modalkit::editing::action::EditInfo;
use modalkit::editing::base::{CloseFlags, WordStyle, WriteFlags};
use modalkit::editing::completion::CompletionList;
use crate::base::{IambBufferId, IambInfo, ProgramStore}; use crate::base::{IambBufferId, IambInfo, IambResult, ProgramStore};
const WELCOME_TEXT: &str = include_str!("welcome.md"); const WELCOME_TEXT: &str = include_str!("welcome.md");
@@ -63,6 +65,19 @@ impl WindowOps<IambInfo> for WelcomeState {
self.tbox.close(flags, store) self.tbox.close(flags, store)
} }
fn write(
&mut self,
path: Option<&str>,
flags: WriteFlags,
store: &mut ProgramStore,
) -> IambResult<EditInfo> {
self.tbox.write(path, flags, store)
}
fn get_completions(&self) -> Option<CompletionList> {
self.tbox.get_completions()
}
fn get_cursor_word(&self, style: &WordStyle) -> Option<String> { fn get_cursor_word(&self, style: &WordStyle) -> Option<String> {
self.tbox.get_cursor_word(style) self.tbox.get_cursor_word(style)
} }

View File

@@ -1,3 +1,4 @@
use std::collections::HashMap;
use std::convert::TryFrom; use std::convert::TryFrom;
use std::fmt::{Debug, Formatter}; use std::fmt::{Debug, Formatter};
use std::fs::File; use std::fs::File;
@@ -5,25 +6,27 @@ use std::io::BufWriter;
use std::str::FromStr; use std::str::FromStr;
use std::sync::mpsc::{sync_channel, Receiver, SyncSender}; use std::sync::mpsc::{sync_channel, Receiver, SyncSender};
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::{Duration, Instant};
use futures::{stream::FuturesUnordered, StreamExt};
use gethostname::gethostname; use gethostname::gethostname;
use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender}; use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender};
use tokio::task::JoinHandle; use tokio::task::JoinHandle;
use tracing::error; use tracing::{error, warn};
use matrix_sdk::{ use matrix_sdk::{
config::{RequestConfig, StoreConfig, SyncSettings}, config::{RequestConfig, SyncSettings},
encryption::verification::{SasVerification, Verification}, encryption::verification::{SasVerification, Verification},
event_handler::Ctx, event_handler::Ctx,
reqwest, reqwest,
room::{Invited, Messages, MessagesOptions, Room as MatrixRoom, RoomMember}, room::{Invited, Messages, MessagesOptions, Room as MatrixRoom, RoomMember},
ruma::{ ruma::{
api::client::{ api::client::{
room::create_room::v3::{Request as CreateRoomRequest, RoomPreset}, room::create_room::v3::{CreationContent, Request as CreateRoomRequest, RoomPreset},
room::Visibility, room::Visibility,
space::get_hierarchy::v1::Request as SpaceHierarchyRequest, space::get_hierarchy::v1::Request as SpaceHierarchyRequest,
}, },
assign,
events::{ events::{
key::verification::{ key::verification::{
done::{OriginalSyncKeyVerificationDoneEvent, ToDeviceKeyVerificationDoneEvent}, done::{OriginalSyncKeyVerificationDoneEvent, ToDeviceKeyVerificationDoneEvent},
@@ -32,21 +35,33 @@ use matrix_sdk::{
start::{OriginalSyncKeyVerificationStartEvent, ToDeviceKeyVerificationStartEvent}, start::{OriginalSyncKeyVerificationStartEvent, ToDeviceKeyVerificationStartEvent},
VerificationMethod, VerificationMethod,
}, },
presence::PresenceEvent,
reaction::ReactionEventContent,
room::{ room::{
encryption::RoomEncryptionEventContent,
member::OriginalSyncRoomMemberEvent,
message::{MessageType, RoomMessageEventContent}, message::{MessageType, RoomMessageEventContent},
name::RoomNameEventContent, name::RoomNameEventContent,
redaction::{OriginalSyncRoomRedactionEvent, SyncRoomRedactionEvent}, redaction::{OriginalSyncRoomRedactionEvent, SyncRoomRedactionEvent},
}, },
tag::Tags, tag::Tags,
typing::SyncTypingEvent, typing::SyncTypingEvent,
AnyInitialStateEvent,
AnyMessageLikeEvent, AnyMessageLikeEvent,
AnyTimelineEvent, AnyTimelineEvent,
EmptyStateKey,
InitialStateEvent,
SyncMessageLikeEvent, SyncMessageLikeEvent,
SyncStateEvent, SyncStateEvent,
}, },
room::RoomType,
serde::Raw,
EventEncryptionAlgorithm,
OwnedEventId,
OwnedRoomId, OwnedRoomId,
OwnedRoomOrAliasId, OwnedRoomOrAliasId,
OwnedUserId, OwnedUserId,
RoomId,
RoomVersionId, RoomVersionId,
}, },
Client, Client,
@@ -57,19 +72,314 @@ use matrix_sdk::{
use modalkit::editing::action::{EditInfo, InfoMessage, UIError}; use modalkit::editing::action::{EditInfo, InfoMessage, UIError};
use crate::{ use crate::{
base::{AsyncProgramStore, IambError, IambResult, Receipts, VerifyAction}, base::{
AsyncProgramStore,
ChatStore,
CreateRoomFlags,
CreateRoomType,
EventLocation,
IambError,
IambResult,
Receipts,
RoomFetchStatus,
VerifyAction,
},
message::MessageFetchResult, message::MessageFetchResult,
ApplicationSettings, ApplicationSettings,
}; };
const IAMB_DEVICE_NAME: &str = "iamb"; const IAMB_DEVICE_NAME: &str = "iamb";
const IAMB_USER_AGENT: &str = "iamb"; const IAMB_USER_AGENT: &str = "iamb";
const REQ_TIMEOUT: Duration = Duration::from_secs(60); const MIN_MSG_LOAD: u32 = 50;
fn initial_devname() -> String { fn initial_devname() -> String {
format!("{} on {}", IAMB_DEVICE_NAME, gethostname().to_string_lossy()) format!("{} on {}", IAMB_DEVICE_NAME, gethostname().to_string_lossy())
} }
pub async fn create_room(
client: &Client,
room_alias_name: Option<&str>,
rt: CreateRoomType,
flags: CreateRoomFlags,
) -> IambResult<OwnedRoomId> {
let mut creation_content = None;
let mut initial_state = vec![];
let mut is_direct = false;
let mut preset = None;
let mut invite = vec![];
let visibility = if flags.contains(CreateRoomFlags::PUBLIC) {
Visibility::Public
} else {
Visibility::Private
};
match rt {
CreateRoomType::Direct(user) => {
invite.push(user);
is_direct = true;
preset = Some(RoomPreset::TrustedPrivateChat);
},
CreateRoomType::Space => {
let mut cc = CreationContent::new();
cc.room_type = Some(RoomType::Space);
let raw_cc = Raw::new(&cc).map_err(IambError::from)?;
creation_content = Some(raw_cc);
},
CreateRoomType::Room => {},
}
// Set up encryption.
if flags.contains(CreateRoomFlags::ENCRYPTED) {
// XXX: Once matrix-sdk uses ruma 0.8, then this can skip the cast.
let algo = EventEncryptionAlgorithm::MegolmV1AesSha2;
let content = RoomEncryptionEventContent::new(algo);
let encr = InitialStateEvent { content, state_key: EmptyStateKey };
let encr_raw = Raw::new(&encr).map_err(IambError::from)?;
let encr_raw = encr_raw.cast::<AnyInitialStateEvent>();
initial_state.push(encr_raw);
}
let request = assign!(CreateRoomRequest::new(), {
room_alias_name,
creation_content,
initial_state: initial_state.as_slice(),
invite: invite.as_slice(),
is_direct,
visibility,
preset,
});
let resp = client.create_room(request).await.map_err(IambError::from)?;
if is_direct {
if let Some(room) = client.get_room(&resp.room_id) {
room.set_is_direct(true).await.map_err(IambError::from)?;
} else {
error!(
room_id = resp.room_id.as_str(),
"Couldn't set is_direct for new direct message room"
);
}
}
return Ok(resp.room_id);
}
async fn load_plan(store: &AsyncProgramStore) -> HashMap<OwnedRoomId, Option<String>> {
let mut locked = store.lock().await;
let ChatStore { need_load, rooms, .. } = &mut locked.application;
let mut plan = HashMap::new();
for room_id in std::mem::take(need_load).into_iter() {
let info = rooms.get_or_default(room_id.clone());
if info.recently_fetched() || info.fetching {
need_load.insert(room_id);
continue;
} else {
info.fetch_last = Instant::now().into();
info.fetching = true;
}
let fetch_id = match &info.fetch_id {
RoomFetchStatus::Done => continue,
RoomFetchStatus::HaveMore(fetch_id) => Some(fetch_id.clone()),
RoomFetchStatus::NotStarted => None,
};
plan.insert(room_id, fetch_id);
}
return plan;
}
async fn load_older_one(
client: Client,
room_id: &RoomId,
fetch_id: Option<String>,
limit: u32,
) -> MessageFetchResult {
if let Some(room) = client.get_room(room_id) {
let mut opts = match &fetch_id {
Some(id) => MessagesOptions::backward().from(id.as_str()),
None => MessagesOptions::backward(),
};
opts.limit = limit.into();
let Messages { end, chunk, .. } = room.messages(opts).await.map_err(IambError::from)?;
let msgs = chunk.into_iter().filter_map(|ev| {
match ev.event.deserialize() {
Ok(AnyTimelineEvent::MessageLike(msg)) => Some(msg),
Ok(AnyTimelineEvent::State(_)) => None,
Err(_) => None,
}
});
Ok((end, msgs.collect()))
} else {
Err(IambError::UnknownRoom(room_id.to_owned()).into())
}
}
async fn load_insert(room_id: OwnedRoomId, res: MessageFetchResult, store: AsyncProgramStore) {
let mut locked = store.lock().await;
let ChatStore { need_load, presences, rooms, .. } = &mut locked.application;
let info = rooms.get_or_default(room_id.clone());
info.fetching = false;
match res {
Ok((fetch_id, msgs)) => {
for msg in msgs.into_iter() {
let sender = msg.sender().to_owned();
let _ = presences.get_or_default(sender);
match msg {
AnyMessageLikeEvent::RoomEncrypted(msg) => {
info.insert_encrypted(msg);
},
AnyMessageLikeEvent::RoomMessage(msg) => {
info.insert(msg);
},
AnyMessageLikeEvent::Reaction(ev) => {
info.insert_reaction(ev);
},
_ => continue,
}
}
info.fetch_id = fetch_id.map_or(RoomFetchStatus::Done, RoomFetchStatus::HaveMore);
},
Err(e) => {
warn!(room_id = room_id.as_str(), err = e.to_string(), "Failed to load older messages");
// Wait and try again.
need_load.insert(room_id);
},
}
}
async fn load_older(client: &Client, store: &AsyncProgramStore) -> usize {
let limit = MIN_MSG_LOAD;
// Fetch each room separately, so they don't block each other.
load_plan(store)
.await
.into_iter()
.map(|(room_id, fetch_id)| {
let client = client.clone();
let store = store.clone();
async move {
let res = load_older_one(client, room_id.as_ref(), fetch_id, limit).await;
load_insert(room_id, res, store).await;
}
})
.collect::<FuturesUnordered<_>>()
.count()
.await
}
async fn load_older_forever(client: &Client, store: &AsyncProgramStore) {
// Load older messages every 2 seconds.
let mut interval = tokio::time::interval(Duration::from_secs(2));
loop {
interval.tick().await;
load_older(client, store).await;
}
}
async fn refresh_rooms(client: &Client, store: &AsyncProgramStore) {
let mut names = vec![];
let mut spaces = vec![];
let mut rooms = vec![];
let mut dms = vec![];
for room in client.invited_rooms().into_iter() {
let name = room.display_name().await.unwrap_or(DisplayName::Empty).to_string();
names.push((room.room_id().to_owned(), name));
if room.is_direct() {
let tags = room.tags().await.unwrap_or_default();
dms.push(Arc::new((room.into(), tags)));
} else if room.is_space() {
spaces.push(room.into());
} else {
let tags = room.tags().await.unwrap_or_default();
rooms.push(Arc::new((room.into(), tags)));
}
}
for room in client.joined_rooms().into_iter() {
let name = room.display_name().await.unwrap_or(DisplayName::Empty).to_string();
names.push((room.room_id().to_owned(), name));
if room.is_direct() {
let tags = room.tags().await.unwrap_or_default();
dms.push(Arc::new((room.into(), tags)));
} else if room.is_space() {
spaces.push(room.into());
} else {
let tags = room.tags().await.unwrap_or_default();
rooms.push(Arc::new((room.into(), tags)));
}
}
let mut locked = store.lock().await;
locked.application.sync_info.spaces = spaces;
locked.application.sync_info.rooms = rooms;
locked.application.sync_info.dms = dms;
for (room_id, name) in names {
locked.application.set_room_name(&room_id, &name);
}
}
async fn refresh_rooms_forever(client: &Client, store: &AsyncProgramStore) {
let mut interval = tokio::time::interval(Duration::from_secs(5));
loop {
interval.tick().await;
refresh_rooms(client, store).await;
}
}
async fn refresh_receipts_forever(client: &Client, store: &AsyncProgramStore) {
// Update the displayed read receipts every 5 seconds.
let mut interval = tokio::time::interval(Duration::from_secs(5));
let mut sent = HashMap::<OwnedRoomId, OwnedEventId>::default();
loop {
interval.tick().await;
let receipts = update_receipts(client).await;
let read = store.lock().await.application.set_receipts(receipts).await;
for (room_id, read_till) in read.into_iter() {
if let Some(read_sent) = sent.get(&room_id) {
if read_sent == &read_till {
// Skip unchanged receipts.
continue;
}
}
if let Some(room) = client.get_joined_room(&room_id) {
if room.read_receipt(&read_till).await.is_ok() {
sent.insert(room_id, read_till);
}
}
}
}
}
#[derive(Debug)] #[derive(Debug)]
pub enum LoginStyle { pub enum LoginStyle {
SessionRestore(Session), SessionRestore(Session),
@@ -125,17 +435,13 @@ async fn update_receipts(client: &Client) -> Vec<(OwnedRoomId, Receipts)> {
pub type FetchedRoom = (MatrixRoom, DisplayName, Option<Tags>); pub type FetchedRoom = (MatrixRoom, DisplayName, Option<Tags>);
pub enum WorkerTask { pub enum WorkerTask {
ActiveRooms(ClientReply<Vec<FetchedRoom>>),
DirectMessages(ClientReply<Vec<FetchedRoom>>),
Init(AsyncProgramStore, ClientReply<()>), Init(AsyncProgramStore, ClientReply<()>),
LoadOlder(OwnedRoomId, Option<String>, u32, ClientReply<MessageFetchResult>),
Login(LoginStyle, ClientReply<IambResult<EditInfo>>), Login(LoginStyle, ClientReply<IambResult<EditInfo>>),
GetInviter(Invited, ClientReply<IambResult<Option<RoomMember>>>), GetInviter(Invited, ClientReply<IambResult<Option<RoomMember>>>),
GetRoom(OwnedRoomId, ClientReply<IambResult<FetchedRoom>>), GetRoom(OwnedRoomId, ClientReply<IambResult<FetchedRoom>>),
JoinRoom(String, ClientReply<IambResult<OwnedRoomId>>), JoinRoom(String, ClientReply<IambResult<OwnedRoomId>>),
Members(OwnedRoomId, ClientReply<IambResult<Vec<RoomMember>>>), Members(OwnedRoomId, ClientReply<IambResult<Vec<RoomMember>>>),
SpaceMembers(OwnedRoomId, ClientReply<IambResult<Vec<OwnedRoomId>>>), SpaceMembers(OwnedRoomId, ClientReply<IambResult<Vec<OwnedRoomId>>>),
Spaces(ClientReply<Vec<(MatrixRoom, DisplayName)>>),
TypingNotice(OwnedRoomId), TypingNotice(OwnedRoomId),
Verify(VerifyAction, SasVerification, ClientReply<IambResult<EditInfo>>), Verify(VerifyAction, SasVerification, ClientReply<IambResult<EditInfo>>),
VerifyRequest(OwnedUserId, ClientReply<IambResult<EditInfo>>), VerifyRequest(OwnedUserId, ClientReply<IambResult<EditInfo>>),
@@ -144,28 +450,12 @@ pub enum WorkerTask {
impl Debug for WorkerTask { impl Debug for WorkerTask {
fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), std::fmt::Error> { fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), std::fmt::Error> {
match self { match self {
WorkerTask::ActiveRooms(_) => {
f.debug_tuple("WorkerTask::ActiveRooms").field(&format_args!("_")).finish()
},
WorkerTask::DirectMessages(_) => {
f.debug_tuple("WorkerTask::DirectMessages")
.field(&format_args!("_"))
.finish()
},
WorkerTask::Init(_, _) => { WorkerTask::Init(_, _) => {
f.debug_tuple("WorkerTask::Init") f.debug_tuple("WorkerTask::Init")
.field(&format_args!("_")) .field(&format_args!("_"))
.field(&format_args!("_")) .field(&format_args!("_"))
.finish() .finish()
}, },
WorkerTask::LoadOlder(room_id, from, n, _) => {
f.debug_tuple("WorkerTask::LoadOlder")
.field(room_id)
.field(from)
.field(n)
.field(&format_args!("_"))
.finish()
},
WorkerTask::Login(style, _) => { WorkerTask::Login(style, _) => {
f.debug_tuple("WorkerTask::Login") f.debug_tuple("WorkerTask::Login")
.field(style) .field(style)
@@ -199,9 +489,6 @@ impl Debug for WorkerTask {
.field(&format_args!("_")) .field(&format_args!("_"))
.finish() .finish()
}, },
WorkerTask::Spaces(_) => {
f.debug_tuple("WorkerTask::Spaces").field(&format_args!("_")).finish()
},
WorkerTask::TypingNotice(room_id) => { WorkerTask::TypingNotice(room_id) => {
f.debug_tuple("WorkerTask::TypingNotice").field(room_id).finish() f.debug_tuple("WorkerTask::TypingNotice").field(room_id).finish()
}, },
@@ -237,21 +524,6 @@ impl Requester {
return response.recv(); return response.recv();
} }
pub fn load_older(
&self,
room_id: OwnedRoomId,
fetch_id: Option<String>,
limit: u32,
) -> MessageFetchResult {
let (reply, response) = oneshot();
self.tx
.send(WorkerTask::LoadOlder(room_id, fetch_id, limit, reply))
.unwrap();
return response.recv();
}
pub fn login(&self, style: LoginStyle) -> IambResult<EditInfo> { pub fn login(&self, style: LoginStyle) -> IambResult<EditInfo> {
let (reply, response) = oneshot(); let (reply, response) = oneshot();
@@ -260,14 +532,6 @@ impl Requester {
return response.recv(); return response.recv();
} }
pub fn direct_messages(&self) -> Vec<FetchedRoom> {
let (reply, response) = oneshot();
self.tx.send(WorkerTask::DirectMessages(reply)).unwrap();
return response.recv();
}
pub fn get_inviter(&self, invite: Invited) -> IambResult<Option<RoomMember>> { pub fn get_inviter(&self, invite: Invited) -> IambResult<Option<RoomMember>> {
let (reply, response) = oneshot(); let (reply, response) = oneshot();
@@ -292,14 +556,6 @@ impl Requester {
return response.recv(); return response.recv();
} }
pub fn active_rooms(&self) -> Vec<FetchedRoom> {
let (reply, response) = oneshot();
self.tx.send(WorkerTask::ActiveRooms(reply)).unwrap();
return response.recv();
}
pub fn members(&self, room_id: OwnedRoomId) -> IambResult<Vec<RoomMember>> { pub fn members(&self, room_id: OwnedRoomId) -> IambResult<Vec<RoomMember>> {
let (reply, response) = oneshot(); let (reply, response) = oneshot();
@@ -316,14 +572,6 @@ impl Requester {
return response.recv(); return response.recv();
} }
pub fn spaces(&self) -> Vec<(MatrixRoom, DisplayName)> {
let (reply, response) = oneshot();
self.tx.send(WorkerTask::Spaces(reply)).unwrap();
return response.recv();
}
pub fn typing_notice(&self, room_id: OwnedRoomId) { pub fn typing_notice(&self, room_id: OwnedRoomId) {
self.tx.send(WorkerTask::TypingNotice(room_id)).unwrap(); self.tx.send(WorkerTask::TypingNotice(room_id)).unwrap();
} }
@@ -349,6 +597,7 @@ pub struct ClientWorker {
initialized: bool, initialized: bool,
settings: ApplicationSettings, settings: ApplicationSettings,
client: Client, client: Client,
load_handle: Option<JoinHandle<()>>,
sync_handle: Option<JoinHandle<()>>, sync_handle: Option<JoinHandle<()>>,
} }
@@ -357,28 +606,27 @@ impl ClientWorker {
let (tx, rx) = unbounded_channel(); let (tx, rx) = unbounded_channel();
let account = &settings.profile; let account = &settings.profile;
// Set up a custom client that only uses HTTP/1. let req_timeout = Duration::from_secs(settings.tunables.request_timeout);
//
// During my testing, I kept stumbling across something weird with sync and HTTP/2 that // Set up the HTTP client.
// will need to be revisited in the future.
let http = reqwest::Client::builder() let http = reqwest::Client::builder()
.user_agent(IAMB_USER_AGENT) .user_agent(IAMB_USER_AGENT)
.timeout(Duration::from_secs(30)) .timeout(req_timeout)
.pool_idle_timeout(Duration::from_secs(60)) .pool_idle_timeout(Duration::from_secs(60))
.pool_max_idle_per_host(10) .pool_max_idle_per_host(10)
.tcp_keepalive(Duration::from_secs(10)) .tcp_keepalive(Duration::from_secs(10))
.http1_only()
.build() .build()
.unwrap(); .unwrap();
let req_config = RequestConfig::new().timeout(req_timeout).retry_timeout(req_timeout);
// Set up the Matrix client for the selected profile. // Set up the Matrix client for the selected profile.
let client = Client::builder() let client = Client::builder()
.http_client(Arc::new(http)) .http_client(Arc::new(http))
.homeserver_url(account.url.clone()) .homeserver_url(account.url.clone())
.store_config(StoreConfig::default())
.sled_store(settings.matrix_dir.as_path(), None) .sled_store(settings.matrix_dir.as_path(), None)
.expect("Failed to setup up sled store for Matrix SDK") .expect("Failed to setup up sled store for Matrix SDK")
.request_config(RequestConfig::new().timeout(REQ_TIMEOUT).retry_timeout(REQ_TIMEOUT)) .request_config(req_config)
.build() .build()
.await .await
.expect("Failed to instantiate Matrix client"); .expect("Failed to instantiate Matrix client");
@@ -387,10 +635,11 @@ impl ClientWorker {
initialized: false, initialized: false,
settings, settings,
client: client.clone(), client: client.clone(),
load_handle: None,
sync_handle: None, sync_handle: None,
}; };
let _ = tokio::spawn(async move { tokio::spawn(async move {
worker.work(rx).await; worker.work(rx).await;
}); });
@@ -416,10 +665,6 @@ impl ClientWorker {
async fn run(&mut self, task: WorkerTask) { async fn run(&mut self, task: WorkerTask) {
match task { match task {
WorkerTask::DirectMessages(reply) => {
assert!(self.initialized);
reply.send(self.direct_messages().await);
},
WorkerTask::Init(store, reply) => { WorkerTask::Init(store, reply) => {
assert_eq!(self.initialized, false); assert_eq!(self.initialized, false);
self.init(store).await; self.init(store).await;
@@ -437,14 +682,6 @@ impl ClientWorker {
assert!(self.initialized); assert!(self.initialized);
reply.send(self.get_room(room_id).await); reply.send(self.get_room(room_id).await);
}, },
WorkerTask::ActiveRooms(reply) => {
assert!(self.initialized);
reply.send(self.active_rooms().await);
},
WorkerTask::LoadOlder(room_id, fetch_id, limit, reply) => {
assert!(self.initialized);
reply.send(self.load_older(room_id, fetch_id, limit).await);
},
WorkerTask::Login(style, reply) => { WorkerTask::Login(style, reply) => {
assert!(self.initialized); assert!(self.initialized);
reply.send(self.login_and_sync(style).await); reply.send(self.login_and_sync(style).await);
@@ -457,10 +694,6 @@ impl ClientWorker {
assert!(self.initialized); assert!(self.initialized);
reply.send(self.space_members(space).await); reply.send(self.space_members(space).await);
}, },
WorkerTask::Spaces(reply) => {
assert!(self.initialized);
reply.send(self.spaces().await);
},
WorkerTask::TypingNotice(room_id) => { WorkerTask::TypingNotice(room_id) => {
assert!(self.initialized); assert!(self.initialized);
self.typing_notice(room_id).await; self.typing_notice(room_id).await;
@@ -497,6 +730,15 @@ impl ClientWorker {
}, },
); );
let _ =
self.client
.add_event_handler(|ev: PresenceEvent, store: Ctx<AsyncProgramStore>| {
async move {
let mut locked = store.lock().await;
locked.application.presences.insert(ev.sender, ev.content.presence);
}
});
let _ = self.client.add_event_handler( let _ = self.client.add_event_handler(
|ev: SyncStateEvent<RoomNameEventContent>, |ev: SyncStateEvent<RoomNameEventContent>,
room: MatrixRoom, room: MatrixRoom,
@@ -507,8 +749,7 @@ impl ClientWorker {
let room_id = room.room_id().to_owned(); let room_id = room.room_id().to_owned();
let room_name = Some(room_name.to_string()); let room_name = Some(room_name.to_string());
let mut locked = store.lock().await; let mut locked = store.lock().await;
let mut info = let mut info = locked.application.rooms.get_or_default(room_id.clone());
locked.application.rooms.entry(room_id.to_owned()).or_default();
info.name = room_name; info.name = room_name;
} }
} }
@@ -523,8 +764,6 @@ impl ClientWorker {
store: Ctx<AsyncProgramStore>| { store: Ctx<AsyncProgramStore>| {
async move { async move {
let room_id = room.room_id(); let room_id = room.room_id();
let room_name = room.display_name().await.ok();
let room_name = room_name.as_ref().map(ToString::to_string);
if let Some(msg) = ev.as_original() { if let Some(msg) = ev.as_original() {
if let MessageType::VerificationRequest(_) = msg.content.msgtype { if let MessageType::VerificationRequest(_) = msg.content.msgtype {
@@ -539,13 +778,34 @@ impl ClientWorker {
} }
let mut locked = store.lock().await; let mut locked = store.lock().await;
let mut info = locked.application.get_room_info(room_id.to_owned());
info.name = room_name; let sender = ev.sender().to_owned();
let _ = locked.application.presences.get_or_default(sender);
let info = locked.application.get_room_info(room_id.to_owned());
info.insert(ev.into_full_event(room_id.to_owned())); info.insert(ev.into_full_event(room_id.to_owned()));
} }
}, },
); );
let _ = self.client.add_event_handler(
|ev: SyncMessageLikeEvent<ReactionEventContent>,
room: MatrixRoom,
store: Ctx<AsyncProgramStore>| {
async move {
let room_id = room.room_id();
let mut locked = store.lock().await;
let sender = ev.sender().to_owned();
let _ = locked.application.presences.get_or_default(sender);
let info = locked.application.get_room_info(room_id.to_owned());
info.insert_reaction(ev.into_full_event(room_id.to_owned()));
}
},
);
let _ = self.client.add_event_handler( let _ = self.client.add_event_handler(
|ev: OriginalSyncRoomRedactionEvent, |ev: OriginalSyncRoomRedactionEvent,
room: MatrixRoom, room: MatrixRoom,
@@ -558,15 +818,53 @@ impl ClientWorker {
let mut locked = store.lock().await; let mut locked = store.lock().await;
let info = locked.application.get_room_info(room_id.to_owned()); let info = locked.application.get_room_info(room_id.to_owned());
let key = if let Some(k) = info.keys.get(&ev.redacts) { match info.keys.get(&ev.redacts) {
k None => return,
} else { Some(EventLocation::Message(key)) => {
return;
};
if let Some(msg) = info.messages.get_mut(key) { if let Some(msg) = info.messages.get_mut(key) {
let ev = SyncRoomRedactionEvent::Original(ev); let ev = SyncRoomRedactionEvent::Original(ev);
msg.event.redact(ev, room_version); msg.redact(ev, room_version);
}
},
Some(EventLocation::Reaction(event_id)) => {
if let Some(reactions) = info.reactions.get_mut(event_id) {
reactions.remove(&ev.redacts);
}
info.keys.remove(&ev.redacts);
},
}
}
},
);
let _ = self.client.add_event_handler(
|ev: OriginalSyncRoomMemberEvent,
room: MatrixRoom,
client: Client,
store: Ctx<AsyncProgramStore>| {
async move {
let room_id = room.room_id();
let user_id = ev.state_key;
let ambiguous_name =
ev.content.displayname.as_deref().unwrap_or_else(|| user_id.localpart());
let ambiguous = client
.store()
.get_users_with_display_name(room_id, ambiguous_name)
.await
.map(|users| users.len() > 1)
.unwrap_or_default();
let mut locked = store.lock().await;
let info = locked.application.get_room_info(room_id.to_owned());
if ambiguous {
info.display_names.remove(&user_id);
} else if let Some(display) = ev.content.displayname {
info.display_names.insert(user_id, display);
} else {
info.display_names.remove(&user_id);
} }
} }
}, },
@@ -628,11 +926,12 @@ impl ClientWorker {
let request = client let request = client
.encryption() .encryption()
.get_verification_request(&ev.sender, &ev.content.transaction_id) .get_verification_request(&ev.sender, &ev.content.transaction_id)
.await .await;
.unwrap();
if let Some(request) = request {
request.accept().await.unwrap(); request.accept().await.unwrap();
} }
}
}, },
); );
@@ -684,18 +983,17 @@ impl ClientWorker {
}, },
); );
self.load_handle = tokio::spawn({
let client = self.client.clone(); let client = self.client.clone();
let _ = tokio::spawn(async move {
// Update the displayed read receipts ever 5 seconds.
let mut interval = tokio::time::interval(Duration::from_secs(5));
loop { async move {
interval.tick().await; let load = load_older_forever(&client, &store);
let rcpt = refresh_receipts_forever(&client, &store);
let receipts = update_receipts(&client).await; let room = refresh_rooms_forever(&client, &store);
store.lock().await.application.set_receipts(receipts).await; let ((), (), ()) = tokio::join!(load, rcpt, room);
} }
}); })
.into();
self.initialized = true; self.initialized = true;
} }
@@ -721,53 +1019,42 @@ impl ClientWorker {
}, },
} }
let handle = tokio::spawn(async move { self.sync_handle = tokio::spawn(async move {
loop { loop {
let settings = SyncSettings::default(); let settings = SyncSettings::default();
let _ = client.sync(settings).await; let _ = client.sync(settings).await;
} }
}); })
.into();
self.sync_handle = Some(handle);
self.client
.sync_once(SyncSettings::default())
.await
.map_err(IambError::from)?;
Ok(Some(InfoMessage::from("Successfully logged in!"))) Ok(Some(InfoMessage::from("Successfully logged in!")))
} }
async fn direct_message(&mut self, user: OwnedUserId) -> IambResult<FetchedRoom> { async fn direct_message(&mut self, user: OwnedUserId) -> IambResult<OwnedRoomId> {
for (room, name, tags) in self.direct_messages().await { for room in self.client.rooms() {
if !room.is_direct() {
continue;
}
if room.get_member(user.as_ref()).await.map_err(IambError::from)?.is_some() { if room.get_member(user.as_ref()).await.map_err(IambError::from)?.is_some() {
return Ok((room, name, tags)); return Ok(room.room_id().to_owned());
} }
} }
let mut request = CreateRoomRequest::new(); let rt = CreateRoomType::Direct(user.clone());
let invite = [user.clone()]; let flags = CreateRoomFlags::ENCRYPTED;
request.is_direct = true;
request.invite = &invite;
request.visibility = Visibility::Private;
request.preset = Some(RoomPreset::PrivateChat);
match self.client.create_room(request).await { create_room(&self.client, None, rt, flags).await.map_err(|e| {
Ok(resp) => self.get_room(resp.room_id).await,
Err(e) => {
error!( error!(
user_id = user.as_str(), user_id = user.as_str(),
err = e.to_string(), err = e.to_string(),
"Failed to create direct message room" "Failed to create direct message room"
); );
let msg = format!("Could not open a room with {}", user); let msg = format!("Could not open a room with {user}");
let err = UIError::Failure(msg); UIError::Failure(msg)
})
Err(err)
},
}
} }
async fn get_inviter(&mut self, invited: Invited) -> IambResult<Option<RoomMember>> { async fn get_inviter(&mut self, invited: Invited) -> IambResult<Option<RoomMember>> {
@@ -799,9 +1086,7 @@ impl ClientWorker {
}, },
} }
} else if let Ok(user) = OwnedUserId::try_from(name.as_str()) { } else if let Ok(user) = OwnedUserId::try_from(name.as_str()) {
let room = self.direct_message(user).await?.0; self.direct_message(user).await
return Ok(room.room_id().to_owned());
} else { } else {
let msg = format!("{:?} is not a valid room or user name", name.as_str()); let msg = format!("{:?} is not a valid room or user name", name.as_str());
let err = UIError::Failure(msg); let err = UIError::Failure(msg);
@@ -810,97 +1095,6 @@ impl ClientWorker {
} }
} }
async fn direct_messages(&self) -> Vec<FetchedRoom> {
let mut rooms = vec![];
for room in self.client.invited_rooms().into_iter() {
if !room.is_direct() {
continue;
}
let name = room.display_name().await.unwrap_or(DisplayName::Empty);
let tags = room.tags().await.unwrap_or_default();
rooms.push((room.into(), name, tags));
}
for room in self.client.joined_rooms().into_iter() {
if !room.is_direct() {
continue;
}
let name = room.display_name().await.unwrap_or(DisplayName::Empty);
let tags = room.tags().await.unwrap_or_default();
rooms.push((room.into(), name, tags));
}
return rooms;
}
async fn active_rooms(&self) -> Vec<FetchedRoom> {
let mut rooms = vec![];
for room in self.client.invited_rooms().into_iter() {
if room.is_space() || room.is_direct() {
continue;
}
let name = room.display_name().await.unwrap_or(DisplayName::Empty);
let tags = room.tags().await.unwrap_or_default();
rooms.push((room.into(), name, tags));
}
for room in self.client.joined_rooms().into_iter() {
if room.is_space() || room.is_direct() {
continue;
}
let name = room.display_name().await.unwrap_or(DisplayName::Empty);
let tags = room.tags().await.unwrap_or_default();
rooms.push((room.into(), name, tags));
}
return rooms;
}
async fn load_older(
&mut self,
room_id: OwnedRoomId,
fetch_id: Option<String>,
limit: u32,
) -> MessageFetchResult {
if let Some(room) = self.client.get_room(room_id.as_ref()) {
let mut opts = match &fetch_id {
Some(id) => MessagesOptions::backward().from(id.as_str()),
None => MessagesOptions::backward(),
};
opts.limit = limit.into();
let Messages { end, chunk, .. } = room.messages(opts).await.map_err(IambError::from)?;
let msgs = chunk.into_iter().filter_map(|ev| {
match ev.event.deserialize() {
Ok(AnyTimelineEvent::MessageLike(msg)) => {
if let AnyMessageLikeEvent::RoomMessage(msg) = msg {
Some(msg)
} else {
None
}
},
Ok(AnyTimelineEvent::State(_)) => None,
Err(_) => None,
}
});
Ok((end, msgs.collect()))
} else {
Err(IambError::UnknownRoom(room_id).into())
}
}
async fn members(&mut self, room_id: OwnedRoomId) -> IambResult<Vec<RoomMember>> { async fn members(&mut self, room_id: OwnedRoomId) -> IambResult<Vec<RoomMember>> {
if let Some(room) = self.client.get_room(room_id.as_ref()) { if let Some(room) = self.client.get_room(room_id.as_ref()) {
Ok(room.active_members().await.map_err(IambError::from)?) Ok(room.active_members().await.map_err(IambError::from)?)
@@ -921,32 +1115,6 @@ impl ClientWorker {
Ok(rooms) Ok(rooms)
} }
async fn spaces(&self) -> Vec<(MatrixRoom, DisplayName)> {
let mut spaces = vec![];
for room in self.client.invited_rooms().into_iter() {
if !room.is_space() {
continue;
}
let name = room.display_name().await.unwrap_or(DisplayName::Empty);
spaces.push((room.into(), name));
}
for room in self.client.joined_rooms().into_iter() {
if !room.is_space() {
continue;
}
let name = room.display_name().await.unwrap_or(DisplayName::Empty);
spaces.push((room.into(), name));
}
return spaces;
}
async fn typing_notice(&mut self, room_id: OwnedRoomId) { async fn typing_notice(&mut self, room_id: OwnedRoomId) {
if let Some(room) = self.client.get_joined_room(room_id.as_ref()) { if let Some(room) = self.client.get_joined_room(room_id.as_ref()) {
let _ = room.typing_notice(true).await; let _ = room.typing_notice(true).await;
@@ -1007,12 +1175,12 @@ impl ClientWorker {
let methods = vec![VerificationMethod::SasV1]; let methods = vec![VerificationMethod::SasV1];
let request = identity.request_verification_with_methods(methods); let request = identity.request_verification_with_methods(methods);
let _req = request.await.map_err(IambError::from)?; let _req = request.await.map_err(IambError::from)?;
let info = format!("Sent verification request to {}", user_id); let info = format!("Sent verification request to {user_id}");
Ok(InfoMessage::from(info).into()) Ok(Some(InfoMessage::from(info)))
}, },
None => { None => {
let msg = format!("Could not find identity information for {}", user_id); let msg = format!("Could not find identity information for {user_id}");
let err = UIError::Failure(msg); let err = UIError::Failure(msg);
Err(err) Err(err)