Add JSDocs to app.js (ref #1)

This commit is contained in:
Tulir Asokan
2017-11-21 17:26:12 +02:00
parent 674ea9eab4
commit cb0d931652
2 changed files with 100 additions and 15 deletions
+87 -5
View File
@@ -19,7 +19,15 @@ const MatrixUser = require("./matrix-user")
const TelegramUser = require("./telegram-user") const TelegramUser = require("./telegram-user")
const Portal = require("./portal") const Portal = require("./portal")
/**
* The base class for the bridge.
*/
class MautrixTelegram { class MautrixTelegram {
/**
* Create a MautrixTelegram instance with the given config data.
*
* @param config The data from the config file.
*/
constructor(config) { constructor(config) {
this.config = config this.config = config
@@ -49,32 +57,57 @@ class MautrixTelegram {
}) })
} }
/**
* Start the bridge.
*/
async run() { async run() {
console.log("Appservice listening on port %s", this.config.appservice.port) console.log("Appservice listening on port %s", this.config.appservice.port)
await this.bridge.run(this.config.appservice.port, {}) await this.bridge.run(this.config.appservice.port, {})
const userEntries = await this.bridge.getUserStore() const userEntries = await this.bridge.getUserStore()
.select({ .select({ type: "matrix" })
type: "matrix",
})
for (const entry of userEntries) { for (const entry of userEntries) {
const user = MatrixUser.fromEntry(this, entry) const user = MatrixUser.fromEntry(this, entry)
this.matrixUsersByID.set(entry.id, user) this.matrixUsersByID.set(entry.id, user)
} }
} }
/**
* The {@link MatrixClient} object for the appservice bot.
*/
get bot() { get bot() {
return this.bridge.getBot() return this.bridge.getBot()
} }
/**
* The {@link Intent} object for the appservice bot.
*/
get botIntent() { get botIntent() {
return this.bridge.getIntent() return this.bridge.getIntent()
} }
/**
* Get the {@link Intent} for the Telegram user with the given ID.
*
* This does not care if a {@link TelegramUser} object for the user ID exists.
* It simply returns an intent for a Matrix puppet user with the correct MXID.
*
* @param {number} id The ID of the Telegram user.
* @returns {Intent} The Matrix puppet intent for the given Telegram user.
*/
getIntentForTelegramUser(id) { getIntentForTelegramUser(id) {
return this.bridge.getIntentFromLocalpart( return this.bridge.getIntentFromLocalpart(
this.config.bridge.username_template.replace("${ID}", id)) this.config.bridge.username_template.replace("${ID}", id))
} }
/**
* Get a {@link Portal} by Telegram peer.
*
* This will either get the room from the room cache or the bridge room database.
* If the room is not found, a new {@link Portal} object is created.
*
* @param {TelegramPeer} peer The TelegramPeer object whose portal to get.
* @returns {Promise<Portal>} The Portal object.
*/
async getPortalByPeer(peer) { async getPortalByPeer(peer) {
let portal = this.portalsByPeerID.get(peer.id) let portal = this.portalsByPeerID.get(peer.id)
if (portal) { if (portal) {
@@ -109,6 +142,16 @@ class MautrixTelegram {
return portal return portal
} }
/**
* Get a {@link Portal} by Matrix room ID.
*
* This will either get the room from the room cache or the bridge room database.
* If the room is not found, this function WILL NOT create a new room,
* but rather just return {@linkplain undefined}.
*
* @param {string} id The Matrix room ID of the portal to get.
* @returns {Promise<Portal>} The Portal object.
*/
async getPortalByRoomID(id) { async getPortalByRoomID(id) {
let portal = this.portalsByRoomID.get(id) let portal = this.portalsByRoomID.get(id)
if (portal) { if (portal) {
@@ -116,6 +159,7 @@ class MautrixTelegram {
} }
// Check if we have it stored in the by-peer map // Check if we have it stored in the by-peer map
// FIXME this is probably useless
for (const [_, portalByPeer] of this.portalsByPeerID) { for (const [_, portalByPeer] of this.portalsByPeerID) {
if (portalByPeer.roomID === id) { if (portalByPeer.roomID === id) {
this.portalsByRoomID.set(id, portal) this.portalsByRoomID.set(id, portal)
@@ -146,6 +190,15 @@ class MautrixTelegram {
return portal return portal
} }
/**
* Get a {@link TelegramUser} by ID.
*
* This will either get the user from the user cache or the bridge user database.
* If the user is not found, a new {@link TelegramUser} instance is created.
*
* @param {number} id The internal Telegram ID of the user to get.
* @returns {Promise<TelegramUser>} The TelegramUser object.
*/
async getTelegramUser(id) { async getTelegramUser(id) {
let user = this.telegramUsersByID.get(id) let user = this.telegramUsersByID.get(id)
if (user) { if (user) {
@@ -172,6 +225,15 @@ class MautrixTelegram {
return user return user
} }
/**
* Get a {@link MatrixUser} by ID.
*
* This will either get the user from the user cache or the bridge user database.
* If the user is not found, a new {@link MatrixUser} instance is created.
*
* @param {string} id The MXID of the Matrix user to get.
* @returns {Promise<MatrixUser>} The MatrixUser object.
*/
async getMatrixUser(id) { async getMatrixUser(id) {
let user = this.matrixUsersByID.get(id) let user = this.matrixUsersByID.get(id)
if (user) { if (user) {
@@ -198,6 +260,11 @@ class MautrixTelegram {
return user return user
} }
/**
* Save a user to the bridge user database.
*
* @param {MatrixUser|TelegramUser} user The user object to save.
*/
putUser(user) { putUser(user) {
const entry = user.toEntry() const entry = user.toEntry()
return this.bridge.getUserStore() return this.bridge.getUserStore()
@@ -207,6 +274,11 @@ class MautrixTelegram {
}, entry) }, entry)
} }
/**
* Save a room to the bridge room database.
*
* @param {Room} room The Room object to save.
*/
putRoom(room) { putRoom(room) {
const entry = room.toEntry() const entry = room.toEntry()
return this.bridge.getRoomStore() return this.bridge.getRoomStore()
@@ -216,6 +288,11 @@ class MautrixTelegram {
}, entry) }, entry)
} }
/**
* Handle a single received Matrix event.
*
* @param evt The Matrix event that occurred.
*/
async handleMatrixEvent(evt) { async handleMatrixEvent(evt) {
const asBotID = this.bridge.getBot() const asBotID = this.bridge.getBot()
.getUserId() .getUserId()
@@ -251,8 +328,7 @@ class MautrixTelegram {
const args = evt.content.body.substr(prefixLength) const args = evt.content.body.substr(prefixLength)
.split(" ") .split(" ")
const command = args.shift() const command = args.shift()
commands.run( commands.run(user, command, args,
user, command, args,
reply => this.botIntent.sendText( reply => this.botIntent.sendText(
evt.room_id, evt.room_id,
reply.replace("$cmdprefix", cmdprefix)), reply.replace("$cmdprefix", cmdprefix)),
@@ -271,6 +347,12 @@ class MautrixTelegram {
} }
} }
/**
* Check whether the given user ID is allowed to use this bridge.
*
* @param {string} userID The full Matrix ID to check (@user:homeserver.tld)
* @returns {boolean} Whether or not the user should be allowed to use the bridge.
*/
checkWhitelist(userID) { checkWhitelist(userID) {
if (!this.config.bridge.whitelist || this.config.bridge.whitelist.length === 0) { if (!this.config.bridge.whitelist || this.config.bridge.whitelist.length === 0) {
return true return true
+13 -10
View File
@@ -15,25 +15,26 @@
// You should have received a copy of the GNU General Public License // You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>. // along with this program. If not, see <http://www.gnu.org/licenses/>.
const { AppServiceRegistration } = require("matrix-appservice-bridge") const { AppServiceRegistration } = require("matrix-appservice-bridge")
const commander = require("commander") const program = require("commander")
const YAML = require("yamljs") const YAML = require("yamljs")
const fs = require("fs") const fs = require("fs")
const MautrixTelegram = require("./app") const MautrixTelegram = require("./app")
const pkg = require("../package.json") const pkg = require("../package.json")
commander program
.version(pkg.version) .version(pkg.version)
.option("-c, --config <path>", "the file to load the config from. defaults to ./config.yaml") .option("-c, --config <path>", "the file to load the config from. defaults to ./config.yaml")
.option("-g, --generate-registration", "generate a registration based on the config") .option("-g, --generate-registration", "generate a registration based on the config")
.option("-r, --registration <path>", "the file to save the registration to. defaults to ./registration.yaml") .option("-r, --registration <path>", "the file to save the registration to. defaults to ./registration.yaml")
.parse(process.argv) .parse(process.argv)
commander.registration = commander.registration || "./registration.yaml" // commander doesn't seem to set default values automatically.
commander.config = commander.config || "./config.yaml" program.registration = program.registration || "./registration.yaml"
program.config = program.config || "./config.yaml"
const config = YAML.load(commander.config) const config = YAML.load(program.config)
if (commander.generateRegistration) { if (program.generateRegistration) {
const registration = { const registration = {
id: config.appservice.id, id: config.appservice.id,
hs_token: AppServiceRegistration.generateToken(), hs_token: AppServiceRegistration.generateToken(),
@@ -53,10 +54,12 @@ if (commander.generateRegistration) {
sender_localpart: config.bridge.bot_username, sender_localpart: config.bridge.bot_username,
rate_limited: false, rate_limited: false,
} }
fs.writeFileSync(commander.registration, YAML.stringify(registration, 10))
config.appservice.registration = commander.registration fs.writeFileSync(program.registration, YAML.stringify(registration, 10))
fs.writeFileSync(commander.config, YAML.stringify(config, 10)) config.appservice.registration = program.registration
console.log("Registration generated and saved to", commander.registration) fs.writeFileSync(program.config, YAML.stringify(config, 10))
console.log("Registration generated and saved to", program.registration)
process.exit() process.exit()
} }