narou 3.2.5.1 → 3.3.0

data/lib/command/csv.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #

data/lib/command/diff.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #
@@ -78,6 +79,7 @@ module Command
     end
 
     def execute(argv)
+      disable_logging
       short_number_option_parse(argv)
       super
       @options["number"] ||= 1
@@ -88,7 +90,7 @@ module Command
         @novel_data = latest
       else
         target = argv.shift
-        @novel_data = Downloader.get_data_by_target(target)
+        @novel_data = Downloader.get_data_by_target(target) || {}
         id = @novel_data["id"]
       end
       unless id
@@ -134,13 +136,13 @@ module Command
 
     def clean_diff(id)
       cache_root_dir = Downloader.get_cache_root_dir(id)
-      print @novel_data["title"] + " の"
+      stream_io.print @novel_data["title"] + " の"
       unless File.exist?(cache_root_dir)
-        puts "差分はひとつもありません"
+        stream_io.puts "差分はひとつもありません"
         return
       end
       FileUtils.remove_entry_secure(cache_root_dir)
-      puts "差分を削除しました"
+      stream_io.puts "差分を削除しました"
     end
 
     def exec_difftool(id)
@@ -148,7 +150,7 @@ module Command
       diff_cmd = create_difftool_command_string(*temp_paths.map { |temp|
         path = temp.path
         if Helper.os_windows?
-          path.encode!(Encoding::Windows_31J)
+          path = path.encode(Encoding::Windows_31J)
         end
         %!"#{path}"!
       })
@@ -160,7 +162,7 @@ module Command
       ensure
         temp_paths.map(&:delete)
       end
-      puts res[0] unless res[0].empty?
+      stream_io.puts res[0] unless res[0].empty?
       error res[1] unless res[1].empty?
     end
 
@@ -195,14 +197,14 @@ module Command
         cache_dir = list ? list[nth - 1] : nil
       end
       unless cache_dir
-        puts "#{@novel_data["title"]} の差分データがありません"
+        stream_io.puts "#{@novel_data["title"]} の差分データがありません"
         return nil
       end
       cache_section_list = Dir.glob("#{cache_dir}/*.yaml").sort_by { |path|
         File.basename(path, ".yaml").split(" ", 2)[0].to_i
       }
       if cache_section_list.length == 0
-        puts "#{@novel_data["title"]} は最新話のみのアップデートのようです"
+        stream_io.puts "#{@novel_data["title"]} は最新話のみのアップデートのようです"
         return nil
       end
       novel_dir = Downloader.get_novel_section_save_dir(Downloader.get_novel_data_dir_by_target(id))
@@ -273,19 +275,19 @@ module Command
 
     def display_diff_list(id)
       list = get_diff_list(id)
-      print "#{list[:title]} の"
+      stream_io.print "#{list[:title]} の"
       if list[:list].empty?
-        puts "差分はひとつもありません"
+        stream_io.puts "差分はひとつもありません"
         return
       end
-      puts "差分一覧"
+      stream_io.puts "差分一覧"
       list[:list].each do |data|
-        puts "<bold><yellow>#{data[:version_string]}   -#{data[:number]}</yellow></bold>".termcolor
+        stream_io.puts "<bold><yellow>#{data[:version_string]}   -#{data[:number]}</yellow></bold>".termcolor
         data[:objects].each do |object|
-          puts "   第#{object[:index]}部分　#{object[:subtitle]}"
+          stream_io.puts "   第#{object[:index]}部分　#{object[:subtitle]}"
         end
         if data[:objects].empty?
-          puts "   (最新話のみのアップデート)"
+          stream_io.puts "   (最新話のみのアップデート)"
         end
       end
     end
@@ -296,7 +298,7 @@ module Command
         cache_root_dir = Downloader.get_cache_root_dir(id)
         next unless File.exist?(cache_root_dir)
         FileUtils.remove_entry_secure(cache_root_dir)
