mirror of
https://github.com/matrix-org/dendrite
synced 2024-06-01 03:59:03 +02:00
Add getting a single event report
This commit is contained in:
parent
b62095b80f
commit
7df4d11233
|
@ -272,6 +272,7 @@ type ClientRoomserverAPI interface {
|
|||
score int64,
|
||||
) (int64, error)
|
||||
QueryAdminEventReports(ctx context.Context, from, limit uint64, backwards bool, userID, roomID string) ([]QueryAdminEventReportsResponse, int64, error)
|
||||
QueryAdminEventReport(ctx context.Context, reportID uint64) (QueryAdminEventReportResponse, error)
|
||||
}
|
||||
|
||||
type UserRoomserverAPI interface {
|
||||
|
|
|
@ -363,6 +363,11 @@ type QueryAdminEventReportsResponse struct {
|
|||
ReceivedTS spec.Timestamp `json:"received_ts"`
|
||||
}
|
||||
|
||||
type QueryAdminEventReportResponse struct {
|
||||
QueryAdminEventReportsResponse
|
||||
EventJSON json.RawMessage `json:"event_json"`
|
||||
}
|
||||
|
||||
// MarshalJSON stringifies the room ID and StateKeyTuple keys so they can be sent over the wire in HTTP API mode.
|
||||
func (r *QueryBulkStateContentResponse) MarshalJSON() ([]byte, error) {
|
||||
se := make(map[string]string)
|
||||
|
|
|
@ -1109,3 +1109,8 @@ func (r *Queryer) RoomsWithACLs(ctx context.Context) ([]string, error) {
|
|||
func (r *Queryer) QueryAdminEventReports(ctx context.Context, from uint64, limit uint64, backwards bool, userID, roomID string) ([]api.QueryAdminEventReportsResponse, int64, error) {
|
||||
return r.DB.QueryAdminEventReports(ctx, from, limit, backwards, userID, roomID)
|
||||
}
|
||||
|
||||
// QueryAdminEventReport returns a single event report.
|
||||
func (r *Queryer) QueryAdminEventReport(ctx context.Context, reportID uint64) (api.QueryAdminEventReportResponse, error) {
|
||||
return r.DB.QueryAdminEventReport(ctx, reportID)
|
||||
}
|
||||
|
|
|
@ -196,6 +196,7 @@ type Database interface {
|
|||
// RoomsWithACLs returns all room IDs for rooms with ACLs
|
||||
RoomsWithACLs(ctx context.Context) ([]string, error)
|
||||
QueryAdminEventReports(ctx context.Context, from uint64, limit uint64, backwards bool, userID string, roomID string) ([]api.QueryAdminEventReportsResponse, int64, error)
|
||||
QueryAdminEventReport(ctx context.Context, reportID uint64) (api.QueryAdminEventReportResponse, error)
|
||||
}
|
||||
|
||||
type UserRoomKeys interface {
|
||||
|
|
|
@ -75,10 +75,17 @@ OFFSET $3
|
|||
LIMIT $4
|
||||
`
|
||||
|
||||
const selectReportedEventSQL = `
|
||||
SELECT id, room_nid, event_nid, reporting_user_nid, event_sender_nid, reason, score, received_ts
|
||||
FROM roomserver_reported_events
|
||||
WHERE id = $1
|
||||
`
|
||||
|
||||
type reportedEventsStatements struct {
|
||||
insertReportedEventsStmt *sql.Stmt
|
||||
selectReportedEventsDescStmt *sql.Stmt
|
||||
selectReportedEventsAscStmt *sql.Stmt
|
||||
selectReportedEventStmt *sql.Stmt
|
||||
}
|
||||
|
||||
func CreateReportedEventsTable(db *sql.DB) error {
|
||||
|
@ -93,6 +100,7 @@ func PrepareReportedEventsTable(db *sql.DB) (tables.ReportedEvents, error) {
|
|||
{&s.insertReportedEventsStmt, insertReportedEventSQL},
|
||||
{&s.selectReportedEventsDescStmt, selectReportedEventsDescSQL},
|
||||
{&s.selectReportedEventsAscStmt, selectReportedEventsAscSQL},
|
||||
{&s.selectReportedEventStmt, selectReportedEventSQL},
|
||||
}.Prepare(db)
|
||||
}
|
||||
|
||||
|
@ -178,3 +186,26 @@ func (r *reportedEventsStatements) SelectReportedEvents(
|
|||
|
||||
return result, count, rows.Err()
|
||||
}
|
||||
|
||||
func (r *reportedEventsStatements) SelectReportedEvent(
|
||||
ctx context.Context,
|
||||
txn *sql.Tx,
|
||||
reportID uint64,
|
||||
) (api.QueryAdminEventReportResponse, error) {
|
||||
stmt := sqlutil.TxStmt(txn, r.selectReportedEventStmt)
|
||||
|
||||
var row api.QueryAdminEventReportResponse
|
||||
if err := stmt.QueryRowContext(ctx, reportID).Scan(
|
||||
&row.ID,
|
||||
&row.RoomNID,
|
||||
&row.EventNID,
|
||||
&row.ReportingUserNID,
|
||||
&row.SenderNID,
|
||||
&row.Reason,
|
||||
&row.Score,
|
||||
&row.ReceivedTS,
|
||||
); err != nil {
|
||||
return api.QueryAdminEventReportResponse{}, err
|
||||
}
|
||||
return row, nil
|
||||
}
|
||||
|
|
|
@ -2040,6 +2040,68 @@ func (d *Database) QueryAdminEventReports(ctx context.Context, from uint64, limi
|
|||
return reports, count, nil
|
||||
}
|
||||
|
||||
func (d *Database) QueryAdminEventReport(ctx context.Context, reportID uint64) (api.QueryAdminEventReportResponse, error) {
|
||||
|
||||
report, err := d.ReportedEventsTable.SelectReportedEvent(ctx, nil, reportID)
|
||||
if err != nil {
|
||||
return api.QueryAdminEventReportResponse{}, err
|
||||
}
|
||||
|
||||
// Get a map from EventStateKeyNID to userID
|
||||
userNIDMap, err := d.EventStateKeys(ctx, []types.EventStateKeyNID{report.ReportingUserNID, report.SenderNID})
|
||||
if err != nil {
|
||||
logrus.WithError(err).Error("unable to map userNIDs to userIDs")
|
||||
return report, err
|
||||
}
|
||||
|
||||
roomIDs, err := d.RoomsTable.BulkSelectRoomIDs(ctx, nil, []types.RoomNID{report.RoomNID})
|
||||
if err != nil {
|
||||
return report, err
|
||||
}
|
||||
|
||||
if len(roomIDs) != 1 {
|
||||
return report, fmt.Errorf("expected one roomID, got %d", len(roomIDs))
|
||||
}
|
||||
|
||||
// TODO: replace this with something more efficient, as it loads the entire state snapshot.
|
||||
stateContent, err := d.GetBulkStateContent(ctx, roomIDs, []gomatrixserverlib.StateKeyTuple{
|
||||
{EventType: spec.MRoomName, StateKey: ""},
|
||||
{EventType: spec.MRoomCanonicalAlias, StateKey: ""},
|
||||
}, false)
|
||||
if err != nil {
|
||||
return report, err
|
||||
}
|
||||
|
||||
eventIDMap, err := d.EventIDs(ctx, []types.EventNID{report.EventNID})
|
||||
if err != nil {
|
||||
logrus.WithError(err).Error("unable to map eventNIDs to eventIDs")
|
||||
return report, err
|
||||
}
|
||||
if len(eventIDMap) != 1 {
|
||||
return report, fmt.Errorf("expected %d eventIDs, got %d", 1, len(eventIDMap))
|
||||
}
|
||||
|
||||
eventJSONs, err := d.EventJSONTable.BulkSelectEventJSON(ctx, nil, []types.EventNID{report.EventNID})
|
||||
if err != nil {
|
||||
return report, err
|
||||
}
|
||||
if len(eventJSONs) != 1 {
|
||||
return report, fmt.Errorf("expected %d eventJSONs, got %d", 1, len(eventJSONs))
|
||||
}
|
||||
|
||||
roomName, canonicalAlias := findRoomNameAndCanonicalAlias(stateContent, roomIDs[0])
|
||||
|
||||
report.Sender = userNIDMap[report.SenderNID]
|
||||
report.UserID = userNIDMap[report.ReportingUserNID]
|
||||
report.RoomID = roomIDs[0]
|
||||
report.RoomName = roomName
|
||||
report.CanonicalAlias = canonicalAlias
|
||||
report.EventID = eventIDMap[report.EventNID]
|
||||
report.EventJSON = eventJSONs[0].EventJSON
|
||||
|
||||
return report, nil
|
||||
}
|
||||
|
||||
// findRoomNameAndCanonicalAlias loops over events to find the corresponding room name and canonicalAlias
|
||||
// for a given roomID.
|
||||
func findRoomNameAndCanonicalAlias(events []tables.StrippedEvent, roomID string) (name, canonicalAlias string) {
|
||||
|
|
|
@ -74,10 +74,17 @@ LIMIT $3
|
|||
OFFSET $4
|
||||
`
|
||||
|
||||
const selectReportedEventSQL = `
|
||||
SELECT id, room_nid, event_nid, reporting_user_nid, event_sender_nid, reason, score, received_ts
|
||||
FROM roomserver_reported_events
|
||||
WHERE id = $1
|
||||
`
|
||||
|
||||
type reportedEventsStatements struct {
|
||||
insertReportedEventsStmt *sql.Stmt
|
||||
selectReportedEventsDescStmt *sql.Stmt
|
||||
selectReportedEventsAscStmt *sql.Stmt
|
||||
selectReportedEventStmt *sql.Stmt
|
||||
}
|
||||
|
||||
func CreateReportedEventsTable(db *sql.DB) error {
|
||||
|
@ -92,6 +99,7 @@ func PrepareReportedEventsTable(db *sql.DB) (tables.ReportedEvents, error) {
|
|||
{&s.insertReportedEventsStmt, insertReportedEventSQL},
|
||||
{&s.selectReportedEventsDescStmt, selectReportedEventsDescSQL},
|
||||
{&s.selectReportedEventsAscStmt, selectReportedEventsAscSQL},
|
||||
{&s.selectReportedEventStmt, selectReportedEventSQL},
|
||||
}.Prepare(db)
|
||||
}
|
||||
|
||||
|
@ -178,3 +186,26 @@ func (r *reportedEventsStatements) SelectReportedEvents(
|
|||
|
||||
return result, count, rows.Err()
|
||||
}
|
||||
|
||||
func (r *reportedEventsStatements) SelectReportedEvent(
|
||||
ctx context.Context,
|
||||
txn *sql.Tx,
|
||||
reportID uint64,
|
||||
) (api.QueryAdminEventReportResponse, error) {
|
||||
stmt := sqlutil.TxStmt(txn, r.selectReportedEventStmt)
|
||||
|
||||
var row api.QueryAdminEventReportResponse
|
||||
if err := stmt.QueryRowContext(ctx, reportID).Scan(
|
||||
&row.ID,
|
||||
&row.RoomNID,
|
||||
&row.EventNID,
|
||||
&row.ReportingUserNID,
|
||||
&row.SenderNID,
|
||||
&row.Reason,
|
||||
&row.Score,
|
||||
&row.ReceivedTS,
|
||||
); err != nil {
|
||||
return api.QueryAdminEventReportResponse{}, err
|
||||
}
|
||||
return row, nil
|
||||
}
|
||||
|
|
|
@ -147,6 +147,11 @@ type ReportedEvents interface {
|
|||
reportingUserID types.EventStateKeyNID,
|
||||
roomNID types.RoomNID,
|
||||
) ([]api.QueryAdminEventReportsResponse, int64, error)
|
||||
SelectReportedEvent(
|
||||
ctx context.Context,
|
||||
txn *sql.Tx,
|
||||
reportID uint64,
|
||||
) (api.QueryAdminEventReportResponse, error)
|
||||
}
|
||||
|
||||
type MembershipState int64
|
||||
|
|
Loading…
Reference in a new issue