Changeset 1485 in ProjectBuilder for projects/casparbuster/devel/bin/cbusterize

Timestamp:

Apr 23, 2012, 11:04:02 AM (12 years ago)

Author:

Bruno Cornec

Message:

• Adds a Plugin module to support plugin management
• cbusterize now works with both -s and -p options
• One plugin example added (dhcpd)

-cb script started. Not working now

File:

• projects/casparbuster/devel/bin/cbusterize (modified) (15 diffs)

projects/casparbuster/devel/bin/cbusterize

@@ -13 +13 @@
    --help   |-h         brief help message
    --man            full documentation
-   --source |-s <file/dir>  directory or file to copy in the CasparBuster tree
+   --force   |-f            force copy of files, even if they exist
+   --source |-s <file/dir>  directory or files to copy in the CasparBuster tree (',' separated if many)
+   --plugin |-p <plugin name>   plugin defining what to copy in the CasparBuster tree (',' separated if many)
    --machine|-m <machine>   machine to consider in the subtree
 
@@ -41 +43 @@
 =item B<--source> I<path>
 
-Specify the path to the source file or directory to manage with CasparBuster.
+Specify the path to the source file or directory to manage with CasparBuster. Multiple paths can be specified separated by ','.
+
+=item B<--plugin> I<name>
+
+Specify the name of the plugin to manage with CasparBuster. Multiple plugins can be specified separated by ','.
+A plugin defines a set of files (with their mode and owner), a set of directories (with their mode and owner) and a set of scripts to launch once the files are copied remotely.
 
 =back
@@ -102 +109 @@
 use CasparBuster::Version;
 use CasparBuster::Env;
+use CasparBuster::Plugin;
 #use Cwd 'realpath';
 use File::Find;
@@ -124 +132 @@
 my $plugin = undef;
 my $quiet = undef;
+my $force = undef;
 my $log = undef;
 my $LOG = undef;
@@ -141 +150 @@
     'help|h'      => \$help,
     'quiet|q'     => \$quiet,
+    'force|f'     => \$force,
     'man'         => \$man,
-    'log-files|l=s' => \$log,
+    'logfile|l=s' => \$log,
     'source|s=s'  => \$source,
     'plugin|p=s'  => \$plugin,
@@ -176 +186 @@
 my %cb;
 my $cb = \%cb;
-($cb->{'basedir'},$cb->{'database'},$cb->{'usemachines'},$cb->{'pluginsdir'},$cb->{'cms'}) = pb_conf_get("cbbasedir","cbdatabase","cbusemachines","cbpluginssubdir","cbcms");
+($cb->{'basedir'},$cb->{'usemachines'},$cb->{'cms'}) = pb_conf_get("cbbasedir","cbusemachines","cbcms");
 pb_log(2,"%cb: ",Dumper($cb));
 
@@ -184 +194 @@
 
 if (defined $plugin) {
-    # Load plugin conf
+    # Load plugins
+    cb_plugin_load();
 }
 
@@ -190 +201 @@
 eval { $basedir =~ s/(\$ENV.+\})/$1/eeg };
 
-pb_log(1, "DEBUG MODE, not doing anything, just printing\nDEBUG: basedir = $basedir\nDEBUG: source  = $source\n");
-pb_log(1, "DEBUG: machine = $machine\n") if (defined ($machine));
+pb_log(1, "DEBUG MODE, not doing anything, just printing\nDEBUG: basedir = $basedir\n");
+pb_log(1, "DEBUG: source = $source\n") if (defined $source);
+pb_log(1, "DEBUG: machine = $machine\n") if (defined $machine);
+
+# Use potentially a remote account if defined
+my $account = undef;
+my $remote = $machine;
+($account) = pb_conf_get_if("cbaccount") if (defined $machine);
+$remote = $account->{$machine}."@".$machine if ((defined $account) && (defined $account->{$machine}));
+pb_log(1, "DEBUG: remote = $remote\n") if (defined $remote);
 
 # Create basedir if it doesn't exist