-        puts "#{data["title"]} の差分を削除しました"
+        stream_io.puts "#{data["title"]} の差分を削除しました"
       end
     end
 
@@ -306,8 +308,8 @@ module Command
     def display_diff_on_oneself(id)
       require_relative "../diffviewer"
       temp_paths = create_temp_files(id) or return
-      puts "#{@novel_data["title"]} の差分を表示します"
-      DiffViewer.new(*temp_paths).view
+      stream_io.puts "#{@novel_data["title"]} の差分を表示します"
+      stream_io.puts DiffViewer.new(*temp_paths).result
     ensure
       temp_paths.map(&:delete) if temp_paths
     end

data/lib/command/download.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #
@@ -11,6 +12,10 @@ module Command
   class Download < CommandBase
     SUPPORT_NOVEL_SITES = %w(小説家になろう(小説を読もう) ノクターンノベルズ ムーンライトノベルズ Arcadia ハーメルン 暁 カクヨム)
 
+    def self.oneline_help
+      "指定した小説をダウンロードします"
+    end
+
     def initialize
       super("[<target> <target2> ...] [options]")
       @opt.separator <<-EOS
@@ -138,8 +143,14 @@ module Command
           mistook_count += 1
           next
         end
-        unless @options["no-convert"]
-          convert_status = Convert.execute!(download_target)
+        if @options["no-convert"]
+          Narou.concurrency_call do
+            after_process(download_target)
+          end
+        else
+          convert_status = Convert.execute!(download_target) do
+            after_process(download_target)
+          end
           if convert_status > 0
             # 変換に失敗したか、中断された
             data = Downloader.get_data_by_target(download_target)   # 新規はDL後に取得しないとデータが存在しない
@@ -148,26 +159,26 @@ module Command
             raise Interrupt if convert_status == Narou::EXIT_INTERRUPT
           end
         end
-        if @options["mail"]
-          Mail.execute!(download_target)
-        end
-        if @options["freeze"]
-          Freeze.execute!(download_target)
-        elsif @options["remove"]
-          # --freeze オプションが指定された場合は --remove オプションは無視する
-          Remove.execute!(download_target, "-y")
-        end
       end
       exit mistook_count if mistook_count > 0
     rescue Interrupt
       puts "ダウンロードを中断しました"
+      Narou::Worker.cancel if Narou.concurrency_enabled?
       exit Narou::EXIT_INTERRUPT
     ensure
       Database.instance.save_database
     end
 
-    def self.oneline_help
-      "指定した小説をダウンロードします"
+    def after_process(target)
+      if @options["mail"]
+        Mail.execute!(target, io: $stdout2)
+      end
+      if @options["freeze"]
+        Freeze.execute!(target)
+      elsif @options["remove"]
+        # --freeze オプションが指定された場合は --remove オプションは無視する
+        Remove.execute!(target, "-y")
+      end
     end
   end
 end

data/lib/command/folder.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #
@@ -27,10 +28,7 @@ module Command
 
     def execute(argv)
       super
-      if argv.empty?
-        puts @opt.help
-        return
-      end
+      display_help! if argv.empty?
       tagname_to_ids(argv)
       argv.each do |target|
         dir = Downloader.get_novel_data_dir_by_target(target)

data/lib/command/freeze.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #
@@ -46,10 +47,7 @@ module Command
 
     def execute(argv)
       super
-      if argv.empty?
-        puts @opt.help
-        return
-      end
+      display_help! if argv.empty?
       tagname_to_ids(argv)
       frozen_list = Inventory.load("freeze")
       argv.each do |target|

data/lib/command/help.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #
@@ -11,7 +12,8 @@ module Command
       "このヘルプを表示します"
     end
 
-    def execute(argv)
+    def execute(_argv)
+      disable_logging
       if Narou.already_init?
         display_help
       else
@@ -20,23 +22,23 @@ module Command
     end
 
     def display_help
-      puts <<-EOS.termcolor
-#{HEADER}
+      stream_io.puts(<<~HELP.termcolor)
+        #{HEADER}
 
