379 lines
7.5 KiB
JavaScript
379 lines
7.5 KiB
JavaScript
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;
|
|
}
|
|
}
|
|
} |