unit module IO::Capture::Simple;

my $stdout = $*OUT;
my $stderr = $*ERR;

sub capture(Callable $code) is export {
    my ($out, $err);

    capture_on($out, $err );
    $code.();
    capture_off;

    $out, $err ;
}

sub capture_on($out is rw, $err is rw ) is export {
    capture_stdout_on($out);
    capture_stderr_on($err);
}

sub capture_off is export {
    $*OUT = $stdout;
    $*ERR = $stderr;
}

sub capture_stdout(Callable $code) is export {
    my $result;

    my $*OUT = class {
        method print(*@args) {
            $result ~= @args.join;
        }
        method flush {}
    }

    $code.();

    $result;
}

sub capture_stdout_on($target is rw) is export {
    $*OUT = class {
        method print(*@args) {
            $target ~= @args.join;
        }
        method flush {}
    }
}

sub capture_stdout_off is export {
    $*OUT = $stdout;
}

sub capture_stderr(Callable $code) is export {
    my $result;

    my $*ERR = class {
        method print(*@args) {
            $result ~= @args.join;
        }
    }

    $code.();

    $result;
}

sub capture_stderr_on($target is rw) is export {
    $*ERR = class {
        method print(*@args) {
            $target ~= @args.join;
        }
    }
}

sub capture_stderr_off is export {
    $*ERR = $stderr;
}