ale/discordbot
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

repair search images

Browse Source 
Signed-off-by: ale <ale@manalejandro.com>
This commit is contained in:
ale 2025-06-08 03:55:22 +02:00
parent 618553a205
commit a9420500af
Signed by: ale
GPG Key ID: 244A9C4DAB1C0C81
1 changed files with 72 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
index.js
View File

@ -196,7 +196,13 @@ function isAudioFile(url) {
async function validateMultimediaUrl(url) {
try {
const response = await fetch(url, { method: 'HEAD', timeout: 5000 });
const response = await fetch(url, {
method: 'HEAD',
timeout: 5000,
headers: {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
}
});
const contentType = response.headers.get('content-type') || '';
const isMultimedia = contentType.startsWith('image/') ||
contentType.startsWith('video/') ||
@ -212,6 +218,43 @@ async function validateMultimediaUrl(url) {
}
}
// More lenient validation for search results
async function validateImageUrl(url) {
try {
// First check if the URL looks like an image file
const urlLower = url.toLowerCase();
const imageExtensions = ['.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp'];
const hasImageExtension = imageExtensions.some(ext => urlLower.includes(ext));
// If it has an image extension or is from known image hosting sites, consider it valid
const knownImageHosts = ['imgur.com', 'i.imgur.com', 'cdn.', 'images.', 'img.', 'static.', 'upload.wikimedia.org'];
const isFromImageHost = knownImageHosts.some(host => urlLower.includes(host));
if (hasImageExtension || isFromImageHost) {
return { valid: true, contentType: 'image/unknown' };
}
// For other URLs, try a quick HEAD request but don't be too strict
const response = await fetch(url, {
method: 'HEAD',
timeout: 3000,
headers: {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
}
});
const contentType = response.headers.get('content-type') || '';
return {
valid: response.ok && (contentType.startsWith('image/') || contentType === ''),
contentType: contentType || 'image/unknown'
};
} catch (error) {
// If validation fails, assume it might be valid anyway for search results
return { valid: true, contentType: 'image/unknown', error: error.message };
}
}
// Bitcoin transaction monitoring functions
async function initBitcoinMonitoring() {
try {
@ -421,14 +464,14 @@ client.on(Events.InteractionCreate, async interaction => {
if (mediaType === 'image') {
// Enhanced image search with better filtering
const searchTerm = `${query} filetype:jpg OR filetype:png OR filetype:gif OR filetype:webp`;
const results = await googleIt({ query: searchTerm, limit: count * 5 });
const results = await googleIt({ query: searchTerm, limit: count * 3 });
const imageResults = [];
for (const result of results) {
const url = result.link;
// Validate if the URL is actually an image
const validation = await validateMultimediaUrl(url);
if (validation.valid && validation.contentType.startsWith('image/')) {
// Use more lenient validation for image search results
const validation = await validateImageUrl(url);
if (validation.valid) {
imageResults.push({
title: result.title,
url: url,
@ -439,7 +482,30 @@ client.on(Events.InteractionCreate, async interaction => {
}
if (imageResults.length === 0) {
await interaction.editReply(`No valid image results found for "${query}"`);
// If no results found, try a simpler search without file type restrictions
const simpleResults = await googleIt({ query: query, limit: count * 2 });
for (const result of simpleResults) {
const url = result.link;
const urlLower = url.toLowerCase();
// Check if URL contains image indicators
const imageIndicators = ['.jpg', '.jpeg', '.png', '.gif', '.webp', 'image', 'img', 'photo'];
const hasImageIndicator = imageIndicators.some(indicator => urlLower.includes(indicator));
if (hasImageIndicator) {
imageResults.push({
title: result.title,
url: url,
contentType: 'image/unknown'
});
if (imageResults.length >= count) break;
}
}
}
if (imageResults.length === 0) {
await interaction.editReply(`No valid image results found for "${query}". Try a more specific search term.`);
return;
}