discourse/lib/temporary_db.rb

# frozen_string_literal: true

class TemporaryDb
  PG_TEMP_PATH = "/tmp/pg_schema_tmp"
  PG_CONF = "#{PG_TEMP_PATH}/postgresql.conf"
  PG_SOCK_PATH = "#{PG_TEMP_PATH}/sockets"

  def port_available?(port)
    TCPServer.open(port).close
    true
  rescue Errno::EADDRINUSE
    false
  end

  def pg_bin_path
    return @pg_bin_path if @pg_bin_path

    ["13", "12", "11", "10"].each do |v|
      bin_path = "/usr/lib/postgresql/#{v}/bin"
      if File.exist?("#{bin_path}/pg_ctl")
        @pg_bin_path = bin_path
        break
      end
    end
    if !@pg_bin_path
      bin_path = "/Applications/Postgres.app/Contents/Versions/latest/bin"
      if File.exist?("#{bin_path}/pg_ctl")
        @pg_bin_path = bin_path
      end
    end
    if !@pg_bin_path
      puts "Can not find postgres bin path"
      exit 1
    end
    @pg_bin_path
  end

  def initdb_path
    return @initdb_path if @initdb_path

    @initdb_path = `which initdb 2> /dev/null`.strip
    if @initdb_path.length == 0
      @initdb_path = "#{pg_bin_path}/initdb"
    end

    @initdb_path
  end

  def find_free_port(range)
    range.each do |port|
      return port if port_available?(port)
    end
  end

  def pg_port
    @pg_port ||= find_free_port(11000..11900)
  end

  def pg_ctl_path
    return @pg_ctl_path if @pg_ctl_path

    @pg_ctl_path = `which pg_ctl 2> /dev/null`.strip
    if @pg_ctl_path.length == 0
      @pg_ctl_path = "#{pg_bin_path}/pg_ctl"
    end

    @pg_ctl_path
  end

  def start
    FileUtils.rm_rf PG_TEMP_PATH
    `#{initdb_path} -D '#{PG_TEMP_PATH}' --auth-host=trust --locale=en_US.UTF-8 -E UTF8 2> /dev/null`

    FileUtils.mkdir PG_SOCK_PATH
    conf = File.read(PG_CONF)
    File.write(PG_CONF, conf + "\nport = #{pg_port}\nunix_socket_directories = '#{PG_SOCK_PATH}'")

    puts "Starting postgres on port: #{pg_port}"
    ENV['DISCOURSE_PG_PORT'] = pg_port.to_s

    Thread.new do
      `#{pg_ctl_path} -D '#{PG_TEMP_PATH}' start`
    end

    puts "Waiting for PG server to start..."
    while !`#{pg_ctl_path} -D '#{PG_TEMP_PATH}' status`.include?('server is running')
      sleep 0.1
    end
    @started = true

    `createuser -h localhost -p #{pg_port} -s -D -w discourse 2> /dev/null`
    `createdb -h localhost -p #{pg_port} discourse`

    puts "PG server is ready and DB is loaded"
  end

  def stop
    @started = false
    `#{pg_ctl_path} -D '#{PG_TEMP_PATH}' stop`
  end

  def with_env(&block)
    old_host = ENV["PGHOST"]
    old_user = ENV["PGUSER"]
    old_port = ENV["PGPORT"]
    old_dev_db = ENV["DISCOURSE_DEV_DB"]
    old_rails_db = ENV["RAILS_DB"]

    ENV["PGHOST"] = "localhost"
    ENV["PGUSER"] = "discourse"
    ENV["PGPORT"] = pg_port.to_s
    ENV["DISCOURSE_DEV_DB"] = "discourse"
    ENV["RAILS_DB"] = "discourse"

    yield
  ensure
    ENV["PGHOST"] = old_host
    ENV["PGUSER"] = old_user
    ENV["PGPORT"] = old_port
    ENV["DISCOURSE_DEV_DB"] = old_dev_db
    ENV["RAILS_DB"] = old_rails_db
  end

  def remove
    raise "Error: the database must be stopped before it can be removed" if @started
    FileUtils.rm_rf PG_TEMP_PATH
  end

  def migrate
    if !@started
      raise "Error: the database must be started before it can be migrated."
    end
    ActiveRecord::Base.establish_connection(
      adapter: 'postgresql',
      database: 'discourse',
      port: pg_port,
      host: 'localhost'
    )

    puts "Running migrations on blank database!"

    old_stdout = $stdout.clone
    old_stderr = $stderr.clone
    $stdout.reopen(File.new('/dev/null', 'w'))
    $stderr.reopen(File.new('/dev/null', 'w'))

    SeedFu.quiet = true
    Rake::Task["db:migrate"].invoke
  ensure
    $stdout.reopen(old_stdout) if old_stdout
    $stderr.reopen(old_stderr) if old_stderr
  end
end