CRBT-Team/Purplet

import type { LocalizationMap } from 'purplet/types';
import { ApplicationCommandType } from 'purplet/types';
import { $appCommand } from './command';
import type {
  Message,
  MessageCommandInteraction,
  PartialUser,
  UserCommandInteraction,
} from '../structures';
import type { CommandPermissionsInput} from '../utils/permissions';
import { resolveCommandPermissions } from '../utils/permissions';

export interface ContextCommandOptions extends CommandPermissionsInput {
  name: string;
  nameLocalizations?: LocalizationMap;
}

export interface UserCommandOptions extends ContextCommandOptions {
  handle(this: UserCommandInteraction, target: PartialUser): void;
}

export function $userContextCommand(opts: UserCommandOptions) {
  return $appCommand({
    command: {
      type: ApplicationCommandType.User,
      name: opts.name,
      name_localizations: opts.nameLocalizations,
      ...resolveCommandPermissions(opts),
    },
    handle(this: UserCommandInteraction) {
      opts.handle.call(this, this.targetUser);
    },
  });
}

export interface MessageCommandOptions extends ContextCommandOptions {
  handle(this: MessageCommandInteraction, target: Message): void;
}

export function $messageContextCommand(opts: MessageCommandOptions) {
  return $appCommand({
    command: {
      type: ApplicationCommandType.Message,
      name: opts.name,
      name_localizations: opts.nameLocalizations,
      ...resolveCommandPermissions(opts),
    },
    handle(this: MessageCommandInteraction) {
      opts.handle.call(this, this.target);
    },
  });
}