cifrado 0.1

data/lib/cifrado.rb

@@ -0,0 +1,48 @@
+require 'cifrado/version'
+require 'cifrado/core_ext/ruby18_base64'
+require 'logger'
+require 'shellwords'
+require 'uri'
+require 'thor'
+require 'fileutils'
+require 'pathname'
+require 'securerandom'
+require 'cifrado/config'
+require 'cifrado/utils'
+require 'cifrado/cli/progressbar'
+
+module Cifrado
+  
+  if !defined? Log or Log.nil?
+    shell = Thor::Shell::Color.new
+    Log = Logger.new($stdout)
+    Log.formatter = proc do |severity, datetime, progname, msg|
+      if severity == 'ERROR' or severity == 'FATAL'
+        "#{shell.set_color(severity, :red, true)}: #{msg}\n"
+      elsif severity == 'WARN'
+        "#{shell.set_color(severity, :yellow, true)}: #{msg}\n"
+      elsif severity == 'INFO'
+        "#{msg}\n"
+      else
+        "#{severity}: #{msg}\n"
+      end
+    end
+    Log.level = Logger::INFO unless ENV['DEBUG']
+    Log.debug "Initializing logger"
+  end
+
+end
+
+vendored_fog = File.join(File.dirname(__FILE__), 'cifrado/vendor/fog/lib')
+if File.directory?(vendored_fog)
+  fog_path = vendored_fog
+else
+  fog_path = File.join(File.dirname(__FILE__), '/../', 'vendor/fog/lib')
+end
+
+$:.insert 0, fog_path
+require 'fog/openstack'
+require 'cifrado/swift_client'
+require 'cifrado/file_splitter'
+require 'cifrado/crypto_services'
+require 'cifrado/cli'

data/lib/cifrado/cli.rb

