cocoapods-service.ts

import { EOL } from "os";
import * as path from "path";
import * as _ from "lodash";
import { PlatformTypes, PODFILE_NAME, NS_BASE_PODFILE } from "../constants";
import { regExpEscape, getHash } from "../common/helpers";
import { IPluginData } from "../definitions/plugins";
import {
	IRubyFunction,
	IProjectData,
	ICocoaPodsPlatformManager,
	ICocoaPodsService,
	IPodfilePlatformData,
} from "../definitions/project";
import { IPlatformData } from "../definitions/platform";
import { IConfiguration, IXcconfigService } from "../declarations";
import {
	IFileSystem,
	IChildProcess,
	IErrors,
	ISpawnResult,
} from "../common/declarations";
import { injector } from "../common/yok";

export class CocoaPodsService implements ICocoaPodsService {
	private static PODFILE_POST_INSTALL_SECTION_NAME = "post_install";
	private static INSTALLER_BLOCK_PARAMETER_NAME = "installer";
	private getCocoaPodsFromPodfile: Function;
	constructor(
		private $cocoaPodsPlatformManager: ICocoaPodsPlatformManager,
		private $fs: IFileSystem,
		private $childProcess: IChildProcess,
		private $errors: IErrors,
		private $logger: ILogger,
		private $config: IConfiguration,
		private $xcconfigService: IXcconfigService
	) {
		this.getCocoaPodsFromPodfile = _.memoize(
			this._getCocoaPodsFromPodfile,
			getHash
		);
	}

	public getPodfileHeader(targetName: string): string {
		return `use_frameworks!${EOL}${EOL}target "${targetName}" do${EOL}`;
	}

	public getPodfileFooter(): string {
		return `${EOL}end`;
	}

	public getProjectPodfilePath(projectRoot: string): string {
		return path.join(projectRoot, PODFILE_NAME);
	}

	public async executePodInstall(
		projectRoot: string,
		xcodeProjPath: string
	): Promise<ISpawnResult> {
		this.$logger.info("Installing pods...");
		let podTool = this.$config.USE_POD_SANDBOX ? "sandbox-pod" : "pod";
		const args = ["install"];

		if (process.platform === "darwin" && process.arch === "arm64") {
			// check if pod is installed as an x86_64 binary or a native arm64 one
			// we run the following:
			// arch -x86_64 pod --version
			// if it's an arm64 binary, we'll get something like this as a result:
			// arch: posix_spawnp: pod: Bad CPU type in executable
			// in which case, we should run it natively.
			const res: string = await this.$childProcess
				.exec("arch -x86_64 pod --version", null, {
					showStderr: true,
				})
				.then((res) => res.stdout + " " + res.stderr)
				.catch((err) => err.message);

			if (!res.includes("Bad CPU type in executable")) {
				this.$logger.trace(
					"Running on arm64 but pod is installed under rosetta2 - running pod through rosetta2"
				);
				args.unshift(podTool);
				args.unshift("-x86_64");
				podTool = "arch";
			}
		}
		// cocoapods print a lot of non-error information on stderr. Pipe the `stderr` to `stdout`, so we won't polute CLI's stderr output.
		const podInstallResult = await this.$childProcess.spawnFromEvent(
			podTool,
			args,
			"close",
			{ cwd: projectRoot, stdio: ["pipe", process.stdout, process.stdout] },
			{ throwError: false }
		);

		if (podInstallResult.exitCode !== 0) {
			// https://github.com/CocoaPods/CocoaPods/blob/92aaf0f1120d32f3487960b485fb69fcaf61486c/lib/cocoapods/resolver.rb#L498
			// TODO add article
			const versionResolutionHint =
				podInstallResult.exitCode === 31
					? `For more information on resolving CocoaPod issues in NativeScript read.`
					: "";
			this.$errors.fail(`'${podTool} install' command failed.${
				podInstallResult.stderr ? " Error is: " + podInstallResult.stderr : ""
			}
${versionResolutionHint}`);
		}

		return podInstallResult;
	}

