require 'msfdb_helpers/db_interface'
module MsfdbHelpers
  class Standalone < DbInterface

    def initialize(options:, db_conf:, connection_string:)
      @options = options
      @db_conf = db_conf
      begin
        @conn = PG.connect(connection_string)
      rescue PG::ConnectionBad
        print_error('Could not connect to standalone PostgreSQL instance. Ensure that the connection string is valid, and that the database is accessible')
        raise
      end

      conninfo = @conn.conninfo_hash
      @options[:db_port] = conninfo[:port]
      @options[:db_host] = conninfo[:host]
      super(options)
    end

    def init(msf_pass, msftest_pass)
      create_db_users(msf_pass, msftest_pass)
    end

    def delete
      @conn.exec("DROP DATABASE IF EXISTS #{@options[:msf_db_name]};")
      @conn.exec("DROP DATABASE IF EXISTS #{@options[:msftest_db_name]};")
      @conn.exec("DROP USER IF EXISTS #{@options[:msf_db_user]};")
      @conn.exec("DROP USER IF EXISTS #{@options[:msftest_db_user]};")
      FileUtils.rm_r(@db_conf, force: true)
    end

    def start
      true
    end

    def stop
      puts 'A standalone database cannot be stopped by msfdb'
      false
    end

    def restart
      raise NotImplementedError
    end

    def exists?
      !@conn.nil?
    end

    def status
      # Search for the database name
      is_initialized = @conn.exec_params('select * from pg_catalog.pg_database where datname = $1', [@options[:msf_db_name]]).any?
      if !is_initialized
        DatabaseStatus::NEEDS_INIT
      else
        DatabaseStatus::RUNNING
      end
    end

    def write_db_client_auth_config
      raise NotImplementedError
    end

    def self.requirements
      []
    end

    private

    def create_db_users(msf_pass, msftest_pass)
      @conn.exec("create user #{@options[:msf_db_user]} with password '#{msf_pass}'")
      @conn.exec("create user #{@options[:msftest_db_user]} with password '#{msftest_pass}'")
      @conn.exec("alter role #{@options[:msf_db_user]} createdb")
      @conn.exec("alter role #{@options[:msftest_db_user]} createdb")
      @conn.exec("alter role #{@options[:msf_db_user]} with password '#{msf_pass}'")
      @conn.exec("alter role #{@options[:msftest_db_user]} with password '#{msftest_pass}'")
      @conn.exec("CREATE DATABASE #{@options[:msf_db_name]}")
      @conn.exec("CREATE DATABASE #{@options[:msftest_db_name]}")

      # In PostgreSQL 15+, users need explicit permission to create objects in the public schema.
      # Grant all privileges on the public schema to the database users.
      conninfo = @conn.conninfo_hash
      [@options[:msf_db_name], @options[:msftest_db_name]].each_with_index do |db_name, idx|
        db_user = idx == 0 ? @options[:msf_db_user] : @options[:msftest_db_user]
        db_conn = PG.connect(host: conninfo[:host], port: conninfo[:port], dbname: db_name,
                             user: conninfo[:user], password: conninfo[:password])
        db_conn.exec("GRANT ALL ON SCHEMA public TO #{db_user}")
        db_conn.finish
      end

      @conn.finish
    end
  end
end