MacAppUtilities.java

package net.filebot.platform.mac;

import static ca.weblite.objc.util.CocoaUtils.*;
import static net.filebot.Logging.*;

import java.awt.EventQueue;
import java.awt.SecondaryLoop;
import java.awt.Toolkit;
import java.awt.Window;
import java.io.File;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.function.Consumer;
import java.util.logging.Level;

import javax.swing.JMenuBar;
import javax.swing.UIManager;

import com.apple.eawt.Application;
import com.apple.eawt.FullScreenUtilities;
import com.apple.eawt.QuitStrategy;
import com.apple.eio.FileManager;
import com.sun.jna.Pointer;

import ca.weblite.objc.Client;
import ca.weblite.objc.Proxy;

public class MacAppUtilities {

	private static Client _objc;

	public static Client objc() {
		if (_objc == null) {
			_objc = new Client();
		}
		return _objc;
	}

	public static Object NSData_initWithBase64Encoding(String text) {
		return objc().sendProxy("NSData", "data").send("initWithBase64Encoding:", text);
	}

	public static String NSURL_bookmarkDataWithOptions(String path) {
		return objc().sendProxy("NSURL", "fileURLWithPath:", path).sendProxy("bookmarkDataWithOptions:includingResourceValuesForKeys:relativeToURL:error:", 2048, null, null, null).sendString("base64Encoding");
	}

	public static Object NSURL_URLByResolvingBookmarkData_startAccessingSecurityScopedResource(String text) {
		return objc().sendProxy("NSURL", "URLByResolvingBookmarkData:options:relativeToURL:bookmarkDataIsStale:error:", NSData_initWithBase64Encoding(text), 1024, null, false, null).send("startAccessingSecurityScopedResource");
	}

	public static List<File> NSOpenPanel_openPanel_runModal(String title, boolean multipleMode, boolean canChooseDirectories, boolean canChooseFiles, String[] allowedFileTypes) {
		final List<File> result = new ArrayList<File>();

		final EventQueue eventQueue = Toolkit.getDefaultToolkit().getSystemEventQueue();
		final SecondaryLoop secondaryLoop = eventQueue.createSecondaryLoop();

		// WARNING: dispatch_sync seems to work on most Mac always causes a deadlock and freezes the application on others (in particular MBP with 2 graphics chips)
		dispatch_async(() -> {
			Pointer pool = createAutoreleasePool();
			Proxy peer = objc().sendProxy("NSOpenPanel", "openPanel");
			peer.send("retain");

			peer.send("setTitle:", title);
			peer.send("setAllowsMultipleSelection:", multipleMode ? 1 : 0);
			peer.send("setCanChooseDirectories:", canChooseDirectories ? 1 : 0);
			peer.send("setCanChooseFiles:", canChooseFiles ? 1 : 0);

			if (allowedFileTypes != null) {
				Proxy mutableArray = objc().sendProxy("NSMutableArray", "arrayWithCapacity:", allowedFileTypes.length);
				for (String type : allowedFileTypes) {
					mutableArray.send("addObject:", type);
				}
				peer.send("setAllowedFileTypes:", mutableArray);
			}

			if (peer.sendInt("runModal") != 0) {
				Proxy nsArray = peer.getProxy("URLs");
				int size = nsArray.sendInt("count");
				for (int i = 0; i < size; i++) {
					Proxy url = nsArray.sendProxy("objectAtIndex:", i);
					String path = url.sendString("path");
					result.add(new File(path));
				}
			}

			drainAutoreleasePool(pool);
			secondaryLoop.exit();
		});

		// Enter the loop to block the current event handler, but leave UI responsive
		if (!secondaryLoop.enter()) {
			throw new IllegalStateException("SecondaryLoop");
		}

		return result;
	}

	public static void setWindowCanFullScreen(Window window) {
		try {
			FullScreenUtilities.setWindowCanFullScreen(window, true);
		} catch (Throwable t) {
			debug.log(Level.WARNING, t.getMessage(), t);
		}
	}

	public static void requestForeground() {
		try {
			Application.getApplication().requestForeground(true);
		} catch (Throwable t) {
			debug.log(Level.WARNING, t.getMessage(), t);
		}
	}

	public static void revealInFinder(File file) {
		try {
			FileManager.revealInFinder(file);
		} catch (Throwable t) {
			debug.log(Level.WARNING, t.getMessage(), t);
		}
	}

	public static void moveToTrash(File file) {
		try {
			FileManager.moveToTrash(file);
		} catch (Throwable t) {
			debug.log(Level.WARNING, t.getMessage(), t);
		}
	}

	public static void setDefaultMenuBar(JMenuBar menu) {
		try {
			Application.getApplication().setDefaultMenuBar(menu);
		} catch (Throwable t) {
			debug.log(Level.WARNING, t.getMessage(), t);
		}
	}

	public static void setQuitStrategyCloseAll() {
		try {
			Application.getApplication().setQuitStrategy(QuitStrategy.CLOSE_ALL_WINDOWS);
		} catch (Throwable t) {
			debug.log(Level.WARNING, t.getMessage(), t);
		}
	}

	public static void setOpenFileHandler(Consumer<List<File>> handler) {
		try {
			Application.getApplication().setOpenFileHandler(evt -> {
				List<File> files = evt.getFiles();
				if (files.size() > 0) {
					handler.accept(files);
				}
			});
		} catch (Throwable t) {
			debug.log(Level.WARNING, t.getMessage(), t);
		}
	}

	public static void initializeApplication() {
		// improved UI defaults
		UIManager.put("TitledBorder.border", UIManager.getBorder("InsetBorder.aquaVariant"));

		// make sure Application Quit Events get forwarded to normal Window Listeners
		setQuitStrategyCloseAll();
	}

	public static boolean isLockedFolder(File folder) {
		// write permissions my not be available even after sandbox has granted permission (e.g. when accessing files of another user)
		return folder.isDirectory() && !folder.canRead();
	}

	public static boolean askUnlockFolders(Window owner, Collection<File> files) {
		return DropToUnlock.showUnlockFoldersDialog(owner, files);
	}

	private MacAppUtilities() {
		throw new UnsupportedOperationException();
	}

}