	public async mergePodXcconfigFile(
		projectData: IProjectData,
		platformData: IPlatformData
	): Promise<void> {
		const podFilesRootDirName = path.join(
			"Pods",
			"Target Support Files",
			`Pods-${projectData.projectName}`
		);
		const podFolder = path.join(platformData.projectRoot, podFilesRootDirName);
		if (this.$fs.exists(podFolder)) {
			const pluginsXcconfigFilePaths = this.$xcconfigService.getPluginsXcconfigFilePaths(
				platformData.projectRoot
			);
			for (const configuration in pluginsXcconfigFilePaths) {
				const pluginsXcconfigFilePath = pluginsXcconfigFilePaths[configuration];
				const podXcconfigFilePath = path.join(
					podFolder,
					`Pods-${projectData.projectName}.${configuration}.xcconfig`
				);
				await this.$xcconfigService.mergeFiles(
					podXcconfigFilePath,
					pluginsXcconfigFilePath
				);
			}
		}
	}

	public async applyPodfileFromAppResources(
		projectData: IProjectData,
		platformData: IPlatformData
	): Promise<void> {
		const { projectRoot, normalizedPlatformName } = platformData;
		const mainPodfilePath = path.join(
			projectData.appResourcesDirectoryPath,
			normalizedPlatformName,
			PODFILE_NAME
		);
		const projectPodfilePath = this.getProjectPodfilePath(projectRoot);
		if (
			this.$fs.exists(projectPodfilePath) ||
			this.$fs.exists(mainPodfilePath)
		) {
			await this.applyPodfileToProject(
				NS_BASE_PODFILE,
				mainPodfilePath,
				projectData,
				platformData
			);
		}
	}

	public async applyPodfileArchExclusions(
		projectData: IProjectData,
		platformData: IPlatformData
	): Promise<void> {
		const { projectRoot } = platformData;
		const exclusionsPodfile = path.join(projectRoot, "Podfile-exclusions");

		if (!this.$fs.exists(exclusionsPodfile)) {
			const exclusions = `
post_install do |installer|
  installer.pods_project.build_configurations.each do |config|
    config.build_settings.delete "VALID_ARCHS"
    config.build_settings["EXCLUDED_ARCHS_x86_64"] = "arm64 arm64e"
    config.build_settings["EXCLUDED_ARCHS[sdk=iphonesimulator*]"] = "i386 armv6 armv7 armv7s armv8 $(EXCLUDED_ARCHS_$(NATIVE_ARCH_64_BIT))"
    config.build_settings["EXCLUDED_ARCHS[sdk=iphoneos*]"] = "i386 armv6 armv7 armv7s armv8 x86_64"
  end
end`.trim();
			this.$fs.writeFile(exclusionsPodfile, exclusions);
		}

		await this.applyPodfileToProject(
			"NativeScript-CLI-Architecture-Exclusions",
			exclusionsPodfile,
			projectData,
			platformData
		);

		// clean up
		this.$fs.deleteFile(exclusionsPodfile);
	}

