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
91
92
93
94
95
96
97
98
|
// Copyright Node.js contributors. All rights reserved. MIT License.
import { existsSync } from "./_fs_exists.ts";
import { mkdir, mkdirSync } from "./_fs_mkdir.ts";
import {
ERR_INVALID_CALLBACK,
ERR_INVALID_OPT_VALUE_ENCODING,
} from "../_errors.ts";
export type mkdtempCallback = (
err: Error | undefined,
directory?: string,
) => void;
// https://nodejs.org/dist/latest-v15.x/docs/api/fs.html#fs_fs_mkdtemp_prefix_options_callback
export function mkdtemp(prefix: string, callback: mkdtempCallback): void;
export function mkdtemp(
prefix: string,
options: { encoding: string } | string,
callback: mkdtempCallback,
): void;
export function mkdtemp(
prefix: string,
optionsOrCallback: { encoding: string } | string | mkdtempCallback,
maybeCallback?: mkdtempCallback,
): void {
const callback: mkdtempCallback | undefined =
typeof optionsOrCallback == "function" ? optionsOrCallback : maybeCallback;
if (!callback) throw new ERR_INVALID_CALLBACK(callback);
const encoding: string | undefined = parseEncoding(optionsOrCallback);
const path = tempDirPath(prefix);
mkdir(
path,
{ recursive: false, mode: 0o700 },
(err: Error | null | undefined) => {
if (err) callback(err);
else callback(undefined, decode(path, encoding));
},
);
}
// https://nodejs.org/dist/latest-v15.x/docs/api/fs.html#fs_fs_mkdtempsync_prefix_options
export function mkdtempSync(
prefix: string,
options?: { encoding: string } | string,
): string {
const encoding: string | undefined = parseEncoding(options);
const path = tempDirPath(prefix);
mkdirSync(path, { recursive: false, mode: 0o700 });
return decode(path, encoding);
}
function parseEncoding(
optionsOrCallback?: { encoding: string } | string | mkdtempCallback,
): string | undefined {
let encoding: string | undefined;
if (typeof optionsOrCallback == "function") encoding = undefined;
else if (optionsOrCallback instanceof Object) {
encoding = optionsOrCallback?.encoding;
} else encoding = optionsOrCallback;
if (encoding) {
try {
new TextDecoder(encoding);
} catch (error) {
throw new ERR_INVALID_OPT_VALUE_ENCODING(encoding);
}
}
return encoding;
}
function decode(str: string, encoding?: string): string {
if (!encoding) return str;
else {
const decoder = new TextDecoder(encoding);
const encoder = new TextEncoder();
return decoder.decode(encoder.encode(str));
}
}
const CHARS = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
function randomName(): string {
return [...Array(6)].map(() =>
CHARS[Math.floor(Math.random() * CHARS.length)]
).join("");
}
function tempDirPath(prefix: string): string {
let path: string;
do {
path = prefix + randomName();
} while (existsSync(path));
return path;
}
|
