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 = '15-4.2.3'.freeze

LIB_PG_QUERY_SHA256SUM = '8b820d63442b1677ce4f0df2a95b3fafdbc520a82901def81217559ec4df9e6b'.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/* #{extdir}/")

system("rmdir #{extdir}/postgres")

system("cp -a #{libdir}/pg_query.h #{extdir}/include")

# Make sure every .c file in the top-level directory is its own translation unit

system("mv #{extdir}/*{_conds,_defs,_helper}.c #{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