module EventStore

# File lib/event_store.rb, line 111
def self.clear!
  return unless connected?
  EventStore.db.from(fully_qualified_table).delete
  EventStore.db.from(fully_qualified_names_table).delete
  @redis.map(&:flushdb)
end

# File lib/event_store.rb, line 49
def self.connect(*args)
  @db ||= Sequel.connect(*args)
end

# File lib/event_store.rb, line 175
def self.connect_db
  self.connect(@db_config)
end

# File lib/event_store.rb, line 107
def self.connected?
  !!EventStore.db
end

# File lib/event_store.rb, line 179
def self.create_db
  connect_db
  table = Sequel.qualify(schema, "schema_info")
  @db.run("CREATE SCHEMA IF NOT EXISTS #{schema}")
  Sequel::Migrator.run(@db, File.expand_path(File.join('..','..','db', self.migrations_dir), __FILE__), table: table)
end

# File lib/event_store.rb, line 152
def self.custom_config(database_config, redis_config, table_name = 'events', environment = 'production')
  database_config = database_config.each_with_object({}) {|(k,v), memo| memo[k.to_s] = v}
  redis_config    = redis_config.each_with_object({}) {|(k,v), memo| memo[k.to_s] = v}

  self.redis_connect(redis_config)

  @adapter         = database_config["adapter"].to_s
  @environment     = environment
  @db_config       = database_config
  @table_name      = table_name
  @schema          = database_config["schema"].to_s
  @use_names_table = database_config.fetch("use_names_table", true)
  connect_db
end

# File lib/event_store.rb, line 40
def self.db
  @db
end

# File lib/event_store.rb, line 26
def self.db_config
  raw_db_config[@environment.to_s][@adapter.to_s]
end

# File lib/event_store.rb, line 143
def self.escape_bytea(binary_string)
  binary_string.unpack('H*').join
end

# File lib/event_store.rb, line 103
def self.fully_qualified_names_table
  @fully_qualified_names_table ||= Sequel.lit "#{schema}.#{lookup_table_name}"
end

# File lib/event_store.rb, line 95
def self.fully_qualified_table
  @fully_qualified_table ||= qualified_table_name
end

# File lib/event_store.rb, line 72
def self.insert_table_name(date)
  return fully_qualified_table unless partitioning?

  partition_name = date.strftime("#{table_name}#{partition_name_suffix}")
  qualified_table_name(partition_name)
end

# File lib/event_store.rb, line 64
def self.local_redis_config
  @redis_connection ||= raw_db_config['redis']
end

# File lib/event_store.rb, line 91
def self.lookup_table_name
  @lookup_table_name ||= raw_db_config['lookup_table_name'] || "fully_qualified_names"
end

# File lib/event_store.rb, line 171
def self.migrations_dir
  @adapter == 'vertica' ? 'migrations' : 'pg_migrations'
end

# File lib/event_store.rb, line 83
def self.partition_name_suffix
  @db_config["partition_name_suffix"]
end

# File lib/event_store.rb, line 79
def self.partitioning?
  @db_config["partitioning"]
end

# File lib/event_store.rb, line 118
def self.postgres(environment = 'test', table_name = 'events', schema = 'event_store_test')
  @schema         = schema
  @table_name     = table_name
  @environment    = environment.to_s
  @adapter        = 'postgres'
  @db_config      ||= self.db_config
  custom_config(@db_config, local_redis_config, @table_name, environment)
end

# File lib/event_store.rb, line 99
def self.qualified_table_name(name = table_name)
  Sequel.lit "#{schema}.#{name}"
end

# File lib/event_store.rb, line 30
def self.raw_db_config
  if @raw_db_config.nil?
    file_path = File.expand_path(__FILE__ + '/../../db/database.yml')
    @config_file = File.open(file_path,'r')
    @raw_db_config = YAML.load(@config_file)
    @config_file.close
  end
  @raw_db_config
end

# File lib/event_store.rb, line 44
def self.redis(hostname)
  hash = Zlib::crc32(hostname)
  @redis[hash % @redis.length]
end

# File lib/event_store.rb, line 53
def self.redis_connect(config_hash)
  if config_hash["hosts"]
    generic_config = config_hash.reject { |k, _| k == "hosts" }
    @redis = config_hash["hosts"].map { |hostname|
      Redis.new(generic_config.merge("host" => hostname))
    }
  else
    @redis ||= [Redis.new(config_hash)]
  end
end

# File lib/event_store.rb, line 68
def self.schema
  @schema ||= raw_db_config[@environment][@adapter]['schema']
end

# File lib/event_store.rb, line 87
def self.table_name
  @table_name ||= raw_db_config['table_name']
end

# File lib/event_store.rb, line 147
def self.unescape_bytea(binary_string)
  binary_string = [binary_string[1..-1]].pack("H*") if binary_string[0] == "x"
  [binary_string].pack("H*")
end

# File lib/event_store.rb, line 167
def self.use_names_table?
  @use_names_table
end

# File lib/event_store.rb, line 133
def self.vertica(environment = 'test', table_name = 'events', schema = 'event_store_test')
  @schema         = schema
  @table_name     = table_name
  @environment    = environment.to_s
  @adapter        = 'vertica'
  @db_config      ||= self.db_config
  @db_config['host'] ||= ENV['VERTICA_HOST'] || vertica_host
  custom_config(@db_config, local_redis_config, @table_name, environment)
end

# File lib/event_store.rb, line 186
def self.vertica_host
  File.read File.expand_path(File.join('..','..','db', 'vertica_host_address.txt'), __FILE__)
end