Initial public release of Warp.
Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b
This commit is contained in:
David Stern
@@ -0,0 +1,228 @@
|
||||
#include "WarpDockTilePlugin.h"
|
||||
|
||||
@implementation WarpDockTilePlugIn {
|
||||
NSFileHandle *_logFileHandle;
|
||||
}
|
||||
|
||||
@synthesize iconChangedObserver;
|
||||
@synthesize defaultsObserver;
|
||||
|
||||
- (void)logMessage:(NSString *)message {
|
||||
if (_logFileHandle) {
|
||||
NSDateFormatter *formatter = [[NSDateFormatter alloc] init];
|
||||
[formatter setDateFormat:@"yyyy-MM-dd HH:mm:ss.SSS"];
|
||||
NSString *timestamp = [formatter stringFromDate:[NSDate date]];
|
||||
NSString *logEntry = [NSString stringWithFormat:@"[%@] %@\n", timestamp, message];
|
||||
[_logFileHandle writeData:[logEntry dataUsingEncoding:NSUTF8StringEncoding]];
|
||||
[_logFileHandle synchronizeFile];
|
||||
}
|
||||
}
|
||||
|
||||
- (instancetype)init {
|
||||
self = [super init];
|
||||
if (self) {
|
||||
@try {
|
||||
NSDateFormatter *formatter = [[NSDateFormatter alloc] init];
|
||||
[formatter setDateFormat:@"yyyy-MM-dd_HH-mm-ss"];
|
||||
NSString *timestamp = [formatter stringFromDate:[NSDate date]];
|
||||
NSString *logPath = [NSString stringWithFormat:@"/tmp/warp_docktile_%@.log", timestamp];
|
||||
NSError *error = nil;
|
||||
[[NSFileManager defaultManager] createFileAtPath:logPath contents:nil attributes:nil];
|
||||
_logFileHandle = [NSFileHandle fileHandleForWritingAtPath:logPath];
|
||||
[self logMessage:@"WarpDockTilePlugin initialized"];
|
||||
} @catch (NSException *exception) {
|
||||
NSLog(@"Exception during initialization: %@\nStack trace: %@",
|
||||
exception.reason,
|
||||
exception.callStackSymbols);
|
||||
}
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
- (void)updateAppIcon:(NSDockTile *)tile {
|
||||
@try {
|
||||
[self logMessage:@"updateAppIcon called"];
|
||||
// Retrieve the bundle ID for the main app from the Info.plist file in the plugin bundle
|
||||
NSBundle *pluginBundle = [NSBundle bundleForClass:[self class]];
|
||||
NSString *path = [[pluginBundle bundlePath] stringByAppendingPathComponent:@"Contents/Info.plist"];
|
||||
NSDictionary *dict = [NSDictionary dictionaryWithContentsOfFile:path];
|
||||
NSString *bundleId = dict[@"MainAppBundleIdentifier"];
|
||||
BOOL isDev = [bundleId containsString:@"Dev"];
|
||||
BOOL isPreview = [bundleId containsString:@"Preview"];
|
||||
BOOL isLocal = [bundleId containsString:@"Local"];
|
||||
[self logMessage:[NSString stringWithFormat:@"Plugin Bundle ID: %@", bundleId]];
|
||||
|
||||
// Initialize the user defaults for the main app
|
||||
NSUserDefaults *hostDefaults = [[NSUserDefaults alloc] initWithSuiteName:bundleId];
|
||||
[hostDefaults synchronize];
|
||||
|
||||
// Get the icon name from the user defaults
|
||||
NSString* appIconName = [hostDefaults stringForKey:@"AppIcon"];
|
||||
|
||||
// Check if the user has set a non-default icon. If the AppIcon key is nil, empty, or "Default", reset to the
|
||||
// icon bundled with the app by setting the content to "nil". Using the icon bundled in the app will allow macOS
|
||||
// to handle the icon, including respecting "Icon & Widget Style" setting and applying a color filter. Non-
|
||||
// default icons do not respect that setting.
|
||||
NSString* cleanName = [[appIconName stringByTrimmingCharactersInSet:[NSCharacterSet characterSetWithCharactersInString:@"\""]] lowercaseString];
|
||||
if (!appIconName || [appIconName length] == 0 || [cleanName isEqualToString:@"default"]) {
|
||||
[self logMessage:@"User has default icon, resetting dock tile to system default"];
|
||||
[tile setContentView:nil];
|
||||
[tile display];
|
||||
return;
|
||||
}
|
||||
|
||||
NSString* iconFileName = [self convertAppIconNameToFileName:appIconName isDev:isDev isLocal:isLocal isPreview:isPreview];
|
||||
[self logMessage:[NSString stringWithFormat:@"Icon file name: %@", iconFileName]];
|
||||
|
||||
// Load the icon image
|
||||
NSImage* currentImage = [self LoadDockTileImage:iconFileName];
|
||||
|
||||
// Set the image on the dock tile
|
||||
// 256 x 256 is the preferred size for retina dock icons.
|
||||
NSImageView* imageView = [[NSImageView alloc] initWithFrame:NSMakeRect(0, 0, 256, 256)];
|
||||
[imageView setImage:currentImage];
|
||||
[imageView setImageScaling:NSImageScaleProportionallyUpOrDown];
|
||||
[tile setContentView:imageView];
|
||||
[tile display];
|
||||
|
||||
[self logMessage:[NSString stringWithFormat:@"Dock tile updated with icon: %@", iconFileName]];
|
||||
} @catch (NSException *exception) {
|
||||
[self logMessage:[NSString stringWithFormat:@"Exception updating dock tile icon: %@\nStack trace: %@\nTile: %@",
|
||||
exception.reason,
|
||||
exception.callStackSymbols,
|
||||
tile ? @"valid" : @"nil"]];
|
||||
}
|
||||
}
|
||||
|
||||
// See app_icon.rs for the rust version of this conversion.
|
||||
- (NSString*)convertAppIconNameToFileName:(NSString*)appIconName isDev:(BOOL)isDev isLocal:(BOOL)isLocal isPreview:(BOOL)isPreview {
|
||||
// First remove quotes and convert to lowercase
|
||||
NSString* cleanName = [[appIconName stringByTrimmingCharactersInSet:[NSCharacterSet characterSetWithCharactersInString:@"\""]] lowercaseString];
|
||||
|
||||
NSDictionary* mapping = @{
|
||||
@"aurora": @"aurora",
|
||||
@"classic1": @"classic_1",
|
||||
@"classic2": @"classic_2",
|
||||
@"classic3": @"classic_3",
|
||||
@"comets": @"comets",
|
||||
@"glasssky": @"glass_sky",
|
||||
@"glitch": @"glitch",
|
||||
@"glow": @"glow",
|
||||
@"holographic": @"holographic",
|
||||
@"mono": @"mono",
|
||||
@"neon": @"neon",
|
||||
@"original": @"original",
|
||||
@"starburst": @"starburst",
|
||||
@"sticker": @"sticker",
|
||||
@"warpone": @"blue",
|
||||
@"cow": @"cow"
|
||||
};
|
||||
|
||||
NSString* fileName = mapping[cleanName];
|
||||
|
||||
// If the mapping doesn't exist, return the default icon
|
||||
// conditional on whether this is a local, dev, or preview build.
|
||||
return fileName ?: isLocal ? @"local" : isDev ? @"dev" : isPreview ? @"preview" : @"warp_2";
|
||||
}
|
||||
|
||||
// Helper function to load named image from the plugin's resource bundle
|
||||
- (NSImage*)LoadDockTileImage:(NSString*)imageName {
|
||||
NSBundle* pluginBundle = [NSBundle bundleForClass:[self class]];
|
||||
NSString* imagePath = [pluginBundle pathForResource:imageName ofType:@"png"];
|
||||
[self logMessage:[NSString stringWithFormat:@"Image path: %@", imagePath]];
|
||||
if (imagePath == nil) {
|
||||
[self logMessage:[NSString stringWithFormat:@"Could not find image named %@ in the plugin resources", imageName]];
|
||||
return nil;
|
||||
}
|
||||
return [[NSImage alloc] initWithContentsOfFile:imagePath];
|
||||
}
|
||||
|
||||
// Protocol method that is invoked by the system when the dock for Warp is updated.
|
||||
// Note that we listen for direct changes to the AppIcon key in the user defaults.
|
||||
- (void)setDockTile:(NSDockTile *)dockTile {
|
||||
@try {
|
||||
[self logMessage:[NSString stringWithFormat:@"setDockTile called with tile: %@", dockTile ? @"valid" : @"nil"]];
|
||||
if (dockTile) {
|
||||
// Get the bundle ID for setting up user defaults observation
|
||||
NSBundle *pluginBundle = [NSBundle bundleForClass:[WarpDockTilePlugIn class]];
|
||||
NSString *path = [[pluginBundle bundlePath] stringByAppendingPathComponent:@"Contents/Info.plist"];
|
||||
NSDictionary *dict = [NSDictionary dictionaryWithContentsOfFile:path];
|
||||
NSString *bundleId = dict[@"MainAppBundleIdentifier"];
|
||||
|
||||
[self logMessage:[NSString stringWithFormat:@"Main app bundleId: %@", bundleId]];
|
||||
|
||||
// Set up user defaults observer
|
||||
NSUserDefaults *hostDefaults = [[NSUserDefaults alloc] initWithSuiteName:bundleId];
|
||||
[hostDefaults addObserver:self
|
||||
forKeyPath:@"AppIcon"
|
||||
options:NSKeyValueObservingOptionNew
|
||||
context:(__bridge void * _Nullable)(dockTile)];
|
||||
self.defaultsObserver = hostDefaults;
|
||||
|
||||
[self logMessage:[NSString stringWithFormat:@"Host defaults: %@", hostDefaults]];
|
||||
|
||||
// Make sure the icon is updated from the get-go as well.
|
||||
[self updateAppIcon:dockTile];
|
||||
} else {
|
||||
[self logMessage:@"No docktile, clearing icon observer"];
|
||||
[[NSDistributedNotificationCenter defaultCenter] removeObserver:self.iconChangedObserver];
|
||||
self.iconChangedObserver = nil;
|
||||
if (self.defaultsObserver) {
|
||||
[(NSUserDefaults *)self.defaultsObserver removeObserver:self forKeyPath:@"AppIcon"];
|
||||
self.defaultsObserver = nil;
|
||||
}
|
||||
}
|
||||
} @catch (NSException *exception) {
|
||||
[self logMessage:[NSString stringWithFormat:@"Exception in setDockTile: %@\nStack trace: %@\nDockTile: %@",
|
||||
exception.reason,
|
||||
exception.callStackSymbols,
|
||||
dockTile ? @"valid" : @"nil"]];
|
||||
}
|
||||
}
|
||||
|
||||
- (void)dealloc {
|
||||
@try {
|
||||
[self logMessage:@"WarpDockTilePlugin deallocating"];
|
||||
if (self.iconChangedObserver) {
|
||||
[[NSDistributedNotificationCenter defaultCenter] removeObserver:self.iconChangedObserver];
|
||||
self.iconChangedObserver = nil;
|
||||
}
|
||||
if (self.defaultsObserver) {
|
||||
[(NSUserDefaults *)self.defaultsObserver removeObserver:self forKeyPath:@"AppIcon"];
|
||||
self.defaultsObserver = nil;
|
||||
}
|
||||
|
||||
if (_logFileHandle) {
|
||||
[self logMessage:@"Closing log file"];
|
||||
[_logFileHandle closeFile];
|
||||
_logFileHandle = nil;
|
||||
}
|
||||
} @catch (NSException *exception) {
|
||||
NSLog(@"Exception during deallocation: %@\nStack trace: %@",
|
||||
exception.reason,
|
||||
exception.callStackSymbols);
|
||||
}
|
||||
}
|
||||
|
||||
// KVO callback method
|
||||
- (void)observeValueForKeyPath:(NSString *)keyPath
|
||||
ofObject:(id)object
|
||||
change:(NSDictionary<NSKeyValueChangeKey,id> *)change
|
||||
context:(void *)context {
|
||||
@try {
|
||||
if ([keyPath isEqualToString:@"AppIcon"]) {
|
||||
[self logMessage:@"AppIcon value changed in user defaults"];
|
||||
NSDockTile *dockTile = (__bridge NSDockTile *)context;
|
||||
[self updateAppIcon:dockTile];
|
||||
}
|
||||
} @catch (NSException *exception) {
|
||||
[self logMessage:[NSString stringWithFormat:@"Exception in KVO handler: %@\nStack trace: %@\nKeyPath: %@\nObject: %@\nChange: %@",
|
||||
exception.reason,
|
||||
exception.callStackSymbols,
|
||||
keyPath,
|
||||
object,
|
||||
change]];
|
||||
}
|
||||
}
|
||||
|
||||
@end
|
||||
Reference in New Issue
Block a user