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
|
//
// TextFileOutputStream.swift
// MullvadVPN
//
// Created by pronebird on 02/08/2020.
// Copyright © 2020 Mullvad VPN AB. All rights reserved.
//
import Foundation
class TextFileOutputStream: TextOutputStream {
private let writer: DispatchIO
private let encoding: String.Encoding
private let queue = DispatchQueue.global(qos: .utility)
class func standardOutputStream(encoding: String.Encoding = .utf8) -> TextFileOutputStream {
return TextFileOutputStream(fileDescriptor: FileHandle.standardOutput.fileDescriptor, encoding: encoding)
}
init(fileDescriptor: Int32, encoding: String.Encoding = .utf8) {
self.encoding = encoding
writer = DispatchIO(type: .stream, fileDescriptor: fileDescriptor, queue: queue) { (errno) in
if errno != 0 {
print("TextFileOutputStream: closed channel with error: \(errno)")
}
}
}
init?(fileURL: URL, createFile: Bool, filePermissions: mode_t = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH, encoding: String.Encoding = .utf8) {
var oflag: Int32 = O_WRONLY
var mode: mode_t = .zero
if createFile {
oflag |= O_CREAT
mode = filePermissions
}
let queue = self.queue
let writer = fileURL.path.withCString { (filePathPointer) -> DispatchIO? in
return DispatchIO(type: .stream, path: filePathPointer, oflag: oflag, mode: mode, queue: queue, cleanupHandler: { (errno) in
if errno != 0 {
print("TextFileOutputStream: closed channel with error: \(errno)")
}
})
}
if let writer = writer {
self.writer = writer
self.encoding = encoding
} else {
return nil
}
}
deinit {
writer.close()
}
func write(_ string: String) {
string.data(using: encoding)?.withUnsafeBytes { (bytes) in
writer.write(offset: .zero, data: DispatchData(bytes: bytes), queue: queue) { (done, data, errno) in
if errno != 0 {
print("TextFileOutputStream: write error: \(errno)")
}
}
}
}
}
|
