new commands update for the bot
This commit is contained in:
93
src/Bot.js
Normal file
93
src/Bot.js
Normal file
@@ -0,0 +1,93 @@
|
||||
// src/bot.js
|
||||
const { Client, GatewayIntentBits } = require('discord.js');
|
||||
// src/Bot.js - Update your import statements
|
||||
const CommandHandler = require('./commands/CommandHandler.js'); // Add .js extension
|
||||
const PlayerService = require('./services/PlayerService.js'); // Add .js extension
|
||||
class Bot {
|
||||
constructor() {
|
||||
this.client = new Client({
|
||||
intents: [
|
||||
GatewayIntentBits.Guilds,
|
||||
GatewayIntentBits.GuildMessages,
|
||||
GatewayIntentBits.MessageContent,
|
||||
],
|
||||
});
|
||||
|
||||
this.playerService = new PlayerService();
|
||||
this.commandHandler = new CommandHandler(this.playerService);
|
||||
|
||||
this.setupEventHandlers();
|
||||
}
|
||||
|
||||
setupEventHandlers() {
|
||||
this.client.on('error', this.handleError.bind(this));
|
||||
this.client.once('ready', this.handleReady.bind(this));
|
||||
this.client.on('interactionCreate', this.handleInteraction.bind(this));
|
||||
}
|
||||
|
||||
async start(token) {
|
||||
console.log('Starting bot...');
|
||||
try {
|
||||
await this.client.login(token);
|
||||
console.log('Login successful');
|
||||
} catch (error) {
|
||||
console.error('Login failed:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
handleError(error) {
|
||||
console.error('Discord client error:', error);
|
||||
}
|
||||
|
||||
handleReady() {
|
||||
console.log(`Logged in as ${this.client.user.tag}!`);
|
||||
}
|
||||
|
||||
async handleInteraction(interaction) {
|
||||
if (!interaction.isCommand()) return;
|
||||
|
||||
try {
|
||||
switch (interaction.commandName) {
|
||||
case 'ping':
|
||||
await this.commandHandler.handlePing(interaction);
|
||||
break;
|
||||
case 'finduser':
|
||||
await this.commandHandler.handleFindUser(interaction);
|
||||
break;
|
||||
case 'matchhistory':
|
||||
await this.commandHandler.handleMatchHistory(interaction);
|
||||
break;
|
||||
default:
|
||||
await interaction.reply({
|
||||
content: 'Unknown command',
|
||||
ephemeral: true
|
||||
});
|
||||
}
|
||||
} catch (error) {
|
||||
await this.handleCommandError(interaction, error);
|
||||
}
|
||||
}
|
||||
|
||||
async handleCommandError(interaction, error) {
|
||||
console.error('Command error:', error);
|
||||
|
||||
if (error.code === 10062) {
|
||||
console.log('Interaction expired');
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
const message = 'An error occurred while processing your command.';
|
||||
if (interaction.deferred) {
|
||||
await interaction.editReply({ content: message, ephemeral: true });
|
||||
} else if (!interaction.replied) {
|
||||
await interaction.reply({ content: message, ephemeral: true });
|
||||
}
|
||||
} catch (e) {
|
||||
console.error('Error while sending error message:', e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = Bot;
|
||||
235
src/commands/CommandHandler.js
Normal file
235
src/commands/CommandHandler.js
Normal file
@@ -0,0 +1,235 @@
|
||||
// src/commands/CommandHandler.js
|
||||
const { EmbedBuilder, ActionRowBuilder, ButtonBuilder, ButtonStyle } = require('discord.js');
|
||||
|
||||
class CommandHandler {
|
||||
constructor(playerService) {
|
||||
this.playerService = playerService;
|
||||
}
|
||||
|
||||
async handlePing(interaction) {
|
||||
await interaction.reply('Pong!');
|
||||
}
|
||||
|
||||
async handleFindUser(interaction) {
|
||||
await interaction.deferReply();
|
||||
|
||||
const username = interaction.options.getString('username');
|
||||
const gameFilter = interaction.options.getString('game');
|
||||
|
||||
const userData = await this.playerService.findUserByUsername(username);
|
||||
if (!userData || !userData.success) {
|
||||
await interaction.editReply('User not found or an error occurred while fetching data.');
|
||||
return;
|
||||
}
|
||||
|
||||
const playerData = typeof userData.player_data === 'string'
|
||||
? JSON.parse(userData.player_data)
|
||||
: userData.player_data;
|
||||
|
||||
const embed = await this.createUserEmbed(playerData, gameFilter);
|
||||
const row = this.createActionRow(playerData.username);
|
||||
|
||||
await interaction.editReply({ embeds: [embed], components: [row] });
|
||||
}
|
||||
|
||||
async handleMatchHistory(interaction) {
|
||||
await interaction.deferReply();
|
||||
|
||||
const username = interaction.options.getString('username');
|
||||
const gameFilter = interaction.options.getString('game');
|
||||
|
||||
try {
|
||||
const userData = await this.playerService.findUserByUsername(username);
|
||||
if (!userData || !userData.success) {
|
||||
await interaction.editReply('User not found or an error occurred while fetching data.');
|
||||
return;
|
||||
}
|
||||
|
||||
const playerData = typeof userData.player_data === 'string'
|
||||
? JSON.parse(userData.player_data)
|
||||
: userData.player_data;
|
||||
|
||||
const matches = Object.values(playerData.matches || {})
|
||||
.filter(match => !gameFilter || match.game_name.toLowerCase() === gameFilter.toLowerCase())
|
||||
.sort((a, b) => new Date(b.start_time) - new Date(a.start_time))
|
||||
.slice(0, 10);
|
||||
|
||||
if (matches.length === 0) {
|
||||
await interaction.editReply(
|
||||
gameFilter
|
||||
? `No matches found for ${username} in ${gameFilter}`
|
||||
: `No matches found for ${username}`
|
||||
);
|
||||
return;
|
||||
}
|
||||
|
||||
const embed = this.createMatchHistoryEmbed(playerData, matches);
|
||||
const row = this.createActionRow(playerData.username);
|
||||
|
||||
await interaction.editReply({ embeds: [embed], components: [row] });
|
||||
} catch (error) {
|
||||
console.error('Error in handleMatchHistory:', error);
|
||||
await interaction.editReply('An error occurred while processing the match history.');
|
||||
}
|
||||
}
|
||||
|
||||
createUserEmbed(playerData, gameFilter) {
|
||||
const profile = playerData.profile || {};
|
||||
const embed = new EmbedBuilder()
|
||||
.setTitle(`User: ${playerData.username || 'Unknown'}`)
|
||||
.setDescription(profile.bio || 'No bio available')
|
||||
.setColor('#0099ff');
|
||||
|
||||
// Add thumbnail (avatar)
|
||||
if (profile.avatar) {
|
||||
embed.setThumbnail(
|
||||
`https://www.vrbattles.gg/assets/uploads/profile/${profile.avatar}`
|
||||
);
|
||||
}
|
||||
|
||||
// Add badge image using PlayerService
|
||||
if (profile.current_level_badge) {
|
||||
const badgeImageUrl = this.playerService.getBadgeImageUrl(profile.current_level_badge);
|
||||
embed.setImage(badgeImageUrl);
|
||||
}
|
||||
|
||||
// Add profile fields
|
||||
const profileFields = [];
|
||||
if (profile.country) profileFields.push({ name: 'Country', value: profile.country, inline: true });
|
||||
if (profile.level) profileFields.push({ name: 'Level', value: profile.level.toString(), inline: true });
|
||||
if (profile.xp) profileFields.push({ name: 'Total XP', value: profile.xp.toString(), inline: true });
|
||||
if (profile.prestige) profileFields.push({ name: 'Prestige', value: profile.prestige.toString(), inline: true });
|
||||
|
||||
if (profileFields.length > 0) {
|
||||
embed.addFields(profileFields);
|
||||
}
|
||||
|
||||
// Add game stats if available
|
||||
const games = playerData.stats?.games;
|
||||
if (games) {
|
||||
Object.entries(games).forEach(([gameName, gameData]) => {
|
||||
if (gameFilter && gameName.toLowerCase() !== gameFilter.toLowerCase()) return;
|
||||
|
||||
embed.addFields({ name: gameName, value: '\u200B' });
|
||||
Object.entries(gameData).forEach(([mode, stats]) => {
|
||||
if (mode === 'Overall') return;
|
||||
|
||||
const statFields = [];
|
||||
if (stats.matches) statFields.push(`Matches: ${stats.matches}`);
|
||||
if (stats.wins) statFields.push(`Wins: ${stats.wins}`);
|
||||
if (stats.losses) statFields.push(`Losses: ${stats.losses}`);
|
||||
if (stats.win_rate) statFields.push(`Win Rate: ${stats.win_rate}`);
|
||||
|
||||
if (statFields.length > 0) {
|
||||
embed.addFields({
|
||||
name: mode,
|
||||
value: statFields.join(' | '),
|
||||
inline: true
|
||||
});
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// Add teams if available
|
||||
const teams = playerData.teams;
|
||||
if (teams && Object.keys(teams).length > 0) {
|
||||
const teamList = Object.values(teams)
|
||||
.map(team => `- **${team.team_name}** (${team.game_name} - ${team.game_mode})`)
|
||||
.join('\n');
|
||||
|
||||
embed.addFields(
|
||||
{ name: '\u200B', value: '**Teams**' },
|
||||
{ name: 'Team List', value: teamList || 'No teams available' }
|
||||
);
|
||||
}
|
||||
|
||||
// Add recent matches if available
|
||||
const matches = playerData.matches;
|
||||
if (matches && Object.keys(matches).length > 0) {
|
||||
const matchList = Object.values(matches)
|
||||
.sort((a, b) => new Date(b.start_time) - new Date(a.start_time))
|
||||
.slice(0, 3)
|
||||
.map(match => {
|
||||
const date = new Date(match.start_time).toLocaleDateString();
|
||||
const team1 = match.team1_name || 'Team 1';
|
||||
const team2 = match.team2_name || 'Team 2';
|
||||
const score = match.score || 'N/A';
|
||||
return `- **${team1}** vs **${team2}** on ${date} - Result: ${score}`;
|
||||
})
|
||||
.join('\n');
|
||||
|
||||
embed.addFields(
|
||||
{ name: '\u200B', value: '**Recent Matches**' },
|
||||
{ name: 'Match History', value: matchList || 'No recent matches' }
|
||||
);
|
||||
}
|
||||
|
||||
return embed;
|
||||
}
|
||||
|
||||
createMatchHistoryEmbed(playerData, matches) {
|
||||
const embed = new EmbedBuilder()
|
||||
.setTitle(`Match History for ${playerData.username}`)
|
||||
.setColor('#0099ff')
|
||||
.setDescription(`Last ${matches.length} matches`);
|
||||
|
||||
if (playerData.profile?.avatar) {
|
||||
embed.setThumbnail(
|
||||
`https://www.vrbattles.gg/assets/uploads/profile/${playerData.profile.avatar}`
|
||||
);
|
||||
}
|
||||
|
||||
// Add match stats
|
||||
const wins = matches.filter(m => {
|
||||
const isTeam1 = m.team1_name === playerData.username;
|
||||
return m.winner_id === (isTeam1 ? "1" : "2");
|
||||
}).length;
|
||||
|
||||
const winRate = ((wins / matches.length) * 100).toFixed(1);
|
||||
|
||||
embed.addFields(
|
||||
{ name: 'Recent Win Rate', value: `${winRate}%`, inline: true },
|
||||
{ name: 'Matches Analyzed', value: matches.length.toString(), inline: true },
|
||||
{ name: 'Recent Wins', value: wins.toString(), inline: true }
|
||||
);
|
||||
|
||||
// Add individual matches
|
||||
matches.forEach((match, index) => {
|
||||
const date = new Date(match.start_time).toLocaleDateString();
|
||||
const isTeam1 = match.team1_name === playerData.username;
|
||||
const opponent = isTeam1 ? match.team2_name : match.team1_name;
|
||||
const won = match.winner_id === (isTeam1 ? "1" : "2");
|
||||
const score = match.score || '0 - 0';
|
||||
const gameMode = match.game_mode ? ` (${match.game_mode})` : '';
|
||||
|
||||
const matchResult = won ? '✅ Victory' : '❌ Defeat';
|
||||
|
||||
embed.addFields({
|
||||
name: `Match ${index + 1} - ${matchResult}`,
|
||||
value: `🎮 **${match.game_name}${gameMode}**\n` +
|
||||
`🆚 vs ${opponent}\n` +
|
||||
`📊 Score: ${score}\n` +
|
||||
`📅 ${date}`,
|
||||
inline: false
|
||||
});
|
||||
});
|
||||
|
||||
return embed;
|
||||
}
|
||||
|
||||
createActionRow(username) {
|
||||
return new ActionRowBuilder().addComponents(
|
||||
new ButtonBuilder()
|
||||
.setLabel('🔵 View Profile')
|
||||
.setStyle(ButtonStyle.Link)
|
||||
.setURL(this.playerService.getProfileUrl(username)),
|
||||
new ButtonBuilder()
|
||||
.setLabel('🟡 View Stats')
|
||||
.setStyle(ButtonStyle.Link)
|
||||
.setURL(this.playerService.getStatsUrl(username))
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = CommandHandler;
|
||||
55
src/services/PlayerService.js
Normal file
55
src/services/PlayerService.js
Normal file
@@ -0,0 +1,55 @@
|
||||
// src/services/PlayerService.js
|
||||
const axios = require('axios');
|
||||
|
||||
class PlayerService {
|
||||
constructor() {
|
||||
this.baseUrl = 'https://www.vrbattles.gg';
|
||||
}
|
||||
|
||||
async findUserByUsername(username) {
|
||||
try {
|
||||
const response = await axios.get(
|
||||
`${this.baseUrl}/api/get_player_data_by_username/${encodeURIComponent(username)}`,
|
||||
{
|
||||
timeout: 5000
|
||||
}
|
||||
);
|
||||
return response.data;
|
||||
} catch (error) {
|
||||
console.error('Error fetching user data:', error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
getBadgeImageUrl(badgeName) {
|
||||
if (!badgeName) {
|
||||
return `${this.baseUrl}/assets/images/badges/xp_badges/A/BADGE_A1_72p.png`;
|
||||
}
|
||||
|
||||
badgeName = badgeName.toUpperCase().replace(/ /g, '_');
|
||||
let badgeUrl = `${this.baseUrl}/assets/images/badges/xp_badges/`;
|
||||
|
||||
if (badgeName.startsWith('A')) {
|
||||
badgeUrl += 'A/';
|
||||
} else if (badgeName.startsWith('B')) {
|
||||
badgeUrl += 'B/';
|
||||
} else if (badgeName.startsWith('C')) {
|
||||
badgeUrl += 'C/';
|
||||
} else {
|
||||
badgeUrl += 'A/';
|
||||
}
|
||||
|
||||
badgeUrl += `BADGE_${badgeName}_72p.png`;
|
||||
return badgeUrl;
|
||||
}
|
||||
|
||||
getProfileUrl(username) {
|
||||
return `${this.baseUrl}/profile/${username}`;
|
||||
}
|
||||
|
||||
getStatsUrl(username) {
|
||||
return `${this.baseUrl}/profile/${username}/stats`;
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = PlayerService;
|
||||
58
src/tests/test.js
Normal file
58
src/tests/test.js
Normal file
@@ -0,0 +1,58 @@
|
||||
// src/test.js
|
||||
const PlayerService = require('./services/PlayerService');
|
||||
|
||||
async function testPlayerService() {
|
||||
console.log('Testing PlayerService...');
|
||||
|
||||
const playerService = new PlayerService();
|
||||
|
||||
try {
|
||||
// Test user lookup
|
||||
console.log('\nTesting user lookup...');
|
||||
const userData = await playerService.findUserByUsername('uthcowboys');
|
||||
console.log('User data received:', !!userData);
|
||||
console.log('Success:', userData?.success);
|
||||
|
||||
if (userData && userData.success) {
|
||||
const playerData = JSON.parse(userData.player_data);
|
||||
console.log('\nBasic user info:');
|
||||
console.log('Username:', playerData.username);
|
||||
console.log('Level:', playerData.profile?.level);
|
||||
console.log('Country:', playerData.profile?.country);
|
||||
|
||||
// Test matches
|
||||
console.log('\nRecent matches:');
|
||||
const matches = Object.values(playerData.matches || {})
|
||||
.sort((a, b) => new Date(b.start_time) - new Date(a.start_time))
|
||||
.slice(0, 3);
|
||||
|
||||
matches.forEach(match => {
|
||||
console.log(`- ${match.game_name} (${match.game_mode}): ${match.score} on ${new Date(match.start_time).toLocaleDateString()}`);
|
||||
});
|
||||
|
||||
// Test stats
|
||||
console.log('\nGame stats:');
|
||||
const games = playerData.stats?.games || {};
|
||||
Object.entries(games).forEach(([gameName, gameData]) => {
|
||||
console.log(`\n${gameName}:`);
|
||||
Object.entries(gameData).forEach(([mode, stats]) => {
|
||||
if (mode !== 'Overall') {
|
||||
console.log(` ${mode}:`);
|
||||
console.log(` Matches: ${stats.matches}`);
|
||||
console.log(` Win Rate: ${stats.win_rate}`);
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Test failed:', error);
|
||||
}
|
||||
}
|
||||
|
||||
// Run tests
|
||||
console.log('Starting tests...\n');
|
||||
testPlayerService().then(() => {
|
||||
console.log('\nTests completed.');
|
||||
}).catch(error => {
|
||||
console.error('Tests failed:', error);
|
||||
});
|
||||
69
src/tests/testcommands.js
Normal file
69
src/tests/testcommands.js
Normal file
@@ -0,0 +1,69 @@
|
||||
// src/testCommands.js
|
||||
const CommandHandler = require('./commands/CommandHandler');
|
||||
const PlayerService = require('./services/PlayerService');
|
||||
|
||||
// Mock Discord interaction
|
||||
class MockInteraction {
|
||||
constructor(commandName, options = {}) {
|
||||
this.commandName = commandName;
|
||||
this.deferred = false;
|
||||
this.replied = false;
|
||||
this.options = {
|
||||
getString: (name) => options[name]
|
||||
};
|
||||
}
|
||||
|
||||
async deferReply() {
|
||||
this.deferred = true;
|
||||
console.log('Interaction deferred');
|
||||
}
|
||||
|
||||
async reply(content) {
|
||||
this.replied = true;
|
||||
console.log('Reply sent:', content);
|
||||
}
|
||||
|
||||
async editReply(content) {
|
||||
console.log('Edit reply:', content);
|
||||
if (content.embeds) {
|
||||
console.log('\nEmbed fields:');
|
||||
content.embeds.forEach(embed => {
|
||||
console.log('Title:', embed.data.title);
|
||||
console.log('Description:', embed.data.description);
|
||||
if (embed.data.fields) {
|
||||
embed.data.fields.forEach(field => {
|
||||
console.log(`${field.name}: ${field.value}`);
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async function testCommands() {
|
||||
console.log('Testing commands...\n');
|
||||
|
||||
const playerService = new PlayerService();
|
||||
const commandHandler = new CommandHandler(playerService);
|
||||
|
||||
// Test finduser command
|
||||
console.log('Testing /finduser command...');
|
||||
const findUserInteraction = new MockInteraction('finduser', {
|
||||
username: 'uthcowboys'
|
||||
});
|
||||
await commandHandler.handleFindUser(findUserInteraction);
|
||||
|
||||
// Test matchhistory command
|
||||
console.log('\nTesting /matchhistory command...');
|
||||
const matchHistoryInteraction = new MockInteraction('matchhistory', {
|
||||
username: 'uthcowboys'
|
||||
});
|
||||
await commandHandler.handleMatchHistory(matchHistoryInteraction);
|
||||
}
|
||||
|
||||
// Run tests
|
||||
testCommands().then(() => {
|
||||
console.log('\nCommand tests completed.');
|
||||
}).catch(error => {
|
||||
console.error('Command tests failed:', error);
|
||||
});
|
||||
12
src/utils/helpers.js
Normal file
12
src/utils/helpers.js
Normal file
@@ -0,0 +1,12 @@
|
||||
// src/utils/helpers.js
|
||||
function safeGet(obj, path, defaultValue = null) {
|
||||
try {
|
||||
return path.split('.').reduce((acc, part) => acc && acc[part], obj) ?? defaultValue;
|
||||
} catch (e) {
|
||||
return defaultValue;
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
safeGet
|
||||
};
|
||||
Reference in New Issue
Block a user