1 | import { Transform } from 'stream'
|
2 | import { Reviver } from '@naturalcycles/js-lib'
|
3 | import { TransformTyped } from '../stream.model'
|
4 |
|
5 | export interface TransformJsonParseOptions {
|
6 | |
7 |
|
8 |
|
9 |
|
10 |
|
11 | strict?: boolean
|
12 |
|
13 | reviver?: Reviver
|
14 | }
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 |
|
26 |
|
27 |
|
28 |
|
29 |
|
30 | export function transformJsonParse<OUT = any>(
|
31 | opt: TransformJsonParseOptions = {},
|
32 | ): TransformTyped<string | Buffer, OUT> {
|
33 | const { strict = true, reviver } = opt
|
34 |
|
35 | return new Transform({
|
36 | writableObjectMode: false,
|
37 | readableObjectMode: true,
|
38 | transform(chunk: string, _, cb) {
|
39 | try {
|
40 | const data = JSON.parse(chunk, reviver)
|
41 | cb(null, data)
|
42 | } catch (err) {
|
43 | if (strict) {
|
44 | cb(err as Error)
|
45 | } else {
|
46 | console.error(err)
|
47 | cb()
|
48 | }
|
49 | }
|
50 | },
|
51 | })
|
52 | }
|
53 |
|
54 |
|
55 | export const bufferReviver: Reviver = (k, v) => {
|
56 | if (v !== null && typeof v === 'object' && v.type === 'Buffer' && Array.isArray(v.data)) {
|
57 | return Buffer.from(v.data)
|
58 | }
|
59 |
|
60 | return v
|
61 | }
|