module SetecAstronomy
  module KeePass
    class Database

      attr_reader :header, :groups, :entries

      def self.open(path)
        self.new(File.read(path))
      end

      def initialize(raw_db)
        @header = Header.new(raw_db[0..124])
        @encrypted_payload = raw_db[124..-1]
      end

      def entry(title)
        @entries.detect { |e| e.title == title }
      end

      def unlock(master_password)
        @final_key = header.final_key(master_password)
        decrypt_payload
        payload_io = StringIO.new(@payload)
        @groups = Group.extract_from_payload(header, payload_io)
        @entries = Entry.extract_from_payload(header, payload_io)
        true
      rescue OpenSSL::Cipher::CipherError
        false
      end

      def search(pattern)
        entries.select { |e| e.title =~ /#{pattern}/ }
      end

      def valid?
        @header.valid?
      end

      def decrypt_payload
        @payload = AESCrypt.decrypt(@encrypted_payload, @final_key, header.encryption_iv, 'AES-256-CBC')
      end
    end
  end
end