@@ -198 +217 @@
         pb_log(1, "DEBUG: Creating recursively directory $basedir\n");
     } else {
-        pb_mkdir_p($basedir) || die "Unable to recursively create $basedir";
-    }
-}
-
+        pb_mkdir_p($basedir) || die "Unable to recursively create $basedir: $!";
+    }
+}
+
+if (defined $source) {
+    foreach my $f (split(/,/,$source)) {    
+        cb_busterize($f);
+    }
+}
+
+# Now handle plugins if any
+my $cbp = ();
+
+if (defined $plugin) {
+    foreach my $p (split(/,/,$plugin)) {    
+        pb_log(1,"Getting context for plugin $p\n");
+        $cbp = cb_plugin_get($p,$cbp);
+        pb_log(2,"cbp: ".Dumper($cbp)."\n");
+        foreach my $k ((keys %{$cbp->{$plugin}->{'files'}}),(keys %{$cbp->{$plugin}->{'dirs'}})) {
+            cb_busterize($k);
+        }
+    }
+}
+
+sub cb_busterize {
+
+my $source = shift;
+
+pb_log(2,"Entering cb_busterize source: $source\n");
 # Is the source a file or a dir ? Split the source parameter in 2
 my $srcdir = undef;
 my $srcfile = undef;
-# TODO: That should be remote !!
-if (-d $source) {
-    $srcdir = $source;
+my $cmd = undef;
+
+if (not defined $machine) {
+    if (-d $source) {
+        $srcdir = $source;
+    } else {
+        $srcdir = dirname($source);
+        $srcfile = basename($source);
+    }
 } else {
-    $srcdir = dirname($source);
-    $srcfile = basename($source);
-    }
+    $cmd = "ssh -q $remote \'echo \"if [ -d $source ]; then exit 0; else exit -1; fi\" | sudo bash\'";
+    my $res = pb_system($cmd,"","quiet");
+    pb_log(2,"DEBUG: Found res = $res\n");
+    if ($res == 0) {
+        $srcdir = $source;
+        pb_log(1,"DEBUG: Found remote dir = $source\n");
+    } else {
+        $srcdir = dirname($source);
+        $srcfile = basename($source);
+        pb_log(1,"DEBUG: Found remote file = $source\n");
+    }
+}
 
 pb_log(1,"DEBUG: Found srcdir = $srcdir\n");
@@ -230 +289 @@
 $target = basename($target) if ((not defined $srcfile) && (-d $target));
 
-# Create target if it doesn't exist
-if (not -d $target) {
+# Create target if it doesn't exist when we have to copy a file
+if ((not -d $target) && (defined $srcfile)) {
     if ($debug) {
         pb_log(1,"DEBUG: Creating recursively directory $target\n");
     } else {
-        pb_mkdir_p($target) || die "Unable to recursively create $target";
+        pb_mkdir_p($target) || die "Unable to recursively create $target: $!";
         pb_vcs_add($scheme,$target);
         pb_log(0,"INFO: Created $target and added it to your $scheme system\n");
@@ -242 +301 @@
 
 # We need to know where to get the content from
-my $cmd;
 my $cmdopt = "";
 
@@ -249 +307 @@
 
 if (defined $machine) {
-    $cmd = "scp -p -q $cmdopt $machine:$source $target";
+    $cmd = "scp -p -q $cmdopt $remote:$source $target";
 } else {
     $cmd = "cp -p $cmdopt $source $target";
@@ -257 +315 @@
 if (defined $srcfile) {
     # File case
-    if (! -f "$target/$srcfile") {
+    if ((! -f "$target/$srcfile") || (defined $force)){
         if ($debug) {
             pb_log(1,"DEBUG: launching $cmd\n");
@@ -270 +328 @@
 } else {
     # Directory case
-    # TODO: if targetlocal dir alredy exists, take the parent
     if ($debug) {
         pb_log(1,"DEBUG: launching $cmd\n");
@@ -279 +336 @@
     }
 }
+pb_log(2,"Exiting cb_busterize\n");
+}