	public async applyPodfileToProject(
		moduleName: string,
		podfilePath: string,
		projectData: IProjectData,
		platformData: IPlatformData
	): Promise<void> {
		const nativeProjectPath = platformData.projectRoot;
		if (!this.$fs.exists(podfilePath)) {
			this.removePodfileFromProject(
				moduleName,
				podfilePath,
				projectData,
				nativeProjectPath
			);
			return;
		}

		const {
			podfileContent,
			replacedFunctions,
			podfilePlatformData,
		} = this.buildPodfileContent(
			podfilePath,
			moduleName,
			projectData,
			platformData
		);
		const pathToProjectPodfile = this.getProjectPodfilePath(nativeProjectPath);
		const projectPodfileContent = this.$fs.exists(pathToProjectPodfile)
			? this.$fs.readText(pathToProjectPodfile).trim()
			: "";

		if (projectPodfileContent.indexOf(podfileContent) === -1) {
			// Remove old occurences of the plugin from the project's Podfile.
			this.removePodfileFromProject(
				moduleName,
				podfilePath,
				projectData,
				nativeProjectPath
			);
			let finalPodfileContent = this.$fs.exists(pathToProjectPodfile)
				? this.getPodfileContentWithoutTarget(
						projectData,
						this.$fs.readText(pathToProjectPodfile)
				  )
				: "";

			if (
				podfileContent.indexOf(
					CocoaPodsService.PODFILE_POST_INSTALL_SECTION_NAME
				) !== -1
			) {
				finalPodfileContent = this.addPostInstallHook(
					replacedFunctions,
					finalPodfileContent,
					podfileContent
				);
			}

			if (podfilePlatformData) {
				finalPodfileContent = this.$cocoaPodsPlatformManager.addPlatformSection(
					projectData,
					podfilePlatformData,
					finalPodfileContent
				);
			}

			finalPodfileContent = `${finalPodfileContent.trim()}${EOL}${EOL}${podfileContent.trim()}${EOL}`;
			this.saveProjectPodfile(
				projectData,
				finalPodfileContent,
				nativeProjectPath
			);
		}
	}

	public removePodfileFromProject(
		moduleName: string,
		podfilePath: string,
		projectData: IProjectData,
		projectRoot: string
	): void {
		if (this.$fs.exists(this.getProjectPodfilePath(projectRoot))) {
			let projectPodFileContent = this.$fs.readText(
				this.getProjectPodfilePath(projectRoot)
			);
			// Remove the data between #Begin Podfile and #EndPodfile
			const regExpToRemove = new RegExp(
				`${this.getPluginPodfileHeader(
					podfilePath
				)}[\\s\\S]*?${this.getPluginPodfileEnd()}`,
				"mg"
			);
			projectPodFileContent = projectPodFileContent.replace(regExpToRemove, "");
			projectPodFileContent = this.removePostInstallHook(
				moduleName,
				projectPodFileContent
			);
			projectPodFileContent = this.$cocoaPodsPlatformManager.removePlatformSection(
				moduleName,
				projectPodFileContent,
				podfilePath
			);

			const defaultPodfileBeginning = this.getPodfileHeader(
				projectData.projectName
			);
			const defaultContentWithPostInstallHook = `${defaultPodfileBeginning}${this.getPostInstallHookHeader()}end${EOL}end`;
			const defaultContentWithoutPostInstallHook = `${defaultPodfileBeginning}${EOL}end`;
			const trimmedProjectPodFileContent = projectPodFileContent.trim();
			if (
				!trimmedProjectPodFileContent ||
				trimmedProjectPodFileContent === defaultContentWithPostInstallHook ||
				trimmedProjectPodFileContent === defaultContentWithoutPostInstallHook
			) {
				this.$fs.deleteFile(this.getProjectPodfilePath(projectRoot));
			} else {
				this.$fs.writeFile(
					this.getProjectPodfilePath(projectRoot),
					projectPodFileContent
				);
			}
		}
	}

	public getPluginPodfilePath(pluginData: IPluginData): string {
		const pluginPlatformsFolderPath = pluginData.pluginPlatformsFolderPath(
			PlatformTypes.ios
		);
		const pluginPodFilePath = path.join(
			pluginPlatformsFolderPath,
			PODFILE_NAME
		);
		return pluginPodFilePath;
	}

