142 lines
3.3 KiB
Ruby
Executable File
142 lines
3.3 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
|
|
|
require "base64"
|
|
require "digest/md5"
|
|
require "fileutils"
|
|
require "stringio"
|
|
require "zlib"
|
|
|
|
if File.read("lib/rscons/version.rb") =~ /VERSION = "(.+)"/
|
|
VERSION = $1
|
|
else
|
|
raise "Could not determine version."
|
|
end
|
|
PROG_NAME = "rscons"
|
|
START_FILE = "bin/#{PROG_NAME}"
|
|
LIB_DIR = "lib"
|
|
DIST = "dist"
|
|
|
|
files_processed = {}
|
|
combined_file = []
|
|
|
|
combine_files = lambda do |file|
|
|
File.read(file, mode: "rb").each_line do |line|
|
|
if line =~ /^\s*require(?:_relative)?\s*"(.*)"$/
|
|
require_name = $1
|
|
if require_name =~ %r{^#{PROG_NAME}(?:/.*)?$}
|
|
path = "#{LIB_DIR}/#{require_name}.rb"
|
|
if File.exist?(path)
|
|
unless files_processed[path]
|
|
files_processed[path] = true
|
|
combine_files[path]
|
|
end
|
|
else
|
|
raise "require path #{path.inspect} not found"
|
|
end
|
|
else
|
|
combined_file << line
|
|
end
|
|
else
|
|
combined_file << line
|
|
end
|
|
end
|
|
end
|
|
|
|
combine_files[START_FILE]
|
|
|
|
# Strip Ruby comment lines and empty lines to save some space, but do not
|
|
# remove lines that are in heredoc sections. This isn't terribly robust to be
|
|
# used in the wild, but works for the heredoc instances for this project.
|
|
stripped = []
|
|
heredoc_end = nil
|
|
combined_file.each do |line|
|
|
if line =~ /<<-?([A-Z]+)/
|
|
heredoc_end = $1
|
|
end
|
|
if heredoc_end and line =~ /^\s*#{heredoc_end}/
|
|
heredoc_end = nil
|
|
end
|
|
if line !~ /#\sspecs/
|
|
if heredoc_end or line !~ /^\s*(#[^!].*)?$/
|
|
stripped << line
|
|
end
|
|
end
|
|
end
|
|
|
|
license = File.read("LICENSE.txt").gsub(/^(.*?)$/) do |line|
|
|
if line.size > 0
|
|
"# #{line}"
|
|
else
|
|
"#"
|
|
end
|
|
end
|
|
|
|
compressed_script = Zlib::Deflate.deflate(stripped.join)
|
|
hash = Digest::MD5.hexdigest(compressed_script)
|
|
encoded_compressed_script = Base64.encode64(compressed_script).gsub("\n", "")
|
|
encoded_compressed_script_io = StringIO.new(encoded_compressed_script)
|
|
commented_encoded_compressed_script = ""
|
|
until encoded_compressed_script_io.eof?
|
|
line = encoded_compressed_script_io.read(64)
|
|
commented_encoded_compressed_script += "##{line}\n"
|
|
end
|
|
|
|
FileUtils.rm_rf(DIST)
|
|
FileUtils.mkdir_p(DIST)
|
|
File.open("#{DIST}/#{PROG_NAME}", "wb", 0755) do |fh|
|
|
fh.write(<<EOF)
|
|
#!/usr/bin/env ruby
|
|
|
|
#{license}
|
|
|
|
BASE64CHARS = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
|
|
|
def base64_decode(s)
|
|
out = ""
|
|
v = 0
|
|
bits = 0
|
|
s.each_char do |c|
|
|
if cv = BASE64CHARS.index(c)
|
|
v = (v << 6) | cv
|
|
bits += 6
|
|
elsif c == "="
|
|
break
|
|
end
|
|
if bits >= 8
|
|
out += (v >> (bits - 8)).chr
|
|
v &= 0xFFFFFFFF >> (32 - (bits - 8))
|
|
bits -= 8
|
|
end
|
|
end
|
|
out
|
|
end
|
|
|
|
script = File.join(File.dirname(__FILE__), ".rscons-#{VERSION}-#{hash}.rb")
|
|
unless File.exist?(script)
|
|
if File.read(__FILE__, mode: "rb") =~ /^#==>(.*)/m
|
|
require "zlib"
|
|
encoded_compressed = $1
|
|
compressed = base64_decode(encoded_compressed)
|
|
if ENV["rscons_dist_specs"]
|
|
require "digest/md5"
|
|
if Digest::MD5.hexdigest(compressed) != "#{hash}"
|
|
raise "Hash mismatch when decompressing rscons executable"
|
|
end
|
|
end
|
|
inflated = Zlib::Inflate.inflate(compressed)
|
|
File.open(script, "wb") do |fh|
|
|
fh.write(inflated)
|
|
end
|
|
else
|
|
raise "Error expanding rscons executable"
|
|
end
|
|
end
|
|
load script
|
|
if __FILE__ == $0
|
|
Rscons::Cli.new.run(ARGV)
|
|
end
|
|
#==>
|
|
#{commented_encoded_compressed_script}
|
|
EOF
|
|
end
|