import cv2
import enum
import numpy as np
import pytesseract
import win32gui
import win32ui
import win32con
import time
from ctypes import windll
from PIL import Image

class State(enum.Enum):
	START = 1
	FISHING = 2
	KEYS = 3
	CATCH = 4
	UNKNOWN = 5
	RESTART = 6
	END = 7


class Screenshot(object):
	def __init__(self, screen=None, image=None):
		self.screen = screen
		self.image = image

	def takeScreenshot(self):
		try:
			left, top, right, bot = win32gui.GetWindowRect(hwnd)
			w = right - left
			h = bot - top

			hwndDC = win32gui.GetWindowDC(hwnd)
			mfcDC  = win32ui.CreateDCFromHandle(hwndDC)
			saveDC = mfcDC.CreateCompatibleDC()

			saveBitMap = win32ui.CreateBitmap()
			saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)

			saveDC.SelectObject(saveBitMap)
			saveDC.BitBlt((0,0),(w, h),mfcDC,(0,0),win32con.SRCCOPY)

			bmpinfo = saveBitMap.GetInfo()
			bmpstr = saveBitMap.GetBitmapBits(True)

			self.screen = Image.frombuffer(
				'RGB',
				(bmpinfo['bmWidth'], bmpinfo['bmHeight']),
				bmpstr, 'raw', 'BGRX', 0, 1)
		finally:
			win32gui.DeleteObject(saveBitMap.GetHandle())
			saveDC.DeleteDC()
			mfcDC.DeleteDC()
			win32gui.ReleaseDC(hwnd, hwndDC)

	def crop(self, area):
		img = self.screen.crop(area) # crop image from screenshot
		img = np.array(img) # convert to cv2 image
		self.image = img[:, :, ::-1].copy()

	def convertToBinary(self):
		img_gray = cv2.cvtColor(self.image, cv2.COLOR_BGR2GRAY) # convert to grayscale
		self.image = cv2.threshold(img_gray,128,255,cv2.THRESH_BINARY_INV|cv2.THRESH_OTSU)[1]

	def getText(self, config = ''):
		return pytesseract.image_to_string(self.image, config=config)

	def getStartPos(self):
		for x in range(0, 50):
			for y in range(0, 50):
				r, g, b = self.image[x, y]
				color = int(r) + int(g) + int(b)
				if color_min < color < color_max:
					return (x+8, y+7)
		return (8, 7)

def GetWindow(title_name, exact = False):
    def enum_handler(hwnd, all_windows):
        all_windows.append((hwnd, win32gui.GetWindowText(hwnd)))
    windows = []
    hwnds = []
    win32gui.EnumWindows(enum_handler, windows)
    if exact:
        hwnds = [hwnd for hwnd, title in windows if title_name == title.lower()]
    else:
        hwnds = [hwnd for hwnd, title in windows if title_name in title.lower()]
    return hwnds[0] if hwnds else None

def GetState():
	screen.takeScreenshot()
	screen.crop(areas['START'])
	screen.convertToBinary()
	if screen.getText().lower().startswith('start'):
		return State.START
	screen.crop(areas['FISHING'])
	screen.convertToBinary()
	if screen.getText().lower().startswith('fishing'):
		return State.FISHING
	screen.crop(areas['BITE'])
	screen.convertToBinary()
	if screen.getText().lower().startswith('bite'):
		return State.FISHING
	return State.UNKNOWN

def Start():
	current_state = State.START
	state_loops = 0
	while current_state != State.END:
		if current_state == State.START:
			screen.takeScreenshot()

			screen.crop(areas['START'])
			screen.convertToBinary()
			if screen.getText().lower().startswith('start'):
				# PRESS SPACE
				print('START - Press space!')
				current_state = State.FISHING
				state_loops = 0
				time.sleep(30)
			else:
				current_state = State.UNKNOWN
		elif current_state == State.FISHING:
			print('FISHING')
			screen.takeScreenshot()

			screen.crop(areas['BITE'])
			screen.convertToBinary()
			if screen.getText().lower().startswith('bite'):
				# PRESS SPACE
				print('BITE - Press space!')
				current_state = State.KEYS
				time.sleep(3)
				continue

			# Check if still fishing
			screen.crop(areas['FISHING'])
			screen.convertToBinary()
			if not screen.getText().lower().startswith('fishing'):
				current_state = State.UNKNOWN
			time.sleep(2)
		elif current_state == State.KEYS:
			screen.takeScreenshot()

			screen.crop(areas['KEYS'])
			pos = screen.getStartPos()
			img = screen.image[pos[0]:pos[0]+y_len, pos[1]:pos[1]+x_len]

			# join letters to remove gaps between
			letters = []
			for i in range(0, 10):
				x_pos = i*l_len + i*gap_len
				letter = img[:, x_pos:x_pos+l_len]
				letters.append(letter)
			img = cv2.hconcat(letters)

			# cv2.imwrite('img.jpg', img)

			keys = screen.getText(config).upper()
			if keys and len(keys) > 1:
				# PRESS KEYS
				print('KEYS: {0}'.format(keys))
				current_state = State.CATCH
			state_loops = state_loops + 1
			if state_loops > 5:
				current_state = State.UNKNOWN
		elif current_state == State.CATCH:
			# PRESS SPACE
			print('CATCHED - START AGAIN!')
			current_state = State.START
			time.sleep(5)
		elif current_state == State.UNKNOWN:
			state = GetState()
			print('UNKNOWN STATUS - retrieved {0}'.format(str(state)))
			if state_loops > 5:
				current_state = State.RESTART
			elif state != State.UNKNOWN:
				current_state = state
				state_loops = 0
			else:
				state_loops = state_loops + 1
			time.sleep(1)
		elif current_state == State.RESTART:
			# TAKE OUT FISHING ROD AND START AGAIN
			# PRESS RIGHT MOUSE BUTTON
			print('RESTART!')
			current_state = State.START
			time.sleep(5)
		else:
			# TAKE OUT FISHING ROD AND START AGAIN
			# PRESS RIGHT MOUSE BUTTON
			print('DEFAULT STATUS = END')
			current_state = State.END

# begin of script
pytesseract.pytesseract.tesseract_cmd = 'C:/Program Files/Tesseract-OCR/tesseract'
config = '-c tessedit_char_whitelist=WASD --psm 13'
color_min = 150
color_max = 200
x_len = 355
y_len = 22
l_len = 18
gap_len = 19
areas = {
	'START': (1098, 52, 1160, 82),
	'FISHING': (830, 52, 912, 85),
	'BITE': (842, 52, 891, 82),
	'KEYS': (750, 330, 1160, 430),
}
hwnd = GetWindow('black desert')
screen = Screenshot()

if hwnd:
	Start()