From e8611aec412a626fa803fc88eefe08a84578ec00 Mon Sep 17 00:00:00 2001
From: Tobias Schmidt <tobidt@gmail.com>
Date: Sat, 15 Apr 2017 00:07:48 -0300
Subject: [PATCH] Render tarball sha256 checksum if available for release

---
 content/css/docs.css    |  5 +++--
 content/download.html   | 19 ++++++----------
 lib/helpers/download.rb | 50 ++++++++++++++++++++++++++++++++++++-----
 3 files changed, 55 insertions(+), 19 deletions(-)

diff --git a/content/css/docs.css b/content/css/docs.css
index 85f16bb8..5304a7f7 100644
--- a/content/css/docs.css
+++ b/content/css/docs.css
@@ -233,8 +233,9 @@ footer {
 }
 
 .downloads .checksum {
-  color: #aaa;
-  font-style: italic;
+  font-family: monospace;
+  font-size: .6em;
+  vertical-align: bottom;
 }
 
 /* Docs-related styles. */
diff --git a/content/download.html b/content/download.html
index 8da12a8e..14b16461 100644
--- a/content/download.html
+++ b/content/download.html
@@ -29,11 +29,6 @@ title: Download
       the <a href="/docs/introduction/getting_started/">installation instructions</a>.
     </p>
 
-    <div class="alert alert-info" role="alert">
-      <strong>Work in progress!</strong>
-      We will provide more precompiled binary versions as well as checksums soon.
-    </div>
-
     <div class="panel panel-default download-selection">
       <div class="panel-body">
         Operating system <%= dropdown(:os, Downloads.operating_systems, :popular, popular: %w(darwin linux windows)) %>
@@ -64,13 +59,13 @@ title: Download
         </tr>
       </thead>
       <tbody>
-      <% release.assets.each do |asset| %>
-        <tr data-os="<%= asset.os %>" data-arch="<%= asset.arch %>">
-          <td class="filename"><a class="download" href="<%= asset.url %>"><%= asset.name %></a></td>
-          <td><%= asset.os %></td>
-          <td><%= asset.arch %></td>
-          <td><%= format_bytes asset.size %></td>
-          <td class="checksum">not available yet</td>
+      <% release.binaries.each do |binary| %>
+        <tr data-os="<%= binary.os %>" data-arch="<%= binary.arch %>">
+          <td class="filename"><a class="download" href="<%= binary.url %>"><%= binary.name %></a></td>
+          <td><%= binary.os %></td>
+          <td><%= binary.arch %></td>
+          <td><%= format_bytes binary.size %></td>
+          <td class="checksum"><%= Downloads.checksum(release, binary.name) %></td>
         </tr>
       <% end %>
       </tbody>
diff --git a/lib/helpers/download.rb b/lib/helpers/download.rb
index 9e7601f6..69d6128a 100644
--- a/lib/helpers/download.rb
+++ b/lib/helpers/download.rb
@@ -1,9 +1,11 @@
+require 'fileutils'
 require 'json'
+require 'open-uri'
 
 module Downloads
   # repositories returns a list of all repositories with releases.
   def self.repositories
-    @repositories ||= begin
+    @_repositories ||= begin
       repos = Dir.glob('downloads/*').map { |dir| Repository.new(dir) }
       repos.sort_by { |r| r.name == 'prometheus' ? '0' : r.name }
     end
@@ -13,7 +15,7 @@ module Downloads
   # provided for.
   def self.operating_systems
     repositories.inject([]) do |list, repo|
-      list += repo.releases.map { |r| r.assets.map(&:os) }.flatten
+      list += repo.releases.map { |r| r.binaries.map(&:os) }.flatten
     end.uniq.sort
   end
 
@@ -21,10 +23,38 @@ module Downloads
   # provided for.
   def self.architectures
     repositories.inject([]) do |list, repo|
-      list += repo.releases.map { |r| r.assets.map(&:arch) }.flatten
+      list += repo.releases.map { |r| r.binaries.map(&:arch) }.flatten
     end.uniq.sort
   end
 
+  # checksum returns the checksum for a given filename of a given release. It
+  # might try to download the sha256sums.txt from the given release if available
+  # and not already cached.
+  def self.checksum(release, name)
+    @_checksums ||= {}
+    @_checksums[release.id] ||= begin
+      asset = release.assets.find { |a| a['name'] == 'sha256sums.txt' }
+
+      if asset
+        cache = ['downloads', '.cache', release.id, 'sha256sums.txt'].join('/')
+        unless File.exists?(cache)
+          FileUtils.mkdir_p(File.dirname(cache))
+          File.open(cache, 'wb') do |file|
+            file.write(URI.parse(asset['browser_download_url']).read)
+          end
+        end
+
+        File.readlines(cache).each_with_object({}) do |line, memo|
+          checksum, filename = line.split(/\s+/)
+          memo[filename] = checksum
+        end
+      else
+        {}
+      end
+    end
+    @_checksums[release.id][name]
+  end
+
   class Repository
     def initialize(dir)
       @repo = JSON.parse(File.read(File.join(dir, 'repo.json')))
@@ -66,6 +96,10 @@ module Downloads
       @data = data
     end
 
+    def id
+      @data['id']
+    end
+
     def name
       @data['name']
     end
@@ -79,11 +113,17 @@ module Downloads
     end
 
     def assets
-      @data['assets'].map { |d| Asset.new(d) }
+      @data['assets']
+    end
+
+    def binaries
+      assets.
+        select { |d| d['name'] && %w[.tar.gz .zip].any? { |ext| d['name'].end_with?(ext) } }.
+        map { |d| Binary.new(d) }
     end
   end
 
-  class Asset
+  class Binary
     def initialize(data)
       @data = data
     end