tiramisu/seyfert
5
0
Fork 0
mirror of https://github.com/tiramisulabs/seyfert.git synced 2025-07-03 05:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: button builder types

Browse Source
This commit is contained in:
Marcos Susaña 2024-03-12 00:15:05 -04:00
parent 10d66a221e
commit b8d511e3bf
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/builders/Button.ts
View File

@ -1,7 +1,6 @@
import { throwError } from '..';
import {
ComponentType,
type APIButtonComponent,
type APIButtonComponentWithCustomId,
type APIButtonComponentWithURL,
type APIMessageComponentEmoji,
@ -13,6 +12,14 @@ import { resolvePartialEmoji } from '../structures/extra/functions';
export type ButtonStylesForID = Exclude<ButtonStyle, ButtonStyle.Link>;
export type ButtonSelect<T extends ButtonStyle> = T extends ButtonStyle.Link
? Omit<Button<false>, 'setCustomId' | 'setURL' | 'setStyle'> & {
setStyle(style: ButtonStyle.Link): Omit<Button<false>, 'setCustomId' | 'setURL' | 'setStyle'>;
}
: Omit<Button<true>, 'setCustomId' | 'setURL' | 'setStyle'> & {
setStyle(style: ButtonStylesForID): Omit<Button<true>, 'setCustomId' | 'setURL' | 'setStyle'>;
};
/**
* Represents a button component.
* @template Type - The type of the button component.
@ -31,10 +38,10 @@ export class Button<Type extends boolean = boolean> {
* @param id - The custom ID to set.
* @returns The modified Button instance.
*/
setCustomId(id: string) {
setCustomId(id: string): ButtonSelect<ButtonStylesForID> {
// @ts-expect-error
this.data.custom_id = id;
return this;
return this as ButtonSelect<ButtonStylesForID>;
}
/**
@ -42,10 +49,10 @@ export class Button<Type extends boolean = boolean> {
* @param url - The URL to set.
* @returns The modified Button instance.
*/
setURL(url: string) {
setURL(url: string): ButtonSelect<ButtonStyle.Link> {
// @ts-expect-error
this.data.url = url;
return this;
return this as ButtonSelect<ButtonStyle.Link>;
}
/**
@ -80,6 +87,8 @@ export class Button<Type extends boolean = boolean> {
return this;
}
setStyle(style: ButtonStyle.Link): Omit<Button, 'setCustomId'>;
setStyle(style: ButtonStylesForID): Omit<Button, 'setURL'>;
setStyle(style: ButtonStyle) {
this.data.style = style;
return this;
@ -90,6 +99,6 @@ export class Button<Type extends boolean = boolean> {
* @returns The JSON representation of the Button instance.
*/
toJSON() {
return { ...this.data } as unknown as APIButtonComponent;
return { ...this.data } as When<Type, APIButtonComponentWithCustomId, APIButtonComponentWithURL>;
}
}