123 lines
3.0 KiB
JavaScript
123 lines
3.0 KiB
JavaScript
import knex from './knex.js';
|
|
import redis from './redis.js';
|
|
import initLogger from './logger.js';
|
|
|
|
const logger = initLogger();
|
|
|
|
export function curateEntity(entity, context) {
|
|
if (!entity) {
|
|
return null;
|
|
}
|
|
|
|
return {
|
|
id: entity.id,
|
|
name: entity.name,
|
|
slug: entity.slug,
|
|
type: entity.type,
|
|
url: entity.url,
|
|
isIndependent: entity.independent,
|
|
hasLogo: entity.has_logo,
|
|
parent: curateEntity(entity.parent, context),
|
|
children: context?.children?.filter((child) => child.parent_id === entity.id).map((child) => curateEntity({ ...child, parent: entity }, { parent: entity })) || [],
|
|
...context?.append?.[entity.id],
|
|
};
|
|
}
|
|
|
|
export async function fetchEntities(options) {
|
|
const entities = await knex('entities')
|
|
.modify((builder) => {
|
|
if (options.query) {
|
|
builder.where((whereBuilder) => {
|
|
whereBuilder
|
|
.whereLike('name', options.query)
|
|
.orWhereLike('slug', options.query);
|
|
});
|
|
}
|
|
|
|
if (options.type === 'primary') {
|
|
builder
|
|
.where('type', 'network')
|
|
.orWhere('independent', true)
|
|
.orWhereNull('parent_id');
|
|
|
|
return;
|
|
}
|
|
|
|
if (options.type) {
|
|
builder.where('type', options.type);
|
|
}
|
|
})
|
|
.orderBy(...(options.order || ['name', 'asc']))
|
|
.limit(options.limit || 1000);
|
|
|
|
return entities.map((entityEntry) => curateEntity(entityEntry));
|
|
}
|
|
|
|
export async function fetchEntitiesById(entityIds, options = {}) {
|
|
const [entities, children] = await Promise.all([
|
|
knex('entities')
|
|
.select('entities.*', knex.raw('row_to_json(parents) as parent'))
|
|
.whereIn('entities.id', entityIds)
|
|
.leftJoin('entities as parents', 'parents.id', 'entities.parent_id')
|
|
.modify((builder) => {
|
|
if (options.order) {
|
|
builder.orderBy(...options.order);
|
|
}
|
|
})
|
|
.groupBy('entities.id', 'parents.id'),
|
|
options.includeChildren ? knex('entities')
|
|
.whereIn('entities.parent_id', entityIds)
|
|
.orderBy('slug') : [],
|
|
]);
|
|
|
|
if (options.order) {
|
|
return entities.map((entityEntry) => curateEntity(entityEntry, {
|
|
append: options.append,
|
|
children,
|
|
}));
|
|
}
|
|
|
|
const curatedEntities = entityIds.map((entityId) => {
|
|
const entity = entities.find((entityEntry) => entityEntry.id === entityId);
|
|
|
|
if (!entity) {
|
|
console.warn(`Can't match entity ${entityId}`);
|
|
return null;
|
|
}
|
|
|
|
return curateEntity(entity, {
|
|
append: options.append,
|
|
children,
|
|
});
|
|
}).filter(Boolean);
|
|
|
|
return curatedEntities;
|
|
}
|
|
|
|
export async function getEntityIdsBySlug(slugs, domain) {
|
|
const ids = await Promise.all(slugs.map(async (slug) => {
|
|
if (!slug) {
|
|
return null;
|
|
}
|
|
|
|
if (Number(slug)) {
|
|
return Number(slug); // already an ID or missing
|
|
}
|
|
|
|
const id = await redis.hGet(`traxxx:${domain}:id_by_slug`, slug);
|
|
|
|
return Number(id);
|
|
}));
|
|
|
|
return ids.filter(Boolean);
|
|
}
|
|
|
|
export async function cacheEntityIds() {
|
|
const entities = await knex('entities').select('id', 'slug', 'type');
|
|
|
|
await redis.del('traxxx:entities:id_by_slug');
|
|
await redis.hSet('traxxx:entities:id_by_slug', entities.map((entity) => [entity.type === 'network' ? `_${entity.slug}` : entity.slug, entity.id]));
|
|
|
|
logger.info('Cached entity IDs by slug');
|
|
}
|