OverSight/MainApp/RulesWindowController.m

//
//  RulesWindowController.m
//  OverSight
//
//  Created by Patrick Wardle on 7/7/16.
//  Copyright (c) 2016 Objective-See. All rights reserved.
//


#import "Consts.h"
#import "RuleRow.h"
#import "Utilities.h"
#import "AppDelegate.h"
#import "../Shared/Logging.h"
#import "RulesWindowController.h"
#import "../Shared/XPCProtocol.h"

@interface RulesWindowController ()

@end

@implementation RulesWindowController

@synthesize items;

//automatically called when nib is loaded
// ->just center window
-(void)awakeFromNib
{
    //load whitelisted items
    self.items = [NSMutableArray arrayWithContentsOfFile:[[APP_SUPPORT_DIRECTORY stringByExpandingTildeInPath] stringByAppendingPathComponent:FILE_WHITELIST]];
    
    return;
}



/*
-(void)windowDidLoad
{
    //super
    [super windowDidLoad];
    
    //load whitelisted items
    self.items = [NSMutableArray arrayWithContentsOfFile:[[APP_SUPPORT_DIRECTORY stringByExpandingTildeInPath] stringByAppendingPathComponent:FILE_WHITELIST]];

    return;
}
*/
 
//table delegate
// ->return number of rows
-(NSInteger)numberOfRowsInTableView:(NSTableView *)tableView
{
    //count
    return self.items.count;
}

//table delegate method
// ->return cell for row
-(NSView *)tableView:(NSTableView *)tableView viewForTableColumn:(NSTableColumn *)tableColumn row:(NSInteger)row
{
    //cell
    NSTableCellView *tableCell = nil;
    
    //process name
    NSString* processName = nil;
    
    //process path
    NSString* processPath = nil;
    
    //process icon
    NSImage* processIcon = nil;
    
    //app bundle
    NSBundle* appBundle = nil;
    
    //sanity check
    if(row >= self.items.count)
    {
        //bail
        goto bail;
    }
    
    //grab process path
    processPath = [self.items objectAtIndex:row];
    
    //try find an app bundle
    appBundle = findAppBundle(processPath);
    if(nil != appBundle)
    {
        //grab name from app's bundle
        processName = [appBundle infoDictionary][@"CFBundleName"];
    }
    
    //still nil?
    // ->just grab from path
    if(nil == processName)
    {
        //from path
        processName = [processPath lastPathComponent];
    }
    
    //grab icon
    processIcon = getIconForProcess(processPath);
    
    //init table cell
    tableCell = [tableView makeViewWithIdentifier:@"itemCell" owner:self];
    if(nil == tableCell)
    {
        //bail
        goto bail;
    }
    
    //set icon
    tableCell.imageView.image = processIcon;
    
    //set (main) text
    tableCell.textField.stringValue = processName;
    
    //set sub text
    [[tableCell viewWithTag:TABLE_ROW_SUB_TEXT_TAG] setStringValue:processPath];
    
    //set detailed text color to gray
    ((NSTextField*)[tableCell viewWithTag:TABLE_ROW_SUB_TEXT_TAG]).textColor = [NSColor grayColor];
    
//bail
bail:
    
    // Return the result
    return tableCell;

}

//automatically invoked
// ->create custom (sub-classed) NSTableRowView
-(NSTableRowView *)tableView:(NSTableView *)tableView rowViewForRow:(NSInteger)row
{
    //row view
    RuleRow* rowView = nil;
    
    //row ID
    static NSString* const kRowIdentifier = @"RowView";
    
    //try grab existing row view
    rowView = [tableView makeViewWithIdentifier:kRowIdentifier owner:self];
    
    //make new if needed
    if(nil == rowView)
    {
        //create new
        // ->size doesn't matter
        rowView = [[RuleRow alloc] initWithFrame:NSZeroRect];
        
        //set row ID
        rowView.identifier = kRowIdentifier;
    }
    
    return rowView;
}

//delete a whitelist item
// ->gotta invoke the login item, as it can do that via XPC
-(IBAction)deleteRule:(id)sender
{
    //index of selected row
    NSInteger selectedRow = 0;
    
    //rule
    NSString* processPath = nil;
    
    //grab selected row
    selectedRow = [self.tableView rowForView:sender];

    //extract selected item
    processPath = self.items[selectedRow];
    
    //remove from items
    [self.items removeObject:processPath];
    
    //reload table
    [self.tableView reloadData];
    
    //restart login item in background
    // ->pass in process name so it can un-whitelist via XPC
    dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0),
    ^{
       //restart
       [((AppDelegate*)[[NSApplication sharedApplication] delegate]) startLoginItem:YES args:@[processPath]];
    });
    
    return;
}

@end