#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

begin
  require 'dbt'
rescue LoadError
  # Ignored
end

if Object.const_defined?('Dbt')
  module BuildrPlus
    module DbtExtension
      module ProjectExtension
        include Extension

        after_define do |project|
          # Make sure all the data sources in the configuration file are mapped to idea project
          Dbt::Buildr.add_idea_data_sources_from_configuration_file(project) if project.ipr?
        end
      end
    end
  end

  class Buildr::Project
    include BuildrPlus::DbtExtension::ProjectExtension
  end
end

#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

begin
  require 'dbt'
rescue LoadError
  # Ignored
end

if Object.const_defined?('Dbt')
  module BuildrPlus
    module DbtExtension
      module ProjectExtension
        include Extension

        first_time do
          Dbt::Config.driver = 'postgres' if BuildrPlus::DbConfig.pgsql?
        end

        after_define do |project|
          # Make sure all the data sources in the configuration file are mapped to idea project
          Dbt::Buildr.add_idea_data_sources_from_configuration_file(project) if project.ipr?
        end
      end
    end
  end

  class Buildr::Project
    include BuildrPlus::DbtExtension::ProjectExtension
  end
end

Make sure DbtDriver is set of postgres build
Make sure DbtDriver is set of postgres build
