root/lang/ruby/misc/periodic_forwarder

#! /usr/bin/env ruby
# -*- mode: ruby; coding: mule-utf-8 -*-

require 'pathname'
require 'optparse'
#require 'const_loader'

#
# = periodic-forwarder
#
# スクリプトに置かれているディレクトリに応じて、該当する periodic なディ
# レクトリを指定パス内から探し出し、これを run-parts あるいは periodic
# で実行する
#
# ex)
#
#  /etc/cron.daily/THIS_SCRIPT -> 
#     periodic #{BASE_DIR}/SITE/#{FILLING_PATH}/daily/
#
# THIS_SCRIPT は link で構わない。APPEND を与えればリダイレクトやパイプに
# よって結果を処理することも可能。
#
# == requirements
# `which` , and `periodic` or `run-parts` command
# 

Version = 0.1

class PeriodicForwarder
#  include ConstLoader

  def initialize
    @opt  = {}

    @conf = {
      'append'       => nil,
      'base_dir'     => nil,
      'filling_path' => nil,
    }

    @periodic        = nil
    @periodic_runner = periodic_runner
  end

  def run()
    begin
      load_conf
    rescue
    end

    opts().parse!( ARGV )
    @period = period( parent_dir )
    exec( collect_periodic_dirs )
  end

  #
  # periodic / run-parts の判別
  #
  def periodic_runner
    return system( 'which run-parts > /dev/null' ) ? 'run-parts' : 'periodic'
  end

  #
  # 指定ディレクトリの親ディレクトリ
  #
  # 無指定の場合はスクリプト自身の置かれているディレクトリ
  #
  def parent_dir( dir = nil )
    dir = ( dir.nil? ) ? File.expand_path( __FILE__ ) : dir

    return Pathname.new( dir ).parent.basename.to_s
  end

  #
  # 指定ディレクトリの意味する期間
  #
  def period( dir )
    period = nil

    if ( /cron\.(.*)/ =~ dir )
      period = $1
    else
      period = dir
    end

    return period
  end

  #
  # base_dir 以下で @periodic と同じ名前を持つディレクトリを集める
  #
  def collect_periodic_dirs( base_dir = nil )
    dir = ( base_dir || @conf['base_dir'] )

    if ( dir.nil? )
      abort( 'base_dir is not defined.' )
    end

    dirs = []
    Dir.foreach( dir ) { |e|
      target = File.expand_path( File.join( dir, e.to_s, @conf['filling_path'].to_s, @period.to_s ) ) + '/'
      if ( File.exist?( target ) and File.directory?( target ) )
        dirs.push( target )
      end
    }

    return dirs
  end

  #
  # 対象ディレクトリ群に対して片っ端から @periodic を実行
  #
  # [Param] dirs
  #
  def exec( dirs )
    dirs.each { |dir|
      cmd = "#{@periodic_runner} #{dir}"
      if ( @conf['append'] ) 
        cmd += " #{@conf['append']}" 
      end
      system( cmd )
    }
  end

  #
  # オプション定義
  #
  def opts
    opt = OptionParser.new()
#     opt.on( '-c', 'CONFIG FILE' ) { |file|
#       if ( File.exist?( file ) )
#         @opt['conf'] = file
#       end
#     }
    opt.on( '-a', '--append=APPEND' ) { |append|
      @conf['append'] = append
    }
    opt.on( '-b', '--base-dir=BASE_DIR' ) { |base_dir|
      if ( File.exist?( base_dir ) )
        @conf['base_dir'] = base_dir
      end
    }
    opt.on( '-f', '--filling-path=FILLING_PATH' ) { |filling_path|
      if ( File.exist?( filling_path ) )
        @conf['filling_path'] = filling_path
      end
    }

    return opt
  end

end # of class App

if ( __FILE__ == $0 )
  begin
    load( File.basename( __FILE__ ) + '.conf' )
  rescue LoadError
  end
  app = PeriodicForwarder.new
  app.run()
end