# -*- coding: utf-8 -*-
import sys
import os
import hashlib
import shutil
import threading
import time
from datetime import datetime, timedelta

from PyQt5.QtWidgets import (
    QApplication, QMainWindow, QPushButton, QVBoxLayout, QWidget,
    QFileDialog, QProgressBar, QLabel, QTextEdit, QHBoxLayout,
    QFrame, QSplitter, QGraphicsDropShadowEffect, QGridLayout
)
from PyQt5.QtCore import Qt, pyqtSignal, QObject, QTimer, QEasingCurve, QPropertyAnimation, QRect
from PyQt5.QtGui import QFont, QPalette, QColor, QPixmap, QPainter, QBrush, QPen, QLinearGradient

LOG_FILE = 'backup_log.txt'
HISTORY_FILE = 'backup_history.txt'
MAX_BACKUPS = 5  # 자동 정리: 최대 백업본 개수

def file_hash(path):
    """파일의 SHA256 해시값 계산 (중복 체크용)"""
    h = hashlib.sha256()
    with open(path, 'rb') as f:
        while chunk := f.read(8192):
            h.update(chunk)
    return h.hexdigest()

class BackupWorker(QObject):
    progress = pyqtSignal(int)
    file_copied = pyqtSignal(str)
    status = pyqtSignal(str)
    finished = pyqtSignal()
    stopped = pyqtSignal()

    def __init__(self, src, dst, pause_event, stop_event):
        super().__init__()
        self.src = src
        self.dst = dst
        self.pause_event = pause_event
        self.stop_event = stop_event

    def run(self):
        try:
            self.status.emit("파일 리스트 수집 중...")
            file_list = []
            for root, dirs, files in os.walk(self.src):
                for name in files:
                    full = os.path.join(root, name)
                    rel = os.path.relpath(full, self.src)
                    file_list.append(rel)
            total = len(file_list)
            copied = 0

            # 히스토리 불러오기 (중복, 증분 체크)
            old_hashes = self._load_backup_history()

            # 새 백업본 폴더 (타임스탬프)
            backup_name = datetime.now().strftime('%Y%m%d_%H%M%S')
            backup_path = os.path.join(self.dst, backup_name)
            os.makedirs(backup_path, exist_ok=True)
            self.status.emit(f"백업 폴더 생성: {backup_path}")

            new_hashes = {}

            for idx, rel_path in enumerate(file_list):
                src_file = os.path.join(self.src, rel_path)
                dst_file = os.path.join(backup_path, rel_path)

                if self.stop_event.is_set():
                    self.status.emit("백업 중지됨")
                    self.stopped.emit()
                    return

                self.pause_event.wait()  # 일시정지 시 block

                # 파일 해시로 변경 여부 확인 (증분/중복)
                hash_val = file_hash(src_file)
                new_hashes[rel_path] = hash_val
                if rel_path in old_hashes and old_hashes[rel_path] == hash_val:
                    self.status.emit(f"[SKIP] {rel_path} (변경 없음)")
                else:
                    os.makedirs(os.path.dirname(dst_file), exist_ok=True)
                    shutil.copy2(src_file, dst_file)
                    self.file_copied.emit(rel_path)
                    self.status.emit(f"[COPY] {rel_path}")

                copied += 1
                progress = int(copied / total * 100)
                self.progress.emit(progress)

            self._save_backup_history(new_hashes)
            self.status.emit("백업 완료!")
            self._auto_cleanup()
            self.finished.emit()
        except Exception as e:
            self.status.emit(f"오류 발생: {e}")
            self.finished.emit()

    def _load_backup_history(self):
        if not os.path.exists(HISTORY_FILE):
            return {}
        try:
            with open(HISTORY_FILE, 'r', encoding='utf-8') as f:
                lines = f.readlines()
                return dict(line.strip().split('\t') for line in lines if '\t' in line)
        except:
            return {}

    def _save_backup_history(self, hashes):
        with open(HISTORY_FILE, 'w', encoding='utf-8') as f:
            for rel, hashv in hashes.items():
                f.write(f"{rel}\t{hashv}\n")

    def _auto_cleanup(self):
        """MAX_BACKUPS 초과시 가장 오래된 백업 삭제"""
        try:
            backups = sorted([
                d for d in os.listdir(self.dst)
                if os.path.isdir(os.path.join(self.dst, d))
            ])
            if len(backups) > MAX_BACKUPS:
                for b in backups[:-MAX_BACKUPS]:
                    b_path = os.path.join(self.dst, b)
                    shutil.rmtree(b_path)
                    self.status.emit(f"오래된 백업 삭제: {b_path}")
        except Exception as e:
            self.status.emit(f"[정리오류] {e}")

