Move filebucket to the Application Controller paradigm

Signed-off-by: Brice Figureau <brice-puppet@daysofwonder.com>

1 files changed, 92 insertions, 0 deletions

diff --git a/lib/puppet/application/filebucket.rb b/lib/puppet/application/filebucket.rb
new file mode 100644
index 000000000..f456e0dcc
--- /dev/null
+++ b/lib/puppet/application/filebucket.rb
@@ -0,0 +1,92 @@
+require 'puppet'
+require 'puppet/application'
+require 'puppet/network/client'
+
+Puppet::Application.new(:filebucket) do
+
+    should_not_parse_config
+
+    option("--version", "-V") do |arg|
+        puts "%s" % Puppet.version
+        exit
+    end
+
+    option("--bucket BUCKET","-b")
+    option("--debug","-d")
+    option("--local","-l")
+    option("--remote","-r")
+    option("--verbose","-v")
+
+    dispatch do
+        ARGV.shift
+    end
+
+    command(:get) do
+        md5 = ARGV.shift
+        out = @client.getfile(md5)
+        print out
+    end
+
+    command(:backup) do
+        ARGV.each do |file|
+            unless FileTest.exists?(file)
+                $stderr.puts "%s: no such file" % file
+                next
+            end
+            unless FileTest.readable?(file)
+                $stderr.puts "%s: cannot read file" % file
+                next
+            end
+            md5 = @client.backup(file)
+            puts "%s: %s" % [file, md5]
+        end
+    end
+
+    command(:restore) do
+        file = ARGV.shift
+        md5 = ARGV.shift
+        @client.restore(file, md5)
+    end
+
+    setup do
+        Puppet::Log.newdestination(:console)
+
+        @client = nil
+        @server = nil
+
+        trap(:INT) do
+            $stderr.puts "Cancelling"
+            exit(1)
+        end
+
+        if options[:debug]
+            Puppet::Log.level = :debug
+        elsif options[:verbose]
+            Puppet::Log.level = :info
+        end
+
+        # Now parse the config
+        Puppet.parse_config
+
+        if Puppet.settings.print_configs?
+                exit(Puppet.settings.print_configs ? 0 : 1)
+        end
+
+        begin
+            if options[:local] or options[:bucket]
+                path = options[:bucket] || Puppet[:bucketdir]
+                @client = Puppet::Network::Client.dipper.new(:Path => path)
+            else
+                require 'puppet/network/handler'
+                @client = Puppet::Network::Client.dipper.new(:Server => Puppet[:server])
+            end
+        rescue => detail
+            $stderr.puts detail
+            if Puppet[:trace]
+                puts detail.backtrace
+            end
+            exit(1)
+        end
+    end
+
+end
\ No newline at end of file