Class: Aniview::Interface::AnimeIO

Inherits:
Bridge
  • Object
show all
Includes:
Util, Observable
Defined in:
lib/aniview/interface/animeio/animeio.rb

Instance Method Summary collapse

Methods included from Util

decode_object, encode_object, error_message, format_duration, format_progress, format_size, mounted_filesystem?, parse_format, readline

Methods inherited from Bridge

#make_hash

Constructor Details

#initialize(pref, mpvbridge) ⇒ AnimeIO

Returns a new instance of AnimeIO.



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 22

def initialize(pref, mpvbridge)
  @pref              = pref
  @mpvbridge         = mpvbridge
  @empty_hash        = make_empty_hash
  @watch_log_tag     = "♪"
  @dir_last_modified = {}
  @made_changes      = true

  self.load
  pref_changed!(verbose: false)
  #relisten

  @pref.add_observer(self, :pref_changed!)
  @mpvbridge.add_observer(self, :animefile_changed!)
  #@@local_anime.each { |key, obj| obj.add_observer(self, :animefile_changed!)}
end

Instance Method Details

#addWatched(file) ⇒ Object 



284
285
286
287
288
289
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 284

def addWatched(file)
  @@local_anime[file].watch
  save
  changed
  notify_observers
end

#animefile_changed! ⇒ Object 



92
93
94
95
96
97
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 92

def animefile_changed!
  return unless @mpvbridge.what_changed[:local_anime]
  save
  changed
  notify_observers
end

#cleanup ⇒ Object 



57
58
59
60
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 57

def cleanup
  @folder_listener.stop unless @folder_listener == nil
  @listener.stop unless @listener == nil
end

#index_anime(force: false, verbose: true, very_verbose: false) ⇒ description_of_the_return_value

Parameters:

  • force (defaults to: false) —

    The force

  • verbose (defaults to: true) —

    The verbose

Returns:

  • (description_of_the_return_value) 


107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
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
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 107

def index_anime(force: false, verbose: true, very_verbose: false)
  puts "Indexing anime" if very_verbose
  Thread.new {
    ret = {}
    made_changes = false
    lal = @@local_anime.length

    @anime_dirs.each_with_index { |dir, i|

      avs_file = dir + "/.avs"

      if File.exist?(avs_file)
        puts "found avs_data" if very_verbose
        avs_data = JSON.parse(File.open(avs_file, "r") { |f| f.read })
      else
        puts "no avs_data" if very_verbose
        avs_data = {}
      end


      # check if the directory exists
      unless File.exist? dir 
        unless @dir_last_modified[dir] == "check"
          @dir_last_modified[dir] = "check"
          made_changes = true
        end
        next
      end

      puts "checking #{dir}" if very_verbose

      # skip indexing a directory if it hasn't been modified since the last
      # check (disabled because doesn't save to much time)
      #
      #dir_mtime = File::Stat.new(dir).mtime
      #
      #if @dir_last_modified.key? dir
      #  if @dir_last_modified[dir] == dir_mtime
      #    next
      #  else
      #    @dir_last_modified[dir] = dir_mtime
      #  end
      #else
      #  @dir_last_modified.merge!(dir => dir_mtime)
      #end

      # get all files in the directory and in all subfolders of the directory
      files = Dir.glob("#{dir}*/*.{mkv,avi,mp4}") + Dir.glob("#{dir}/*.{mkv,avi,mp4}")

      files.each_with_index { |file, j|
        if verbose
          pct = Util.format_progress ((j + 1) / Float(files.length)) * ( (i + 1) / Float(@anime_dirs.length)) * 100
          Util.error_message("#{pct}% #{File.basename(file)}") 
        end

        print "indexing #{file} -> " if very_verbose
        params = {:seen => false, :watched_on => 0}

        #check if this anime has a stored watched value
        if avs_data.key?(file)
          puts "found stored data for this file" if very_verbose
          params = {
            :seen => avs_data[file]["seen"],
            :watched_on => avs_data[file]["watched_on"]
          }
          puts params if very_verbose
        else
          puts "no data on this file" if very_verbose
          avs_data.merge!({file => params})
        end

        # we have already indexed this file
        if @@local_anime != nil and @@local_anime.key?(file)
          
          puts "this file is in the index" if very_verbose

          # get its seen data
          params[:seen]       = @@local_anime[file].seen?
          params[:watched_on] = @@local_anime[file].seenOn
          puts params if very_verbose
          # get its last modifed data
          last_modified = File::Stat.new(file).mtime
          # get wether or not it is accessible
          @@local_anime[file].accessible = true

          # if we arent forcing or the anime hasn't been modified since the
          # last check, skip it
          if not force or last_modified == @@local_anime[file].last_modified
            puts "unchanged" if very_verbose
            ret.merge!(file => @@local_anime[file])
            next
          end
        end

        #puts "changed!"

        # okay we are making changes to local_anime
        made_changes = true

        # this is the anime file we are adding
        ret.merge!(
          file => AnimeFile.new(file, params)
        )
      }

      mavsf = @pref.get("avs_files")["create?"]
      oimfs = @pref.get("avs_files")["only_on_mounted_fs?"]
      if mavsf and ((oimfs and Util.mounted_filesystem?(dir)) or not oimfs)
        File.open(avs_file, "w") { |f| f.puts JSON.generate(avs_data)}
      end
    }

    #if @@local_anime != nil
    #  @@local_anime.merge!(ret)
    #else
      @@local_anime = ret
    #end

    #pruned = {}
    #@@local_anime.each{ |a|
    #  pruned.merge!(a[0] => a[1]) if File.exist? a[1].path
    #}
    #@@local_anime = pruned
    made_changes = true if @@local_anime.length != lal
    if @made_changes
      self.save 
      changed 
      notify_observers
    end
    #
  }
