diff --git a/src/plugins/dearrow/index.tsx b/src/plugins/dearrow/index.tsx index 7e50bcae..f896531a 100644 --- a/src/plugins/dearrow/index.tsx +++ b/src/plugins/dearrow/index.tsx @@ -74,8 +74,8 @@ async function embedDidMount(this: Component) { } } -function renderButton(this: Component) { - const { embed } = this.props; +function renderButton(component: Component) { + const { embed } = component.props; if (!embed?.dearrow) return null; return ( @@ -97,7 +97,7 @@ function renderButton(this: Component) { embed.thumbnail.proxyURL = oldThumb; } - this.forceUpdate(); + component.forceUpdate(); }} > {/* Dearrow Icon, taken from https://dearrow.ajay.app/logo.svg (and optimised) */} @@ -134,7 +134,13 @@ export default definePlugin({ authors: [Devs.Ven], embedDidMount, - renderButton: ErrorBoundary.wrap(renderButton, { noop: true }), + renderButton(component: Component) { + return ( + + {renderButton(component)} + + ); + }, patches: [{ find: "this.renderInlineMediaEmbed", @@ -148,7 +154,7 @@ export default definePlugin({ // add dearrow button { match: /children:\[(?=null!=\i\?\i\.renderSuppressButton)/, - replace: "children:[$self.renderButton.call(this)," + replace: "children:[$self.renderButton(this)," } ] }],