Graphite/desktop/bundle/src/mac.rs

use std::collections::HashMap;
use std::error::Error;
use std::fs;
use std::path::{Path, PathBuf};

use crate::common::*;

const APP_ID: &str = "art.graphite.Graphite";

const ICONS_FILE_NAME: &str = "graphite.icns";

const EXEC_PATH: &str = "Contents/MacOS";
const FRAMEWORKS_PATH: &str = "Contents/Frameworks";
const RESOURCES_PATH: &str = "Contents/Resources";
const CEF_FRAMEWORK: &str = "Chromium Embedded Framework.framework";
const GRAPHITE_DOCUMENT_TYPE: &str = "art.graphite.document";
const GRAPHITE_FILE_EXTENSION: &str = "graphite";
const GRAPHITE_MIME_TYPE: &str = "application/graphite+json";

pub fn main() -> Result<(), Box<dyn Error>> {
	let app_bin = build_bin("graphite-desktop-platform-mac", None, Some("main"))?;
	let helper_bin = build_bin("graphite-desktop-platform-mac", Some("helper"), Some("helper"))?;

	let profile_path = profile_path();
	let app_dir = bundle(&profile_path, &app_bin, &helper_bin);

	// TODO: Consider adding more useful cli
	let args: Vec<String> = std::env::args().collect();
	if let Some(pos) = args.iter().position(|a| a == "open") {
		let executable = app_dir.join(EXEC_PATH).join(APP_NAME);
		let extra_args: Vec<&str> = args[pos + 1..].iter().map(|s| s.as_str()).collect();
		run_command(&executable.to_string_lossy(), &extra_args).expect("failed to open app");
	}

	Ok(())
}

fn bundle(out_dir: &Path, app_bin: &Path, helper_bin: &Path) -> PathBuf {
	let app_dir = out_dir.join(APP_NAME).with_extension("app");

	clean_dir(&app_dir);

	create_app(&app_dir, APP_ID, APP_NAME, app_bin, false);

	for helper_type in [None, Some("GPU"), Some("Renderer")] {
		let helper_id_suffix = helper_type.map(|t| format!(".{t}")).unwrap_or_default();
		let helper_id = format!("{APP_ID}.helper{helper_id_suffix}");
		let helper_name_suffix = helper_type.map(|t| format!(" ({t})")).unwrap_or_default();
		let helper_name = format!("{APP_NAME} Helper{helper_name_suffix}");
		let helper_app_dir = app_dir.join(FRAMEWORKS_PATH).join(&helper_name).with_extension("app");
		create_app(&helper_app_dir, &helper_id, &helper_name, helper_bin, true);
	}

	copy_dir(&cef_path().join(CEF_FRAMEWORK), &app_dir.join(FRAMEWORKS_PATH).join(CEF_FRAMEWORK));

	let resource_dir = app_dir.join(RESOURCES_PATH);
	fs::create_dir_all(&resource_dir).expect("failed to create app resource dir");

	let icon_file = workspace_path().join("branding/app-icons").join(ICONS_FILE_NAME);
	fs::copy(icon_file, resource_dir.join(ICONS_FILE_NAME)).expect("failed to copy icon file");

	app_dir
}

fn create_app(app_dir: &Path, id: &str, name: &str, bin: &Path, is_helper: bool) {
	fs::create_dir_all(app_dir.join(EXEC_PATH)).unwrap();

	let app_contents_dir: &Path = &app_dir.join("Contents");
	create_info_plist(app_contents_dir, id, name, is_helper).unwrap();
	fs::copy(bin, app_dir.join(EXEC_PATH).join(name)).unwrap();
}

fn create_info_plist(dir: &Path, id: &str, exec_name: &str, is_helper: bool) -> Result<(), Box<dyn std::error::Error>> {
	let info = InfoPlist {
		cf_bundle_name: exec_name.to_string(),
		cf_bundle_identifier: id.to_string(),
		cf_bundle_display_name: exec_name.to_string(),
		cf_bundle_executable: exec_name.to_string(),
		cf_bundle_icon_file: if is_helper { None } else { Some(ICONS_FILE_NAME.to_string()) },
		cf_bundle_info_dictionary_version: "6.0".to_string(),
		cf_bundle_package_type: "APPL".to_string(),
		cf_bundle_signature: "????".to_string(),
		cf_bundle_version: "0.0.0".to_string(),
		cf_bundle_short_version_string: "0.0".to_string(),
		cf_bundle_development_region: "en".to_string(),
		ls_environment: [("MallocNanoZone".to_string(), "0".to_string())].iter().cloned().collect(),
		ls_file_quarantine_enabled: true,
		ls_minimum_system_version: "11.0".to_string(),
		ls_ui_element: if is_helper { Some("1".to_string()) } else { None },
		ns_supports_automatic_graphics_switching: true,
		cf_bundle_document_types: (!is_helper).then(document_types),
		ut_exported_type_declarations: (!is_helper).then(exported_type_declarations),
	};

	let plist_file = dir.join("Info.plist");
	plist::to_file_xml(plist_file, &info)?;
	Ok(())
}

