git.zarb.org Git - booh/blob - bin/album2booh

   1 #! /usr/bin/ruby
   2 #
   3 #                         *  BOOH  *
   4 #
   5 # A.k.a 'Best web-album Of the world, Or your money back, Humerus'.
   6 #
   7 # The acronyn sucks, however this is a tribute to Dragon Ball by
   8 # Akira Toriyama, where the last enemy beaten by heroes of Dragon
   9 # Ball is named "Boo". But there was already a free software project
  10 # called Boo, so this one will be it "Booh". Or whatever.
  11 #
  12 #
  13 # Copyright (c) 2004-2006 Guillaume Cottenceau <http://zarb.org/~gc/resource/gc_mail.png>
  14 # Copyright (c) 2007 Stephane Fillod
  15 #
  16 # This software may be freely redistributed under the terms of the GNU
  17 # public license version 2.
  18 #
  19 # You should have received a copy of the GNU General Public License
  20 # along with this program; if not, write to the Free Software
  21 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  22
  23 # 'album' structure:
  24 # "captions.txt"
  25 #       <directory><tab><title>
  26 #       <picfilename><tab><title><tab><texte>
  27 # <directory>".captions.txt"
  28 #       <picfilename><tab><title><tab><texte>
  29 # <directory>/<picfilename>".txt"
  30 #       <free text>
  31 #
  32 # Since Booh requires UTF-8, you may need to convert caption files:
  33 #       $ recode ISO-8859-1..UTF-8 captions.txt
  34 #
  35
  36 require 'getoptlong'
  37 require 'gettext'
  38 include GetText
  39 require 'booh/rexml/document'
  40 include REXML
  41
  42 require 'booh/booh-lib'
  43 include Booh
  44
  45 #- options
  46 $options = [
  47     [ '--help',          '-h', GetoptLong::NO_ARGUMENT,       _("Get help message") ],
  48
  49     [ '--config',        '-C', GetoptLong::REQUIRED_ARGUMENT, _("File containing config listing images and videos within directories with captions") ],
  50
  51     [ '--verbose-level', '-v', GetoptLong::REQUIRED_ARGUMENT, _("Set max verbosity level (0: errors, 1: warnings, 2: important messages, 3: other messages)") ],
  52 ]
  53
  54 #- default values for some globals
  55 $switches = []
  56 $stdout.sync = true
  57
  58 def usage
  59     puts _("Usage: %s [OPTION]...") % File.basename($0)
  60     $options.each { |ary|
  61         printf " %3s, %-15s %s\n", ary[1], ary[0], ary[3]
  62     }
  63 end
  64
  65 def handle_options
  66     parser = GetoptLong.new
  67     parser.set_options(*$options.collect { |ary| ary[0..2] })
  68     begin
  69         parser.each_option do |name, arg|
  70             case name
  71             when '--help'
  72                 usage
  73                 exit(0)
  74
  75             when '--config'
  76                 if File.readable?(arg)
  77                     $xmldoc = REXML::Document.new File.new(arg)
  78                     $conffile = arg
  79                 else
  80                     die_ _('Config file does not exist or is unreadable.')
  81                 end
  82
  83             when '--verbose-level'
  84                 $verbose_level = arg.to_i
  85
  86             end
  87         end
  88     rescue
  89         puts $!
  90         usage
  91         exit(1)
  92     end
  93
  94     if !$xmldoc
  95         die_ _("Missing --config parameter.")
  96     end
  97
  98     $source = $xmldoc.root.attributes['source']
  99     $dest = $xmldoc.root.attributes['destination']
 100 end
 101
 102 def utf8_and_entities(string)
 103     return utf8(string).gsub('&agrave;', 'à').
 104                         gsub('&ccedil;', 'ç').
 105                         gsub('&ocirc;',  'ô').
 106                         gsub('&eacute;', 'é').
 107                         gsub('&ecirc;',  'ê').
 108                         gsub('&egrave;', 'è').
 109                         gsub('&Egrave;', 'È').
 110                         gsub('&icirc;',  'î').
 111                         gsub('&lt;',     '<').
 112                         gsub('&gt;',     '>').
 113                         gsub('&ugrave;', 'ù').
 114                         gsub('&quot;',   '"')
 115 end
 116
 117 def parse_album_captionstxt(filepath)
 118     begin
 119         contents = File.open(filepath).readlines
 120         out = {}
 121         out[:legends] = {}
 122         for line in contents
 123             if line =~ /^(.*)\t(.*)\t(.*)/
 124                 out[:legends][$1] = $2 + "\n" + $3
 125             elsif line =~ /^(.*)\t(.*)/
 126                 out[:legends][$1] = $2
 127             end
 128         end
 129         return out
 130     rescue
 131         return nil
 132     end
 133 end
 134
 135 def walk_source_dir
 136
 137     `find #{$source} -type d`.sort.each { |dir|
 138         dir.chomp!
 139         msg 2, _("Handling %s from config list...") % dir
 140
 141         if !infos = parse_album_captionstxt("#{dir}/captions.txt")
 142             next
 143         end
 144
 145         #- place xml document on proper node
 146         xmldir = $xmldoc.elements["//dir[@path='#{utf8(dir)}']"]
 147
 148         #if infos.has_key?(:title)
 149         #    type = find_subalbum_info_type(xmldir)
 150         #    xmldir.add_attribute("#{type}-caption", utf8_and_entities(infos[:title]))
 151         #end
 152
 153         xmldir.elements.each { |element|
 154             if %w(image video).include?(element.name)
 155                 if infos[:legends].has_key?(element.attributes['filename'])
 156                     element.add_attribute('caption', utf8_and_entities(infos[:legends][element.attributes['filename']]))
 157                 end
 158             end
 159         }
 160     }
 161 end
 162
 163
 164 handle_options
 165
 166 walk_source_dir
 167
 168 ios = File.open("#{$conffile}.merged", "w")
 169 $xmldoc.write(ios)
 170 ios.close