Merge pull request #7 from adjust/development

Added build scripts for iOS and Android

Author: nonelse

Adjust.unitypackage

@@ -0,0 +1,96 @@
+using UnityEngine;
+using System.Collections;
+using UnityEditor.Callbacks;
+using UnityEditor;
+using System.Diagnostics;
+
+public class AdjustEditor : MonoBehaviour {
+
+	[PostProcessBuild]
+	public static void OnPostprocessBuild(BuildTarget target, string pathToBuiltProject) {
+		var exitCode = RunPostBuildScript (preBuild: false, pathToBuiltProject: pathToBuiltProject);
+
+		if (exitCode == -1) {
+			return;
+		}
+		if (exitCode != 0) {
+			var errorMessage = GenerateErrorScriptMessage(exitCode);
+
+			UnityEngine.Debug.LogError ("adjust: " + errorMessage);
+		}
+	}
+
+	[MenuItem("Adjust/Fix AndroidManifest.xml")]
+	static void FixAndroidManifest() {
+		#if UNITY_ANDROID
+		var exitCode = RunPostBuildScript (preBuild: true);
+		if (exitCode == 1) {
+			EditorUtility.DisplayDialog("Adjust", 
+			                            string.Format("AndroidManifest.xml changed or created at {0}/Plugins/Android/ .",Application.dataPath),
+			                            "OK");
+		} else if (exitCode == 0) {
+			EditorUtility.DisplayDialog("Adjust", 
+			                            "AndroidManifest.xml did not needed to be changed.",
+			                            "OK");
+		} else {
+			EditorUtility.DisplayDialog("Adjust", 
+			                            GenerateErrorScriptMessage(exitCode),
+			                            "OK");
+			
+		}
+		#else
+		EditorUtility.DisplayDialog("Adjust", "Option only valid for the Android platform.", "OK");
+		#endif
+	}
+
+	static int RunPostBuildScript (bool preBuild, string pathToBuiltProject = "") {
+		string pathToScript = null;
+		string arguments = null;
+
+		#if UNITY_ANDROID
+		pathToScript = "/Editor/AdjustPostBuildAndroid";
+		arguments = Application.dataPath;
+		if (preBuild)
+			arguments = "--pre-build " + arguments;
+		#elif UNITY_IOS
+		pathToScript = "/Editor/AdjustPostBuildiOS";
+		arguments = pathToBuiltProject;
+		#else
+		return -1;
+		#endif
+
+		Process proc = new Process();
+		proc.EnableRaisingEvents=false; 
+		proc.StartInfo.FileName = Application.dataPath + pathToScript;
+		proc.StartInfo.Arguments = arguments;
+		proc.Start();
+		proc.WaitForExit();
+		return proc.ExitCode;
+	}
+
+	static string GenerateErrorScriptMessage(int exitCode) {
+		#if UNITY_ANDROID
+		if (exitCode == 1) {
+			return "The AndroidManifest.xml file was only changed or created after building the package. " +
+          		"PLease build again the Android Unity package so it can use the new file";
+		}  
+		#endif
+
+		if (exitCode != 0) {
+			var message = "Build script exited with error." +
+				" Please check the Adjust log file for more information at {0}";
+			string projectPath = Application.dataPath.Substring (0, Application.dataPath.Length - 7);
+			string logFile = null;
+			#if UNITY_ANDROID
+			logFile = projectPath + "/AdjustPostBuildAndroidLog.txt";
+			#elif UNITY_IOS
+			logFile = projectPath + "/AdjustPostBuiliOSLog.txt";
+			#else
+			return null;
+			#endif
+			return string.Format(message, logFile);
+		} 
+
+		return null;
+	}
+}

Assets/Editor.meta

