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

   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 #
  15 # This software may be freely redistributed under the terms of the GNU
  16 # public license version 2.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program; if not, write to the Free Software
  20 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21
  22 require 'getoptlong'
  23 require 'gettext'
  24 include GetText
  25 require 'rexml/document'
  26 include REXML
  27
  28 require 'booh/booh-lib'
  29 require 'booh/html-merges'
  30
  31 #- bind text domain as soon as possible because some _() functions are called early to build data structures
  32 bindtextdomain("booh")
  33
  34 #- options
  35 $options = [
  36     [ '--help',          '-h', GetoptLong::NO_ARGUMENT,       _("Get help message") ],
  37     [ '--version',       '-V', GetoptLong::NO_ARGUMENT,       _("Print version and exit") ],
  38
  39     [ '--source',        '-s', GetoptLong::REQUIRED_ARGUMENT, _("Directory which contains original images/videos as files or subdirs") ],
  40     [ '--destination',   '-d', GetoptLong::REQUIRED_ARGUMENT, _("Directory which will contain the web-album") ],
  41 #    [ '--clean',         '-c', GetoptLong::NO_ARGUMENT,       _("Clean destination directory") ],
  42
  43     [ '--theme',         '-t', GetoptLong::REQUIRED_ARGUMENT, _("Select HTML theme to use") ],
  44     [ '--config',        '-C', GetoptLong::REQUIRED_ARGUMENT, _("File containing config listing images and videos within directories with captions") ],
  45     [ '--config-skel',   '-k', GetoptLong::REQUIRED_ARGUMENT, _("Filename where the script will output a config skeleton") ],
  46     [ '--merge-config',  '-M', GetoptLong::REQUIRED_ARGUMENT, _("File containing config listing, where to merge new/removed images/videos from --source, and change theme info") ],
  47     [ '--merge-config-onedir',  '-O', GetoptLong::REQUIRED_ARGUMENT, _("File containing config listing, for merging the subdir specified with --dir") ],
  48     [ '--merge-config-subdirs', '-U', GetoptLong::REQUIRED_ARGUMENT, _("File containing config listing, for merging the new subdirs down the subdir specified with --dir") ],
  49     [ '--dir',           '-D', GetoptLong::REQUIRED_ARGUMENT, _("Directory for merge with --merge-config-onedir or --merge-config-subdirs") ],
  50     [ '--use-config',    '-u', GetoptLong::REQUIRED_ARGUMENT, _("File containing config listing, where to change theme info") ],
  51     [ '--force',         '-f', GetoptLong::NO_ARGUMENT,       _("Force generation of album even if the GUI marked some directories as already generated") ],
  52
  53     [ '--sizes',         '-S', GetoptLong::REQUIRED_ARGUMENT, _("Specify the list of images sizes to use instead of all specified in the theme (this is a comma-separated list)") ],
  54     [ '--thumbnails-per-row', '-T', GetoptLong::REQUIRED_ARGUMENT, _("Specify the amount of thumbnails per row in the thumbnails page (if applicable in theme)") ],
  55     [ '--thumbnails-per-page', '-p', GetoptLong::REQUIRED_ARGUMENT, _("Specify the amount of thumbnails per page in the thumbnails page, after which split occurs") ],
  56     [ '--optimize-for-32', '-o', GetoptLong::NO_ARGUMENT,       _("Resize images with optimized sizes for 3/2 aspect ratio rather than 4/3 (typical aspect ratio of pictures from non digital cameras are 3/2 when pictures from digital cameras are 4/3)") ],
  57     [ '--index-link',    '-l', GetoptLong::REQUIRED_ARGUMENT, _("Specify the HTML markup to use on the bottom of pages for a small link returning to wherever you see fit in your website (or somewhere else)") ],
  58     [ '--made-with',     '-n', GetoptLong::REQUIRED_ARGUMENT, _("Specify the HTML markup to use on the bottom of pages for a small 'made with' message") ],
  59     [ '--comments-format','-c', GetoptLong::REQUIRED_ARGUMENT, _("Specify comments format to use for images instead of only filename when creating new albums; use ImageMagick's format") ],
  60
  61     [ '--mproc',         '-m', GetoptLong::REQUIRED_ARGUMENT, _("Specify the number of processors for multi-processors machines") ],
  62
  63     [ '--for-gui',       '-g', GetoptLong::NO_ARGUMENT,       _("Do the minimum work to be able to see the album under the GUI (don't generate all thumbnails)") ],
  64
  65     [ '--verbose-level', '-v', GetoptLong::REQUIRED_ARGUMENT, _("Set max verbosity level (0: errors, 1: warnings, 2: important messages, 3: other messages)") ],
  66     [ '--info-pipe',     '-i', GetoptLong::REQUIRED_ARGUMENT, _("Name a file where to write information about what's going on (used by the GUI)") ],
  67 ]
  68
  69 #- default values for some globals
  70 $switches = []
  71 $stdout.sync = true
  72 $no_identify = false
  73 $ignore_videos = false
  74 $forgui = false
  75 $hardlinks_ok = true
  76
  77 def usage
  78     puts _("Usage: %s [OPTION]...") % File.basename($0)
  79     $options.each { |ary|
  80         printf " %3s, %-18s %s\n", ary[1], ary[0], ary[3]
  81     }
  82 end
  83
  84 def handle_options
  85     parser = GetoptLong.new
  86     parser.set_options(*$options.collect { |ary| ary[0..2] })
  87     begin
  88         parser.each_option do |name, arg|
  89             case name
  90             when '--help'
  91                 usage
  92                 exit(0)
  93
  94             when '--version'
  95                 puts _("Booh version %s
  96
  97 Copyright (c) 2005 Guillaume Cottenceau.
  98 This is free software; see the source for copying conditions.  There is NO
  99 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.") % $VERSION
 100
 101                 exit(0)
 102
 103             when '--source'
 104                 $source = File.expand_path(arg.sub(%r|/$|, ''))
 105                 if !File.directory?($source)
 106                     die _("Argument to --source must be a directory")
 107                 end
 108             when '--destination'
 109                 $dest = File.expand_path(arg.sub(%r|/$|, ''))
 110                 if File.exists?($dest) && !File.directory?($dest)
 111                     die _("If --destination exists, it must be a directory")
 112                 end
 113                 if $dest != make_dest_filename($dest)
 114                     die _("Sorry, destination directory can't contain non simple alphanumeric characters.")
 115                 end
 116 #            when '--clean'
 117 #                system("rm -rf #{$dest}")
 118
 119             when '--theme'
 120                 $theme = arg
 121             when '--config'
 122                 arg = File.expand_path(arg)
 123                 if File.readable?(arg)
 124                     $xmldoc = REXML::Document.new File.new(arg)
 125                     $mode = 'use_config'
 126                 else
 127                     die _('Config file does not exist or is unreadable.')
 128                 end
 129             when '--config-skel'
 130                 arg = File.expand_path(arg)
 131                 if File.exists?(arg)
 132                     if File.directory?(arg)
 133                         die _("Config skeleton file (%s) already exists and is a directory! Please change the filename.") % arg
 134                     else
 135                         msg 1, _("Config skeleton file already exists, backuping to %s.backup") % arg
 136                         File.rename(arg, "#{arg}.backup")
 137                     end
 138                 end
 139                 $config_writeto = arg
 140                 $mode = 'gen_config'
 141             when '--merge-config'
 142                 arg = File.expand_path(arg)
 143                 if File.readable?(arg)
 144                     msg 2, _("Merge config notice: backuping current config file to %s.backup") % arg
 145                     $xmldoc = REXML::Document.new File.new(arg)
 146                     File.rename(arg, "#{arg}.backup")
 147                     $config_writeto = arg
 148                     $mode = 'merge_config'
 149                 else
 150                     die _('Config file does not exist or is unreadable.')
 151                 end
 152             when '--merge-config-onedir'
 153                 arg = File.expand_path(arg)
 154                 if File.readable?(arg)
 155                     msg 2, _("Merge config notice: backuping current config file to %s.backup") % arg
 156                     $xmldoc = REXML::Document.new File.new(arg)
 157                     File.rename(arg, "#{arg}.backup")
 158                     $config_writeto = arg
 159                     $mode = 'merge_config_onedir'
 160                 else
 161                     die _('Config file does not exist or is unreadable.')
 162                 end
 163             when '--merge-config-subdirs'
 164                 arg = File.expand_path(arg)
 165                 if File.readable?(arg)
 166                     msg 2, _("Merge config notice: backuping current config file to %s.backup") % arg
 167                     $xmldoc = REXML::Document.new File.new(arg)
 168                     File.rename(arg, "#{arg}.backup")
 169                     $config_writeto = arg
 170                     $mode = 'merge_config_subdirs'
 171                 else
 172                     die _('Config file does not exist or is unreadable.')
 173                 end
 174             when '--dir'
 175                 arg = File.expand_path(arg)
 176                 if !File.readable?(arg)
 177                     die _('Specified directory to merge with --dir is not readable')
 178                 else
 179                     $onedir = arg
 180                 end
 181             when '--use-config'
 182                 arg = File.expand_path(arg)
 183                 if File.readable?(arg)
 184                     msg 2, _("Use config notice: backuping current config file to %s.backup") % arg
 185                     $xmldoc = REXML::Document.new File.new(arg)
 186                     File.rename(arg, "#{arg}.backup")
 187                     $config_writeto = arg
 188                     $mode = 'use_config_changetheme'
 189                 else
 190                     die _('Config file does not exist or is unreadable.')
 191                 end
 192
 193             when '--sizes'
 194                 $limit_sizes = arg
 195
 196             when '--thumbnails-per-row'
 197                 $N_per_row = arg
 198
 199             when '--thumbnails-per-page'
 200                 $N_per_page = arg
 201
 202             when '--optimize-for-32'
 203                 $optimize_for_32 = true
 204
 205             when '--made-with'
 206                 $madewith = arg
 207
 208             when '--index-link'
 209                 $indexlink = arg
 210
 211             when '--comments-format'
 212                 $commentsformat = arg
 213
 214             when '--force'
 215                 $force = true
 216
 217             when '--mproc'
 218                 $mproc = arg.to_i
 219                 $pids = []
 220
 221             when '--for-gui'
 222                 $forgui = true
 223
 224             when '--verbose-level'
 225                 $verbose_level = arg.to_i
 226
 227             when '--info-pipe'
 228                 $info_pipe = File.open(arg, File::WRONLY)
 229                 $info_pipe.sync = true
 230             end
 231         end
 232     rescue
 233         puts $!
 234         usage
 235         exit(1)
 236     end
 237
 238     if !$source && $xmldoc
 239         $source = from_utf8($xmldoc.root.attributes['source']).sub(%r|/$|, '')
 240         $dest = from_utf8($xmldoc.root.attributes['destination']).sub(%r|/$|, '')
 241         $theme ||= $xmldoc.root.attributes['theme']
 242         $limit_sizes ||= $xmldoc.root.attributes['limit-sizes']
 243         if $mode == 'use_config' || $mode =~ /^merge_config/
 244             $optimize_for_32 = !$xmldoc.root.attributes['optimize-for-32'].nil?
 245             $N_per_row = $xmldoc.root.attributes['thumbnails-per-row']
 246             $N_per_page = $xmldoc.root.attributes['thumbnails-per-page']
 247             $madewith = $xmldoc.root.attributes['made-with']
 248             $indexlink = $xmldoc.root.attributes['index-link']
 249         end
 250     end
 251
 252     if $mode == 'merge_config_onedir' && !$onedir
 253         die _("Missing --dir for --merge-config-onedir")
 254     end
 255     if $mode == 'merge_config_subdirs' && !$onedir
 256         die _("Missing --dir for --merge-config-subdirs")
 257     end
 258
 259     if !$source
 260         usage
 261         exit(0)
 262     end
 263     if !$dest
 264         die _("Missing --destination parameter.")
 265     end
 266     if !$theme
 267         $theme = 'simple'
 268     end
 269
 270     select_theme($theme, $limit_sizes, $optimize_for_32, $N_per_row)
 271
 272     if !$xmldoc
 273         additional_params = ''
 274         if $limit_sizes
 275             additional_params += "limit-sizes='#{$limit_sizes}'"
 276         end
 277         if $optimize_for_32
 278             additional_params += " optimize-for-32='true'"
 279         end
 280         if $N_per_row
 281             additional_params += " thumbnails-per-row='#{$N_per_row}'"
 282         end
 283         if $N_per_page
 284             additional_params += " thumbnails-per-page='#{$N_per_page}'"
 285         end
 286         if $madewith
 287             additional_params += " made-with='#{$madewith}'"
 288         end
 289         if $indexlink
 290             additional_params += " index-link='#{$indexlink}'"
 291         end
 292         $xmldoc = Document.new "<booh version='#{$VERSION}' source='#{utf8($source)}' destination='#{utf8($dest)}' theme='#{$theme}' #{additional_params}/>"
 293         $xmldoc << XMLDecl.new(XMLDecl::DEFAULT_VERSION, $CURRENT_CHARSET)
 294         $mode = 'gen_config'
 295     end
 296
 297     if $mode == 'merge_config' || $mode == 'use_config_changetheme'
 298         $xmldoc.root.add_attribute('theme', $theme)
 299         $xmldoc.root.add_attribute('version', $VERSION)
 300         if $limit_sizes
 301             $xmldoc.root.add_attribute('limit-sizes', $limit_sizes)
 302         else
 303             $xmldoc.root.delete_attribute('limit-sizes')
 304         end
 305         if $optimize_for_32
 306             $xmldoc.root.add_attribute('optimize-for-32', 'true')
 307         else
 308             $xmldoc.root.delete_attribute('optimize-for-32')
 309         end
 310         if $N_per_row
 311             $xmldoc.root.add_attribute('thumbnails-per-row', $N_per_row)
 312         else
 313             $xmldoc.root.delete_attribute('thumbnails-per-row')
 314         end
 315         if $N_per_page
 316             $xmldoc.root.add_attribute('thumbnails-per-page', $N_per_page)
 317         else
 318             $xmldoc.root.delete_attribute('thumbnails-per-page')
 319         end
 320         if $madewith
 321             $xmldoc.root.add_attribute('made-with', $madewith)
 322         else
 323             $xmldoc.root.delete_attribute('made-with')
 324         end
 325         if $indexlink
 326             $xmldoc.root.add_attribute('index-link', $indexlink)
 327         else
 328             $xmldoc.root.delete_attribute('index-link')
 329         end
 330     end
 331
 332     if $madewith
 333         $madewith = $madewith.gsub('%booh', '"http://booh.org/"')
 334     end
 335 end
 336
 337 def read_config
 338     $config = {}
 339 end
 340
 341 def write_config
 342 end
 343
 344 def info(value)
 345     if $info_pipe
 346         $info_pipe.puts(value)
 347     end
 348 end
 349
 350 def die(value)
 351     if $info_pipe
 352         $info_pipe.puts("die: " + value)
 353     end
 354     die_ value
 355 end
 356
 357 def check_installation
 358     if !system("which convert >/dev/null 2>/dev/null")
 359         die _("The program 'convert' is needed. Please install it.
 360 It is generally available with the 'ImageMagick' software package.")
 361     end
 362     if !system("which identify >/dev/null 2>/dev/null")
 363         msg 1, _("the program 'identify' is needed to get images sizes and EXIF data. Please install it.
 364 It is generally available with the 'ImageMagick' software package.")
 365         $no_identify = true
 366     end
 367     missing = %w(transcode mencoder).delete_if { |prg| system("which #{prg} >/dev/null 2>/dev/null") }
 368     if missing != []
 369         msg 1, _("the following program(s) are needed to handle videos: '%s'. Videos will be ignored.") % missing.join(', ')
 370         $ignore_videos = true
 371     end
 372 end
 373
 374 def replace_line(surround, keyword, line)
 375     begin
 376         contents = eval "$#{keyword}"
 377         line.sub!(/#{surround}#{keyword}#{surround}/, contents)
 378     rescue TypeError
 379         die _("No '%s' found for substitution") % keyword
 380     end
 381 end
 382
 383 def build_html_skeletons
 384     $html_images     = File.open("#{$FPATH}/themes/#{$theme}/skeleton_image.html").readlines
 385     $html_thumbnails = File.open("#{$FPATH}/themes/#{$theme}/skeleton_thumbnails.html").readlines
 386     $html_index      = File.open("#{$FPATH}/themes/#{$theme}/skeleton_index.html").readlines
 387     for line in $html_images + $html_thumbnails + $html_index
 388         while line =~ /~~~(\w+)~~~/
 389             replace_line('~~~', $1, line)
 390         end
 391     end
 392 end
 393
 394 def find_caption_value(xmldir, filename)
 395     if cap = xmldir.elements["*[@filename='#{utf8(filename)}']"].attributes['caption']
 396         return cap.gsub("\n", '<br/>')
 397     else
 398         return nil
 399     end
 400 end
 401
 402 def find_captions(xmldir, images)
 403     return images.collect { |img| find_caption_value(xmldir, img) }
 404 end
 405
 406 #- stolen from CVSspam
 407 def urlencode(text)
 408   text.sub(/[^a-zA-Z0-9\-,.*_\/]/) do
 409     "%#{sprintf('%2X', $&[0])}"
 410   end
 411 end
 412
 413 def all_images_sizes
 414     return $limit_sizes =~ /original/ ? $images_size + [ { 'name' => 'original' } ] : $images_size
 415 end
 416
 417 def html_reload_to_thumbnails
 418     html_reload_to_thumbnails = $preferred_size_reloader.clone
 419     html_reload_to_thumbnails.gsub!(/~~theme~~/, $theme)
 420     html_reload_to_thumbnails.gsub!(/~~default_size~~/, $default_size['name'])
 421     html_reload_to_thumbnails.gsub!(/~~all_sizes~~/, all_images_sizes.collect { |s| "\"#{size2js(s['name'])}\"" }.join(', '))
 422     size_auto_chooser = '';
 423     all_images_sizes.find_all { |s| s.has_key?('optimizedforwidth') }.
 424                      sort { |a,b| b['optimizedforwidth'].to_i <=> a['optimizedforwidth'].to_i }.
 425                      each { |s| size_auto_chooser += "if (w + 50 > #{s['optimizedforwidth']}) { return 'thumbnails-#{size2js(s['name'])}-0.html'; }\n" }
 426     html_reload_to_thumbnails.gsub!(/~~size_auto_chooser~~/, size_auto_chooser)
 427     return html_reload_to_thumbnails
 428 end
 429
 430 def discover_iterations(iterations, line)
 431     if line =~ /~~iterate(\d)_open(_max(\d+|N))?~~/
 432         for iter in iterations.values
 433             if iter['open']
 434                 iter['open'] = false
 435                 iter['close_wait'] = $1.to_i
 436             end
 437         end
 438         max = $3 == 'N' ? ($N_per_row || $default_N) : $3
 439         iterations[$1.to_i] = { 'open' => true, 'max' => max, 'opening' => '', 'closing' => '' }
 440         if $1.to_i == 1
 441             line.sub!(/.*/, '~~thumbnails~~')
 442         else
 443             line.sub!(/.*/, '')
 444         end
 445     elsif line =~ /~~iterate(\d)_close~~/
 446         iterations[$1.to_i]['open']  = false;
 447         iterations[$1.to_i]['close'] = true;
 448         line.sub!(/.*/, '')
 449     else
 450         for iter in iterations.values
 451             if iter['open']
 452                 iter['opening'] += line
 453                 line.sub!(/.*/, '')
 454             end
 455             if !iter['close'] && iter['close_wait'] && iterations[iter['close_wait']]['close']
 456                 iter['closing'] += line
 457                 line.sub!(/.*/, '')
 458             end
 459         end
 460     end
 461 end
 462
 463 def reset_iterations(iterations)
 464     for iter in iterations.values
 465         iter['value'] = 0
 466     end
 467 end
 468
 469 def run_iterations(iterations, amount)
 470     html = ''
 471     should_rerun = false
 472     for level in iterations.keys.sort
 473         if iterations[level]['value'] == 0
 474             html += iterations[level]['opening']
 475         elsif level == iterations.keys.max
 476             if !iterations[level]['max'] || iterations[level]['max'] && iterations[level]['value'] + amount <= iterations[level]['max'].to_i
 477                 html += iterations[level]['opening']
 478             else
 479                 should_rerun = true
 480             end
 481         end
 482         iterations[level]['value'] += amount
 483         if iterations[level]['max'] && iterations[level]['value'] > iterations[level]['max'].to_i
 484             iterations[level]['value'] = 0
 485             iterations[level-1]['value'] = 0
 486             html += iterations[level-1]['closing']
 487         end
 488     end
 489     if should_rerun
 490         return html + run_iterations(iterations, amount)
 491     else
 492         return html
 493     end
 494 end
 495
 496 def close_iterations(iterations)
 497     html = ''
 498     for level in iterations.keys.sort.reverse
 499         html += iterations[level]['closing']
 500     end
 501     return html
 502 end
 503
 504 def img_element(fullpath)
 505     if size = get_image_size(fullpath)
 506         sizespec = 'width="' + size[:x].to_s + '" height="' + size[:y].to_s + '"'
 507     else
 508         sizespec = ''
 509     end
 510     return '<img src="' + File.basename(fullpath) + '" ' + sizespec + ' alt="image"/>'
 511 end
 512
 513 def size2js(name)
 514     return name.gsub(/-/, '')
 515 end
 516
 517 def substitute_html_sizes(html, sizeobj, type, suffix)
 518     sizestrings = []
 519     if $images_size.length > 1 || (type == 'image' && $limit_sizes =~ /original/)
 520         for sizeobj2 in $images_size
 521             sizejs = size2js(sizeobj2['name'])
 522             sizen = sizename(sizeobj2['name'])
 523             if sizeobj != sizeobj2
 524                 if type == 'thumbnails'
 525                     sizestrings << '<a href="thumbnails-' + sizejs + suffix + '.html" onclick="set_preferred_size(\'' + sizejs + '\')">' + sizen + '</a>'
 526                 else
 527                     sizestrings << '<a id="link' + sizejs + '" onclick="set_preferred_size(\'' + sizejs + '\')">' + sizen + '</a>'
 528                 end
 529             else
 530                 sizestrings << sizen
 531             end
 532         end
 533         if type == 'image' && $limit_sizes =~ /original/
 534             sizestrings << '<a id="linkoriginal" target="newframe">' + sizename('original') + '</a>'
 535         end
 536     end
 537     html.sub!(/~~sizes~~(.+)~~/) { sizestrings.join($1) }
 538 end
 539
 540 def substitute_navigation(html, xmldir)
 541     if xmldir.parent.name == 'dir'
 542         nav = ''
 543         path = '..'
 544         parent = xmldir.parent
 545         while parent.name == 'dir'
 546             parentcaption = parent.attributes['subdirs-caption'] || File.basename(parent.attributes['path'])
 547             nav = "<a href=\"#{path}/index.html\">#{parentcaption}</a> #{utf8(_(" > "))} #{nav}"
 548             path += '/..'
 549             parent = parent.parent
 550         end
 551         html.gsub!(/~~ifnavigation\?~~(.+?)~~fi~~/) { $1 }
 552         html.gsub!(/~~navigation~~/, nav + (xmldir.attributes['subdirs-caption'] || File.basename(xmldir.attributes['path'])))
 553     else
 554         html.gsub!(/~~ifnavigation\?~~(.+?)~~fi~~/, '')
 555     end
 556 end
 557
 558 def xmldir2destdir(xmldir)
 559     return make_dest_filename(from_utf8(File.basename(xmldir.attributes['path'])))
 560 end
 561
 562 def find_previous_album(xmldir)
 563     relative_pos = ''
 564     begin
 565         #- move to previous dir element if exists
 566         if prevelem = xmldir.previous_element_byname_notattr('dir', 'deleted')
 567             xmldir = prevelem
 568             relative_pos += '../' + xmldir2destdir(xmldir) + '/'
 569             child = nil
 570             #- after having moved to previous dir, we need to go down last subdir until the last one
 571             while child = xmldir.elements['dir']
 572                 while nextchild = child.next_element_byname_notattr('dir', 'deleted')
 573                     child = nextchild
 574                 end
 575                 relative_pos += xmldir2destdir(child) + '/'
 576                 xmldir = child
 577             end
 578         else
 579             #- previous dir doesn't exist, move to previous dir element if exists
 580             xmldir = xmldir.parent
 581             if xmldir.name == 'dir' && !xmldir.attributes['deleted']
 582                 relative_pos += '../'
 583             else
 584                 return nil
 585             end
 586         end
 587     end while !xmldir.child_byname_notattr('image', 'deleted') && !xmldir.child_byname_notattr('video', 'deleted')
 588     return File.reduce_path(relative_pos)
 589 end
 590
 591 def find_next_album(xmldir)
 592     relative_pos = ''
 593     begin
 594         #- first child dir element (catches when initial xmldir has both thumbnails and subdirs)
 595         if firstchild = xmldir.child_byname_notattr('dir', 'deleted')
 596             xmldir = firstchild
 597             relative_pos += xmldir2destdir(xmldir) + '/'
 598         #- next brother
 599         elsif nextbro = xmldir.next_element_byname_notattr('dir', 'deleted')
 600             xmldir = nextbro
 601             relative_pos += '../' + xmldir2destdir(xmldir) + '/'
 602         else
 603             #- go up until we have a next brother or we are finished
 604             begin
 605                 xmldir = xmldir.parent
 606                 relative_pos += '../'
 607             end while xmldir && !xmldir.next_element_byname_notattr('dir', 'deleted')
 608             if xmldir
 609                 xmldir = xmldir.next_element_byname('dir')
 610                 relative_pos += '../' + xmldir2destdir(xmldir) + '/'
 611             else
 612                 return nil
 613             end
 614         end
 615     end while !xmldir.child_byname_notattr('image', 'deleted') && !xmldir.child_byname_notattr('video', 'deleted')
 616     return File.reduce_path(relative_pos)
 617 end
 618
 619 def sub_previous_next_album(previous_album, next_album, html)
 620     if previous_album
 621         html.gsub!(/~~previous_album~~/, '<a href="' + previous_album + 'thumbnails.html">' + utf8(_('previous album')) + '</a>')
 622         html.gsub!(/~~ifprevious_album\?~~(.+?)~~fi~~/) { $1 }
 623     else
 624         html.gsub!(/~~previous_album~~/, '')
 625         html.gsub!(/~~ifprevious_album\?~~(.+?)~~fi~~/, '')
 626     end
 627     if next_album
 628         html.gsub!(/~~next_album~~/, '<a href="' + next_album + 'thumbnails.html">' + utf8(_('next album')) + '</a>')
 629         html.gsub!(/~~ifnext_album\?~~(.+?)~~fi~~/) { $1 }
 630     else
 631         html.gsub!(/~~next_album~~/, '')
 632         html.gsub!(/~~ifnext_album\?~~(.+?)~~fi~~/, '')
 633     end
 634     return html
 635 end
 636
 637 def walk_source_dir
 638
 639     #- preprocess the path->dir, rexml is very slow with that; we seem to improve speed by 7%
 640     optxpath = {}
 641     $xmldoc.elements.each('//dir') { |elem|
 642         optxpath[elem.attributes['path']] = elem
 643     }
 644
 645     examined_dirs = nil
 646     if $mode == 'merge_config_onedir'
 647         examined_dirs = [ $onedir ]
 648     elsif $mode == 'merge_config_subdirs'
 649         examined_dirs = `find '#{$onedir}' -type d -follow`.sort.collect { |v| v.chomp }.delete_if { |v| optxpath.has_key?(utf8(v)) }
 650     else
 651         examined_dirs = `find '#{$source}' -type d -follow`.sort.collect { |v| v.chomp }
 652         if $mode == 'merge_config'
 653             $xmldoc.elements.each('//dir') { |elem|
 654                 if ! examined_dirs.include?(elem.attributes['path'])
 655                     msg 2, _("Merging config: removing directory %s from config, isn't on filesystem anymore") % elem.attributes['path']
 656                     elem.remove
 657                 end
 658             }
 659         end
 660     end
 661     info("directories: #{examined_dirs.length}, sizes: #{$images_size.length}")
 662
 663     examined_dirs.each { |dir|
 664         if dir =~ /'/
 665             die _("Source directory or sub-directories can't contain a single-quote character, sorry: %s") % dir
 666         end
 667         if $mode !~ /^use_config/
 668             Dir.entries(dir).each { |file|
 669                 if file =~ /['"\[\]]/
 670                     die _("Files can't contain any of the characters ', \", [ or ], sorry: %s") % "#{dir}/#{file}"
 671                 end
 672             }
 673         end
 674     }
 675
 676     examined_dirs.each { |dir|
 677         if File.basename(dir) =~ /^\./
 678             msg 1, _("Ignoring directory %s, begins with a dot (indicating a hidden directory)") % dir
 679             next
 680         end
 681
 682         dest_dir = make_dest_filename(dir.sub(/^#{Regexp.quote($source)}/, $dest))
 683
 684         #- place xml document on proper node if exists, else create
 685         xmldir = optxpath[utf8(dir)]
 686         if $mode == 'use_config' || $mode == 'use_config_changetheme'
 687             if !xmldir || (xmldir.attributes['already-generated'] && !$force) || xmldir.attributes['deleted']
 688                 info("walking: #{dir}|#{$source}, 0 elements")
 689                 if xmldir && xmldir.attributes['deleted']
 690                     system("rm -rf '#{dest_dir}'")
 691                 end
 692                 next
 693             end
 694         else
 695             if $mode == 'gen_config' || (($mode == 'merge_config' || $mode == 'merge_config_subdirs') && !xmldir)
 696                 #- add the <dir..> element if necessary
 697                 parent = File.dirname(dir)
 698                 xmldir = $xmldoc.elements["//dir[@path='#{utf8(parent)}']"]
 699                 if !xmldir
 700                     xmldir = $xmldoc.root
 701                 end
 702                 #- need to remove the already-generated mark of the parent because of the sub-albums page containing now one more element
 703                 xmldir.delete_attribute('already-generated')
 704                 xmldir = optxpath[utf8(dir)] = xmldir.add_element('dir', { 'path' => utf8(dir) })
 705             end
 706         end
 707         xmldir.delete_attribute('already-generated')
 708
 709         #- read images/videos entries from config or from directories depending on mode
 710         entries = []
 711         if $mode == 'use_config' || $mode == 'use_config_changetheme'
 712             msg 2, _("Handling %s from config list...") % dir
 713             xmldir.elements.each { |element|
 714                 if %w(image video).include?(element.name) && !element.attributes['deleted']
 715                     entries << from_utf8(element.attributes['filename'])
 716                 end
 717             }
 718         else
 719             msg 2, _("Examining %s...") % dir
 720             entries = Dir.entries(dir).sort
 721             #- populate config in case of gen_config, add new files in case of merge_config
 722             for file in entries
 723                 if file =~ /['"\[\]]/
 724                     msg 1, _("Ignoring %s, contains one of forbidden characters: '\"[]") % "#{dir}/#{file}"
 725                 else
 726                     type = entry2type(file)
 727                     if type && !xmldir.elements["#{type}[@filename='#{utf8(file)}']"]
 728                         #- hack: don't run identify (which is slow) if format only contains default %t
 729                         if $commentsformat && type == 'image' && $commentsformat != '%t'
 730                             comment = utf8(`identify -format "#{$commentsformat}" '#{dir}/#{file}'`.chomp.sub(/\.$/, ''))
 731                         else
 732                             comment = utf8cut(file.sub(/\.[^\.]+$/, ''), 18)
 733                         end
 734                         xmldir.add_element type, { "filename" => utf8(file), "caption" => comment }
 735                     end
 736                 end
 737             end
 738             if $mode != 'gen_config'
 739                 #- cleanup removed files from config and reread entries from config to get proper ordering
 740                 entries = []
 741                 xmldir.elements.each { |element|
 742                     fullpath = "#{dir}/#{from_utf8(element.attributes['filename'])}"
 743                     if %w(image video).include?(element.name)
 744                         if !File.readable?(fullpath)
 745                             msg 1, _("Config merge: removing %s from config; use the backup file to retrieve caption info if this was a mistake") % fullpath
 746                             xmldir.delete(element)
 747                         elsif !element.attributes['deleted']
 748                             entries << from_utf8(element.attributes['filename'])
 749                         end
 750                     end
 751                 }
 752                 #- if there is no more elements here, there is no album here anymore
 753                 if !xmldir.child_byname_notattr('image', 'deleted') && !xmldir.child_byname_notattr('video', 'deleted')
 754                     xmldir.delete_attribute('thumbnails-caption')
 755                     xmldir.delete_attribute('thumbnails-captionfile')
 756                 end
 757             end
 758         end
 759         images = entries.find_all { |e| entry2type(e) == 'image' }
 760         msg 3, _("\t%s images") % images.length
 761         videos = entries.find_all { |e| entry2type(e) == 'video' }
 762         msg 3, _("\t%s videos") % videos.length
 763         info("walking: #{dir}|#{$source}, #{images.length + videos.length} elements")
 764
 765         system("mkdir -p '#{dest_dir}'")
 766
 767         #- generate .htaccess file
 768         if !$forgui
 769             ios = File.open("#{dest_dir}/.htaccess", "w")
 770             ios.write("AddCharset UTF-8 .html\n")
 771             if auth_user_file = xmldir.attributes['password-protect']
 772                 msg 3, _("\tgenerating password protection file #{dest_dir}/.htaccess")
 773                 ios.write("AuthType Basic\nAuthName \"protected area\"\nAuthUserFile #{auth_user_file}\nrequire valid-user\n")
 774             end
 775             ios.close
 776         end
 777
 778         #- pass through if there are no images and videos
 779         if images.size == 0 && videos.size == 0
 780             if !$forgui
 781                 #- cleanup old images/videos, especially if this directory contained images/videos previously.
 782                 themestuff = Dir.entries("#{$FPATH}/themes/#{$theme}").
 783                                 find_all { |e| !%w(. .. skeleton_image.html skeleton_thumbnails.html skeleton_index.html metadata CVS).include?(e) }
 784                 if $mode != 'gen_config'
 785                     rightful_images = [ '.htaccess' ]
 786                     if xmldir.attributes['thumbnails-caption']
 787                         rightful_images << 'thumbnails-thumbnail.jpg'
 788                     end
 789                     xmldir.elements.each('dir') { |child|
 790                         if child.attributes['deleted']
 791                             next
 792                         end
 793                         subdir = make_dest_filename(from_utf8(File.basename(child.attributes['path'])))
 794                         rightful_images << "thumbnails-#{subdir}.jpg"
 795                     }
 796                     to_del = Dir.entries(dest_dir).find_all { |e| !File.directory?("#{dest_dir}/#{e}") && !rightful_images.include?(e) } - themestuff
 797                     if to_del.size > 0
 798                         system("rm -f " + to_del.collect { |e| "'#{dest_dir}/#{e}'" }.join(' '))
 799                     end
 800                 end
 801
 802                 #- copy any resource file that goes with the theme (css, images..)
 803                 themestuff.each { |entry|
 804                     if !File.exists?("#{dest_dir}/#{entry}")
 805                         psys("cp '#{$FPATH}/themes/#{$theme}/#{entry}' '#{dest_dir}'")
 806                     end
 807                 }
 808             end
 809             next
 810         end
 811
 812         msg 2, _("Outputting in %s...") % dest_dir
 813
 814         #- populate data structure with sizes from theme
 815         for sizeobj in $images_size
 816             fullscreen_images ||= {}
 817             fullscreen_images[sizeobj['name']] = []
 818             thumbnail_images ||= {}
 819             thumbnail_images[sizeobj['name']] = []
 820             thumbnail_videos ||= {}
 821             thumbnail_videos[sizeobj['name']] = []
 822         end
 823         #- a special dummy size to keep 'references' to thumbnails in case of panorama, because the GUI will use the regular thumbnails
 824         thumbnail_images['dont-delete-file-for-gui'] = []
 825         if $limit_sizes =~ /original/
 826             fullscreen_images['original'] = []
 827         end
 828
 829         images.size >= 1 and msg 3, _("\tcreating images thumbnails...")
 830
 831         #- create thumbnails for images
 832         images.each { |img|
 833             info("processing element")
 834             base_dest_img = dest_dir + '/' + make_dest_filename(img.sub(/\.[^\.]+$/, ''))
 835             if $forgui
 836                 thumbnail_dest_img = base_dest_img + "-#{$default_size['thumbnails']}.jpg"
 837                 gen_thumbnails_element("#{dir}/#{img}", xmldir, true, [ { 'filename' => thumbnail_dest_img, 'size' => $default_size['thumbnails'] } ])
 838             else
 839                 todo = []
 840                 elem = xmldir.elements["image[@filename='#{utf8(img)}']"]
 841                 for sizeobj in $images_size
 842                     size_fullscreen = sizeobj['fullscreen']
 843                     size_thumbnails = sizeobj['thumbnails']
 844                     fullscreen_dest_img = base_dest_img + "-#{size_fullscreen}.jpg"
 845                     fullscreen_images[sizeobj['name']] << File.basename(fullscreen_dest_img)
 846                     todo << { 'filename' => fullscreen_dest_img, 'size' => size_fullscreen }
 847                     if pano = pano_amount(elem)
 848                         thumbnail_images['dont-delete-file-for-gui'] << File.basename(base_dest_img + "-#{size_thumbnails}.jpg")
 849                         size_thumbnails = size_thumbnails.sub(/(\d+)/) { ($1.to_i * pano).to_i }
 850                     end
 851                     thumbnail_dest_img = base_dest_img + "-#{size_thumbnails}.jpg"
 852                     thumbnail_images[sizeobj['name']] << File.basename(thumbnail_dest_img)
 853                     todo << { 'filename' => thumbnail_dest_img,  'size' => size_thumbnails }
 854                 end
 855                 gen_thumbnails_element("#{dir}/#{img}", xmldir, true, todo)
 856                 if $limit_sizes =~ /original/
 857                     fullscreen_images['original'] << img
 858                 end
 859                 destimg = "#{dest_dir}/#{img}"
 860                 if $limit_sizes =~ /original/ && !File.exists?(destimg)
 861                     if $hardlinks_ok
 862                         if ! sys("ln '#{dir}/#{img}' '#{destimg}'")
 863                             $hardlinks_ok = false
 864                         end
 865                     end
 866                     if ! $hardlinks_ok
 867                         psys("cp '#{dir}/#{img}' '#{destimg}'")
 868                     end
 869                 end
 870             end
 871         }
 872
 873         videos.size >= 1 and msg 3, _("\tcreating videos thumbnails...")
 874
 875         #- create thumbnails for videos
 876         videos.each { |video|
 877             info("processing element")
 878             if $forgui
 879                 thumbnail_dest_img = dest_dir + '/' + make_dest_filename(video.sub(/\.[^\.]+$/, '')) + "-#{$default_size['thumbnails']}.jpg"
 880                 gen_thumbnails_element("#{dir}/#{video}", xmldir, true, [ { 'filename' => thumbnail_dest_img, 'size' => $default_size['thumbnails'] } ])
 881             else
 882                 todo = []
 883                 for sizeobj in $images_size
 884                     size_thumbnails = sizeobj['thumbnails']
 885                     thumbnail_dest_img = dest_dir + '/' + make_dest_filename(video.sub(/\.[^\.]+$/, '')) + "-#{size_thumbnails}.jpg"
 886                     thumbnail_videos[sizeobj['name']] << File.basename(thumbnail_dest_img)
 887                     todo << { 'filename' => thumbnail_dest_img, 'size' => size_thumbnails }
 888                 end
 889                 gen_thumbnails_element("#{dir}/#{video}", xmldir, true, todo)
 890             end
 891             destvideo = "#{dest_dir}/#{video}"
 892             if !File.exists?(destvideo)
 893                 if $hardlinks_ok
 894                     if ! sys("ln '#{dir}/#{video}' '#{destvideo}'")
 895                         $hardlinks_ok = false
 896                     end
 897                 end
 898                 if ! $hardlinks_ok
 899                     psys("cp '#{dir}/#{video}' '#{destvideo}'")
 900                 end
 901             end
 902         }
 903
 904         if !$forgui
 905             themestuff = Dir.entries("#{$FPATH}/themes/#{$theme}").
 906                              find_all { |e| !%w(. .. skeleton_image.html skeleton_thumbnails.html skeleton_index.html metadata CVS).include?(e) }
 907
 908             #- cleanup old images/videos (for when removing elements or sizes)
 909             all_elements = fullscreen_images.collect { |e| e[1] }.flatten.
 910                      concat(thumbnail_images.collect { |e| e[1] }.flatten).
 911                      concat(thumbnail_videos.collect { |e| e[1] }.flatten).
 912                      concat(videos).
 913                      push('.htaccess')
 914             to_del = Dir.entries(dest_dir).find_all { |e| !File.directory?("#{dest_dir}/#{e}") && !all_elements.include?(e) && e !~ /^thumbnails-\w+\.jpg/ } - themestuff
 915             if to_del.size > 0
 916                 msg 3, _("\tcleaning up: #{to_del.join(', ')}")
 917                 system("rm -f " + to_del.collect { |e| "#{dest_dir}/#{e}" }.join(' '))
 918             end
 919
 920             #- copy any resource file that goes with the theme (css, images..)
 921             themestuff.each { |entry|
 922                 if !File.exists?("#{dest_dir}/#{entry}")
 923                     psys("cp '#{$FPATH}/themes/#{$theme}/#{entry}' '#{dest_dir}'")
 924                 end
 925             }
 926
 927             msg 3, _("\tgenerating HTML pages...")
 928             #- fixup max per page
 929             if $N_per_page && $N_per_row
 930                 $N_per_page = $N_per_page.to_i / $N_per_row.to_i * $N_per_row.to_i
 931             end
 932
 933             #- generate thumbnails*.html (page with thumbnails)
 934             image2thumbnailpage4js = []
 935             for sizeobj in $images_size
 936                 info("processing size")
 937                 html = $html_thumbnails.collect { |l| l.clone }
 938                 iterations = {}
 939                 for i in html
 940                     i.sub!(/~~title~~/,
 941                            xmldir.attributes['thumbnails-caption'] || utf8(File.basename(dir)))
 942                     discover_iterations(iterations, i)
 943                 end
 944                 all_pages = []
 945                 html_thumbnails = ''
 946                 html_thumbnails_nojs = ''
 947                 counter = 0
 948                 pagecount = 0
 949                 reset_iterations(iterations)
 950                 #- preprocess the @filename->elem, rexml is very slow with that; we dramatically improve this part of the processing
 951                 optfilename = {}
 952                 xmldir.elements.each('image') { |elem|
 953                     optfilename[elem.attributes['filename']] = elem
 954                 }
 955                 for file in entries
 956                     type = images.include?(file) ? 'image' : videos.include?(file) ? 'video' : nil
 957                     if type
 958                         if type == 'image' && elem = optfilename[utf8(file)]
 959                             if pano = pano_amount(elem)
 960                                 html_elem = run_iterations(iterations, pano)
 961                                 counter += count = pano.ceil
 962                                 html_elem.gsub!(/~~colspan~~/) { "colspan=\"#{count}\"" }
 963                             else
 964                                 html_elem = run_iterations(iterations, 1)
 965                                 counter += 1
 966                                 html_elem.gsub!(/~~colspan~~/, '')
 967                             end
 968                         else
 969                             html_elem = run_iterations(iterations, 1)
 970                             counter += 1
 971                             html_elem.gsub!(/~~colspan~~/, '')
 972                         end
 973                         if type == 'image'
 974                             index = images.index(file)
 975                             html_elem.gsub!(/~~caption_iteration~~/,
 976                                             find_caption_value(xmldir, images[index]) || utf8(images[index]))
 977                             html_elem.gsub!(/~~ifimage\?~~(.+?)~~fi~~/) { $1 }
 978                             html_elem.gsub!(/~~ifvideo\?~~(.+?)~~fi~~/, '')
 979                         elsif type == 'video'
 980                             index = videos.index(file)
 981                             if File.exists?("#{dest_dir}/#{thumbnail_videos[sizeobj['name']][index]}")
 982                                 html_elem.gsub!(/~~image_iteration~~/,
 983                                                 '<a href="' + videos[index] + '">' + img_element("#{dest_dir}/#{thumbnail_videos[sizeobj['name']][index]}") + '</a>')
 984                             else
 985                                 html_elem.gsub!(/~~image_iteration~~/,
 986                                                 '<a href="' + videos[index] + '">' + utf8(_("(no preview)")) + '</a>')
 987                             end
 988                             html_elem.gsub!(/~~caption_iteration~~/, find_caption_value(xmldir, videos[index]) || utf8(videos[index]))
 989                             html_elem.gsub!(/~~ifimage\?~~(.+?)~~fi~~/, '')
 990                             html_elem.gsub!(/~~ifvideo\?~~(.+?)~~fi~~/) { $1 }
 991                         end
 992                         html_thumbnails += html_elem
 993                         html_thumbnails_nojs += html_elem
 994                         if type == 'image'
 995                             html_thumbnails.gsub!(/~~image_iteration~~/,
 996                                                   '<a href="image-' + size2js(sizeobj['name']) + '.html#current=' + fullscreen_images[sizeobj['name']][index] +
 997                                                       '" name="' + fullscreen_images[sizeobj['name']][index] + '">' +
 998                                                       img_element("#{dest_dir}/#{thumbnail_images[sizeobj['name']][index]}") + '</a>')
 999                             html_thumbnails_nojs.gsub!(/~~image_iteration~~/,
1000                                                        '<a href="' + fullscreen_images[sizeobj['name']][index] + '" name="' + fullscreen_images[sizeobj['name']][index] + '">' +
1001                                                        img_element("#{dest_dir}/#{thumbnail_images[sizeobj['name']][index]}") + '</a>')
1002                             #- remember in which thumbnails page is this element, for image->thumbnail link
1003                             if sizeobj == $images_size[0]
1004                                 image2thumbnailpage4js << pagecount
1005                             end
1006                         end
1007
1008                         if counter == $N_per_page
1009                             html_thumbnails      += close_iterations(iterations)
1010                             html_thumbnails_nojs += close_iterations(iterations)
1011                             all_pages << [ html_thumbnails, html_thumbnails_nojs ]
1012                             html_thumbnails = ''
1013                             html_thumbnails_nojs = ''
1014                             counter = 0
1015                             reset_iterations(iterations)
1016                             pagecount += 1
1017                         end
1018                     end
1019                 end
1020                 if counter > 0
1021                     html_thumbnails      += close_iterations(iterations)
1022                     html_thumbnails_nojs += close_iterations(iterations)
1023                     all_pages << [ html_thumbnails, html_thumbnails_nojs ]
1024                 end
1025                 for i in html
1026                     i.gsub!(/~~theme~~/, $theme)
1027                     i.gsub!(/~~current_size~~/, sizeobj['name'])
1028                     i.gsub!(/~~current_size_js~~/, size2js(sizeobj['name']))
1029                     i.gsub!(/~~madewith~~/, $madewith || '')
1030                     i.gsub!(/~~indexlink~~/, $indexlink || '')
1031                     substitute_navigation(i, xmldir)
1032                 end
1033                 html_nojs = html.collect { |l| l.clone }
1034                 pagecount = 0
1035                 for page in all_pages
1036                     html_thumbnails, html_thumbnails_nojs = page
1037                     final_html = html.collect { |l| l.clone }
1038                     mstuff = utf8(_("Pages: %s") % (pagecount > 0 ? "<a href=\"thumbnails-#{size2js(sizeobj['name'])}%nojs-#{pagecount - 1}.html\">" + _("<- Previous") + "</a> " : '') +
1039                                                    all_pages.collect_with_index { |p,idx| page == p ? idx + 1 : "<a href=\"thumbnails-#{size2js(sizeobj['name'])}%nojs-#{idx}.html\">#{idx + 1}</a>" }.join(', ') +
1040                                                    (pagecount < all_pages.size - 1 ? " <a href=\"thumbnails-#{size2js(sizeobj['name'])}%nojs-#{pagecount + 1}.html\">" + _("Next ->") + "</a> " : ''))
1041                     for i in final_html
1042                         i.sub!(/~~run_slideshow~~/,
1043                                images.size <= 1 ? '' : '<a href="image-' + size2js(sizeobj['name']) + '.html#run_slideshow=1">' + utf8(_("Run slideshow!"))+'</a>')
1044                         i.sub!(/~~thumbnails~~/, html_thumbnails)
1045                         if all_pages.size == 1
1046                             i.gsub!(/~~ifmultiplepages\?~~.*~~fi~~/, '')
1047                         else
1048                             i.gsub!(/~~ifmultiplepages\?~~(.+?)~~fi~~/) { $1 }
1049                             i.gsub!(/~~multiplepagesstuff~~/, mstuff.gsub('%nojs', ''))
1050                         end
1051                         substitute_html_sizes(i, sizeobj, 'thumbnails', "-#{pagecount}")
1052                     end
1053                     ios = File.open("#{dest_dir}/thumbnails-#{size2js(sizeobj['name'])}-#{pagecount}.html", "w")
1054                     ios.write(final_html)
1055                     ios.close
1056                     final_html_nojs = html_nojs.collect { |l| l.clone }
1057                     for i in final_html_nojs
1058                         i.sub!(/~~run_slideshow~~/, utf8(_("<i>Click on an image to view it larger</i>")))
1059                         i.sub!(/~~thumbnails~~/, html_thumbnails_nojs)
1060                         if all_pages.size == 1
1061                             i.gsub!(/~~ifmultiplepages\?~~.*~~fi~~/, '')
1062                         else
1063                             i.gsub!(/~~ifmultiplepages\?~~(.+?)~~fi~~/) { $1 }
1064                             i.gsub!(/~~multiplepagesstuff~~/, mstuff.gsub('%nojs', '-nojs'))
1065                         end
1066                         substitute_html_sizes(i, sizeobj, 'thumbnails', "-nojs-#{pagecount}")
1067                     end
1068                     ios = File.open("#{dest_dir}/thumbnails-#{size2js(sizeobj['name'])}-nojs-#{pagecount}.html", "w")
1069                     ios.write(final_html_nojs)
1070                     ios.close
1071                     pagecount += 1
1072                 end
1073             end
1074
1075             info("finished processing sizes")
1076
1077             #- generate "main" thumbnails.html page that will reload to correct size thanks to cookie
1078             ios = File.open("#{dest_dir}/thumbnails.html", "w")
1079             ios.write(html_reload_to_thumbnails)
1080             ios.close
1081
1082             #- generate image.html (page with fullscreen images)
1083             if images.size > 0
1084                 captions4js = find_captions(xmldir, images).collect { |e| e ? '"' + e.gsub('"', '\"') + '"' : '""' }.join(', ')
1085                 thumbnailspage4js = image2thumbnailpage4js.collect { |e| "\"#{e}\"" }.join(', ')
1086                 for sizeobj in $images_size
1087                     html = $html_images.collect { |l| l.clone }
1088                     images4js = fullscreen_images[sizeobj['name']].collect { |e| "\"#{e}\"" }.join(', ')
1089                     otherimages4js = ''
1090                     othersizes = []
1091                     for sizeobj2 in all_images_sizes
1092                         if sizeobj != sizeobj2
1093                             otherimages4js += "var images_#{size2js(sizeobj2['name'])} = new Array(" + fullscreen_images[sizeobj2['name']].collect { |e| "\"#{e}\"" }.join(', ') + ")\n"
1094                             othersizes << "\"#{size2js(sizeobj2['name'])}\""
1095                         end
1096                     end
1097                     for i in html
1098                         i.gsub!(/~~images~~/, images4js)
1099                         i.gsub!(/~~other_images~~/, otherimages4js)
1100                         i.gsub!(/~~thumbnailspages~~/, thumbnailspage4js)
1101                         i.gsub!(/~~other_sizes~~/, othersizes.join(', '))
1102                         i.gsub!(/~~captions~~/, captions4js)
1103                         i.gsub!(/~~title~~/, xmldir.attributes['thumbnails-caption'] || utf8(File.basename(dir)))
1104                         i.gsub!(/~~thumbnails~~/, '<a href="thumbnails-' + size2js(sizeobj['name']) + '.html" id="thumbnails">' + utf8(_('return to thumbnails')) + '</a>')
1105                         i.gsub!(/~~theme~~/, $theme)
1106                         i.gsub!(/~~current_size~~/, size2js(sizeobj['name']))
1107                         i.gsub!(/~~madewith~~/, $madewith || '')
1108                         i.gsub!(/~~indexlink~~/, $indexlink || '')
1109                         substitute_html_sizes(i, sizeobj, 'image', '')
1110                         substitute_navigation(i, xmldir)
1111                     end
1112                     ios = File.open("#{dest_dir}/image-#{size2js(sizeobj['name'])}.html", "w")
1113                     ios.write(html)
1114                     ios.close
1115                 end
1116             end
1117         end
1118     }
1119
1120     msg 3, ''
1121
1122     #- add attributes to <dir..> elements needing so
1123     if $mode != 'use_config'
1124         msg 3, _("\tfixating configuration file...")
1125         $xmldoc.elements.each('//dir') { |element|
1126             path = captionpath = element.attributes['path']
1127             descendant_element = element.elements['descendant::image'] || element.elements['descendant::video']
1128             if !descendant_element
1129                 msg 3, _("\t\tremoving %s, no element in it") % path
1130                 element.remove  #- means we have a directory with nothing interesting in it
1131             else
1132                 captionfile = "#{descendant_element.parent.attributes['path']}/#{descendant_element.attributes['filename']}"
1133                 basename = File.basename(path)
1134                 if element.elements['dir']
1135                     if !element.attributes['subdirs-caption']
1136                         element.add_attribute('subdirs-caption', basename)
1137                     end
1138                     if !element.attributes['subdirs-captionfile']
1139                         element.add_attribute('subdirs-captionfile', captionfile)
1140                     end
1141                 end
1142                 if element.child_byname_notattr('image', 'deleted') || element.child_byname_notattr('video', 'deleted')
1143                     if !element.attributes['thumbnails-caption']
1144                         element.add_attribute('thumbnails-caption', basename)
1145                     end
1146                     if !element.attributes['thumbnails-captionfile']
1147                         element.add_attribute('thumbnails-captionfile', captionfile)
1148                     end
1149                 end
1150             end
1151         }
1152     end
1153
1154     #- write down to disk config if necessary
1155     if $config_writeto
1156         ios = File.open($config_writeto, "w")
1157         $xmldoc.write(ios, 0)
1158         ios.close
1159     end
1160
1161     if $forgui
1162         msg 3, _(" completed necessary stuff for GUI, exiting.")
1163         return
1164     end
1165
1166     #- second pass to create index.html files and previous/next links
1167     info("creating index.html")
1168     msg 3, _("\trescanning directories to generate all 'index.html' files...")
1169
1170     #- recompute the memoization because elements mights have been removed (the ones with no element in them)
1171     optxpath = {}
1172     $xmldoc.elements.each('//dir') { |elem|
1173         optxpath[elem.attributes['path']] = elem
1174     }
1175
1176     examined_dirs.each { |dir|
1177         info("index.html: #{dir}|#{$source}")
1178
1179         xmldir = optxpath[utf8(dir)]
1180         if !xmldir || (xmldir.attributes['already-generated'] && !$force) || xmldir.attributes['deleted']
1181             next
1182         end
1183         dest_dir = make_dest_filename(dir.sub(/^#{Regexp.quote($source)}/, $dest))
1184
1185         previous_album = find_previous_album(xmldir)
1186         next_album = find_next_album(xmldir)
1187
1188         if xmldir.elements['dir']
1189             html = $html_index.collect { |l| l.clone }
1190             iterations = {}
1191             for i in html
1192                 caption = xmldir.attributes['subdirs-caption']
1193                 i.gsub!(/~~title~~/, caption)
1194                 substitute_navigation(i, xmldir)
1195                 discover_iterations(iterations, i)
1196             end
1197
1198             html_index = ''
1199             reset_iterations(iterations)
1200
1201             #- deal with "current" album (directs to "thumbnails" page)
1202             if xmldir.attributes['thumbnails-caption']
1203                 thumbnail = "#{dest_dir}/thumbnails-thumbnail.jpg"
1204                 gen_thumbnails_subdir(from_utf8(xmldir.attributes['thumbnails-captionfile']), xmldir, false,
1205                                       [ { 'filename' => thumbnail, 'size' => $albums_thumbnail_size } ], 'thumbnails')
1206                 html_index += run_iterations(iterations, 1)
1207                 html_index.gsub!(/~~image_iteration~~/, "<a href=\"thumbnails.html\">" + img_element(thumbnail) + '</a>')
1208                 html_index.gsub!(/~~caption_iteration~~/, xmldir.attributes['thumbnails-caption'])
1209             end
1210
1211             #- deal with sub-albums (direct to subdirs/index.html pages)
1212             xmldir.elements.each('dir') { |child|
1213                 if child.attributes['deleted']
1214                     next
1215                 end
1216                 subdir = make_dest_filename(from_utf8(File.basename(child.attributes['path'])))
1217                 thumbnail = "#{dest_dir}/thumbnails-#{subdir}.jpg"
1218                 html_index += run_iterations(iterations, 1)
1219                 captionfile, caption = find_subalbum_caption_info(child)
1220                 gen_thumbnails_subdir(captionfile, child, false,
1221                                       [ { 'filename' => thumbnail, 'size' => $albums_thumbnail_size } ], find_subalbum_info_type(child))
1222                 html_index.gsub!(/~~caption_iteration~~/, caption)
1223                 html_index.gsub!(/~~image_iteration~~/, "<a href=\"#{subdir}/index.html\">" + img_element(thumbnail) + '</a>')
1224             }
1225
1226             html_index += close_iterations(iterations)
1227
1228             for i in html
1229                 i.gsub!(/~~thumbnails~~/, html_index)
1230                 i.gsub!(/~~madewith~~/, $madewith || '')
1231                 i.gsub!(/~~indexlink~~/, $indexlink || '')
1232             end
1233
1234         else
1235             html = html_reload_to_thumbnails
1236         end
1237
1238         ios = File.open("#{dest_dir}/index.html", "w")
1239         ios.write(html)
1240         ios.close
1241
1242         #- substitute multiple "return to albums", previous/next correctly
1243         #- the following two statements are dramatical optimizations to executing for each substInFile callback
1244         dirpresent = xmldir.elements['dir']
1245         parentname = xmldir.parent.name
1246         if xmldir.child_byname_notattr('image', 'deleted') || xmldir.child_byname_notattr('video', 'deleted')
1247             for suffix in [ '', '-nojs' ]
1248                 for sizeobj in $images_size
1249                     Dir.glob("#{dest_dir}/thumbnails-#{size2js(sizeobj['name'])}#{suffix}-*.html") do |file|
1250                         substInFile(file) { |line|
1251                             sub_previous_next_album(previous_album, next_album, line)
1252                             if dirpresent
1253                                 line.sub!(/~~return_to_albums~~/, '<a href="index.html">' + utf8(_('return to albums')) + '</a>')
1254                             else
1255                                 if parentname == 'dir'
1256                                     line.sub!(/~~return_to_albums~~/, '<a href="../index.html">' + utf8(_('return to albums')) + '</a>')
1257                                 else
1258                                     line.sub!(/~~return_to_albums~~/, '')
1259                                 end
1260                             end
1261                             line
1262                         }
1263                         if suffix == '' && xmldir.child_byname_notattr('image', 'deleted')
1264                             substInFile("#{dest_dir}/image-#{size2js(sizeobj['name'])}.html") { |line|
1265                                 sub_previous_next_album(previous_album, next_album, line)
1266                             }
1267                         end
1268                     end
1269                 end
1270             end
1271         end
1272     }
1273
1274     msg 3, _(" all done.")
1275 end
1276
1277 handle_options
1278 read_config
1279 check_installation
1280
1281 build_html_skeletons
1282
1283 walk_source_dir