From 26f0d61bb1924c42f65cfd817d45838f7057a5ed Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Sat, 21 Nov 2015 16:01:49 +0900
Subject: [PATCH] Add download button for mobile

---
 assets/index.html    | 1 +
 assets/index.ja.html | 1 +
 assets/index.ru.html | 1 +
 web.lua              | 8 +++++++-
 4 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/assets/index.html b/assets/index.html
index 8cde0f5..59701d1 100644
--- a/assets/index.html
+++ b/assets/index.html
@@ -48,6 +48,7 @@
         <label><input type="radio" name="scale" value="2"> 2x</label>
       </fieldset>
       <input type="submit"/>
+      <input type="submit" name="download" value="Download"/>
     </form>
     <div class="help">
       <ul class="padding-left">
diff --git a/assets/index.ja.html b/assets/index.ja.html
index 5936d14..fcc9612 100644
--- a/assets/index.ja.html
+++ b/assets/index.ja.html
@@ -45,6 +45,7 @@
         <label><input type="radio" name="scale" value="2"> 2x</label>
       </fieldset>
       <input type="submit" value="実行"/>
+      <input type="submit" name="download" value="ダウンロード"/>
     </form>
     <div class="help">
       <ul class="padding-left">
diff --git a/assets/index.ru.html b/assets/index.ru.html
index 7d45890..dfde011 100644
--- a/assets/index.ru.html
+++ b/assets/index.ru.html
@@ -47,6 +47,7 @@
         <label><input type="radio" name="scale" value="2"> 2x</label>
       </fieldset>
       <input type="submit"/>
+      <input type="submit" name="download" value="Download"/>
     </form>
     <div class="help">
       <ul class="padding-left">
diff --git a/web.lua b/web.lua
index ccf7b7d..6e859e7 100644
--- a/web.lua
+++ b/web.lua
@@ -156,6 +156,8 @@ function APIHandler:post()
    local noise = tonumber(self:get_argument("noise", "0"))
    local white_noise = tonumber(self:get_argument("white_noise", "0"))
    local style = self:get_argument("style", "art")
+   local download = (self:get_argument("download", "")):len()
+
    if style ~= "art" then
       style = "photo" -- style must be art or photo
    end
@@ -189,8 +191,12 @@ function APIHandler:post()
       local name = uuid() .. ".png"
       local blob = image_loader.encode_png(x, alpha)
       self:set_header("Content-Disposition", string.format('filename="%s"', name))
-      self:set_header("Content-Type", "image/png")
       self:set_header("Content-Length", string.format("%d", #blob))
+      if download > 0 then
+	 self:set_header("Content-Type", "application/octet-stream")
+      else
+	 self:set_header("Content-Type", "image/png")
+      end
       self:write(blob)
    else
       if not x then