Add more Synapse admin API wrappers (#181)

Co-authored-by: Tulir Asokan <tulir@maunium.net>
pull/190/head
G-ht 2024-02-24 14:06:27 +01:00 committed by GitHub
parent 581aa80155
commit cbd1334724
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 326 additions and 2 deletions

237
synapseadmin/roomapi.go Normal file
View File

@ -0,0 +1,237 @@
// Copyright (c) 2023 Tulir Asokan
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
package synapseadmin
import (
"context"
"encoding/json"
"net/http"
"strconv"
"maunium.net/go/mautrix"
"maunium.net/go/mautrix/event"
"maunium.net/go/mautrix/id"
)
type ReqListRoom struct {
SearchTerm string
OrderBy string
Direction mautrix.Direction
From int
Limit int
}
func (req *ReqListRoom) BuildQuery() map[string]string {
query := map[string]string{
"from": strconv.Itoa(req.From),
}
if req.SearchTerm != "" {
query["search_term"] = req.SearchTerm
}
if req.OrderBy != "" {
query["order_by"] = req.OrderBy
}
if req.Direction != 0 {
query["dir"] = string(req.Direction)
}
if req.Limit != 0 {
query["limit"] = strconv.Itoa(req.Limit)
}
return query
}
type RoomInfo struct {
RoomID id.RoomID `json:"room_id"`
Name string `json:"name"`
CanonicalAlias id.RoomAlias `json:"canonical_alias"`
JoinedMembers int `json:"joined_members"`
JoinedLocalMembers int `json:"joined_local_members"`
Version string `json:"version"`
Creator id.UserID `json:"creator"`
Encryption id.Algorithm `json:"encryption"`
Federatable bool `json:"federatable"`
Public bool `json:"public"`
JoinRules event.JoinRule `json:"join_rules"`
GuestAccess event.GuestAccess `json:"guest_access"`
HistoryVisibility event.HistoryVisibility `json:"history_visibility"`
StateEvents int `json:"state_events"`
RoomType event.RoomType `json:"room_type"`
}
type RespListRooms struct {
Rooms []RoomInfo `json:"rooms"`
Offset int `json:"offset"`
TotalRooms int `json:"total_rooms"`
NextBatch int `json:"next_batch"`
PrevBatch int `json:"prev_batch"`
}
// ListRooms returns a list of rooms on the server.
//
// https://matrix-org.github.io/synapse/latest/admin_api/rooms.html#list-room-api
func (cli *Client) ListRooms(ctx context.Context, req ReqListRoom) (RespListRooms, error) {
var resp RespListRooms
var reqURL string
reqURL = cli.BuildURLWithQuery(mautrix.SynapseAdminURLPath{"v1", "rooms"}, req.BuildQuery())
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodGet,
URL: reqURL,
ResponseJSON: &resp,
})
return resp, err
}
type RespRoomMessages = mautrix.RespMessages
// RoomMessages returns a list of messages in a room.
//
// https://matrix-org.github.io/synapse/latest/admin_api/rooms.html#room-messages-api
func (cli *Client) RoomMessages(ctx context.Context, roomID id.RoomID, from, to string, dir mautrix.Direction, filter *mautrix.FilterPart, limit int) (resp *RespRoomMessages, err error) {
query := map[string]string{
"from": from,
"dir": string(dir),
}
if filter != nil {
filterJSON, err := json.Marshal(filter)
if err != nil {
return nil, err
}
query["filter"] = string(filterJSON)
}
if to != "" {
query["to"] = to
}
if limit != 0 {
query["limit"] = strconv.Itoa(limit)
}
urlPath := cli.BuildURLWithQuery(mautrix.SynapseAdminURLPath{"v1", "rooms", roomID, "messages"}, query)
_, err = cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodGet,
URL: urlPath,
ResponseJSON: &resp,
})
return resp, err
}
type ReqDeleteRoom struct {
Purge bool `json:"purge,omitempty"`
Block bool `json:"block,omitempty"`
Message string `json:"message,omitempty"`
RoomName string `json:"room_name,omitempty"`
NewRoomUserID id.UserID `json:"new_room_user_id,omitempty"`
}
type RespDeleteRoom struct {
DeleteID string `json:"delete_id"`
}
// DeleteRoom deletes a room from the server, optionally blocking it and/or purging all data from the database.
//
// This calls the async version of the endpoint, which will return immediately and delete the room in the background.
//
// https://matrix-org.github.io/synapse/latest/admin_api/rooms.html#version-2-new-version
func (cli *Client) DeleteRoom(ctx context.Context, roomID id.RoomID, req ReqDeleteRoom) (RespDeleteRoom, error) {
reqURL := cli.BuildAdminURL("v2", "rooms", roomID)
var resp RespDeleteRoom
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodDelete,
URL: reqURL,
ResponseJSON: &resp,
RequestJSON: &req,
})
return resp, err
}
type RespRoomsMembers struct {
Members []id.UserID `json:"members"`
Total int `json:"total"`
}
// RoomMembers gets the full list of members in a room.
//
// https://matrix-org.github.io/synapse/latest/admin_api/rooms.html#room-members-api
func (cli *Client) RoomMembers(ctx context.Context, roomID id.RoomID) (RespRoomsMembers, error) {
reqURL := cli.BuildAdminURL("v1", "rooms", roomID, "members")
var resp RespRoomsMembers
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodGet,
URL: reqURL,
ResponseJSON: &resp,
})
return resp, err
}
type ReqMakeRoomAdmin struct {
UserID id.UserID `json:"user_id"`
}
// MakeRoomAdmin promotes a user to admin in a room. This requires that a local user has permission to promote users in the room.
//
// https://matrix-org.github.io/synapse/latest/admin_api/rooms.html#make-room-admin-api
func (cli *Client) MakeRoomAdmin(ctx context.Context, roomIDOrAlias string, req ReqMakeRoomAdmin) error {
reqURL := cli.BuildAdminURL("v1", "rooms", roomIDOrAlias, "make_room_admin")
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodPost,
URL: reqURL,
RequestJSON: &req,
})
return err
}
type ReqJoinUserToRoom struct {
UserID id.UserID `json:"user_id"`
}
// JoinUserToRoom makes a local user join the given room.
//
// https://matrix-org.github.io/synapse/latest/admin_api/room_membership.html
func (cli *Client) JoinUserToRoom(ctx context.Context, roomID id.RoomID, req ReqJoinUserToRoom) error {
reqURL := cli.BuildAdminURL("v1", "join", roomID)
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodPost,
URL: reqURL,
RequestJSON: &req,
})
return err
}
type ReqBlockRoom struct {
Block bool `json:"block"`
}
// BlockRoom blocks or unblocks a room.
//
// https://matrix-org.github.io/synapse/latest/admin_api/rooms.html#block-room-api
func (cli *Client) BlockRoom(ctx context.Context, roomID id.RoomID, req ReqBlockRoom) error {
reqURL := cli.BuildAdminURL("v1", "rooms", roomID, "block")
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodPut,
URL: reqURL,
RequestJSON: &req,
})
return err
}
// RoomsBlockResponse represents the response containing wether a room is blocked or not
type RoomsBlockResponse struct {
Block bool `json:"block"`
UserID id.UserID `json:"user_id"`
}
// GetRoomBlockStatus gets whether a room is currently blocked.
//
// https://matrix-org.github.io/synapse/latest/admin_api/rooms.html#get-block-status
func (cli *Client) GetRoomBlockStatus(ctx context.Context, roomID id.RoomID) (RoomsBlockResponse, error) {
var resp RoomsBlockResponse
reqURL := cli.BuildAdminURL("v1", "rooms", roomID, "block")
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodGet,
URL: reqURL,
ResponseJSON: &resp,
})
return resp, err
}

