import tkinter as tk
from tkinter import ttk, messagebox
import socket
import threading
import ipaddress
from datetime import datetime


COMMON_PORTS = {
    21: "FTP", 22: "SSH", 23: "Telnet", 25: "SMTP", 53: "DNS",
    80: "HTTP", 110: "POP3", 143: "IMAP", 443: "HTTPS", 465: "SMTPS",
    587: "SMTP(TLS)", 993: "IMAPS", 995: "POP3S", 1433: "MSSQL",
    3306: "MySQL", 3389: "RDP", 5432: "PostgreSQL", 6379: "Redis",
    8080: "HTTP-Alt", 8443: "HTTPS-Alt", 27017: "MongoDB",
}


class App066:
    """ポートスキャナー"""

    def __init__(self, root):
        self.root = root
        self.root.title("ポートスキャナー")
        self.root.geometry("860x600")
        self.root.configure(bg="#0d1117")
        self._scanning = False
        self._results = []
        self._build_ui()

    def _build_ui(self):
        header = tk.Frame(self.root, bg="#161b22", pady=6)
        header.pack(fill=tk.X)
        tk.Label(header, text="🔍 ポートスキャナー",
                 font=("Noto Sans JP", 12, "bold"),
                 bg="#161b22", fg="#f0f6fc").pack(side=tk.LEFT, padx=12)
        tk.Label(header,
                 text="※ ローカルホストや自分が管理するサーバーのみにスキャンしてください",
                 bg="#161b22", fg="#f48771", font=("Arial", 8)).pack(
            side=tk.LEFT, padx=4)

        # 設定
        cfg_f = tk.Frame(self.root, bg="#0d1117", pady=6)
        cfg_f.pack(fill=tk.X, padx=8)

        tk.Label(cfg_f, text="ホスト:", bg="#0d1117", fg="#c9d1d9",
                 font=("Arial", 9)).pack(side=tk.LEFT)
        self.host_var = tk.StringVar(value="127.0.0.1")
        ttk.Entry(cfg_f, textvariable=self.host_var,
                  width=20).pack(side=tk.LEFT, padx=4)

        tk.Label(cfg_f, text="開始ポート:", bg="#0d1117", fg="#c9d1d9",
                 font=("Arial", 9)).pack(side=tk.LEFT, padx=(8, 2))
        self.start_port = tk.IntVar(value=1)
        ttk.Spinbox(cfg_f, from_=1, to=65535, textvariable=self.start_port,
                    width=7).pack(side=tk.LEFT)

        tk.Label(cfg_f, text="終了ポート:", bg="#0d1117", fg="#c9d1d9",
                 font=("Arial", 9)).pack(side=tk.LEFT, padx=(8, 2))
        self.end_port = tk.IntVar(value=1024)
        ttk.Spinbox(cfg_f, from_=1, to=65535, textvariable=self.end_port,
                    width=7).pack(side=tk.LEFT)

        tk.Label(cfg_f, text="タイムアウト(ms):", bg="#0d1117", fg="#c9d1d9",
                 font=("Arial", 9)).pack(side=tk.LEFT, padx=(8, 2))
        self.timeout_var = tk.IntVar(value=200)
        ttk.Spinbox(cfg_f, from_=50, to=5000, textvariable=self.timeout_var,
                    width=6).pack(side=tk.LEFT)

        tk.Label(cfg_f, text="並列数:", bg="#0d1117", fg="#c9d1d9",
                 font=("Arial", 9)).pack(side=tk.LEFT, padx=(8, 2))
        self.threads_var = tk.IntVar(value=100)
        ttk.Spinbox(cfg_f, from_=1, to=500, textvariable=self.threads_var,
                    width=5).pack(side=tk.LEFT)

        # プリセット
        preset_f = tk.Frame(self.root, bg="#0d1117", pady=2)
        preset_f.pack(fill=tk.X, padx=8)
        tk.Label(preset_f, text="プリセット:", bg="#0d1117", fg="#8b949e",
                 font=("Arial", 9)).pack(side=tk.LEFT)
        for label, start, end in [
                ("よく使うポート(Top20)", 0, 0),
                ("1-1024", 1, 1024),
                ("1-10000", 1, 10000)]:
            ttk.Button(preset_f, text=label,
                       command=lambda s=start, e=end: self._set_preset(s, e)
                       ).pack(side=tk.LEFT, padx=2)
        self.only_common_var = tk.BooleanVar(value=False)
        tk.Checkbutton(preset_f, text="よく使うポートのみ",
                       variable=self.only_common_var,
                       bg="#0d1117", fg="#c9d1d9", selectcolor="#161b22",
                       activebackground="#0d1117").pack(side=tk.LEFT, padx=8)

        # ボタン
        btn_f = tk.Frame(self.root, bg="#0d1117", pady=4)
        btn_f.pack(fill=tk.X, padx=8)
        self.scan_btn = ttk.Button(btn_f, text="▶ スキャン開始",
                                    command=self._start)
        self.scan_btn.pack(side=tk.LEFT, padx=4)
        ttk.Button(btn_f, text="⏹ 停止",
                   command=self._stop).pack(side=tk.LEFT)
        ttk.Button(btn_f, text="📋 コピー",
                   command=self._copy_results).pack(side=tk.LEFT, padx=8)

        # 進捗
        self.progress = ttk.Progressbar(self.root, mode="determinate")
        self.progress.pack(fill=tk.X, padx=8)

        # 結果
        result_f = tk.Frame(self.root, bg="#0d1117")
        result_f.pack(fill=tk.BOTH, expand=True, padx=8, pady=4)
        cols = ("port", "state", "service", "banner")
        self.tree = ttk.Treeview(result_f, columns=cols, show="headings",
                                  selectmode="browse")
        self.tree.heading("port",    text="ポート")
        self.tree.heading("state",   text="状態")
        self.tree.heading("service", text="サービス")
        self.tree.heading("banner",  text="バナー")
        self.tree.column("port",    width=70,  anchor="center")
        self.tree.column("state",   width=70,  anchor="center")
        self.tree.column("service", width=100, anchor="w")
        self.tree.column("banner",  width=400, anchor="w")
        sb = ttk.Scrollbar(result_f, command=self.tree.yview)
        self.tree.configure(yscrollcommand=sb.set)
        sb.pack(side=tk.RIGHT, fill=tk.Y)
        self.tree.pack(fill=tk.BOTH, expand=True)
        self.tree.tag_configure("open",   foreground="#3fb950")
        self.tree.tag_configure("closed", foreground="#6e7681")

        self.status_var = tk.StringVar(value="スキャン設定を確認してください")
        tk.Label(self.root, textvariable=self.status_var,
                 bg="#21262d", fg="#8b949e", font=("Arial", 9),
                 anchor="w", padx=8).pack(fill=tk.X, side=tk.BOTTOM)

    def _set_preset(self, start, end):
        if start == 0:
            self.only_common_var.set(True)
        else:
            self.only_common_var.set(False)
            self.start_port.set(start)
            self.end_port.set(end)

    def _start(self):
        if self._scanning:
            return
        host = self.host_var.get().strip()
        if not host:
            messagebox.showerror("エラー", "ホストを入力してください")
            return
        self._scanning = True
        self._results = []
        self.tree.delete(*self.tree.get_children())

        if self.only_common_var.get():
            ports = sorted(COMMON_PORTS.keys())
        else:
            p1 = self.start_port.get()
            p2 = self.end_port.get()
            ports = list(range(p1, p2 + 1))

        self.progress["maximum"] = len(ports)
        self.progress["value"] = 0
        self.status_var.set(f"スキャン中: {host}  {len(ports)} ポート...")

        threading.Thread(target=self._do_scan,
                          args=(host, ports), daemon=True).start()

    def _do_scan(self, host, ports):
        timeout = self.timeout_var.get() / 1000
        max_threads = self.threads_var.get()
        semaphore = threading.Semaphore(max_threads)
        lock = threading.Lock()
        scanned = [0]
        open_count = [0]

        def check_port(port):
            if not self._scanning:
                return
            with semaphore:
                if not self._scanning:
                    return
                try:
                    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                    sock.settimeout(timeout)
                    result = sock.connect_ex((host, port))
                    banner = ""
                    if result == 0:
                        try:
                            sock.send(b"HEAD / HTTP/1.0\r\n\r\n")
                            banner = sock.recv(128).decode("utf-8",
                                                            errors="replace").strip()
                            banner = banner.splitlines()[0][:60] if banner else ""
                        except Exception:
                            pass
                    sock.close()
                    with lock:
                        scanned[0] += 1
                        if result == 0:
                            open_count[0] += 1
                            service = COMMON_PORTS.get(port, "")
                            self._results.append((port, "OPEN", service, banner))
                            self.root.after(0, self._add_row, port, "OPEN",
                                             service, banner)
                        self.root.after(0, self.progress.configure,
                                         {"value": scanned[0]})
                except Exception:
                    with lock:
                        scanned[0] += 1

        threads = []
        for port in ports:
            if not self._scanning:
                break
            t = threading.Thread(target=check_port, args=(port,), daemon=True)
            threads.append(t)
            t.start()

        for t in threads:
            t.join()

        self.root.after(0, self._scan_done, host, open_count[0], len(ports))

    def _add_row(self, port, state, service, banner):
        self.tree.insert("", tk.END,
                          values=(port, state, service, banner),
                          tags=("open" if state == "OPEN" else "closed",))

    def _scan_done(self, host, open_count, total):
        self._scanning = False
        self.status_var.set(
            f"スキャン完了: {host}  オープン {open_count} / {total} ポート")

    def _stop(self):
        self._scanning = False
        self.status_var.set("スキャンを停止しました")

    def _copy_results(self):
        lines = [f"{p}\t{s}\t{srv}\t{b}" for p, s, srv, b in self._results]
        self.root.clipboard_clear()
        self.root.clipboard_append("\n".join(lines))
        self.status_var.set("結果をコピーしました")


if __name__ == "__main__":
    root = tk.Tk()
    app = App066(root)
    root.mainloop()
