From eb765e246332ea29ba5470f7f265ec84f1a396b3 Mon Sep 17 00:00:00 2001
From: Justin Edmund <justin@jedmund.com>
Date: Sat, 24 Dec 2022 14:16:45 -0800
Subject: [PATCH] Update tasks

---
 lib/tasks/export_character.rake | 20 +++++++++++---------
 lib/tasks/export_summon.rake    | 20 ++++++++++++++------
 lib/tasks/export_weapon.rake    | 14 ++++++++------
 3 files changed, 33 insertions(+), 21 deletions(-)

diff --git a/lib/tasks/export_character.rake b/lib/tasks/export_character.rake
index 41c5ef0..cdcd3e1 100644
--- a/lib/tasks/export_character.rake
+++ b/lib/tasks/export_character.rake
@@ -22,20 +22,22 @@ namespace :granblue do
 
       # Set up filepath
       dir = "#{Rails.root}/export/"
+      filename = "#{dir}/character-#{size}.txt"
       FileUtils.mkdir(dir) unless Dir.exist?(dir)
 
-      unless File.exist?("#{dir}/character-#{size}.txt")
-        File.open("#{dir}/character-#{size}.txt", 'w') do |f|
-          Character.all.each do |c|
-            f.write("#{build_chara_url("#{c.granblue_id}_01", size)} \n")
-            f.write("#{build_chara_url("#{c.granblue_id}_02", size)} \n")
-            f.write("#{build_chara_url("#{c.granblue_id}_03", size)} \n") if c.flb
-            f.write("#{build_chara_url("#{c.granblue_id}_04", size)} \n") if c.ulb
-          end
+      # Write to file
+      File.open(filename, 'w+') do |f|
+        Character.all.each do |c|
+          f.write("#{build_chara_url("#{c.granblue_id}_01", size)} \n")
+          f.write("#{build_chara_url("#{c.granblue_id}_02", size)} \n")
+          f.write("#{build_chara_url("#{c.granblue_id}_03", size)} \n") if c.flb
+          f.write("#{build_chara_url("#{c.granblue_id}_04", size)} \n") if c.ulb
         end
       end
 
-      puts "Wrote #{Character.count} character URLs for \"#{size}\" size"
+      # CLI Output
+      count = `wc -l #{filename}`.split.first.to_i
+      puts "Wrote #{count} character URLs for \"#{size}\" size"
     end
   end
 end
diff --git a/lib/tasks/export_summon.rake b/lib/tasks/export_summon.rake
index e5e1d0c..5942ab1 100644
--- a/lib/tasks/export_summon.rake
+++ b/lib/tasks/export_summon.rake
@@ -22,18 +22,26 @@ namespace :granblue do
 
       # Set up filepath
       dir = "#{Rails.root}/export/"
+      filename = "#{dir}/summon-#{size}.txt"
       FileUtils.mkdir(dir) unless Dir.exist?(dir)
 
-      unless File.exist?("#{dir}/summon-#{size}.txt")
-        File.open("#{dir}/summon-#{size}.txt", 'w') do |f|
-          Summon.all.each do |s|
-            f.write("#{build_summon_url("#{s.granblue_id}_01", size)} \n")
-            f.write("#{build_summon_url("#{s.granblue_id}_02", size)} \n") if (s.series == 3 || s.series == 0) && s.ulb
+      # Write to file
+      File.open(filename, 'w') do |f|
+        Summon.all.each do |s|
+          series = s.series.to_i
+          f.write("#{build_summon_url(s.granblue_id.to_s, size)} \n")
+
+          # Download second images only for Providence ULBs and Primal summons
+          if series == 3 || (series == 0 && s.ulb)
+            f.write("#{build_summon_url("#{s.granblue_id}_02",
+                                        size)} \n")
           end
         end
       end
 
-      puts "Wrote #{Summon.count} summon URLs for \"#{size}\" size"
+      # CLI output
+      count = `wc -l #{filename}`.split.first.to_i
+      puts "Wrote #{count} summon URLs for \"#{size}\" size"
     end
   end
 end
diff --git a/lib/tasks/export_weapon.rake b/lib/tasks/export_weapon.rake
index 8a6a808..9c4292e 100644
--- a/lib/tasks/export_weapon.rake
+++ b/lib/tasks/export_weapon.rake
@@ -22,17 +22,19 @@ namespace :granblue do
 
       # Set up filepath
       dir = "#{Rails.root}/export/"
+      filename = "#{dir}/weapon-#{size}.txt"
       FileUtils.mkdir(dir) unless Dir.exist?(dir)
 
-      unless File.exist?("#{dir}/weapon-#{size}.txt")
-        File.open("#{dir}/weapon-#{size}.txt", 'w') do |f|
-          Weapon.all.each do |w|
-            f.write("#{build_weapon_url(w.granblue_id.to_s, size)} \n")
-          end
+      # Write to file
+      File.open(filename, 'w') do |f|
+        Weapon.all.each do |w|
+          f.write("#{build_weapon_url(w.granblue_id.to_s, size)} \n")
         end
       end
 
-      puts "Wrote #{Weapon.count} weapon URLs for \"#{size}\" size"
+      # CLI output
+      count = `wc -l #{filename}`.split.first.to_i
+      puts "Wrote #{count} weapon URLs for \"#{size}\" size"
     end
   end
 end