#!/usr/bin/env ruby

require 'tmpdir'

require "fileutils"

LOCK_FILE_PATH = File.join(Dir.tmpdir, "gitorious_graphgenerator_lockfile")

if File.exist?(LOCK_FILE_PATH)

    $stderr.puts "Lockfile '#{LOCK_FILE_PATH}' exists!"

    exit(1)

end

ENV["PATH"] = "/usr/local/bin/:/opt/local/bin:#{ENV["PATH"]}"

FileUtils.touch(LOCK_FILE_PATH)

ENV["RAILS_ENV"] ||= "production"

require File.dirname(__FILE__) + "/../config/environment"

log = Logger.new(File.join(RAILS_ROOT, "log", "graph_generator.log"))

log.formatter = Logger::Formatter.new

log.level = Logger::INFO

log.formatter.datetime_format = "%Y-%m-%d %H:%M:%S"

STATUS_FILE = File.join(RAILS_ROOT, "tmp", "graph_generator.status")

update_status = !File.exist?(STATUS_FILE)

tmpdir = File.join(RAILS_ROOT, "tmp", "graph_generator")

if !File.directory?(tmpdir)

  FileUtils.mkdir_p(tmpdir)

elsif !update_status

  mtime = File.mtime(STATUS_FILE).utc

  days = (Time.now.utc - mtime) / (3600*24)

  if days >= 5

    log.info "Cleaning '#{tmpdir}'..."

    Dir.glob(File.join(tmpdir, "*.status")) do |file|

      FileUtils.rm(file)

    end

    update_status = true

  end

end

if update_status

  FileUtils.touch(STATUS_FILE)

end

log.info "Starting graph generation run..."

Repository.find(:all).each_with_index do |repository, index|

	begin

		Gitorious::Graphs::Builder.generate_all_for(repository)

		if index % 5 == 0

			# GC every five cycles

			GC.start

		end

		sleep(0.1)

	rescue => e

		log.fatal "Error generating graphs for repo##{repository.id}"

    exception_backtrace_string = "#{e.class.name} #{e.message}\n#{e.backtrace.join("\n  ")}"

    log.fatal exception_backtrace_string

    $stderr.puts exception_backtrace_string

#    exit(1)		

	end

end

log.info "Done with graph generation run"

FileUtils.rm(LOCK_FILE_PATH)