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 conduwuit::{Err, Event, Result, err};
|
||||
use conduwuit::{Err, Event, PduEvent, Result, err};
|
||||
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};
|
||||
|
||||
|
@ -14,6 +14,7 @@ pub(crate) async fn get_room_event_route(
|
|||
) -> Result<get_room_event::v3::Response> {
|
||||
let event_id = &body.event_id;
|
||||
let room_id = &body.room_id;
|
||||
let sender_user = body.sender_user();
|
||||
|
||||
let event = services
|
||||
.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.")));
|
||||
}
|
||||
|
||||
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!(
|
||||
event.event_id() == event_id && event.room_id() == room_id,
|
||||
"Fetched PDU must match requested"
|
||||
|
|
|
@ -40,6 +40,7 @@ pub(crate) async fn get_supported_versions_route(
|
|||
"v1.11".to_owned(),
|
||||
],
|
||||
unstable_features: BTreeMap::from_iter([
|
||||
("fi.mau.msc2815".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) */
|
||||
("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,
|
||||
..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 {
|
||||
name: "publicroomids",
|
||||
..descriptor::RANDOM_SMALL
|
||||
|
|
|
@ -19,6 +19,8 @@ pub(super) struct Data {
|
|||
pduid_pdu: Arc<Map>,
|
||||
userroomid_highlightcount: Arc<Map>,
|
||||
userroomid_notificationcount: Arc<Map>,
|
||||
/// Stores the original content of redacted PDUs.
|
||||
pduid_originalcontent: Arc<Map>,
|
||||
pub(super) db: Arc<Database>,
|
||||
services: Services,
|
||||
}
|
||||
|
@ -38,6 +40,7 @@ impl Data {
|
|||
pduid_pdu: db["pduid_pdu"].clone(),
|
||||
userroomid_highlightcount: db["userroomid_highlightcount"].clone(),
|
||||
userroomid_notificationcount: db["userroomid_notificationcount"].clone(),
|
||||
pduid_originalcontent: db["pduid_originalcontent"].clone(), // Initialize new table
|
||||
db: args.db.clone(),
|
||||
services: Services {
|
||||
short: args.depend::<rooms::short::Service>("rooms::short"),
|
||||
|
@ -177,6 +180,24 @@ impl Data {
|
|||
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(
|
||||
&self,
|
||||
pdu_id: &RawPduId,
|
||||
|
|
|
@ -260,6 +260,25 @@ impl Service {
|
|||
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.
|
||||
///
|
||||
/// 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)
|
||||
.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)
|
||||
.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,
|
||||
reason: &PduEvent,
|
||||
shortroomid: ShortRoomId,
|
||||
keep_original_content: bool,
|
||||
) -> Result {
|
||||
// TODO: Don't reserialize, keep original json
|
||||
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?;
|
||||
|
||||
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)?;
|
||||
|
||||
let obj = utils::to_canonical_object(&pdu).map_err(|e| {
|
||||
|
|
Loading…
Add table
Reference in a new issue