class ModernButton(QPushButton):
    def __init__(self, text, color="#4A90E2", hover_color="#357ABD"):
        super().__init__(text)
        self.setStyleSheet(f"""
            QPushButton {{
                background: qlineargradient(x1:0, y1:0, x2:0, y2:1,
                    stop:0 {color}, stop:1 {hover_color});
                border: none;
                border-radius: 12px;
                color: white;
                font-weight: bold;
                font-size: 13px;
                padding: 12px 24px;
                min-height: 20px;
            }}
            QPushButton:hover {{
                background: qlineargradient(x1:0, y1:0, x2:0, y2:1,
                    stop:0 {hover_color}, stop:1 {color});
                transform: translateY(-2px);
            }}
            QPushButton:pressed {{
                background: {hover_color};
                transform: translateY(0px);
            }}
            QPushButton:disabled {{
                background: #CCCCCC;
                color: #666666;
            }}
        """)
        
        # 그림자 효과 추가
        shadow = QGraphicsDropShadowEffect()
        shadow.setBlurRadius(10)
        shadow.setOffset(0, 3)
        shadow.setColor(QColor(0, 0, 0, 60))
        self.setGraphicsEffect(shadow)

class ModernProgressBar(QProgressBar):
    def __init__(self):
        super().__init__()
        self.setStyleSheet("""
            QProgressBar {
                border: none;
                border-radius: 8px;
                background-color: #E0E0E0;
                text-align: center;
                font-weight: bold;
                color: #333333;
                height: 20px;
            }
            QProgressBar::chunk {
                background: qlineargradient(x1:0, y1:0, x2:1, y2:0,
                    stop:0 #4A90E2, stop:0.5 #5BA0F2, stop:1 #4A90E2);
                border-radius: 8px;
            }
        """)

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("💾 Modern Backup Tool")
        self.setGeometry(100, 100, 900, 700)
        self.setMinimumSize(800, 600)

        self.src_dir = ""
        self.dst_dir = ""
        self.backup_thread = None
        self.worker = None
        self.pause_event = threading.Event()
        self.pause_event.set()
        self.stop_event = threading.Event()
        self.timer = QTimer()
        self.timer.timeout.connect(self.scheduled_backup)

        self.setup_ui()
        self.apply_dark_theme()

    def setup_ui(self):
        # 메인 위젯
        main_widget = QWidget()
        self.setCentralWidget(main_widget)
        
        # 메인 레이아웃
        main_layout = QVBoxLayout(main_widget)
        main_layout.setSpacing(20)
        main_layout.setContentsMargins(30, 30, 30, 30)

        # 헤더
        header_frame = QFrame()
        header_frame.setFixedHeight(80)
        header_layout = QVBoxLayout(header_frame)
        
        title_label = QLabel("Modern Backup Tool")
        title_label.setStyleSheet("""
            QLabel {
                font-size: 28px;
                font-weight: bold;
                color: #FFFFFF;
                margin-bottom: 5px;
            }
        """)
        
        subtitle_label = QLabel("프로페셔널 백업 솔루션")
        subtitle_label.setStyleSheet("""
            QLabel {
                font-size: 14px;
                color: #AAAAAA;
            }
        """)
        
        header_layout.addWidget(title_label)
        header_layout.addWidget(subtitle_label)
        main_layout.addWidget(header_frame)

        # 폴더 선택 섹션
        folder_frame = QFrame()
        folder_frame.setStyleSheet("""
            QFrame {
                background-color: #2D2D2D;
                border-radius: 15px;
                padding: 20px;
            }
        """)
        folder_layout = QGridLayout(folder_frame)
        
        self.btn_src = ModernButton("📂 원본 폴더 선택", "#28A745", "#1E7E34")
        self.btn_dst = ModernButton("💾 백업 폴더 선택", "#17A2B8", "#117A8B")
        
        self.src_label = QLabel("원본 폴더: 선택되지 않음")
        self.dst_label = QLabel("백업 폴더: 선택되지 않음")
        
        for label in [self.src_label, self.dst_label]:
            label.setStyleSheet("""
                QLabel {
                    color: #CCCCCC;
                    font-size: 12px;
                    padding: 5px;
                    background-color: #1E1E1E;
                    border-radius: 5px;
                }
            """)
        
        folder_layout.addWidget(self.btn_src, 0, 0)
        folder_layout.addWidget(self.src_label, 0, 1)
        folder_layout.addWidget(self.btn_dst, 1, 0)
        folder_layout.addWidget(self.dst_label, 1, 1)
        
        main_layout.addWidget(folder_frame)

        # 컨트롤 섹션
        control_frame = QFrame()
        control_frame.setStyleSheet("""
            QFrame {
                background-color: #2D2D2D;
                border-radius: 15px;
                padding: 20px;
            }
        """)
        control_layout = QHBoxLayout(control_frame)
        
        self.btn_start = ModernButton("▶️ 시작", "#28A745", "#1E7E34")
        self.btn_pause = ModernButton("⏸️ 일시정지", "#FFC107", "#D39E00")
        self.btn_stop = ModernButton("⏹️ 중지", "#DC3545", "#B02A37")
        self.btn_schedule = ModernButton("⏰ 5분마다 자동 백업", "#6F42C1", "#5A2D91")
        
        for btn in [self.btn_start, self.btn_pause, self.btn_stop, self.btn_schedule]:
            control_layout.addWidget(btn)
        
        main_layout.addWidget(control_frame)

        # 진행률 섹션
        progress_frame = QFrame()
        progress_frame.setStyleSheet("""
            QFrame {
                background-color: #2D2D2D;
                border-radius: 15px;
                padding: 20px;
            }
        """)
        progress_layout = QVBoxLayout(progress_frame)
        
        progress_title = QLabel("백업 진행 상태")
        progress_title.setStyleSheet("""
            QLabel {
                font-size: 16px;
                font-weight: bold;
                color: #FFFFFF;
                margin-bottom: 10px;
            }
        """)
        
        self.progress = ModernProgressBar()
        self.label_file = QLabel("대기 중...")
        self.label_file.setStyleSheet("""
            QLabel {
                color: #AAAAAA;
                font-size: 12px;
                padding: 5px;
            }
        """)
        
        progress_layout.addWidget(progress_title)
        progress_layout.addWidget(self.progress)
        progress_layout.addWidget(self.label_file)
        
        main_layout.addWidget(progress_frame)

        # 로그 섹션
        log_frame = QFrame()
        log_frame.setStyleSheet("""
            QFrame {
                background-color: #2D2D2D;
                border-radius: 15px;
                padding: 20px;
            }
        """)
        log_layout = QVBoxLayout(log_frame)
        
        log_title = QLabel("활동 로그")
        log_title.setStyleSheet("""
            QLabel {
                font-size: 16px;
                font-weight: bold;
                color: #FFFFFF;
                margin-bottom: 10px;
            }
        """)
        
        self.text_status = QTextEdit()
        self.text_status.setReadOnly(True)
        self.text_status.setStyleSheet("""
            QTextEdit {
                background-color: #1E1E1E;
                color: #FFFFFF;
                border: 1px solid #404040;
                border-radius: 8px;
                font-family: 'Courier New', monospace;
                font-size: 11px;
                padding: 10px;
            }
        """)
        
        log_layout.addWidget(log_title)
        log_layout.addWidget(self.text_status)
        
        main_layout.addWidget(log_frame)

        # 시그널 연결
        self.btn_src.clicked.connect(self.select_src)
        self.btn_dst.clicked.connect(self.select_dst)
        self.btn_start.clicked.connect(self.start_backup)
        self.btn_pause.clicked.connect(self.pause_resume)
        self.btn_stop.clicked.connect(self.stop_backup)
        self.btn_schedule.clicked.connect(self.toggle_schedule)

        self.scheduled = False

    def apply_dark_theme(self):
        self.setStyleSheet("""
            QMainWindow {
                background: qlineargradient(x1:0, y1:0, x2:0, y2:1,
                    stop:0 #1A1A1A, stop:1 #2D2D2D);
            }
            QWidget {
                background: transparent;
            }
        """)

    def select_src(self):
        d = QFileDialog.getExistingDirectory(self, "원본 폴더 선택")
        if d:
            self.src_dir = d
            self.src_label.setText(f"원본 폴더: {os.path.basename(d)}")
            self.log_status(f"✅ 원본 폴더 선택: {self.src_dir}")

    def select_dst(self):
        d = QFileDialog.getExistingDirectory(self, "백업 폴더 선택")
        if d:
            self.dst_dir = d
            self.dst_label.setText(f"백업 폴더: {os.path.basename(d)}")
            self.log_status(f"✅ 백업 폴더 선택: {self.dst_dir}")

    def start_backup(self):
        if not self.src_dir or not self.dst_dir:
            self.log_status("❌ 원본/백업 폴더를 선택하세요.")
            return
        self.log_status("🚀 백업 시작!")
        self.progress.setValue(0)
        self.stop_event.clear()
        self.pause_event.set()
        self.worker = BackupWorker(self.src_dir, self.dst_dir, self.pause_event, self.stop_event)
        self.worker.progress.connect(self.progress.setValue)
        self.worker.file_copied.connect(self.update_current_file)
        self.worker.status.connect(self.log_status)
        self.worker.finished.connect(self.backup_done)
        self.worker.stopped.connect(self.backup_stopped)

        self.backup_thread = threading.Thread(target=self.worker.run, daemon=True)
        self.backup_thread.start()

    def update_current_file(self, filename):
        self.label_file.setText(f"현재 파일: {filename}")

    def pause_resume(self):
        if not self.worker:
            return
        if self.pause_event.is_set():
            self.pause_event.clear()
            self.btn_pause.setText("▶️ 재개")
            self.log_status("⏸️ 일시정지 중...")
        else:
            self.pause_event.set()
            self.btn_pause.setText("⏸️ 일시정지")
            self.log_status("▶️ 재개")

    def stop_backup(self):
        if self.worker:
            self.stop_event.set()
            self.pause_event.set()
            self.log_status("⏹️ 중지 요청됨.")

    def backup_done(self):
        self.log_status("✅ 백업이 정상적으로 완료되었습니다.")
        self.label_file.setText("완료!")
        self.worker = None

    def backup_stopped(self):
        self.log_status("⚠️ 백업이 중지되었습니다.")
        self.label_file.setText("중지됨")
        self.worker = None

    def log_status(self, msg):
        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        line = f"[{now}] {msg}"
        self.text_status.append(line)
        with open(LOG_FILE, 'a', encoding='utf-8') as f:
            f.write(line + '\n')

    def toggle_schedule(self):
        if self.scheduled:
            self.timer.stop()
            self.scheduled = False
            self.btn_schedule.setText("⏰ 5분마다 자동 백업")
            self.log_status("⏰ 스케줄링 중지")
        else:
            self.timer.start(5 * 60 * 1000)  # 5분
            self.scheduled = True
            self.btn_schedule.setText("⏰ 스케줄 중지")
            self.log_status("⏰ 5분마다 자동 백업 시작")

    def scheduled_backup(self):
        if not self.worker:
            self.start_backup()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    
    # 애플리케이션 스타일 설정
    app.setStyle('Fusion')
    
    # 다크 팔레트 적용
    palette = QPalette()
    palette.setColor(QPalette.Window, QColor(26, 26, 26))
    palette.setColor(QPalette.WindowText, QColor(255, 255, 255))
    palette.setColor(QPalette.Base, QColor(30, 30, 30))
    palette.setColor(QPalette.AlternateBase, QColor(45, 45, 45))
    palette.setColor(QPalette.ToolTipBase, QColor(255, 255, 255))
    palette.setColor(QPalette.ToolTipText, QColor(255, 255, 255))
    palette.setColor(QPalette.Text, QColor(255, 255, 255))
    palette.setColor(QPalette.Button, QColor(45, 45, 45))
    palette.setColor(QPalette.ButtonText, QColor(255, 255, 255))
    palette.setColor(QPalette.BrightText, QColor(255, 0, 0))
    palette.setColor(QPalette.Link, QColor(42, 130, 218))
    palette.setColor(QPalette.Highlight, QColor(42, 130, 218))
    palette.setColor(QPalette.HighlightedText, QColor(0, 0, 0))
    app.setPalette(palette)
    
    win = MainWindow()
    win.show()
    sys.exit(app.exec_())