summaryrefslogtreecommitdiff
path: root/runtime/js/40_write_file.js
blob: a32ef441bc0a53d73040fcb3bc7b1cadf7e48c72 (plain)
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
99
100
101
102
103
104
105
106
107
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
"use strict";
((window) => {
  const core = window.__bootstrap.core;
  const ops = core.ops;
  const { abortSignal } = window.__bootstrap;
  const { pathFromURL } = window.__bootstrap.util;
  const { open } = window.__bootstrap.files;
  const { ReadableStreamPrototype } = window.__bootstrap.streams;
  const { ObjectPrototypeIsPrototypeOf } = window.__bootstrap.primordials;

  function writeFileSync(
    path,
    data,
    options = {},
  ) {
    options.signal?.throwIfAborted();
    ops.op_write_file_sync(
      pathFromURL(path),
      options.mode,
      options.append ?? false,
      options.create ?? true,
      options.createNew ?? false,
      data,
    );
  }

  async function writeFile(
    path,
    data,
    options = {},
  ) {
    let cancelRid;
    let abortHandler;
    if (options.signal) {
      options.signal.throwIfAborted();
      cancelRid = ops.op_cancel_handle();
      abortHandler = () => core.tryClose(cancelRid);
      options.signal[abortSignal.add](abortHandler);
    }
    try {
      if (ObjectPrototypeIsPrototypeOf(ReadableStreamPrototype, data)) {
        const file = await open(path, {
          mode: options.mode,
          append: options.append ?? false,
          create: options.create ?? true,
          createNew: options.createNew ?? false,
          write: true,
        });
        await data.pipeTo(file.writable, {
          signal: options.signal,
        });
      } else {
        await core.opAsync(
          "op_write_file_async",
          pathFromURL(path),
          options.mode,
          options.append ?? false,
          options.create ?? true,
          options.createNew ?? false,
          data,
          cancelRid,
        );
      }
    } finally {
      if (options.signal) {
        options.signal[abortSignal.remove](abortHandler);

        // always throw the abort error when aborted
        options.signal.throwIfAborted();
      }
    }
  }

  function writeTextFileSync(
    path,
    data,
    options = {},
  ) {
    const encoder = new TextEncoder();
    return writeFileSync(path, encoder.encode(data), options);
  }

  function writeTextFile(
    path,
    data,
    options = {},
  ) {
    if (ObjectPrototypeIsPrototypeOf(ReadableStreamPrototype, data)) {
      return writeFile(
        path,
        data.pipeThrough(new TextEncoderStream()),
        options,
      );
    } else {
      const encoder = new TextEncoder();
      return writeFile(path, encoder.encode(data), options);
    }
  }

  window.__bootstrap.writeFile = {
    writeTextFile,
    writeTextFileSync,
    writeFile,
    writeFileSync,
  };
})(this);