From 36761f0185a18fe5e32ef1c0b46938365d6d15dd Mon Sep 17 00:00:00 2001
From: Justin Edmund <justin@jedmund.com>
Date: Thu, 2 Feb 2023 02:27:16 -0800
Subject: [PATCH] Update download tasks

* Renames old `download_images` task to `download_all_images`
* Creates a new task for downloading images for a specific list of granblue IDs
---
 lib/tasks/download_all_images.rake |  41 +++++++++
 lib/tasks/download_images.rake     | 131 ++++++++++++++++++++++++-----
 2 files changed, 152 insertions(+), 20 deletions(-)
 create mode 100644 lib/tasks/download_all_images.rake

diff --git a/lib/tasks/download_all_images.rake b/lib/tasks/download_all_images.rake
new file mode 100644
index 0000000..524449e
--- /dev/null
+++ b/lib/tasks/download_all_images.rake
@@ -0,0 +1,41 @@
+namespace :granblue do
+  def _progress_reporter(count:, total:, result:, bar_len: 40, multi: true)
+    filled_len = (bar_len * count / total).round
+    status = File.basename(result)
+    percents = (100.0 * count / total).round(1)
+    bar = '=' * filled_len + '-' * (bar_len - filled_len)
+
+    if !multi
+      print("[#{bar}] #{percents}% ...#{' ' * 14}#{status}\n")
+    else
+      print "\n"
+    end
+  end
+
+  desc 'Downloads images for the given object type at the given size'
+  task :download_all_images, %i[object size] => :environment do |_t, args|
+    require 'open-uri'
+
+    filename = "export/#{args[:object]}-#{args[:size]}.txt"
+    count = `wc -l #{filename}`.split.first.to_i
+
+    path = "#{Rails.root}/download/#{args[:object]}-#{args[:size]}"
+    FileUtils.mkdir_p(path) unless Dir.exist?(path)
+
+    puts "Downloading #{count} images from #{args[:object]}-#{args[:size]}.txt..."
+    if File.exist?(filename)
+      File.readlines(filename).each_with_index do |line, i|
+        download = URI.parse(line.strip).open
+        download_URI = "#{path}/#{download.base_uri.to_s.split('/')[-1]}"
+        if File.exist?(download_URI)
+          puts "Skipping #{line}"
+        else
+          IO.copy_stream(download, "#{path}/#{download.base_uri.to_s.split('/')[-1]}")
+          _progress_reporter(count: i, total: count, result: download_URI, bar_len: 40, multi: false)
+        end
+      rescue StandardError => e
+        puts "#{e}: #{line}"
+      end
+    end
+  end
+end
diff --git a/lib/tasks/download_images.rake b/lib/tasks/download_images.rake
index 934c296..ec61cb2 100644
--- a/lib/tasks/download_images.rake
+++ b/lib/tasks/download_images.rake
@@ -12,29 +12,120 @@ namespace :granblue do
     end
   end
 
-  desc 'Exports a list of character URLs for a given size'
-  task :download_images, %i[object size] => :environment do |_t, args|
-    require 'open-uri'
+  def build_weapon_url(id, size)
+    # Set up URL
+    base_url = 'http://gbf.game-a.mbga.jp/assets/img/sp/assets/weapon'
+    extension = '.jpg'
 
-    filename = "export/#{args[:object]}-#{args[:size]}.txt"
-    count = `wc -l #{filename}`.split.first.to_i
+    directory = 'ls' if size.to_s == 'main'
+    directory = 'm' if size.to_s == 'grid'
+    directory = 's' if size.to_s == 'square'
 
-    path = "#{Rails.root}/download/#{args[:object]}-#{args[:size]}"
-    FileUtils.mkdir_p(path) unless Dir.exist?(path)
+    "#{base_url}/#{directory}/#{id}#{extension}"
+  end
 