@@ -0,0 +1,162 @@
+#!/usr/bin/env python
+
+import argparse
+import errno
+import shutil
+import os.path
+import sys
+from xml.dom.minidom import parse, parseString
+
+def main():
+    parser = argparse.ArgumentParser(
+            description="Adjust post build android script")
+    parser.add_argument('assets_path',
+            help="path to the assets folder of unity3d")
+    # pre build must be set manually
+    parser.add_argument('--pre-build', action="store_true",
+            help="used to check and change the AndroidManifest.xml to conform to the Adjust SDK")
+    exit_code = 0
+    with open('AdjustPostBuildAndroidLog.txt','w') as fileLog:
+        # log function with file injected
+        LogFunc = LogInput(fileLog)
+        
+        # get the path of the android plugin folder
+        android_plugin_path, pre_build = parse_input(LogFunc, parser)
+
+        # try to open an existing manifest file
+        try:
+            manifest_path = os.path.join(android_plugin_path + "AndroidManifest.xml")
+            edited_xml = None
+            with open(manifest_path,'r+') as mf:
+                check_dic = check_manifest(LogFunc, mf)
+                # check if manifest has all changes needed
+                all_check = check_dic["has_adjust_receiver"] and \
+                            check_dic["has_internet_permission"] and \
+                            check_dic["has_wifi_permission"]
+                # edit manifest if has any change missing
+                if not all_check:
+                    # warn unity if it was pos-build, if something is missing
+                    if not pre_build:
+                        LogFunc("Android manifest used in unity did not " + \
+                                "had all the changes adjust SDK needs. " + \
+                                "Please build again the package.")
+                    edited_xml = edit_manifest(LogFunc, mf, check_dic)
+            # write changed xml
+            if edited_xml:
+                with open(manifest_path,'w+') as mf:
+                    edited_xml.writexml(mf)
+                exit_code = 1
+        except IOError as ioe:
+            # if it does not exist 
+            if ioe.errno == errno.ENOENT:
+                # warn unity that needed manifest wasn't used
+                if not pre_build:
+                    LogFunc("Used default Android manifest file from " + \
+                            "unity. Please build again the package to " +
+                            "include the changes for adjust SDK")
+                copy_adjust_manifest(LogFunc, android_plugin_path)
+                exit_code = 1
+            else:
+                LogFunc(ioe)
+        except Exception as e:
+            LogFunc(e)
+
+    # exit with return code for unity
+    sys.exit(exit_code)
+
+def edit_manifest(Log, manifest_file, check_dic):
+    manifest_xml = check_dic["manifest_xml"]
+
+    # add the adjust install referrer to the application element
+    if not check_dic["has_adjust_receiver"]:
+        receiver_string = """<?xml version="1.0" ?>
+        <receiver
+            xmlns:android="http://schemas.android.com/apk/res/android"
+            android:name="com.adjust.sdk.ReferrerReceiver"
+            android:exported="true" >
+            <intent-filter>
+                <action android:name="com.android.vending.INSTALL_REFERRER" />
+            </intent-filter>
+        </receiver>
+        """
+        receiver_xml = parseString(receiver_string)
+        receiver_xml.documentElement.removeAttribute("xmlns:android")
+
+        for app_element in manifest_xml.getElementsByTagName("application"):
+            app_element.appendChild(receiver_xml.documentElement)
+        Log("added adjust install referrer receiver")
+
+    # add the internet permission to the manifest element
+    if not check_dic["has_internet_permission"]:
+        ip_element = manifest_xml.createElement("uses-permission")
+        ip_element.setAttribute("android:name", "android.permission.INTERNET")
+        manifest_xml.documentElement.appendChild(ip_element)
+        Log("added internet permission")
+
+    # add the access wifi state permission to the manifest element
+    if not check_dic["has_wifi_permission"]:
+        ip_element = manifest_xml.createElement("uses-permission")
+        ip_element.setAttribute("android:name", "android.permission.ACCESS_WIFI_STATE")
+        manifest_xml.documentElement.appendChild(ip_element)
+        Log("added access wifi permission")
+
+    #Log(manifest_xml.toxml())
+    return manifest_xml
+
+def check_manifest(Log, manifest_file):
+    
+    manifest_xml = parse(manifest_file)
+    #Log(manifest_xml.toxml())
+    
+    has_adjust_receiver = has_element_attr(manifest_xml,
+            "receiver", "android:name", "com.adjust.sdk.ReferrerReceiver")
+    Log("has adjust install referrer receiver?: {0}", has_adjust_receiver)
+
+    has_internet_permission = has_element_attr(manifest_xml,
+            "uses-permission", "android:name", "android.permission.INTERNET")
+    Log("has internet permission?: {0}", has_internet_permission)
+    
+    has_wifi_permission = has_element_attr(manifest_xml,
+            "uses-permission", "android:name", "android.permission.ACCESS_WIFI_STATE")
+    Log("has wifi permission?: {0}", has_wifi_permission)
+
+    return {"manifest_xml" : manifest_xml,
+            "has_adjust_receiver" : has_adjust_receiver,
+            "has_internet_permission" : has_internet_permission,
+            "has_wifi_permission" : has_wifi_permission}
+
+def has_element_attr(xml_dom, tag_name, attr_name, attr_value):
+    for node in xml_dom.getElementsByTagName(tag_name):
+        attr_dom = node.getAttribute(attr_name)
+        if attr_dom == attr_value:
+            return True
+    return False
+
+def copy_adjust_manifest(Log, android_plugin_path):
+    adjust_manifest_path = os.path.join(android_plugin_path, "AdjustAndroidManifest.xml")
+    new_manifest_path = os.path.join(android_plugin_path, "AndroidManifest.xml")
+    try:
+        shutil.copyfile(adjust_manifest_path, new_manifest_path)
+    except Exception as e:
+        Log(e)
+    else:
+        Log("Manifest copied from {0} to {1}", 
+                adjust_manifest_path, new_manifest_path)
+
+def LogInput(writeObject):
+    def Log(message, *args):
+        messageNLine = str(message) + "\n"
+        writeObject.write(messageNLine.format(*args))
+    return Log
+
+def parse_input(Log, parser):
+    args = parser.parse_args()
+    assets_path = args.assets_path
+
+    android_plugin_path = os.path.join(assets_path, "Plugins/Android/")
+    Log("Android plugin path: {0}", android_plugin_path)
+
+    return android_plugin_path, args.pre_build
+
+if __name__ == "__main__":
+    main()

