@@ -8,36 +8,76 @@
88
99import Foundation
1010
11−//let directory = "/Users/ilya/Developer/Dotrix/Dotrix/"
12−//let output = "/Users/ilya/Desktop/foo.tex"
13−
1411let args = Array(CommandLine.arguments.dropFirst())
15−let temp = FileManager.default.temporaryDirectory.appendingPathComponent("SourceDocsSwift/build\(Date().timeIntervalSince1970)")
16−
17−print(temp)
12+//let temp = FileManager.default.temporaryDirectory.appendingPathComponent("SourceDocsSwift/build\(Date().timeIntervalSince1970)", isDirectory: true)
1813
1914var output = "docs.tex"
2015var directory: String?
16+var style: TeXer.TexType = .laTeX(2)
2117
2218var i = 0
2319while i < args.count {
2420 switch args[i] {
25− case "-o":
21+ case "-o", "--output":
2622 if i + 1 < args.count {
2723 i += 1
2824 output = args[i]
2925 }
30− case "-i":
26+ case "-i", "--input":
3127 if i + 1 < args.count {
3228 i += 1
3329 directory = args[i]
3430 }
31+ case "-s", "--style":
32+ if i + 1 < args.count {
33+ i += 1
34+ switch args[i].lowercased() {
35+ case "latex", "xetex", "tex":
36+ style = .laTeX(style.levelOffset)
37+ case "tech", "techdoc", "techdocs":
38+ style = .techDoc(style.levelOffset)
39+ default:
40+ print("""
41+ That is not a valid style.
42+
43+ Valid styles are:
44+ latex
45+ techdoc
46+ """)
47+ exit(1)
48+ }
49+ }
50+ case "-l", "--level":
51+ if i + 1 < args.count {
52+ i += 1
53+ guard let l = Int(args[i]) else {
54+ print("""
55+ That is not a valid int.
56+ """)
57+ exit(1)
58+ }
59+ if l < 1 {
60+ print("""
61+ Level must be greater or equal to 1.
62+ """)
63+ exit(1)
64+ }
65+ style.levelOffset = l
66+ }
3567 default:
3668 break
3769 }
3870 i += 1
3971}
4072
73+var outputIsDirectory: ObjCBool = false
74+
75+FileManager.default.fileExists(atPath: output, isDirectory: &outputIsDirectory)
76+
77+if outputIsDirectory.boolValue {
78+ output.append("/docs.tex")
79+}
80+
4181let decoder = JSONDecoder()
4282
4383guard let xcbTagetData = Shell.launch(executable: "/usr/bin/xcodebuild", arguments: ["-list", "-json"], launchDirectory: directory) else {
@@ -59,8 +99,8 @@ let targets = xcbResponse.targets
5999var files: [String: [Object]] = [:]
60100
61101for target in targets {
62− try? FileManager.default.removeItem(at: temp)
63− guard let skResponseData = Shell.launch(executable: "/usr/local/bin/sourcekitten", arguments: ["doc", "--", "-target", target, "CONFIGURATION_BUILD_DIR=\(temp)"], launchDirectory: directory) else {
102+// try? FileManager.default.removeItem(at: temp)
103+ guard let skResponseData = Shell.launch(executable: "/usr/local/bin/sourcekitten", arguments: ["doc", "--", "-target", target], launchDirectory: directory) else {
64104 print("""
65105 Make sure sourcekitten is installed
66106
@@ -80,9 +120,15 @@ for target in targets {
80120
81121
82122
83−let tex = TeXer.convert(objects: files.flatMap {$0.value})
84−
85−try? tex.write(toFile: output, atomically: true, encoding: .utf8)
123+let tex = TeXer.convert(objects: files.flatMap {$0.value}, with: style)
124+do {
125+ try tex.write(toFile: output, atomically: true, encoding: .utf8)
126+} catch {
127+ print("""
128+ There was an error writing the outuput file.
129+ """)
130+ exit(1)
131+}
86132
87133print("""
88134+-------+