DP-23: Visitor パターン

import java.util.ArrayList;
import java.util.List;

public class VisitorPatternSample {

    // Visitor インターフェース: 各要素を訪問するメソッドを定義する
    interface FileSystemVisitor {
        void visitFile(FileNode file);
        void visitDirectory(DirectoryNode directory);
    }

    // Element インターフェース: Visitor を受け入れる accept メソッドを定義する
    interface FileSystemNode {
        String getName();
        void accept(FileSystemVisitor visitor);
    }

    // 具体要素: ファイルを表すクラス
    static class FileNode implements FileSystemNode {
        private final String name;
        private final long sizeBytes;

        FileNode(String name, long sizeBytes) {
            this.name = name;
            this.sizeBytes = sizeBytes;
        }

        @Override
        public String getName() { return name; }

        long getSizeBytes() { return sizeBytes; }

        @Override
        public void accept(FileSystemVisitor visitor) {
            // ファイルへの訪問を Visitor に委譲する
            visitor.visitFile(this);
        }
    }

    // 具体要素: ディレクトリを表すクラス
    static class DirectoryNode implements FileSystemNode {
        private final String name;
        private final List<FileSystemNode> children = new ArrayList<>();

        DirectoryNode(String name) { this.name = name; }

        @Override
        public String getName() { return name; }

        void add(FileSystemNode node) { children.add(node); }

        List<FileSystemNode> getChildren() { return children; }

        @Override
        public void accept(FileSystemVisitor visitor) {
            // ディレクトリ自身への訪問
            visitor.visitDirectory(this);
            // 子要素も再帰的に訪問する
            for (FileSystemNode child : children) {
                child.accept(visitor);
            }
        }
    }

    // 具体 Visitor 1: ファイル一覧を表示する
    static class ListVisitor implements FileSystemVisitor {
        private int depth = 0;

        // インデントを手動で作成する（Java 8 では "  ".repeat() が使えないため）
        private String indent() {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < depth; i++) {
                sb.append("  ");
            }
            return sb.toString();
        }

        @Override
        public void visitFile(FileNode file) {
            System.out.println(indent() + "[FILE] " + file.getName()
                    + " (" + file.getSizeBytes() + " bytes)");
        }

        @Override
        public void visitDirectory(DirectoryNode directory) {
            System.out.println(indent() + "[DIR]  " + directory.getName() + "/");
            depth++;
        }
    }

    // 具体 Visitor 2: ファイルの合計サイズを計算する
    static class SizeCalculatorVisitor implements FileSystemVisitor {
        private long totalSize = 0;

        @Override
        public void visitFile(FileNode file) {
            totalSize += file.getSizeBytes();
        }

        @Override
        public void visitDirectory(DirectoryNode directory) {
            // ディレクトリ自体はサイズを持たない
        }

        long getTotalSize() { return totalSize; }
    }

    // 具体 Visitor 3: 指定した拡張子のファイルを数える
    static class FileCountVisitor implements FileSystemVisitor {
        private final String extension;
        private int count = 0;

        FileCountVisitor(String extension) {
            this.extension = extension.toLowerCase();
        }

        @Override
        public void visitFile(FileNode file) {
            if (file.getName().toLowerCase().endsWith(extension)) {
                count++;
            }
        }

        @Override
        public void visitDirectory(DirectoryNode directory) {
            // ディレクトリ自体はカウントしない
        }

        int getCount() { return count; }
    }

    public static void main(String[] args) {
        // ファイルシステム構造を構築する
        DirectoryNode root = new DirectoryNode("project");
        DirectoryNode src = new DirectoryNode("src");
        DirectoryNode docs = new DirectoryNode("docs");

        src.add(new FileNode("Main.java", 2048));
        src.add(new FileNode("Utils.java", 1024));
        docs.add(new FileNode("README.md", 256));
        docs.add(new FileNode("design.pdf", 10240));

        root.add(src);
        root.add(docs);
        root.add(new FileNode("pom.xml", 384));

        // Visitor 1: ファイル一覧
        System.out.println("=== ファイル一覧 ===");
        root.accept(new ListVisitor());

        // Visitor 2: 合計サイズ
        System.out.println("\n=== 合計サイズ ===");
        SizeCalculatorVisitor sizeCalc = new SizeCalculatorVisitor();
        root.accept(sizeCalc);
        System.out.println("合計: " + sizeCalc.getTotalSize() + " bytes");

        // Visitor 3: .java ファイル数
        System.out.println("\n=== .java ファイル数 ===");
        FileCountVisitor javaCount = new FileCountVisitor(".java");
        root.accept(javaCount);
        System.out.println(".java ファイル数: " + javaCount.getCount());
    }
}