#!/usr/bin/perl

use strict;
use warnings;

use Test::More tests => 4;

BEGIN
{
    use File::Spec;
    use lib File::Spec->catdir(File::Spec->curdir(), "t", "lib");
}

use File::Find::Object::TreeCreate;
use File::Find::Object;

use File::Path;

{
    my $tree =
    {
        'name' => "prune--traverse-2/",
        'subs' =>
        [
            {
                'name' => "b.doc",
                'contents' => "This file was spotted in the wild.",
            },
            {
                'name' => "a/",
            },
            {
                'name' => "foo/",
                'subs' =>
                [
                    {
                        'name' => "please-prune-me/",
                        'subs' =>
                        [
                            {
                                'name' => "a-non-reachable-dir/",
                                'subs' =>
                                [
                                    {
                                        'name' => "dir1/",
                                    },
                                    {
                                        'name' => "dir2/",
                                    },
                                    {
                                        'name' =>
                                            "if-we-get-this-its-wrong.txt",
                                        'content' => "Hi ho!",
                                    },
                                ],
                            },
                            {
                                'name' => "h.rnd",
                                'contents' => "This file is empty.",
                            },
                            {
                                'name' => "lambda.calculus",
                                'contents' => '\f \x (f (f x))'
                            },
                        ],
                    },
                ],
            },
        ],
    };

    my $t = File::Find::Object::TreeCreate->new();
    $t->create_tree("./t/sample-data/", $tree);
    my $ff =
        File::Find::Object->new(
            {},
            $t->get_path("./t/sample-data/prune--traverse-2")
        );
    my @results;
    for my $i (1 .. 7)
    {
        my $file = $ff->next();
        # We're doing that because get_current_node_files_list() used to
        # call ->_recurse() which caused some subtle bugs.
        my $files_in_node = $ff->get_current_node_files_list();

        if ($file eq
            $t->get_path("t/sample-data/prune--traverse-2/foo/please-prune-me")
           )
        {
            $ff->set_traverse_to(
                [
                    grep { $_ !~ /non-reachable/ }
                    @{$ff->get_current_node_files_list()}
                ]
            );

            # TEST
            is_deeply ($ff->get_traverse_to(), ["h.rnd", "lambda.calculus"],
                "Testing ->get_traverse_to()"
            );
        }
        push @results, $file;
    }
    # TEST
    is_deeply(
        \@results,
        [(map { $t->get_path("t/sample-data/prune--traverse-2/$_") }
            ("",
            qw(
                a
                b.doc
                foo
                foo/please-prune-me
                foo/please-prune-me/h.rnd
                foo/please-prune-me/lambda.calculus
            )))
        ],
        "Checking for regular, lexicographically sorted order",
    );

    rmtree($t->get_path("./t/sample-data/prune--traverse-2"))
}

{
    my $tree =
    {
        'name' => "prune--traverse-2/",
        'subs' =>
        [
            {
                'name' => "b.doc",
                'contents' => "This file was spotted in the wild.",
            },
            {
                'name' => "a/",
            },
            {
                'name' => "foo/",
                'subs' =>
                [
                    {
                        'name' => "please-prune-me/",
                        'subs' =>
                        [
                            {
                                'name' => "a-non-reachable-dir/",
                                'subs' =>
                                [
                                    {
                                        'name' => "dir1/",
                                    },
                                    {
                                        'name' => "dir2/",
                                    },
                                    {
                                        'name' =>
                                            "if-we-get-this-its-wrong.txt",
                                        'content' => "Hi ho!",
                                    },
                                ],
                            },
                            {
                                'name' => "h.rnd",
                                'contents' => "This file is empty.",
                            },
                            {
                                'name' => "lambda.calculus",
                                'contents' => '\f \x (f (f x))'
                            },
                        ],
                    },
                    {
                        'name' => "zardoz/",
                        'subs' =>
                        [
                            {
                                'name' => "p.txt",
                                'contents' => "Intentionally Left Boring.",
                            },
                        ],
                    },
                ],
            },
            {
                'name' => "zardoz/",
                'subs' =>
                [
                    {
                        'name' => "p.txt",
                        'contents' => "Intentionally Left Boring.",
                    },
                ],
            },

        ],
    };

    my $t = File::Find::Object::TreeCreate->new();
    $t->create_tree("./t/sample-data/", $tree);
    my $ff =
        File::Find::Object->new(
            {},
            $t->get_path("./t/sample-data/prune--traverse-2")
        );
    my @results;
    for my $i (1 .. 9)
    {
        my $file = $ff->next();
        if ($file eq
            $t->get_path("t/sample-data/prune--traverse-2/foo/please-prune-me")
           )
        {
            $ff->prune();
        }
        push @results, $file;
    }

    # TEST
    ok (!defined($ff->next()), "Testing that the scan has completed.");

    # TEST
    is_deeply(
        \@results,
        [(map { $t->get_path("t/sample-data/prune--traverse-2/$_") }
            ("",
            qw(
                a
                b.doc
                foo
                foo/please-prune-me
                foo/zardoz
                foo/zardoz/p.txt
                zardoz
                zardoz/p.txt
            )))
        ],
        "Checking for regular, lexicographically sorted order",
    );

    rmtree($t->get_path("./t/sample-data/prune--traverse-2"))
}