From be28a8e6249a911634ffecf6856585b3b9229d86 Mon Sep 17 00:00:00 2001
From: Toks <tttoksss@gmail.com>
Date: Fri, 13 Jun 2014 16:59:08 -0400
Subject: [PATCH] Fix Seiga sample/thumbnail rewriting

---
 app/logical/downloads/strategies/nico_seiga.rb | 7 ++++---
 app/logical/sources/strategies/nico_seiga.rb   | 2 ++
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/app/logical/downloads/strategies/nico_seiga.rb b/app/logical/downloads/strategies/nico_seiga.rb
index de55ad2d5..cf6f481df 100644
--- a/app/logical/downloads/strategies/nico_seiga.rb
+++ b/app/logical/downloads/strategies/nico_seiga.rb
@@ -30,9 +30,10 @@ module Downloads
       end
 
       def rewrite_thumbnails(url, headers)
-        if url =~ %r{/thumb/(\d+)}
-          id = $1
-          url = "http://seiga.nicovideo.jp/image/source?id=" + id
+        if url =~ %r{/thumb/\d+}
+          source = ::Sources::Strategies::NicoSeiga.new(url)
+          source.get
+          return [source.image_url, headers]
         end
 
         return [url, headers]
diff --git a/app/logical/sources/strategies/nico_seiga.rb b/app/logical/sources/strategies/nico_seiga.rb
index 32f5cad7c..696016599 100644
--- a/app/logical/sources/strategies/nico_seiga.rb
+++ b/app/logical/sources/strategies/nico_seiga.rb
@@ -89,6 +89,8 @@ module Sources
             "http://seiga.nicovideo.jp/seiga/im#{$1}"
           elsif url =~ %r{\Ahttp://lohas\.nicoseiga\.jp/priv/(\d+)}i
             "http://seiga.nicovideo.jp/seiga/im#{$1}"
+          elsif url =~ %r{\Ahttp://lohas\.nicoseiga\.jp//?thumb/(\d+)}i
+            "http://seiga.nicovideo.jp/seiga/im#{$1}"
           elsif url =~ %r{/seiga/im\d+}
             url
           else