-
Notifications
You must be signed in to change notification settings - Fork 150
Expand file tree
/
Copy pathbinary.ts
More file actions
90 lines (84 loc) · 2.99 KB
/
binary.ts
File metadata and controls
90 lines (84 loc) · 2.99 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import { Scalar } from '../../nodes/Scalar.ts'
import { stringifyString } from '../../stringify/stringifyString.ts'
import type { ScalarTag } from '../types.ts'
declare global {
interface Uint8Array {
/** https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Uint8Array/toBase64 */
toBase64(options?: unknown): string
}
interface Uint8ArrayConstructor {
/** https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Uint8Array/fromBase64 */
fromBase64(string: string, options?: unknown): Uint8Array
}
}
export const binary: ScalarTag = {
identify: value => value instanceof Uint8Array, // Buffer inherits from Uint8Array
default: false,
tag: 'tag:yaml.org,2002:binary',
/**
* Returns a Buffer in Node.js and an Uint8Array elsewhere
*
* To use the resulting buffer as an image, you'll want to do something like:
*
* const blob = new Blob([buffer], { type: 'image/jpeg' })
* document.querySelector('#photo').src = URL.createObjectURL(/proxy?url=https%3A%2F%2Fgithub.com%2Fblob)
*/
resolve(src, onError) {
if (typeof Buffer === 'function') {
return Buffer.from(src, 'base64')
} else if (typeof Uint8Array.fromBase64 === 'function') {
return Uint8Array.fromBase64(src)
} else if (typeof atob === 'function') {
// On IE 11, atob() can't handle newlines
const str = atob(src.replace(/[\n\r]/g, ''))
const buffer = new Uint8Array(str.length)
for (let i = 0; i < str.length; ++i) buffer[i] = str.charCodeAt(i)
return buffer
} else {
onError(
'This environment does not support reading binary tags; either Buffer or atob is required'
)
return src
}
},
stringify({ comment, type, value }, ctx, onComment, onChompKeep) {
if (!value) return ''
const buf = value as Uint8Array // checked earlier by binary.identify()
let str: string
if (typeof buf.toBase64 === 'function') {
str = buf.toBase64()
} else if (typeof Buffer === 'function') {
str =
buf instanceof Buffer
? buf.toString('base64')
: Buffer.from(buf.buffer).toString('base64')
} else if (typeof btoa === 'function') {
let s = ''
for (let i = 0; i < buf.length; ++i) s += String.fromCharCode(buf[i])
str = btoa(s)
} else {
throw new Error(
'This environment does not support writing binary tags; either Buffer or btoa is required'
)
}
type ??= Scalar.BLOCK_LITERAL
if (type !== Scalar.QUOTE_DOUBLE) {
const lineWidth = Math.max(
ctx.options.lineWidth - ctx.indent.length,
ctx.options.minContentWidth
)
const n = Math.ceil(str.length / lineWidth)
const lines = new Array(n)
for (let i = 0, o = 0; i < n; ++i, o += lineWidth) {
lines[i] = str.substr(o, lineWidth)
}
str = lines.join(type === Scalar.BLOCK_LITERAL ? '\n' : ' ')
}
return stringifyString(
{ comment, type, value: str } as Scalar,
ctx,
onComment,
onChompKeep
)
}
}