A downloads manager for Jekyll

# Title: Download tag for Jekyll
# Author: Brett Terpstra http://brettterpstra.com
# Description: Create updateable download cards for downloadable projects. Reads from a downloads.csv file in the root folder
#
# Example: {% download id %} to read download [id] from the CSV


module Jekyll
  class DownloadTag < Liquid::Tag
    require 'csv'
    @title = nil
    @file = ''
    @info = ''
    @icon = nil
    @template = 'link'

    def initialize(tag_name, markup, tokens)
      if markup =~ /^(\d+)( .*)?$/
        req_id = $1.strip
        @format = $2.strip.to_i unless $2.nil?
        CSV.read("downloads.csv").each do |line|
          if line[0] == req_id
            id, title, file, version, description, info, icon, updated = line
            @file = file
            @version = version == '' ? '' : %Q{ v#{version}}
            @title = %Q{#{title}#{@version}}
            @icon = icon == '' ? '' : %Q{<img src="#{icon}" width="100" height="100">}
            @updated = updated == '' ? '' : %Q{<p class="dl_updated">Updated #{updated.sub(/[\d:]+ -\d+ (\d+)/,"\\1").strip}.</p>}
            @description = description == '' ? '' : %Q{<p class="dl_description">#{description}</p>#{@updated}}
            @info = info == '' ? '' : %Q{<p class="dl_info"><a href="#{info}" title="More information on #{title}">More info&hellip;</a></p>}
            break
          end
        end
      end
      super
    end
# TODO: Make this read templates
# Example:
#     Dir.chdir(includes_dir) do
#       choices = Dir['**/*'].reject { |x| File.symlink?(x) }
#       if choices.include?(file)
#         source = File.read(file)
#         partial = Liquid::Template.parse(source)
#         context.stack do
#           rtn = rtn + partial.render(context)
#         end
#       else
#         rtn = rtn + "Included file '#{file}' not found in _includes directory"
#       end
#     end
    def render(context)
      output = super
      # TODO: Enable template selection
      if @title
        download = %Q{<div class="download"><h4>#{@title}</h4><p class="dl_icon"><a href="#{@file}" title="Download #{@title}">#{@icon}</a></p><div class="dl_body"><p class="dl_link"><a href="#{@file}">Download #{@title}</a></p>#{@description}#{@info}</div></div>}
      else
        "Error processing input, expected syntax: {% download title filename [url/to/related/post] %}"
      end
    end
  end
end

Liquid::Template.register_tag('download', Jekyll::DownloadTag)

desc "Find a download ID"
task :find_download, :term do |t, args|
  raise "### You haven't created a download csv yet." unless File.exists?('downloads.csv')
  results = CSV.read("downloads.csv").delete_if {|row|
    row[0].strip =~ /^\d+$/ && row[1] + " " + row[4] =~ /.*#{args.term}.*/i ? false : true
  }
  results.sort! { |a,b|
    a[0].to_i <=> b[0].to_i
  }.map! { |res|
    res[0] + ": " + res[1] + " v" + res[3]
  }
  results_menu(results,"download")
  print ("Select download")
  while line = Readline.readline(": ", true)
    if !line || line =~ /^[a-z]/i
      puts "## Canceled"
      Process.exit 0
    end
    line = line.to_i
    if (line > 0 && line <= results.length)
      id = results[line.to_i - 1].match(/^\d+/)[0]
      download_tag = "{% download #{id} %}"
      %x{echo "#{download_tag}\\c"| pbcopy}
      puts %Q{Download tag in clipboard: "#{download_tag}"}
      Process.exit 0
    else
      puts "## Selection out of range"
      Process.exit 0
    end
  end
end

# creates a user menu from a hash or array
def results_menu(res, type = "file")
  counter = 1
  puts
  res.reverse!
  res.each do |match|
    match = match.class == String ? match : match[:path]
    if type == "file"
      display = match.sub(/^.*?\/([^\/]+)$/,"\\1")
      display.gsub!(/^[\d-]+/,'')
      display.gsub!(/\.(md|markdown)$/,'')
      display.gsub!(/-/,' ')
    elsif type == "download"
      display = match.sub(/^\d+:\s*/,'')
    else
      display = match.strip
    end
    printf("%2d ) %s\n", counter, display)
    counter += 1
  end
  puts
end