Rakefile

# FreeBSD Ports
#
# Douglas Thrift
#
# Rakefile

require 'date'
require 'ostruct'
require 'pathname'
require 'set'

def in_portdir(&block)
  portdir = Pathname.new(Rake.original_dir)
  raise "#{portdir} is not a port directory" unless portdir.relative_path_from(@portsdir).to_s =~ %r{[^/]+/[^/]+}
  Dir.chdir(portdir) do
    block.call(portdir)
  end
end

def in_portstree(portstree, &block)
  portstree_zfs = `zfs get -H -o name mountpoint #{portstree.path}`.chomp
  sh "sudo zfs snapshot #{portstree_zfs}@clean"
  Dir.chdir(portstree.path) do
    block.call(portstree.path)
  end
ensure
  sh "sudo zfs rollback #{portstree_zfs}@clean"
  sh "sudo zfs destroy #{portstree_zfs}@clean"
end

def update_portstree_map(map = {})
  `poudriere ports -l -q`.each_line do |line|
    line.chomp!
    if line =~ /^([^ ]+) +([^ ]+) +([0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:[0-9]{2}) +([^ ]+)$/
      map[$1.to_sym] = OpenStruct.new(method: $2, timestamp: DateTime.parse($3), path: Pathname.new($4))
    else
      raise "unexpected line: #{line}"
    end
  end
  map
end

def clean_version(version)
  if version =~ /^(\d+\.\d+-RELEASE)(?:-p\d+)$/
    $1
  else
    version
  end
end

portstree_map = update_portstree_map
portsdir = @portsdir = Pathname.pwd
distdir = portsdir.join('distfiles')

task default: %i(update_githead)

desc 'Create or update the port checksum file (distinfo)'
task :makesum do
  in_portdir do
    sh "make makesum DEV_WARNING_WAIT=0 DISTDIR=#{distdir}"
  end
end

%w(md5 sha1 sha256).each do |sum|
  desc "Show the #{sum.upcase} sums of the distfiles"
  task sum do
    sh "mkdir -p #{distdir}" unless distdir.exist?
    files = distdir.glob('**/*').select {|dist| dist.file?}.map {|dist| dist.relative_path_from(portsdir)}
    sh "#{sum} #{files.join(' ')}" unless files.empty?
  end
end

desc 'Remove the ports distfiles'
task :distclean do
  in_portdir do
    sh "make distclean DISTDIR=#{distdir}"
  end
end

desc 'Verify the port using portlint'
task portlint: :update_githead do
  in_portdir do |portdir|
    if githead_portstree.path.join(portdir.relative_path_from(portsdir), 'Makefile').exist?
      sh "portlint -C"
    else
      sh "portlint -A"
    end
  end
end

desc 'Show the port differences'
task diff: :update_svnhead do |_, args|
  patch = args.extras.find {|arg| arg == 'patch'}
  in_portdir do |portdir|
    new_files = Set.new(`git ls-files`.each_line.map {|line| Pathname.new(line.chomp)})
    in_portstree(svnhead_portstree) do |path|
      rel_portdir = portdir.relative_path_from(portsdir)
      svnhead_portdir = path.join(rel_portdir)
      if svnhead_portdir.exist?
        old_files = Set.new(`svn ls -R #{svnhead_portdir}`.each_line.reject {|line| line =~ %r{/$}}.map {|line| Pathname.new(line.chomp)})
      else
        old_files = Set.new
      end
      sh "sudo mkdir -p #{svnhead_portdir}"
      Dir.chdir(svnhead_portdir) do
        new_files.each do |file|
          portdir_file = portdir.join(file)
          sh "sudo mkdir -p #{file.dirname}" unless file.dirname.exist?
          sh "sudo cp #{portdir_file} #{file}"
        end
        (new_files - old_files).each do |file|
          sh "sudo svn add --parents #{file}"
        end
        (old_files - new_files).each do |file|
          sh "sudo svn remove #{file}"
        end
        unless old_files.empty?
          if patch
            sh "svn diff > #{portdir.dirname.join("#{`make -V PKGNAME`.chomp}.diff")}"
          else
            sh "svn diff"
          end
        end
      end
      if old_files.empty?
        if patch
          sh "svn diff #{rel_portdir} > #{portsdir.join("#{portdir.basename}.diff")}"
        else
          sh "svn diff #{rel_portdir}"
        end
      end
    end
  end
end

desc 'Test the port with poudriere'
task :testport, %i(jailname flavor) => :update_githead do |_, args|
  this_version = clean_version(`uname -r`.chomp)
  this_arch = `uname -m`.chomp
  this_jailname = nil
  `poudriere jail -l -q`.each_line do |line|
    line.chomp!
    if line =~ /^([^ ]+) +([^ ]+) +([^ ]+) +([^ ]+).+$/
      jailname, version, arch, method = $1, $2, $3, $4
      if clean_version(version) == this_version && arch == this_arch && method == 'ftp'
        this_jailname = jailname
        break
      end
    else
      raise "unexpected line: #{line}"
    end
  end
  args.with_defaults(jailname: this_jailname)
  in_portdir do |portdir|
    files = `git ls-files`.each_line.map {|line| Pathname.new(line.chomp)}
    in_portstree(githead_portstree) do |path|
      rel_portdir = portdir.relative_path_from(portsdir)
      githead_portdir = path.join(rel_portdir)
      sh "sudo mkdir -p #{githead_portdir}"
      Dir.chdir(githead_portdir) do
        files.each do |file|
          portdir_file = portdir.join(file)
          sh "sudo mkdir -p #{file.dirname}" unless file.dirname.exist?
          sh "sudo cp #{portdir_file} #{file}"
        end
      end
      sh "sudo poudriere testport -j #{args.jailname} -o #{rel_portdir}#{args.flavor && "@#{args.flavor}"} -p githead"
    end
  end
end

{githead: 'git+https', svnhead: 'svn+https'}.each do |name, method|
  define_method("#{name}_portstree") do
    portstree_map[name]
  end

  desc "Create or update the #{name} portstree"
  task "update_#{name}" do |_, args|
    force = args.extras.find {|arg| arg == 'force'}
    portstree = portstree_map[name]
    if portstree
      raise "unexpected portstree method for #{name}: #{portstree[:method]} (expected: #{method})" if portstree[:method] != method
      if force || portstree.timestamp < Date.today
        sh "sudo poudriere ports -u -p #{name}"
      end
    else
      sh "sudo poudriere ports -c -m #{method} -p #{name}"
      update_portstree_map(portstree_map)
    end
  end
end