CWMP/Store/YAML.pm

# Dobrica Pavlinusic, <dpavlin@rot13.org> 10/26/07 21:37:12 CEST
package CWMP::Store::YAML;

use strict;
use warnings;

use Data::Dump qw/dump/;
use YAML qw/LoadFile DumpFile/;
use Hash::Merge qw/merge/;
use Carp qw/confess/;

=head1 NAME

CWMP::Store::YAML - use YAML as storage

=head1 METHODS

=head2 open

  $store->open({
        path => 'var/',
        debug => 1,
        clean => 1,
  });

=cut

my $path;

my $debug = 0;

sub open {
        my $self = shift;

        my $args = shift;

        $debug = $args->{debug};
        $path = $args->{path} || confess "no path?";

        warn "open ",dump( $args ) if $debug;

        $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

  $store->update_uid_state( $uid, $state );

=cut

sub update_uid_state {
        my ( $self, $uid, $state ) = @_;

        my $file = "$path/$uid.yml";

        my $old_state = $self->get_state( $uid );

        my $combined = merge( $state, $old_state );

#       warn "## ",dump( $old_state, $state, $combined );

        DumpFile( $file, $combined ) || die "can't write $file: $!";

}

=head2 get_state

  $store->get_state( $uid );

=cut

sub get_state {
        my ( $self, $uid ) = @_;

        my $file = "$path/$uid.yml";

        if ( -e $file ) {
                return LoadFile( $file );
        }

        return;
}

=head2 all_uids

  my @uids = $store->all_uids;

=cut

sub all_uids {
        my $self = shift;

        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;
}

1;
1	dpavlin	136	# Dobrica Pavlinusic, <dpavlin@rot13.org> 10/26/07 21:37:12 CEST
2			package CWMP::Store::YAML;
3
4			use strict;
5			use warnings;
6
7			use Data::Dump qw/dump/;
8			use YAML qw/LoadFile DumpFile/;
9	dpavlin	139	use Hash::Merge qw/merge/;
10	dpavlin	140	use Carp qw/confess/;
11	dpavlin	136
12			=head1 NAME
13
14			CWMP::Store::YAML - use YAML as storage
15
16			=head1 METHODS
17
18			=head2 open
19
20	dpavlin	147	$store->open({
21			path => 'var/',
22			debug => 1,
23			clean => 1,
24			});
25
26	dpavlin	136	=cut
27
28	dpavlin	140	my $path;
29	dpavlin	136
30	dpavlin	147	my $debug = 0;
31	dpavlin	136
32			sub open {
33			my $self = shift;
34
35	dpavlin	140	my $args = shift;
36	dpavlin	136
37	dpavlin	140	$debug = $args->{debug};
38			$path = $args->{path} \|\| confess "no path?";
39
40			warn "open ",dump( $args ) if $debug;
41
42			$path = "$path/yaml";
43
44			if ( ! -e $path ) {
45			mkdir $path \|\| die "can't create $path: $!";
46	dpavlin	147	warn "created $path directory\n" if $debug;
47			} elsif ( $args->{clean} ) {
48			warn "removed old $path\n" if $debug;
49			foreach my $uid ( $self->all_uids ) {
50			my $file = "$path/$uid.yml";
51			unlink $file \|\| die "can't remove $file: $!";
52			}
53	dpavlin	136	}
54
55	dpavlin	147
56	dpavlin	136	}
57
58			=head2 update_uid_state
59
60			$store->update_uid_state( $uid, $state );
61
62			=cut
63
64			sub update_uid_state {
65			my ( $self, $uid, $state ) = @_;
66
67	dpavlin	140	my $file = "$path/$uid.yml";
68	dpavlin	136
69	dpavlin	139	my $old_state = $self->get_state( $uid );
70	dpavlin	136
71	dpavlin	139	my $combined = merge( $state, $old_state );
72
73			# warn "## ",dump( $old_state, $state, $combined );
74
75			DumpFile( $file, $combined ) \|\| die "can't write $file: $!";
76
77	dpavlin	136	}
78
79			=head2 get_state
80
81			$store->get_state( $uid );
82
83			=cut
84
85			sub get_state {
86			my ( $self, $uid ) = @_;
87
88	dpavlin	140	my $file = "$path/$uid.yml";
89	dpavlin	136
90			if ( -e $file ) {
91			return LoadFile( $file );
92			}
93
94			return;
95			}
96
97			=head2 all_uids
98
99			my @uids = $store->all_uids;
100
101			=cut
102
103			sub all_uids {
104			my $self = shift;
105
106	dpavlin	140	opendir(my $d, $path) \|\| die "can't opendir $path: $!";
107			my @uids = grep { /\.yml$/ && -f "$path/$_" } readdir($d);
108	dpavlin	136	closedir $d;
109
110	dpavlin	137	return map { my $l = $_; $l =~ s/\.yml$//; $l } @uids;
111	dpavlin	136	}
112
113			1;