interu-backup 3.0.16

data/lib/backup/configuration/storage/rsync.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+
+module Backup
+  module Configuration
+    module Storage
+      class RSync < Base
+        class << self
+
+          ##
+          # Server credentials
+          attr_accessor :username, :password
+
+          ##
+          # Server IP Address and SSH port
+          attr_accessor :ip, :port
+
+          ##
+          # Path to store backups to
+          attr_accessor :path
+
+        end
+      end
+    end
+  end
+end

data/lib/backup/configuration/storage/s3.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+
+module Backup
+  module Configuration
+    module Storage
+      class S3 < Base
+        class << self
+
+          ##
+          # Amazon Simple Storage Service (S3) Credentials
+          attr_accessor :access_key_id, :secret_access_key
+
+          ##
+          # Amazon S3 bucket name and path
+          attr_accessor :bucket, :path
+
+          ##
+          # Region of the specified S3 bucket
+          attr_accessor :region
+
+        end
+      end
+    end
+  end
+end

data/lib/backup/configuration/storage/scp.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+
+module Backup
+  module Configuration
+    module Storage
+      class SCP < Base
+        class << self
+
+          ##
+          # Server credentials
+          attr_accessor :username, :password
+
+          ##
+          # Server IP Address and SCP port
+          attr_accessor :ip, :port
+
+          ##
+          # Path to store backups to
+          attr_accessor :path
+
+        end
+      end
+    end
+  end
+end

data/lib/backup/configuration/storage/sftp.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+
+module Backup
+  module Configuration
+    module Storage
+      class SFTP < Base
+        class << self
+
+          ##
+          # Server credentials
+          attr_accessor :username, :password
+
+          ##
+          # Server IP Address and SFTP port
+          attr_accessor :ip, :port
+
+          ##
+          # Path to store backups to
+          attr_accessor :path
+
+        end
+      end
+    end
+  end
+end

data/lib/backup/configuration/syncer/rsync.rb

@@ -0,0 +1,45 @@
+# encoding: utf-8
+
+module Backup
+  module Configuration
+    module Syncer
+      class RSync < Base
+        class << self
+
+          ##
+          # Server credentials
+          attr_accessor :username, :password
+
+          ##
+          # Server IP Address and SSH port
+          attr_accessor :ip
+
+          ##
+          # The SSH port to connect to
+          attr_accessor :port
+
+          ##
+          # Directories to sync
+          attr_accessor :directories
+
+          ##
+          # Path to store the synced files/directories to
+          attr_accessor :path
+
+          ##
+          # Flag for mirroring the files/directories
+          attr_accessor :mirror
+
+          ##
+          # Flag for compressing (only compresses for the transfer)
+          attr_accessor :compress
+
+          ##
+          # Additional options for the rsync cli
+          attr_accessor :additional_options
+
+        end
+      end
+    end
+  end
+end

data/lib/backup/configuration/syncer/s3.rb

@@ -0,0 +1,33 @@
+# encoding: utf-8
+
+module Backup
+  module Configuration
+    module Syncer
+      class S3 < Base
+        class << self
+
+          ##
+          # Amazon Simple Storage Service (S3) Credentials
+          attr_accessor :access_key_id, :secret_access_key
+
+          ##
+          # Amazon S3 bucket name and path to sync to
+          attr_accessor :bucket, :path
+
+          ##
+          # Directories to sync
+          attr_accessor :directories
+
+          ##
+          # Flag to enable mirroring
+          attr_accessor :mirror
+
+          ##
+          # Additional options for the s3sync cli
+          attr_accessor :additional_options
+
+        end
+      end
+    end
+  end
+end

data/lib/backup/database/base.rb

@@ -0,0 +1,33 @@
+# encoding: utf-8
+
+module Backup
+  module Database
+    class Base
+      include Backup::CLI
+      include Backup::Configuration::Helpers
+
+      ##
+      # Contains the path to where the database should be dumped
+      attr_accessor :dump_path
+
+      ##
+      # Allows the user to specify the path to a "dump" utility
+      # in case it cannot be auto-detected by Backup
+      attr_accessor :utility_path
+
+      ##
+      # Defines the @dump_path and ensures it exists by creating it
+      def prepare!
+        @dump_path = File.join(TMP_PATH, TRIGGER, self.class.name.split('::').last)
+        mkdir(dump_path)
+      end
+
+      ##
+      # Logs a message to the console and log file to inform
+      # the client that Backup is dumping the database
+      def log!
+        Logger.message("#{ self.class } started dumping and archiving \"#{ name }\".")
+      end
+    end
+  end
+end

