WindowSizer.m

/*
 ShiftIt: Resize windows with Hotkeys
 Copyright (C) 2010  Aravind
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
 
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.
 
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 */

#import "WindowSizer.h"
#import <Carbon/Carbon.h>

@interface WindowSizer ()

- (void)disableCursorMode;
- (void)resizeWindowToCursorLocation:(NSTimer *)timer;

@end

@implementation WindowSizer

-(void)chooseScreen{
    CGDirectDisplayID tempId[2];
    CGDisplayCount tempCount;
    CGError error = CGGetDisplaysWithRect(CGRectMake(_windowPosition.x, _windowPosition.y, _windowSize.width, _windowSize.height), 2, tempId, &tempCount);
    if(error == kCGErrorSuccess){
        CGRect screenBounds = CGDisplayBounds(tempId[0]);
        if(tempCount == 1){
            _screenPosition.x = screenBounds.origin.x;
            _screenPosition.y = screenBounds.origin.y;
            _screenSize.width = screenBounds.size.width;
            _screenSize.height = screenBounds.size.height;
        }else if (tempCount == 2) {
            CGRect screenBounds1 = CGDisplayBounds(tempId[1]);
            int screenChosen = 0;
            int delta = abs(screenBounds.origin.x - (_windowPosition.x+_windowSize.width));
            int delta2 = 0;
            if(delta > screenBounds.size.width){
                delta = abs(_windowPosition.x - (screenBounds.origin.x+screenBounds.size.width));
                delta2 = abs(_windowPosition.x+_windowSize.width - screenBounds1.origin.x);
            }else {
                delta2 = abs(_windowPosition.x - (screenBounds1.origin.x+screenBounds1.size.width));
            }
            if (delta2> delta) {
                screenChosen = 1;
            }
            if(screenChosen == 0){
                _screenPosition.x = screenBounds.origin.x;
                _screenPosition.y = screenBounds.origin.y;
                _screenSize.width = screenBounds.size.width;
                _screenSize.height = screenBounds.size.height;
            }else {
                _screenPosition.x = screenBounds1.origin.x;
                _screenPosition.y = screenBounds1.origin.y;
                _screenSize.width = screenBounds1.size.width;
                _screenSize.height = screenBounds1.size.height;
            }
        }       
    }
}

-(void)getVisibleScreenParams{
    NSArray * tempArray = [NSScreen screens];
    if ([tempArray count] ==1) {
        NSScreen * tempScreen = (NSScreen*)[tempArray objectAtIndex:0];
        _screenVisibleSize = tempScreen.visibleFrame.size;
        _screenVisiblePosition = tempScreen.visibleFrame.origin;
    }else {
        for(int i = 0;i<[tempArray count]; i++){
            NSScreen * tempScreen = (NSScreen*)[tempArray objectAtIndex:i];
            if (tempScreen.frame.origin.x == _screenPosition.x) {
                _screenVisibleSize = tempScreen.visibleFrame.size;
                _screenVisiblePosition = tempScreen.visibleFrame.origin;
                break;
            }
        }
    }
	
    
}

-(BOOL)getWindowParameters{
    NSLog(@"Get Window Parameter");
    BOOL error = FALSE;
	AXUIElementRef _focusedApp;
	CFTypeRef _position;
	CFTypeRef _size;
    
	AXUIElementCopyAttributeValue(_systemWideElement,(CFStringRef)kAXFocusedApplicationAttribute,(CFTypeRef*)&_focusedApp);
	if(AXUIElementCopyAttributeValue((AXUIElementRef)_focusedApp,(CFStringRef)NSAccessibilityFocusedWindowAttribute,(CFTypeRef*)&_focusedWindow) == kAXErrorSuccess) {
		if(CFGetTypeID(_focusedWindow) == AXUIElementGetTypeID()) {
			if(AXUIElementCopyAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)&_position) != kAXErrorSuccess) {
				NSLog(@"Can't Retrieve Window Position");
                error = TRUE;
			}else {
                if(AXValueGetType(_position) == kAXValueCGPointType) {
                    AXValueGetValue(_position, kAXValueCGPointType, (void*)&_windowPosition);
                    NSLog(@"position x:%f, y:%f", _windowPosition.x, _windowPosition.y);
                }else {
                    error = TRUE;
                }                
            }
			
			if(AXUIElementCopyAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)&_size) != kAXErrorSuccess) {
				NSLog(@"Can't Retrieve Window Size");
                error = TRUE;
			}else {
                if(AXValueGetType(_size) == kAXValueCGSizeType) {
                    AXValueGetValue(_size, kAXValueCGSizeType, (void*)&_windowSize);
                    NSLog(@"size width:%f, height:%f", _windowSize.width, _windowSize.height);
                }else {
                    error = TRUE;
                }
            }
		}
	}else {
		NSLog(@"Problem with App");
	}
    if(!error){
        // macOS mojave doesn't seem to have GetMBarHeight() anymore
        // 25 should work for intel macs, but not for notch silicon macs
        _menuBarHeight = 25;
        [self chooseScreen];
        NSLog(@"Get Window Parameter Success");
    }else {
        NSLog(@"Get Window Parameter Failed");
    }
    return !error;
}

