5
0
mirror of https://github.com/wailsapp/wails.git synced 2025-05-04 07:43:11 +08:00
wails/v3/pkg/application/systemtray_darwin.m
Lea Anthony beacf06c7d
Fix macOS systray DPI scaling and offset.
Added new systray examples.
2024-12-13 23:46:27 +11:00

245 lines
8.5 KiB
Objective-C

//go:build darwin
#include "Cocoa/Cocoa.h"
#include "menuitem_darwin.h"
#include "systemtray_darwin.h"
extern void systrayClickCallback(long, int);
// StatusItemController.m
@implementation StatusItemController
- (void)statusItemClicked:(id)sender {
NSEvent *event = [NSApp currentEvent];
systrayClickCallback(self.id, event.type);
}
@end
// Create a new system tray
void* systemTrayNew(long id) {
StatusItemController *controller = [[StatusItemController alloc] init];
controller.id = id;
NSStatusItem *statusItem = [[[NSStatusBar systemStatusBar] statusItemWithLength:NSVariableStatusItemLength] retain];
[statusItem setTarget:controller];
[statusItem setAction:@selector(statusItemClicked:)];
NSButton *button = statusItem.button;
[button sendActionOn:(NSEventMaskLeftMouseDown|NSEventMaskRightMouseDown)];
return (void*)statusItem;
}
void systemTraySetLabel(void* nsStatusItem, char *label) {
if( label == NULL ) {
return;
}
// Set the label on the main thread
dispatch_async(dispatch_get_main_queue(), ^{
NSStatusItem *statusItem = (NSStatusItem *)nsStatusItem;
statusItem.button.title = [NSString stringWithUTF8String:label];
free(label);
});
}
void systemTraySetANSILabel(void* nsStatusItem, void* label) {
if( label == NULL ) {
return;
}
NSMutableAttributedString* attributedString = (NSMutableAttributedString*) label;
// Set the label
NSStatusItem *statusItem = (NSStatusItem *)nsStatusItem;
[statusItem setAttributedTitle:attributedString];
// [attributedString release];
}
void* appendAttributedString(void *currentString, char *title, char *FG, char *BG) {
NSMutableAttributedString* newString = createAttributedString(title, FG, BG);
if( currentString != NULL ) {
NSMutableAttributedString* current = (NSMutableAttributedString*)currentString;
[current appendAttributedString:newString];
newString = current;
}
return (void*)newString;
}
void* createAttributedString(char *title, char *FG, char *BG) {
NSMutableDictionary *dictionary = [NSMutableDictionary new];
// RGBA
if(FG != NULL && strlen(FG) > 0) {
unsigned short r, g, b, a;
// white by default
r = g = b = a = 255;
int count = sscanf(FG, "#%02hx%02hx%02hx%02hx", &r, &g, &b, &a);
if (count > 0) {
NSColor *colour = [NSColor colorWithCalibratedRed:(CGFloat)r / 255.0
green:(CGFloat)g / 255.0
blue:(CGFloat)b / 255.0
alpha:(CGFloat)a / 255.0];
dictionary[NSForegroundColorAttributeName] = colour;
}
}
// Calculate BG colour
if(BG != NULL && strlen(BG) > 0) {
unsigned short r, g, b, a;
// white by default
r = g = b = a = 255;
int count = sscanf(BG, "#%02hx%02hx%02hx%02hx", &r, &g, &b, &a);
if (count > 0) {
NSColor *colour = [NSColor colorWithCalibratedRed:(CGFloat)r / 255.0
green:(CGFloat)g / 255.0
blue:(CGFloat)b / 255.0
alpha:(CGFloat)a / 255.0];
dictionary[NSBackgroundColorAttributeName] = colour;
}
}
NSMutableAttributedString *attributedString = [[NSMutableAttributedString alloc] initWithString:[NSString stringWithUTF8String:title] attributes:dictionary];
return (void*)attributedString;
}
// Create an nsimage from a byte array
NSImage* imageFromBytes(const unsigned char *bytes, int length) {
NSData *data = [NSData dataWithBytes:bytes length:length];
NSImage *image = [[NSImage alloc] initWithData:data];
return image;
}
// Set the icon on the system tray
void systemTraySetIcon(void* nsStatusItem, void* nsImage, int position, bool isTemplate) {
// Set the icon on the main thread
dispatch_async(dispatch_get_main_queue(), ^{
NSStatusItem *statusItem = (NSStatusItem *)nsStatusItem;
NSImage *image = (NSImage *)nsImage;
NSStatusBar *statusBar = [NSStatusBar systemStatusBar];
CGFloat thickness = [statusBar thickness];
[image setSize:NSMakeSize(thickness, thickness)];
if( isTemplate ) {
[image setTemplate:YES];
}
statusItem.button.image = [image autorelease];
statusItem.button.imagePosition = position;
});
}
// Destroy system tray
void systemTrayDestroy(void* nsStatusItem) {
// Remove the status item from the status bar and its associated menu
dispatch_async(dispatch_get_main_queue(), ^{
NSStatusItem *statusItem = (NSStatusItem *)nsStatusItem;
[[NSStatusBar systemStatusBar] removeStatusItem:statusItem];
[statusItem release];
});
}
void showMenu(void* nsStatusItem, void *nsMenu) {
// Show the menu on the main thread
dispatch_async(dispatch_get_main_queue(), ^{
NSStatusItem *statusItem = (NSStatusItem *)nsStatusItem;
[statusItem popUpStatusItemMenu:(NSMenu *)nsMenu];
// Post a mouse up event so the statusitem defocuses
NSEvent *event = [NSEvent mouseEventWithType:NSEventTypeLeftMouseUp
location:[NSEvent mouseLocation]
modifierFlags:0
timestamp:[[NSProcessInfo processInfo] systemUptime]
windowNumber:0
context:nil
eventNumber:0
clickCount:1
pressure:1];
[NSApp postEvent:event atStart:NO];
[statusItem.button highlight:NO];
});
}
void systemTrayGetBounds(void* nsStatusItem, NSRect *rect, void **outScreen) {
NSStatusItem *statusItem = (NSStatusItem *)nsStatusItem;
NSStatusBarButton *button = statusItem.button;
// Get mouse location and find the screen it's on
NSPoint mouseLocation = [NSEvent mouseLocation];
NSScreen *screen = nil;
NSArray *screens = [NSScreen screens];
for (NSScreen *candidate in screens) {
NSRect frame = [candidate frame];
if (NSPointInRect(mouseLocation, frame)) {
screen = candidate;
break;
}
}
if (!screen) {
screen = [NSScreen mainScreen];
}
// Get button frame in screen coordinates
NSRect buttonFrame = button.frame;
NSRect buttonFrameScreen = [button.window convertRectToScreen:buttonFrame];
*rect = buttonFrameScreen;
*outScreen = (void*)screen;
}
NSRect NSScreen_frame(void* screen) {
return [(NSScreen*)screen frame];
}
int statusBarHeight() {
NSMenu *mainMenu = [NSApp mainMenu];
CGFloat menuBarHeight = [mainMenu menuBarHeight];
return (int)menuBarHeight;
}
void systemTrayPositionWindow(void* nsStatusItem, void* nsWindow, int offset) {
// Get the status item's button
NSStatusBarButton *button = [(NSStatusItem*)nsStatusItem button];
// Get the frame in screen coordinates
NSRect frame = [button.window convertRectToScreen:button.frame];
// Get the screen that contains the status item
NSScreen *screen = [button.window screen];
if (screen == nil) {
screen = [NSScreen mainScreen];
}
// Get screen's backing scale factor (DPI)
CGFloat scaleFactor = [screen backingScaleFactor];
// Get the window's frame
NSRect windowFrame = [(NSWindow*)nsWindow frame];
// Calculate the horizontal position (centered under the status item)
CGFloat windowX = frame.origin.x + (frame.size.width - windowFrame.size.width) / 2;
// If the window would go off the right edge of the screen, adjust it
if (windowX + windowFrame.size.width > screen.frame.origin.x + screen.frame.size.width) {
windowX = screen.frame.origin.x + screen.frame.size.width - windowFrame.size.width;
}
// If the window would go off the left edge of the screen, adjust it
if (windowX < screen.frame.origin.x) {
windowX = screen.frame.origin.x;
}
// Get screen metrics
NSRect screenFrame = [screen frame];
NSRect visibleFrame = [screen visibleFrame];
// Calculate the vertical position
CGFloat scaledOffset = offset * scaleFactor;
CGFloat windowY = visibleFrame.origin.y + visibleFrame.size.height - windowFrame.size.height - scaledOffset;
// Set the window's frame
windowFrame.origin.x = windowX;
windowFrame.origin.y = windowY;
[(NSWindow*)nsWindow setFrame:windowFrame display:YES animate:NO];
}