#!/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

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)