Add AppCode support (fixes #147, #49) (#204)

InjectionBundle/InjectionClient.h

@@ -23,6 +23,7 @@ typedef NS_ENUM(int, InjectionCommand) {
  InjectionSigned,
  InjectionLoad,
  InjectionInject,
+ InjectionIdeProcPath,
  InjectionXprobe,
  InjectionEval,
  InjectionVaccineSettingChanged,

InjectionBundle/InjectionClient.mm

@@ -181,6 +181,10 @@ - (void)runInBackground {
  case InjectionUntrace:
  [SwiftTrace removeAllPatches];
  break;
+ case InjectionIdeProcPath: {
+ [SwiftEval sharedInstance].lastIdeProcPath = [self readString];
+ break;
+ }
  default: {
  NSString *changed = [self readString];
  dispatch_async(dispatch_get_main_queue(), ^{

InjectionBundle/SwiftEval.swift

@@ -164,6 +164,8 @@ public class SwiftEval: NSObject {
  }
 
  @objc public var injectionNumber = 0
+ @objc public var lastIdeProcPath = ""
+
  static var compileByClass = [String: (String, String)]()
 
  static var buildCacheFile = "/tmp/eval_builds.plist"
@@ -173,10 +175,10 @@ public class SwiftEval: NSObject {
  // Largely obsolete section used find Xcode paths from source file being injected.
 
  let sourceURL = URL(fileURLWithPath: classNameOrFile.hasPrefix("/") ? classNameOrFile : #file)
- guard let derivedData = findDerivedData(url: URL(fileURLWithPath: NSHomeDirectory())) ??
+ guard let derivedData = findDerivedData(url: URL(fileURLWithPath: NSHomeDirectory()), ideProcPath: self.lastIdeProcPath) ??
  (self.projectFile != nil ?
- findDerivedData(url: URL(fileURLWithPath: self.projectFile!)) :
- findDerivedData(url: sourceURL)) else {
+ findDerivedData(url: URL(fileURLWithPath: self.projectFile!), ideProcPath: self.lastIdeProcPath) :
+ findDerivedData(url: sourceURL, ideProcPath: self.lastIdeProcPath)) else {
  throw evalError("Could not locate derived data. Is the project under your home directory?")
  }
  guard let (projectFile, logsDir) =
@@ -549,20 +551,53 @@ public class SwiftEval: NSObject {
  return (compileCommand, sourceFile.replacingOccurrences(of: "\\$", with: "$"))
  }
 
- func findDerivedData(url: URL) -> URL? {
+ func getAppCodeDerivedData(procPath: String) -> String {
+ //Default with current year
+ let derivedDataPath = { (year: Int, pathSelector: String) -> String in
+ "Library/Caches/\(year > 2019 ? "JetBrains/" : "")\(pathSelector)/DerivedData"
+ }
+
+ let year = Calendar.current.component(.year, from: Date())
+ let month = Calendar.current.component(.month, from: Date())
+
+ let defaultPath = derivedDataPath(year, "AppCode\(month / 4 == 0 ? year - 1 : year).\(month / 4 + (month / 4 == 0 ? 3 : 0))")
+
+ var plistPath = URL(fileURLWithPath: procPath)
+ plistPath.deleteLastPathComponent()
+ plistPath.deleteLastPathComponent()
+ plistPath = plistPath.appendingPathComponent("Info.plist")
+
+ guard let dictionary = NSDictionary(contentsOf: plistPath) as? Dictionary<String, Any> else { return defaultPath }
+ guard let jvmOptions = dictionary["JVMOptions"] as? Dictionary<String, Any> else { return defaultPath }
+ guard let properties = jvmOptions["Properties"] as? Dictionary<String, Any> else { return defaultPath }
+ guard let pathSelector: String = properties["idea.paths.selector"] as? String else { return defaultPath }
+
+ let components = pathSelector.replacingOccurrences(of: "AppCode", with: "").components(separatedBy: ".")
+ guard components.count == 2 else { return defaultPath }
+
+ guard let realYear = Int(components[0]) else { return defaultPath }
+ return derivedDataPath(realYear, pathSelector)
+ }
+
+ func findDerivedData(url: URL, ideProcPath: String) -> URL? {
  if url.path == "/" {
  return nil
  }
 
- for relative in ["DerivedData", "build/DerivedData",
- "Library/Developer/Xcode/DerivedData"] {
+ var relativeDirs = ["DerivedData", "build/DerivedData"]
+ if ideProcPath.lowercased().contains("appcode") {
+ relativeDirs.append(getAppCodeDerivedData(procPath: ideProcPath))
+ } else {
+ relativeDirs.append("Library/Developer/Xcode/DerivedData")
+ }
+ for relative in relativeDirs {
  let derived = url.appendingPathComponent(relative)
  if FileManager.default.fileExists(atPath: derived.path) {
  return derived
  }
  }
 
- return findDerivedData(url: url.deletingLastPathComponent())
+ return findDerivedData(url: url.deletingLastPathComponent(), ideProcPath: ideProcPath)
  }
 
  func findProject(for source: URL, derivedData: URL) -> (projectFile: URL, logsDir: URL)? {

InjectionIII/FileWatcher.swift

@@ -10,7 +10,7 @@ import Foundation
 
 let INJECTABLE_PATTERN = "[^~]\\.(mm?|swift|storyboard|xib)$"
 
-public typealias InjectionCallback = (_ filesChanged: NSArray) -> Void
+public typealias InjectionCallback = (_ filesChanged: NSArray, _ ideProcPath: String) -> Void
 
 public class FileWatcher: NSObject {
  var fileEvents: FSEventStreamRef! = nil
@@ -66,7 +66,11 @@ public class FileWatcher: NSObject {
  }
 
  if changed.count != 0 {
- callback(Array(changed) as NSArray)
+ var path = ""
+ if let application = NSWorkspace.shared.frontmostApplication {
+ path = getProcPath(pid: application.processIdentifier)
+ }
+ callback(Array(changed) as NSArray, path)
  }
  }
 
@@ -75,4 +79,14 @@ public class FileWatcher: NSObject {
  FSEventStreamInvalidate(fileEvents)
  FSEventStreamRelease(fileEvents)
  }
+
+ func getProcPath(pid: pid_t) -> String {
+ let pathBuffer = UnsafeMutablePointer<UInt8>.allocate(capacity: Int(MAXPATHLEN))
+ defer {
+ pathBuffer.deallocate()
+ }
+ proc_pidpath(pid, pathBuffer, UInt32(MAXPATHLEN))
+ let path = String(cString: pathBuffer)
+ return path
+ }
 }

InjectionIII/InjectionIII-Bridging-Header.h

@@ -13,6 +13,7 @@
 #import "InjectionClient.h"
 
 #import "DDHotKeyCenter.h"
+#import <libproc.h>
 
 #ifdef XPROBE_PORT
 #import "../XprobePlugin/Classes/XprobePluginMenuController.h"

InjectionIII/InjectionServer.swift

@@ -15,10 +15,10 @@ var projectInjected = ["": ["": Date.timeIntervalSinceReferenceDate]]
 let MIN_INJECTION_INTERVAL = 1.0
 
 public class InjectionServer: SimpleSocket {
- var injector: ((_ changed: NSArray) -> Void)? = nil
+ var injector: ((_ changed: NSArray, _ ideProcPath:String) -> Void)? = nil
  var fileWatchers = [FileWatcher]()
  var pending = [String]()
-
+ var lastIdeProcPath = ""
  override public class func error(_ message: String) -> Int32 {
  let saveno = errno
  DispatchQueue.main.sync {
@@ -162,7 +162,7 @@ public class InjectionServer: SimpleSocket {
  var testCache = [String: [String]]()
 
  injector = {
- (changed: NSArray) in
+ (changed: NSArray, ideProcPath: String) in
  var changed = changed as! [String]
 
  if UserDefaults.standard.bool(forKey: UserDefaultsTDDEnabled) {
@@ -195,7 +195,7 @@ public class InjectionServer: SimpleSocket {
  }
  }
  }
-
+ self.lastIdeProcPath = ideProcPath
  if (automatic) {
  self.injectPending()
  }
@@ -259,6 +259,7 @@ public class InjectionServer: SimpleSocket {
  @objc public func injectPending() {
  for swiftSource in pending {
  injectionQueue.async {
+ self.sendCommand(.ideProcPath, with: self.lastIdeProcPath)
  self.sendCommand(.inject, with:swiftSource)
  }
  }