File Coverage

blib/lib/Catmandu/Fix/wd_language.pm

Criterion	Covered	Total	%
statement	25	32	78.1
branch	8	18	44.4
condition	1	3	33.3
subroutine	5	5	100.0
pod	0	1	0.0
total	39	59	66.1

package Catmandu::Fix::wd_language;
#ABSTRACT: Limit string values in a Wikidata entity record to a selected language
our $VERSION = '0.06'; #VERSION
use Catmandu::Sane;
use Moo;

with 'Catmandu::Fix::Base';

has language => (is => 'ro', required => 1);
has force => (is => 'ro');

around BUILDARGS => sub {
    my ($orig, $class, $language) = @_;
    $orig->($class, { language => $language });
};

sub emit {
    my ($self, $fixer) = @_;

    my $language = $self->language;
    my $var  = $fixer->var;
    my $code = $fixer->capture( sub { _fix_code($_[0],$language) } );

    return "${code}->(${var})";
}

sub _fix_code {
    my ($data, $language) = @_;

    foreach my $what (qw(labels descriptions)) {
        next unless exists $data->{$what};
        my $field = $data->{$what};
        if (ref $field) { # keep simple strings as given
            my $string = $field->{$language};
            if (defined $string) {
                $data->{$what} = ref $string ? $string->{value} : $string;
            } else {
                delete $data->{$what};
            }
        }
    }

    if (exists $data->{labels}) {
        $data->{label} = delete $data->{labels};
    }

    if (exists $data->{descriptions}) {
        $data->{description} = delete $data->{descriptions};
    }

    if (ref $data->{aliases} and ref $data->{aliases} eq 'HASH') {
        my $aliases = $data->{aliases}->{$language};
        if (defined $aliases) {
            $data->{aliases} = [
                map { ref $_ ? $_->{value} : $_ } @$aliases
            ];
        } else {
            $data->{aliases} = [ ];
        }
    }

    # TODO: only delete of string of requested language was found (or force)

    $data;
}

1;

__END__