---

import keyboard
import time
import threading
import pyautogui
from pynput.mouse import Controller
import json
import os
import traceback
import colorsys

try:
    import win32api
    import win32con
    DIRECT_INPUT = True
except ImportError:
    DIRECT_INPUT = False

class ParkourBot:
    def __init__(self):
        self.running = False
        self.paused = False
        self.current_pitch = 0
        self.min_pitch = 15
        self.step_down = 2
        
        self.mouse = Controller()
        
        self.timings = {
            'camera_up_angle': 30,
            'camera_up_delay': 0.05,
            'first_w_time': 0.08,       # Первое W
            'w_gap': 0.03,              # Пауза между W
            'second_w_time': 0.05,      # Второе W (удержание перед прыжком)
            'jump_hold': 0.1,
            'post_jump_forward': 0.15,
            'landing_delay': 0.3,
            'stabilization_time': 0.1,
            'camera_reset_delay': 0.1,
            'block_check_delay': 0.05,
            'scan_rotation_speed': 0.02,
            'scan_angle_step': 10,
            'resume_delay': 0.3,
            'color_tolerance': 15,
        }
        
        self.base_colors = [
            (255, 23, 23),
            (255, 29, 30),
            (188, 17, 18),
            (196, 17, 18),
        ]
        
        self.generate_color_palette()
        
        self.settings_file = os.path.join(
            os.path.dirname(os.path.abspath(__file__)), 
            'parkour_settings.json'
        )
        self.load_settings()
    
    def generate_color_palette(self):
        self.block_colors = []
        for base in self.base_colors:
            self.block_colors.append(base)
            r, g, b = base
            h, s, v = colorsys.rgb_to_hsv(r/255, g/255, b/255)
            for brightness in [0.8, 0.9, 1.1, 1.2]:
                new_v = min(1.0, max(0.0, v * brightness))
                new_r, new_g, new_b = colorsys.hsv_to_rgb(h, s, new_v)
                color = (int(new_r * 255), int(new_g * 255), int(new_b * 255))
                if color not in self.block_colors:
                    self.block_colors.append(color)
            for saturation in [0.8, 0.9, 1.1, 1.2]:
                new_s = min(1.0, max(0.0, s * saturation))
                new_r, new_g, new_b = colorsys.hsv_to_rgb(h, new_s, v)
                color = (int(new_r * 255), int(new_g * 255), int(new_b * 255))
                if color not in self.block_colors:
                    self.block_colors.append(color)
        print("Сгенерировано оттенков красного: " + str(len(self.block_colors)))
    
    def load_settings(self):
        try:
            if os.path.exists(self.settings_file):
                with open(self.settings_file, 'r', encoding='utf-8') as f:
                    saved = json.load(f)
                    for key in self.timings:
                        if key in saved:
                            self.timings[key] = saved[key]
                    if 'base_colors' in saved:
                        self.base_colors = [tuple(c) for c in saved['base_colors']]
                        self.generate_color_palette()
                print("Настройки загружены")
        except Exception as e:
            print("Стандартные настройки: " + str(e))
    
    def save_settings(self):
        try:
            save_data = {}
            for key, value in self.timings.items():
                save_data[key] = value
            save_data['base_colors'] = self.base_colors
            
            try:
                with open(self.settings_file, 'w', encoding='utf-8') as f:
                    json.dump(save_data, f, indent=4, ensure_ascii=False)
                print("Настройки сохранены")
            except PermissionError:
                home_dir = os.path.expanduser('~')
                backup_file = os.path.join(home_dir, 'parkour_settings.json')
                with open(backup_file, 'w', encoding='utf-8') as f:
                    json.dump(save_data, f, indent=4, ensure_ascii=False)
                print("Настройки сохранены в " + backup_file)
        except Exception as e:
            print("Ошибка сохранения: " + str(e))
    
    def release_keys(self):
        try:
            keyboard.release('w')
            keyboard.release('space')
        except:
            pass
    
    def move_mouse(self, dx, dy):
        try:
            if DIRECT_INPUT:
                win32api.mouse_event(win32con.MOUSEEVENTF_MOVE, int(dx), int(dy), 0, 0)
            else:
                pos = self.mouse.position
                self.mouse.position = (pos[0] + dx, pos[1] + dy)
        except:
            pass
    
    def color_match_percent(self, color1, color2):
        try:
            tolerance = self.timings['color_tolerance']
            dr = abs(color1[0] - color2[0])
            dg = abs(color1[1] - color2[1])
            db = abs(color1[2] - color2[2])
            max_diff = 255 * 3
            actual_diff = dr + dg + db
            similarity = 100 - (actual_diff / max_diff * 100)
            return similarity >= (100 - tolerance)
        except:
            return False
    
    def find_block(self):
        try:
            screenshot = pyautogui.screenshot()
            w, h = screenshot.size
            cx, cy = w // 2, h // 2
            pixel = screenshot.getpixel((cx, cy))
            for color in self.block_colors:
                if self.color_match_percent(pixel[:3], color):
                    return True, (cx, cy)
            for dx in [-5, 0, 5]:
                for dy in [-5, 0, 5]:
                    if dx == 0 and dy == 0:
                        continue
                    try:
                        pixel = screenshot.getpixel((cx + dx, cy + dy))
                        for color in self.block_colors:
                            if self.color_match_percent(pixel[:3], color):
                                return True, (cx + dx, cy + dy)
                    except:
                        continue
            return False, None
        except:
            return False, None
    
    def reset_camera(self):
        if self.current_pitch > 0:
            steps = 5
            step = self.current_pitch / steps
            for _ in range(steps):
                if not self.running or self.paused:
                    break
                self.move_mouse(0, -step)
                time.sleep(self.timings['camera_reset_delay'] / steps)
            self.current_pitch = 0
    
    def rotate_scan(self):
        for _ in range(2):
            if not self.running or self.paused:
                break
            for _ in range(0, 180, self.timings['scan_angle_step']):
                if not self.running or self.paused:
                    break
                self.move_mouse(self.timings['scan_angle_step'], 0)
                time.sleep(self.timings['scan_rotation_speed'])
                found, pos = self.find_block()
                if found:
                    return True, pos
        return False, None
    
    def look_down(self):
        if self.current_pitch < self.min_pitch:
            self.move_mouse(0, self.step_down)
            self.current_pitch += self.step_down
    
    def do_jump(self):
        """Прыжок: подъём камеры -> W W -> прыжок (камера не опускается)"""
        print("=" * 40)
        print("ПРЫЖОК")
        print("=" * 40)
        
        # 1. Поднимаем камеру
        print("  Подъём камеры...")
        angle = self.timings['camera_up_angle']
        for _ in range(5):
            self.move_mouse(0, -angle)
            time.sleep(0.005)
        time.sleep(self.timings['camera_up_delay'])
        
        # 2. Ускорение - двойное W
        print("  Ускорение (W W)...")
        # Первое W
        keyboard.press('w')
        time.sleep(self.timings['first_w_time'])
        keyboard.release('w')
        # Пауза между W
        time.sleep(self.timings['w_gap'])
        # Второе W (удерживаем)
        keyboard.press('w')
        time.sleep(self.timings['second_w_time'])
        
        # 3. Прыжок (W всё ещё зажат!)
        print("  Прыжок!")
        keyboard.press('space')
        time.sleep(self.timings['jump_hold'])
        keyboard.release('space')
        
        # 4. Летим вперёд с поднятой камерой
        time.sleep(self.timings['post_jump_forward'])
        
        # 5. Отпускаем W
        keyboard.release('w')
        
        # 6. Приземление (камера всё ещё поднята)
        print("  Приземление...")
        time.sleep(self.timings['landing_delay'])
        
        # 7. Опускаем камеру только после приземления
        for _ in range(5):
            self.move_mouse(0, angle)
            time.sleep(0.005)
        
        time.sleep(self.timings['stabilization_time'])
        
        print("Готово")
        print("=" * 40)
    
    def test_color_detection(self):
        print("=" * 50)
        print("ТЕСТ ОПРЕДЕЛЕНИЯ ЦВЕТОВ")
        print("=" * 50)
        print("Допуск цвета: " + str(self.timings['color_tolerance']) + "%")
        print("Наведитесь на блок и нажмите Enter")
        input("> ")
        
        try:
            screenshot = pyautogui.screenshot()
            w, h = screenshot.size
            cx, cy = w // 2, h // 2
            pixel = screenshot.getpixel((cx, cy))
            
            print("Цвет под прицелом: RGB" + str(pixel[:3]))
            
            best_match = None
            best_similarity = 0
            
            for color in self.block_colors:
                dr = abs(pixel[0] - color[0])
                dg = abs(pixel[1] - color[1])
                db = abs(pixel[2] - color[2])
                similarity = 100 - ((dr + dg + db) / (255 * 3) * 100)
                
                if similarity > best_similarity:
                    best_similarity = similarity
                    best_match = color
            
            if best_match:
                print("Лучшее совпадение: RGB" + str(best_match))
                print("Схожесть: " + str(round(best_similarity, 1)) + "%")
                
                if best_similarity >= (100 - self.timings['color_tolerance']):
                    print("Блок ОБНАРУЖЕН!")
                else:
                    print("Блок НЕ обнаружен")
            else:
                print("Нет совпадений с палитрой")
                
        except Exception as e:
            print("Ошибка теста: " + str(e))
    
    def test_jump_sequence(self):
        """Тест последовательности прыжка"""
        print("=" * 50)
        print("ТЕСТ ПРЫЖКА")
        print("=" * 50)
        print("1. Подъём камеры")
        print("2. W W (двойное ускорение)")
        print("3. Прыжок")
        print("4. Приземление")
        print("5. Опускание камеры")
        print("Переключитесь на Minecraft!")
        
        for i in range(3, 0, -1):
            print(str(i) + "...")
            time.sleep(1)
        
        print("ТЕСТ!")
        self.do_jump()
        print("Готово")
    
    def main_loop(self):
        print("Запуск...")
        
        while self.running:
            if self.paused:
                time.sleep(0.1)
                continue
            
            try:
                found, _ = self.find_block()
                
                if found:
                    print("Блок найден!")
                    self.do_jump()
                    time.sleep(self.timings['resume_delay'])
                else:
                    found_scan, _ = self.rotate_scan()
                    
                    if found_scan:
                        self.do_jump()
                    else:
                        self.look_down()
                        if self.current_pitch >= self.min_pitch:
                            self.reset_camera()
                    
                    time.sleep(self.timings['block_check_delay'])
                    
            except Exception as e:
                print("Ошибка: " + str(e))
                self.release_keys()
                time.sleep(0.5)
    
    def setup_menu(self):
        print("=" * 50)
        print("НАСТРОЙКА БОТА")
        print("=" * 50)
        
        if os.path.exists(self.settings_file):
            use = input("Загрузить сохранённые настройки? (Y/n): ").strip().lower()
            if use == 'n':
                print("Настройка заново...")
            else:
                print("Загружены сохранённые настройки")
                self.show_settings()
                return
        
        print("Вводите значения или Enter (пропустить)")
        print("Введите 0 для сброса на стандартное")
        print("Камера поднимается и НЕ опускается до приземления!")
        
        settings = {
            "ЦВЕТА": [
                ('color_tolerance', 'Допуск цвета (%)', 5, 50, '%'),
            ],
            "ПОДЪЁМ КАМЕРЫ (держит весь прыжок)": [
                ('camera_up_angle', 'Сила подъёма камеры', 10, 100, 'пикс'),
                ('camera_up_delay', 'Задержка после подъёма', 0.01, 0.2, 'сек'),
            ],
            "УСКОРЕНИЕ (ДВОЙНОЕ W)": [
                ('first_w_time', 'Первое W (разгон)', 0.03, 0.2, 'сек'),
                ('w_gap', 'Пауза между W', 0.01, 0.1, 'сек'),
                ('second_w_time', 'Второе W (удержание)', 0.03, 0.2, 'сек'),
            ],
            "ПРЫЖОК": [
                ('jump_hold', 'Длительность прыжка', 0.05, 0.3, 'сек'),
                ('post_jump_forward', 'Полёт вперёд', 0.05, 0.5, 'сек'),
            ],
            "ПРИЗЕМЛЕНИЕ": [
                ('landing_delay', 'Ожидание приземления', 0.1, 0.8, 'сек'),
                ('stabilization_time', 'Стабилизация', 0.05, 0.3, 'сек'),
            ],
        }
        
        defaults = self.timings.copy()
        
        for group, params in settings.items():
            print("--- " + group + " ---")
            for item in params:
                key, desc, min_v, max_v, unit = item
                cur = self.timings[key]
                print(desc)
                print("Текущее: " + str(cur) + " " + unit + " (от " + str(min_v) + " до " + str(max_v) + ")")
                
                inp = input("> ").strip()
                
                if inp == "":
                    print("Оставлено: " + str(cur) + " " + unit)
                elif inp == "0":
                    self.timings[key] = defaults[key]
                    print("Сброшено: " + str(defaults[key]) + " " + unit)
                else:
                    try:
                        val = float(inp)
                        if min_v <= val <= max_v:
                            self.timings[key] = val
                            print("Установлено: " + str(val) + " " + unit)
                        else:
                            print("Значение должно быть от " + str(min_v) + " до " + str(max_v))
                    except:
                        print("Введите число!")
        
        print("=" * 50)
        print("Хотите протестировать прыжок?")
        test = input("(Y/n): ").strip().lower()
        if test != 'n':
            self.test_jump_sequence()
        
        self.save_settings()
        print("Готово")
    
    def show_settings(self):
        print("Текущие настройки:")
        print("  Камера: " + str(self.timings['camera_up_angle']) + " пикс")
        print("  W W: " + str(self.timings['first_w_time']) + " + " + str(self.timings['w_gap']) + " + " + str(self.timings['second_w_time']) + " сек")
        print("  Прыжок: " + str(self.timings['jump_hold']) + " сек")
        print("  Допуск цвета: " + str(self.timings['color_tolerance']) + "%")
    
    def start(self):
        self.setup_menu()
        
        print("=" * 50)
        print("БОТ ЗАПУЩЕН!")
        print("F8 - Пауза | F9 - Стоп | F10 - Тест цвета")
        print("=" * 50)
        
        self.running = True
        self.paused = False
        
        thread = threading.Thread(target=self.main_loop)
        thread.daemon = True
        thread.start()
        
        f8_state = False
        f10_state = False
        
        while self.running:
            try:
                if keyboard.is_pressed('F8'):
                    if not f8_state:
                        f8_state = True
                        self.paused = not self.paused
                        if self.paused:
                            print("ПАУЗА")
                            self.release_keys()
                        else:
                            print("ПРОДОЛЖЕНИЕ")
                            time.sleep(0.3)
                else:
                    f8_state = False
                
                if keyboard.is_pressed('F10'):
                    if not f10_state:
                        f10_state = True
                        was_paused = self.paused
                        self.paused = True
                        self.test_color_detection()
                        self.paused = was_paused
                else:
                    f10_state = False
                
                if keyboard.is_pressed('F9'):
                    self.running = False
                    print("СТОП")
                    break
                
                time.sleep(0.05)
                
            except KeyboardInterrupt:
                break
        
        self.release_keys()
        print("Выход")

if __name__ == "__main__":
    print("=" * 50)
    print("БОТ ПАРКУРА MINECRAFT")
    print("=" * 50)
    print("Порядок прыжка:")
    print("  1. Подъём камеры")
    print("  2. W W (двойное ускорение)")
    print("  3. Прыжок (камера поднята)")
    print("  4. Приземление")
    print("  5. Опускание камеры")
    print("=" * 50)
    
    input("Нажмите Enter для запуска...")
    
    bot = ParkourBot()
    bot.start()