File Coverage

blib/lib/Mojolicious/Plugin/NoIndex.pm

Criterion	Covered	Total	%
statement	62	62	100.0
branch	27	32	84.3
condition	10	18	55.5
subroutine	4	4	100.0
pod	1	1	100.0
total	104	117	88.8

package Mojolicious::Plugin::NoIndex;

# ABSTRACT: add meta tag to HTML output to define a policy for robots

use Mojo::Base 'Mojolicious::Plugin';

use Carp qw(croak);

our $VERSION = '0.02';

my %routes;
my %routes_special;


sub register {
    my ($self, $app, $config) = @_;

    if ( !$config || !%{$config} ) {
        $config = { all_routes => 1 };
    }

    my $default_value = $config->{default} // 'noindex';

    %routes         = ();
    %routes_special = ();

    ROUTENAME:
    for my $route_name ( keys %{ $config->{routes} || {} } ) {
        my $value = $config->{routes}->{$route_name};
        $value    = $default_value if $value eq '1';

        if ( $route_name =~ m{\A(.*?)\#\#\#(.*)}xms ) {
            my %conditions = map{ my ($param, $regex) = split /=/, $_; $param => qr/\A(?:$regex)\z/  } split /\&/, $2 // '';
            push @{ $routes_special{$1}->{conditions} }, \%conditions;
            $routes_special{$1}->{value} = $value;

            next ROUTENAME;
        }

        $routes{$route_name} = $value;
    }

    for my $value ( keys %{ $config->{by_value} || {} } ) {

        ROUTE:
        for my $route_name ( @{ $config->{by_value}->{$value} || [] } ) {
            if ( $route_name =~ m{\A(.*?)\#\#\#(.*)}xms ) {
                my %conditions = map{ my ($param, $regex) = split /=/, $_; $param => qr/\A(?:$regex)\z/  } split /\&/, $2 // '';
                push @{ $routes_special{$1}->{conditions} }, \%conditions;
                $routes_special{$1}->{value} = $value;

                next ROUTE;
            }

            $routes{$route_name} = $value;
        }
    }

    $app->hook(
        after_render => sub {
            my ($c, $content, $format) = @_;

            return if !$format;
            return if $format ne 'html';

            my $route = $c->current_route;
            my $value;

            if ( $routes{$route} ) {
                $value = $routes{$route} // $default_value;
            }

            if ( $routes_special{$route} ) {
                my $matched;

                CONDITION:
                for my $conditions ( @{ $routes_special{$route}->{conditions} || [] } ) {
                    for my $param ( keys %{ $conditions || {} } ) {
                        if ( $c->param( $param ) !~ $conditions->{$param} ) {
                            next CONDITION;
                        }
                    }

                    $value = $routes_special{$route}->{value} // $default_value;
                    last CONDITION;
                }
            }

            if ( $config->{all_routes} && !$value ) {
                $value = $default_value;
            }

            return if !$value;

            $$content =~ s{]+ name="robots" [^>]+ >}{}x;
            $$content =~ s{\K}{};
        }
    );
}

1;

__END__