require 'bundler/gem_tasks'
require 'rake/clean'
require 'rake/extensiontask'
require 'rspec/core/rake_task'
require 'rubocop/rake_task'
require 'open-uri'

LIB_PG_QUERY_TAG = '16-5.1.0'.freeze
LIB_PG_QUERY_SHA256SUM = '31f25b573da1c966bc762b0313b0a50cdd03aabdbaf666d90469eddcb1656df7'.freeze

Rake::ExtensionTask.new 'pg_query' do |ext|
  ext.lib_dir = 'lib/pg_query'
end

RSpec::Core::RakeTask.new
RuboCop::RakeTask.new

task spec: :compile

task default: %i[spec lint]
task test: :spec
task lint: :rubocop

CLEAN.include 'tmp/**/*'
CLEAN.include 'ext/pg_query/*.o'
CLEAN.include 'lib/pg_query/pg_query.bundle'

task :update_source do
  workdir = File.join(__dir__, 'tmp')
  libdir = File.join(workdir, 'libpg_query-' + LIB_PG_QUERY_TAG)
  filename = File.join(workdir, 'libpg_query-' + LIB_PG_QUERY_TAG + '.tar.gz')
  testfilesdir = File.join(__dir__, 'spec/files')
  extdir = File.join(__dir__, 'ext/pg_query')
  extbakdir = File.join(workdir, 'extbak')

  unless File.exist?(filename)
    system("mkdir -p #{workdir}")
    File.open(filename, 'wb') do |target_file|
      URI.open('https://codeload.github.com/pganalyze/libpg_query/tar.gz/' + LIB_PG_QUERY_TAG, 'rb') do |read_file|
        target_file.write(read_file.read)
      end
    end

    checksum = Digest::SHA256.hexdigest(File.read(filename))

    if checksum != LIB_PG_QUERY_SHA256SUM
      raise "SHA256 of #{filename} does not match: got #{checksum}, expected #{LIB_PG_QUERY_SHA256SUM}"
    end
  end

  unless Dir.exist?(libdir)
    system("tar -xzf #{filename} -C #{workdir}") || raise('ERROR')
  end

  # Backup important files from ext dir
  system("rm -fr #{extbakdir}")
  system("mkdir -p #{extbakdir}")
  system("cp -a #{extdir}/pg_query_ruby{.c,.sym,_freebsd.sym} #{extdir}/extconf.rb #{extbakdir}")

  FileUtils.rm_rf extdir

  # Reduce everything down to one directory
  system("mkdir -p #{extdir}")
  system("cp -a #{libdir}/src/* #{extdir}/")
  system("mv #{extdir}/postgres/include #{extdir}/include/postgres")
  system("mv #{extdir}/postgres/* #{extdir}/")
  system("rmdir #{extdir}/postgres")
  system("cp -a #{libdir}/pg_query.h #{extdir}/include")
  # Protobuf definitions
  system("protoc --proto_path=#{libdir}/protobuf --ruby_out=#{File.join(__dir__, 'lib/pg_query')} #{libdir}/protobuf/pg_query.proto")
  system("mkdir -p #{extdir}/include/protobuf")
  system("cp -a #{libdir}/protobuf/*.h #{extdir}/include/protobuf")
  system("cp -a #{libdir}/protobuf/*.c #{extdir}/")
  # Protobuf library code
  system("mkdir -p #{extdir}/include/protobuf-c")
  system("cp -a #{libdir}/vendor/protobuf-c/*.h #{extdir}/include")
  system("cp -a #{libdir}/vendor/protobuf-c/*.h #{extdir}/include/protobuf-c")
  system("cp -a #{libdir}/vendor/protobuf-c/*.c #{extdir}/")
  # xxhash library code
  system("mkdir -p #{extdir}/include/xxhash")
  system("cp -a #{libdir}/vendor/xxhash/*.h #{extdir}/include")
  system("cp -a #{libdir}/vendor/xxhash/*.h #{extdir}/include/xxhash")
  system("cp -a #{libdir}/vendor/xxhash/*.c #{extdir}/")
  # Other support files
  system("cp -a #{libdir}/testdata/* #{testfilesdir}")
  # Copy back the custom ext files
  system("cp -a #{extbakdir}/pg_query_ruby{.c,.sym,_freebsd.sym} #{extbakdir}/extconf.rb #{extdir}")
end