-(IBAction)shiftToCursorLocation:(id)sender{
    NSLog(@"-- Shifting to Cursor Location");
	
	const NSTimeInterval ResizeUpdateTimeInterval = 0.1;
	
	cursorModeFlag = !cursorModeFlag;
	
	if (cursorModeFlag) {
		[self resizeWindowToCursorLocation:nil];	// call it once first to update immediately
		
		if (resizeTimer == nil) {
			resizeTimer = [NSTimer scheduledTimerWithTimeInterval:ResizeUpdateTimeInterval
												  target:self
												selector:@selector(resizeWindowToCursorLocation:)
												userInfo:nil
												 repeats:YES];
		}
		
		// register mouse click listener
		mouseClickMonitor = [NSEvent addGlobalMonitorForEventsMatchingMask:
							 (NSLeftMouseDownMask | NSRightMouseDownMask | NSOtherMouseDownMask)
																  handler:^(NSEvent *incomingEvent) {
																	  NSLog(@"Mouse click detected, disabling cursor mode");
																	  [self disableCursorMode];
																  }];
	} else {
		[self disableCursorMode];
	}
}

- (void)disableCursorMode {
	cursorModeFlag = NO;
	[resizeTimer invalidate];
	resizeTimer = nil;
	if (mouseClickMonitor) {
		[NSEvent removeMonitor:mouseClickMonitor];
		mouseClickMonitor = nil;
	}
}

- (void)resizeWindowToCursorLocation:(NSTimer *)timer {
	if([self getWindowParameters]){
		[self getVisibleScreenParams];
		
		// get mouse coordinates
		NSPoint mouseLocation;
		mouseLocation = [NSEvent mouseLocation]; //get current mouse position
		
		CFTypeRef _size;
		
		NSLog(@"Mouse location: (%f, %f)", mouseLocation.x, mouseLocation.y);
		NSLog(@"Screen size (%f, %f)", _screenSize.width, _screenSize.height);
		_windowSize.width = mouseLocation.x - _windowPosition.x;
		_windowSize.height = (_screenSize.height - mouseLocation.y) - _windowPosition.y;
		_size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
		
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
	}
	_focusedWindow = NULL;
}

