rootstrap/ios-base

import Foundation
 
let baseProjectName = "ios-base"
var projectName = "RSDemoProject"
let baseDomain = "com.rootstrap"
var bundleDomain = baseDomain
let baseCompany = "Rootstrap Inc."
var companyName = baseCompany
 
let whiteList: [String] = [".DS_Store", "UserInterfaceState.xcuserstate", "init.swift"]
let fileManager = FileManager.default
var currentFolder: String {
  return fileManager.currentDirectoryPath
}
 
enum SetupStep: Int {
  case nameEntry = 1
  case bundleDomainEntry
  case companyNameEntry
  
  var question: String {
    switch self {
    case .nameEntry: return "Enter a name for the project"
    case .bundleDomainEntry: return "Enter the reversed domain of your organization"
    case .companyNameEntry: return "Enter the Company name to use on file's headers"
    }
  }
}
 
// Helper methods
 
func prompt(message: String) -> String? {
  print("\n" + message)
  let answer = readLine()
Force unwrapping should be avoided.
  return answer == nil || answer == "" ? nil : answer!
}
 
func setup(step: SetupStep, defaultValue: String) -> String {
Line should be 90 characters or less: currently 108 characters
  let result = prompt(message: "\(step.rawValue). " + step.question + " (leave blank for \(defaultValue)).")
  guard let res = result else {
    print(defaultValue)
    return defaultValue
  }
  return res
}
 
func shell(_ args: String...) -> (output: String, exitCode: Int32) {
  let task = Process()
  task.launchPath = "/usr/bin/env"
  task.arguments = args
  task.currentDirectoryPath = currentFolder
  let pipe = Pipe()
  task.standardOutput = pipe
  task.launch()
  task.waitUntilExit()
  let data = pipe.fileHandleForReading.readDataToEndOfFile()
  let output = String(data: data, encoding: .utf8) ?? ""
  return (output, task.terminationStatus)
}
 
extension URL {
  var fileName: String {
    let urlValues = try? resourceValues(forKeys: [.nameKey])
    return urlValues?.name ?? ""
  }
  
  var isDirectory: Bool {
    let urlValues = try? resourceValues(forKeys: [.isDirectoryKey])
    return urlValues?.isDirectory ?? false
  }
  
  func rename(from oldName: String, to newName: String) {
    if fileName.contains(oldName) {
      let newName = fileName.replacingOccurrences(of: oldName, with: newName)
Line should be 90 characters or less: currently 117 characters
Force tries should be avoided.
      try! fileManager.moveItem(at: self, to: URL(fileURLWithPath: newName, relativeTo: deletingLastPathComponent()))
    }
  }
  
  func replaceOccurrences(of value: String, with newValue: String) {
    guard let fileContent = try? String(contentsOfFile: path, encoding: .utf8) else {
      print("Unable to read file at: \(self)")
      return
    }
    let updatedContent = fileContent.replacingOccurrences(of: value, with: newValue)
Force tries should be avoided.
    try! updatedContent.write(to: self, atomically: true, encoding: .utf8)
  }
  
  func setupForNewProject() {
    replaceOccurrences(of: baseProjectName, with: projectName)
    replaceOccurrences(of: baseDomain, with: bundleDomain)
    rename(from: baseProjectName, to: projectName)
  }
}
 
// Helper functions
 
func changeOrganizationName() {
  let pbxProjectPath = "\(currentFolder)/\(baseProjectName).xcodeproj/project.pbxproj"
  guard
    fileManager.fileExists(atPath: pbxProjectPath),
    companyName != baseCompany
  else { return }
 
  print("\nUpdating company name to '\(companyName)'...")
  
  let filterKey = "ORGANIZATIONNAME"
  let organizationNameFilter = "\(filterKey) = \"\(baseCompany)\""
  let organizationNameReplacement = "\(filterKey) = \"\(companyName)\""
  let fileUrl = URL(fileURLWithPath: pbxProjectPath)
  fileUrl.replaceOccurrences(
    of: organizationNameFilter,
    with: organizationNameReplacement
  )
}
 
// Project Initialization
 
print("""
+-----------------------------------------+
|                                         |
|       < New iOS Project Setup >         |
|                                         |
+-----------------------------------------+
""")
 
projectName = setup(step: .nameEntry, defaultValue: projectName)
bundleDomain = setup(step: .bundleDomainEntry, defaultValue: baseDomain)
companyName = setup(step: .companyNameEntry, defaultValue: baseCompany)
 
//Remove current git tracking
_ = shell("rm", "-rf", ".git")
 
changeOrganizationName()
 
print("\nRenaming to '\(projectName)'...")
Line should be 90 characters or less: currently 138 characters
Force unwrapping should be avoided.
let enumerator = fileManager.enumerator(at: URL(fileURLWithPath: currentFolder), includingPropertiesForKeys: [.nameKey, .isDirectoryKey])!
var directories: [URL] = []
while let itemURL = enumerator.nextObject() as? URL {
  guard !whiteList.contains(itemURL.fileName) else { continue }
  if itemURL.isDirectory {
    directories.append(itemURL)
  } else {
    itemURL.setupForNewProject()
  }
}
 
for dir in directories.reversed() {
  dir.rename(from: baseProjectName, to: projectName)
}
//TODO: Rename current dir
let currentURL = URL(fileURLWithPath: currentFolder)
currentURL.rename(from: baseProjectName, to: projectName)
 
print("Opening new project...")
_ = shell("open", "\(projectName).xcworkspace")
// Initialization Done!
print("************** ALL SET! *******************")