Assets/Editor/AdjustEditor.cs

@@ -0,0 +1,117 @@
+#!/usr/bin/env python
+
+import sys
+import re
+from subprocess import Popen, PIPE
+import argparse
+
+from mod_pbxproj import XcodeProject
+
+def main():
+    parser = argparse.ArgumentParser(description="Adjust post build iOS script")
+    parser.add_argument('ios_project_path', help="path to the folder of the iOS project generated by unity3d")
+    
+    with open('AdjustPostBuildiOSLog.txt','w') as fileLog:
+        # log function with file injected
+        LogFunc = LogInput(fileLog)
+       
+        # path of the Xcode SDK on the system
+        xcode_sdk_path = get_xcode_sdk_path(LogFunc)
+
+        # path for unity iOS Xcode project and adFramework on the system
+        unity_xcode_project_path, adFramework_path = get_paths(LogFunc, parser, xcode_sdk_path)
+
+        # edit the Xcode project using mod_pbxproj
+        #  add the adFramework library
+        #  change the compilation flags of the adjust project files to support non-ARC
+        edit_unity_xcode_project(LogFunc, unity_xcode_project_path, adFramework_path)
+
+        # change the Xcode project directly
+        #  allow objective-c exceptions
+        rewrite_unity_xcode_project(LogFunc, unity_xcode_project_path)
+    sys.exit(0)
+
+def LogInput(writeObject):
+    def Log(message, *args):
+        messageNLine = (message if message else "None") + "\n"
+        writeObject.write(messageNLine.format(*args))
+    return Log
+
+def get_paths(Log, parser, xcode_sdk_path):
+    args = parser.parse_args()
+    ios_project_path = args.ios_project_path
+
+    unity_xcode_project_path = ios_project_path + "/Unity-iPhone.xcodeproj/project.pbxproj"
+    Log("Unity3d Xcode project path: {0}", unity_xcode_project_path)
+    
+    adFramework_path = xcode_sdk_path + "/System/Library/Frameworks/AdSupport.framework"
+    Log("adFramework path: {0}", adFramework_path)
+
+    return unity_xcode_project_path, adFramework_path
+
+def edit_unity_xcode_project(Log, unity_xcode_project_path, adFramework_path):
+    # load unity iOS pbxproj project file
+    unity_XcodeProject = XcodeProject.Load(unity_xcode_project_path)
+    
+    # add adFramework to unity
+    unity_XcodeProject.add_file_if_doesnt_exist(adFramework_path, tree="SDKROOT", create_build_files=True,weak=True)
+    Log("added adSupport framework")
+
+    # regex for adjust sdk files
+    re_adjust_files = re.compile(r"AI.*\.m|.*\+AI.*\.m|Adjust\.m|AdjustUnity\.mm")
+
+    # iterate all objects in the unity Xcode iOS project file
+    for key in unity_XcodeProject.get_ids():
+        obj = unity_XcodeProject.get_obj(key)
+        
+        name = obj.get('name')
+        isa = obj.get('isa')
+        path = obj.get('path')
+        fileref = obj.get('fileRef')
+
+        #Log("key: {0}, name: {1}, isa: {2}, path: {3}, fileref: {4}", key, name, isa, path, fileref)
+
+        #check if file reference match any adjust file
+        adjust_file_match = re_adjust_files.match(name if name else "")
+        if (adjust_file_match):
+            #Log("file match, group: {0}", adjust_file_match.group())
+            # get the build file, from the file reference id
+            build_files = unity_XcodeProject.get_build_files(key)
+            for build_file in build_files:
+                # add the ARC compiler flag to the adjust file if doesn't exist
+                build_file.add_compiler_flag('-fobjc-arc')
+                Log("added ARC flag to file {0}", name)
+
+    # save changes
+    unity_XcodeProject.saveFormat3_2()
+
+def rewrite_unity_xcode_project(Log, unity_xcode_project_path):
+    unity_xcode_lines = []
+    # allow objective-c exceptions
+    re_objc_excep = re.compile(r"\s*GCC_ENABLE_OBJC_EXCEPTIONS *= *NO.*")
+    with open(unity_xcode_project_path) as upf:
+        for line in upf:
+            if re_objc_excep.match(line):
+                #Log("matched line: {0}", re_objc_excep.match(line).group())
+                line = line.replace("NO","YES")
+                Log("Objective-c exceptions enabled")
+            unity_xcode_lines.append(line)
+    with open(unity_xcode_project_path, "w+") as upf:
+        upf.writelines(unity_xcode_lines)
+
+def get_xcode_sdk_path(Log):
+    # outputs all info from xcode
+    proc = Popen(["xcodebuild", "-version", "-sdk"], stdout=PIPE, stderr=PIPE)
+    out, err = proc.communicate()
+    
+    if proc.returncode not in [0, 66]:
+        Log("Could not retrieve Xcode sdk path. code: {0}, err: {1}", proc.returncode, err)
+        return None
+
+    match = re.search("iPhoneOS.*?Path: (?P<sdk_path>.*?)\n", out, re.DOTALL)
+    xcode_sdk_path = match.group('sdk_path') if match else None
+    Log("Xcode sdk path: {0}", xcode_sdk_path)
+    return xcode_sdk_path
+
+if __name__ == "__main__":
+    main()