set up initial member, topic, and pl sync

This commit is contained in:
emerson 2021-12-05 20:22:28 -05:00
parent e201b54d90
commit 98a096e18f
No known key found for this signature in database
GPG key ID: 270669502DA603E3
3 changed files with 73 additions and 9 deletions

View file

@ -4,28 +4,53 @@ import { IRCUser } from "./IRCUser.js";
import { IRCMessage } from "./Message.js";
export class Channel {
private server: Server
public name: string
private matrixUsers: Map<string, MatrixUser>
private ircUsers: Map<string, IRCUser>
private nickToMXid: Map<string, string>
private powerLevels: Map<string, number>
private topic: Map<string, string>;
private memberCount: number;
private modes: Map<string, string>
private messages: Map<string, IRCMessage>;
private tsToEventId: Map<number, string>;
constructor(public roomId: string, private server: Server, initialIRCUser: IRCUser) {
constructor(public roomId: string, initialIRCUser: IRCUser) {
this.server = initialIRCUser.server
this.name = roomId;
this.matrixUsers = new Map();
this.ircUsers = new Map();
this.ircUsers.set(initialIRCUser.nick, initialIRCUser);
this.nickToMXid = new Map();
this.powerLevels = new Map();
this.topic = new Map();
this.memberCount = 0;
this.topic = new Map([['text', ''], ['timestamp', '0']]);
this.modes = new Map();
this.modes.set('n', '');
this.messages = new Map();
this.tsToEventId = new Map();
}
handleMatrixEvent(event: any) {
if (!event["type"])
return;
if (event["type"] === "m.room.member") {
const thisMatrixUser = this.server.getOrCreateMatrixUser(event["sender"]);
this.matrixUsers.set(thisMatrixUser.nick, thisMatrixUser);
}
else if (event["type"] === "m.room.topic") {
this.topic.set("text", event["content"]["topic"]);
this.topic.set("timestamp", event["origin_server_ts"].toString())
}
else if (event["type"] === "m.room.power_levels") {
const allUsers = event["content"]["users"];
for (const [mxid, pl] of Object.entries(allUsers)) {
const thisMatrixUser = this.server.getOrCreateMatrixUser(event["sender"]);
this.matrixUsers.set(thisMatrixUser.nick, thisMatrixUser);
this.powerLevels.set(thisMatrixUser.nick, Number(pl));
}
}
else {
console.log(event);
}
}
}

View file

@ -1,25 +1,23 @@
import axios from "axios";
import { Channel } from "./Channel.js";
import { Client } from "./Client.js";
import { MatrixUser } from "./MatrixUser.js";
import { IRCMessage } from "./Message.js";
import { Server } from "./Server.js";
export class IRCUser {
private clients: Set<Client>
private channels: Map<string, Channel>
private server: Server
private matrixUser: MatrixUser|null
public server: Server
public nick: string
private ident: string
private hostname: string
public accountName: string
public isAuthed: boolean
private txnIdStore: Set<string>
private nextBatch: string
constructor(private initialClient: Client, public mxid: string, private accessToken: string) {
this.clients = new Set([initialClient]);
this.channels = new Map();
this.server = initialClient.server;
this.matrixUser = null;
const mxidSplit = mxid.split(':')
this.nick = mxidSplit[0].substr(1);
this.ident = this.nick;
@ -27,6 +25,8 @@ export class IRCUser {
this.accountName = mxid.slice(1);
this.isAuthed = false;
this.txnIdStore = new Set();
this.nextBatch = "";
this.doSync();
}
verifyCredentials(accessToken: string): boolean {
@ -36,4 +36,22 @@ export class IRCUser {
getMask(): string {
return `${this.nick}!${this.ident}@${this.hostname}`;
}
doSync(): void {
let endpoint = `https://matrix.org/_matrix/client/v3/sync?access_token=${this.accessToken}`;
if (this.nextBatch !== "")
endpoint = `${endpoint}&since=${this.nextBatch}`;
axios.get(endpoint).then(response => {
const data = response.data;
this.nextBatch = data.next_batch;
const rooms = data.rooms;
if (rooms['join']) {
for (const roomId of Object.keys(rooms.join)) {
const targetChannel = this.server.matrixRooms.get(roomId) || new Channel(roomId, this);
rooms.join[roomId].state.events.forEach((nextEvent: any) => targetChannel.handleMatrixEvent(nextEvent));
}
}
})
}
}

View file

@ -21,4 +21,25 @@ export class Server {
this.ircUsers = new Map();
this.nickToMxid = new Map();
}
getOrCreateMatrixUser(mxid: string): MatrixUser {
let maybeMatrixUser = this.matrixUsers.get(mxid);
if (maybeMatrixUser) {
return maybeMatrixUser;
}
let potentialNick = mxid.split(":")[0].substr(1);
if (!this.nickToMxid.has(potentialNick)) {
const newMatrixUser = new MatrixUser(mxid, potentialNick);
this.matrixUsers.set(mxid, newMatrixUser);
this.nickToMxid.set(potentialNick, mxid);
return newMatrixUser;
}
const homeserverArray = mxid.split(":")[1].split('.');
const baseDomainNum = homeserverArray.length - 2;
potentialNick = `${potentialNick}-${homeserverArray[baseDomainNum]}`;
const newMatrixUser = new MatrixUser(mxid, potentialNick);
this.matrixUsers.set(mxid, newMatrixUser);
this.nickToMxid.set(potentialNick, mxid);
return newMatrixUser;
}
}