#!/usr/bin/env ruby
#
# Copyright © 2018-2022 Nick Bowler
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

preprocess do
    system("git", "submodule", "update", "--init")

    # Remove dead annex keys from processing
    @items.delete_if do |item|
        l = File.readlink(item.raw_filename)
        true if !File.exists?(item.raw_filename) and l =~ %r{/annex/objects/}
    rescue Errno::EINVAL
    end

    commit = nil
    Open3.popen2("git", "rev-parse", "HEAD") do |stdin, stdout, result|
        stdin.close
        val = stdout.gets
        stdout.close
        commit = val.chomp if result.value.success?
    end

    if commit then
        @items.each do |item|
            item[:gitrev] = commit if item_source(item)
        end
    end

    @items.find_all('/weblog/*.md').each do |item|
        item[:kind] ||= 'article'
    end

    @items.each do |item|
        item[:created_at] ||=
            item[:published] || File.stat(item.raw_filename).mtime
        item[:updated_at] ||=
            item[:updated] || File.stat(item.raw_filename).mtime
    end

    archivedirs = {}
    @items.find_all('/archive/**/*').each do |item|
        dir = File.dirname(item.identifier)
        while dir != "/"
            archivedirs[dir] = true
            dir = File.dirname(dir)
        end
    end

    archivedirs.keys.sort_by{ |s| -s.length }.each do |dir|
        attrs = {
            pattern: "#{dir}/*{,/index.lst}",
            title: "Index of #{dir}",
            gitrev: commit,
        }

        @items.create("", attrs, "#{dir}/index.lst")
    end

    create_wkd_items(@items["/pubring.gpg"])
    create_hkp_items(@items["/pubring.gpg"])
end

postprocess do
    # Gzip all text items for nginx http_static
    gzip_site = !ENV['GZIP_SITE'].to_s.empty?
    if gzip_site
        reps = @items.flat_map(&:reps).each do |rep|
            next if rep.binary?
            rep.instance_variable_get(:@item_rep).raw_paths[:last].each do |f|
                if system("gzip", "-nck9", f, [:out]=>["#{f}.gz.tmp", "wb"])
                    system("touch", "-r", f, "#{f}.gz.tmp")
                    File.rename("#{f}.gz.tmp", "#{f}.gz")
                else
                    File.unlink("#{f}.gz.tmp")
                end
            end
        end
    end

    # Register URLs for git-annex keys
    unless (uribase = ENV['ANNEX_URI_BASE'].to_s.chomp("/")).empty?
        Open3.popen2("git", "-c", "core.bare=false", "annex", "registerurl") do
        |stdin, stdout, result|
            @items.each do |item|
                next if item.raw_filename.nil?

                l = File.readlink(item.raw_filename)
                next unless l =~ %r{/annex/objects/}

                key = File.basename(l)

                # Find output reps corresponding to this key, if any
                item.reps.each do |rep|
                    next if rep.raw_path.nil?
                    next unless
                        FileUtils.identical?(item.raw_filename, rep.raw_path)

                    loop do
                        STDOUT.write(stdout.read_nonblock(100))
                    rescue EOFError, IO::WaitReadable
                        break
                    end

                    stdin.printf("%s %s%s\n", key, uribase, rep.path)
                end
            rescue Errno::EINVAL
            end

            stdin.close
            loop do
                STDOUT.write(stdout.readpartial(100))
            rescue EOFError
                break
            end

            unless (rc = result.value).success?
                printf("git annex registerurl failed: %s\n", rc.to_s)
            end
        end
    end
end

compile '/**/index.lst' do
    layout '/listing.erb'
    layout '/default.xml'
    layout '/default.xsl'
    layout '/embed-svg.xsl'
    filter :relativize_paths, type: :xml
    filter :xhtml_compat, fix_doctype: true
    write item.identifier.without_ext + ".xhtml"
end

compile '/license/gpl*.md' do
    filter :kramdown, auto_ids: false, header_offset: -1
    layout '/default.xml'
    layout '/gpl.xsl'
    layout '/default.xsl'
    filter :relativize_paths, type: :xml
    filter :xhtml_compat
    write to_xhtml
end

compile '/**/*.md' do
    filter :erb
    filter :kramdown, auto_ids: false, header_offset: 1
    snapshot :rawbody
    layout '/default.xml'
    layout '/default.xsl'
    filter :relativize_paths, type: :xml
    filter :xhtml_compat
    write to_xhtml
end

compile '/license/cc*.sgml' do
    filter :sgml2xml
    layout '/creativecommons.xsl'
    layout '/default.xml'
    layout '/default.xsl', "section-links": "yes"
    filter :relativize_paths, type: :xml
    filter :xhtml_compat
    write to_xhtml
end

compile '/license/cc*.xhtml' do
    layout '/creativecommons.xsl'
    layout '/default.xml'
    layout '/default.xsl', "section-links": "yes"
    filter :relativize_paths, type: :xml
    filter :xhtml_compat
    write to_xhtml
end

compile '/images/*.jpg', rep: :large do
    w, h = FastImage.size(item.raw_filename)
    filename = item.identifier.without_ext + '-t1200.' + item.identifier.ext
    filter :imgresize, width: [w, 1200].min, height: [h, 1200].min, cache: filename
    write filename
end

compile '/images/*.jpg', rep: :medium do
    w, h = FastImage.size(item.raw_filename)
    filename = item.identifier.without_ext + '-t800.' + item.identifier.ext
    if w > 900 or h > 900
        filter :imgresize, width: 800, height: 800, cache: filename
        write filename
    end
end

compile '/images/*.jpg', rep: :info do
    filter :imginfo
    layout '/imginfo.xsl'
    layout '/default.xml'
    layout '/default.xsl'
    filter :relativize_paths, type: :xml
    filter :xhtml_compat
    write to_xhtml
end

compile '/**/*.scss' do
    filter :sass, syntax: :scss, load_paths: ["."]
    filter :css_source, uribase: \
        "https://git.draconx.ca/gitweb/homepage.git/blob/" +
        @item[:gitrev] + ":"
    filter :css_clean_selectors, \
        preserve_comments: true, \
        preserve_hacks: true
    snapshot :before_darkmode
    filter :css_darkmode
    write @item.identifier.without_ext + '.css'
end

compile '/style.scss', rep: :dark do
    filter :compiled_content, snapshot: :before_darkmode
    filter :css_darkmode, alternate: true
    write "/dark.css"
end

compile '/gpg/*' do
    filter :wkd_export_armor
    write "/pubring/#{@item.identifier.components.last}.asc"
    write "/pubring/#{@item[:id64]}.asc" if @item[:id64]
    write "/pubring/#{@item[:id32]}.asc" if @item[:id32]
end

compile '/gpg/*', rep: :hu do
    if @item[:wkd_hash]
        filter :wkd_export
        write "/pubring/wkd/" + @item[:wkd_hash]
    end
end
ignore '/*.gpg'

compile '/**/*.svg' do
    filter :scour, comment_stripping: true
    write @item.identifier.to_s
end

compile '/icons/**/*.svg', rep: :icon32 do
    filter :svg2png, width: 32, height: 32
    write @item.identifier.without_ext + "-32.png"
end

compile '/**/*' do
    filter :copybin if @item.binary?
    write @item.identifier.to_s
end

layout '/**/*.xsl', :xsl
layout '/**/*', :erb