data/lib/backup/database/mongodb.rb

@@ -0,0 +1,179 @@
+# encoding: utf-8
+
+module Backup
+  module Database
+    class MongoDB < Base
+
+      ##
+      # Name of the database that needs to get dumped
+      attr_accessor :name
+
+      ##
+      # Credentials for the specified database
+      attr_accessor :username, :password
+
+      ##
+      # Connectivity options
+      attr_accessor :host, :port
+
+      ##
+      # IPv6 support (disabled by default)
+      attr_accessor :ipv6
+
+      ##
+      # Collections to dump, collections that aren't specified won't get dumped
+      attr_accessor :only_collections
+
+      ##
+      # Additional "mongodump" options
+      attr_accessor :additional_options
+
+      ##
+      # 'lock' dump meaning wrapping mongodump with fsync & lock
+      attr_accessor :lock
+
+      ##
+      # Creates a new instance of the MongoDB database object
+      def initialize(&block)
+        load_defaults!
+
+        @only_collections   ||= Array.new
+        @additional_options ||= Array.new
+        @ipv6               ||= false
+        @lock               ||= false
+
+        instance_eval(&block)
+        prepare!
+      end
+
+      ##
+      # Builds the MongoDB credentials syntax to authenticate the user
+      # to perform the database dumping process
+      def credential_options
+        %w[username password].map do |option|
+          next if send(option).nil? or send(option).empty?
+          "--#{option}='#{send(option)}'"
+        end.compact.join("\s")
+      end
+
+      ##
+      # Builds the MongoDB connectivity options syntax to connect the user
+      # to perform the database dumping process
+      def connectivity_options
+        %w[host port].map do |option|
+          next if send(option).nil? or (send(option).respond_to?(:empty?) and send(option).empty?)
+          "--#{option}='#{send(option)}'"
+        end.compact.join("\s")
+      end
+
+      ##
+      # Builds a MongoDB compatible string for the
+      # additional options specified by the user
+      def additional_options
+        @additional_options.join("\s")
+      end
+
+      ##
+      # Returns an array of collections to dump
+      def collections_to_dump
+        @only_collections
+      end
+
+      ##
+      # Returns the MongoDB database selector syntax
+      def database
+        "--db='#{ name }'"
+      end
+
+      ##
+      # Returns the mongodump syntax for enabling ipv6
+      def ipv6
+        @ipv6.eql?(true) ? '--ipv6' : ''
+      end
+
+      ##
+      # Returns the MongoDB syntax for determining where to output all the database dumps,
+      # e.g. ~/Backup/.tmp/MongoDB/<databases here>/<database collections>
+      def dump_directory
+        "--out='#{ dump_path }'"
+      end
+
+      ##
+      # Builds the full mongodump string based on all attributes
+      def mongodump
+        "#{ utility(:mongodump) } #{ database } #{ credential_options } " +
+        "#{ connectivity_options } #{ ipv6 } #{ additional_options } #{ dump_directory }"
+      end
+
+      ##
+      # Performs the mongodump command and outputs the data to the
+      # specified path based on the 'trigger'. If the user hasn't specified any
+      # specific collections to dump, it'll dump everything. If the user has specified
+      # collections to dump, it'll loop through the array of collections and invoke the
+      # 'mongodump' command once per collection
+      def perform!
+        log!
+
+        begin
+          lock_database if @lock.eql?(true)
+          if collections_to_dump.is_a?(Array) and not collections_to_dump.empty?
+            specific_collection_dump!
+          else
+            dump!
+          end
+          unlock_database if @lock.eql?(true)
+        rescue => exception
+          unlock_database if @lock.eql?(true)
+          raise exception
+        end
+      end
+
+      ##
+      # Builds and runs the mongodump command
+      def dump!
+        run(mongodump)
+      end
+
+      ##
+      # For each collection in the @only_collections array, it'll
+      # build the whole 'mongodump' command, append the '--collection' option,
+      # and run the command built command
+      def specific_collection_dump!
+        collections_to_dump.each do |collection|
+          run("#{mongodump} --collection='#{collection}'")
+        end
+      end
+
+      ##
+      # Builds the full mongo string based on all attributes
+      def mongo_shell
+        [utility(:mongo), database, credential_options, connectivity_options, ipv6].join(' ')
+      end
+
+      ##
+      # Locks and FSyncs the database to bring it up to sync
+      # and ensure no 'write operations' are performed during the
+      # dump process
+      def lock_database
+        lock_command = <<-EOS
+          echo 'use admin
+          db.runCommand({"fsync" : 1, "lock" : 1})' | #{mongo_shell}
+        EOS
+
+        run(lock_command)
+      end
+
+      ##
+      # Unlocks the (locked) database
+      def unlock_database
+        unlock_command = <<-EOS
+          echo 'use admin
+          db.$cmd.sys.unlock.findOne()' | #{mongo_shell}
+        EOS
+
+        run(unlock_command)
+      end
+
+    end
+  end
+end