@@ -0,0 +1,140 @@
+module Cifrado
+  class CLI < Thor
+
+    include Cifrado
+    include Cifrado::Utils
+
+    attr_reader :config
+
+    check_unknown_options!
+
+    class_option :username
+    class_option :quiet
+    class_option :password
+    class_option :auth_url
+    class_option :tenant
+    class_option :config
+    class_option :region
+    class_option :insecure, :type => :boolean, :desc => "Insecure SSL connections"
+    class_option :debug,    :type => :boolean
+
+    private
+    def secure_password
+      @config[:password] + @config[:secure_random]
+    end
+
+    def bwlimit
+      (options[:bwlimit] * 1024 * 1024)/8 if options[:bwlimit]
+    end
+
+    def client_instance
+
+      if options[:quiet] and Log.level < Logger::WARN
+        Log.level = Logger::WARN
+      end
+
+      if options[:debug]
+        Log.level = Logger::DEBUG
+      end
+
+      config = check_options
+      if options[:insecure]
+        Log.warn "SSL verification DISABLED"
+      end
+      client = Cifrado::SwiftClient.new :username => config[:username], 
+                                        :api_key  => config[:password],
+                                        :auth_url => config[:auth_url],
+                                        :tenant   => config[:tenant],
+                                        :region   => config[:region],
+                                        :password_salt => config[:secure_random],
+                                        :connection_options => { 
+                                          :ssl_verify_peer => !options[:insecure] 
+                                        }
+      @client = client
+      @config = config
+      # Validate connection
+      client.test_connection
+      client
+    end
+
+    def check_options
+      config_file = options[:config] || File.join(ENV['HOME'], '.config/cifrado/cifradorc')
+      config = {}
+
+      if File.exist?(config_file)
+        begin
+          Log.debug "Configuration file found: #{config_file}"
+          Cifrado::Log.debug "Trying to read config file #{config_file}"
+          config = YAML.load_file(config_file)
+          Cifrado::Log.debug "Config #{config_file} read"
+          original_config = config.dup
+        rescue => e
+          Cifrado::Log.error "Error loading config file"
+          raise e
+        end
+      end
+
+      config[:username]        = options[:username] || config[:username]
+      config[:password]        = options[:password] || config[:password]
+      config[:auth_url]        = options[:auth_url] || config[:auth_url] 
+      config[:tenant]          = options[:tenant]   || config[:tenant] 
+      config[:region]          = options[:region]   || config[:region] 
+      config[:secure_random]   = config[:secure_random]
+      [:username, :password, :auth_url, :tenant].each do |opt|
+        if config[opt].nil?
+          Log.error "#{opt.to_s.capitalize} not provided."
+          Log.error "Use --#{opt.to_s.gsub('_', '-')} option or run 'cifrado setup' first."
+          raise "Missing setting"
+        end
+      end
+      unless config[:secure_random]
+        raise Exception.new("secure_random key not found in #{config_file}")
+      end
+
+      config
+    end
+
+  end
+end
+
+require 'cifrado/cli/stat'
+require 'cifrado/cli/download'
+require 'cifrado/cli/list'
+require 'cifrado/cli/post'
+require 'cifrado/cli/delete'
+require 'cifrado/cli/setup'
+require 'cifrado/cli/upload'
+require 'cifrado/cli/set_acl'
+require 'cifrado/cli/jukebox'
+require 'cifrado/cli/cinema'
+require 'cifrado/cli/saio'
+
+at_exit do
+  include Cifrado::Utils
+  include Cifrado
+  e = $!
+  if e
+    if e.is_a? Excon::Errors::Unauthorized
+      Log.error "Unauthorized"
+      Log.error "Double check the username, password and auth_url."
+    elsif e.is_a? Excon::Errors::SocketError
+      if e.message =~ /Unable to verify certificate|hostname (was|does) not match (with )?the server/
+        Log.error "Unable to verify SSL certificate."
+        Log.error "If the server is using a self-signed certificate, try using --insecure."
+        Log.error "Please be aware of the security implications."
+      else
+        Log.error e.message
+      end
+    elsif e.is_a? RuntimeError
+      Log.error e.message
+    elsif e.is_a? Interrupt
+      Log.info
+      Log.info 'At your command, Sir!'
+    else
+      Log.fatal e.message
+    end
+    system 'stty echo'
+    prettify_backtrace e
+    exit! 1
+  end
+end

data/lib/cifrado/cli/cinema.rb

