platform.rs

   1use super::{
   2    event::key_to_native, screen::Screen, status_item::StatusItem, BoolExt as _, Dispatcher,
   3    FontSystem, Window,
   4};
   5use crate::{
   6    executor,
   7    keymap_matcher::KeymapMatcher,
   8    platform::{self, CursorStyle},
   9    Action, AppVersion, ClipboardItem, Event, Menu, MenuItem,
  10};
  11use anyhow::{anyhow, Result};
  12use block::ConcreteBlock;
  13use cocoa::{
  14    appkit::{
  15        NSApplication, NSApplicationActivationPolicy::NSApplicationActivationPolicyRegular,
  16        NSEventModifierFlags, NSMenu, NSMenuItem, NSModalResponse, NSOpenPanel, NSPasteboard,
  17        NSPasteboardTypeString, NSSavePanel, NSWindow,
  18    },
  19    base::{id, nil, selector, YES},
  20    foundation::{
  21        NSArray, NSAutoreleasePool, NSBundle, NSData, NSInteger, NSProcessInfo, NSString,
  22        NSUInteger, NSURL,
  23    },
  24};
  25use core_foundation::{
  26    base::{CFType, CFTypeRef, OSStatus, TCFType as _},
  27    boolean::CFBoolean,
  28    data::CFData,
  29    dictionary::{CFDictionary, CFDictionaryRef, CFMutableDictionary},
  30    string::{CFString, CFStringRef},
  31};
  32use ctor::ctor;
  33use objc::{
  34    class,
  35    declare::ClassDecl,
  36    msg_send,
  37    runtime::{Class, Object, Sel},
  38    sel, sel_impl,
  39};
  40use postage::oneshot;
  41use ptr::null_mut;
  42use std::{
  43    cell::{Cell, RefCell},
  44    convert::TryInto,
  45    ffi::{c_void, CStr, OsStr},
  46    os::{raw::c_char, unix::ffi::OsStrExt},
  47    path::{Path, PathBuf},
  48    ptr,
  49    rc::Rc,
  50    slice, str,
  51    sync::Arc,
  52};
  53use time::UtcOffset;
  54
  55#[allow(non_upper_case_globals)]
  56const NSUTF8StringEncoding: NSUInteger = 4;
  57
  58#[allow(non_upper_case_globals)]
  59pub const NSViewLayerContentsRedrawDuringViewResize: NSInteger = 2;
  60
  61const MAC_PLATFORM_IVAR: &str = "platform";
  62static mut APP_CLASS: *const Class = ptr::null();
  63static mut APP_DELEGATE_CLASS: *const Class = ptr::null();
  64
  65#[ctor]
  66unsafe fn build_classes() {
  67    APP_CLASS = {
  68        let mut decl = ClassDecl::new("GPUIApplication", class!(NSApplication)).unwrap();
  69        decl.add_ivar::<*mut c_void>(MAC_PLATFORM_IVAR);
  70        decl.add_method(
  71            sel!(sendEvent:),
  72            send_event as extern "C" fn(&mut Object, Sel, id),
  73        );
  74        decl.register()
  75    };
  76
  77    APP_DELEGATE_CLASS = {
  78        let mut decl = ClassDecl::new("GPUIApplicationDelegate", class!(NSResponder)).unwrap();
  79        decl.add_ivar::<*mut c_void>(MAC_PLATFORM_IVAR);
  80        decl.add_method(
  81            sel!(applicationDidFinishLaunching:),
  82            did_finish_launching as extern "C" fn(&mut Object, Sel, id),
  83        );
  84        decl.add_method(
  85            sel!(applicationDidBecomeActive:),
  86            did_become_active as extern "C" fn(&mut Object, Sel, id),
  87        );
  88        decl.add_method(
  89            sel!(applicationDidResignActive:),
  90            did_resign_active as extern "C" fn(&mut Object, Sel, id),
  91        );
  92        decl.add_method(
  93            sel!(applicationWillTerminate:),
  94            will_terminate as extern "C" fn(&mut Object, Sel, id),
  95        );
  96        decl.add_method(
  97            sel!(handleGPUIMenuItem:),
  98            handle_menu_item as extern "C" fn(&mut Object, Sel, id),
  99        );
 100        decl.add_method(
 101            sel!(validateMenuItem:),
 102            validate_menu_item as extern "C" fn(&mut Object, Sel, id) -> bool,
 103        );
 104        decl.add_method(
 105            sel!(menuWillOpen:),
 106            menu_will_open as extern "C" fn(&mut Object, Sel, id),
 107        );
 108        decl.add_method(
 109            sel!(application:openURLs:),
 110            open_urls as extern "C" fn(&mut Object, Sel, id, id),
 111        );
 112        decl.register()
 113    }
 114}
 115
 116#[derive(Default)]
 117pub struct MacForegroundPlatform(RefCell<MacForegroundPlatformState>);
 118
 119#[derive(Default)]
 120pub struct MacForegroundPlatformState {
 121    become_active: Option<Box<dyn FnMut()>>,
 122    resign_active: Option<Box<dyn FnMut()>>,
 123    quit: Option<Box<dyn FnMut()>>,
 124    event: Option<Box<dyn FnMut(crate::Event) -> bool>>,
 125    menu_command: Option<Box<dyn FnMut(&dyn Action)>>,
 126    validate_menu_command: Option<Box<dyn FnMut(&dyn Action) -> bool>>,
 127    will_open_menu: Option<Box<dyn FnMut()>>,
 128    open_urls: Option<Box<dyn FnMut(Vec<String>)>>,
 129    finish_launching: Option<Box<dyn FnOnce()>>,
 130    menu_actions: Vec<Box<dyn Action>>,
 131}
 132
 133impl MacForegroundPlatform {
 134    unsafe fn create_menu_bar(
 135        &self,
 136        menus: Vec<Menu>,
 137        delegate: id,
 138        actions: &mut Vec<Box<dyn Action>>,
 139        keystroke_matcher: &KeymapMatcher,
 140    ) -> id {
 141        let application_menu = NSMenu::new(nil).autorelease();
 142        application_menu.setDelegate_(delegate);
 143
 144        for menu_config in menus {
 145            let menu = NSMenu::new(nil).autorelease();
 146            menu.setTitle_(ns_string(menu_config.name));
 147            menu.setDelegate_(delegate);
 148
 149            for item_config in menu_config.items {
 150                menu.addItem_(self.create_menu_item(
 151                    item_config,
 152                    delegate,
 153                    actions,
 154                    keystroke_matcher,
 155                ));
 156            }
 157
 158            let menu_item = NSMenuItem::new(nil).autorelease();
 159            menu_item.setSubmenu_(menu);
 160            application_menu.addItem_(menu_item);
 161
 162            if menu_config.name == "Window" {
 163                let app: id = msg_send![APP_CLASS, sharedApplication];
 164                app.setWindowsMenu_(menu);
 165            }
 166        }
 167
 168        application_menu
 169    }
 170
 171    unsafe fn create_menu_item(
 172        &self,
 173        item: MenuItem,
 174        delegate: id,
 175        actions: &mut Vec<Box<dyn Action>>,
 176        keystroke_matcher: &KeymapMatcher,
 177    ) -> id {
 178        match item {
 179            MenuItem::Separator => NSMenuItem::separatorItem(nil),
 180            MenuItem::Action { name, action } => {
 181                let keystrokes = keystroke_matcher
 182                    .bindings_for_action_type(action.as_any().type_id())
 183                    .find(|binding| binding.action().eq(action.as_ref()))
 184                    .map(|binding| binding.keystrokes());
 185
 186                let item;
 187                if let Some(keystrokes) = keystrokes.flatten() {
 188                    if keystrokes.len() == 1 {
 189                        let keystroke = &keystrokes[0];
 190                        let mut mask = NSEventModifierFlags::empty();
 191                        for (modifier, flag) in &[
 192                            (keystroke.cmd, NSEventModifierFlags::NSCommandKeyMask),
 193                            (keystroke.ctrl, NSEventModifierFlags::NSControlKeyMask),
 194                            (keystroke.alt, NSEventModifierFlags::NSAlternateKeyMask),
 195                            (keystroke.shift, NSEventModifierFlags::NSShiftKeyMask),
 196                        ] {
 197                            if *modifier {
 198                                mask |= *flag;
 199                            }
 200                        }
 201
 202                        item = NSMenuItem::alloc(nil)
 203                            .initWithTitle_action_keyEquivalent_(
 204                                ns_string(name),
 205                                selector("handleGPUIMenuItem:"),
 206                                ns_string(key_to_native(&keystroke.key).as_ref()),
 207                            )
 208                            .autorelease();
 209                        item.setKeyEquivalentModifierMask_(mask);
 210                    }
 211                    // For multi-keystroke bindings, render the keystroke as part of the title.
 212                    else {
 213                        use std::fmt::Write;
 214
 215                        let mut name = format!("{name} [");
 216                        for (i, keystroke) in keystrokes.iter().enumerate() {
 217                            if i > 0 {
 218                                name.push(' ');
 219                            }
 220                            write!(&mut name, "{}", keystroke).unwrap();
 221                        }
 222                        name.push(']');
 223
 224                        item = NSMenuItem::alloc(nil)
 225                            .initWithTitle_action_keyEquivalent_(
 226                                ns_string(&name),
 227                                selector("handleGPUIMenuItem:"),
 228                                ns_string(""),
 229                            )
 230                            .autorelease();
 231                    }
 232                } else {
 233                    item = NSMenuItem::alloc(nil)
 234                        .initWithTitle_action_keyEquivalent_(
 235                            ns_string(name),
 236                            selector("handleGPUIMenuItem:"),
 237                            ns_string(""),
 238                        )
 239                        .autorelease();
 240                }
 241
 242                let tag = actions.len() as NSInteger;
 243                let _: () = msg_send![item, setTag: tag];
 244                actions.push(action);
 245                item
 246            }
 247            MenuItem::Submenu(Menu { name, items }) => {
 248                let item = NSMenuItem::new(nil).autorelease();
 249                let submenu = NSMenu::new(nil).autorelease();
 250                submenu.setDelegate_(delegate);
 251                for item in items {
 252                    submenu.addItem_(self.create_menu_item(
 253                        item,
 254                        delegate,
 255                        actions,
 256                        keystroke_matcher,
 257                    ));
 258                }
 259                item.setSubmenu_(submenu);
 260                item.setTitle_(ns_string(name));
 261                item
 262            }
 263        }
 264    }
 265}
 266
 267impl platform::ForegroundPlatform for MacForegroundPlatform {
 268    fn on_become_active(&self, callback: Box<dyn FnMut()>) {
 269        self.0.borrow_mut().become_active = Some(callback);
 270    }
 271
 272    fn on_resign_active(&self, callback: Box<dyn FnMut()>) {
 273        self.0.borrow_mut().resign_active = Some(callback);
 274    }
 275
 276    fn on_quit(&self, callback: Box<dyn FnMut()>) {
 277        self.0.borrow_mut().quit = Some(callback);
 278    }
 279
 280    fn on_event(&self, callback: Box<dyn FnMut(crate::Event) -> bool>) {
 281        self.0.borrow_mut().event = Some(callback);
 282    }
 283
 284    fn on_open_urls(&self, callback: Box<dyn FnMut(Vec<String>)>) {
 285        self.0.borrow_mut().open_urls = Some(callback);
 286    }
 287
 288    fn run(&self, on_finish_launching: Box<dyn FnOnce()>) {
 289        self.0.borrow_mut().finish_launching = Some(on_finish_launching);
 290
 291        unsafe {
 292            let app: id = msg_send![APP_CLASS, sharedApplication];
 293            let app_delegate: id = msg_send![APP_DELEGATE_CLASS, new];
 294            app.setDelegate_(app_delegate);
 295
 296            let self_ptr = self as *const Self as *const c_void;
 297            (*app).set_ivar(MAC_PLATFORM_IVAR, self_ptr);
 298            (*app_delegate).set_ivar(MAC_PLATFORM_IVAR, self_ptr);
 299
 300            let pool = NSAutoreleasePool::new(nil);
 301            app.run();
 302            pool.drain();
 303
 304            (*app).set_ivar(MAC_PLATFORM_IVAR, null_mut::<c_void>());
 305            (*app.delegate()).set_ivar(MAC_PLATFORM_IVAR, null_mut::<c_void>());
 306        }
 307    }
 308
 309    fn on_menu_command(&self, callback: Box<dyn FnMut(&dyn Action)>) {
 310        self.0.borrow_mut().menu_command = Some(callback);
 311    }
 312
 313    fn on_will_open_menu(&self, callback: Box<dyn FnMut()>) {
 314        self.0.borrow_mut().will_open_menu = Some(callback);
 315    }
 316
 317    fn on_validate_menu_command(&self, callback: Box<dyn FnMut(&dyn Action) -> bool>) {
 318        self.0.borrow_mut().validate_menu_command = Some(callback);
 319    }
 320
 321    fn set_menus(&self, menus: Vec<Menu>, keystroke_matcher: &KeymapMatcher) {
 322        unsafe {
 323            let app: id = msg_send![APP_CLASS, sharedApplication];
 324            let mut state = self.0.borrow_mut();
 325            let actions = &mut state.menu_actions;
 326            app.setMainMenu_(self.create_menu_bar(
 327                menus,
 328                app.delegate(),
 329                actions,
 330                keystroke_matcher,
 331            ));
 332        }
 333    }
 334
 335    fn prompt_for_paths(
 336        &self,
 337        options: platform::PathPromptOptions,
 338    ) -> oneshot::Receiver<Option<Vec<PathBuf>>> {
 339        unsafe {
 340            let panel = NSOpenPanel::openPanel(nil);
 341            panel.setCanChooseDirectories_(options.directories.to_objc());
 342            panel.setCanChooseFiles_(options.files.to_objc());
 343            panel.setAllowsMultipleSelection_(options.multiple.to_objc());
 344            panel.setResolvesAliases_(false.to_objc());
 345            let (done_tx, done_rx) = oneshot::channel();
 346            let done_tx = Cell::new(Some(done_tx));
 347            let block = ConcreteBlock::new(move |response: NSModalResponse| {
 348                let result = if response == NSModalResponse::NSModalResponseOk {
 349                    let mut result = Vec::new();
 350                    let urls = panel.URLs();
 351                    for i in 0..urls.count() {
 352                        let url = urls.objectAtIndex(i);
 353                        if url.isFileURL() == YES {
 354                            if let Ok(path) = ns_url_to_path(url) {
 355                                result.push(path)
 356                            }
 357                        }
 358                    }
 359                    Some(result)
 360                } else {
 361                    None
 362                };
 363
 364                if let Some(mut done_tx) = done_tx.take() {
 365                    let _ = postage::sink::Sink::try_send(&mut done_tx, result);
 366                }
 367            });
 368            let block = block.copy();
 369            let _: () = msg_send![panel, beginWithCompletionHandler: block];
 370            done_rx
 371        }
 372    }
 373
 374    fn prompt_for_new_path(&self, directory: &Path) -> oneshot::Receiver<Option<PathBuf>> {
 375        unsafe {
 376            let panel = NSSavePanel::savePanel(nil);
 377            let path = ns_string(directory.to_string_lossy().as_ref());
 378            let url = NSURL::fileURLWithPath_isDirectory_(nil, path, true.to_objc());
 379            panel.setDirectoryURL(url);
 380
 381            let (done_tx, done_rx) = oneshot::channel();
 382            let done_tx = Cell::new(Some(done_tx));
 383            let block = ConcreteBlock::new(move |response: NSModalResponse| {
 384                let mut result = None;
 385                if response == NSModalResponse::NSModalResponseOk {
 386                    let url = panel.URL();
 387                    if url.isFileURL() == YES {
 388                        result = ns_url_to_path(panel.URL()).ok()
 389                    }
 390                }
 391
 392                if let Some(mut done_tx) = done_tx.take() {
 393                    let _ = postage::sink::Sink::try_send(&mut done_tx, result);
 394                }
 395            });
 396            let block = block.copy();
 397            let _: () = msg_send![panel, beginWithCompletionHandler: block];
 398            done_rx
 399        }
 400    }
 401}
 402
 403pub struct MacPlatform {
 404    dispatcher: Arc<Dispatcher>,
 405    fonts: Arc<FontSystem>,
 406    pasteboard: id,
 407    text_hash_pasteboard_type: id,
 408    metadata_pasteboard_type: id,
 409}
 410
 411impl MacPlatform {
 412    pub fn new() -> Self {
 413        Self {
 414            dispatcher: Arc::new(Dispatcher),
 415            fonts: Arc::new(FontSystem::new()),
 416            pasteboard: unsafe { NSPasteboard::generalPasteboard(nil) },
 417            text_hash_pasteboard_type: unsafe { ns_string("zed-text-hash") },
 418            metadata_pasteboard_type: unsafe { ns_string("zed-metadata") },
 419        }
 420    }
 421
 422    unsafe fn read_from_pasteboard(&self, kind: id) -> Option<&[u8]> {
 423        let data = self.pasteboard.dataForType(kind);
 424        if data == nil {
 425            None
 426        } else {
 427            Some(slice::from_raw_parts(
 428                data.bytes() as *mut u8,
 429                data.length() as usize,
 430            ))
 431        }
 432    }
 433}
 434
 435unsafe impl Send for MacPlatform {}
 436unsafe impl Sync for MacPlatform {}
 437
 438impl platform::Platform for MacPlatform {
 439    fn dispatcher(&self) -> Arc<dyn platform::Dispatcher> {
 440        self.dispatcher.clone()
 441    }
 442
 443    fn activate(&self, ignoring_other_apps: bool) {
 444        unsafe {
 445            let app = NSApplication::sharedApplication(nil);
 446            app.activateIgnoringOtherApps_(ignoring_other_apps.to_objc());
 447        }
 448    }
 449
 450    fn hide(&self) {
 451        unsafe {
 452            let app = NSApplication::sharedApplication(nil);
 453            let _: () = msg_send![app, hide: nil];
 454        }
 455    }
 456
 457    fn hide_other_apps(&self) {
 458        unsafe {
 459            let app = NSApplication::sharedApplication(nil);
 460            let _: () = msg_send![app, hideOtherApplications: nil];
 461        }
 462    }
 463
 464    fn unhide_other_apps(&self) {
 465        unsafe {
 466            let app = NSApplication::sharedApplication(nil);
 467            let _: () = msg_send![app, unhideAllApplications: nil];
 468        }
 469    }
 470
 471    fn quit(&self) {
 472        // Quitting the app causes us to close windows, which invokes `Window::on_close` callbacks
 473        // synchronously before this method terminates. If we call `Platform::quit` while holding a
 474        // borrow of the app state (which most of the time we will do), we will end up
 475        // double-borrowing the app state in the `on_close` callbacks for our open windows. To solve
 476        // this, we make quitting the application asynchronous so that we aren't holding borrows to
 477        // the app state on the stack when we actually terminate the app.
 478
 479        use super::dispatcher::{dispatch_async_f, dispatch_get_main_queue};
 480
 481        unsafe {
 482            dispatch_async_f(dispatch_get_main_queue(), ptr::null_mut(), Some(quit));
 483        }
 484
 485        unsafe extern "C" fn quit(_: *mut c_void) {
 486            let app = NSApplication::sharedApplication(nil);
 487            let _: () = msg_send![app, terminate: nil];
 488        }
 489    }
 490
 491    fn screens(&self) -> Vec<Rc<dyn platform::Screen>> {
 492        Screen::all()
 493            .into_iter()
 494            .map(|screen| Rc::new(screen) as Rc<_>)
 495            .collect()
 496    }
 497
 498    fn open_window(
 499        &self,
 500        id: usize,
 501        options: platform::WindowOptions,
 502        executor: Rc<executor::Foreground>,
 503    ) -> Box<dyn platform::Window> {
 504        Box::new(Window::open(id, options, executor, self.fonts()))
 505    }
 506
 507    fn key_window_id(&self) -> Option<usize> {
 508        Window::key_window_id()
 509    }
 510
 511    fn add_status_item(&self) -> Box<dyn platform::Window> {
 512        Box::new(StatusItem::add(self.fonts()))
 513    }
 514
 515    fn fonts(&self) -> Arc<dyn platform::FontSystem> {
 516        self.fonts.clone()
 517    }
 518
 519    fn write_to_clipboard(&self, item: ClipboardItem) {
 520        unsafe {
 521            self.pasteboard.clearContents();
 522
 523            let text_bytes = NSData::dataWithBytes_length_(
 524                nil,
 525                item.text.as_ptr() as *const c_void,
 526                item.text.len() as u64,
 527            );
 528            self.pasteboard
 529                .setData_forType(text_bytes, NSPasteboardTypeString);
 530
 531            if let Some(metadata) = item.metadata.as_ref() {
 532                let hash_bytes = ClipboardItem::text_hash(&item.text).to_be_bytes();
 533                let hash_bytes = NSData::dataWithBytes_length_(
 534                    nil,
 535                    hash_bytes.as_ptr() as *const c_void,
 536                    hash_bytes.len() as u64,
 537                );
 538                self.pasteboard
 539                    .setData_forType(hash_bytes, self.text_hash_pasteboard_type);
 540
 541                let metadata_bytes = NSData::dataWithBytes_length_(
 542                    nil,
 543                    metadata.as_ptr() as *const c_void,
 544                    metadata.len() as u64,
 545                );
 546                self.pasteboard
 547                    .setData_forType(metadata_bytes, self.metadata_pasteboard_type);
 548            }
 549        }
 550    }
 551
 552    fn read_from_clipboard(&self) -> Option<ClipboardItem> {
 553        unsafe {
 554            if let Some(text_bytes) = self.read_from_pasteboard(NSPasteboardTypeString) {
 555                let text = String::from_utf8_lossy(text_bytes).to_string();
 556                let hash_bytes = self
 557                    .read_from_pasteboard(self.text_hash_pasteboard_type)
 558                    .and_then(|bytes| bytes.try_into().ok())
 559                    .map(u64::from_be_bytes);
 560                let metadata_bytes = self
 561                    .read_from_pasteboard(self.metadata_pasteboard_type)
 562                    .and_then(|bytes| String::from_utf8(bytes.to_vec()).ok());
 563
 564                if let Some((hash, metadata)) = hash_bytes.zip(metadata_bytes) {
 565                    if hash == ClipboardItem::text_hash(&text) {
 566                        Some(ClipboardItem {
 567                            text,
 568                            metadata: Some(metadata),
 569                        })
 570                    } else {
 571                        Some(ClipboardItem {
 572                            text,
 573                            metadata: None,
 574                        })
 575                    }
 576                } else {
 577                    Some(ClipboardItem {
 578                        text,
 579                        metadata: None,
 580                    })
 581                }
 582            } else {
 583                None
 584            }
 585        }
 586    }
 587
 588    fn open_url(&self, url: &str) {
 589        unsafe {
 590            let url = NSURL::alloc(nil)
 591                .initWithString_(ns_string(url))
 592                .autorelease();
 593            let workspace: id = msg_send![class!(NSWorkspace), sharedWorkspace];
 594            msg_send![workspace, openURL: url]
 595        }
 596    }
 597
 598    fn write_credentials(&self, url: &str, username: &str, password: &[u8]) -> Result<()> {
 599        let url = CFString::from(url);
 600        let username = CFString::from(username);
 601        let password = CFData::from_buffer(password);
 602
 603        unsafe {
 604            use security::*;
 605
 606            // First, check if there are already credentials for the given server. If so, then
 607            // update the username and password.
 608            let mut verb = "updating";
 609            let mut query_attrs = CFMutableDictionary::with_capacity(2);
 610            query_attrs.set(kSecClass as *const _, kSecClassInternetPassword as *const _);
 611            query_attrs.set(kSecAttrServer as *const _, url.as_CFTypeRef());
 612
 613            let mut attrs = CFMutableDictionary::with_capacity(4);
 614            attrs.set(kSecClass as *const _, kSecClassInternetPassword as *const _);
 615            attrs.set(kSecAttrServer as *const _, url.as_CFTypeRef());
 616            attrs.set(kSecAttrAccount as *const _, username.as_CFTypeRef());
 617            attrs.set(kSecValueData as *const _, password.as_CFTypeRef());
 618
 619            let mut status = SecItemUpdate(
 620                query_attrs.as_concrete_TypeRef(),
 621                attrs.as_concrete_TypeRef(),
 622            );
 623
 624            // If there were no existing credentials for the given server, then create them.
 625            if status == errSecItemNotFound {
 626                verb = "creating";
 627                status = SecItemAdd(attrs.as_concrete_TypeRef(), ptr::null_mut());
 628            }
 629
 630            if status != errSecSuccess {
 631                return Err(anyhow!("{} password failed: {}", verb, status));
 632            }
 633        }
 634        Ok(())
 635    }
 636
 637    fn read_credentials(&self, url: &str) -> Result<Option<(String, Vec<u8>)>> {
 638        let url = CFString::from(url);
 639        let cf_true = CFBoolean::true_value().as_CFTypeRef();
 640
 641        unsafe {
 642            use security::*;
 643
 644            // Find any credentials for the given server URL.
 645            let mut attrs = CFMutableDictionary::with_capacity(5);
 646            attrs.set(kSecClass as *const _, kSecClassInternetPassword as *const _);
 647            attrs.set(kSecAttrServer as *const _, url.as_CFTypeRef());
 648            attrs.set(kSecReturnAttributes as *const _, cf_true);
 649            attrs.set(kSecReturnData as *const _, cf_true);
 650
 651            let mut result = CFTypeRef::from(ptr::null());
 652            let status = SecItemCopyMatching(attrs.as_concrete_TypeRef(), &mut result);
 653            match status {
 654                security::errSecSuccess => {}
 655                security::errSecItemNotFound | security::errSecUserCanceled => return Ok(None),
 656                _ => return Err(anyhow!("reading password failed: {}", status)),
 657            }
 658
 659            let result = CFType::wrap_under_create_rule(result)
 660                .downcast::<CFDictionary>()
 661                .ok_or_else(|| anyhow!("keychain item was not a dictionary"))?;
 662            let username = result
 663                .find(kSecAttrAccount as *const _)
 664                .ok_or_else(|| anyhow!("account was missing from keychain item"))?;
 665            let username = CFType::wrap_under_get_rule(*username)
 666                .downcast::<CFString>()
 667                .ok_or_else(|| anyhow!("account was not a string"))?;
 668            let password = result
 669                .find(kSecValueData as *const _)
 670                .ok_or_else(|| anyhow!("password was missing from keychain item"))?;
 671            let password = CFType::wrap_under_get_rule(*password)
 672                .downcast::<CFData>()
 673                .ok_or_else(|| anyhow!("password was not a string"))?;
 674
 675            Ok(Some((username.to_string(), password.bytes().to_vec())))
 676        }
 677    }
 678
 679    fn delete_credentials(&self, url: &str) -> Result<()> {
 680        let url = CFString::from(url);
 681
 682        unsafe {
 683            use security::*;
 684
 685            let mut query_attrs = CFMutableDictionary::with_capacity(2);
 686            query_attrs.set(kSecClass as *const _, kSecClassInternetPassword as *const _);
 687            query_attrs.set(kSecAttrServer as *const _, url.as_CFTypeRef());
 688
 689            let status = SecItemDelete(query_attrs.as_concrete_TypeRef());
 690
 691            if status != errSecSuccess {
 692                return Err(anyhow!("delete password failed: {}", status));
 693            }
 694        }
 695        Ok(())
 696    }
 697
 698    fn set_cursor_style(&self, style: CursorStyle) {
 699        unsafe {
 700            let cursor: id = match style {
 701                CursorStyle::Arrow => msg_send![class!(NSCursor), arrowCursor],
 702                CursorStyle::ResizeLeftRight => {
 703                    msg_send![class!(NSCursor), resizeLeftRightCursor]
 704                }
 705                CursorStyle::ResizeUpDown => msg_send![class!(NSCursor), resizeUpDownCursor],
 706                CursorStyle::PointingHand => msg_send![class!(NSCursor), pointingHandCursor],
 707                CursorStyle::IBeam => msg_send![class!(NSCursor), IBeamCursor],
 708            };
 709            let _: () = msg_send![cursor, set];
 710        }
 711    }
 712
 713    fn should_auto_hide_scrollbars(&self) -> bool {
 714        #[allow(non_upper_case_globals)]
 715        const NSScrollerStyleOverlay: NSInteger = 1;
 716
 717        unsafe {
 718            let style: NSInteger = msg_send![class!(NSScroller), preferredScrollerStyle];
 719            style == NSScrollerStyleOverlay
 720        }
 721    }
 722
 723    fn local_timezone(&self) -> UtcOffset {
 724        unsafe {
 725            let local_timezone: id = msg_send![class!(NSTimeZone), localTimeZone];
 726            let seconds_from_gmt: NSInteger = msg_send![local_timezone, secondsFromGMT];
 727            UtcOffset::from_whole_seconds(seconds_from_gmt.try_into().unwrap()).unwrap()
 728        }
 729    }
 730
 731    fn path_for_auxiliary_executable(&self, name: &str) -> Result<PathBuf> {
 732        unsafe {
 733            let bundle: id = NSBundle::mainBundle();
 734            if bundle.is_null() {
 735                Err(anyhow!("app is not running inside a bundle"))
 736            } else {
 737                let name = ns_string(name);
 738                let url: id = msg_send![bundle, URLForAuxiliaryExecutable: name];
 739                if url.is_null() {
 740                    Err(anyhow!("resource not found"))
 741                } else {
 742                    ns_url_to_path(url)
 743                }
 744            }
 745        }
 746    }
 747
 748    fn app_path(&self) -> Result<PathBuf> {
 749        unsafe {
 750            let bundle: id = NSBundle::mainBundle();
 751            if bundle.is_null() {
 752                Err(anyhow!("app is not running inside a bundle"))
 753            } else {
 754                Ok(path_from_objc(msg_send![bundle, bundlePath]))
 755            }
 756        }
 757    }
 758
 759    fn app_version(&self) -> Result<platform::AppVersion> {
 760        unsafe {
 761            let bundle: id = NSBundle::mainBundle();
 762            if bundle.is_null() {
 763                Err(anyhow!("app is not running inside a bundle"))
 764            } else {
 765                let version: id = msg_send![bundle, objectForInfoDictionaryKey: ns_string("CFBundleShortVersionString")];
 766                let len = msg_send![version, lengthOfBytesUsingEncoding: NSUTF8StringEncoding];
 767                let bytes = version.UTF8String() as *const u8;
 768                let version = str::from_utf8(slice::from_raw_parts(bytes, len)).unwrap();
 769                version.parse()
 770            }
 771        }
 772    }
 773
 774    fn os_name(&self) -> &'static str {
 775        "macOS"
 776    }
 777
 778    fn os_version(&self) -> Result<crate::AppVersion> {
 779        unsafe {
 780            let process_info = NSProcessInfo::processInfo(nil);
 781            let version = process_info.operatingSystemVersion();
 782            Ok(AppVersion {
 783                major: version.majorVersion as usize,
 784                minor: version.minorVersion as usize,
 785                patch: version.patchVersion as usize,
 786            })
 787        }
 788    }
 789}
 790
 791unsafe fn path_from_objc(path: id) -> PathBuf {
 792    let len = msg_send![path, lengthOfBytesUsingEncoding: NSUTF8StringEncoding];
 793    let bytes = path.UTF8String() as *const u8;
 794    let path = str::from_utf8(slice::from_raw_parts(bytes, len)).unwrap();
 795    PathBuf::from(path)
 796}
 797
 798unsafe fn get_foreground_platform(object: &mut Object) -> &MacForegroundPlatform {
 799    let platform_ptr: *mut c_void = *object.get_ivar(MAC_PLATFORM_IVAR);
 800    assert!(!platform_ptr.is_null());
 801    &*(platform_ptr as *const MacForegroundPlatform)
 802}
 803
 804extern "C" fn send_event(this: &mut Object, _sel: Sel, native_event: id) {
 805    unsafe {
 806        if let Some(event) = Event::from_native(native_event, None) {
 807            let platform = get_foreground_platform(this);
 808            if let Some(callback) = platform.0.borrow_mut().event.as_mut() {
 809                if callback(event) {
 810                    return;
 811                }
 812            }
 813        }
 814
 815        msg_send![super(this, class!(NSApplication)), sendEvent: native_event]
 816    }
 817}
 818
 819extern "C" fn did_finish_launching(this: &mut Object, _: Sel, _: id) {
 820    unsafe {
 821        let app: id = msg_send![APP_CLASS, sharedApplication];
 822        app.setActivationPolicy_(NSApplicationActivationPolicyRegular);
 823
 824        let platform = get_foreground_platform(this);
 825        let callback = platform.0.borrow_mut().finish_launching.take();
 826        if let Some(callback) = callback {
 827            callback();
 828        }
 829    }
 830}
 831
 832extern "C" fn did_become_active(this: &mut Object, _: Sel, _: id) {
 833    let platform = unsafe { get_foreground_platform(this) };
 834    if let Some(callback) = platform.0.borrow_mut().become_active.as_mut() {
 835        callback();
 836    }
 837}
 838
 839extern "C" fn did_resign_active(this: &mut Object, _: Sel, _: id) {
 840    let platform = unsafe { get_foreground_platform(this) };
 841    if let Some(callback) = platform.0.borrow_mut().resign_active.as_mut() {
 842        callback();
 843    }
 844}
 845
 846extern "C" fn will_terminate(this: &mut Object, _: Sel, _: id) {
 847    let platform = unsafe { get_foreground_platform(this) };
 848    if let Some(callback) = platform.0.borrow_mut().quit.as_mut() {
 849        callback();
 850    }
 851}
 852
 853extern "C" fn open_urls(this: &mut Object, _: Sel, _: id, urls: id) {
 854    let urls = unsafe {
 855        (0..urls.count())
 856            .into_iter()
 857            .filter_map(|i| {
 858                let url = urls.objectAtIndex(i);
 859                match CStr::from_ptr(url.absoluteString().UTF8String() as *mut c_char).to_str() {
 860                    Ok(string) => Some(string.to_string()),
 861                    Err(err) => {
 862                        log::error!("error converting path to string: {}", err);
 863                        None
 864                    }
 865                }
 866            })
 867            .collect::<Vec<_>>()
 868    };
 869    let platform = unsafe { get_foreground_platform(this) };
 870    if let Some(callback) = platform.0.borrow_mut().open_urls.as_mut() {
 871        callback(urls);
 872    }
 873}
 874
 875extern "C" fn handle_menu_item(this: &mut Object, _: Sel, item: id) {
 876    unsafe {
 877        let platform = get_foreground_platform(this);
 878        let mut platform = platform.0.borrow_mut();
 879        if let Some(mut callback) = platform.menu_command.take() {
 880            let tag: NSInteger = msg_send![item, tag];
 881            let index = tag as usize;
 882            if let Some(action) = platform.menu_actions.get(index) {
 883                callback(action.as_ref());
 884            }
 885            platform.menu_command = Some(callback);
 886        }
 887    }
 888}
 889
 890extern "C" fn validate_menu_item(this: &mut Object, _: Sel, item: id) -> bool {
 891    unsafe {
 892        let mut result = false;
 893        let platform = get_foreground_platform(this);
 894        let mut platform = platform.0.borrow_mut();
 895        if let Some(mut callback) = platform.validate_menu_command.take() {
 896            let tag: NSInteger = msg_send![item, tag];
 897            let index = tag as usize;
 898            if let Some(action) = platform.menu_actions.get(index) {
 899                result = callback(action.as_ref());
 900            }
 901            platform.validate_menu_command = Some(callback);
 902        }
 903        result
 904    }
 905}
 906
 907extern "C" fn menu_will_open(this: &mut Object, _: Sel, _: id) {
 908    unsafe {
 909        let platform = get_foreground_platform(this);
 910        let mut platform = platform.0.borrow_mut();
 911        if let Some(mut callback) = platform.will_open_menu.take() {
 912            callback();
 913            platform.will_open_menu = Some(callback);
 914        }
 915    }
 916}
 917
 918unsafe fn ns_string(string: &str) -> id {
 919    NSString::alloc(nil).init_str(string).autorelease()
 920}
 921
 922unsafe fn ns_url_to_path(url: id) -> Result<PathBuf> {
 923    let path: *mut c_char = msg_send![url, fileSystemRepresentation];
 924    if path.is_null() {
 925        Err(anyhow!(
 926            "url is not a file path: {}",
 927            CStr::from_ptr(url.absoluteString().UTF8String()).to_string_lossy()
 928        ))
 929    } else {
 930        Ok(PathBuf::from(OsStr::from_bytes(
 931            CStr::from_ptr(path).to_bytes(),
 932        )))
 933    }
 934}
 935
 936mod security {
 937    #![allow(non_upper_case_globals)]
 938    use super::*;
 939
 940    #[link(name = "Security", kind = "framework")]
 941    extern "C" {
 942        pub static kSecClass: CFStringRef;
 943        pub static kSecClassInternetPassword: CFStringRef;
 944        pub static kSecAttrServer: CFStringRef;
 945        pub static kSecAttrAccount: CFStringRef;
 946        pub static kSecValueData: CFStringRef;
 947        pub static kSecReturnAttributes: CFStringRef;
 948        pub static kSecReturnData: CFStringRef;
 949
 950        pub fn SecItemAdd(attributes: CFDictionaryRef, result: *mut CFTypeRef) -> OSStatus;
 951        pub fn SecItemUpdate(query: CFDictionaryRef, attributes: CFDictionaryRef) -> OSStatus;
 952        pub fn SecItemDelete(query: CFDictionaryRef) -> OSStatus;
 953        pub fn SecItemCopyMatching(query: CFDictionaryRef, result: *mut CFTypeRef) -> OSStatus;
 954    }
 955
 956    pub const errSecSuccess: OSStatus = 0;
 957    pub const errSecUserCanceled: OSStatus = -128;
 958    pub const errSecItemNotFound: OSStatus = -25300;
 959}
 960
 961#[cfg(test)]
 962mod tests {
 963    use crate::platform::Platform;
 964
 965    use super::*;
 966
 967    #[test]
 968    fn test_clipboard() {
 969        let platform = build_platform();
 970        assert_eq!(platform.read_from_clipboard(), None);
 971
 972        let item = ClipboardItem::new("1".to_string());
 973        platform.write_to_clipboard(item.clone());
 974        assert_eq!(platform.read_from_clipboard(), Some(item));
 975
 976        let item = ClipboardItem::new("2".to_string()).with_metadata(vec![3, 4]);
 977        platform.write_to_clipboard(item.clone());
 978        assert_eq!(platform.read_from_clipboard(), Some(item));
 979
 980        let text_from_other_app = "text from other app";
 981        unsafe {
 982            let bytes = NSData::dataWithBytes_length_(
 983                nil,
 984                text_from_other_app.as_ptr() as *const c_void,
 985                text_from_other_app.len() as u64,
 986            );
 987            platform
 988                .pasteboard
 989                .setData_forType(bytes, NSPasteboardTypeString);
 990        }
 991        assert_eq!(
 992            platform.read_from_clipboard(),
 993            Some(ClipboardItem::new(text_from_other_app.to_string()))
 994        );
 995    }
 996
 997    fn build_platform() -> MacPlatform {
 998        let mut platform = MacPlatform::new();
 999        platform.pasteboard = unsafe { NSPasteboard::pasteboardWithUniqueName(nil) };
1000        platform
1001    }
1002}