	private addPostInstallHook(
		replacedFunctions: IRubyFunction[],
		finalPodfileContent: string,
		pluginPodfileContent: string
	): string {
		const postInstallHookStart = this.getPostInstallHookHeader();
		let postInstallHookContent = "";
		_.each(replacedFunctions, (rubyFunction) => {
			let functionExecution = rubyFunction.functionName;
			if (
				rubyFunction.functionParameters &&
				rubyFunction.functionParameters.length
			) {
				functionExecution = `${functionExecution} ${CocoaPodsService.INSTALLER_BLOCK_PARAMETER_NAME}`;
			}

			postInstallHookContent += `  ${functionExecution}${EOL}`;
		});

		if (postInstallHookContent) {
			const index = finalPodfileContent.indexOf(postInstallHookStart);
			if (index !== -1) {
				const regExp = new RegExp(
					`(${regExpEscape(postInstallHookStart)}[\\s\\S]*?)(\\bend\\b)`,
					"m"
				);
				finalPodfileContent = finalPodfileContent.replace(
					regExp,
					`$1${postInstallHookContent.trimRight()}${EOL}$2`
				);
			} else {
				if (finalPodfileContent.length > 0) {
					finalPodfileContent += `${EOL}${EOL}`;
				}
				const postInstallHook = `${postInstallHookStart}${postInstallHookContent}end`;
				finalPodfileContent = `${finalPodfileContent}${postInstallHook}`;
			}
		}

		return finalPodfileContent;
	}

	private getPodfileContentWithoutTarget(
		projectData: IProjectData,
		projectPodfileContent: string
	): string {
		const podFileHeader = this.getPodfileHeader(projectData.projectName);

		if (_.startsWith(projectPodfileContent, podFileHeader)) {
			projectPodfileContent = projectPodfileContent.substr(
				podFileHeader.length
			);

			const podFileFooter = this.getPodfileFooter();
			// Only remove the final end in case the file starts with the podFileHeader
			if (_.endsWith(projectPodfileContent, podFileFooter)) {
				projectPodfileContent = projectPodfileContent.substr(
					0,
					projectPodfileContent.length - podFileFooter.length
				);
			}
		}

		return projectPodfileContent.trim();
	}

	private saveProjectPodfile(
		projectData: IProjectData,
		projectPodfileContent: string,
		projectRoot: string
	): void {
		projectPodfileContent = this.getPodfileContentWithoutTarget(
			projectData,
			projectPodfileContent
		);
		const podFileHeader = this.getPodfileHeader(projectData.projectName);
		const podFileFooter = this.getPodfileFooter();
		const contentToWrite = `${podFileHeader}${projectPodfileContent}${podFileFooter}`;
		const projectPodfilePath = this.getProjectPodfilePath(projectRoot);
		this.$fs.writeFile(projectPodfilePath, contentToWrite);
	}

	private removePostInstallHook(
		moduleName: string,
		projectPodFileContent: string
	): string {
		const regExp = new RegExp(
			`^.*?${this.getHookBasicFuncNameForPlugin(
				CocoaPodsService.PODFILE_POST_INSTALL_SECTION_NAME,
				moduleName
			)}.*?$\\r?\\n`,
			"gm"
		);
		projectPodFileContent = projectPodFileContent.replace(regExp, "");
		return projectPodFileContent;
	}

	private getHookBasicFuncNameForPlugin(
		hookName: string,
		pluginName: string
	): string {
		// nativescript-hook and nativescript_hook should have different names, so replace all _ with ___ first and then replace all special symbols with _
		// This will lead to a clash in case plugins are called nativescript-hook and nativescript___hook
		const replacedPluginName = pluginName
			.replace(/_/g, "___")
			.replace(/[^A-Za-z0-9_]/g, "_");
		return `${hookName}${replacedPluginName}`;
	}

	private replaceHookContent(
		hookName: string,
		podfileContent: string,
		pluginName: string
	): { replacedContent: string; newFunctions: IRubyFunction[] } {
		const hookStart = `${hookName} do`;

		const hookDefinitionRegExp = new RegExp(
			`${hookStart} *(\\|(\\w+)\\|)?`,
			"g"
		);
		const newFunctions: IRubyFunction[] = [];

		const replacedContent = podfileContent.replace(
			hookDefinitionRegExp,
			(
				substring: string,
				firstGroup: string,
				secondGroup: string,
				index: number
			): string => {
				const newFunctionName = `${this.getHookBasicFuncNameForPlugin(
					hookName,
					pluginName
				)}_${newFunctions.length}`;
				let newDefinition = `def ${newFunctionName}`;

				const rubyFunction: IRubyFunction = { functionName: newFunctionName };
				// firstGroup is the block parameter, secondGroup is the block parameter name.
				if (firstGroup && secondGroup) {
					newDefinition = `${newDefinition} (${secondGroup})`;
					rubyFunction.functionParameters = secondGroup;
				}

				newFunctions.push(rubyFunction);
				return newDefinition;
			}
		);

		return { replacedContent, newFunctions };
	}

