SpontaneousCMS/spontaneous

module Spontaneous
  module Utils
    module Database
      class MySQLDumper

        def initialize(database)
          @database = database
        end

        def name
          "mysql"
        end

        def load(path)
          system(load_command(path))
        end

        def load_command(path)
          options = [
            "mysql",
            option(:password),
            option(:user),
            option(:default_character_set),
            database_name
          ]
          if path =~ /\.gz$/
            options = ["gunzip", "<", path, "|"].concat(options)
          end

          command = options.join(" ")
        end

        def dump(path, tables = nil)
          system(dump_command(path, tables))
        end

        def dump_command(path, tables = nil)
          options = [
            option(:password),
            option(:user),
            option(:default_character_set),
            option(:ignore_table),
            database_name
          ]
          unless tables.nil?
            options.push(tables.join(" "))
          end

          options.push( "| gzip") if path =~ /\.gz$/

          command = %(mysqldump #{options.join(" ")} > #{path} )
        end

        def database_name
          @database.opts[:database]
        end

        def user
          @database.opts[:user]
        end

        def password
          @database.opts[:password]
        end

        def default_character_set
          "UTF8"
        end

        def ignore_table
          [database_name, revision_archive_table].join(".")
        end

        def dumpfilename(time = Time.now)
          "#{time.strftime('%Y%m%d%H%M%S')}.#{name}.gz"
        end

        # The whole point of this table's existance is to reduce the
        # size of the db dump for syncing
        def revision_archive_table
          Spontaneous::instance.model.archive_dataset.first_source.to_s
        end

        def option(option, add_if_nil=false)
          value = self.send(option)
          if !value.nil? or add_if_nil
            "--#{option.to_s.gsub(/_/, '-')}=#{value}"
          else
            ""
          end
        end
      end
    end
  end
end