mirror of
https://github.com/wassname/talk.git
synced 2026-06-30 01:58:00 +08:00
Wyatt Johnson
118 lines
3.4 KiB
JavaScript
118 lines
3.4 KiB
JavaScript
const DataLoader = require('dataloader');
|
|
const url = require('url');
|
|
|
|
const errors = require('../../errors');
|
|
const scraper = require('../../services/scraper');
|
|
const util = require('./util');
|
|
|
|
const AssetModel = require('../../models/asset');
|
|
const AssetsService = require('../../services/assets');
|
|
|
|
/**
|
|
* Retrieves assets by an array of ids.
|
|
* @param {Object} context the context of the request
|
|
* @param {Array} ids array of ids to lookup
|
|
*/
|
|
const genAssetsByID = (context, ids) => AssetModel.find({
|
|
id: {
|
|
$in: ids
|
|
}
|
|
}).then(util.singleJoinBy(ids, 'id'));
|
|
|
|
/**
|
|
* [getAssetsByQuery description]
|
|
* @param {Object} context the context of the request
|
|
* @param {Object} query the query
|
|
* @return {Promise} resolves the assets
|
|
*/
|
|
const getAssetsByQuery = async (context, query) => {
|
|
|
|
// If we are requesting based on a limit, ask for one more than we want.
|
|
const limit = query.limit;
|
|
if (limit) {
|
|
query.limit += 1;
|
|
}
|
|
|
|
const nodes = await AssetsService.search(query);
|
|
|
|
// The hasNextPage is always handled the same (ask for one more than we need,
|
|
// if there is one more, than there is more).
|
|
let hasNextPage = false;
|
|
if (limit && nodes.length > limit) {
|
|
|
|
// There was one more than we expected! Set hasNextPage = true and remove
|
|
// the last item from the array that we requested.
|
|
hasNextPage = true;
|
|
nodes.splice(limit, 1);
|
|
}
|
|
|
|
return {
|
|
startCursor: nodes && nodes.length > 0 ? nodes[0].created_at : null,
|
|
endCursor: nodes && nodes.length > 0 ? nodes[nodes.length - 1].created_at : null,
|
|
hasNextPage,
|
|
nodes,
|
|
};
|
|
};
|
|
|
|
/**
|
|
* This endpoint find or creates an asset at the given url when it is loaded.
|
|
* @param {Object} context the context of the request
|
|
* @param {String} asset_url the url passed in from the query
|
|
* @returns {Promise} resolves to the asset
|
|
*/
|
|
const findOrCreateAssetByURL = async (context, asset_url) => {
|
|
|
|
// Verify that the asset_url is parsable.
|
|
let parsed_asset_url = url.parse(asset_url);
|
|
if (!parsed_asset_url.protocol) {
|
|
throw errors.ErrInvalidAssetURL;
|
|
}
|
|
|
|
let asset = await AssetsService.findOrCreateByUrl(asset_url);
|
|
|
|
// If the asset wasn't scraped before, scrape it! Otherwise just return
|
|
// the asset.
|
|
if (!asset.scraped) {
|
|
await scraper.create(asset);
|
|
}
|
|
|
|
return asset;
|
|
};
|
|
|
|
const getAssetsForMetrics = async ({loaders: {Comments}}) => {
|
|
return Comments.getByQuery({action_type: 'FLAG'})
|
|
.then((connection) => connection.nodes);
|
|
};
|
|
|
|
const findByUrl = async (context, asset_url) => {
|
|
|
|
// Verify that the asset_url is parsable.
|
|
let parsed_asset_url = url.parse(asset_url);
|
|
if (!parsed_asset_url.protocol) {
|
|
throw errors.ErrInvalidAssetURL;
|
|
}
|
|
|
|
return AssetsService.findByUrl(asset_url);
|
|
};
|
|
|
|
/**
|
|
* Creates a set of loaders based on a GraphQL context.
|
|
* @param {Object} context the context of the GraphQL request
|
|
* @return {Object} object of loaders
|
|
*/
|
|
|
|
module.exports = (context) => ({
|
|
Assets: {
|
|
|
|
// TODO: decide whether we want to move these to mutators or not, as in fact
|
|
// this operation create a new asset if one isn't found.
|
|
getByURL: (url) => findOrCreateAssetByURL(context, url),
|
|
|
|
findByUrl: (url) => findByUrl(context, url),
|
|
getByQuery: (query) => getAssetsByQuery(context, query),
|
|
getByID: new DataLoader((ids) => genAssetsByID(context, ids)),
|
|
getForMetrics: () => getAssetsForMetrics(context),
|
|
getAll: new util.SingletonResolver(() => AssetModel.find({}))
|
|
}
|
|
});
|