#!/usr/bin/env python3

import sys
import os
import pygame
import argparse
import shutil


class Image:
	def __init__(self, path, filename):
		self.image = pygame.image.load(path + filename)
		self.path = path
		self.filename = filename

	def fit(self, surface, screen_width, screen_height):
		width = self.image.get_width()
		height = self.image.get_height()

		delta_w = width - screen_width
		delta_h = height - screen_height

		scale_w = 1

		if delta_w > 0:
			scale_w = screen_width / width

		scale_h = 1

		if delta_h > 0:
			scale_h = screen_height / height

		scale = min(scale_w, scale_h)

		return pygame.transform.scale_by(self.image, scale)

	def draw(self, surface):
		image = self.fit(self.image, surface.get_width(), surface.get_height())

		x = (surface.get_width() - image.get_width()) * 0.5
		y = surface.get_height() * 0.5 - image.get_height() * 0.5

		surface.blit(image, (x, y))


def load_image(index, images, screen):
	screen.fill((0, 0, 0))
	images[index].draw(screen)
	pygame.display.update()


def next_image(current_image, skip, images, screen):
	if len(images) < 1:
		screen.fill((0, 0, 0))
		pygame.display.update()
		return

	index = (current_image + skip) % len(images)

	load_image(index, images, screen)

	return index


def sanitize_path(path):
	if not path.endswith('/'):
		path += '/'

	return path


def main():
	pygame.init()

	parser = argparse.ArgumentParser(prog='Image Tinder', description='Swipe through your images and press M if you have a match.')
	parser.add_argument('image_path', help='The folder with your images')
	parser.add_argument('target_path_match', help='The folder for your matched images')
	parser.add_argument('target_path_remain', help='The folder for the remaining images')

	args = parser.parse_args()
	path = sanitize_path(args.image_path)
	path_matches = sanitize_path(args.target_path_match)
	path_remain = sanitize_path(args.target_path_remain)

	if not os.path.isdir(args.target_path_match):
		raise Exception('Match path %s does not exist!' % args.target_path_match)

	if not os.path.isdir(args.target_path_remain):
		raise Exception('Remain path %s does not exist!' % args.target_path_remain)

	images = []
	matches = []

	for file in os.listdir(args.image_path):
		filepath = path + file

		if not filepath.lower().endswith('.jpg') and not filepath.lower().endswith('.jpeg'):
			continue

		images.append(Image(path, file))

	screen = pygame.display.set_mode(pygame.display.list_modes()[0], pygame.FULLSCREEN)

	current_image = 0

	load_image(current_image, images, screen)

	while True:
		for event in pygame.event.get():
			if event.type == pygame.QUIT:
				return

			elif event.type == pygame.KEYDOWN:
				match event.key:
					case pygame.K_ESCAPE:
						return

					case pygame.K_LEFT:
						current_image = next_image(current_image, -1, images, screen)

					case pygame.K_RIGHT:
						current_image = next_image(current_image, 1, images, screen)

					case pygame.K_m:
						if len(images) > 0:
							matches.append(images[current_image])
							images.pop(current_image)
							current_image = next_image(current_image, 0, images, screen)

					case pygame.K_RETURN:
						for m in matches:
							shutil.move(m.path + m.filename, path_matches + m.filename)

						for i in images:
							shutil.move(i.path + i.filename, path_remain + i.filename)

						return


if __name__ == '__main__':
	main()