Class: Command::Diff

Inherits:
CommandBase show all
Defined in:
lib/command/diff.rb

Instance Method Summary collapse

Methods inherited from CommandBase

execute!, #load_local_settings

Constructor Details

#initializeDiff

Returns a new instance of Diff.



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# File 'lib/command/diff.rb', line 21

def initialize
  super("[<target>] [options ...]")
  @opt.separator <<-EOS

  ・指定した小説のアップデート前後の変更部分を setting コマンドで指定した difftool で表示します。
  ・引数を指定しなかった場合は直前に更新した小説の変更点を表示します。

  Example:
narou diff          # 直前に更新した小説の変更点を表示
narou diff 6
narou diff 6 -n 2   # 最新から2番目の差分との比較
narou diff 6 -2     # -n 2 の省略した記述方法
narou diff 6 2013.02.21@01;39;46   # 差分を直接指定

# 差分表示用プログラムの指定
narou setting difftool="C:\\Program Files\\WinMerge\\WinMergeU.exe"

# difftoolに渡す引数(指定しなければ単純に新旧ファイルを引数に呼び出す)
# 特殊な変数 %NEW : DLした最新データの差分用ファイルパス
#            %OLD : 古い方の差分用ファイルパス
narou setting difftool.arg='-e -x -ub -dl "NEW" -dr "OLD" %NEW %OLD'

  Options:
  EOS

  @options["number"] = 1
  @opt.on("-n NUM", "--number", "比較する差分を遡って指定する。最新のアップデートと直前のデータを比較するなら-n 1、2個前のアップデートなら-n 2。(デフォルトは-n 1)", Integer) { |number|
    @options["number"] = number if number > 1
  }
  @opt.on("-l", "--list", "指定した小説の差分一覧を表示する") {
    @options["list"] = true
  }
  @opt.on("-c", "--clean", "指定した小説の差分データを全て削除する") {
    @options["clean"] = true
  }
  @opt.on("--all-clean", "凍結済を除く全小説の差分データを削除する") {
    @options["all-clean"] = true
  }
end

Instance Method Details

#clean_all_diffObject



239
240
241
242
243
244
245
246
247
# File 'lib/command/diff.rb', line 239

def clean_all_diff
  Database.instance.each do |id, data|
    next if Narou.novel_frozen?(id)
    cache_root_dir = Downloader.get_cache_root_dir(id)
    next unless File.exists?(cache_root_dir)
    FileUtils.remove_entry_secure(cache_root_dir)
    puts "#{data["title"]} の差分を削除しました"
  end
end

#clean_diff(id) ⇒ Object



119
120
121
122
123
124
125
126
127
128
# File 'lib/command/diff.rb', line 119

def clean_diff(id)
  cache_root_dir = Downloader.get_cache_root_dir(id)
  print Database.instance.get_data("id", id)["title"] + ""
  unless File.exists?(cache_root_dir)
    puts "差分はひとつもありません"
    return
  end
  FileUtils.remove_entry_secure(cache_root_dir)
  puts "差分を削除しました"
end

#create_difftool_command_string(left, right) ⇒ Object



151
152
153
154
155
156
157
158
159
160
161
162
# File 'lib/command/diff.rb', line 151

def create_difftool_command_string(left, right)
  diff_arg = GlobalSetting.get["global_setting"]["difftool.arg"]
  diff_cmd = %!"#{@difftool}" !
  if diff_arg
    diff_arg.gsub!("%NEW", left)
    diff_arg.gsub!("%OLD", right)
    diff_cmd += diff_arg
  else
    diff_cmd += [left, right].join(" ")
  end
  diff_cmd
end

#create_temp_files(id) ⇒ Object



169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
# File 'lib/command/diff.rb', line 169

def create_temp_files(id)
  if @options["view_diff_version"]
    cache_root_dir = Downloader.get_cache_root_dir(id)
    if cache_root_dir
      cache_dir = File.join(cache_root_dir, @options["view_diff_version"])
      cache_dir = nil unless File.exists?(cache_dir)
    end
  else
    nth = @options["number"]
    list = get_sorted_cache_list(id)
    cache_dir = list[nth - 1] if list
  end
  unless cache_dir
    puts "差分データがありません"
    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 "最新話のみのアップデートのようです"
    return nil
  end
  novel_dir = Downloader.get_novel_section_save_dir(Downloader.get_novel_data_dir_by_target(id))
  latest_novel_sections = []
  cache_sections = []
  cache_section_list.each do |path|
    match_latest_path = File.join(novel_dir, File.basename(path))
    latest_novel_sections << YAML.load_file(match_latest_path) if File.exists?(match_latest_path)
    cache_sections << YAML.load_file(path)
  end

  novel_info = Database.instance[id]

  sections = latest_novel_sections
  temp_new = Tempfile.open(["new", ".txt"])
  temp_new.write(Template.get("diff.txt", binding))
  temp_new.close

  sections = cache_sections
  temp_old = Tempfile.open(["old", ".txt"])
  temp_old.write(Template.get("diff.txt", binding))
  temp_old.close

  [temp_new, temp_old]
