wikibase-cli/lib/fetch_and_log_ids_data.js

A command-line interface to Wikibase

import { isEntitySchemaId, simplifyEntity } from 'wikibase-sdk'
import { exitOnMissingInstance } from '#lib/exit_on_missing'
import optionsFactory from '#lib/parse_simplify_options'
import { getEntitiesByBatches } from './get_entities_by_batches.js'
import { getEntitiesSchemas } from './get_entities_schemas.js'
import getTtlEntities from './get_ttl_entities.js'
import logNdjson from './log_ndjson.js'
import parseProps from './parse_props.js'
import pickSelectedSubprops from './pick_selected_subprops.js'
import program from './program.js'

export function fetchAndLogIdsData (ids) {
  exitOnMissingInstance(program.instance)
  const requestedPropsAndSubProps = parseProps(program.props)
  const requestedProps = Object.keys(requestedPropsAndSubProps)
  const options = optionsFactory(program)
  const { simplify: simplifyFlag, format } = program

  if (format === 'ttl') return getTtlEntities(ids)

  const entitiesSchemasIds = ids.filter(isEntitySchemaId)

  if (entitiesSchemasIds.length > 0) return getEntitiesSchemas(entitiesSchemasIds)

  const onResponse = entities => {
    entities = entities.map(entity => pickSelectedSubprops(entity, requestedPropsAndSubProps))
    if (simplifyFlag) {
      entities = entities.map(entity => simplifyEntity(entity, options))
    }
    logNdjson(entities)
  }

  return getEntitiesByBatches({ ids, props: requestedProps, onResponse })
}