import tkinter as tk
from tkinter import ttk, filedialog, messagebox
from PIL import Image, ImageDraw, UnidentifiedImageError
import os
import sys

# tkinterdnd2 をインポート (フォールバック対応)
try:
    from tkinterdnd2 import DND_FILES, TkinterDnD
except ImportError:
    TkinterDnD = None
    print("tkinterdnd2ライブラリが見つかりません。ドラッグアンドドロップ機能は無効になります。", file=sys.stderr)
    print("インストールするには 'pip install tkinterdnd2' を実行してください。", file=sys.stderr)


class AutoRoundedTransparentIconGenerator:
    def __init__(self, master_root):
        self.master = master_root
        self.master.title("Auto Rounded Transparent Icon Generator")
        self.master.geometry("580x680") # 少し高さを増やす

        self.style = ttk.Style()
        self.style.theme_use('clam')

        self.BG_COLOR = "white"
        self.PRIMARY_COLOR = "#00529B"
        self.SECONDARY_COLOR = "#0078D4"
        self.LIGHT_BLUE_BG = "#F0F8FF"
        self.TEXT_COLOR_ON_LIGHT_BG = self.PRIMARY_COLOR
        self.TEXT_COLOR_ON_DARK_BG = "white"
        self.BUTTON_ACTIVE_BG = "#003C75"
        self.DISABLED_FG_COLOR = "#A0A0A0"
        self.DISABLED_BG_COLOR = "#E0E0E0"
        self.ERROR_COLOR = "red"
        self.SUCCESS_COLOR = "green"

        self.FONT_NORMAL = ("Segoe UI", 10)
        self.FONT_BOLD = ("Segoe UI Bold", 11)
        self.FONT_LARGE = ("Segoe UI", 14)
        self.FONT_TITLE = ("Segoe UI Semibold", 16)

        self.master.configure(bg=self.BG_COLOR)

        self.style.configure("TFrame", background=self.BG_COLOR)
        self.style.configure("TLabel", background=self.BG_COLOR, foreground=self.TEXT_COLOR_ON_LIGHT_BG, font=self.FONT_NORMAL)
        self.style.configure("TLabelframe", background=self.BG_COLOR, bordercolor=self.SECONDARY_COLOR)
        self.style.configure("TLabelframe.Label", background=self.BG_COLOR, foreground=self.PRIMARY_COLOR, font=self.FONT_BOLD)
        self.style.configure("TButton", background=self.SECONDARY_COLOR, foreground=self.TEXT_COLOR_ON_DARK_BG,
                             font=self.FONT_BOLD, padding=6, relief="raised", borderwidth=1, bordercolor=self.PRIMARY_COLOR)
        self.style.map("TButton", background=[('active', self.BUTTON_ACTIVE_BG), ('disabled', self.DISABLED_BG_COLOR)],
                       foreground=[('disabled', self.DISABLED_FG_COLOR)])
        self.style.configure("TEntry", fieldbackground="white", foreground=self.TEXT_COLOR_ON_LIGHT_BG, insertcolor=self.TEXT_COLOR_ON_LIGHT_BG,
                             bordercolor=self.SECONDARY_COLOR, font=self.FONT_NORMAL)
        self.style.map("TEntry", bordercolor=[('focus', self.PRIMARY_COLOR)], fieldbackground=[('readonly', "#F0F0F0")])
        self.style.configure("TCheckbutton", background=self.BG_COLOR, foreground=self.TEXT_COLOR_ON_LIGHT_BG, font=self.FONT_NORMAL) # チェックボタンのスタイル
        self.style.map("TCheckbutton",
                       indicatorcolor=[('selected', self.PRIMARY_COLOR), ('!selected', self.SECONDARY_COLOR)],
                       foreground=[('disabled', self.DISABLED_FG_COLOR)])


        self.input_path = tk.StringVar()
        self.output_path = tk.StringVar()
        self.radius_var = tk.IntVar(value=25)
        self.make_fully_round_var = tk.BooleanVar(value=False) # ★ 完全に丸くするオプション用

        self.dnd_enabled = TkinterDnD is not None

        main_frame = ttk.Frame(self.master, padding="10")
        main_frame.pack(expand=True, fill=tk.BOTH)

        title_label = ttk.Label(main_frame, text="画像角丸・透過ツール", font=self.FONT_TITLE, foreground=self.PRIMARY_COLOR)
        title_label.pack(pady=(10,15))

        drop_frame_container = tk.Frame(main_frame, bg=self.LIGHT_BLUE_BG,
                                        highlightbackground=self.PRIMARY_COLOR,
                                        highlightthickness=2, relief="solid",
                                        width=500, height=160) # 少し高さを調整
        drop_frame_container.pack(pady=10, padx=10, fill="x")
        drop_frame_container.pack_propagate(False)

        self.drop_label_text = tk.StringVar()
        drop_label_message = "ここに画像をドラッグ＆ドロップ" if self.dnd_enabled else "ドラッグ＆ドロップ無効"
        self.drop_label_text.set(drop_label_message + "\nまたは下から選択")

        drop_label = tk.Label(drop_frame_container, textvariable=self.drop_label_text,
                              font=self.FONT_LARGE, wraplength=480, justify="center",
                              bg=self.LIGHT_BLUE_BG, fg=self.TEXT_COLOR_ON_LIGHT_BG)
        drop_label.pack(expand=True, fill="both", padx=10, pady=10)

        if self.dnd_enabled:
            drop_label.drop_target_register(DND_FILES)
            drop_label.dnd_bind('<<Drop>>', self.handle_drop)
            drop_frame_container.drop_target_register(DND_FILES)
            drop_frame_container.dnd_bind('<<Drop>>', self.handle_drop)
        else:
            drop_label.config(text="D&D無効: tkinterdnd2未導入\n下からファイルを選択してください")

        settings_frame = ttk.Frame(main_frame)
        settings_frame.pack(pady=(5,0), padx=5, fill="x") # 上のpady調整

        input_outer_frame = ttk.Frame(settings_frame)
        input_outer_frame.pack(pady=(5,2), fill="x")
        ttk.Label(input_outer_frame, text="入力画像:").pack(side=tk.LEFT, padx=(0,5))
        self.input_entry = ttk.Entry(input_outer_frame, textvariable=self.input_path, width=45, state="readonly")
        self.input_entry.pack(side=tk.LEFT, expand=True, fill="x", ipady=2, padx=(0,5))
        ttk.Button(input_outer_frame, text="参照", command=self.select_input_file, width=6).pack(side=tk.LEFT)

        output_outer_frame = ttk.Frame(settings_frame)
        output_outer_frame.pack(pady=(2,5), fill="x")
        ttk.Label(output_outer_frame, text="出力先:  ").pack(side=tk.LEFT, padx=(0,5))
        self.output_entry = ttk.Entry(output_outer_frame, textvariable=self.output_path, width=45, state="readonly")
        self.output_entry.pack(side=tk.LEFT, expand=True, fill="x", ipady=2, padx=(0,5))
        ttk.Button(output_outer_frame, text="参照", command=self.select_output_file, width=6).pack(side=tk.LEFT)

        # --- 角丸設定フレーム ---
        radius_options_frame = ttk.LabelFrame(main_frame, text="角丸設定", padding="10")
        radius_options_frame.pack(pady=10, padx=10, fill="x")

        # ★ 完全に丸くするチェックボックス
        self.fully_round_check = ttk.Checkbutton(radius_options_frame, text="完全に丸くする (円形切り抜き)",
                                                 variable=self.make_fully_round_var,
                                                 command=self.toggle_radius_slider_state,
                                                 style="TCheckbutton")
        self.fully_round_check.pack(anchor=tk.W, pady=(0,5))


        radius_slider_frame = ttk.Frame(radius_options_frame) # スケールとラベルを同じ行に
        radius_slider_frame.pack(fill="x")

        ttk.Label(radius_slider_frame, text="角の丸み:", style="TLabel").pack(side=tk.LEFT, padx=(0,10))
        self.radius_scale = tk.Scale(radius_slider_frame, from_=5, to_=200, orient=tk.HORIZONTAL, # ★ to_ を変更
                                     variable=self.radius_var,
                                     bg=self.BG_COLOR, fg=self.TEXT_COLOR_ON_LIGHT_BG, troughcolor=self.SECONDARY_COLOR,
                                     highlightthickness=0, activebackground=self.PRIMARY_COLOR, sliderrelief="flat",
                                     font=self.FONT_NORMAL, length=300) # length調整
        self.radius_scale.pack(side=tk.LEFT, expand=True, fill="x")

        button_frame = ttk.Frame(main_frame)
        button_frame.pack(pady=(10,5), padx=10, fill="x") # padx追加
        self.process_button = ttk.Button(button_frame, text="画像を処理", command=self.process_image, style="TButton")
        self.process_button.pack(expand=True, fill="x", ipady=5)

        self.status_label = ttk.Label(main_frame, text="", font=self.FONT_NORMAL, anchor=tk.CENTER)
        self.status_label.pack(pady=(5,10), fill="x")

        self.update_process_button_state()
        self.toggle_radius_slider_state() # 初期状態でスライダーの状態を更新

    # ★ スライダーの有効/無効を切り替えるメソッド
    def toggle_radius_slider_state(self):
        if self.make_fully_round_var.get():
            self.radius_scale.config(state=tk.DISABLED)
            # 必要であれば、完全に丸くするときの半径の目安をラベルに表示するなど
        else:
            self.radius_scale.config(state=tk.NORMAL)

    def update_process_button_state(self):
        # (変更なし)
        if self.input_path.get() and self.output_path.get():
            self.process_button.config(state=tk.NORMAL)
        else:
            self.process_button.config(state=tk.DISABLED)

    def handle_drop(self, event):
        # (変更なし、ただしSyntaxWarning修正済み)
        if not self.dnd_enabled: return

        filepath = event.data
        if filepath.startswith('{') and filepath.endswith('}'):
            filepath = filepath[1:-1]

        filepaths = []
        if r'\} \{' in filepath: # raw文字列
            filepaths = [p[1:-1] for p in filepath.split(r'\} \{')]
        else:
            filepaths.append(filepath)

        if filepaths:
            first_filepath = filepaths[0].strip()
            if os.path.isfile(first_filepath):
                if self.is_image_file(first_filepath):
                    self.input_path.set(first_filepath)
                    self.drop_label_text.set(f"読込: {os.path.basename(first_filepath)}")
                    self.status_label.config(text="画像が読み込まれました。", foreground=self.PRIMARY_COLOR)

                    input_dir = os.path.dirname(first_filepath)
                    base, ext = os.path.splitext(os.path.basename(first_filepath))
                    default_output_name = f"{base}_rounded.png"
                    self.output_path.set(os.path.join(input_dir, default_output_name))
                else:
                    self.status_label.config(text="エラー: 画像ファイルではありません。", foreground=self.ERROR_COLOR)
                    drop_message_base = "ここに画像をドラッグ＆ドロップ" if self.dnd_enabled else "D&D無効"
                    self.drop_label_text.set(drop_message_base + "\nまたは下から選択")
            else:
                self.status_label.config(text="エラー: 有効なファイルではありません。", foreground=self.ERROR_COLOR)
                drop_message_base = "ここに画像をドラッグ＆ドロップ" if self.dnd_enabled else "D&D無効"
                self.drop_label_text.set(drop_message_base + "\nまたは下から選択")
        self.update_process_button_state()

    def is_image_file(self, filepath):
        # (変更なし)
        try:
            img_test = Image.open(filepath)
            img_test.verify()
            img_test.close()
            return True
        except (IOError, UnidentifiedImageError, FileNotFoundError, SyntaxError):
            return False

    def select_input_file(self):
        # (変更なし)
        filepath = filedialog.askopenfilename(
            title="入力画像を選択",
            filetypes=(("画像ファイル", "*.png *.jpg *.jpeg *.gif *.bmp *.tiff"),
                       ("すべてのファイル", "*.*"))
        )
        if filepath:
            self.input_path.set(filepath)
            self.drop_label_text.set(f"読込: {os.path.basename(filepath)}")
            input_dir = os.path.dirname(filepath)
            base, ext = os.path.splitext(os.path.basename(filepath))
            default_output_name = f"{base}_rounded.png"
            self.output_path.set(os.path.join(input_dir, default_output_name))
            self.status_label.config(text="入力画像を選択しました。", foreground=self.PRIMARY_COLOR)
        self.update_process_button_state()

    def select_output_file(self):
        # (変更なし)
        input_filename = os.path.basename(self.input_path.get())
        if input_filename:
            name, _ = os.path.splitext(input_filename)
            default_name = f"{name}_rounded.png"
        else:
            default_name = "rounded_image.png"

        filepath = filedialog.asksaveasfilename(
            title="出力先を選択",
            defaultextension=".png",
            initialfile=default_name,
            filetypes=(("PNG画像", "*.png"), ("すべてのファイル", "*.*"))
        )
        if filepath:
            self.output_path.set(filepath)
            self.status_label.config(text="出力先を選択しました。", foreground=self.PRIMARY_COLOR)
        self.update_process_button_state()

    def round_corners(self, image, radius):
        # (変更なし)
        image = image.convert("RGBA")
        width, height = image.size
        mask = Image.new('L', (width, height), 0)
        draw = ImageDraw.Draw(mask)
        draw.rounded_rectangle((0, 0, width, height), radius=radius, fill=255)
        image.putalpha(mask)
        return image

    def process_image(self):
        input_p = self.input_path.get()
        output_p = self.output_path.get()
        # ★ radiusの取得方法を変更
        # radius = self.radius_var.get() # スライダーの値は直接使わない場合がある

        if not input_p or not output_p:
            messagebox.showerror("エラー", "入力画像と出力先を選択してください。")
            return

        try:
            self.status_label.config(text="処理中...", foreground=self.PRIMARY_COLOR)
            self.master.update_idletasks()

            img = Image.open(input_p)
            max_possible_radius = min(img.width, img.height) // 2

            actual_radius = 0
            if self.make_fully_round_var.get():
                actual_radius = max_possible_radius
                self.radius_var.set(actual_radius) # スライダーにも反映（表示のため）
                print(f"完全に丸くするため、半径を {actual_radius} に設定しました。")
            else:
                slider_radius_val = self.radius_var.get()
                if slider_radius_val > max_possible_radius:
                    actual_radius = max_possible_radius
                    self.radius_var.set(actual_radius) # スライダーの値を調整して反映
                    if slider_radius_val != actual_radius: # 実際に調整された場合のみ警告
                        messagebox.showwarning("半径調整", f"指定された半径({slider_radius_val})は大きすぎたため、{actual_radius}ピクセルに調整しました。")
                else:
                    actual_radius = slider_radius_val

            if actual_radius < 0: actual_radius = 0 # 半径が負にならないように

            rounded_img = self.round_corners(img.copy(), actual_radius)
            rounded_img.save(output_p, "PNG")
            img.close()

            self.status_label.config(text=f"処理完了: {os.path.basename(output_p)} を保存しました。", foreground=self.SUCCESS_COLOR)
            drop_message_base = "ここに画像をドラッグ＆ドロップ" if self.dnd_enabled else "D&D無効"
            self.drop_label_text.set(drop_message_base + "\nまたは下から選択")

        except FileNotFoundError:
            messagebox.showerror("エラー", f"入力ファイルが見つかりません: {input_p}")
            self.status_label.config(text="エラー: 入力ファイルが見つかりません。", foreground=self.ERROR_COLOR)
        except UnidentifiedImageError:
             messagebox.showerror("エラー", f"画像ファイルとして認識できません: {input_p}")
             self.status_label.config(text="エラー: 画像ファイル形式が無効です。", foreground=self.ERROR_COLOR)
        except IOError as e:
            messagebox.showerror("エラー", f"ファイル入出力エラー: {e}\n画像が破損しているかサポート外の形式の可能性。")
            self.status_label.config(text=f"エラー: ファイルIOエラー ({e})", foreground=self.ERROR_COLOR)
        except Exception as e:
            messagebox.showerror("エラー", f"予期せぬエラーが発生しました: {e}")
            self.status_label.config(text=f"エラー: {e}", foreground=self.ERROR_COLOR)
        finally:
            self.update_process_button_state()

    def run(self):
        self.master.mainloop()

if __name__ == "__main__":
    if TkinterDnD:
        root = TkinterDnD.Tk()
    else:
        root = tk.Tk()

    app = AutoRoundedTransparentIconGenerator(root)
    app.run()