end

#display_diff_list(id) ⇒ Object



216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
# File 'lib/command/diff.rb', line 216

def display_diff_list(id)
  cache_list = get_sorted_cache_list(id)
  print Database.instance.get_data("id", id)["title"] + ""
  if cache_list.empty?
    puts "差分はひとつもありません"
    return
  end
  puts "差分一覧"
  cache_list.each.with_index(1) do |cache_path, i|
    puts ("<bold><yellow>" + File.basename(cache_path) + "   -#{i}</yellow></bold>").termcolor
    cache_section_list = Dir.glob(File.join(cache_path, "*.yaml"))
    if cache_section_list.length > 0
      cache_section_list.map { |section_path|
        File.basename(section_path, ".yaml").split(" ", 2)
      }.sort_by { |v| v[0].to_i }.each { |index, subtitle|
        puts "#{index}部分 #{subtitle}"
      }
    else
      puts "   (最新話のみのアップデート)"
    end
  end
end

#exec_difftool(id) ⇒ Object



130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
# File 'lib/command/diff.rb', line 130

def exec_difftool(id)
  temp_paths = create_temp_files(id) or return
  diff_cmd = create_difftool_command_string(*temp_paths.map { |temp|
    path = temp.path
    if Helper.os_windows?
      path.encode!(Encoding::Windows_31J)
    end
    %!"#{path}"!
  })
  begin
    res = Open3.capture3(diff_cmd)
  rescue Errno::ENOENT => e
    error e.message
    exit 1
  ensure
    temp_paths.each { |tmp| tmp.delete }
  end
  puts res[0] unless res[0].empty?
  error res[1] unless res[1].empty?
end

#execute(argv) ⇒ Object



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# File 'lib/command/diff.rb', line 72

def execute(argv)
  short_number_option_parse(argv)
  super
  if argv.empty?
    latest = Database.instance.sort_by_last_update.first
    return unless latest
    id = latest["id"]
  else
    target = argv.shift
    id = Downloader.get_id_by_target(target)
  end
  unless id
    error "#{target} は存在しません"
    return
  end
  view_diff_version = argv.shift
  if view_diff_version
    if invalid_diff_version_string?(view_diff_version)
      error "差分指定の書式が違います(正しい例:2013.02.21@01;39;46)"
      return
    end
    @options["view_diff_version"] = view_diff_version
  end
  if @options["list"]
    display_diff_list(id)
    return
  end
  if @options["clean"]
    clean_diff(id)
    return
  end
  if @options["all-clean"]
    clean_all_diff
    return
  end
  @difftool = GlobalSetting.get["global_setting"]["difftool"]
  unless @difftool
    error "difftool が設定されていません。narou setting で difftool を設定して下さい"
    return
  end
  exec_difftool(id)
end

#get_sorted_cache_list(id) ⇒ Object



164
165
166
167
# File 'lib/command/diff.rb', line 164

def get_sorted_cache_list(id)
  list = Downloader.get_cache_list(id) or return nil
  list.sort_by { |v| File.basename(v) }.reverse
end

#invalid_diff_version_string?(str) ⇒ Boolean

Returns:

  • (Boolean)


115
116
117
# File 'lib/command/diff.rb', line 115

def invalid_diff_version_string?(str)
  str !~ /^\d{4}\.\d{2}\.\d{2}@\d{2};\d{2};\d{2}$/
end

#oneline_helpObject



17
18
19
# File 'lib/command/diff.rb', line 17

def oneline_help
  "アップデートされた小説の変更点を表示します"
end

#short_number_option_parse(argv) ⇒ Object

引数の中の -数字 オプション(-n 数字の省略形)を -n 数字 に変換する



62
63
64
65
66
67
68
69
70
# File 'lib/command/diff.rb', line 62

def short_number_option_parse(argv)
  argv.map! { |arg|
    if arg =~ /^-(\d+)$/
      ["-n", $1.to_s]
    else
      arg
    end
  }.flatten!
end