-(IBAction)shiftToLeftHalf:(id)sender{
    NSLog(@"Shifting To Left Half");
	if([self getWindowParameters]){
		[self getVisibleScreenParams];
        CFTypeRef _position;
        CFTypeRef _size;
        
		_windowPosition.x = _screenVisiblePosition.x;
		_windowPosition.y = ((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = ((_screenVisibleSize.width)/2);
        _windowSize.height = _screenVisibleSize.height;
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        NSLog(@"size2 width:%f, height:%f", _windowSize.width, _windowSize.height);
		
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
        
    }
    NSLog(@"Shifted To Left Half");
    _focusedWindow = NULL;
    
}

-(IBAction)shiftToRightHalf:(id)sender{
    NSLog(@"Shifting To Right Half");
	if([self getWindowParameters]){
		[self getVisibleScreenParams];        
        CFTypeRef _position;
        CFTypeRef _size;
		_windowPosition.x = _screenVisiblePosition.x +(_screenVisibleSize.width/2);
		_windowPosition.y = ((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = ((_screenVisibleSize.width)/2);
        _windowSize.height = _screenVisibleSize.height;
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Right Half");
    _focusedWindow = NULL;
    
}

-(IBAction)shiftToTopHalf:(id)sender{
    NSLog(@"Shifting To Top Half");
	if([self getWindowParameters]){
		[self getVisibleScreenParams];        
        CFTypeRef _position;
        CFTypeRef _size;
		_windowPosition.x = _screenVisiblePosition.x;
		_windowPosition.y = ((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = _screenVisibleSize.width;
        _windowSize.height = (_screenVisibleSize.height/2);
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Top Half");
    _focusedWindow = NULL;
    
}

-(IBAction)shiftToBottomHalf:(id)sender{
    NSLog(@"Shifting To Bottom Half");
	if([self getWindowParameters]){
		[self getVisibleScreenParams];
        
        CFTypeRef _position;
        CFTypeRef _size;
        
		_windowPosition.x = _screenVisiblePosition.x;
		_windowPosition.y = (_screenVisibleSize.height/2)+((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = _screenVisibleSize.width;
        _windowSize.height = (_screenVisibleSize.height/2);
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Bottom Half");
    _focusedWindow = NULL;
}

-(IBAction)shiftToTopLeft:(id)sender{
    NSLog(@"Shifting To Top Left");
	if([self getWindowParameters]){
        [self getVisibleScreenParams];
        CFTypeRef _position;
        CFTypeRef _size;
        
		_windowPosition.x = _screenVisiblePosition.x;
		_windowPosition.y = ((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = (_screenVisibleSize.width/2);
        _windowSize.height = (_screenVisibleSize.height/2);
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Top Left");
    _focusedWindow = NULL;
    
}

-(IBAction)shiftToTopRight:(id)sender{
    NSLog(@"Shifting To Top Right");
	if([self getWindowParameters]){
        [self getVisibleScreenParams];
        
        CFTypeRef _position;
        CFTypeRef _size;
        
		_windowPosition.x = _screenVisiblePosition.x +(_screenVisibleSize.width/2);
		_windowPosition.y = ((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = (_screenVisibleSize.width/2);
        _windowSize.height = (_screenVisibleSize.height/2);
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Top Right");
    _focusedWindow = NULL;
    
}

-(IBAction)shiftToBottomLeft:(id)sender{
    NSLog(@"Shifting To Botom Left");
	if([self getWindowParameters]){
        [self getVisibleScreenParams];
        
        CFTypeRef _position;
        CFTypeRef _size;
        
		_windowPosition.x = _screenVisiblePosition.x;
		_windowPosition.y = (_screenVisibleSize.height/2)+((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = (_screenVisibleSize.width/2);
        _windowSize.height = (_screenVisibleSize.height/2);
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Bottom Left");
    _focusedWindow = NULL;
    
}

-(IBAction)shiftToBottomRight:(id)sender{
    NSLog(@"Shifting To Botom Right");
	if([self getWindowParameters]){
        [self getVisibleScreenParams];
        
        CFTypeRef _position;
        CFTypeRef _size;
        
		_windowPosition.x = _screenVisiblePosition.x +(_screenVisibleSize.width/2);
		_windowPosition.y = (_screenVisibleSize.height/2)+((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = (_screenVisibleSize.width/2);
        _windowSize.height = (_screenVisibleSize.height/2);
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Bottom Right");
    _focusedWindow = NULL;
    
}

-(IBAction)shiftToCenter:(id)sender{
    NSLog(@"Shifting To Center");
	if([self getWindowParameters]){
        [self getVisibleScreenParams];
        
        CFTypeRef _position;
        
		_windowPosition.x = _screenVisiblePosition.x+(_screenVisibleSize.width/2)-(_windowSize.width/2);
		_windowPosition.y = ((_screenVisiblePosition.x ==0)? _menuBarHeight:0)+(_screenVisibleSize.height/2)-(_windowSize.height/2);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
    }
    NSLog(@"Shifted To Center");
    _focusedWindow = NULL;
}

-(IBAction)fullScreen:(id)sender{
    NSLog(@"Shifting To Full Screen");
	if([self getWindowParameters]){
        [self getVisibleScreenParams];
        CFTypeRef _position;
        CFTypeRef _size;
        
		_windowPosition.x = _screenVisiblePosition.x;
		_windowPosition.y = ((_screenVisiblePosition.x ==0)? _menuBarHeight:0);
		_position = (CFTypeRef)(AXValueCreate(kAXValueCGPointType, (const void *)&_windowPosition));
		
        _windowSize.width = _screenVisibleSize.width;
        _windowSize.height = _screenVisibleSize.height;
        _size = (CFTypeRef)(AXValueCreate(kAXValueCGSizeType, (const void *)&_windowSize));					
        
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilityPositionAttribute,(CFTypeRef*)_position) != kAXErrorSuccess){
			NSLog(@"Position cannot be changed");
		}
		if(AXUIElementSetAttributeValue((AXUIElementRef)_focusedWindow,(CFStringRef)NSAccessibilitySizeAttribute,(CFTypeRef*)_size) != kAXErrorSuccess){
			NSLog(@"Size cannot be modified");
		}
    }
    NSLog(@"Shifted To Full Screen");
    _focusedWindow = NULL;
}

-(WindowSizer *)init{
	if(self = [super init]){;
		_systemWideElement = AXUIElementCreateSystemWide();
	}
	return self;
}

@end