SIMBL swizzling in finder - objective-c

Question

I manages to integrate Icon Overlay like dropbox in Mac Os Finder with a SIMBL plugin !
I Use the swizzle method to override some finder function.
Here is my swizzle method :
void PluginSwizzleInstanceMethod(Class cls, SEL oldSel, SEL newSel)
{
Method mnew = class_getInstanceMethod(cls, newSel);
Method mold = class_getInstanceMethod(cls, oldSel);
method_exchangeImplementations(mnew, mold);
}
So for the moment i override the drawWithFrame method like this :
Class cls = NSClassFromString(#"TIconAndTextCell");
SEL oldSel = #selector(drawWithFrame:inView:);
SEL newSel = #selector(PluginDrawWithFrame:inView:);
PluginSwizzleInstanceMethod(cls, oldSel, newSel);
all my icons are displayed but not on my desktop or on the view that show items as icon ...
Why ?
thank you ,

Answer 1

I just managed to do some stuff for the Desktop, so I'll post in case that's still relevant for someone:
So as you experienced, Finder draws icons differently on the Desktop.
One class that is responsible for that is TDesktopIcon.
One can, for example, change icons by overriding the drawIconInContext: method, and setting the thumbnailImage property to whatever NSImage* he likes.
- (void) FO_drawIconInContext:(struct CGContext *)arg1 {
NSImage *image;
// set image...
[(TDesktopIcon*)self setThumbnailImage:image];
[self FO_drawIconInContext:arg1];
}
Getting the url for each item was not as straight forward as with TIconAndTextCell or IKImageBrowserCell.
In order to get the URLs, you need to override the prepareToDrawNode: method of TDesktopViewController.
You get the URL from arg1 as follows:
- (void) FO_prepareToDrawNode:(const struct TFENode *)arg1 {
struct OpaqueNodeRef *opr = arg1->fNodeRef;
struct FINode *fiNode = [FINode nodeFromNodeRef:opr];
NSURL *url = [fiNode previewItemURL];
// save url somewhere (I use NSUserDefaults)
[self FO_prepareToDrawNode:arg1];
}
Maybe there's a better way, but that's what I figured out...

Answer 2

Got it! (THIS CODE ONLY WORKS FOR 10.7 THOUGH) I was having issues because the icon was being drawn before the final image was rendered to the thumbnail image.... so I got these artifacts where the original icon was missing....but the overlay was always there...it was the order of draw operations I had wrong...!
I even have the name for the target node... you need the TFENodeHelper class to translate the TFENode into a filePath...
//
// DesktopViewIconOverlay.h
// FinderIconOverlayExample
//
// Created by Orbitus007 on 2/20/13.
//
//
#import <Foundation/Foundation.h>
#import <Quartz/Quartz.h>
#import "Finder.h"
#interface DesktopViewIconOverlay : NSObject
+ (void)pluginLoad;
- (void) FO_prepareToDrawNode:(const struct TFENode *)arg1;
- (void) FO_drawIconInContext:(struct CGContext *)arg1;
#end
//
// DesktopViewIconOverlay.m
// FinderIconOverlayExample
//
// Created by Orbitus007 on 2/20/13.
//
//
#import "DesktopViewIconOverlay.h"
#import "FinderIconOverlayExample.h"
#include <objc/objc.h>
#include <objc/runtime.h>
#import <Quartz/Quartz.h>
#import "TFENodeHelper.h"
static TFENodeHelper *gNodeHelper;
#implementation DesktopViewIconOverlay
+ (void)pluginLoad
{
// Create helper object
gNodeHelper = [[TFENodeHelper alloc] init];
if (gNodeHelper == nil) {
NSLog(#"Failed to instantiate 'TFENodeHelper' class");
return;
}
Method old, new;
Class self_class = [self class];
Class finder_class = [objc_getClass("TDesktopIcon") class];
class_addMethod(finder_class, #selector(FO_drawIconInContext:),
class_getMethodImplementation(self_class, #selector(FO_drawIconInContext:)),"v#:#");
old = class_getInstanceMethod(finder_class, #selector(drawIconInContext:));
new = class_getInstanceMethod(finder_class, #selector(FO_drawIconInContext:));
method_exchangeImplementations(old, new);
finder_class = [objc_getClass("TDesktopViewController") class];
class_addMethod(finder_class, #selector(FO_prepareToDrawNode:),
class_getMethodImplementation(self_class, #selector(FO_prepareToDrawNode:)),"v#:#");
old = class_getInstanceMethod(finder_class, #selector(prepareToDrawNode:));
new = class_getInstanceMethod(finder_class, #selector(FO_prepareToDrawNode:));
method_exchangeImplementations(old, new);
}
- (void) FO_prepareToDrawNode:(const struct TFENode *)arg1 {
NSString *path = [gNodeHelper pathForNode:arg1];
NSLog(#"Path = %#", path);
[[NSUserDefaults standardUserDefaults] setValue:path forKey:#"TDesktopIconURL"];
//struct OpaqueNodeRef *opr = arg1->fNodeRef;
//NSLog(#"path = %#", [[FINode nodeWithFENode:arg1] fullPath]);
//id fiNode = [FINode nodeFromNodeRef:opr];
//NSURL *url = [fiNode previewItemURL];
//[[NSUserDefaults standardUserDefaults] setValue:[url path] forKey:#"TDesktopIconURL"];
//NSLog(#"sending ", arg1->fNodeRef);
// save url somewhere (I use NSUserDefaults)
[self FO_prepareToDrawNode:arg1];
}
- (void) FO_drawIconInContext:(struct CGContext *)arg1
{
[self FO_drawIconInContext:arg1];
NSString *iconPath = [[NSUserDefaults standardUserDefaults] valueForKey:#"TDesktopIconURL"];
NSLog(#"recieved %#", iconPath);
if (![iconPath isEqualToString:#"/Users/h0xff/Desktop/Restart Finder.app"])
return;
NSString *imagePath = #"/Users/h0xff/Desktop/020513_icons/128_synced_green.png";
NSImage *overlay = [[NSImage alloc] initWithContentsOfFile:imagePath];
NSImage *mainImage = [(TDesktopIcon*)self thumbnailImage];
float width = 256.0;
float height = 256.0;
NSImage *finalImage = [[NSImage alloc] initWithSize:NSMakeSize(width, height)];
[finalImage lockFocus];
// draw the base image
[mainImage drawInRect:NSMakeRect(0, 0, width, height)
fromRect:NSZeroRect operation:NSCompositeSourceOver fraction:1.0];
// draw the overlay image at some offset point
[overlay drawInRect:NSMakeRect(0, 0, [overlay size].width/1.5, [overlay size].height/1.5)
fromRect:NSZeroRect operation:NSCompositeSourceOver fraction:1.0];
[finalImage unlockFocus];
// set image...
[(TDesktopIcon*)self setThumbnailImage:finalImage];
[self FO_drawIconInContext:arg1];
}
#end
This code was derived from Alexey Zhuchkov and many others.... please ask questions if you have any!
//
// TFENodeHelper.h
// FinderMenu
//
// Helper class to get full path from TFENode
//
// Created by Alexey Zhuchkov on 11/4/12.
// Copyright (c) 2012 InfiniteLabs. All rights reserved.
//
#import <Foundation/Foundation.h>
// Forward declarations
struct TFENode;
typedef enum {
TFENodeCtorTypeUnknown,
TFENodeCtorTypeNodeWithFENode, // nodeWithFENode
TFENodeCtorTypeNodeFromNodeRef, // nodeFromNodeRef
} TFENodeCtorType;
typedef enum {
TFENodePathMethodTypeUnknown,
TFENodePathMethodTypeFullPath, // fullPath
TFENodePathMethodTypePreviewItemURL, // previewItemURL
} TFENodePathMethodType;
#interface TFENodeHelper : NSObject {
#private
Class _class;
TFENodeCtorType _ctorType;
TFENodePathMethodType _pathMethodType;
}
- (NSString *)pathForNode:(const struct TFENode *)node;
#end
//
// TFENodeHelper.m
// FinderMenu
//
// Created by Alexey Zhuchkov on 11/4/12.
// Copyright (c) 2012 InfiniteLabs. All rights reserved.
//
#import "TFENodeHelper.h"
#import <objc/runtime.h>
#import "Finder.h"
#implementation TFENodeHelper
- (id)init {
self = [super init];
if (self) {
_class = NSClassFromString(#"FINode");
_ctorType = TFENodeCtorTypeUnknown;
_pathMethodType = TFENodePathMethodTypeUnknown;
if (_class) {
if (class_getClassMethod(_class, #selector(nodeWithFENode:))) {
_ctorType = TFENodeCtorTypeNodeWithFENode;
} else if (class_getClassMethod(_class, #selector(nodeFromNodeRef:))) {
_ctorType = TFENodeCtorTypeNodeFromNodeRef;
}
if (class_getInstanceMethod(_class, #selector(fullPath))) {
_pathMethodType = TFENodePathMethodTypeFullPath;
} else if (class_getInstanceMethod(_class, #selector(previewItemURL))) {
_pathMethodType = TFENodePathMethodTypePreviewItemURL;
}
}
if (!_class
|| (_ctorType == TFENodePathMethodTypeUnknown)
|| (_pathMethodType == TFENodePathMethodTypeUnknown)) {
[self release];
return nil;
}
}
return self;
}
- (NSString *)pathForNode:(const struct TFENode *)node {
FINode *fiNode = nil;
NSString *path = nil;
switch (_ctorType) {
case TFENodeCtorTypeNodeWithFENode:
fiNode = [_class nodeWithFENode:node];
break;
case TFENodeCtorTypeNodeFromNodeRef:
fiNode = [_class nodeFromNodeRef:node->fNodeRef];
break;
default:
break;
}
NSURL *url;
if (fiNode) {
switch (_pathMethodType) {
case TFENodePathMethodTypeFullPath:
path = [fiNode fullPath];
break;
case TFENodePathMethodTypePreviewItemURL:
url = [fiNode previewItemURL];
path = [url path];
default:
break;
}
}
return path;
}
#end