File Coverage

blib/lib/CGI/Application/Plugin/Config/YAML.pm

Criterion	Covered	Total	%
statement	10	12	83.3
branch			n/a
condition			n/a
subroutine	4	4	100.0
pod			n/a
total	14	16	87.5

package CGI::Application::Plugin::Config::YAML;

use strict;
use warnings;
use vars qw($VERSION @EXPORT);
use CGI::Application;
use Config::YAML;

$VERSION = '0.01';

@EXPORT = qw(
    config_file
    config_param
    config
    config_fold
    config_read
    get_hash
);

sub import {
    my $pkg  = shift;
    my $call = caller;
    no strict 'refs';
    foreach my $sym (@EXPORT) {
        *{"${call}::$sym"} = \&{$sym};
    }
}

sub config_file {
    my $self = shift;
    my $file_name = shift;

    if ( defined $file_name ) {
        $self->{__CONFIG_YAML}->{__FILE_NAME} = $file_name;
        $self->{__CONFIG_YAML}->{__FILE_CHANGED} = 1;
    }
    else {
        $file_name = $self->{__CONFIG_YAML}->{__FILE_NAME};
    }

    if ( ! defined $file_name ) {
        $ENV{CGIAPP_CONFIG_FILE} =~ /(.*)/;
        $file_name = $1;
    }

    $file_name;
}

sub config_param {
    my $self = shift;
    my @params = @_;
    my $conf = $self->config();

    if ( scalar(@params) == 0 ) {
        return scalar($self->get_hash);
    }
    elsif ( scalar(@params) == 1 ) {
        return $conf->get($params[0]);
    }
    else {
        my %params = (@params);
        $conf->set($_ => $params{$_}) foreach (keys %params);
        if ( $conf->write ) {
            return;
        }
        else{
            die "Config-Plugin: Could not write to config file (" . $self->config_file . ")! ";
        }
    }
}

sub config_fold {
    my ($self, $data) = @_;
    my $conf = $self->config();
    $conf->fold($data);
    return;
}

sub config_read {
    my ($self, $file) = @_;
    my $conf = $self->config();
    $conf->read($file);
    return;
}

sub config {
    my $self = shift;
    my $create = !$self->{__CONFIG_YAML}->{__CONFIG_OBJ} || $self->{__CONFIG_YAML}->{__FILE_CHANGED};
    if ( $create ) {
        my $file_name = $self->config_file or die "No config file specified!";

        my $conf;
        eval{
            $conf = Config::YAML->new(config => $file_name);
        };

        die "Could not create Config::YAML object for file $file_name! $@" if $@;

        $self->{__CONFIG_YAML}->{__CONFIG_OBJ} = $conf;
        $self->{__CONFIG_YAML}->{__FILE_CHANGED} = 0;
    }

    return $self->{__CONFIG_YAML}->{__CONFIG_OBJ};
}

sub get_hash {
    my $self = shift;
    my $yaml;

    open(FH,'<',$self->config_file) or die "Can't open $self->config_file; $!\n";
    while (my $line = ) {
        next if ($line =~ /^\-{3,}/);
        next if ($line =~ /^#/);
        next if ($line =~ /^$/);
        $yaml .= $line;
    }
    close(FH);

    my $tmpyaml = YAML::Load($yaml);
    return $tmpyaml;
}

1;

__END__