Add comments to script generated code

evan-liu · Nov 28, 2016 · 8d1dad7 · 8d1dad7
1 parent 5da9697
commit 8d1dad7
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 7 deletions.
diff --git a/Scripts/anchors.swift b/Scripts/anchors.swift
@@ -42,7 +42,7 @@ func main() {
 
     let pwdURL = URL(fileURLWithPath: fileManager.currentDirectoryPath)
     let scriptURL = URL(fileURLWithPath: #file, relativeTo: pwdURL)
-    let sourcesURL = scriptURL.deletingLastPathComponent().deletingLastPathComponent().appendingPathComponent("Sources")
+    let sourcesURL = scriptURL.appendingPathComponent("../../Sources")
     let codeFileURL = sourcesURL.appendingPathComponent("ConstraintMaker+Anchors.swift")
     let code = try! String(contentsOf: codeFileURL)
 
@@ -74,14 +74,15 @@ func main() {
         "centerXWithinMargins", 
         "centerYWithinMargins",
     ]
-    let equals = anchors
+    var equals = anchors
         .map { seedCode.replacingOccurrences(of: seedAnchor, with: $0) }
-        .joined(separator: "\n\n")
 
+    let equalsString = equals.joined(separator: "\n\n")
     let relations = ["greaterThanOrEqualTo", "lessThanOrEqualTo"]
-        .map { equals.replacingOccurrences(of: "equalTo", with: $0) }
+        .map { equalsString.replacingOccurrences(of: "equalTo", with: $0) }
 
-    let finalCode = header + ([equals] + relations).joined(separator: "\n\n")
+    equals.insert("//== ↓ Generated by `swift Scripts/anchors.swift` ↓ ==", at: 1)
+    let finalCode = header + (equals + relations).joined(separator: "\n\n")
 
     do {
         try finalCode.write(to: codeFileURL, atomically: true, encoding: .utf8)

diff --git a/Scripts/helpers.swift b/Scripts/helpers.swift
@@ -42,7 +42,7 @@ func main() {
 
     let pwdURL = URL(fileURLWithPath: fileManager.currentDirectoryPath)
     let scriptURL = URL(fileURLWithPath: #file, relativeTo: pwdURL)
-    let sourcesURL = scriptURL.deletingLastPathComponent().deletingLastPathComponent().appendingPathComponent("Sources")
+    let sourcesURL = scriptURL.appendingPathComponent("../../Sources")
     let codeFileURL = sourcesURL.appendingPathComponent("ConstraintMaker+Helpers.swift")
     let code = try! String(contentsOf: codeFileURL)
 
@@ -57,7 +57,7 @@ func main() {
     let relations = ["greaterThanOrEqualTo", "lessThanOrEqualTo"]
         .map { seedCode.replacingOccurrences(of: "equalTo", with: $0) }
 
-    let finalCode = header + ([seedCode] + relations).joined(separator: "\n\n")
+    let finalCode = header + ([seedCode, "//== ↓ Generated by `swift Scripts/helpers.swift` ↓ =="] + relations).joined(separator: "\n\n")
 
     do {
         try finalCode.write(to: codeFileURL, atomically: true, encoding: .utf8)

diff --git a/Sources/ConstraintMaker+Anchors.swift b/Sources/ConstraintMaker+Anchors.swift
@@ -52,6 +52,8 @@ extension ConstraintMaker {
 
 }
 
+//== ↓ Generated by `swift Scripts/anchors.swift` ↓ ==
+
 extension ConstraintMaker {
 
     /// Make item.right equalTo item2.attribute * multiplier + constant

diff --git a/Sources/ConstraintMaker+Helpers.swift b/Sources/ConstraintMaker+Helpers.swift
@@ -109,6 +109,8 @@ extension ConstraintMaker {
 
 }
 
+//== ↓ Generated by `swift Scripts/helpers.swift` ↓ ==
+
 extension ConstraintMaker {
 
     /// Make item.with greaterThanOrEqualTo one value