	private getPluginPodfileHeader(pluginPodFilePath: string): string {
		// escape special + from the podfile path (pnpm)
		pluginPodFilePath = pluginPodFilePath.replace(/\+/g, "\\+");
		return `# Begin Podfile - ${pluginPodFilePath}`;
	}

	private getPluginPodfileEnd(): string {
		return `# End Podfile${EOL}`;
	}

	private getPostInstallHookHeader() {
		return `${CocoaPodsService.PODFILE_POST_INSTALL_SECTION_NAME} do |${CocoaPodsService.INSTALLER_BLOCK_PARAMETER_NAME}|${EOL}`;
	}

	private buildPodfileContent(
		pluginPodFilePath: string,
		pluginName: string,
		projectData: IProjectData,
		platformData: IPlatformData
	): {
		podfileContent: string;
		replacedFunctions: IRubyFunction[];
		podfilePlatformData: IPodfilePlatformData;
	} {
		const pluginPodfileContent = this.$fs.readText(pluginPodFilePath);
		const data = this.replaceHookContent(
			CocoaPodsService.PODFILE_POST_INSTALL_SECTION_NAME,
			pluginPodfileContent,
			pluginName
		);
		const cocoapodsData = this.$cocoaPodsPlatformManager.replacePlatformRow(
			data.replacedContent,
			pluginPodFilePath
		);
		const podfilePlatformData = cocoapodsData.podfilePlatformData;
		let replacedContent = cocoapodsData.replacedContent;

		if (
			projectData.nsConfig &&
			projectData.nsConfig.overridePods &&
			!this.isMainPodFile(pluginPodFilePath, projectData, platformData)
		) {
			replacedContent = this.overridePodsFromFile(
				replacedContent,
				projectData,
				platformData
			);
		}

		return {
			podfileContent: `${this.getPluginPodfileHeader(
				pluginPodFilePath
			)}${EOL}${replacedContent}${EOL}${this.getPluginPodfileEnd()}`,
			replacedFunctions: data.newFunctions,
			podfilePlatformData,
		};
	}

	private getMainPodFilePath(
		projectData: IProjectData,
		platformData: IPlatformData
	): string {
		return path.join(
			projectData.appResourcesDirectoryPath,
			platformData.normalizedPlatformName,
			PODFILE_NAME
		);
	}

	private isMainPodFile(
		podFilePath: string,
		projectData: IProjectData,
		platformData: IPlatformData
	): boolean {
		const mainPodfilePath = this.getMainPodFilePath(projectData, platformData);

		return podFilePath === mainPodfilePath;
	}

	private overridePodsFromFile(
		podfileContent: string,
		projectData: IProjectData,
		platformData: IPlatformData
	): string {
		const mainPodfilePath = this.getMainPodFilePath(projectData, platformData);

		if (this.$fs.exists(mainPodfilePath)) {
			const mainPodfileContent = this.$fs.readText(mainPodfilePath);
			const pods = this.getCocoaPodsFromPodfile(mainPodfileContent);
			_.forEach(pods, (pod) => {
				podfileContent = podfileContent.replace(
					new RegExp(`^[ ]*pod\\s*["']${pod}['"].*$`, "gm"),
					"#$&"
				);
			});
		}

		return podfileContent;
	}

	private _getCocoaPodsFromPodfile(podfileContent: string): Array<string> {
		const pods = [];
		const podsRegex = /^\s*pod\s*["'](.*?)['"].*$/gm;

		let match = podsRegex.exec(podfileContent);
		while (match != null) {
			const podName: string = match[1];
			if (podName) {
				pods.push(podName);
			}

			match = podsRegex.exec(podfileContent);
		}

		return pods;
	}
}

injector.register("cocoapodsService", CocoaPodsService);