Using contents from extension to create the playgrounds

Author: insidegui

PACore/Source/PlayAlways.swift

@@ -22,9 +22,11 @@ public struct PlayAlways {
     }
 
     public let platform: String
+    public let contents: String
 
-    public init(platform: String) {
+    public init(platform: String, contents: String = "") {
         self.platform = platform.lowercased()
+        self.contents = contents
     }
 
     var dateString: String {
@@ -46,7 +48,11 @@ public struct PlayAlways {
     }
 
     var importHeader: String {
-        return "//: Playground - noun: a place where people can play\n\nimport \(systemFramework)\n\nvar str = \"Hello, playground\""
+        return "//: Playground - noun: a place where people can play\n\nimport \(systemFramework)\n\n"
+    }
+    
+    var effectiveContents: String {
+        return importHeader + (contents.isEmpty ? "var str = \"Hello, playground\"" : contents)
     }
 
     var contentHeader: String {
@@ -98,7 +104,7 @@ public struct PlayAlways {
 
         if createPlaygroundFolder(playgroundDir.path) &&
             writeFile("contents.xcplayground", at: playgroundDir.path, content: contentHeader) &&
-            writeFile("Contents.swift", at: playgroundDir.path, content: importHeader) {
+            writeFile("Contents.swift", at: playgroundDir.path, content: effectiveContents) {
 
             return playgroundDir
         }

PlayAways/AppRouter.swift

@@ -20,7 +20,7 @@ final class AppRouter {
         self.persistence = persistence
     }
 
-    func createPlayground(with options: MenuOptions) {
+    func createPlayground(with options: MenuOptions, contents: String = "") {
         let platform: PlaygroundPlatform
         var showOptions = false
 
@@ -57,16 +57,16 @@ final class AppRouter {
 
             guard let url = panel.url else { return }
 
-            createPlayground(for: platform, at: url, usingDefaultName: false)
+            createPlayground(for: platform, at: url, usingDefaultName: false, contents: contents)
         } else {
             let url = storageLocationShowingPanelIfNeeded()
 
-            createPlayground(for: platform, at: url, usingDefaultName: true)
+            createPlayground(for: platform, at: url, usingDefaultName: true, contents: contents)
         }
 
     }
 
-    private func createPlayground(for platform: PlaygroundPlatform, at location: URL?, usingDefaultName defaultName: Bool) {
+    private func createPlayground(for platform: PlaygroundPlatform, at location: URL?, usingDefaultName defaultName: Bool, contents: String) {
         var playgroundUrl: URL?
 
         guard let location = location else {
@@ -85,7 +85,7 @@ final class AppRouter {
         }
 
         do {
-            playgroundUrl = try playgroundMaker.createPlayground(named: name, at: directory, platform: platform)
+            playgroundUrl = try playgroundMaker.createPlayground(named: name, at: directory, platform: platform, contents: contents)
         } catch {
             handle(error)
         }

PlayAways/ExtensionHandler.swift

@@ -29,7 +29,7 @@ final class ExtensionHandler {
                 options = .iOSWithPanel
             }
 
-            self?.router?.createPlayground(with: options)
+            self?.router?.createPlayground(with: options, contents: buffer)
         }
     }
 

PlayAways/PlayAlwaysMaker.swift

@@ -11,8 +11,8 @@ import PACore
 
 final class PlayAlwaysMaker: PlaygroundMaker {
 
-    func createPlayground(named name: String?, at destination: String, platform: PlaygroundPlatform) throws -> URL {
-        let maker = PlayAlways(platform: platform.rawValue)
+    func createPlayground(named name: String?, at destination: String, platform: PlaygroundPlatform, contents: String = "") throws -> URL {
+        let maker = PlayAlways(platform: platform.rawValue, contents: contents)
         return try maker.createPlayground(fileName: name, atDestination: destination)
     }
 

PlayAways/PlaygroundMaker.swift

@@ -15,5 +15,5 @@ public enum PlaygroundPlatform: String {
 }
 
 protocol PlaygroundMaker {
-    func createPlayground(named name: String?, at destination: String, platform: PlaygroundPlatform) throws -> URL
+    func createPlayground(named name: String?, at destination: String, platform: PlaygroundPlatform, contents: String) throws -> URL
 }