99 lines
3.4 KiB
TypeScript
99 lines
3.4 KiB
TypeScript
/*
|
|
* Vencord, a modification for Discord's desktop app
|
|
* Copyright (c) 2022 Vendicated and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
import ErrorBoundary from "@components/ErrorBoundary";
|
|
import { Logger } from "@utils/Logger";
|
|
import { IconComponent } from "@utils/types";
|
|
import { Channel, Message } from "@vencord/discord-types";
|
|
import type { ComponentType, MouseEventHandler } from "react";
|
|
|
|
import { useSettings } from "./Settings";
|
|
|
|
const logger = new Logger("MessagePopover");
|
|
|
|
export interface MessagePopoverButtonItem {
|
|
key?: string,
|
|
label: string,
|
|
icon: ComponentType<any>,
|
|
message: Message,
|
|
channel: Channel,
|
|
onClick?: MouseEventHandler<HTMLButtonElement>,
|
|
onContextMenu?: MouseEventHandler<HTMLButtonElement>;
|
|
}
|
|
|
|
export type MessagePopoverButtonFactory = (message: Message) => MessagePopoverButtonItem | null;
|
|
export type MessagePopoverButtonData = {
|
|
render: MessagePopoverButtonFactory;
|
|
/**
|
|
* This icon is used only for Settings UI. Your render function must still return an icon,
|
|
* and it can be different from this one.
|
|
*/
|
|
icon: IconComponent;
|
|
};
|
|
|
|
export const MessagePopoverButtonMap = new Map<string, MessagePopoverButtonData>();
|
|
|
|
/**
|
|
* The icon argument is used only for Settings UI. Your render function must still return an icon,
|
|
* and it can be different from this one.
|
|
*/
|
|
export function addMessagePopoverButton(
|
|
identifier: string,
|
|
render: MessagePopoverButtonFactory,
|
|
icon: IconComponent
|
|
) {
|
|
MessagePopoverButtonMap.set(identifier, { render, icon });
|
|
}
|
|
|
|
export function removeMessagePopoverButton(identifier: string) {
|
|
MessagePopoverButtonMap.delete(identifier);
|
|
}
|
|
|
|
function VencordPopoverButtons(props: { Component: React.ComponentType<MessagePopoverButtonItem>, message: Message; }) {
|
|
const { Component, message } = props;
|
|
|
|
const { messagePopoverButtons } = useSettings(["uiElements.messagePopoverButtons.*"]).uiElements;
|
|
|
|
const elements = Array.from(MessagePopoverButtonMap.entries())
|
|
.filter(([key]) => messagePopoverButtons[key]?.enabled !== false)
|
|
.map(([key, { render }]) => {
|
|
try {
|
|
// FIXME: this should use proper React to ensure hooks work
|
|
const item = render(message);
|
|
if (!item) return null;
|
|
|
|
return (
|
|
<ErrorBoundary noop>
|
|
<Component key={key} {...item} />
|
|
</ErrorBoundary>
|
|
);
|
|
} catch (err) {
|
|
logger.error(`[${key}]`, err);
|
|
return null;
|
|
}
|
|
});
|
|
|
|
return <>{elements}</>;
|
|
}
|
|
|
|
export function _buildPopoverElements(
|
|
Component: React.ComponentType<MessagePopoverButtonItem>,
|
|
message: Message
|
|
) {
|
|
return <VencordPopoverButtons Component={Component} message={message} />;
|
|
}
|