200 lines
9.5 KiB
JavaScript
200 lines
9.5 KiB
JavaScript
import {Config} from "data";
|
|
import WebpackModules from "./webpackmodules";
|
|
import DiscordModules from "./discordmodules";
|
|
import DiscordClasses from "./discordclasses";
|
|
import Utilities from "./utilities";
|
|
import Patcher from "./patcher";
|
|
import BDLogo from "../ui/icons/bdlogo";
|
|
import Logger from "common/logger";
|
|
|
|
const React = DiscordModules.React;
|
|
const Tooltip = WebpackModules.getByDisplayName("Tooltip");
|
|
const Anchor = WebpackModules.getByDisplayName("Anchor");
|
|
|
|
const Developers = [
|
|
/* Zerebos#7790 */
|
|
"249746236008169473",
|
|
|
|
/* Strencher#1044 */
|
|
"415849376598982656"
|
|
];
|
|
|
|
const DeveloperBadge = function DeveloperBadge({type, size = 16}) {
|
|
return React.createElement(Tooltip, {color: "primary", position: "top", text: "BetterDiscord Developer"},
|
|
props => React.createElement(Anchor, Object.assign({className: `bd-${type}-badge`, href: "https://github.com/BetterDiscord/BetterDiscord", title: "BetterDiscord", target: "_blank"}, props),
|
|
React.createElement(BDLogo, {size, className: "bd-logo"})
|
|
)
|
|
);
|
|
};
|
|
|
|
export default new class ComponentPatcher {
|
|
warn(...message) {return Logger.warn("ComponentPatcher", ...message);}
|
|
error(...message) {return Logger.error("ComponentPatcher", ...message);}
|
|
debug(...message) {return Logger.debug("ComponentPatcher", ...message);}
|
|
|
|
initialize() {
|
|
Utilities.suppressErrors(this.patchSocial.bind(this), "BD Social Patch")();
|
|
Utilities.suppressErrors(this.patchGuildPills.bind(this), "BD Guild Pills Patch")();
|
|
Utilities.suppressErrors(this.patchGuildListItems.bind(this), "BD Guild List Items Patch")();
|
|
Utilities.suppressErrors(this.patchMessageHeader.bind(this), "BD Message Header Patch")();
|
|
Utilities.suppressErrors(this.patchMemberList.bind(this), "BD Member List Patch")();
|
|
Utilities.suppressErrors(this.patchProfile.bind(this), "BD Profile Badges Patch")();
|
|
}
|
|
|
|
patchSocial() {
|
|
if (this.socialPatch) return;
|
|
const TabBar = WebpackModules.getByDisplayName("TabBar");
|
|
if (!TabBar) return;
|
|
this.socialPatch = Patcher.after("ComponentPatcher", TabBar.prototype, "render", (thisObject, args, returnValue) => {
|
|
const children = returnValue.props.children;
|
|
if (!children || !children.length || children.length < 3) return;
|
|
if (children[children.length - 3].type.displayName !== "Separator") return;
|
|
if (!children[children.length - 2].type.toString().includes("socialLinks")) return;
|
|
if (Anchor) {
|
|
const original = children[children.length - 2].type;
|
|
const newOne = function() {
|
|
const returnVal = original(...arguments);
|
|
returnVal.props.children.push(
|
|
DiscordModules.React.createElement(Anchor, {className: "bd-social-link", href: "https://twitter.com/_BetterDiscord_", title: "BetterDiscord", target: "_blank"},
|
|
DiscordModules.React.createElement(BDLogo, {size: "16px", className: "bd-social-logo"})
|
|
)
|
|
);
|
|
return returnVal;
|
|
};
|
|
children[children.length - 2].type = newOne;
|
|
}
|
|
|
|
const additional = DiscordModules.React.createElement("div", {className: `${DiscordClasses.Text.colorMuted} ${DiscordClasses.Text.size12}`}, `BetterDiscord ${Config.version}`);
|
|
|
|
|
|
const originalVersions = children[children.length - 1].type;
|
|
children[children.length - 1].type = function() {
|
|
const returnVal = originalVersions(...arguments);
|
|
returnVal.props.children.splice(1, 0, additional);
|
|
return returnVal;
|
|
};
|
|
});
|
|
}
|
|
|
|
patchGuildListItems() {
|
|
if (this.guildListItemsPatch) return;
|
|
const ListNavigators = WebpackModules.getByProps("ListNavigatorProvider");
|
|
const GuildComponent = WebpackModules.find(m => m.type && m.type.toString().includes("guildNode") && m.type.toString().includes("treeitem"));
|
|
if (!GuildComponent || typeof(GuildComponent.type) !== "function") return this.warn("Failed to get Guild component.");
|
|
if (!ListNavigators || typeof(ListNavigators.ListNavigatorProvider) !== "function") return this.warn("Failed to get ListNavigatorProvider component.");
|
|
|
|
this.guildListItemsPatch = Patcher.after("ComponentPatcher", GuildComponent, "type", (_, [props], returnValue) => {
|
|
if (!returnValue || !returnValue.props) return;
|
|
|
|
try {
|
|
returnValue.props.className += " bd-guild";
|
|
if (props.unread) returnValue.props.className += " bd-unread";
|
|
if (props.selected) returnValue.props.className += " bd-selected";
|
|
if (props.mediaState.audio) returnValue.props.className += " bd-audio";
|
|
if (props.mediaState.video) returnValue.props.className += " bd-video";
|
|
if (props.badge) returnValue.props.className += " bd-badge";
|
|
if (props.animatable) returnValue.props.className += " bd-animatable";
|
|
if (props.unavailable) returnValue.props.className += " bd-unavailable";
|
|
if (props.mediaState.screenshare) returnValue.props.className += " bd-screenshare";
|
|
if (props.mediaState.liveStage) returnValue.props.className += " bd-live-stage";
|
|
if (props.muted) returnValue.props.className += " bd-muted";
|
|
}
|
|
catch (err) {
|
|
Logger.error("ComponentPatcher:Guilds", `Error inside BDGuild:`, err);
|
|
this.guildListItemsPatch();
|
|
}
|
|
});
|
|
|
|
const {useState} = DiscordModules.React;
|
|
function useForceUpdate() {
|
|
const [, setValue] = useState(false);
|
|
return () => setValue(v => !v); // update the state to force render
|
|
}
|
|
|
|
let hasForced = false;
|
|
this.cancelForceUpdate = Patcher.after("ComponentPatcher", ListNavigators, "ListNavigatorProvider", (_, __, returnValue) => {
|
|
if (returnValue.props.value.id !== "guildsnav") return;
|
|
|
|
const originalParent = Utilities.findInTree(returnValue, m => m?.props?.className, {walkable: ["children", "props"]});
|
|
if (!originalParent) return;
|
|
const original = originalParent.type;
|
|
originalParent.type = e => {
|
|
const forceUpdate = useForceUpdate();
|
|
if (!hasForced) {
|
|
hasForced = true;
|
|
setTimeout(() => {
|
|
forceUpdate();
|
|
this.cancelForceUpdate();
|
|
}, 1);
|
|
}
|
|
|
|
return Reflect.apply(original, null, [e]);
|
|
};
|
|
});
|
|
}
|
|
|
|
patchGuildPills() {
|
|
if (this.guildPillPatch) return;
|
|
const guildPill = WebpackModules.find(m => m.default.displayName === "AnimatedHalfPill");
|
|
if (!guildPill) return;
|
|
this.guildPillPatch = Patcher.after("ComponentPatcher", guildPill, "default", (_, args, returnValue) => {
|
|
const props = args[0];
|
|
if (props.unread) returnValue.props.className += " bd-unread";
|
|
if (props.selected) returnValue.props.className += " bd-selected";
|
|
if (props.hovered) returnValue.props.className += " bd-hovered";
|
|
return returnValue;
|
|
});
|
|
}
|
|
|
|
patchMessageHeader() {
|
|
if (this.messageHeaderPatch) return;
|
|
const MessageTimestamp = WebpackModules.getModule(m => m?.default?.toString().indexOf("showTimestampOnHover") > -1);
|
|
this.messageHeaderPatch = Patcher.after("ComponentPatcher", MessageTimestamp, "default", (_, [{message}], returnValue) => {
|
|
const userId = Utilities.getNestedProp(message, "author.id");
|
|
if (Developers.indexOf(userId) < 0) return;
|
|
const children = Utilities.getNestedProp(returnValue, "props.children.1.props.children");
|
|
if (!Array.isArray(children)) return;
|
|
|
|
children.splice(2, 0,
|
|
React.createElement(DeveloperBadge, {
|
|
type: "chat"
|
|
})
|
|
);
|
|
});
|
|
}
|
|
|
|
patchMemberList() {
|
|
if (this.memberListPatch) return;
|
|
const MemberListItem = WebpackModules.findByDisplayName("MemberListItem");
|
|
if (!MemberListItem?.prototype?.renderDecorators) return;
|
|
this.memberListPatch = Patcher.after("ComponentPatcher", MemberListItem.prototype, "renderDecorators", (thisObject, args, returnValue) => {
|
|
const user = Utilities.getNestedProp(thisObject, "props.user");
|
|
const children = Utilities.getNestedProp(returnValue, "props.children");
|
|
if (!children || Developers.indexOf(user.id) < 0) return;
|
|
if (!Array.isArray(children)) return;
|
|
children.push(
|
|
React.createElement(DeveloperBadge, {
|
|
type: "member"
|
|
})
|
|
);
|
|
});
|
|
}
|
|
|
|
patchProfile() {
|
|
if (this.profilePatch) return;
|
|
const UserProfileBadgeList = WebpackModules.getModule(m => m?.default?.displayName === "UserProfileBadgeList");
|
|
this.profilePatch = Patcher.after("ComponentPatcher", UserProfileBadgeList, "default", (_, [{user}], res) => {
|
|
if (Developers.indexOf(user?.id) < 0) return;
|
|
const children = Utilities.getNestedProp(res, "props.children");
|
|
if (!Array.isArray(children)) return;
|
|
|
|
children.unshift(
|
|
React.createElement(DeveloperBadge, {
|
|
type: "profile",
|
|
size: 18
|
|
})
|
|
);
|
|
});
|
|
}
|
|
|
|
}; |