- <bold><green>Usage: narou &lt;command&gt; [arguments...] [options...]
-              [--no-color] [--multiple] [--time] [--backtrace]</green></bold>
+         <bold><green>Usage: narou &lt;command&gt; [arguments...] [options...]
+                      [--no-color] [--multiple] [--time] [--backtrace]</green></bold>
 
- コマンドの簡単な説明:
-      EOS
+         コマンドの簡単な説明:
+      HELP
       cmd_list = Command.get_list
       cmd_list.each do |key, command|
         oneline = command.oneline_help.split("\n")
-        puts "   <bold><green>#{key.ljust(12)}</green></bold> #{oneline.shift}".termcolor
+        stream_io.puts "   <bold><green>#{key.ljust(12)}</green></bold> #{oneline.shift}".termcolor
         oneline.each do |h|
-          puts " " * 16 + h
+          stream_io.puts " " * 16 + h
         end
       end
-      puts <<-EOS.termcolor
+      stream_io.puts(<<-HELP.termcolor)
 
   各コマンドの詳細は narou &lt;command&gt; -h を参照してください。
   各コマンドは先頭の一文字か二文字でも指定できます。
@@ -47,18 +49,18 @@ module Command
     --multiple   引数の区切りにスペースの他に","も使えるようにする
     --time       実行時間表示
     --backtrace  エラー発生時詳細情報を表示
-      EOS
+      HELP
     end
 
     def display_help_first_time
-      puts <<-EOS.termcolor
-#{HEADER}
+      stream_io.puts(<<~HELP.termcolor)
+        #{HEADER}
 
- <bold><green>Usage: narou init</green></bold>
+         <bold><green>Usage: narou init</green></bold>
 
-   まだこのフォルダは初期化されていません。
-   <bold><yellow>narou init</yellow></bold> コマンドを実行して初期化を行いましょう。
-      EOS
+           まだこのフォルダは初期化されていません。
+           <bold><yellow>narou init</yellow></bold> コマンドを実行して初期化を行いましょう。
+      HELP
     end
   end
 end

data/lib/command/init.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #
@@ -111,7 +112,7 @@ module Command
 
     def rewrite_aozoraepub3_files(aozora_path, line_height)
       # chuki_tag.txt の書き換え
-      custom_chuki_tag_path = File.join(Narou.get_preset_dir, "custom_chuki_tag.txt")
+      custom_chuki_tag_path = File.join(Narou.preset_dir, "custom_chuki_tag.txt")
       chuki_tag_path = File.join(aozora_path, "chuki_tag.txt")
       custom_chuki_tag = File.read(custom_chuki_tag_path, mode: "r:BOM|UTF-8")
       chuki_tag = File.read(chuki_tag_path, mode: "r:BOM|UTF-8")
@@ -130,7 +131,7 @@ module Command
       dst = ["AozoraEpub3.ini", "template/OPS/css_custom/vertical_font.css"]
       puts "(次のファイルをコピーor上書きしました)"
       src.size.times do |i|
-        src_full_path = File.join(Narou.get_preset_dir, src[i])
+        src_full_path = File.join(Narou.preset_dir, src[i])
         dst_full_path = File.join(aozora_path, dst[i])
         Helper.erb_copy(src_full_path, dst_full_path, binding)
         puts dst_full_path

data/lib/command/inspect.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #

data/lib/command/list.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #
@@ -132,8 +133,8 @@ module Command
                 when "nonfrozen"
                   !frozen
                 else
-                  error "不明なフィルターです(#{item})"
-                  warn "filters = #{filter_type_help}"
+                  stream_io.error "不明なフィルターです(#{item})"
+                  stream_io.puts "filters = #{filter_type_help}"
                   exit Narou::EXIT_ERROR_CODE
                 end
         sum + (apply ? 1 : 0)
@@ -203,19 +204,20 @@ module Command
 
     def output_lines(took_lines)
       if STDOUT.tty?
