mirror of
https://github.com/element-hq/element-web.git
synced 2024-12-13 02:51:56 +03:00
228 lines
7.9 KiB
TypeScript
228 lines
7.9 KiB
TypeScript
/*
|
|
Copyright 2022 The Matrix.org Foundation C.I.C.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
/// <reference types="cypress" />
|
|
|
|
import type { ISendEventResponse, MatrixClient, Room } from "matrix-js-sdk/src/matrix";
|
|
import { HomeserverInstance } from "../plugins/utils/homeserver";
|
|
import { Credentials } from "./homeserver";
|
|
import Chainable = Cypress.Chainable;
|
|
|
|
interface CreateBotOpts {
|
|
/**
|
|
* A prefix to use for the userid. If unspecified, "bot_" will be used.
|
|
*/
|
|
userIdPrefix?: string;
|
|
/**
|
|
* Whether the bot should automatically accept all invites.
|
|
*/
|
|
autoAcceptInvites?: boolean;
|
|
/**
|
|
* The display name to give to that bot user
|
|
*/
|
|
displayName?: string;
|
|
/**
|
|
* Whether or not to start the syncing client.
|
|
*/
|
|
startClient?: boolean;
|
|
/**
|
|
* Whether or not to generate cross-signing keys
|
|
*/
|
|
bootstrapCrossSigning?: boolean;
|
|
}
|
|
|
|
const defaultCreateBotOptions = {
|
|
userIdPrefix: "bot_",
|
|
autoAcceptInvites: true,
|
|
startClient: true,
|
|
bootstrapCrossSigning: true,
|
|
} as CreateBotOpts;
|
|
|
|
export interface CypressBot extends MatrixClient {
|
|
__cypress_password: string;
|
|
}
|
|
|
|
declare global {
|
|
// eslint-disable-next-line @typescript-eslint/no-namespace
|
|
namespace Cypress {
|
|
interface Chainable {
|
|
/**
|
|
* Returns a new Bot instance
|
|
* @param homeserver the instance on which to register the bot user
|
|
* @param opts create bot options
|
|
*/
|
|
getBot(homeserver: HomeserverInstance, opts: CreateBotOpts): Chainable<CypressBot>;
|
|
/**
|
|
* Returns a new Bot instance logged in as an existing user
|
|
* @param homeserver the instance on which to register the bot user
|
|
* @param username the username for the bot to log in with
|
|
* @param password the password for the bot to log in with
|
|
* @param opts create bot options
|
|
*/
|
|
loginBot(
|
|
homeserver: HomeserverInstance,
|
|
username: string,
|
|
password: string,
|
|
opts: CreateBotOpts,
|
|
): Chainable<MatrixClient>;
|
|
/**
|
|
* Let a bot join a room
|
|
* @param cli The bot's MatrixClient
|
|
* @param roomId ID of the room to join
|
|
*/
|
|
botJoinRoom(cli: MatrixClient, roomId: string): Chainable<Room>;
|
|
/**
|
|
* Let a bot join a room by name
|
|
* @param cli The bot's MatrixClient
|
|
* @param roomName Name of the room to join
|
|
*/
|
|
botJoinRoomByName(cli: MatrixClient, roomName: string): Chainable<Room>;
|
|
/**
|
|
* Send a message as a bot into a room
|
|
* @param cli The bot's MatrixClient
|
|
* @param roomId ID of the room to join
|
|
* @param message the message body to send
|
|
*/
|
|
botSendMessage(cli: MatrixClient, roomId: string, message: string): Chainable<ISendEventResponse>;
|
|
}
|
|
}
|
|
}
|
|
|
|
function setupBotClient(
|
|
homeserver: HomeserverInstance,
|
|
credentials: Credentials,
|
|
opts: CreateBotOpts,
|
|
): Chainable<MatrixClient> {
|
|
opts = Object.assign({}, defaultCreateBotOptions, opts);
|
|
return cy.window({ log: false }).then((win) => {
|
|
const keys = {};
|
|
|
|
const getCrossSigningKey = (type: string) => {
|
|
return keys[type];
|
|
};
|
|
|
|
const saveCrossSigningKeys = (k: Record<string, Uint8Array>) => {
|
|
Object.assign(keys, k);
|
|
};
|
|
|
|
const cli = new win.matrixcs.MatrixClient({
|
|
baseUrl: homeserver.baseUrl,
|
|
userId: credentials.userId,
|
|
deviceId: credentials.deviceId,
|
|
accessToken: credentials.accessToken,
|
|
store: new win.matrixcs.MemoryStore(),
|
|
scheduler: new win.matrixcs.MatrixScheduler(),
|
|
cryptoStore: new win.matrixcs.MemoryCryptoStore(),
|
|
cryptoCallbacks: { getCrossSigningKey, saveCrossSigningKeys },
|
|
});
|
|
|
|
if (opts.autoAcceptInvites) {
|
|
cli.on(win.matrixcs.RoomMemberEvent.Membership, (event, member) => {
|
|
if (member.membership === "invite" && member.userId === cli.getUserId()) {
|
|
cli.joinRoom(member.roomId);
|
|
}
|
|
});
|
|
}
|
|
|
|
if (!opts.startClient) {
|
|
return cy.wrap(cli);
|
|
}
|
|
|
|
return cy.wrap(
|
|
cli
|
|
.initCrypto()
|
|
.then(() => cli.setGlobalErrorOnUnknownDevices(false))
|
|
.then(() => cli.startClient())
|
|
.then(async () => {
|
|
if (opts.bootstrapCrossSigning) {
|
|
await cli.bootstrapCrossSigning({
|
|
authUploadDeviceSigningKeys: async (func) => {
|
|
await func({
|
|
type: "m.login.password",
|
|
identifier: {
|
|
type: "m.id.user",
|
|
user: credentials.userId,
|
|
},
|
|
password: credentials.password,
|
|
});
|
|
},
|
|
});
|
|
}
|
|
})
|
|
.then(() => cli),
|
|
// extra timeout, as this sometimes takes a while
|
|
{ timeout: 30_000 },
|
|
);
|
|
});
|
|
}
|
|
|
|
Cypress.Commands.add("getBot", (homeserver: HomeserverInstance, opts: CreateBotOpts): Chainable<CypressBot> => {
|
|
opts = Object.assign({}, defaultCreateBotOptions, opts);
|
|
const username = Cypress._.uniqueId(opts.userIdPrefix);
|
|
const password = Cypress._.uniqueId("password_");
|
|
return cy
|
|
.registerUser(homeserver, username, password, opts.displayName)
|
|
.then((credentials) => {
|
|
cy.log(`Registered bot user ${username} with displayname ${opts.displayName}`);
|
|
return setupBotClient(homeserver, credentials, opts);
|
|
})
|
|
.then((client): Chainable<CypressBot> => {
|
|
Object.assign(client, { __cypress_password: password });
|
|
return cy.wrap(client as CypressBot);
|
|
});
|
|
});
|
|
|
|
Cypress.Commands.add(
|
|
"loginBot",
|
|
(
|
|
homeserver: HomeserverInstance,
|
|
username: string,
|
|
password: string,
|
|
opts: CreateBotOpts,
|
|
): Chainable<MatrixClient> => {
|
|
opts = Object.assign({}, defaultCreateBotOptions, { bootstrapCrossSigning: false }, opts);
|
|
return cy.loginUser(homeserver, username, password).then((credentials) => {
|
|
return setupBotClient(homeserver, credentials, opts);
|
|
});
|
|
},
|
|
);
|
|
|
|
Cypress.Commands.add("botJoinRoom", (cli: MatrixClient, roomId: string): Chainable<Room> => {
|
|
return cy.wrap(cli.joinRoom(roomId));
|
|
});
|
|
|
|
Cypress.Commands.add("botJoinRoomByName", (cli: MatrixClient, roomName: string): Chainable<Room> => {
|
|
const room = cli.getRooms().find((r) => r.getDefaultRoomName(cli.getUserId()) === roomName);
|
|
|
|
if (room) {
|
|
return cy.botJoinRoom(cli, room.roomId);
|
|
}
|
|
|
|
return cy.wrap(Promise.reject(`Bot room join failed. Cannot find room '${roomName}'`));
|
|
});
|
|
|
|
Cypress.Commands.add(
|
|
"botSendMessage",
|
|
(cli: MatrixClient, roomId: string, message: string): Chainable<ISendEventResponse> => {
|
|
return cy.wrap(
|
|
cli.sendMessage(roomId, {
|
|
msgtype: "m.text",
|
|
body: message,
|
|
}),
|
|
{ log: false },
|
|
);
|
|
},
|
|
);
|