#include <Cocoa/Cocoa.h>

NSColor* nscolor(gg__Color c) {
	float red = (float)c.r / 255.0f;
	float green = (float)c.g / 255.0f;
	float blue = (float)c.b / 255.0f;
	return [NSColor colorWithDeviceRed:red green:green blue:blue alpha:1.0f];
}

NSString* nsstring(string s) {
	return [[NSString alloc] initWithBytesNoCopy:s.str
	                                      length:s.len
	                                    encoding:NSUTF8StringEncoding
	                                freeWhenDone:false];
}

gg__Size gg_get_screen_size() {
	NSScreen *currentScreen = nil;
	NSWindow *mainWindow = [NSApp mainWindow];
	// 1. Try screen containing the main window
	if (mainWindow) {
		currentScreen = [mainWindow screen];
	}
	// 2. If no main window, try the key window (might be different, e.g., a panel)
	if (!currentScreen) {
		NSWindow *keyWindow = [NSApp keyWindow];
		if (keyWindow) {
			currentScreen = [keyWindow screen];
		}
	}
	// 3. If no relevant window, find the screen containing the mouse cursor
	if (!currentScreen) {
		// Get mouse location in global screen coordinates (bottom-left origin)
		NSPoint mouseLocation = [NSEvent mouseLocation];
		NSArray<NSScreen *> *screens = [NSScreen screens];
		for (NSScreen *screen in screens) {
			// Check if the mouse location is within the screen's frame
			// Note: Both mouseLocation and screen.frame use bottom-left origin coordinates
			if (NSMouseInRect(mouseLocation, [screen frame], NO)) {
				currentScreen = screen;
				break; // Found the screen with the mouse
			}
		}
	}
	// 4. As a last resort, fall back to the main screen
	if (!currentScreen) {
		NSLog(@"Warning: Could not determine current screen based on window or mouse. Falling back to mainScreen.");
		currentScreen = [NSScreen mainScreen];
	}
	// Now get the size of the determined screen
	NSDictionary *description = [currentScreen deviceDescription];
	// Use NSDeviceSize to get pixel dimensions
	NSSize displayPixelSize = [[description objectForKey:NSDeviceSize] sizeValue];
	// Create the V gg.Size object
	gg__Size res;
	res.width = displayPixelSize.width;
	res.height = displayPixelSize.height;
	return res;
}

void gg_macos_resize_window(void *window_ptr, int width, int height) {
	if (window_ptr == nil || width <= 0 || height <= 0) {
		return;
	}
	NSWindow *window = (__bridge NSWindow *)window_ptr;
	[window setContentSize:NSMakeSize((CGFloat)width, (CGFloat)height)];
}

void gg_macos_set_window_resizable(void *window_ptr, bool resizable) {
	if (window_ptr == nil) {
		return;
	}
	NSWindow *window = (__bridge NSWindow *)window_ptr;
	NSWindowStyleMask style = [window styleMask];
	if (resizable) {
		style |= NSWindowStyleMaskResizable;
	} else {
		style &= ~NSWindowStyleMaskResizable;
	}
	[window setStyleMask:style];
}

void darwin_draw_string(int x, int y, string s, gg__TextCfg cfg) {
	NSFont* font = [NSFont userFontOfSize:cfg.size];
	// # NSFont*    font = [NSFont fontWithName:@"Roboto Mono" size:cfg.size];
	if (cfg.mono) {
		// # font = [NSFont fontWithName:@"Roboto Mono" size:cfg.size];
		font = [NSFont fontWithName:@"Menlo" size:cfg.size - 5];
	}
	if (cfg.bold) {
		font = [[NSFontManager sharedFontManager] convertFont:font toHaveTrait:NSBoldFontMask];
	}

	NSDictionary* attr = @{
		NSForegroundColorAttributeName : nscolor(cfg.color),
		// NSParagraphStyleAttributeName: paragraphStyle,
		NSFontAttributeName : font,
	};
	[nsstring(s) drawAtPoint:NSMakePoint(x, y - 15) withAttributes:attr];
}

int darwin_text_width(string s) {
	// println('text_width "${s}" len=${s.len}')
	NSString* n = @"";
	if (s.len == 1) {
		// println('len=1')
		n = [NSString stringWithFormat:@"%c", s.str[0]];
	} else {
		n = nsstring(s);
	}
	/*
	# if (!defaultFont){
	# defaultFont = [NSFont userFontOfSize: ui__DEFAULT_FONT_SIZE];
	# }
	# NSDictionary *attrs = @{
	# NSFontAttributeName: defaultFont,
	# };
	*/
	NSSize size = [n sizeWithAttributes:nil];
	// # printf("!!!%f\n", ceil(size.width));
	return (int)(ceil(size.width));
}

void darwin_draw_rect(float x, float y, float width, float height, gg__Color c) {
	NSColor* color = nscolor(c);
	NSRect rect = NSMakeRect(x, y, width, height);
	[color setFill];
	NSRectFill(rect);
}

static void mark_view_tree_needs_display(NSView *view) {
	if (view == nil) {
		return;
	}
	[view setNeedsDisplay:YES];
	for (NSView *subview in [view subviews]) {
		mark_view_tree_needs_display(subview);
	}
}

void darwin_window_refresh() {
	dispatch_async(dispatch_get_main_queue(), ^{
		NSWindow *window = [NSApp mainWindow];
		if (window == nil) {
			window = [NSApp keyWindow];
		}
		if (window == nil) {
			return;
		}
		NSView *contentView = [window contentView];
		if (contentView == nil) {
			return;
		}
		mark_view_tree_needs_display(contentView);
		[window displayIfNeeded];
	});

	// puts("refresh");
	//[[NSApp mainWindow].contentView drawRect:NSMakeRect(0,0,2000,2000)];
	//[[NSGraphicsContext currentContext] flushGraphics];
}

gg__Image darwin_create_image(string path_) {
	// file = file.trim_space()
	NSString* path = nsstring(path_);
	NSImage* img = [[NSImage alloc] initWithContentsOfFile:path];
	if (img == 0) {
	}
	NSSize size = [img size];
	gg__Image res;
	res.width = size.width;
	res.height = size.height;
	res.path = path_;
	res.ok = true;
	// printf("inited img width=%d\n", res.width) ;
	//  need __bridge_retained so that the pointer is not freed by ARC
	res.data = (__bridge_retained voidptr)(img);
	return res;
}

void darwin_draw_image(float x, float y, float w, float h, gg__Image* img) {
	NSImage* i = (__bridge NSImage*)(img->data);
	[i drawInRect:NSMakeRect(x, y, w, h)];
}

void darwin_draw_circle(float x, float y, float d, gg__Color color) {
	NSColor* c = nscolor(color);
	NSRect rect = NSMakeRect(x, y, d * 2, d * 2);
	NSBezierPath* circlePath = [NSBezierPath bezierPath];
	[circlePath appendBezierPathWithOvalInRect:rect];
	[c setFill];
	// [circlePath stroke];
	[circlePath fill];
	// NSRectFill(rect);
}

void darwin_draw_circle_empty(float x, float y, float d, gg__Color color) {
	NSColor* outlineColor = nscolor(color);
	CGFloat outlineWidth = 1.0; //2.0;

	NSRect rect = NSMakeRect(x, y, d * 2, d * 2);
	NSBezierPath* circlePath = [NSBezierPath bezierPath];
	[circlePath appendBezierPathWithOvalInRect:rect];

	[outlineColor setStroke];
	[circlePath setLineWidth:outlineWidth];
	[circlePath stroke];
}