mirror of
https://github.com/Vendicated/Vencord.git
synced 2025-01-10 18:06:22 +00:00
115 lines
3.8 KiB
TypeScript
115 lines
3.8 KiB
TypeScript
/*
|
|
* Vencord, a Discord client mod
|
|
* Copyright (c) 2024 Vendicated and contributors
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
|
*/
|
|
|
|
import { DataStore } from "@api/index";
|
|
import { definePluginSettings } from "@api/Settings";
|
|
import { Devs } from "@utils/constants";
|
|
import definePlugin, { OptionType } from "@utils/types";
|
|
import { findByCodeLazy, findByPropsLazy } from "@webpack";
|
|
import { ChannelStore, GuildStore } from "@webpack/common";
|
|
|
|
const SummaryStore = findByPropsLazy("allSummaries", "findSummary");
|
|
const createSummaryFromServer = findByCodeLazy(".people)),startId:", ".type}");
|
|
|
|
const settings = definePluginSettings({
|
|
summaryExpiryThresholdDays: {
|
|
type: OptionType.SLIDER,
|
|
description: "The time in days before a summary is removed. Note that only up to 50 summaries are kept per channel",
|
|
markers: [1, 3, 5, 7, 10, 15, 20, 25, 30],
|
|
stickToMarkers: false,
|
|
default: 3,
|
|
}
|
|
});
|
|
|
|
interface Summary {
|
|
count: number;
|
|
end_id: string;
|
|
id: string;
|
|
message_ids: string[];
|
|
people: string[];
|
|
source: number;
|
|
start_id: string;
|
|
summ_short: string;
|
|
topic: string;
|
|
type: number;
|
|
unsafe: boolean;
|
|
}
|
|
|
|
interface ChannelSummaries {
|
|
type: string;
|
|
channel_id: string;
|
|
guild_id: string;
|
|
summaries: Summary[];
|
|
|
|
// custom property
|
|
time?: number;
|
|
}
|
|
|
|
export default definePlugin({
|
|
name: "Summaries",
|
|
description: "Enables Discord's experimental Summaries feature on every server, displaying AI generated summaries of conversations",
|
|
authors: [Devs.mantikafasi],
|
|
settings,
|
|
patches: [
|
|
{
|
|
find: "SUMMARIZEABLE.has",
|
|
replacement: {
|
|
match: /\i\.hasFeature\(\i\.\i\.SUMMARIES_ENABLED\w+?\)/g,
|
|
replace: "true"
|
|
}
|
|
},
|
|
{
|
|
find: "RECEIVE_CHANNEL_SUMMARY(",
|
|
replacement: {
|
|
match: /shouldFetch\((\i),\i\){/,
|
|
replace: "$& if(!$self.shouldFetch($1)) return false;"
|
|
}
|
|
}
|
|
],
|
|
flux: {
|
|
CONVERSATION_SUMMARY_UPDATE(data) {
|
|
const incomingSummaries: ChannelSummaries[] = data.summaries.map((summary: any) => ({ ...createSummaryFromServer(summary), time: Date.now() }));
|
|
|
|
// idk if this is good for performance but it doesnt seem to be a problem in my experience
|
|
DataStore.update("summaries-data", summaries => {
|
|
summaries ??= {};
|
|
summaries[data.channel_id] ? summaries[data.channel_id].unshift(...incomingSummaries) : (summaries[data.channel_id] = incomingSummaries);
|
|
if (summaries[data.channel_id].length > 50)
|
|
summaries[data.channel_id] = summaries[data.channel_id].slice(0, 50);
|
|
|
|
return summaries;
|
|
});
|
|
}
|
|
},
|
|
|
|
async start() {
|
|
await DataStore.update("summaries-data", summaries => {
|
|
summaries ??= {};
|
|
for (const key of Object.keys(summaries)) {
|
|
for (let i = summaries[key].length - 1; i >= 0; i--) {
|
|
if (summaries[key][i].time < Date.now() - 1000 * 60 * 60 * 24 * settings.store.summaryExpiryThresholdDays) {
|
|
summaries[key].splice(i, 1);
|
|
}
|
|
}
|
|
|
|
if (summaries[key].length === 0) {
|
|
delete summaries[key];
|
|
}
|
|
}
|
|
|
|
Object.assign(SummaryStore.allSummaries(), summaries);
|
|
return summaries;
|
|
});
|
|
},
|
|
|
|
shouldFetch(channelId: string) {
|
|
const channel = ChannelStore.getChannel(channelId);
|
|
// SUMMARIES_ENABLED feature is not in discord-types
|
|
const guild = GuildStore.getGuild(channel.guild_id);
|
|
// @ts-ignore
|
|
return guild.hasFeature("SUMMARIES_ENABLED_GA");
|
|
}
|
|
});
|