/* HBQueueController This file is part of the HandBrake source code. Homepage: . It may be used under the terms of the GNU General Public License. */ #import "HBQueueController.h" #import "HBQueueItem.h" #import "HBController.h" #import "HBAppDelegate.h" #import "HBTableView.h" #import "HBQueueItemView.h" #import "NSArray+HBAdditions.h" #import "HBUtilities.h" #import "HBDockTile.h" #import "HBOutputRedirect.h" #import "HBJobOutputFileWriter.h" #import "HBPreferencesController.h" @import HandBrakeKit; // Pasteboard type for or drag operations #define DragDropSimplePboardType @"HBQueueCustomTableViewPboardType" // DockTile update frequency in total percent increment #define dockTileUpdateFrequency 0.1f static void *HBControllerQueueCoreContext = &HBControllerQueueCoreContext; @interface HBQueueController () /// Whether the window is visible or occluded, /// useful to avoid updating the UI needlessly @property (nonatomic) BOOL visible; // Progress @property (nonatomic, strong) NSString *progressInfo; @property (nonatomic, readonly) HBDockTile *dockTile; @property (nonatomic, readwrite) double dockIconProgress; @property (unsafe_unretained) IBOutlet NSTextField *progressTextField; @property (unsafe_unretained) IBOutlet NSTextField *countTextField; @property (unsafe_unretained) IBOutlet HBTableView *tableView; @property (nonatomic) IBOutlet NSToolbarItem *ripToolbarItem; @property (nonatomic) IBOutlet NSToolbarItem *pauseToolbarItem; @property (nonatomic) NSTableCellView *dummyCell; @property (nonatomic) NSLayoutConstraint *dummyCellWidth; @property (nonatomic, readonly) HBDistributedArray *items; @property (nonatomic) HBQueueItem *currentItem; @property (nonatomic) HBJobOutputFileWriter *currentLog; @property (nonatomic, readwrite) BOOL stop; @property (nonatomic, readwrite) NSUInteger pendingItemsCount; @property (nonatomic, readwrite) NSUInteger completedItemsCount; @property (nonatomic) NSArray *dragNodesArray; @end @interface HBQueueController (TouchBar) - (void)_touchBar_updateButtonsStateForQueueCore:(HBState)state; - (void)_touchBar_validateUserInterfaceItems; @end @implementation HBQueueController - (instancetype)initWithURL:(NSURL *)queueURL; { NSParameterAssert(queueURL); if (self = [super initWithWindowNibName:@"Queue"]) { // Load the dockTile and instiante initial text fields _dockTile = [[HBDockTile alloc] initWithDockTile:[[NSApplication sharedApplication] dockTile] image:[[NSApplication sharedApplication] applicationIconImage]]; NSInteger loggingLevel = [[NSUserDefaults standardUserDefaults] integerForKey:@"LoggingLevel"]; // Init a separate instance of libhb for the queue _core = [[HBCore alloc] initWithLogLevel:loggingLevel name:@"QueueCore"]; _core.automaticallyPreventSleep = NO; // Progress _progressInfo = @""; // Load the queue from disk. _items = [[HBDistributedArray alloc] initWithURL:queueURL class:[HBQueueItem class]]; [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(reloadQueue) name:HBDistributedArrayChanged object:_items]; [NSUserNotificationCenter defaultUserNotificationCenter].delegate = self; } return self; } - (void)dealloc { [[NSNotificationCenter defaultCenter] removeObserver:self]; } - (void)windowDidLoad { if (@available (macOS 10.12, *)) { self.window.tabbingMode = NSWindowTabbingModeDisallowed; } // lets setup our queue list table view for drag and drop here [self.tableView registerForDraggedTypes:@[DragDropSimplePboardType]]; [self.tableView setDraggingSourceOperationMask:NSDragOperationEvery forLocal:YES]; [self.tableView setVerticalMotionCanBeginDrag:YES]; [self updateQueueStats]; [self.core addObserver:self forKeyPath:@"state" options:NSKeyValueObservingOptionNew | NSKeyValueObservingOptionInitial context:HBControllerQueueCoreContext]; [self addObserver:self forKeyPath:@"pendingItemsCount" options:NSKeyValueObservingOptionNew | NSKeyValueObservingOptionInitial context:HBControllerQueueCoreContext]; } - (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context { if (context == HBControllerQueueCoreContext) { HBState state = self.core.state; [self updateToolbarButtonsStateForQueueCore:state]; [self.window.toolbar validateVisibleItems]; if (@available(macOS 10.12.2, *)) { [self _touchBar_updateButtonsStateForQueueCore:state]; [self _touchBar_validateUserInterfaceItems]; } } else { [super observeValueForKeyPath:keyPath ofObject:object change:change context:context]; } } #pragma mark Toolbar - (void)updateToolbarButtonsStateForQueueCore:(HBState)state { if (state == HBStatePaused) { _pauseToolbarItem.image = [NSImage imageNamed: @"encode"]; _pauseToolbarItem.label = NSLocalizedString(@"Resume", @"Toolbar Pause Item"); _pauseToolbarItem.toolTip = NSLocalizedString(@"Resume Encoding", @"Toolbar Pause Item"); } else { _pauseToolbarItem.image = [NSImage imageNamed:@"pauseencode"]; _pauseToolbarItem.label = NSLocalizedString(@"Pause", @"Toolbar Pause Item"); _pauseToolbarItem.toolTip = NSLocalizedString(@"Pause Encoding", @"Toolbar Pause Item"); } if (state == HBStateScanning || state == HBStateWorking || state == HBStateSearching || state == HBStateMuxing || state == HBStatePaused) { _ripToolbarItem.image = [NSImage imageNamed:@"stopencode"]; _ripToolbarItem.label = NSLocalizedString(@"Stop", @"Toolbar Start/Stop Item"); _ripToolbarItem.toolTip = NSLocalizedString(@"Stop Encoding", @"Toolbar Start/Stop Item"); } else { _ripToolbarItem.image = [NSImage imageNamed: @"encode"]; _ripToolbarItem.label = NSLocalizedString(@"Start", @"Toolbar Start/Stop Item"); _pauseToolbarItem.toolTip = NSLocalizedString(@"Start Encoding", @"Toolbar Start/Stop Item"); } } - (BOOL)validateMenuItem:(NSMenuItem *)menuItem { SEL action = menuItem.action; if (action == @selector(rip:)) { if (self.core.state == HBStateIdle) { menuItem.title = NSLocalizedString(@"Start Encoding", @"Queue -> start/stop menu"); return (self.pendingItemsCount > 0); } else if (self.core.state != HBStateIdle) { menuItem.title = NSLocalizedString(@"Stop Encoding", @"Queue -> start/stop menu"); return YES; } } if (action == @selector(pause:)) { if (self.core.state != HBStatePaused) { menuItem.title = NSLocalizedString(@"Pause Encoding", @"Queue -> pause/resume menu"); } else { menuItem.title = NSLocalizedString(@"Resume Encoding", @"Queue -> pause/resume men"); } return (self.core.state == HBStateWorking || self.core.state == HBStatePaused); } if (action == @selector(editSelectedQueueItem:) || action == @selector(removeSelectedQueueItem:) || action == @selector(revealSelectedQueueItems:) || action == @selector(revealSelectedQueueItemsSources:)) { return (self.tableView.selectedRow != -1 || self.tableView.clickedRow != -1); } if (action == @selector(resetJobState:)) { return self.tableView.targetedRowIndexes.count > 0; } if (action == @selector(clearAll:)) { return self.items.count > 0; } if (action == @selector(clearCompleted:)) { return self.completedItemsCount > 0; } return YES; } - (BOOL)validateUserIterfaceItemForAction:(SEL)action { HBState s = self.core.state; if (action == @selector(toggleStartCancel:)) { if ((s == HBStateScanning) || (s == HBStatePaused) || (s == HBStateWorking) || (s == HBStateMuxing)) { return YES; } else { return (self.pendingItemsCount > 0); } } if (action == @selector(togglePauseResume:)) { if (s == HBStatePaused) { return YES; } else { return (s == HBStateWorking || s == HBStateMuxing); } } return NO; } - (BOOL)validateToolbarItem:(NSToolbarItem *)theItem { SEL action = theItem.action; return [self validateUserIterfaceItemForAction:action]; } #pragma mark - Public methods - (void)addJob:(HBJob *)item { NSParameterAssert(item); [self addJobsFromArray:@[item]]; } - (void)addJobsFromArray:(NSArray *)jobs; { NSParameterAssert(jobs); NSMutableArray *itemsToAdd = [NSMutableArray array]; for (HBJob *job in jobs) { HBQueueItem *item = [[HBQueueItem alloc] initWithJob:job]; [itemsToAdd addObject:item]; } if (itemsToAdd.count) { [self addQueueItems:itemsToAdd]; } } - (BOOL)jobExistAtURL:(NSURL *)url { NSParameterAssert(url); for (HBQueueItem *item in self.items) { if ((item.state == HBQueueItemStateReady || item.state == HBQueueItemStateWorking) && [item.completeOutputURL isEqualTo:url]) { return YES; } } return NO; } - (NSUInteger)count { return self.items.count; } /** * This method will clear the queue of any encodes that are not still pending * this includes both successfully completed encodes as well as canceled encodes */ - (void)removeCompletedJobs { [self.items beginTransaction]; NSIndexSet *indexes = [self.items indexesOfObjectsUsingBlock:^BOOL(HBQueueItem *item) { return (item.state == HBQueueItemStateCompleted || item.state == HBQueueItemStateCanceled); }]; [self removeQueueItemsAtIndexes:indexes]; [self.items commit]; } /** * This method will clear the queue of all encodes. effectively creating an empty queue */ - (void)removeAllJobs { [self.items beginTransaction]; [self removeQueueItemsAtIndexes:[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, self.items.count)]]; [self.items commit]; } /** * This method will set any item marked as encoding back to pending * currently used right after a queue reload */ - (void)setEncodingJobsAsPending { [self.items beginTransaction]; NSMutableIndexSet *indexes = [NSMutableIndexSet indexSet]; NSUInteger idx = 0; for (HBQueueItem *item in self.items) { // We want to keep any queue item that is pending or was previously being encoded if (item.state == HBQueueItemStateWorking) { item.state = HBQueueItemStateReady; [indexes addIndex:idx]; } idx++; } [self reloadQueueItemsAtIndexes:indexes]; [self.items commit]; } #pragma mark - Private queue editing methods /** * Reloads the queue, this is called * when another HandBrake instances modifies the queue */ - (void)reloadQueue { [self updateQueueStats]; [self.tableView reloadData]; [self.window.undoManager removeAllActions]; } - (void)reloadQueueItemAtIndex:(NSUInteger)idx { [self reloadQueueItemsAtIndexes:[NSIndexSet indexSetWithIndex:idx]]; } - (void)reloadQueueItemsAtIndexes:(NSIndexSet *)indexes { NSIndexSet *columnIndexes = [NSIndexSet indexSetWithIndex:0]; [self.tableView reloadDataForRowIndexes:indexes columnIndexes:columnIndexes]; [self updateQueueStats]; } - (void)addQueueItems:(NSArray *)items { NSParameterAssert(items); NSIndexSet *indexes = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(self.items.count, items.count)]; [self addQueueItems:items atIndexes:indexes]; } - (void)addQueueItems:(NSArray *)items atIndexes:(NSIndexSet *)indexes { NSParameterAssert(items); NSParameterAssert(indexes); [self.items beginTransaction]; [self.tableView beginUpdates]; // Forward NSUInteger currentIndex = indexes.firstIndex; NSUInteger currentObjectIndex = 0; while (currentIndex != NSNotFound) { [self.items insertObject:items[currentObjectIndex] atIndex:currentIndex]; currentIndex = [indexes indexGreaterThanIndex:currentIndex]; currentObjectIndex++; } [self.tableView insertRowsAtIndexes:indexes withAnimation:NSTableViewAnimationSlideDown]; NSUndoManager *undo = self.window.undoManager; [[undo prepareWithInvocationTarget:self] removeQueueItemsAtIndexes:indexes]; if (!undo.isUndoing) { if (items.count == 1) { [undo setActionName:NSLocalizedString(@"Add Job To Queue", @"Queue undo action name")]; } else { [undo setActionName:NSLocalizedString(@"Add Jobs To Queue", @"Queue undo action name")]; } } [self.tableView endUpdates]; [self updateQueueStats]; [self.items commit]; } - (void)removeQueueItemAtIndex:(NSUInteger)index { [self removeQueueItemsAtIndexes:[NSIndexSet indexSetWithIndex:index]]; } - (void)removeQueueItemsAtIndexes:(NSIndexSet *)indexes { NSParameterAssert(indexes); if (indexes.count == 0) { return; } [self.items beginTransaction]; [self.tableView beginUpdates]; NSArray *removeItems = [self.items objectsAtIndexes:indexes]; if (self.items.count > indexes.lastIndex) { [self.items removeObjectsAtIndexes:indexes]; } [self.tableView removeRowsAtIndexes:indexes withAnimation:NSTableViewAnimationSlideUp]; [self.tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:indexes.firstIndex] byExtendingSelection:NO]; NSUndoManager *undo = self.window.undoManager; [[undo prepareWithInvocationTarget:self] addQueueItems:removeItems atIndexes:indexes]; if (!undo.isUndoing) { if (indexes.count == 1) { [undo setActionName:NSLocalizedString(@"Remove Job From Queue", @"Queue undo action name")]; } else { [undo setActionName:NSLocalizedString(@"Remove Jobs From Queue", @"Queue undo action name")]; } } [self.tableView endUpdates]; [self updateQueueStats]; [self.items commit]; } - (void)moveQueueItems:(NSArray *)items toIndex:(NSUInteger)index { [self.items beginTransaction]; [self.tableView beginUpdates]; NSMutableArray *source = [NSMutableArray array]; NSMutableArray *dest = [NSMutableArray array]; for (id object in items.reverseObjectEnumerator) { NSUInteger sourceIndex = [self.items indexOfObject:object]; [self.items removeObjectAtIndex:sourceIndex]; if (sourceIndex < index) { index--; } [self.items insertObject:object atIndex:index]; [source addObject:@(index)]; [dest addObject:@(sourceIndex)]; [self.tableView moveRowAtIndex:sourceIndex toIndex:index]; } NSUndoManager *undo = self.window.undoManager; [[undo prepareWithInvocationTarget:self] moveQueueItemsAtIndexes:source toIndexes:dest]; if (!undo.isUndoing) { if (items.count == 1) { [undo setActionName:NSLocalizedString(@"Move Job in Queue", @"Queue undo action name")]; } else { [undo setActionName:NSLocalizedString(@"Move Jobs in Queue", @"Queue undo action name")]; } } [self.tableView endUpdates]; [self.items commit]; } - (void)moveQueueItemsAtIndexes:(NSArray *)source toIndexes:(NSArray *)dest { [self.items beginTransaction]; [self.tableView beginUpdates]; NSMutableArray *newSource = [NSMutableArray array]; NSMutableArray *newDest = [NSMutableArray array]; for (NSInteger idx = source.count - 1; idx >= 0; idx--) { NSUInteger sourceIndex = [source[idx] integerValue]; NSUInteger destIndex = [dest[idx] integerValue]; [newSource addObject:@(destIndex)]; [newDest addObject:@(sourceIndex)]; id obj = [self.items objectAtIndex:sourceIndex]; [self.items removeObjectAtIndex:sourceIndex]; [self.items insertObject:obj atIndex:destIndex]; [self.tableView moveRowAtIndex:sourceIndex toIndex:destIndex]; } NSUndoManager *undo = self.window.undoManager; [[undo prepareWithInvocationTarget:self] moveQueueItemsAtIndexes:newSource toIndexes:newDest]; if (!undo.isUndoing) { if (source.count == 1) { [undo setActionName:NSLocalizedString(@"Move Job in Queue", @"Queue undo action name")]; } else { [undo setActionName:NSLocalizedString(@"Move Jobs in Queue", @"Queue undo action name")]; } } [self.tableView endUpdates]; [self.items commit]; } - (void)windowDidChangeOcclusionState:(NSNotification *)notification { if ([self.window occlusionState] & NSWindowOcclusionStateVisible) { self.visible = YES; self.progressTextField.stringValue = self.progressInfo; } else { self.visible = NO; } } - (void)updateProgress:(NSString *)info progress:(double)progress hidden:(BOOL)hidden { self.progressInfo = info; if (self.visible) { self.progressTextField.stringValue = info; } [self.controller setQueueInfo:info progress:progress hidden:hidden]; } /** * Updates the queue status label. */ - (void)updateQueueStats { // lets get the stats on the status of the queue array NSUInteger pendingCount = 0; NSUInteger completedCount = 0; for (HBQueueItem *item in self.items) { if (item.state == HBQueueItemStateReady) { pendingCount++; } if (item.state == HBQueueItemStateCompleted) { completedCount++; } } NSString *string; if (pendingCount == 0) { string = NSLocalizedString(@"No encode pending", @"Queue status"); } else if (pendingCount == 1) { string = [NSString stringWithFormat: NSLocalizedString(@"%d encode pending", @"Queue status"), pendingCount]; } else { string = [NSString stringWithFormat: NSLocalizedString(@"%d encodes pending", @"Queue status"), pendingCount]; } self.countTextField.stringValue = string; self.pendingItemsCount = pendingCount; self.completedItemsCount = completedCount; } #pragma mark - Queue Job Processing - (BOOL)_isDiskSpaceLowAtURL:(NSURL *)url { if ([[NSUserDefaults standardUserDefaults] boolForKey:@"HBQueuePauseIfLowSpace"]) { NSURL *volumeURL = nil; NSDictionary *attrs = [url resourceValuesForKeys:@[NSURLIsVolumeKey, NSURLVolumeURLKey] error:NULL]; long long minCapacity = [[[NSUserDefaults standardUserDefaults] stringForKey:@"HBQueueMinFreeSpace"] longLongValue] * 1000000000; volumeURL = [attrs[NSURLIsVolumeKey] boolValue] ? url : attrs[NSURLVolumeURLKey]; if (volumeURL) { [volumeURL removeCachedResourceValueForKey:NSURLVolumeAvailableCapacityKey]; attrs = [volumeURL resourceValuesForKeys:@[NSURLVolumeAvailableCapacityKey] error:NULL]; if (attrs[NSURLVolumeAvailableCapacityKey]) { if ([attrs[NSURLVolumeAvailableCapacityKey] longLongValue] < minCapacity) { return YES; } } } } return NO; } /** * Used to get the next pending queue item and return it if found */ - (HBQueueItem *)getNextPendingQueueItem { for (HBQueueItem *item in self.items) { if (item.state == HBQueueItemStateReady) { return item; } } return nil; } /** * Starts the queue */ - (void)encodeNextQueueItem { [self.items beginTransaction]; self.currentItem = nil; // since we have completed an encode, we go to the next if (self.stop) { [HBUtilities writeToActivityLog:"Queue manually stopped"]; self.stop = NO; [self.core allowSleep]; } else { // Check to see if there are any more pending items in the queue HBQueueItem *nextItem = [self getNextPendingQueueItem]; if (nextItem && [self _isDiskSpaceLowAtURL:nextItem.outputURL]) { // Disk space is low, show an alert [HBUtilities writeToActivityLog:"Queue Stopped, low space on destination disk"]; [self queueLowDiskSpaceAlert]; } // If we still have more pending items in our queue, lets go to the next one else if (nextItem) { // now we mark the queue item as working so another instance can not come along and try to scan it while we are scanning nextItem.state = HBQueueItemStateWorking; // Tell HB to output a new activity log file for this encode self.currentLog = [[HBJobOutputFileWriter alloc] initWithJob:nextItem.job]; if (self.currentLog) { [[HBOutputRedirect stderrRedirect] addListener:self.currentLog]; [[HBOutputRedirect stdoutRedirect] addListener:self.currentLog]; } self.currentItem = nextItem; [self reloadQueueItemAtIndex:[self.items indexOfObject:nextItem]]; // now we can go ahead and scan the new pending queue item [self encodeItem:nextItem]; // erase undo manager history [self.window.undoManager removeAllActions]; } else { [HBUtilities writeToActivityLog:"Queue Done, there are no more pending encodes"]; // Since there are no more items to encode, go to queueCompletedAlerts // for user specified alerts after queue completed [self queueCompletedAlerts]; [self.core allowSleep]; } } [self.items commit]; } - (void)completedItem:(HBQueueItem *)item result:(HBCoreResult)result; { NSParameterAssert(item); [self.items beginTransaction]; // Since we are done with this encode, tell output to stop writing to the // individual encode log. [[HBOutputRedirect stderrRedirect] removeListener:self.currentLog]; [[HBOutputRedirect stdoutRedirect] removeListener:self.currentLog]; self.currentLog = nil; // Check to see if the encode state has not been canceled // to determine if we should send it to external app. if (result != HBCoreResultCanceled) { // Send to tagger [self sendToExternalApp:item]; } // Mark the encode just finished switch (result) { case HBCoreResultDone: item.state = HBQueueItemStateCompleted; break; case HBCoreResultCanceled: item.state = HBQueueItemStateCanceled; break; default: item.state = HBQueueItemStateFailed; break; } if ([self.items containsObject:item]) { [self reloadQueueItemAtIndex:[self.items indexOfObject:item]]; } [self.window.toolbar validateVisibleItems]; [self.items commit]; // Update UI NSString *info = nil; switch (result) { case HBCoreResultDone: info = NSLocalizedString(@"Encode Finished.", @"Queue status"); [self itemCompletedAlerts:item result:result]; break; case HBCoreResultCanceled: info = NSLocalizedString(@"Encode Canceled.", @"Queue status"); break; default: info = NSLocalizedString(@"Encode Failed.", @"Queue status"); [self itemCompletedAlerts:item result:result]; break; } [self updateProgress:info progress:1.0 hidden:YES]; // Restore dock icon [self.dockTile updateDockIcon:-1.0 withETA:@""]; self.dockIconProgress = 0; } /** * Here we actually tell hb_scan to perform the source scan, using the path to source and title number */ - (void)encodeItem:(HBQueueItem *)item { NSParameterAssert(item); // Progress handler void (^progressHandler)(HBState state, HBProgress progress, NSString *info) = ^(HBState state, HBProgress progress, NSString *info) { [self updateProgress:info progress:0 hidden:NO]; }; // Completion handler void (^completionHandler)(HBCoreResult result) = ^(HBCoreResult result) { if (result == HBCoreResultDone) { [self realEncodeItem:item]; } else { [self completedItem:item result:result]; [self encodeNextQueueItem]; } }; // Only scan 10 previews before an encode - additional previews are // only useful for autocrop and static previews, which are already taken care of at this point [self.core scanURL:item.fileURL titleIndex:item.job.titleIdx previews:10 minDuration:0 progressHandler:progressHandler completionHandler:completionHandler]; } /** * This assumes that we have re-scanned and loaded up a new queue item to send to libhb */ - (void)realEncodeItem:(HBQueueItem *)item { NSParameterAssert(item); HBJob *job = item.job; // Reset the title in the job. job.title = self.core.titles.firstObject; NSParameterAssert(job); HBStateFormatter *formatter = [[HBStateFormatter alloc] init]; formatter.title = job.outputFileName; self.core.stateFormatter = formatter; // Progress handler void (^progressHandler)(HBState state, HBProgress progress, NSString *info) = ^(HBState state, HBProgress progress, NSString *info) { if (state == HBStateWorking) { // Update dock icon if (self.dockIconProgress < 100.0 * progress.percent) { [self.dockTile updateDockIcon:progress.percent hours:progress.hours minutes:progress.minutes seconds:progress.seconds]; self.dockIconProgress += dockTileUpdateFrequency; } } else if (state == HBStateMuxing) { [self.dockTile updateDockIcon:1.0 withETA:@""]; } // Update UI [self updateProgress:info progress:progress.percent hidden:NO]; }; // Completion handler void (^completionHandler)(HBCoreResult result) = ^(HBCoreResult result) { [self completedItem:item result:result]; [self encodeNextQueueItem]; }; // We should be all setup so let 'er rip [self.core encodeJob:job progressHandler:progressHandler completionHandler:completionHandler]; // We are done using the title, remove it from the job job.title = nil; } /** * Cancels the current job */ - (void)doCancelCurrentItem { if (self.core.state == HBStateScanning) { [self.core cancelScan]; } else { [self.core cancelEncode]; } } /** * Cancels the current job and starts processing the next in queue. */ - (void)cancelCurrentItemAndContinue { [self doCancelCurrentItem]; } /** * Cancels the current job and stops libhb from processing the remaining encodes. */ - (void)cancelCurrentItemAndStop { self.stop = YES; [self doCancelCurrentItem]; } /** * Finishes the current job and stops libhb from processing the remaining encodes. */ - (void)finishCurrentAndStop { self.stop = YES; } #pragma mark - Encode Done Actions - (void)userNotificationCenter:(NSUserNotificationCenter *)center didActivateNotification:(NSUserNotification *)notification { // Show the file in Finder when a done notification was clicked. NSString *path = notification.userInfo[@"Path"]; if ([path isKindOfClass:[NSString class]] && path.length) { NSURL *fileURL = [NSURL fileURLWithPath:path]; [[NSWorkspace sharedWorkspace] activateFileViewerSelectingURLs:@[fileURL]]; } } - (void)showNotificationWithTitle:(NSString *)title description:(NSString *)description url:(NSURL *)fileURL playSound:(BOOL)playSound { NSUserNotification *notification = [[NSUserNotification alloc] init]; notification.title = title; notification.informativeText = description; notification.soundName = playSound ? NSUserNotificationDefaultSoundName : nil; notification.hasActionButton = YES; notification.actionButtonTitle = NSLocalizedString(@"Show", @"Notification -> Show in Finder"); notification.userInfo = @{ @"Path": fileURL.path }; [[NSUserNotificationCenter defaultUserNotificationCenter] deliverNotification:notification]; } /** * Sends the URL to the external app * selected in the preferences. * * @param job the job of the file to send */ - (void)sendToExternalApp:(HBQueueItem *)item { // This end of encode action is called as each encode rolls off of the queue if ([[NSUserDefaults standardUserDefaults] boolForKey:@"HBSendToAppEnabled"] == YES) { #ifdef __SANDBOX_ENABLED__ BOOL accessingSecurityScopedResource = [item.outputURL startAccessingSecurityScopedResource]; #endif NSWorkspace *workspace = [NSWorkspace sharedWorkspace]; NSString *app = [workspace fullPathForApplication:[[NSUserDefaults standardUserDefaults] objectForKey:@"HBSendToApp"]]; if (app) { if (![workspace openFile:item.completeOutputURL.path withApplication:app]) { [HBUtilities writeToActivityLog:"Failed to send file to: %s", app]; } } else { [HBUtilities writeToActivityLog:"Send file to: app not found"]; } #ifdef __SANDBOX_ENABLED__ if (accessingSecurityScopedResource) { [item.outputURL stopAccessingSecurityScopedResource]; } #endif } } /** * Runs the alert for a single job */ - (void)itemCompletedAlerts:(HBQueueItem *)item result:(HBCoreResult)result { // Both the Notification and Sending to tagger can be done as encodes roll off the queue if ([[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionNotification || [[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionAlertAndNotification) { // If Play System Alert has been selected in Preferences bool playSound = [[NSUserDefaults standardUserDefaults] boolForKey:@"HBAlertWhenDoneSound"]; NSString *title; NSString *description; if (result == HBCoreResultDone) { title = NSLocalizedString(@"Put down that cocktail…", @"Queue notification alert message"); description = [NSString stringWithFormat:NSLocalizedString(@"Your encode %@ is done!", @"Queue done notification message"), item.outputFileName]; } else { title = NSLocalizedString(@"Encode failed", @"Queue done notification failed message"); description = [NSString stringWithFormat:NSLocalizedString(@"Your encode %@ couldn't be completed.", @"Queue done notification message"), item.outputFileName]; } [self showNotificationWithTitle:title description:description url:item.completeOutputURL playSound:playSound]; } } /** * Runs the global queue completed alerts */ - (void)queueCompletedAlerts { // If Play System Alert has been selected in Preferences if ([[NSUserDefaults standardUserDefaults] boolForKey:@"HBAlertWhenDoneSound"] == YES) { NSBeep(); } // If Alert Window or Window and Notification has been selected if ([[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionAlert || [[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionAlertAndNotification) { // On Screen Notification NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"Put down that cocktail…", @"Queue done alert message")]; [alert setInformativeText:NSLocalizedString(@"Your HandBrake queue is done!", @"Queue done alert informative text")]; [NSApp requestUserAttention:NSCriticalRequest]; [alert runModal]; } // If sleep has been selected if ([[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionSleep) { // Sleep NSDictionary *errorDict; NSAppleScript *scriptObject = [[NSAppleScript alloc] initWithSource: @"tell application \"Finder\" to sleep"]; [scriptObject executeAndReturnError: &errorDict]; } // If Shutdown has been selected if ([[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionShutDown) { // Shut Down NSDictionary *errorDict; NSAppleScript *scriptObject = [[NSAppleScript alloc] initWithSource:@"tell application \"Finder\" to shut down"]; [scriptObject executeAndReturnError: &errorDict]; } } - (void)queueLowDiskSpaceAlert { NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"Your destination disk is almost full.", @"Queue -> disk almost full alert message")]; [alert setInformativeText:NSLocalizedString(@"You need to make more space available on your destination disk.",@"Queue -> disk almost full alert informative text")]; [NSApp requestUserAttention:NSCriticalRequest]; [alert runModal]; } #pragma mark - Queue Item Controls /** * Delete encodes from the queue window and accompanying array * Also handling first cancelling the encode if in fact its currently encoding. */ - (IBAction)removeSelectedQueueItem:(id)sender { if ([self.items beginTransaction] == HBDistributedArrayContentReload) { // Do not execture the action if the array changed. [self.items commit]; return; } NSMutableIndexSet *targetedRows = [[self.tableView targetedRowIndexes] mutableCopy]; if (targetedRows.count) { // if this is a currently encoding job, we need to be sure to alert the user, // to let them decide to cancel it first, then if they do, we can come back and // remove it NSIndexSet *workingIndexes = [self.items indexesOfObjectsUsingBlock:^BOOL(HBQueueItem *item) { return item.state == HBQueueItemStateWorking; }]; if ([targetedRows containsIndexes:workingIndexes]) { [targetedRows removeIndexes:workingIndexes]; NSArray *workingItems = [self.items filteredArrayUsingBlock:^BOOL(HBQueueItem *item) { return item.state == HBQueueItemStateWorking; }]; if ([workingItems containsObject:self.currentItem]) { NSString *alertTitle = [NSString stringWithFormat:NSLocalizedString(@"Stop This Encode and Remove It?", @"Queue Stop Alert -> stop and remove message")]; // Which window to attach the sheet to? NSWindow *targetWindow = self.window; if ([sender respondsToSelector: @selector(window)]) { targetWindow = [sender window]; } NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:alertTitle]; [alert setInformativeText:NSLocalizedString(@"Your movie will be lost if you don't continue encoding.", @"Queue Stop Alert -> stop and remove informative text")]; [alert addButtonWithTitle:NSLocalizedString(@"Keep Encoding", @"Queue Stop Alert -> stop and remove first button")]; [alert addButtonWithTitle:NSLocalizedString(@"Stop Encoding and Delete", @"Queue Stop Alert -> stop and remove second button")]; [alert setAlertStyle:NSAlertStyleCritical]; [alert beginSheetModalForWindow:targetWindow completionHandler:^(NSModalResponse returnCode) { if (returnCode == NSAlertSecondButtonReturn) { [self.items beginTransaction]; NSInteger index = [self.items indexOfObject:self.currentItem]; [self cancelCurrentItemAndContinue]; [self removeQueueItemAtIndex:index]; [self.items commit]; } }]; } } // remove the non working items immediately [self removeQueueItemsAtIndexes:targetedRows]; } [self.items commit]; } /** * Show the finished encode in the finder */ - (IBAction)revealSelectedQueueItems:(id)sender { NSIndexSet *targetedRows = [self.tableView targetedRowIndexes]; NSMutableArray *urls = [[NSMutableArray alloc] init]; NSUInteger currentIndex = [targetedRows firstIndex]; while (currentIndex != NSNotFound) { NSURL *url = [[self.items objectAtIndex:currentIndex] completeOutputURL]; [urls addObject:url]; currentIndex = [targetedRows indexGreaterThanIndex:currentIndex]; } [[NSWorkspace sharedWorkspace] activateFileViewerSelectingURLs:urls]; } - (IBAction)revealSelectedQueueItemsSources:(id)sender { NSIndexSet *targetedRows = [self.tableView targetedRowIndexes]; NSMutableArray *urls = [[NSMutableArray alloc] init]; NSUInteger currentIndex = [targetedRows firstIndex]; while (currentIndex != NSNotFound) { NSURL *url = [[self.items objectAtIndex:currentIndex] fileURL]; [urls addObject:url]; currentIndex = [targetedRows indexGreaterThanIndex:currentIndex]; } [[NSWorkspace sharedWorkspace] activateFileViewerSelectingURLs:urls]; } - (void)remindUserOfSleepOrShutdown { if ([[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionSleep) { // Warn that computer will sleep after encoding NSBeep(); [NSApp requestUserAttention:NSCriticalRequest]; NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"The computer will sleep after encoding is done.", @"Queue Done Alert -> sleep message")]; [alert setInformativeText:NSLocalizedString(@"You have selected to sleep the computer after encoding. To turn off sleeping, go to the HandBrake preferences.", @"Queue Done Alert -> sleep informative text")]; [alert addButtonWithTitle:NSLocalizedString(@"OK", @"Queue Done Alert -> sleep first button")]; [alert addButtonWithTitle:NSLocalizedString(@"Preferences…", @"Queue Done Alert -> sleep second button")]; NSInteger response = [alert runModal]; if (response == NSAlertSecondButtonReturn) { [self.delegate showPreferencesWindow:nil]; } } else if ([[NSUserDefaults standardUserDefaults] integerForKey:@"HBAlertWhenDone"] == HBDoneActionShutDown) { // Warn that computer will shut down after encoding NSBeep(); [NSApp requestUserAttention:NSCriticalRequest]; NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"The computer will shut down after encoding is done.", @"Queue Done Alert -> shut down message")]; [alert setInformativeText:NSLocalizedString(@"You have selected to shut down the computer after encoding. To turn off shut down, go to the HandBrake preferences.", @"Queue Done Alert -> shut down informative text")]; [alert addButtonWithTitle:NSLocalizedString(@"OK", @"Queue Done Alert -> shut down first button")]; [alert addButtonWithTitle:NSLocalizedString(@"Preferences…", @"Queue Done Alert -> shut down second button")]; NSInteger response = [alert runModal]; if (response == NSAlertSecondButtonReturn) { [self.delegate showPreferencesWindow:nil]; } } } /** * Rip: puts up an alert before ultimately calling doRip */ - (IBAction)rip:(id)sender { // Rip or Cancel ? if (self.core.state == HBStateWorking || self.core.state == HBStatePaused || self.core.state == HBStateSearching) { [self cancelRip:sender]; } // If there are pending items in the queue, then this is a rip the queue else if (self.pendingItemsCount > 0) { // We check to see if we need to warn the user that the computer will go to sleep // or shut down when encoding is finished [self remindUserOfSleepOrShutdown]; [self.core preventSleep]; [self encodeNextQueueItem]; } } /** * Displays an alert asking user if the want to cancel encoding of current item. * Cancel: returns immediately after posting the alert. Later, when the user * acknowledges the alert, doCancelCurrentItem is called. */ - (IBAction)cancelRip:(id)sender { // Which window to attach the sheet to? NSWindow *window = self.window; if ([sender respondsToSelector:@selector(window)]) { window = [sender window]; } NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"You are currently encoding. What would you like to do?", @"Queue Alert -> cancel rip message")]; [alert setInformativeText:NSLocalizedString(@"Select Continue Encoding to dismiss this dialog without making changes.", @"Queue Alert -> cancel rip informative text")]; [alert addButtonWithTitle:NSLocalizedString(@"Continue Encoding", @"Queue Alert -> cancel rip first button")]; [alert addButtonWithTitle:NSLocalizedString(@"Skip Current Job", @"Queue Alert -> cancel rip second button")]; [alert addButtonWithTitle:NSLocalizedString(@"Stop After Current Job", @"Queue Alert -> cancel rip third button")]; [alert addButtonWithTitle:NSLocalizedString(@"Stop All", @"Queue Alert -> cancel rip fourth button")]; [alert setAlertStyle:NSAlertStyleCritical]; [alert beginSheetModalForWindow:window completionHandler:^(NSModalResponse returnCode) { if (returnCode == NSAlertSecondButtonReturn) { [self cancelCurrentItemAndContinue]; } else if (returnCode == NSAlertThirdButtonReturn) { [self finishCurrentAndStop]; } else if (returnCode == NSAlertThirdButtonReturn + 1) { [self cancelCurrentItemAndStop]; } }]; } /** * Starts or cancels the processing of items depending on the current state */ - (IBAction)toggleStartCancel:(id)sender { HBState s = self.core.state; if ((s == HBStatePaused) || (s == HBStateWorking) || (s == HBStateMuxing)) { [self cancelRip:self]; } else if (self.pendingItemsCount > 0) { [self rip:self]; } } /** * Toggles the pause/resume state of libhb */ - (IBAction)togglePauseResume:(id)sender { HBState s = self.core.state; if (s == HBStatePaused) { [self.core resume]; [self.core preventSleep]; } else if (s == HBStateWorking || s == HBStateMuxing) { [self.core pause]; [self.core allowSleep]; } } /** * Resets the item state to ready. */ - (IBAction)resetJobState:(id)sender { if ([self.items beginTransaction] == HBDistributedArrayContentReload) { // Do not execture the action if the array changed. [self.items commit]; return; } NSIndexSet *targetedRows = [self.tableView targetedRowIndexes]; NSMutableIndexSet *updatedIndexes = [NSMutableIndexSet indexSet]; NSUInteger currentIndex = [targetedRows firstIndex]; while (currentIndex != NSNotFound) { HBQueueItem *item = self.items[currentIndex]; if (item.state == HBQueueItemStateCanceled || item.state == HBQueueItemStateCompleted || item.state == HBQueueItemStateFailed) { item.state = HBQueueItemStateReady; [updatedIndexes addIndex:currentIndex]; } currentIndex = [targetedRows indexGreaterThanIndex:currentIndex]; } [self reloadQueueItemsAtIndexes:updatedIndexes]; [self.items commit]; } - (void)editQueueItem:(HBQueueItem *)item { NSParameterAssert(item); [self.items beginTransaction]; if (item != self.currentItem) { item.state = HBQueueItemStateWorking; NSUInteger row = [self.items indexOfObject:item]; [self reloadQueueItemAtIndex:row]; [self.controller openJob:[item.job copy] completionHandler:^(BOOL result) { [self.items beginTransaction]; if (result) { // Now that source is loaded and settings applied, delete the queue item from the queue NSInteger index = [self.items indexOfObject:item]; item.state = HBQueueItemStateReady; [self removeQueueItemAtIndex:index]; } else { item.state = HBQueueItemStateFailed; NSBeep(); } [self.items commit]; }]; } else { NSBeep(); } [self.items commit]; } /** * Send the selected queue item back to the main window for rescan and possible edit. */ - (IBAction)editSelectedQueueItem:(id)sender { if ([self.items beginTransaction] == HBDistributedArrayContentReload) { // Do not execture the action if the array changed. [self.items commit]; return; } NSInteger row = self.tableView.clickedRow; if (row != NSNotFound) { // if this is a currently encoding item, we need to be sure to alert the user, // to let them decide to cancel it first, then if they do, we can come back and // remove it HBQueueItem *item = self.items[row]; if (item == self.currentItem) { NSString *alertTitle = [NSString stringWithFormat:NSLocalizedString(@"Stop This Encode and Edit It?", @"Queue Edit Alert -> stop and edit message")]; // Which window to attach the sheet to? NSWindow *docWindow = self.window; if ([sender respondsToSelector: @selector(window)]) { docWindow = [sender window]; } NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:alertTitle]; [alert setInformativeText:NSLocalizedString(@"Your movie will be lost if you don't continue encoding.", @"Queue Edit Alert -> stop and edit informative text")]; [alert addButtonWithTitle:NSLocalizedString(@"Keep Encoding", @"Queue Edit Alert -> stop and edit first button")]; [alert addButtonWithTitle:NSLocalizedString(@"Stop Encoding and Edit", @"Queue Edit Alert -> stop and edit second button")]; [alert setAlertStyle:NSAlertStyleCritical]; [alert beginSheetModalForWindow:docWindow completionHandler:^(NSModalResponse returnCode) { if (returnCode == NSAlertSecondButtonReturn) { [self editQueueItem:item]; } }]; } else if (item.state != HBQueueItemStateWorking) { [self editQueueItem:item]; } } [self.items commit]; } - (IBAction)clearAll:(id)sender { [self.items beginTransaction]; NSIndexSet *indexes = [self.items indexesOfObjectsUsingBlock:^BOOL(HBQueueItem *item) { return (item.state != HBQueueItemStateWorking); }]; [self removeQueueItemsAtIndexes:indexes]; [self.items commit]; } - (IBAction)clearCompleted:(id)sender { [self.items beginTransaction]; NSIndexSet *indexes = [self.items indexesOfObjectsUsingBlock:^BOOL(HBQueueItem *item) { return (item.state == HBQueueItemStateCompleted); }]; [self removeQueueItemsAtIndexes:indexes]; [self.items commit]; } #pragma mark - NSTableView data source - (NSView *)tableView:(NSTableView *)tableView viewForTableColumn:(NSTableColumn *)tableColumn row:(NSInteger)row { HBQueueItemView *view = [tableView makeViewWithIdentifier:@"MainCell" owner:self]; HBQueueItem *item = self.items[row]; view.delegate = self; view.item = item; return view; } - (NSInteger)numberOfRowsInTableView:(NSTableView *)tableView { return self.items.count; } - (NSTableCellView *)dummyCell { if (!_dummyCell) { _dummyCell = [self.tableView makeViewWithIdentifier:@"MainCellForSizing" owner: self]; _dummyCellWidth = [NSLayoutConstraint constraintWithItem:_dummyCell attribute:NSLayoutAttributeWidth relatedBy:NSLayoutRelationEqual toItem:nil attribute:NSLayoutAttributeNotAnAttribute multiplier:1.0f constant:500]; [_dummyCell addConstraint:_dummyCellWidth]; } return _dummyCell; } - (CGFloat)tableView:(NSTableView *)tableView heightOfRow:(NSInteger)row { HBQueueItem *item = self.items[row]; if (item.expanded) { CGFloat width = tableView.frame.size.width; self.dummyCellWidth.constant = width; self.dummyCell.textField.preferredMaxLayoutWidth = width - 60; self.dummyCell.textField.attributedStringValue = item.attributedDescription; CGFloat height = self.dummyCell.fittingSize.height; return height; } else { return 20; } } - (void)toggleRowsAtIndexes:(NSIndexSet *)rowIndexes expand:(BOOL)expand { NSMutableIndexSet *rowsToExpand = [NSMutableIndexSet indexSet]; [rowIndexes enumerateIndexesUsingBlock:^(NSUInteger index, BOOL *stop) { HBQueueItem *item = self.items[index]; BOOL expanded = item.expanded; if (expanded != expand) { item.expanded = !expanded; [rowsToExpand addIndex:index]; } HBQueueItemView *itemView = (HBQueueItemView *)[self.tableView viewAtColumn:0 row:index makeIfNecessary:NO]; if (expand) { [itemView expand]; } else { [itemView collapse]; } }]; [self.tableView noteHeightOfRowsWithIndexesChanged:rowsToExpand]; } #pragma mark NSQueueItemView delegate - (void)removeQueueItem:(nonnull HBQueueItem *)item { NSUInteger index = [self.items indexOfObject:item]; [self removeQueueItemAtIndex:index]; } - (void)revealQueueItem:(nonnull HBQueueItem *)item { [[NSWorkspace sharedWorkspace] activateFileViewerSelectingURLs:@[item.completeOutputURL]]; } - (void)toggleQueueItemHeight:(nonnull HBQueueItem *)item { NSInteger row = [self.items indexOfObject:item]; [self toggleRowsAtIndexes:[NSIndexSet indexSetWithIndex:row] expand:!item.expanded]; } #pragma mark NSTableView delegate - (void)HB_deleteSelectionFromTableView:(NSTableView *)tableView { [self removeSelectedQueueItem:tableView]; } - (void)HB_expandSelectionFromTableView:(NSTableView *)tableView { NSIndexSet *rowIndexes = [self.tableView selectedRowIndexes]; [self toggleRowsAtIndexes:rowIndexes expand:YES]; } - (void)HB_collapseSelectionFromTableView:(NSTableView *)tableView; { NSIndexSet *rowIndexes = [self.tableView selectedRowIndexes]; [self toggleRowsAtIndexes:rowIndexes expand:NO]; } - (BOOL)tableView:(NSTableView *)tableView writeRowsWithIndexes:(NSIndexSet *)rowIndexes toPasteboard:(NSPasteboard *)pboard; { NSArray *items = [self.items objectsAtIndexes:rowIndexes]; // Dragging is only allowed of the pending items. if (items[0].state != HBQueueItemStateReady) { return NO; } self.dragNodesArray = items; // Provide data for our custom type, and simple NSStrings. [pboard declareTypes:@[DragDropSimplePboardType] owner:self]; // the actual data doesn't matter since DragDropSimplePboardType drags aren't recognized by anyone but us!. [pboard setData:[NSData data] forType:DragDropSimplePboardType]; return YES; } - (NSDragOperation)tableView:(NSTableView *)tableView validateDrop:(id)info proposedRow:(NSInteger)row proposedDropOperation:(NSTableViewDropOperation)dropOperation { // Don't allow dropping ONTO an item since they can't really contain any children. BOOL isOnDropTypeProposal = dropOperation == NSTableViewDropOn; if (isOnDropTypeProposal) { return NSDragOperationNone; } // We do not let the user drop a pending item before or *above* // already finished or currently encoding items. NSInteger encodingRow = [self.items indexOfObject:self.currentItem]; if (encodingRow != NSNotFound && row <= encodingRow) { return NSDragOperationNone; row = MAX(row, encodingRow); } return NSDragOperationMove; } - (BOOL)tableView:(NSTableView *)tableView acceptDrop:(id)info row:(NSInteger)row dropOperation:(NSTableViewDropOperation)dropOperation { [self moveQueueItems:self.dragNodesArray toIndex:row]; return YES; } @end @implementation HBQueueController (TouchBar) @dynamic touchBar; static NSTouchBarItemIdentifier HBTouchBarMain = @"fr.handbrake.queueWindowTouchBar"; static NSTouchBarItemIdentifier HBTouchBarRip = @"fr.handbrake.rip"; static NSTouchBarItemIdentifier HBTouchBarPause = @"fr.handbrake.pause"; - (NSTouchBar *)makeTouchBar { NSTouchBar *bar = [[NSTouchBar alloc] init]; bar.delegate = self; bar.defaultItemIdentifiers = @[HBTouchBarRip, HBTouchBarPause]; bar.customizationIdentifier = HBTouchBarMain; bar.customizationAllowedItemIdentifiers = @[HBTouchBarRip, HBTouchBarPause]; return bar; } - (NSTouchBarItem *)touchBar:(NSTouchBar *)touchBar makeItemForIdentifier:(NSTouchBarItemIdentifier)identifier { if ([identifier isEqualTo:HBTouchBarRip]) { NSCustomTouchBarItem *item = [[NSCustomTouchBarItem alloc] initWithIdentifier:identifier]; item.customizationLabel = NSLocalizedString(@"Start/Stop Encoding", @"Touch bar"); NSButton *button = [NSButton buttonWithImage:[NSImage imageNamed:NSImageNameTouchBarPlayTemplate] target:self action:@selector(toggleStartCancel:)]; item.view = button; return item; } else if ([identifier isEqualTo:HBTouchBarPause]) { NSCustomTouchBarItem *item = [[NSCustomTouchBarItem alloc] initWithIdentifier:identifier]; item.customizationLabel = NSLocalizedString(@"Pause/Resume Encoding", @"Touch bar"); NSButton *button = [NSButton buttonWithImage:[NSImage imageNamed:NSImageNameTouchBarPauseTemplate] target:self action:@selector(togglePauseResume:)]; item.view = button; return item; } return nil; } - (void)_touchBar_updateButtonsStateForQueueCore:(HBState)state; { NSButton *ripButton = (NSButton *)[[self.touchBar itemForIdentifier:HBTouchBarRip] view]; NSButton *pauseButton = (NSButton *)[[self.touchBar itemForIdentifier:HBTouchBarPause] view]; if (state == HBStateScanning || state == HBStateWorking || state == HBStateSearching || state == HBStateMuxing) { ripButton.image = [NSImage imageNamed:NSImageNameTouchBarRecordStopTemplate]; pauseButton.image = [NSImage imageNamed:NSImageNameTouchBarPauseTemplate]; } else if (state == HBStatePaused) { ripButton.image = [NSImage imageNamed:NSImageNameTouchBarRecordStopTemplate]; pauseButton.image = [NSImage imageNamed:NSImageNameTouchBarPlayTemplate]; } else if (state == HBStateIdle) { ripButton.image = [NSImage imageNamed:NSImageNameTouchBarPlayTemplate]; pauseButton.image = [NSImage imageNamed:NSImageNameTouchBarPauseTemplate]; } } - (void)_touchBar_validateUserInterfaceItems { for (NSTouchBarItemIdentifier identifier in self.touchBar.itemIdentifiers) { NSTouchBarItem *item = [self.touchBar itemForIdentifier:identifier]; NSView *view = item.view; if ([view isKindOfClass:[NSButton class]]) { NSButton *button = (NSButton *)view; BOOL enabled = [self validateUserIterfaceItemForAction:button.action]; button.enabled = enabled; } } } @end