/[pxelator]/couchdb/design-couch.pl
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Annotation of /couchdb/design-couch.pl

Parent Directory Parent Directory | Revision Log Revision Log


Revision 248 - (hide annotations)
Tue Aug 18 12:18:39 2009 UTC (14 years, 7 months ago) by dpavlin
File MIME type: text/plain
File size: 4081 byte(s)
CouchDB view push/pull script
1 dpavlin 248 #!/usr/bin/perl
2    
3     use warnings;
4     use strict;
5    
6     use LWP::UserAgent;
7     use JSON;
8     use Data::Dump qw/dump/;
9     use File::Path qw/mkpath/;
10     use File::Slurp qw//;
11     use File::Find;
12     use HTTP::Request::Common;
13     use MIME::Base64;
14    
15     use lib qw(lib ../lib);
16     use Media::Type::Simple;
17    
18    
19     # design-couch.pl
20     #
21     # 04/26/09 21:12:28 CEST Dobrica Pavlinusic <dpavlin@rot13.org>
22    
23     my ( $command, $database, $design ) = @ARGV;
24     die "usage: $0 [push|pull] database design\n" unless $database && $design;
25    
26     chdir $design || die "can't find $design: $!";
27    
28     my $ua = LWP::UserAgent->new;
29    
30     my $url = "http://localhost:5984/$database/_design/$design";
31    
32     sub create_path {
33     my $path = shift;
34     if ( $path =~ m{/} ) {
35     my $dir = $path;
36     $dir =~ s{/[^/]+$}{};
37     mkpath $dir if ! -e $dir;
38     #warn "# dir $dir";
39     }
40     }
41     sub write_file {
42     my ( $path, $content ) = @_;
43     $path =~ s{^/+}{};
44     create_path $path;
45     File::Slurp::write_file $path, $content;
46     print "$path ", -s $path, " bytes created\n";
47     }
48    
49     sub write_attachment {
50     my ( $path ) = @_;
51     my $file = "_attachemnts/$path";
52     create_path $file;
53     $ua->mirror( "$url/$path", $file );
54     print "detached $file ", -s $file, " bytes\n";
55     }
56    
57    
58     sub unroll {
59     my ( $tree, $path ) = @_;
60    
61     my $ref = ref $tree;
62     if ( $ref eq 'HASH' ) {
63     foreach my $child ( keys %$tree ) {
64     if ( $child eq '_attachments' ) {
65     write_attachment $_ foreach keys %{ $tree->{$child} };
66     } else {
67     unroll( $tree->{$child}, $path ? "$path/$child" : $child );
68     }
69     }
70     } elsif ( $ref ) {
71     warn "UNSUPPORTED $path $ref ", dump( $tree );
72     write_file "$path.json", to_json $tree;
73     } elsif ( $ref eq '' ) {
74    
75     if ( $tree =~ m[^\s*(function(.*){.*}|/\*|//|var)]is ) {
76     $path .= '.js';
77     } elsif ( $tree =~ m{<%=.*%>} ) { # couchapp template
78     $path .= '.html';
79     } else {
80     warn "# can't detect type of $path\n";
81     }
82    
83     write_file $path, $tree;
84     }
85    
86     }
87    
88     if ( $command eq 'pull' ) {
89    
90     warn "# get $url\n";
91     my $response = $ua->get( $url );
92     die $response->status_line if $response->is_error;
93    
94     my $json = $response->decoded_content;
95     write_file "../$database-$design.pull.js", $json;
96    
97     unroll( from_json $json, '' );
98    
99     } elsif ( $command eq 'push' ) {
100    
101     my $json;
102    
103     find({ no_chdir => 1, wanted => sub {
104     my $path = $File::Find::name;
105     return unless -f $path;
106    
107     warn "## $path\n";
108    
109     $path =~ s{^\./}{};
110    
111     if ( $path =~ m{_attachemnts/(.+)} ) {
112    
113     my $filename = $1;
114     my $content_type = 'text/plain';
115     $content_type = type_from_ext($1) if $filename =~ m{\.(\w+)$};
116    
117     my $data = File::Slurp::read_file( $path );
118     $data = encode_base64( $data );
119     # XXX inline attachments must be single line
120     # XXX http://wiki.apache.org/couchdb/HTTP_Document_API
121     $data =~ s/[\n\r]+//gs;
122     $json->{_attachments}->{ $filename } = {
123     content_type => $content_type,
124     data => $data,
125     };
126     return;
127     }
128    
129     my $data = File::Slurp::read_file( $path );
130     $path =~ s[/]['}->{']g;
131     $path =~ s{\.\w+$}{};
132     my $code = "\$json->{'$path'} = \$data;";
133     eval $code;
134     die "ERROR in $code: $@" if $@;
135     # warn "## json = ",dump( $json );
136     }}, '.' );
137    
138     if ( ! defined $json->{_id} ) {
139     warn "creating _id for document\n";
140     $json->{_id} = $$ . '-' . time();
141     }
142     delete( $json->{_rev} ) && warn "removing _rev from document\n";
143    
144     print "push $database/_design/$design\n";
145     write_file "../$database-$design.push.js", to_json $json;
146    
147     warn "# put $url\n";
148     my $response = $ua->request(
149     HTTP::Request::Common::PUT(
150     $url,
151     'Content-Type' => 'application/json',
152     Content => to_json $json,
153     )
154     );
155    
156     if ( $response->code == 409 ) {
157     warn "## update $url\n";
158     my $response = $ua->get( $url );
159     die $response->status_line if $response->is_error;
160    
161     my $data = from_json $response->decoded_content;
162     $json->{$_} = $data->{$_} foreach ( '_rev', '_id' );
163    
164     $response = $ua->request( HTTP::Request::Common::PUT($url, 'Content-Type' => 'application/json', Content => to_json $json ) );
165     die $response->status_line if $response->is_error;
166     warn "push updated $url\n";
167     } else {
168     die $response->status_line if $response->is_error;
169     warn "push new $url\n";
170     }
171    
172     } else {
173     die "$0: unknown command $command";
174     }
175    

Properties

Name Value
svn:executable *

  ViewVC Help
Powered by ViewVC 1.1.26