end

#items ⇒ Object

return an AnimeFile array of all mkvs in any dir specified in $pref



274
275
276
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 274

def items
  makeHash @@local_anime.values
end

#listen_to_all_dirs ⇒ Object 



39
40
41
42
43
44
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 39

def listen_to_all_dirs
  #listen to all
  all_dirs = (@pref.get("anime_locations").split(":")).map { |dir| @pref.parseDir(dir) }
  @folder_listener = FolderListen.to(all_dirs) { index_anime }
  @folder_listener.start
end

#load ⇒ Object 



71
72
73
74
75
76
77
78
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 71

def load
  raw = @pref.get "local_anime"
  if not raw == nil and raw.class == String
    @@local_anime = Util.decode_object raw
  else
    @@local_anime = {}
  end
end

#logWatched(file) ⇒ Object 



278
279
280
281
282
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 278

def logWatched(file)
  open(@pref.parseDir(@pref.get("watch_log")), 'a') do |f|
    f.puts %x(date).chomp + " ./" + File.basename(file) + " " + @watch_log_tag
  end
end

#make_empty_hash ⇒ Object 



62
63
64
65
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 62

def make_empty_hash
  af = AnimeFile.new("empty")
  {AnimeSeries.new("empty", [af]) => [af]}
end

#makeHash(arr) ⇒ Object

convert an AnimeFile array to a hash structured like (string)parent => (AnimeFile array)[child0, child1, child2] and return it



247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 247

def makeHash arr
  #return @empty_hash if arr[0] == nil
  
  keys = {}
  hash = {}

  arr.sort_by {|i| i.title}.each { |file|
    unless keys.key?(parent file.path)
      key = AnimeSeries.new parent(file.path)
      keys.merge! parent(file.path) => key
      hash.merge! key => []
    else
      key = keys[parent file.path]
    end

    hash[key] << file
    key << file
  }
  hash.sort_by{ |j| j[0].title }.to_h
end

#parent(file) ⇒ Object

returns the parent directory of a file



241
242
243
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 241

def parent(file)
  return File.basename(File.dirname(file))
end

#pref_changed!(verbose: true) ⇒ Object 



80
81
82
83
84
85
86
87
88
89
90
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 80

def pref_changed!(verbose: true)
  new_dirs = (@pref.get("anime_locations").split(":")).collect { |dir|
    parsed_dir = @pref.parseDir(dir)
    parsed_dir if File.exist?(parsed_dir)
  }.compact
  if new_dirs != @anime_dirs
    @anime_dirs = new_dirs
    index_anime(verbose: verbose)
    relisten
  end
end

#relisten ⇒ Object 



46
47
48
49
50
51
52
53
54
55
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 46

def relisten
  cleanup

  listen_to_all_dirs

  return if @anime_dirs == nil or @anime_dirs[0] == "" or @anime_dirs.length == 0

  @listener = Listen.to(*@anime_dirs, only: /\.mkv$/) { index_anime }
  @listener.start
end

#rmWatched(path = "") ⇒ Object 



291
292
293
294
295
296
297
298
299
300
301
302
303
304
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 291

def rmWatched(path = "")
  if @@local_anime.key? path
    @@local_anime[path].unwatch
  else
    r = []
    @@local_anime.values.each_with_index{ |v, i| r << {Integer(v.seenOn) => @@local_anime.keys[i]} if v.seen? }
    return if r.length == 0
    last_watched = (r.sort_by{ |h| h.keys.first}.reverse)[0].values[0]
    @@local_anime[last_watched].unwatch
  end
  save
  changed
  notify_observers
end

#save ⇒ Object 



67
68
69
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 67

def save
  @pref.set "local_anime", Util.encode_object(@@local_anime)
end

#unwatched ⇒ Object

return the difference of getAired and getWatched



269
270
271
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 269

def unwatched
  makeHash @@local_anime.reject { |f, a| a.seen? }.values
end

#watch(file) ⇒ Object 



306
307
308
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 306

def watch(file)
  @mpvbridge.play file
end

#watch_list(files) ⇒ Object 



310
311
312
313
314
315
316
317
 
# File 'lib/aniview/interface/animeio/animeio.rb', line 310

def watch_list(files)
  File.open("/tmp/aniview_list.m3u", "w") { |f|
    files.each { |file|
      f.puts file.path
    }
  }
  @mpvbridge.playlist("/tmp/aniview_list.m3u", files.map { |f| [f.path, f] }.to_h)
end