[WIP] threads. #74
1 changed files with 0 additions and 269 deletions
sh!t, that wasn't supposed to make it there...
commit
0425c18cf3
|
|
@ -1,269 +0,0 @@
|
||||||
// @ts-check
|
|
||||||
|
|
||||||
const DiscordTypes = require("discord-api-types/v10")
|
|
||||||
const assert = require("assert/strict")
|
|
||||||
const {z} = require("zod")
|
|
||||||
const {H3Event, defineEventHandler, sendRedirect, createError, getValidatedQuery, readValidatedBody, setResponseHeader} = require("h3")
|
|
||||||
const {randomUUID} = require("crypto")
|
|
||||||
const {LRUCache} = require("lru-cache")
|
|
||||||
const Ty = require("../../types")
|
|
||||||
const uqr = require("uqr")
|
|
||||||
|
|
||||||
const {id: botID} = require("../../../addbot")
|
|
||||||
const {discord, as, sync, select, from, db} = require("../../passthrough")
|
|
||||||
/** @type {import("../pug-sync")} */
|
|
||||||
const pugSync = sync.require("../pug-sync")
|
|
||||||
/** @type {import("../../d2m/actions/create-space")} */
|
|
||||||
const createSpace = sync.require("../../d2m/actions/create-space")
|
|
||||||
/** @type {import("../auth")} */
|
|
||||||
const auth = require("../auth")
|
|
||||||
/** @type {import("../../discord/utils")} */
|
|
||||||
const dUtils = sync.require("../../discord/utils")
|
|
||||||
/** @type {import("../../matrix/utils")} */
|
|
||||||
const mxUtils = sync.require("../../matrix/utils")
|
|
||||||
const {reg} = require("../../matrix/read-registration")
|
|
||||||
|
|
||||||
const schema = {
|
|
||||||
guild: z.object({
|
|
||||||
guild_id: z.string().optional()
|
|
||||||
}),
|
|
||||||
qr: z.object({
|
|
||||||
guild_id: z.string().optional()
|
|
||||||
}),
|
|
||||||
invite: z.object({
|
|
||||||
mxid: z.string().regex(/@([^:]+):([a-z0-9:-]+\.[a-z0-9.:-]+)/),
|
|
||||||
permissions: z.enum(["default", "moderator", "admin"]),
|
|
||||||
guild_id: z.string().optional(),
|
|
||||||
nonce: z.string().optional()
|
|
||||||
}),
|
|
||||||
inviteNonce: z.object({
|
|
||||||
nonce: z.string()
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param {H3Event} event
|
|
||||||
* @returns {import("../../matrix/api")}
|
|
||||||
*/
|
|
||||||
function getAPI(event) {
|
|
||||||
/* c8 ignore next */
|
|
||||||
return event.context.api || sync.require("../../matrix/api")
|
|
||||||
}
|
|
||||||
|
|
||||||
/** @type {LRUCache<string, string>} nonce to guild id */
|
|
||||||
const validNonce = new LRUCache({max: 200})
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param {{type: number, parent_id?: string | null, position?: number}} channel
|
|
||||||
* @param {Map<string, {type: number, parent_id?: string | null, position?: number}>} channels
|
|
||||||
*/
|
|
||||||
function getPosition(channel, channels) {
|
|
||||||
let position = 0
|
|
||||||
|
|
||||||
// Categories always appear below un-categorised channels. Their contents can be ordered.
|
|
||||||
// So categories, and things in them, will have their position multiplied by a big number. The category's big number. The regular position small number sorts within the category.
|
|
||||||
// Categories are size 2000.
|
|
||||||
let foundCategory = channel
|
|
||||||
while (foundCategory.parent_id) {
|
|
||||||
const f = channels.get(foundCategory.parent_id)
|
|
||||||
assert(f)
|
|
||||||
foundCategory = f
|
|
||||||
}
|
|
||||||
if (foundCategory.type === DiscordTypes.ChannelType.GuildCategory) position = ((foundCategory.position || 0) + 1) * 2000
|
|
||||||
|
|
||||||
// Categories always appear above what they contain.
|
|
||||||
if (channel.type === DiscordTypes.ChannelType.GuildCategory) position -= 0.5
|
|
||||||
|
|
||||||
// Within a category, voice channels are always sorted to the bottom. The text/voice split is size 1000 each.
|
|
||||||
if ([DiscordTypes.ChannelType.GuildVoice, DiscordTypes.ChannelType.GuildStageVoice].includes(channel.type)) position += 1000
|
|
||||||
|
|
||||||
// Channels are manually ordered within the text/voice split.
|
|
||||||
if (typeof channel.position === "number") position += channel.position
|
|
||||||
|
|
||||||
// Threads appear below their channel.
|
|
||||||
if ([DiscordTypes.ChannelType.PublicThread, DiscordTypes.ChannelType.PrivateThread, DiscordTypes.ChannelType.AnnouncementThread].includes(channel.type)) {
|
|
||||||
position += 0.5
|
|
||||||
let parent = channels.get(channel.parent_id || "")
|
|
||||||
if (parent && parent["position"]) position += parent["position"]
|
|
||||||
}
|
|
||||||
|
|
||||||
return position
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param {DiscordTypes.APIGuild} guild
|
|
||||||
* @param {Ty.R.Hierarchy[]} rooms
|
|
||||||
* @param {string[]} roles
|
|
||||||
*/
|
|
||||||
function getChannelRoomsLinks(guild, rooms, roles) {
|
|
||||||
let channelIDs = discord.guildChannelMap.get(guild.id)
|
|
||||||
assert(channelIDs)
|
|
||||||
|
|
||||||
let linkedChannels = select("channel_room", ["channel_id", "room_id", "name", "nick", "room_mention_role"], {channel_id: channelIDs}).all()
|
|
||||||
let linkedChannelsWithDetails = linkedChannels.map(c => ({
|
|
||||||
// @ts-ignore
|
|
||||||
/** @type {DiscordTypes.APIGuildChannel} */ channel: discord.channels.get(c.channel_id),
|
|
||||||
...c
|
|
||||||
}))
|
|
||||||
let removedUncachedChannels = dUtils.filterTo(linkedChannelsWithDetails, c => c.channel)
|
|
||||||
let linkedChannelIDs = linkedChannelsWithDetails.map(c => c.channel_id)
|
|
||||||
linkedChannelsWithDetails.sort((a, b) => getPosition(a.channel, discord.channels) - getPosition(b.channel, discord.channels))
|
|
||||||
|
|
||||||
let unlinkedChannelIDs = channelIDs.filter(c => !linkedChannelIDs.includes(c))
|
|
||||||
/** @type {DiscordTypes.APIGuildChannel[]} */ // @ts-ignore
|
|
||||||
let unlinkedChannels = unlinkedChannelIDs.map(c => discord.channels.get(c))
|
|
||||||
let removedWrongTypeChannels = dUtils.filterTo(unlinkedChannels, c => c && [0, 5].includes(c.type))
|
|
||||||
let removedPrivateChannels = dUtils.filterTo(unlinkedChannels, c => {
|
|
||||||
const permissions = dUtils.getPermissions(guild.id, roles, guild.roles, botID, c["permission_overwrites"])
|
|
||||||
return dUtils.hasSomePermissions(permissions, ["Administrator", "ViewChannel"])
|
|
||||||
})
|
|
||||||
unlinkedChannels.sort((a, b) => getPosition(a, discord.channels) - getPosition(b, discord.channels))
|
|
||||||
|
|
||||||
let linkedRoomIDs = linkedChannels.map(c => c.room_id)
|
|
||||||
let unlinkedRooms = [...rooms]
|
|
||||||
let removedLinkedRooms = dUtils.filterTo(unlinkedRooms, r => !linkedRoomIDs.includes(r.room_id))
|
|
||||||
let removedWrongTypeRooms = dUtils.filterTo(unlinkedRooms, r => !r.room_type)
|
|
||||||
let removedEncryptedRooms = dUtils.filterTo(unlinkedRooms, r => !r.encryption && !r["im.nheko.summary.encryption"])
|
|
||||||
// https://discord.com/developers/docs/topics/threads#active-archived-threads
|
|
||||||
// need to filter out linked archived threads from unlinkedRooms, will just do that by comparing against the name
|
|
||||||
let removedArchivedThreadRooms = dUtils.filterTo(unlinkedRooms, r => r.name && !r.name.match(/^\[(🔒)?⛓️\]/))
|
|
||||||
|
|
||||||
return {
|
|
||||||
linkedChannelsWithDetails, unlinkedChannels, unlinkedRooms,
|
|
||||||
removedUncachedChannels, removedWrongTypeChannels, removedPrivateChannels, removedLinkedRooms, removedWrongTypeRooms, removedArchivedThreadRooms, removedEncryptedRooms
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param {string} mxid
|
|
||||||
*/
|
|
||||||
function getInviteTargetSpaces(mxid) {
|
|
||||||
/** @type {{room_id: string, mxid: string, type: string, name: string, topic: string?, avatar: string?}[]} */
|
|
||||||
const spaces =
|
|
||||||
// invited spaces
|
|
||||||
db.prepare("SELECT room_id, invite.mxid, type, name, topic, avatar FROM invite LEFT JOIN guild_space ON invite.room_id = guild_space.space_id WHERE mxid = ? AND space_id IS NULL AND type = 'm.space'").all(mxid)
|
|
||||||
// moderated spaces
|
|
||||||
.concat(db.prepare("SELECT room_id, invite.mxid, type, name, topic, avatar FROM invite LEFT JOIN guild_space ON invite.room_id = guild_space.space_id INNER JOIN member_cache USING (room_id) WHERE member_cache.mxid = ? AND power_level >= 50 AND space_id IS NULL AND type = 'm.space'").all(mxid))
|
|
||||||
const seen = new Set(spaces.map(s => s.room_id))
|
|
||||||
return spaces.filter(s => seen.delete(s.room_id))
|
|
||||||
}
|
|
||||||
|
|
||||||
as.router.get("/guild", defineEventHandler(async event => {
|
|
||||||
const {guild_id} = await getValidatedQuery(event, schema.guild.parse)
|
|
||||||
const session = await auth.useSession(event)
|
|
||||||
const managed = await auth.getManagedGuilds(event)
|
|
||||||
const row = from("guild_active").join("guild_space", "guild_id", "left").select("space_id", "privacy_level", "autocreate").where({guild_id}).get()
|
|
||||||
// @ts-ignore
|
|
||||||
const guild = discord.guilds.get(guild_id)
|
|
||||||
|
|
||||||
// Permission problems
|
|
||||||
if (!guild_id || !guild || !managed.has(guild_id) || !row) {
|
|
||||||
return pugSync.render(event, "guild-access-denied.pug", {guild_id, row})
|
|
||||||
}
|
|
||||||
|
|
||||||
// Self-service guild that hasn't been linked yet - needs a special page encouraging the link flow
|
|
||||||
if (!row.space_id && row.autocreate === 0) {
|
|
||||||
const spaces = session.data.mxid ? getInviteTargetSpaces(session.data.mxid) : []
|
|
||||||
return pugSync.render(event, "guild-not-linked.pug", {guild, guild_id, spaces})
|
|
||||||
}
|
|
||||||
|
|
||||||
const roles = guild.members?.find(m => m.user.id === botID)?.roles || []
|
|
||||||
|
|
||||||
// Easy mode guild that hasn't been linked yet - need to remove elements that would require an existing space
|
|
||||||
if (!row.space_id) {
|
|
||||||
const links = getChannelRoomsLinks(guild, [], roles)
|
|
||||||
return pugSync.render(event, "guild.pug", {guild, guild_id, ...links, ...row})
|
|
||||||
}
|
|
||||||
|
|
||||||
// Linked guild
|
|
||||||
const api = getAPI(event)
|
|
||||||
const rooms = await api.getFullHierarchy(row.space_id)
|
|
||||||
const links = getChannelRoomsLinks(guild, rooms, roles)
|
|
||||||
return pugSync.render(event, "guild.pug", {guild, guild_id, ...links, ...row})
|
|
||||||
}))
|
|
||||||
|
|
||||||
as.router.get("/qr", defineEventHandler(async event => {
|
|
||||||
const {guild_id} = await getValidatedQuery(event, schema.qr.parse)
|
|
||||||
const managed = await auth.getManagedGuilds(event)
|
|
||||||
const row = from("guild_active").join("guild_space", "guild_id", "left").select("space_id", "privacy_level", "autocreate").where({guild_id}).get()
|
|
||||||
// @ts-ignore
|
|
||||||
const guild = discord.guilds.get(guild_id)
|
|
||||||
|
|
||||||
// Permission problems
|
|
||||||
if (!guild_id || !guild || !managed.has(guild_id) || !row) {
|
|
||||||
return pugSync.render(event, "guild-access-denied.pug", {guild_id, row})
|
|
||||||
}
|
|
||||||
|
|
||||||
const nonce = randomUUID()
|
|
||||||
validNonce.set(nonce, guild_id)
|
|
||||||
|
|
||||||
const data = `${reg.ooye.bridge_origin}/invite?nonce=${nonce}`
|
|
||||||
// necessary to scale the svg pixel-perfectly on the page
|
|
||||||
// https://github.com/unjs/uqr/blob/244952a8ba2d417f938071b61e11fb1ff95d6e75/src/svg.ts#L24
|
|
||||||
const generatedSvg = uqr.renderSVG(data, {pixelSize: 3})
|
|
||||||
const svg = generatedSvg.replace(/viewBox="0 0 ([0-9]+) ([0-9]+)"/, `data-nonce="${nonce}" width="$1" height="$2" $&`)
|
|
||||||
assert.notEqual(svg, generatedSvg)
|
|
||||||
|
|
||||||
return svg
|
|
||||||
}))
|
|
||||||
|
|
||||||
as.router.get("/invite", defineEventHandler(async event => {
|
|
||||||
const {nonce} = await getValidatedQuery(event, schema.inviteNonce.parse)
|
|
||||||
const isValid = validNonce.has(nonce)
|
|
||||||
const guild_id = validNonce.get(nonce)
|
|
||||||
const guild = discord.guilds.get(guild_id || "")
|
|
||||||
return pugSync.render(event, "invite.pug", {isValid, nonce, guild_id, guild})
|
|
||||||
}))
|
|
||||||
|
|
||||||
as.router.post("/api/invite", defineEventHandler(async event => {
|
|
||||||
const parsedBody = await readValidatedBody(event, schema.invite.parse)
|
|
||||||
const managed = await auth.getManagedGuilds(event)
|
|
||||||
const api = getAPI(event)
|
|
||||||
|
|
||||||
// Check guild ID or nonce
|
|
||||||
if (parsedBody.guild_id) {
|
|
||||||
var guild_id = parsedBody.guild_id
|
|
||||||
if (!managed.has(guild_id)) throw createError({status: 403, message: "Forbidden", data: "Can't invite users to a guild you don't have Manage Server permissions in"})
|
|
||||||
} else if (parsedBody.nonce) {
|
|
||||||
if (!validNonce.has(parsedBody.nonce)) throw createError({status: 403, message: "Nonce expired", data: "Nonce means number-used-once, and, well, you tried to use it twice..."})
|
|
||||||
let ok = validNonce.get(parsedBody.nonce)
|
|
||||||
assert(ok)
|
|
||||||
var guild_id = ok
|
|
||||||
validNonce.delete(parsedBody.nonce)
|
|
||||||
} else {
|
|
||||||
throw createError({status: 400, message: "Missing guild ID", data: "Passing a guild ID or a nonce is required."})
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check guild is bridged
|
|
||||||
const guild = discord.guilds.get(guild_id)
|
|
||||||
assert(guild)
|
|
||||||
const spaceID = await createSpace.ensureSpace(guild)
|
|
||||||
|
|
||||||
// Check for existing invite to the space
|
|
||||||
let spaceMember
|
|
||||||
try {
|
|
||||||
spaceMember = await api.getStateEvent(spaceID, "m.room.member", parsedBody.mxid)
|
|
||||||
} catch (e) {}
|
|
||||||
|
|
||||||
if (!spaceMember || !["invite", "join"].includes(spaceMember.membership)) {
|
|
||||||
// Invite
|
|
||||||
await api.inviteToRoom(spaceID, parsedBody.mxid)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Permissions
|
|
||||||
const powerLevel =
|
|
||||||
( parsedBody.permissions === "admin" ? 100
|
|
||||||
: parsedBody.permissions === "moderator" ? 50
|
|
||||||
: 0)
|
|
||||||
if (powerLevel) await mxUtils.setUserPowerCascade(spaceID, parsedBody.mxid, powerLevel, api)
|
|
||||||
|
|
||||||
if (parsedBody.guild_id) {
|
|
||||||
setResponseHeader(event, "HX-Refresh", true)
|
|
||||||
return sendRedirect(event, `/guild?guild_id=${guild_id}`, 302)
|
|
||||||
} else {
|
|
||||||
return sendRedirect(event, "/ok?msg=User has been invited.", 302)
|
|
||||||
}
|
|
||||||
}))
|
|
||||||
|
|
||||||
module.exports._getPosition = getPosition
|
|
||||||
module.exports.getInviteTargetSpaces = getInviteTargetSpaces
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue