r257@brr: dpavlin | 2007-11-24 03:16:39 +0100

[perl-cwmp.git] / lib / CWMP / Store / YAML.pm

diff --git a/lib/CWMP/Store/YAML.pm b/lib/CWMP/Store/YAML.pm
index 160cbe8..6b5ea15 100644 (file)
--- a/lib/CWMP/Store/YAML.pm
+++ b/lib/CWMP/Store/YAML.pm
@@ -5,8 +5,9 @@ use strict;
 use warnings;
 
 use Data::Dump qw/dump/;
-use YAML qw/LoadFile DumpFile/;
+use YAML::Syck;
 use Hash::Merge qw/merge/;
+use Carp qw/confess/;
 
 =head1 NAME
 
@@ -16,22 +17,42 @@ CWMP::Store::YAML - use YAML as storage
 
 =head2 open
 
+  $store->open({
+       path => 'var/',
+       debug => 1,
+       clean => 1,
+  });
+
 =cut
 
-my $dir = 'yaml';
+my $path;
 
-my $debug = 1;
+my $debug = 0;
 
 sub open {
        my $self = shift;
 
-       warn "open ",dump( @_ );
+       my $args = shift;
+
+       $debug = $args->{debug};
+       $path = $args->{path} || confess "no path?";
+
+       warn "open ",dump( $args ) if $debug;
 
-       if ( ! -e $dir ) {
-               mkdir $dir || die "can't create $dir: $!";
-               warn "created $dir directory\n";
+       $path = "$path/yaml";
+
+       if ( ! -e $path ) {
+               mkdir $path || die "can't create $path: $!";
+               warn "created $path directory\n" if $debug;
+       } elsif ( $args->{clean} ) {
+               warn "removed old $path\n" if $debug;
+               foreach my $uid ( $self->all_uids ) {
+                       my $file = "$path/$uid.yml";
+                       unlink $file || die "can't remove $file: $!";
+               }
        }
 
+
 }
 
 =head2 update_uid_state
@@ -43,7 +64,7 @@ sub open {
 sub update_uid_state {
        my ( $self, $uid, $state ) = @_;
 
-       my $file = "$dir/$uid.yml";
+       my $file = "$path/$uid.yml";
 
        my $old_state = $self->get_state( $uid );
 
@@ -64,7 +85,7 @@ sub update_uid_state {
 sub get_state {
        my ( $self, $uid ) = @_;
 
-       my $file = "$dir/$uid.yml";
+       my $file = "$path/$uid.yml";
 
        if ( -e $file ) {
                return LoadFile( $file );
@@ -82,8 +103,8 @@ sub get_state {
 sub all_uids {
        my $self = shift;
 
-       opendir(my $d, $dir) || die "can't opendir $dir: $!";
-       my @uids = grep { /\.yml$/ && -f "$dir/$_" } readdir($d);
+       opendir(my $d, $path) || die "can't opendir $path: $!";
+       my @uids = grep { /\.yml$/ && -f "$path/$_" } readdir($d);
        closedir $d;
 
        return map { my $l = $_; $l =~ s/\.yml$//; $l } @uids;