Add measure-ios command

Author: Kudo

.gitignore

@@ -3,3 +3,4 @@ node_modules
 .vscode
 build/
 *.tsbuildinfo
+yarn-error.log

babel.config.js

@@ -0,0 +1,6 @@
+module.exports = {
+  presets: [
+    ['@babel/preset-env', {targets: {node: '8'}}],
+    '@babel/preset-typescript',
+  ],
+};

definitions/xcode/index.d.ts

@@ -0,0 +1,5 @@
+declare module 'xcode' {
+  const module: any;
+
+  export default module;
+}

definitions/xcode/lib/index.d.ts

@@ -0,0 +1,5 @@
+declare module 'xcode/lib/pbxFile' {
+  const module: any;
+
+  export default module;
+}

package.json

@@ -13,14 +13,18 @@
   "repository": "https://github.com/Kudo/react-native-cli-plugin-benchmark",
   "license": "MIT",
   "devDependencies": {
+    "@babel/preset-env": "^7.8.4",
+    "@babel/preset-typescript": "^7.8.3",
     "@react-native-community/cli-platform-android": "^4.0.1",
     "@react-native-community/cli-platform-ios": "^4.0.1",
     "@react-native-community/cli-types": "^3.0.0",
     "@types/glob": "^7.1.1",
     "@types/graceful-fs": "^4.1.3",
+    "@types/jest": "^25.1.3",
     "@types/lodash": "^4.14.149",
     "@types/node": "^13.7.1",
     "@types/node-fetch": "^2.5.4",
+    "jest": "^25.1.0",
     "prettier": "^1.19.1",
     "typescript": "^3.7.5"
   },
@@ -31,6 +35,7 @@
     "execa": "^4.0.0",
     "glob": "^7.1.6",
     "graceful-fs": "^4.2.3",
-    "lodash": "^4.17.15"
+    "lodash": "^4.17.15",
+    "xcode": "^2.1.0"
   }
 }

src/commands/common/ObjcPatcher.ts

@@ -0,0 +1,54 @@
+/**
+ * @format
+ */
+
+import fs from 'fs';
+
+export default class ObjcPatcher {
+  _content: string;
+  _patchSig: string;
+
+  constructor(fromFile: string, patchSignature: string) {
+    this._content = fs.readFileSync(fromFile, {encoding: 'utf8'});
+    this._patchSig = `/* Patched by ObjcPatcher: ${patchSignature} */\n`;
+  }
+
+  isPatched(): boolean {
+    return this._content.indexOf(this._patchSig) >= 0;
+  }
+
+  addImport(file: string) {
+    const lastImportBegin = this._content.lastIndexOf('\n#import');
+    let lastImportEnd = this._content.indexOf('\n', lastImportBegin + 1);
+    const headPart = this._content.substring(0, lastImportEnd);
+    const tailPart = this._content.substring(lastImportEnd);
+    this._content = headPart + `\n#import ${file}` + tailPart;
+    return this;
+  }
+
+  addFunction(code: string) {
+    const lastImplEnd = this._content.lastIndexOf('\n@end');
+    const headPart = this._content.substring(0, lastImplEnd);
+    const tailPart = this._content.substring(lastImplEnd);
+    this._content = headPart + `\n${code}` + tailPart;
+    return this;
+  }
+
+  replace(searchValue: string | RegExp, replaceValue: string) {
+    this._content = this._content.replace(searchValue, replaceValue);
+    return this;
+  }
+
+  write(toFile: string): void {
+    this._addPatchSigIfNeeded();
+    fs.writeFileSync(toFile, this._content);
+  }
+
+  _addPatchSigIfNeeded() {
+    if (this.isPatched()) {
+      return this;
+    }
+    this._content = this._patchSig + this._content;
+    return this;
+  }
+}

src/commands/common/__tests__/objcPatcher.test.ts

@@ -0,0 +1,171 @@
+/**
+ * @format
+ */
+
+import ObjcPatcher from '../ObjcPatcher';
+import fs from 'fs';
+
+jest.mock('fs');
+
+describe('ObjcPatcher', () => {
+  test('addImport() will append imports', () => {
+    const beforePatch = `\
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  return NO;
+}
+`;
+
+    const afterPatch = `\
+/* Patched by ObjcPatcher: foo */
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+#import <React/Foo.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  return NO;
+}
+`;
+    fs.readFileSync = jest.fn().mockReturnValueOnce(beforePatch);
+
+    fs.writeFileSync = jest.fn();
+
+    const patcher = new ObjcPatcher('/foo', 'foo');
+    patcher.addImport('<React/Foo.h>').write('/bar');
+    expect(fs.writeFileSync.mock.calls[0][0]).toBe('/bar');
+    expect(fs.writeFileSync.mock.calls[0][1]).toBe(afterPatch);
+  });
+
+  test('isPatched() returns false if the file has not been patched', () => {
+    const beforePatch = `\
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  return NO;
+}
+`;
+
+    fs.readFileSync = jest.fn().mockReturnValueOnce(beforePatch);
+    const patcher = new ObjcPatcher('/foo', 'foo');
+    expect(patcher.isPatched()).toBe(false);
+  });
+
+  test('isPatched() returns true if the file has been patched', () => {
+    const afterPatch = `\
+/* Patched by ObjcPatcher: foo */
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  return NO;
+}
+`;
+
+    fs.readFileSync = jest.fn().mockReturnValueOnce(afterPatch);
+    const patcher = new ObjcPatcher('/foo', 'foo');
+    expect(patcher.isPatched()).toBe(true);
+  });
+
+  test('addFunction() will add code block before last @end', () => {
+    const beforePatch = `\
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  self.window.rootViewController = rootViewController;
+  return NO;
+}
+@end
+`;
+
+    const afterPatch = `\
+/* Patched by ObjcPatcher: foo */
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  self.window.rootViewController = rootViewController;
+  return NO;
+}
+
+- (void)foo
+{
+}
+@end
+`;
+
+    fs.readFileSync = jest.fn().mockReturnValueOnce(beforePatch);
+
+    fs.writeFileSync = jest.fn();
+
+    const patcher = new ObjcPatcher('/foo', 'foo');
+    const addCode = `
+- (void)foo
+{
+}`;
+    patcher.addFunction(addCode).write('/bar');
+    expect(fs.writeFileSync.mock.calls[0][1]).toBe(afterPatch);
+  });
+
+  test('replace() is simply passthrough String.replace', () => {
+    const beforePatch = `\
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  self.window.rootViewController = rootViewController;
+  return NO;
+}
+`;
+
+    const afterPatch = `\
+/* Patched by ObjcPatcher: foo */
+#import <UIKit/UIKit.h>
+#import <React/RCTRootView.h>
+
+@implementation AppDelegate
+- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
+{
+  self.window.rootViewController = rootViewController;
+  dispatch_async(dispatch_get_main_queue(), ^{
+      NSLog(@"AsyncTask");
+  });
+  return NO;
+}
+`;
+
+    fs.readFileSync = jest.fn().mockReturnValueOnce(beforePatch);
+
+    fs.writeFileSync = jest.fn();
+
+    const patcher = new ObjcPatcher('/foo', 'foo');
+    const patchCode = `\
+  dispatch_async(dispatch_get_main_queue(), ^{
+      NSLog(@"AsyncTask");
+  });`;
+    patcher
+      .replace(
+        /(^\s*self.window.rootViewController = rootViewController;\s*$)/m,
+        `$1\n${patchCode}`,
+      )
+      .write('/bar');
+    expect(fs.writeFileSync.mock.calls[0][1]).toBe(afterPatch);
+  });
+});