const db = require('../db/models'); const ValidationError = require('./notifications/errors/validation'); const Sequelize = db.Sequelize; const Op = Sequelize.Op; /** * @param {string} permission * @param {object} currentUser */ async function checkPermissions(permission, currentUser) { if (!currentUser) { throw new ValidationError('auth.unauthorized'); } const userPermission = currentUser.custom_permissions.find( (cp) => cp.name === permission, ); if (userPermission) { return true; } try { if (!currentUser.app_role) { throw new ValidationError('auth.forbidden'); } const permissions = await currentUser.app_role.getPermissions(); return !!permissions.find((p) => p.name === permission); } catch (e) { throw e; } } module.exports = class SearchService { static async search(searchQuery, currentUser ) { try { if (!searchQuery) { throw new ValidationError('iam.errors.searchQueryRequired'); } const tableColumns = { "users": [ "firstName", "lastName", "phoneNumber", "email", ], "mappers": [ "osu_username", "avatar_url", ], "artists": [ "name", "normalized_name", "image_url", ], "beatmap_sets": [ "title", "artist_name", "background_image_url", ], "beatmaps": [ "title", "version", "artist_name", "background_image_url", ], "game_sessions": [ "session_key", "daily_seed_key", ], "game_rounds": [ "suspicion_reason", ], "leaderboard_entries": [ "scope_key", ], "aggregation_jobs": [ "error_message", ], "api_request_logs": [ "ip_address", "user_agent", "route", "method", ], }; const columnsInt = { "mappers": [ "osu_user_numeric", "total_ranked_loved_playcount", ], "artists": [ "total_ranked_loved_playcount", ], "beatmap_sets": [ "osu_beatmapset_numeric", ], "beatmaps": [ "osu_beatmap_numeric", "playcount", ], "game_sessions": [ "starting_lives", "lives_remaining", "streak_current", "streak_best", "round_time_min_seconds", "round_time_max_seconds", ], "game_rounds": [ "round_index", "time_limit_seconds", ], "leaderboard_entries": [ "score_value", "best_streak", "games_played", ], "aggregation_jobs": [ "items_processed", "items_failed", ], "api_request_logs": [ "status_code", "duration_ms", ], }; let allFoundRecords = []; for (const tableName in tableColumns) { if (tableColumns.hasOwnProperty(tableName)) { const attributesToSearch = tableColumns[tableName]; const attributesIntToSearch = columnsInt[tableName] || []; const whereCondition = { [Op.or]: [ ...attributesToSearch.map(attribute => ({ [attribute]: { [Op.iLike] : `%${searchQuery}%`, }, })), ...attributesIntToSearch.map(attribute => ( Sequelize.where( Sequelize.cast(Sequelize.col(`${tableName}.${attribute}`), 'varchar'), { [Op.iLike]: `%${searchQuery}%` } ) )), ], }; const hasPermission = await checkPermissions(`READ_${tableName.toUpperCase()}`, currentUser); if (!hasPermission) { continue; } const foundRecords = await db[tableName].findAll({ where: whereCondition, attributes: [...tableColumns[tableName], 'id', ...attributesIntToSearch], }); const modifiedRecords = foundRecords.map((record) => { const matchAttribute = []; for (const attribute of attributesToSearch) { if (record[attribute]?.toLowerCase()?.includes(searchQuery.toLowerCase())) { matchAttribute.push(attribute); } } for (const attribute of attributesIntToSearch) { const castedValue = String(record[attribute]); if (castedValue && castedValue.toLowerCase().includes(searchQuery.toLowerCase())) { matchAttribute.push(attribute); } } return { ...record.get(), matchAttribute, tableName, }; }); allFoundRecords = allFoundRecords.concat(modifiedRecords); } } return allFoundRecords; } catch (error) { throw error; } } }