@@ -0,0 +1,81 @@
+module Cifrado
+  class CLI
+
+    desc "cinema CONTAINER VIDEO", 
+         "Stream videos from the target container"
+    option :subtitles, :type => :boolean
+    def cinema(container, video)
+      client = client_instance
+      
+      dir = client.service.directories.get container
+      unless dir
+        raise "Container #{container} not found."
+      end
+      
+      to_play = dir.files.find { |v| v.key =~ /#{video}/ }
+      unless to_play
+        raise "No video matches #{video}"
+      end
+
+      sub_file = nil
+      if options[:subtitles]
+        begin
+          Log.info "Subtitles found, downloading."
+          sub_object = to_play.key.gsub(/\.(avi|mpeg|mov|mkv|ogv|webm)$/, '') + '.srt'
+          sub = client.head container, sub_object
+          sub_file = "/tmp/#{SecureRandom.hex}"
+          client.download container, sub_object, :output => sub_file
+        rescue => e
+          Log.error "Error downloading subtitles, ignoring."
+        end
+      end
+
+      $stderr.reopen('/dev/null', 'w')
+      pipe = IO.popen(player_command(sub_file), 'w')
+      begin
+        cb = Proc.new do |total, bytes, segment|
+          pipe.write segment if bytes > 0 
+        end
+
+        Log.info "#{set_color 'Playing', :bold} #{to_play.key}"
+        r = client.stream container, 
+                          to_play.key, 
+                          :progress_callback => cb
+        Log.debug "Video finished streaming"
+      rescue => e
+        Log.debug "Closing pipe"
+        prettify_backtrace e
+        pipe.close unless pipe.closed?
+        Log.info set_color "\nAdios!", :bold
+      end
+    end
+
+    private
+    def player_command(subtitles = nil)
+      cvlc = '/usr/bin/cvlc'
+      vlc = '/usr/bin/vlc'
+      mplayer = '/usr/bin/mplayer'
+      totem = '/usr/bin/totem'
+
+      if File.exist?(cvlc)
+        Log.debug "Using cvlc player"
+        extra_args = "--sub-file #{subtitles}" if subtitles
+        "/usr/bin/cvlc #{extra_args} -" 
+      elsif File.exist?(mplayer)
+        extra_args = "-sub #{subtitles}" if subtitles
+        Log.debug "Using mplayer player"
+        "/usr/bin/mplayer #{extra_args} -really-quiet -msglevel all=-1 -cache 256 -"
+      elsif File.exist?(vlc)
+        Log.debug "Using vlc player"
+        extra_args = "--sub-file #{subtitles}" if subtitles
+        "/usr/bin/vlc #{extra_args} -"
+      elsif File.exist?(totem)
+        Log.debug "Using totem player"
+        '/usr/bin/totem --enqueue fd://0'
+      else
+        raise "No player available. Install MPlayer, VLC or totem."
+      end
+    end
+
+  end
+end

data/lib/cifrado/cli/delete.rb

@@ -0,0 +1,36 @@
+module Cifrado
+  class CLI
+    desc "delete CONTAINER [OBJECT]", "Delete specific container or object"
+    def delete(container, object = nil)
+      client = client_instance
+      if object
+        Log.info "Deleting file #{object}..."
+        deleted = false
+        begin
+          client.service.delete_object container, object
+          deleted = true
+        rescue Fog::Storage::OpenStack::NotFound
+          Log.debug 'Trying to find hashed object name'
+          file_hash = (Digest::SHA2.new << object).to_s
+          deleted = client.service.delete_object(container, file_hash) rescue nil
+        end
+        if deleted
+          Log.info "File #{object} deleted"
+        else
+          Log.error "File #{object} not found"
+        end
+      else
+        Log.info "Deleting container '#{container}'..."
+        dir = client.service.directories.get(container)
+        if dir
+          dir.files.each do |f|
+            Log.info "Deleting file #{f.key}..."
+            f.destroy
+          end
+          dir.destroy
+          Log.info "Container #{container} deleted"
+        end
+      end
+    end
+  end
+end

data/lib/cifrado/cli/download.rb

@@ -0,0 +1,54 @@
+module Cifrado
+  class CLI
+    desc "download [CONTAINER] [OBJECT]", "Download container, objects"
+    option :decrypt, :type => :boolean
+    option :passphrase, :type => :string, :desc => "Passphrase used to decrypt the file"
+    option :output
+    option :progressbar, :default => :fancy
+    option :bwlimit, :type => :numeric
+    def download(container, object = nil)
+      client = client_instance
+      downloaded = []
+      files = []
+      if object
+        files << object
+      else
+        Log.info "Downloading files from container #{container}"
+        dir = client.service.directories.get(container)
+        files = dir.files if dir
+      end
+      pb = Progressbar.new 1, 1, :style => options[:progressbar]
+      files.each do |f|
+        obj = f.is_a?(String) ? f : f.key
+        if !f.is_a?(String) and f.metadata[:encrypted_name]
+          fname = decrypt_filename f.metadata[:encrypted_name],
+                                   @config[:password] + @config[:secure_random]
+          Log.info "Downloading file #{fname}"
+        else
+          Log.info "Downloading file #{obj}"
+        end
+        if client.file_available?(container, obj)
+          r = client.download container, obj,
+                              :decrypt => options[:decrypt],
+                              :passphrase => options[:passphrase],
+                              :output => options[:output],
+                              :progress_callback => pb.block,
+                              :bwlimit => bwlimit
+          downloaded << obj
+        else 
+          Log.debug 'Trying to find hashed object name'
+          file_hash = (Digest::SHA2.new << obj).to_s
+          r = client.download container, file_hash,
+                              :decrypt => options[:decrypt],
+                              :passphrase => options[:passphrase],
+                              :output => options[:output],
+                              :progress_callback => pb.block,
+                              :bwlimit => bwlimit
+          downloaded << file_hash if r.status == 200
+        end
+      end
+      Log.warn "No files were downloaded." if downloaded.empty?
+      downloaded
+    end
+  end
+end

data/lib/cifrado/cli/jukebox.rb

@@ -0,0 +1,94 @@
+module Cifrado
+  class CLI
+
+    desc "jukebox CONTAINER", 
+         "Play music randomly from the target container"
+    option :match, :type => :string
+    def jukebox(container)
+      client = client_instance
+      
+      dir = client.service.directories.get container
+      unless dir
+        raise "Container #{container} not found."
+      end
+      songs = dir.files
+      last_exit = Time.now.to_f
+
+      Log.info
+      Log.info set_color "Cifrado Jukebox", :bold
+      Log.info "---------------"
+      Log.info
+      Log.info set_color("Ctrl-C once", :bold)+ "   -> next song"
+      Log.info set_color("Ctrl-C twice", :bold)+ "  -> quit"
+      Log.info
+      $stderr.reopen('/dev/null', 'w')
+      pipe = IO.popen(player_command, 'w')
+      songs.shuffle.each do |song|
+        if options[:match] and song.key !~ /#{options[:match]}/i
+          next
+        end
+        begin
+
+          cb = Proc.new do |total, bytes, segment|
+            pipe.write segment if bytes > 0 
+          end
+
+          unless (song.content_type =~ /ogg|mp3/) or \
+                  song.key =~ /(mp3|wav|ogg)$/
+            next
+          end
+          Log.info "#{set_color 'Playing', :bold} song"
+          Log.info "  * #{song.key}"
+          r = client.stream container, 
+                            song.key, 
+                            :progress_callback => cb
+          Log.debug "Song finished streaming"
+        rescue Timeout::Error, Errno::EPIPE, Interrupt => e
+          Log.debug "Closing pipe"
+          prettify_backtrace e
+          pipe.close unless pipe.closed?
+          Log.debug "Opening new pipe"
+          pipe = IO.popen(player_command, 'w')
+          if Time.now.to_f - last_exit < 1
+            Log.info set_color "\nAdios!", :bold
+            return
+          else
+            last_exit = Time.now.to_f
+            Log.info 'Next song...'
+          end
+        end
+      end
+    ensure
+      if pipe and !pipe.closed?
+        Log.debug "Closing pipe, killing mplayer"
+        Process.kill 'SIGKILL', pipe.pid
+        pipe.close 
+      end
+    end
+
+    private
+    def player_command
+      cvlc = '/usr/bin/cvlc'
+      vlc = '/usr/bin/vlc'
+      mplayer = '/usr/bin/mplayer'
+      totem = '/usr/bin/totem'
+
+      if File.exist?(cvlc)
+        Log.debug "Using cvlc player"
+        '/usr/bin/cvlc -' 
+      elsif File.exist?(mplayer)
+        Log.debug "Using mplayer player"
+        '/usr/bin/mplayer -really-quiet -msglevel all=-1 -cache 256 -'
+      elsif File.exist?(vlc)
+        Log.debug "Using vlc player"
+        '/usr/bin/vlc -'
+      elsif File.exist?(totem)
+        Log.debug "Using totem player"
+        '/usr/bin/totem --enqueue fd://0'
+      else
+        raise "No player available. Install MPlayer, VLC or totem."
+      end
+    end
+
+  end
+end