app.py

# app.py

import streamlit as st
import tempfile
import shutil
from pathlib import Path
import git  # GitPython
from core.file_scanner import FileScanner, FileInfo

# =====================================
# セッション状態の初期化
# =====================================
if 'scanned_files' not in st.session_state:
    st.session_state.scanned_files = []  # スキャンしたFileInfoリスト
if 'selected_files' not in st.session_state:
    st.session_state.selected_files = set()  # ユーザーが選択中のファイルパス (相対パス)
if 'cloned_repo_dir' not in st.session_state:
    st.session_state.cloned_repo_dir = None  # クローン先ディレクトリの絶対パス文字列

# =====================================
# タイトル等
# =====================================
st.title("Gitリポジトリ スキャナー")
st.markdown("**ディレクトリ構造をツリー表示し、ファイルを選んでMarkdownダウンロードできます**\n(**ワイドモード推奨**)")

# =====================================
# ツリー構造を生成する関数
# =====================================
def build_tree(paths):
    """
    相対パス(Pathオブジェクト)のリストからツリー状のネスト構造を構築する。
    戻り値は {要素名 -> 子要素のdict or None} という入れ子の辞書。
    """
    tree = {}
    for p in paths:
        parts = p.parts
        current = tree
        for i, part in enumerate(parts):
            if i == len(parts) - 1:
                # ファイルやフォルダの末端
                current[part] = None
            else:
                if part not in current:
                    current[part] = {}
                if isinstance(current[part], dict):
                    current = current[part]
                else:
                    # もしNoneだった場合(同名のファイル/フォルダがあるなど) → 無理やりdictに
                    current[part] = {}
                    current = current[part]
    return tree

def format_tree(tree_dict, prefix=""):
    """
    build_tree()で作ったネスト構造をASCIIアートのツリー文字列にする。
    """
    lines = []
    entries = sorted(tree_dict.keys())
    for i, entry in enumerate(entries):
        is_last = (i == len(entries) - 1)
        marker = "└── " if is_last else "├── "
        # 子要素がある(=dict)ならフォルダ、Noneならファイル
        if isinstance(tree_dict[entry], dict):
            # フォルダとして表示
            lines.append(prefix + marker + entry + "/")
            # 次の階層のプレフィックスを用意
            extension = "    " if is_last else "│   "
            sub_prefix = prefix + extension
            # 再帰的に生成
            lines.extend(format_tree(tree_dict[entry], sub_prefix))
        else:
            # ファイルとして表示
            lines.append(prefix + marker + entry)
    return lines


# =====================================
# ユーザー入力
# =====================================
repo_url = st.text_input("GitリポジトリURL (例: https://github.com/username/repo.git)")

st.subheader("スキャン対象拡張子")
available_exts = [".py", ".js", ".ts", ".sh", ".md", ".txt", ".java", ".cpp"]
chosen_exts = []
for ext in available_exts:
    default_checked = (ext in [".py", ".md"])  # デモ用に .py と .md を初期ON
    if st.checkbox(ext, key=f"ext_{ext}", value=default_checked):
        chosen_exts.append(ext)

# =====================================
# スキャン開始ボタン
# =====================================
if st.button("スキャン開始"):
    if not repo_url.strip():
        st.error("リポジトリURLを入力してください。")
    else:
        # 既にクローン済フォルダがあれば削除
        if st.session_state.cloned_repo_dir and Path(st.session_state.cloned_repo_dir).exists():
            shutil.rmtree(st.session_state.cloned_repo_dir, ignore_errors=True)
        
        # 一時フォルダを作成してクローン
        tmp_dir = tempfile.mkdtemp()
        clone_path = Path(tmp_dir) / "cloned_repo"

        try:
            st.write(f"リポジトリをクローン中: {clone_path}")
            git.Repo.clone_from(repo_url, clone_path)
            st.session_state.cloned_repo_dir = str(clone_path)
        except Exception as e:
            st.error(f"クローン失敗: {e}")
            st.session_state.cloned_repo_dir = None
            st.session_state.scanned_files = []
            st.stop()

        # スキャン
        scanner = FileScanner(base_dir=clone_path, target_extensions=set(chosen_exts))
        found_files = scanner.scan_files()

        st.session_state.scanned_files = found_files
        st.session_state.selected_files = set()

        st.success(f"スキャン完了: {len(found_files)}個のファイルを検出")

# =====================================
# クローン削除ボタン
# =====================================
if st.session_state.cloned_repo_dir:
    if st.button("クローン済みデータを削除"):
        shutil.rmtree(st.session_state.cloned_repo_dir, ignore_errors=True)
        st.session_state.cloned_repo_dir = None
        st.session_state.scanned_files = []
        st.session_state.selected_files = set()
        st.success("クローンしたディレクトリを削除しました")

# =====================================
# スキャン結果がある場合 → ツリー表示 + ファイル選択
# =====================================
if st.session_state.scanned_files:
    base_path = Path(st.session_state.cloned_repo_dir)

    # --- ツリーを作る ---
    # scanned_files は「指定拡張子」だけ取得されているので、そのファイルパスのみでツリーを構築
    rel_paths = [f.path.relative_to(base_path) for f in st.session_state.scanned_files]
    tree_dict = build_tree(rel_paths)
    tree_lines = format_tree(tree_dict)
    ascii_tree = "\n".join(tree_lines)

    st.write("## スキャン結果")
    col_tree, col_files = st.columns([1, 2])  # 左:ツリー, 右:ファイル一覧

    with col_tree:
        st.markdown("**ディレクトリ構造 (指定拡張子のみ)**")
        st.markdown(f"```\n{ascii_tree}\n```")

    with col_files:
        st.markdown("**ファイル一覧 (チェックボックス)**")

        col_btn1, col_btn2 = st.columns(2)
        with col_btn1:
            if st.button("すべて選択"):
                st.session_state.selected_files = set(rel_paths)
        with col_btn2:
            if st.button("すべて解除"):
                st.session_state.selected_files = set()

        for file_info in st.session_state.scanned_files:
            rel_path = file_info.path.relative_to(base_path)
            checked = rel_path in st.session_state.selected_files

            new_checked = st.checkbox(
                f"{rel_path} ({file_info.formatted_size})",
                value=checked,
                key=str(rel_path)  # keyの重複回避
            )
            if new_checked:
                st.session_state.selected_files.add(rel_path)
            else:
                st.session_state.selected_files.discard(rel_path)


# =====================================
# 選択ファイルをまとめてMarkdown化 & ダウンロード
# =====================================
def create_markdown_for_selected(files, selected_paths, base_dir: Path) -> str:
    output = []
    for f in files:
        rel_path = f.path.relative_to(base_dir)
        if rel_path in selected_paths:
            output.append(f"## {rel_path}")
            output.append("------------")
            if f.content is not None:
                output.append(f.content)
            else:
                output.append("# Failed to read content")
            output.append("")  # 空行
    return "\n".join(output)

if st.session_state.scanned_files:
    st.write("## 選択ファイルをダウンロード")
    if st.button("選択ファイルをMarkdownとしてダウンロード(整形後，下にダウンロードボタンが出ます)"):
        base_path = Path(st.session_state.cloned_repo_dir)
        markdown_text = create_markdown_for_selected(
            st.session_state.scanned_files,
            st.session_state.selected_files,
            base_path
        )
        st.download_button(
            label="Markdownダウンロード",
            data=markdown_text,
            file_name="selected_files.md",
            mime="text/markdown"
        )