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
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2019 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
import Foundation
/// Common behavior used to generate source files.
protocol FileGenerator {
/// Types conforming to this protocol must implement this method to write their content into the
/// given file handle.
func write(into handle: FileHandle) throws
}
extension FileGenerator {
/// Generates a file at the given URL, overwriting it if it already exists.
func generateFile(at url: URL) throws {
let fm = FileManager.default
if fm.fileExists(atPath: url.path) {
try fm.removeItem(at: url)
}
fm.createFile(atPath: url.path, contents: nil, attributes: nil)
let handle = try FileHandle(forWritingTo: url)
defer { handle.closeFile() }
try write(into: handle)
}
}
extension FileHandle: TextOutputStream {
/// Writes the provided string as data to a file output stream.
public func write(_ string: String) {
guard let data = string.data(using: .utf8) else { return }
write(data)
}
}
|
