mirror of
https://forgejo.ellis.link/continuwuation/continuwuity.git
synced 2025-07-05 22:35:47 +02:00
Compare commits
3 commits
aa29b81ef6
...
be52d495c9
Author | SHA1 | Date | |
---|---|---|---|
|
be52d495c9 | ||
|
8f7b3bceb0 | ||
|
5c916550fc |
5 changed files with 115 additions and 4 deletions
|
@ -1,7 +1,7 @@
|
||||||
use axum::extract::State;
|
use axum::extract::State;
|
||||||
use conduwuit::{Err, Event, Result, err};
|
use conduwuit::{Err, Event, PduEvent, Result, err};
|
||||||
use futures::{FutureExt, TryFutureExt, future::try_join};
|
use futures::{FutureExt, TryFutureExt, future::try_join};
|
||||||
use ruma::api::client::room::get_room_event;
|
use ruma::api::client::{error::ErrorKind, room::get_room_event};
|
||||||
|
|
||||||
use crate::{Ruma, client::is_ignored_pdu};
|
use crate::{Ruma, client::is_ignored_pdu};
|
||||||
|
|
||||||
|
@ -14,6 +14,7 @@ pub(crate) async fn get_room_event_route(
|
||||||
) -> Result<get_room_event::v3::Response> {
|
) -> Result<get_room_event::v3::Response> {
|
||||||
let event_id = &body.event_id;
|
let event_id = &body.event_id;
|
||||||
let room_id = &body.room_id;
|
let room_id = &body.room_id;
|
||||||
|
let sender_user = body.sender_user();
|
||||||
|
|
||||||
let event = services
|
let event = services
|
||||||
.rooms
|
.rooms
|
||||||
|
@ -33,6 +34,52 @@ pub(crate) async fn get_room_event_route(
|
||||||
return Err!(Request(Forbidden("You don't have permission to view this event.")));
|
return Err!(Request(Forbidden("You don't have permission to view this event.")));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let include_unredacted_content = body
|
||||||
|
.include_unredacted_content // User's file has this field name
|
||||||
|
.unwrap_or(false);
|
||||||
|
|
||||||
|
if include_unredacted_content && event.is_redacted() {
|
||||||
|
let is_server_admin = services
|
||||||
|
.users
|
||||||
|
.is_admin(sender_user)
|
||||||
|
.map(|is_admin| Ok(is_admin));
|
||||||
|
let can_redact_privilege = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.user_can_redact(event_id, sender_user, room_id, false) // federation=false for local check
|
||||||
|
;
|
||||||
|
let (is_server_admin, can_redact_privilege) =
|
||||||
|
try_join(is_server_admin, can_redact_privilege).await?;
|
||||||
|
|
||||||
|
if !is_server_admin && !can_redact_privilege {
|
||||||
|
return Err!(Request(Forbidden(
|
||||||
|
"You don't have permission to view redacted content.",
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
let pdu_id = match services.rooms.timeline.get_pdu_id(event_id).await {
|
||||||
|
| Ok(id) => id,
|
||||||
|
| Err(e) => {
|
||||||
|
return Err(e);
|
||||||
|
},
|
||||||
|
};
|
||||||
|
let original_content = services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.get_original_pdu_content(&pdu_id)
|
||||||
|
.await?;
|
||||||
|
if let Some(original_content) = original_content {
|
||||||
|
// If the original content is available, we can return it.
|
||||||
|
// event.content = to_raw_value(&original_content)?;
|
||||||
|
event = PduEvent::from_id_val(event_id, original_content)?;
|
||||||
|
} else {
|
||||||
|
return Err(conduwuit::Error::BadRequest(
|
||||||
|
ErrorKind::UnredactedContentDeleted { content_keep_ms: None },
|
||||||
|
"The original unredacted content is not in the database.",
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
debug_assert!(
|
debug_assert!(
|
||||||
event.event_id() == event_id && event.room_id() == room_id,
|
event.event_id() == event_id && event.room_id() == room_id,
|
||||||
"Fetched PDU must match requested"
|
"Fetched PDU must match requested"
|
||||||
|
|
|
@ -40,6 +40,7 @@ pub(crate) async fn get_supported_versions_route(
|
||||||
"v1.11".to_owned(),
|
"v1.11".to_owned(),
|
||||||
],
|
],
|
||||||
unstable_features: BTreeMap::from_iter([
|
unstable_features: BTreeMap::from_iter([
|
||||||
|
("fi.mau.msc2815".to_owned(), true),
|
||||||
("org.matrix.e2e_cross_signing".to_owned(), true),
|
("org.matrix.e2e_cross_signing".to_owned(), true),
|
||||||
("org.matrix.msc2285.stable".to_owned(), true), /* private read receipts (https://github.com/matrix-org/matrix-spec-proposals/pull/2285) */
|
("org.matrix.msc2285.stable".to_owned(), true), /* private read receipts (https://github.com/matrix-org/matrix-spec-proposals/pull/2285) */
|
||||||
("uk.half-shot.msc2666.query_mutual_rooms".to_owned(), true), /* query mutual rooms (https://github.com/matrix-org/matrix-spec-proposals/pull/2666) */
|
("uk.half-shot.msc2666.query_mutual_rooms".to_owned(), true), /* query mutual rooms (https://github.com/matrix-org/matrix-spec-proposals/pull/2666) */
|
||||||
|
|
|
@ -121,6 +121,15 @@ pub(super) static MAPS: &[Descriptor] = &[
|
||||||
index_size: 512,
|
index_size: 512,
|
||||||
..descriptor::SEQUENTIAL
|
..descriptor::SEQUENTIAL
|
||||||
},
|
},
|
||||||
|
Descriptor {
|
||||||
|
name: "pduid_originalcontent",
|
||||||
|
cache_disp: CacheDisp::SharedWith("pduid_pdu"),
|
||||||
|
key_size_hint: Some(16),
|
||||||
|
val_size_hint: Some(1520),
|
||||||
|
block_size: 2048,
|
||||||
|
index_size: 512,
|
||||||
|
..descriptor::RANDOM
|
||||||
|
},
|
||||||
Descriptor {
|
Descriptor {
|
||||||
name: "publicroomids",
|
name: "publicroomids",
|
||||||
..descriptor::RANDOM_SMALL
|
..descriptor::RANDOM_SMALL
|
||||||
|
|
|
@ -19,6 +19,8 @@ pub(super) struct Data {
|
||||||
pduid_pdu: Arc<Map>,
|
pduid_pdu: Arc<Map>,
|
||||||
userroomid_highlightcount: Arc<Map>,
|
userroomid_highlightcount: Arc<Map>,
|
||||||
userroomid_notificationcount: Arc<Map>,
|
userroomid_notificationcount: Arc<Map>,
|
||||||
|
/// Stores the original content of redacted PDUs.
|
||||||
|
pduid_originalcontent: Arc<Map>,
|
||||||
pub(super) db: Arc<Database>,
|
pub(super) db: Arc<Database>,
|
||||||
services: Services,
|
services: Services,
|
||||||
}
|
}
|
||||||
|
@ -38,6 +40,7 @@ impl Data {
|
||||||
pduid_pdu: db["pduid_pdu"].clone(),
|
pduid_pdu: db["pduid_pdu"].clone(),
|
||||||
userroomid_highlightcount: db["userroomid_highlightcount"].clone(),
|
userroomid_highlightcount: db["userroomid_highlightcount"].clone(),
|
||||||
userroomid_notificationcount: db["userroomid_notificationcount"].clone(),
|
userroomid_notificationcount: db["userroomid_notificationcount"].clone(),
|
||||||
|
pduid_originalcontent: db["pduid_originalcontent"].clone(), // Initialize new table
|
||||||
db: args.db.clone(),
|
db: args.db.clone(),
|
||||||
services: Services {
|
services: Services {
|
||||||
short: args.depend::<rooms::short::Service>("rooms::short"),
|
short: args.depend::<rooms::short::Service>("rooms::short"),
|
||||||
|
@ -177,6 +180,24 @@ impl Data {
|
||||||
self.pduid_pdu.get(pdu_id).await.deserialized()
|
self.pduid_pdu.get(pdu_id).await.deserialized()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Stores the original content of a PDU that is about to be redacted.
|
||||||
|
pub(super) async fn store_redacted_pdu_content(
|
||||||
|
&self,
|
||||||
|
pdu_id: &RawPduId,
|
||||||
|
pdu_json: &CanonicalJsonObject,
|
||||||
|
) -> Result<()> {
|
||||||
|
self.pduid_originalcontent.raw_put(pdu_id, Json(pdu_json));
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the original content of a redacted PDU.
|
||||||
|
pub(super) async fn get_original_pdu_content(
|
||||||
|
&self,
|
||||||
|
pdu_id: &RawPduId,
|
||||||
|
) -> Result<Option<CanonicalJsonObject>> {
|
||||||
|
self.pduid_originalcontent.get(pdu_id).await.deserialized()
|
||||||
|
}
|
||||||
|
|
||||||
pub(super) async fn append_pdu(
|
pub(super) async fn append_pdu(
|
||||||
&self,
|
&self,
|
||||||
pdu_id: &RawPduId,
|
pdu_id: &RawPduId,
|
||||||
|
|
|
@ -260,6 +260,25 @@ impl Service {
|
||||||
self.db.replace_pdu(pdu_id, pdu_json, pdu).await
|
self.db.replace_pdu(pdu_id, pdu_json, pdu).await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Stores the content of a to-be redacted pdu.
|
||||||
|
#[tracing::instrument(skip(self), level = "debug")]
|
||||||
|
pub async fn store_redacted_pdu_content(
|
||||||
|
&self,
|
||||||
|
pdu_id: &RawPduId,
|
||||||
|
pdu_json: &CanonicalJsonObject,
|
||||||
|
) -> Result<()> {
|
||||||
|
self.db.store_redacted_pdu_content(pdu_id, pdu_json).await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the original content of a redacted PDU.
|
||||||
|
#[tracing::instrument(skip(self), level = "debug")]
|
||||||
|
pub async fn get_original_pdu_content(
|
||||||
|
&self,
|
||||||
|
pdu_id: &RawPduId,
|
||||||
|
) -> Result<Option<CanonicalJsonObject>> {
|
||||||
|
self.db.get_original_pdu_content(pdu_id).await
|
||||||
|
}
|
||||||
|
|
||||||
/// Creates a new persisted data unit and adds it to a room.
|
/// Creates a new persisted data unit and adds it to a room.
|
||||||
///
|
///
|
||||||
/// By this point the incoming event should be fully authenticated, no auth
|
/// By this point the incoming event should be fully authenticated, no auth
|
||||||
|
@ -472,7 +491,7 @@ impl Service {
|
||||||
.user_can_redact(redact_id, &pdu.sender, &pdu.room_id, false)
|
.user_can_redact(redact_id, &pdu.sender, &pdu.room_id, false)
|
||||||
.await?
|
.await?
|
||||||
{
|
{
|
||||||
self.redact_pdu(redact_id, pdu, shortroomid).await?;
|
self.redact_pdu(redact_id, pdu, shortroomid, true).await?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
@ -485,7 +504,7 @@ impl Service {
|
||||||
.user_can_redact(redact_id, &pdu.sender, &pdu.room_id, false)
|
.user_can_redact(redact_id, &pdu.sender, &pdu.room_id, false)
|
||||||
.await?
|
.await?
|
||||||
{
|
{
|
||||||
self.redact_pdu(redact_id, pdu, shortroomid).await?;
|
self.redact_pdu(redact_id, pdu, shortroomid, true).await?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
@ -1033,6 +1052,7 @@ impl Service {
|
||||||
event_id: &EventId,
|
event_id: &EventId,
|
||||||
reason: &PduEvent,
|
reason: &PduEvent,
|
||||||
shortroomid: ShortRoomId,
|
shortroomid: ShortRoomId,
|
||||||
|
keep_original_content: bool,
|
||||||
) -> Result {
|
) -> Result {
|
||||||
// TODO: Don't reserialize, keep original json
|
// TODO: Don't reserialize, keep original json
|
||||||
let Ok(pdu_id) = self.get_pdu_id(event_id).await else {
|
let Ok(pdu_id) = self.get_pdu_id(event_id).await else {
|
||||||
|
@ -1054,6 +1074,19 @@ impl Service {
|
||||||
|
|
||||||
let room_version_id = self.services.state.get_room_version(&pdu.room_id).await?;
|
let room_version_id = self.services.state.get_room_version(&pdu.room_id).await?;
|
||||||
|
|
||||||
|
if keep_original_content && !pdu.is_redacted() {
|
||||||
|
let original_pdu_json = utils::to_canonical_object(&pdu).map_err(|e| {
|
||||||
|
err!(Database(error!(
|
||||||
|
?event_id,
|
||||||
|
?e,
|
||||||
|
"Failed to convert PDU to canonical JSON for original content storage"
|
||||||
|
)))
|
||||||
|
})?;
|
||||||
|
self.db
|
||||||
|
.store_redacted_pdu_content(&pdu_id, &original_pdu_json)
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
pdu.redact(&room_version_id, reason)?;
|
pdu.redact(&room_version_id, reason)?;
|
||||||
|
|
||||||
let obj = utils::to_canonical_object(&pdu).map_err(|e| {
|
let obj = utils::to_canonical_object(&pdu).map_err(|e| {
|
||||||
|
|
Loading…
Add table
Reference in a new issue