-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #173 from zazuko/loader
extract loader package
- Loading branch information
Showing
12 changed files
with
141 additions
and
90 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
--- | ||
"@zazuko/vocabulary-loader": major | ||
--- | ||
|
||
First release |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
--- | ||
"@zazuko/prefixes": patch | ||
--- | ||
|
||
Export required type |
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,90 @@ | ||
/* eslint-disable no-console */ | ||
import { Stream, DataFactory, DatasetCore, DatasetCoreFactory } from 'rdf-js' | ||
import { Readable } from 'stream' | ||
import rdf from '@zazuko/env' | ||
import type { Environment } from '@rdfjs/environment/Environment.js' | ||
import ParserN3 from '@rdfjs/parser-n3' | ||
import fromStream from 'rdf-dataset-ext/fromStream.js' | ||
import addAll from 'rdf-dataset-ext/addAll.js' | ||
import toStream from 'rdf-dataset-ext/toStream.js' | ||
import { loadDatasetStream } from './lib/loadDataset.js' | ||
|
||
export type Datasets<P extends Record<string, string>> = Record<keyof P, DatasetCore> | ||
|
||
export interface VocabulariesOptions<P extends Record<string, string>> { | ||
only?: (keyof P)[] | null | ||
factory?: Environment<DatasetCoreFactory | DataFactory> | ||
} | ||
|
||
export interface VocabulariesDatasetOptions<P extends Record<string, string>> extends VocabulariesOptions<P> { | ||
stream?: false | ||
} | ||
|
||
export interface VocabulariesStreamOptions<P extends Record<string, string>> extends VocabulariesOptions<P> { | ||
stream: true | ||
} | ||
|
||
export interface Loader<P extends Record<string, string>> { | ||
(options?: VocabulariesDatasetOptions<P>): Promise<Datasets<P>> | ||
(options: VocabulariesStreamOptions<P>): Promise<Stream & Readable> | ||
} | ||
|
||
export function create<P extends Record<string, string>>(prefixMap: P): Loader<P> { | ||
return async function vocabularies({ only = null, factory = rdf, stream = false }: VocabulariesDatasetOptions<P> | VocabulariesStreamOptions<P> = {}) { | ||
let selectedPrefixes: (keyof typeof prefixMap)[] = [] | ||
|
||
if (!!only && Array.isArray(only)) { | ||
only.forEach((prefix: keyof P) => { | ||
if (prefix in prefixMap) { | ||
selectedPrefixes.push(prefix) | ||
} else { | ||
console.warn(`unknown prefix '${String(prefix)}'`) | ||
} | ||
}) | ||
} | ||
if (!selectedPrefixes.length) { | ||
selectedPrefixes = Object.keys(prefixMap) | ||
} | ||
|
||
const promises = selectedPrefixes.map((prefix) => loadFile(prefix, { customSelection: !!only, factory })) | ||
const datasets = await Promise.all(promises) | ||
|
||
if (stream !== false) { | ||
let combinedDataset = factory.dataset() | ||
datasets.forEach((dataset) => { | ||
if (dataset && dataset.size) { | ||
combinedDataset = addAll(combinedDataset, dataset) | ||
} | ||
}) | ||
// eslint-disable-next-line @typescript-eslint/no-explicit-any | ||
return toStream(combinedDataset) as any | ||
} | ||
|
||
// eslint-disable-next-line @typescript-eslint/no-explicit-any | ||
const result: Datasets<P> = {} as any | ||
datasets.forEach((dataset, i) => { | ||
if (dataset && dataset.size) { | ||
result[selectedPrefixes[i]] = dataset | ||
} | ||
}) | ||
return result | ||
} | ||
} | ||
|
||
interface LoadFileOptions { | ||
customSelection?: boolean | ||
factory: Environment<DatasetCoreFactory> | ||
} | ||
|
||
export async function loadFile<P extends Record<string, string>>(prefix: keyof P, { customSelection, factory }: LoadFileOptions) { | ||
try { | ||
const parserN3 = new ParserN3() | ||
const readStream = await loadDatasetStream(String(prefix)) | ||
const quadStream = parserN3.import(readStream) | ||
return fromStream(factory.dataset(), quadStream) | ||
} catch { | ||
if (customSelection) { | ||
console.warn(`unavailable prefix '${String(prefix)}'`) | ||
} | ||
} | ||
} |
2 changes: 1 addition & 1 deletion
2
packages/vocabularies/lib/loadDataset.ts → packages/loader/lib/loadDataset.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
{ | ||
"name": "@zazuko/vocabulary-loader", | ||
"version": "0.0.0", | ||
"type": "module", | ||
"main": "index.js", | ||
"files": [ | ||
"**/*.js", | ||
"**/*.d.ts" | ||
], | ||
"dependencies": { | ||
"@zazuko/env": "^1.6.1", | ||
"@rdfjs/parser-n3": "^2.0.1", | ||
"@types/rdfjs__environment": "^0.1", | ||
"rdf-dataset-ext": "^1.1.0" | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,84 +1,6 @@ | ||
/* eslint-disable no-console */ | ||
import { Stream, DataFactory, DatasetCore, DatasetCoreFactory } from 'rdf-js' | ||
import { Readable } from 'stream' | ||
import rdf from '@zazuko/env' | ||
import type { Environment } from '@rdfjs/environment/Environment.js' | ||
import prefixes from '@zazuko/prefixes' | ||
import ParserN3 from '@rdfjs/parser-n3' | ||
import fromStream from 'rdf-dataset-ext/fromStream.js' | ||
import addAll from 'rdf-dataset-ext/addAll.js' | ||
import toStream from 'rdf-dataset-ext/toStream.js' | ||
import { loadDatasetStream } from './lib/loadDataset.js' | ||
import { create } from '@zazuko/vocabulary-loader' | ||
import prefixes from './prefixes.js' | ||
|
||
export type Datasets = Record<keyof typeof prefixes, DatasetCore> | ||
export { loadFile, Datasets } from '@zazuko/vocabulary-loader' | ||
|
||
interface VocabulariesOptions { | ||
only?: (keyof typeof prefixes)[] | null | ||
factory?: Environment<DatasetCoreFactory | DataFactory> | ||
} | ||
|
||
interface VocabulariesDatasetOptions extends VocabulariesOptions { | ||
stream?: false | ||
} | ||
|
||
interface VocabulariesStreamOptions extends VocabulariesOptions { | ||
stream: true | ||
} | ||
|
||
export async function vocabularies (options?: VocabulariesDatasetOptions): Promise<Datasets> | ||
export async function vocabularies (options: VocabulariesStreamOptions): Promise<Stream & Readable> | ||
export async function vocabularies({ only = null, factory = rdf, stream = false }: VocabulariesDatasetOptions | VocabulariesStreamOptions = {}) { | ||
let selectedPrefixes: (keyof typeof prefixes)[] = [] | ||
|
||
if (!!only && Array.isArray(only)) { | ||
only.forEach((prefix: keyof typeof prefixes) => { | ||
if (prefix in prefixes) { | ||
selectedPrefixes.push(prefix) | ||
} else { | ||
console.warn(`unknown prefix '${prefix}'`) | ||
} | ||
}) | ||
} | ||
if (!selectedPrefixes.length) { | ||
selectedPrefixes = Object.keys(prefixes) | ||
} | ||
|
||
const promises = selectedPrefixes.map((prefix) => loadFile(prefix, { customSelection: !!only, factory })) | ||
const datasets = await Promise.all(promises) | ||
|
||
if (stream !== false) { | ||
let combinedDataset = factory.dataset() | ||
datasets.forEach((dataset) => { | ||
if (dataset && dataset.size) { | ||
combinedDataset = addAll(combinedDataset, dataset) | ||
} | ||
}) | ||
return toStream(combinedDataset) as any | ||
} | ||
|
||
const result: Datasets = {} | ||
datasets.forEach((dataset, i) => { | ||
if (dataset && dataset.size) { | ||
result[selectedPrefixes[i]] = dataset | ||
} | ||
}) | ||
return result | ||
} | ||
|
||
interface LoadFileOptions { | ||
customSelection?: boolean | ||
factory: Environment<DatasetCoreFactory> | ||
} | ||
|
||
export async function loadFile(prefix: keyof typeof prefixes, { customSelection, factory }: LoadFileOptions) { | ||
try { | ||
const parserN3 = new ParserN3() | ||
const readStream = await loadDatasetStream(prefix) | ||
const quadStream = parserN3.import(readStream) | ||
return fromStream(factory.dataset(), quadStream) | ||
} catch { | ||
if (customSelection) { | ||
console.warn(`unavailable prefix '${prefix}'`) | ||
} | ||
} | ||
} | ||
export const vocabularies = create(prefixes) |