-    puts "Downloading #{count} images from #{args[:object]}-#{args[:size]}.txt..."
-    if File.exist?(filename)
-      File.readlines(filename).each_with_index do |line, i|
-        download = URI.parse(line.strip).open
-        download_URI = "#{path}/#{download.base_uri.to_s.split('/')[-1]}"
-        if File.exist?(download_URI)
-          puts "Skipping #{line}"
-        else
-          IO.copy_stream(download, "#{path}/#{download.base_uri.to_s.split('/')[-1]}")
-          _progress_reporter(count: i, total: count, result: download_URI, bar_len: 40, multi: false)
-        end
-      rescue StandardError => e
-        puts "#{e}: #{line}"
+  def build_summon_url(id, size)
+    # Set up URL
+    base_url = 'http://gbf.game-a.mbga.jp/assets/img/sp/assets/summon'
+    extension = '.jpg'
+
+    directory = 'party_main' if size.to_s == 'main'
+    directory = 'party_sub' if size.to_s == 'grid'
+    directory = 's' if size.to_s == 'square'
+
+    "#{base_url}/#{directory}/#{id}#{extension}"
+  end
+
+  def build_chara_url(id, size)
+    # Set up URL
+    base_url = 'http://gbf.game-a.mbga.jp/assets/img/sp/assets/npc'
+    extension = '.jpg'
+
+    directory = 'f' if size.to_s == 'main'
+    directory = 'm' if size.to_s == 'grid'
+    directory = 's' if size.to_s == 'square'
+
+    "#{base_url}/#{directory}/#{id}#{extension}"
+  end
+
+  def download_images(url, size, path)
+    begin
+      download = URI.parse(url).open
+      download_URI = "#{path}/#{download.base_uri.to_s.split('/')[-1]}"
+      if File.exist?(download_URI)
+        puts "\tSkipping #{size}\t#{url}"
+      else
+        puts "\tDownloading #{size}\t#{url}..."
+        IO.copy_stream(download, "#{path}/#{download.base_uri.to_s.split('/')[-1]}")
+      end
+    rescue OpenURI::HTTPError
+      puts "\t404 returned\t#{url}"
+    end
+  end
+
+  def download_chara_images(id)
+    sizes = %w[main grid square]
+
+    url = {
+      'main': build_chara_url(id, 'main'),
+      'grid': build_chara_url(id, 'grid'),
+      'square': build_chara_url(id, 'square')
+    }
+
+    puts "Character #{id}"
+    sizes.each do |size|
+      path = "#{Rails.root}/download/character-#{size}"
+      download_images(url[size.to_sym], size, path)
+    end
+  end
+
+  def download_weapon_images(id)
+    sizes = %w[main grid square]
+
+    url = {
+      'main': build_weapon_url(id, 'main'),
+      'grid': build_weapon_url(id, 'grid'),
+      'square': build_weapon_url(id, 'square')
+    }
+
+    puts "Weapon #{id}"
+    sizes.each do |size|
+      path = "#{Rails.root}/download/weapon-#{size}"
+      download_images(url[size.to_sym], size, path)
+    end
+  end
+
+  def download_summon_images(id)
+    sizes = %w[main grid square]
+
+    url = {
+      'main': build_summon_url(id, 'main'),
+      'grid': build_summon_url(id, 'grid'),
+      'square': build_summon_url(id, 'square')
+    }
+
+    puts "Summon #{id}"
+    sizes.each do |size|
+      path = "#{Rails.root}/download/character-#{size}"
+      download_images(url[size.to_sym], size, path)
+    end
+  end
+
+  desc 'Downloads images for the given Granblue_IDs'
+  task :download_images, %i[object] => :environment do |_t, args|
+    object = args[:object]
+    list = args.extras
+
+    list.each do |id|
+      if object == 'character'
+        download_chara_images("#{id}_01")
+        download_chara_images("#{id}_02")
+        download_chara_images("#{id}_03")
+        download_chara_images("#{id}_04")
+      elsif object == 'weapon'
+        download_weapon_images(id)
+      elsif object == 'summon'
+        download_summon_images(id)
       end
     end
   end