-        puts header
-        puts took_lines.values.join("\n").termcolor
+        stream_io.puts header
+        stream_io.puts took_lines.values.join("\n").termcolor
       elsif @options["echo"]
         # pipeにそのまま出力するときはansicolorコードが邪魔なので削除
-        puts header
-        puts TermColorLight.strip_tag(took_lines.values.join("\n"))
+        stream_io.puts header
+        stream_io.puts TermColorLight.strip_tag(took_lines.values.join("\n"))
       else
         # pipeに接続するときはIDを渡す
-        puts took_lines.keys.join(" ")
+        stream_io.puts took_lines.keys.join(" ")
       end
     end
 
     def execute(argv)
+      disable_logging
       super
       database_values = Database.instance.get_object.values
       limit =

data/lib/command/list/novel_decorator.rb

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
 #
 # Copyright 2013 whiteleaf. All rights reserved.
 #

data/lib/command/log.rb

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+
+#
+# Copyright 2013 whiteleaf. All rights reserved.
+#
+
+require_relative "log/tail"
+
+module Command
+  class Log < CommandBase
+    NUM = 20
+
+    def self.oneline_help
+      "保存したログを表示します"
+    end
+
+    def initialize
+      super("[options] [<path>]")
+      @opt.separator <<~HELP
+
+        ・#{Narou.log_dir} に保存されたログを表示します。
+        ・logging 設定が有効な場合のみ、ログは保存されます。
+        ・<path> を指定しない場合、最新のログが表示されます。
+        ・デフォルトでは末尾#{NUM}行のログを表示します。
+
+          Examples:
+            narou s logging=true # ログの保存を有効にする
+
+            narou log
+            narou log -n 100     # 100行分のログを表示
+            narou log -t         # tail -f オプションのようにログを流し続ける
+
+            narou log log/narou.txt # 直接ファイルを指定可能
+
+            # concurrency 設定（更新・変換同時実行）が有効時は変換ログが別ファイルに
+            # 分かれるので、変換ログを表示したい場合は -c オプションを付ける
+            # （指定しなかった場合は通常ログの方を表示）
+            narou log -c         # 変換ログを表示
+
+          Options:
+      HELP
+      @opt.on("-n", "--num NUM", Integer, "表示する行数を指定する") do |num|
+        @options["num"] = num
+      end
+      @opt.on("-t", "--tail", "ログを流し続ける") do
+        @options["tail"] = true
+      end
+      @opt.on("-c", "--source-convert", "変換ログを表示する。<path> を直接指定した場合は無視") do
+        @options["source-convert"] = true
+      end
+    end
+
+    def execute(argv)
+      disable_logging
+      super
+      path = resolve_path(argv.first) || latest_log_path
+      unless path
+        error "表示できるログファイルが一つも見つかりませんでした"
+        exit Narou::EXIT_ERROR_CODE
+      end
+      stream_io.puts "<cyan>#{path}</cyan>".termcolor
+      tail = Tail.new(path, @options["num"] || NUM)
+      tail.stream_io = stream_io
+      if @options["tail"]
+        tail.stream
+      else
+        tail.display
+      end
+    rescue Interrupt
+      stream_io.puts
+    end
+
+    def latest_log_path
+      Narou.log_dir.glob("*").sort_by(&:mtime).reverse.find do |path|
+        path_matches_source?(path)
+      end
+    end
+
+    def path_matches_source?(path)
+      ext = path.extname
+      base = path.basename(ext)
+      ended_with_convert = base.to_s.end_with?("_convert")
+      if @options["source-convert"]
+        ended_with_convert
+      else
+        !ended_with_convert
+      end
+    end
+
+    def resolve_path(path)
+      return nil if path.blank?
+      fullpath = Pathname(path).expand_path
+      unless fullpath.exist?
+        error "#{path} が存在しません"
+        exit Narou::EXIT_ERROR_CODE
+      end
+      fullpath
+    end
+  end
+end