X-Git-Url: https://git.draconx.ca/gitweb/homepage.git/blobdiff_plain/de605446b3ed408672aecd74462a25de41c1367c..ec1703ff7346b6d3edeee73f1c6dfa750b7aeed4:/lib/helpers.rb

diff --git a/lib/helpers.rb b/lib/helpers.rb
index d2f3a7c..df48f4c 100644
--- a/lib/helpers.rb
+++ b/lib/helpers.rb
@@ -18,10 +18,12 @@
 require 'nokogiri'
 
 use_helper Nanoc::Helpers::Breadcrumbs
+use_helper Nanoc::Helpers::Blogging
 
 Xmlns = {
     'xhtml' => 'http://www.w3.org/1999/xhtml'
 }.freeze
+$counters = {}
 
 def to_xhtml(subpath = "", item = @item)
     if item.identifier =~ '/index.*'
@@ -35,12 +37,10 @@ end
 
 def item_source(item = @item)
     filename = "content" + item.identifier
+    filebase = filename.chomp(File.extname(filename))
 
-    if File.file? filename then
-        return filename
-    end
-
-    return nil
+    [ if item.binary? then filebase + ".yaml" end,
+      filename].compact.find { |f| File.file? f }
 end
 
 def item_uri(item = @item, rep: :default)
@@ -63,3 +63,24 @@ def item_longdesc(item)
     p = xml.xpath('//xhtml:p', Xmlns)
     if p.empty? then nil else p[0].xpath('string(.)') end
 end
+
+def counter(name = :default, item = @item)
+    $counters[item] ||= {}
+    $counters[item][name] ||= 0
+
+    name.to_s.capitalize + " " + ($counters[item][name] += 1).to_s
+end
+
+def human_filesize(size)
+    units = ["B", "KiB", "MiB", "GiB"]
+
+    for unit in units
+        if (size < 1024)
+            break
+        end
+
+        size /= 1024.0
+    end
+
+    sprintf("%.1f %s", size + 0.05, unit)
+end