View File

@ -21,7 +21,6 @@ import (
type ReqResetPassword struct {
// The user whose password to reset.
UserID id.UserID `json:"-"`
// The new password for the user. Required.
NewPassword string `json:"new_password"`
// Whether all the user's existing devices should be logged out after the password change.
@ -86,7 +85,7 @@ type RespUserInfo struct {
UserID id.UserID `json:"name"`
DisplayName string `json:"displayname"`
AvatarURL id.ContentURIString `json:"avatar_url"`
Guest int `json:"is_guest"`
Guest bool `json:"is_guest"`
Admin bool `json:"admin"`
Deactivated bool `json:"deactivated"`
Erased bool `json:"erased"`
@ -109,3 +108,91 @@ func (cli *Client) GetUserInfo(ctx context.Context, userID id.UserID) (resp *Res
})
return
}
type ReqDeleteUser struct {
Erase bool `json:"erase"`
}
// DeactivateAccount deactivates a specific local user account.
//
// https://matrix-org.github.io/synapse/latest/admin_api/user_admin_api.html#deactivate-account
func (cli *Client) DeactivateAccount(ctx context.Context, userID id.UserID, req ReqDeleteUser) error {
reqURL := cli.BuildAdminURL("v1", "deactivate", userID)
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodPost,
URL: reqURL,
RequestJSON: &req,
})
return err
}
type ReqCreateOrModifyAccount struct {
Password string `json:"password,omitempty"`
LogoutDevices *bool `json:"logout_devices,omitempty"`
Deactivated *bool `json:"deactivated,omitempty"`
Admin *bool `json:"admin,omitempty"`
Locked *bool `json:"locked,omitempty"`
Displayname string `json:"displayname,omitempty"`
AvatarURL id.ContentURIString `json:"avatar_url,omitempty"`
UserType string `json:"user_type,omitempty"`
}
// CreateOrModifyAccount creates or modifies an account on the server.
//
// https://matrix-org.github.io/synapse/latest/admin_api/user_admin_api.html#create-or-modify-account
func (cli *Client) CreateOrModifyAccount(ctx context.Context, userID id.UserID, req ReqCreateOrModifyAccount) error {
reqURL := cli.BuildAdminURL("v2", "users", userID)
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodPut,
URL: reqURL,
RequestJSON: &req,
})
return err
}
type RatelimitOverride struct {
MessagesPerSecond int `json:"messages_per_second"`
BurstCount int `json:"burst_count"`
}
type ReqSetRatelimit = RatelimitOverride
// SetUserRatelimit overrides the message sending ratelimit for a specific user.
//
// https://matrix-org.github.io/synapse/latest/admin_api/user_admin_api.html#set-ratelimit
func (cli *Client) SetUserRatelimit(ctx context.Context, userID id.UserID, req ReqSetRatelimit) error {
reqURL := cli.BuildAdminURL("v1", "users", userID, "override_ratelimit")
_, err := cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodPost,
URL: reqURL,
RequestJSON: &req,
})
return err
}
type RespUserRatelimit = RatelimitOverride
// GetUserRatelimit gets the ratelimit override for the given user.
//
// https://matrix-org.github.io/synapse/latest/admin_api/user_admin_api.html#get-status-of-ratelimit
func (cli *Client) GetUserRatelimit(ctx context.Context, userID id.UserID) (resp RespUserRatelimit, err error) {
_, err = cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodGet,
URL: cli.BuildAdminURL("v1", "users", userID, "override_ratelimit"),
ResponseJSON: &resp,
})
return
}
// DeleteUserRatelimit deletes the ratelimit override for the given user, returning them to the default ratelimits.
//
// https://matrix-org.github.io/synapse/latest/admin_api/user_admin_api.html#delete-ratelimit
func (cli *Client) DeleteUserRatelimit(ctx context.Context, userID id.UserID) (err error) {
_, err = cli.MakeFullRequest(ctx, mautrix.FullRequest{
Method: http.MethodDelete,
URL: cli.BuildAdminURL("v1", "users", userID, "override_ratelimit"),
})
return
}