fn document_types() -> Vec<DocumentType> {
	vec![
		DocumentType {
			cf_bundle_type_name: "Graphite Document".to_string(),
			cf_bundle_type_role: "Editor".to_string(),
			cf_bundle_type_extensions: Some(vec![GRAPHITE_FILE_EXTENSION.to_string()]),
			cf_bundle_type_icon_file: Some(ICONS_FILE_NAME.to_string()),
			ls_handler_rank: Some("Owner".to_string()),
			ls_item_content_types: vec![GRAPHITE_DOCUMENT_TYPE.to_string()],
		},
		DocumentType {
			cf_bundle_type_name: "SVG Image".to_string(),
			cf_bundle_type_role: "Editor".to_string(),
			cf_bundle_type_extensions: Some(vec!["svg".to_string()]),
			cf_bundle_type_icon_file: None,
			ls_handler_rank: Some("Alternate".to_string()),
			ls_item_content_types: vec!["public.svg-image".to_string()],
		},
		DocumentType {
			cf_bundle_type_name: "Image".to_string(),
			cf_bundle_type_role: "Editor".to_string(),
			cf_bundle_type_extensions: None,
			cf_bundle_type_icon_file: None,
			ls_handler_rank: Some("Alternate".to_string()),
			ls_item_content_types: vec!["public.image".to_string()],
		},
	]
}

fn exported_type_declarations() -> Vec<ExportedTypeDeclaration> {
	vec![ExportedTypeDeclaration {
		ut_type_identifier: GRAPHITE_DOCUMENT_TYPE.to_string(),
		ut_type_description: "Graphite Document".to_string(),
		ut_type_conforms_to: vec!["public.json".to_string()],
		ut_type_tag_specification: TypeTagSpecification {
			public_filename_extension: vec![GRAPHITE_FILE_EXTENSION.to_string()],
			public_mime_type: GRAPHITE_MIME_TYPE.to_string(),
		},
	}]
}

#[derive(serde::Serialize)]
struct InfoPlist {
	#[serde(rename = "CFBundleName")]
	cf_bundle_name: String,
	#[serde(rename = "CFBundleIdentifier")]
	cf_bundle_identifier: String,
	#[serde(rename = "CFBundleDisplayName")]
	cf_bundle_display_name: String,
	#[serde(rename = "CFBundleExecutable")]
	cf_bundle_executable: String,
	#[serde(rename = "CFBundleIconFile")]
	#[serde(skip_serializing_if = "Option::is_none")]
	cf_bundle_icon_file: Option<String>,
	#[serde(rename = "CFBundleInfoDictionaryVersion")]
	cf_bundle_info_dictionary_version: String,
	#[serde(rename = "CFBundlePackageType")]
	cf_bundle_package_type: String,
	#[serde(rename = "CFBundleSignature")]
	cf_bundle_signature: String,
	#[serde(rename = "CFBundleVersion")]
	cf_bundle_version: String,
	#[serde(rename = "CFBundleShortVersionString")]
	cf_bundle_short_version_string: String,
	#[serde(rename = "CFBundleDevelopmentRegion")]
	cf_bundle_development_region: String,
	#[serde(rename = "LSEnvironment")]
	ls_environment: HashMap<String, String>,
	#[serde(rename = "LSFileQuarantineEnabled")]
	ls_file_quarantine_enabled: bool,
	#[serde(rename = "LSMinimumSystemVersion")]
	ls_minimum_system_version: String,
	#[serde(rename = "LSUIElement")]
	#[serde(skip_serializing_if = "Option::is_none")]
	ls_ui_element: Option<String>,
	#[serde(rename = "NSSupportsAutomaticGraphicsSwitching")]
	ns_supports_automatic_graphics_switching: bool,
	#[serde(rename = "CFBundleDocumentTypes")]
	#[serde(skip_serializing_if = "Option::is_none")]
	cf_bundle_document_types: Option<Vec<DocumentType>>,
	#[serde(rename = "UTExportedTypeDeclarations")]
	#[serde(skip_serializing_if = "Option::is_none")]
	ut_exported_type_declarations: Option<Vec<ExportedTypeDeclaration>>,
}

#[derive(serde::Serialize)]
struct DocumentType {
	#[serde(rename = "CFBundleTypeName")]
	cf_bundle_type_name: String,
	#[serde(rename = "CFBundleTypeRole")]
	cf_bundle_type_role: String,
	#[serde(rename = "CFBundleTypeExtensions")]
	#[serde(skip_serializing_if = "Option::is_none")]
	cf_bundle_type_extensions: Option<Vec<String>>,
	#[serde(rename = "CFBundleTypeIconFile")]
	#[serde(skip_serializing_if = "Option::is_none")]
	cf_bundle_type_icon_file: Option<String>,
	#[serde(rename = "LSHandlerRank")]
	#[serde(skip_serializing_if = "Option::is_none")]
	ls_handler_rank: Option<String>,
	#[serde(rename = "LSItemContentTypes")]
	ls_item_content_types: Vec<String>,
}

#[derive(serde::Serialize)]
struct ExportedTypeDeclaration {
	#[serde(rename = "UTTypeIdentifier")]
	ut_type_identifier: String,
	#[serde(rename = "UTTypeDescription")]
	ut_type_description: String,
	#[serde(rename = "UTTypeConformsTo")]
	ut_type_conforms_to: Vec<String>,
	#[serde(rename = "UTTypeTagSpecification")]
	ut_type_tag_specification: TypeTagSpecification,
}

#[derive(serde::Serialize)]
struct TypeTagSpecification {
	#[serde(rename = "public.filename-extension")]
	public_filename_extension: Vec<String>,
	#[serde(rename = "public.mime-type")]
	public_mime_type: String,
}