data/lib/backup/database/mysql.rb

@@ -0,0 +1,104 @@
+# encoding: utf-8
+
+module Backup
+  module Database
+    class MySQL < Base
+
+      ##
+      # Name of the database that needs to get dumped
+      attr_accessor :name
+
+      ##
+      # Credentials for the specified database
+      attr_accessor :username, :password
+
+      ##
+      # Connectivity options
+      attr_accessor :host, :port, :socket
+
+      ##
+      # Tables to skip while dumping the database
+      attr_accessor :skip_tables
+
+      ##
+      # Tables to dump, tables that aren't specified won't get dumped
+      attr_accessor :only_tables
+
+      ##
+      # Additional "mysqldump" options
+      attr_accessor :additional_options
+
+      ##
+      # Creates a new instance of the MySQL adapter object
+      def initialize(&block)
+        load_defaults!
+
+        @skip_tables        ||= Array.new
+        @only_tables        ||= Array.new
+        @additional_options ||= Array.new
+
+        instance_eval(&block)
+        prepare!
+      end
+
+      ##
+      # Builds the MySQL syntax for specifying which tables to skip
+      # during the dumping of the database
+      def tables_to_skip
+        skip_tables.map do |table|
+          "--ignore-table='#{name}.#{table}'"
+        end.join("\s")
+      end
+
+      ##
+      # Builds the MySQL syntax for specifying which tables to dump
+      # during the dumping of the database
+      def tables_to_dump
+        only_tables.join("\s")
+      end
+
+      ##
+      # Builds the credentials MySQL syntax to authenticate the user
+      # to perform the database dumping process
+      def credential_options
+        %w[username password].map do |option|
+          next if send(option).nil? or send(option).empty?
+          "--#{option}='#{send(option)}'".gsub('--username', '--user')
+        end.compact.join("\s")
+      end
+
+      ##
+      # Builds the MySQL connectivity options syntax to connect the user
+      # to perform the database dumping process
+      def connectivity_options
+        %w[host port socket].map do |option|
+          next if send(option).nil? or (send(option).respond_to?(:empty?) and send(option).empty?)
+          "--#{option}='#{send(option)}'"
+        end.compact.join("\s")
+      end
+
+      ##
+      # Builds a MySQL compatible string for the additional options
+      # specified by the user
+      def options
+        additional_options.join("\s")
+      end
+
+      ##
+      # Builds the full mysqldump string based on all attributes
+      def mysqldump
+        "#{ utility(:mysqldump) } #{ credential_options } #{ connectivity_options } " +
+        "#{ options } #{ name } #{ tables_to_dump } #{ tables_to_skip }"
+      end
+
+      ##
+      # Performs the mysqldump command and outputs the
+      # data to the specified path based on the 'trigger'
+      def perform!
+        log!
+        run("#{mysqldump} > '#{File.join(dump_path, name)}.sql'")
+      end
+
+    end
+  end
+end