rfs 0.1

data/lib/RenameFileSet.bak.rb

@@ -0,0 +1,372 @@
+require 'rubygems'
+require_gem 'fxruby'
+require 'innate/mkdirs'
+require 'innate/roman'
+require 'innate/titlecase'
+
+include Fox
+
+#TO DO:
+# change pattern to combobox with some useful patterns
+# \d.+(\d+).+
+# (\d+[AB])
+
+#user interface
+class RenameWindow < FXMainWindow
+  def initialize(app)
+    super(app, 'Rename File Set', nil, nil, DECOR_ALL, 20, 100, 1000, 500)
+    
+    FXHorizontalFrame.new(self, LAYOUT_FILL | PACK_UNIFORM_WIDTH) { |mainFrame|
+      # navigation on left
+      @dirList = FXDirList.new mainFrame, nil, 0, LAYOUT_FILL | FRAME_NORMAL
+      # controls in centre
+      FXVerticalFrame.new(mainFrame, LAYOUT_FILL | PACK_UNIFORM_WIDTH) { |controls|
+        FXLabel.new controls, 'Search (replace first capture):', nil, LAYOUT_FILL_X | JUSTIFY_LEFT
+        @search = FXTextField.new controls, 0, nil, LAYOUT_FILL
+        
+        FXGroupBox.new(controls, 'Replacement Method', LAYOUT_FILL | FRAME_GROOVE | JUSTIFY_LEFT ) { |group|
+          @selReplace = FXRadioButton.new(group, 'Replace with:') { |r|
+            connectRadio r
+          }
+          @replace = FXTextField.new group, 30, nil, PACK_UNIFORM_WIDTH
+          @selAdd = FXRadioButton.new(group, 'Add to captured digits: ', nil, LAYOUT_FILL_Y) { |r|
+            connectRadio r
+          }
+          @add = FXSpinner.new(group, 10, nil, FRAME_NORMAL) { |s| s.range = (-500..500)}
+          @selRoman = FXRadioButton.new(group, 'Convert to Roman Numerals', nil, LAYOUT_FILL_X ) { |r|
+            connectRadio r
+          }
+          @selTapeNumbers = FXRadioButton.new(group, 'Convert Tape Numbers', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+          @selFill = FXRadioButton.new(group, 'Replace capture with last match capture:', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+          @fill = FXTextField.new group, 30, nil, PACK_UNIFORM_WIDTH
+          @selFile = FXRadioButton.new(group, 'Replace from files.txt', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+          @selCount = FXRadioButton.new(group, 'Count', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+          @selMarkChange = FXRadioButton.new(group, 'Mark Changes', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+          @selFromFullName = FXRadioButton.new(group, 'Replace with capture in full name:', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+          @fullNamePattern = FXTextField.new group, 30, nil, PACK_UNIFORM_WIDTH
+          @selMoveUp = FXRadioButton.new(group, 'Move to Parent Directory', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+          @selCapitalize = FXRadioButton.new(group, 'Capitalize', nil, LAYOUT_FILL_X) { |r|
+            connectRadio r
+          }
+        }
+        
+        FXButton.new(controls, '&List Files', nil, nil, 0, LAYOUT_FILL_X | BUTTON_NORMAL) { |b|
+          b.connect SEL_COMMAND, method(:onListClick)
+        }
+        FXButton.new(controls, '&Preview', nil, nil, 0, LAYOUT_FILL_X | FRAME_RAISED | FRAME_THICK) { |b|
+          b.connect SEL_COMMAND, method(:onPreviewClick)
+        }
+        FXButton.new(controls, '&Apply', nil, nil, 0, LAYOUT_FILL_X | FRAME_RAISED | FRAME_THICK) { |b|
+          b.connect SEL_COMMAND, method(:onReplaceClick)
+        }
+      }
+      # results on right
+      @results = FXText.new mainFrame, nil, 0, LAYOUT_FILL | FRAME_NORMAL | TEXT_READONLY
+    }
+    
+    @selReplace.checkState = TRUE
+    onRadioUpdate @selReplace, nil, nil
+  end
+  
+  def connectRadio(r)
+    r.connect SEL_COMMAND, method(:onRadioUpdate)
+  end
+
+  def onPreviewClick(sender, sel, ptr)
+    start false
+  end
+  
+  def onReplaceClick(sender, sel, ptr)
+    start true
+  end
+  
+  def onListClick(sender, sel, ptr)
+    start false, true
+    @dirList.setFocus
+  end
+  
+  def onRadioUpdate(sender, sel, ptr)
+    sender.parent.each_child do |c|
+      if FXRadioButton === c and c != sender
+        c.checkState = FALSE
+      end
+    end
+    @selReplace.checked? ? @replace.enable : @replace.disable
+    @selAdd.checked? ? @add.enable : @add.disable      
+    @selFill.checked? ? @fill.enable : @fill.disable
+    @selFromFullName.checked? ? @fullNamePattern.enable : @fullNamePattern.disable
+  end
+end
+
+#the real work of it
+class RenameWindow
+      
+  def start(commit, listonly = false)
+    @results.text = ''
+    traverseList @dirList, commit, listonly
+  end
+  
+  def traverseList(list, commit, listonly)
+    list.each do |i|
+       if i.selected?
+        doReplace @dirList.itemPathname(i), commit, listonly
+      end
+      traverseList(i, commit, listonly) if i.expanded?
+    end
+  end
+   
+  def subst(src, matchdata, repl)
+    return src unless matchdata and matchdata[1]
+    t = repl
+    c = 1
+    matchdata.to_a[c..-1].each do |m|
+      t.gsub!("%#{c}", m ? m : '')
+      c += 1
+    end
+    src[0, matchdata.offset(1)[0]] + (t == nil ? '' : t) + src[matchdata.offset(1)[1], src.length]
+  end
+  
+  def result(t)
+    @results.text += t.to_s + "\n"
+  end
+  
+  def rename(path, oldfn, newfn, commit)
+    unless oldfn == newfn
+      t = newfn
+      if commit
+        begin
+          fn = File.join(path, newfn)
+          mkdirs File.split(fn).first
+          File.rename(File.join(path, oldfn), fn)
+        rescue
+          t += ' - Error'
+        else
+          t += ' - OK'
+        end
+      end
+      result t
+    end
+  end
+  
+  def rename_each_match(re, path, commit)
+    return unless Regexp === re
+    dir = Dir.new(path) rescue return
+    dir.each do |file|
+      next if /^\.+$/ =~ file
+      md = re.match(file)
+      if md
+        newfn = yield(file, md) 
+        rename path, file, newfn, commit if newfn
+      end
+    end
+  end
+
+  def new_regexp(pattern)
+    begin
+      Regexp.new(pattern)
+    rescue
+      result "Error parsing regular expression:\n#{$!}\n------------"
+      nil
+    end
+  end
+
+  def doReplace(path, commit, listonly)
+    re = new_regexp @search.text
+    
+    if listonly
+      Dir.new(path).each do |file, md|
+        result "#{(File.directory?(File.join(path, file)) ? '\\' : '')}#{file}"
+      end
+      return
+    end
+
+    if @selReplace.checked?
+      rename_each_match(re, path, commit) do |file, md|
+        if md.length >= 2
+          subst(file, md, @replace.text)
+        else
+          file.sub(re, @replace.text)
+        end
+      end
+    elsif @selFile.checked?
+      fn = File.join(path, 'files.txt')
+      return unless File.exist? fn
+      File.open(fn) do |f|
+        rename_each_match(re, path, commit) do |file, md|
+          if md.length >= 2
+            t = f.readline.chomp rescue '--no data--'
+            subst(file, md, t)
+          end
+        end
+      end
+    elsif @selAdd.checked?
+      rename_each_match(re, path, commit) do |file, md|
+        if md.length >= 2
+          int = md[1].to_i rescue int = md[1] rescue int = 0
+          subst(file, md, (int + @add.value).to_s)
+        end
+      end
+    elsif @selCount.checked?
+      count = 0
+      rename_each_match(re, path, commit) do |file, md|
+        if md.length >= 2
+          count += 1
+          subst(file, md, count.to_s)
+        end
+      end
+    elsif @selFromFullName.checked?
+      r = new_regexp(@fullNamePattern.text)
+      if r
+        rename_each_match(re, path, commit) do |file, md|
+          fn = path + '\\' + file
+          m = r.match(fn)
+          if m
+            if m.length >= 2
+              replace = m[1]
+            else
+              replace = m
+            end
+            if md.length >= 2
+              subst(file, md, replace)
+            else
+              file.sub(re, replace)
+            end
+          else
+            result "no match: #{fn}"
+            nil
+          end
+        end
+      end
+    elsif @selMoveUp.checked?
+      p = File.split(path).first
+      unless p == '\\' 
+        rename_each_match(re, path, commit) do |file, md|
+          d = File.join(p, file)
+          File.rename File.join(path, file), d if commit
+          result d
+          nil
+        end
+      else
+        nil
+      end
+    elsif @selRoman.checked?
+      rename_each_match(re, path, commit) do |file, md|
+        if md.length >= 2
+          t = md[1].to_i rescue t = md[1] rescue t = 0
+          subst(file, md, t.roman) if Integer === t 
+        end
+      end
+    elsif @selMarkChange.checked?
+      prev = ''
+      rename_each_match(re, path, commit) do |file, md|
+        if md.length >= 2
+          unless prev == md[1]
+            prev = md[1]
+            subst(file, /()\..+?$/.match(file), '-change-')
+          end
+        end
+      end
+    elsif @selTapeNumbers.checked?
+      rename_each_match(re, path, commit) do |file, md|
+        if md.length >= 2
+          t = md[1]
+          n = t.to_i * 2 + (/a/i =~ t[-1,1] ? -1 : 0)
+          n += 2 if n < 1
+          subst(file, md, n.to_s)
+        end
+      end
+    elsif @selCapitalize.checked?
+      always_lower = %w{ vol
+        mp3 wav txt avi nfo pdf rar zip rb ape sfk mpg rm qt doc }
+      lower = %w{ as of in the and at or a to is on by for with an } + always_lower
+      split_chars = /[ �\-().]/
+      upper = /^[MDCLXVIW]+$/
+      rename_each_match(re, path, commit) do |file, md|
+        o = md[1]
+        prev = ''
+        if md.length >= 2
+          res = o.split(split_chars).inject([]) do |r, w|
+            if r.empty? or prev =~ /^([\d.]+|.*[;.!_]|)$/
+              if always_lower.include?(w.downcase)
+                r << w.downcase
+              elsif w =~ upper
+                r << w.upcase
+              else
+                r << w.capitalize
+              end
+            else
+              if lower.include?(w.downcase)
+                r << w.downcase
+              elsif w =~ upper
+                r << w.upcase
+              else
+                r << w.capitalize
+              end
+            end
+            prev = w
+            r
+          end
+          r = res.inject('') do |s, w|
+            unless w.empty?
+              until o[s.length, 1] =~ Regexp.new(Regexp.escape(w[0,1]), 'i') do
+                s << o[s.length, 1]
+              end
+              s << w
+            end
+            s
+          end
+          subst(file, md, r)
+        end
+      end
+    elsif @selFill.checked?
+      begin
+        fillRE = Regexp.new(@fill.text)
+      rescue
+        result "Error parsing regular expression:\n#{$!}\n------------"
+        return
+      end
+      fillText = ''
+      dir = Dir.new(path) rescue return
+      dir.each do |file|
+        next if /^\.+$/ =~ file
+        md = re.match(file)
+        if md or fillText
+          if md and md.length >= 2 and md[1].length > 0 and not (fillRE and fillRE.match(file))
+            fillText = md[1]
+            result "source: #{file}"
+          else
+            md = fillRE.match(file)
+            if md and md.length >= 2
+              t = subst(file, md, fillText)
+              rename path, file, t, commit
+            else
+              result "* No capture to fill * #{file}" 
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+if $0 == __FILE__
+  theApp = FXApp.new
+  theMainWindow = RenameWindow.new theApp 
+  theApp.create
+  theMainWindow.show
+  theApp.run
+end

data/lib/errors.rb

@@ -0,0 +1,15 @@
+class RenamerError < RuntimeError 
+end
+
+class ActionError < RenamerError
+  def initialize *opts
+    if opts.length > 0
+      super
+    else
+      super 'Unrecognized or missing action'
+    end
+  end    
+end
+
+class DirectoryExistsError < RuntimeError
+end

data/lib/filters.rb

@@ -0,0 +1,68 @@
+
+class Filter
+  def self.add(orig, nxt)
+    if orig
+      orig.add nxt
+    else
+      nxt
+    end
+  end    
+  
+  attr_accessor :next, :name, :regexp, :full_name
+  
+  def initialize(regexp, nxt = nil, full_name = false)
+    @next = nxt
+    @regexp = regexp
+    @full_name = full_name
+  end
+
+  def add(nxt)
+    if @next
+      @next.add nxt
+    else
+      @next = nxt
+    end
+    self
+  end
+
+  def filter(path, file)
+    unless @regexp =~ text(path, file)
+      next_filter(path, file)
+    else
+      false
+    end
+  end
+
+  def each(&b)
+    yield self
+    @next.each(&b) if @next
+  end
+  
+protected
+  def text path, file
+    if @full_name
+      File.expand_path(File.join(path, file))
+    else
+      file
+    end
+  end 
+
+  def next_filter path, file
+    if @next
+      @next.filter path, file
+    else
+      true
+    end
+  end
+end
+
+class FilterNonMatches < Filter
+  def filter(path, file)
+    if @regexp =~ text(path, file)
+      next_filter(path, file)
+    else
+      false
+    end
+  end
+end
+