members.tsā¢2.52 kB
import { z } from 'zod';
import type { ToolHandler } from '../tool-types.js';
import { Routes } from 'discord-api-types/v10';
import { REST } from '@discordjs/rest';
import { DiscordClient } from '../discord.js';
export function getMemberTool(dc: DiscordClient): ToolHandler {
const input = z.object({ guild_id: z.string(), user_id: z.string() });
return {
name: 'discord_get_member',
description: 'Get a guild member.',
inputSchema: input,
async *handler({ input }: { input: any }){
const { guild_id, user_id } = input as any;
const rest = (dc as any)['rest'] as REST;
const res = await rest.get(Routes.guildMember(guild_id, user_id));
yield { content: [{ type: 'text', text: JSON.stringify(res) }] };
}
};
}
export function addMemberRoleTool(dc: DiscordClient): ToolHandler {
const input = z.object({ guild_id: z.string(), user_id: z.string(), role_id: z.string() });
return {
name: 'discord_add_member_role',
description: 'Add a role to a member.',
inputSchema: input,
async *handler({ input }: { input: any }){
const { guild_id, user_id, role_id } = input as any;
const rest = (dc as any)['rest'] as REST;
await rest.put(Routes.guildMemberRole(guild_id, user_id, role_id));
yield { content: [{ type: 'text', text: 'ok' }] };
}
};
}
export function removeMemberRoleTool(dc: DiscordClient): ToolHandler {
const input = z.object({ guild_id: z.string(), user_id: z.string(), role_id: z.string() });
return {
name: 'discord_remove_member_role',
description: 'Remove a role from a member.',
inputSchema: input,
async *handler({ input }: { input: any }){
const { guild_id, user_id, role_id } = input as any;
const rest = (dc as any)['rest'] as REST;
await rest.delete(Routes.guildMemberRole(guild_id, user_id, role_id));
yield { content: [{ type: 'text', text: 'ok' }] };
}
};
}
export function kickMemberTool(dc: DiscordClient): ToolHandler {
const input = z.object({ guild_id: z.string(), user_id: z.string(), reason: z.string().optional() });
return {
name: 'discord_kick_member',
description: 'Kick (remove) a member from a guild.',
inputSchema: input,
async *handler({ input }: { input: any }){
const { guild_id, user_id, reason } = input as any;
const rest = (dc as any)['rest'] as REST;
await rest.delete(Routes.guildMember(guild_id, user_id), { reason });
yield { content: [{ type: 'text', text: 'ok' }] };
}
};
}