#!/usr/bin/env ruby -KU

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

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

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

action = ARGV[0]

abort("Need create_repos or load_readmes command") unless action

def failure_msg(msg)

  $stderr.puts "\e[1;31m#{msg}\e[0m"

end

project_count = Project.count

Project.find(:all).each_with_index do |project, index|

  case action

  when "create_repos"

    next unless project.wiki_repository.blank?

    project.transaction do

      puts "Creating wiki repo for #{project.slug} (#{index+1}/#{project_count})"

      wiki_repo = Repository.new({

        :name => project.slug + Repository::WIKI_NAME_SUFFIX,

        :kind => Repository::KIND_WIKI,

        :mainline => false,

        :user => project.user,

      })

      project.wiki_repository = wiki_repo

      wiki_repo.save!

    end

  when "load_readmes"

    wiki_repo = project.wiki_repository

    git_dir = File.join(GitoriousConfig["repository_base_path"], project.mainline_repository.gitdir)

    unless File.exist?(git_dir)

      failure_msg "Skipping #{project.slug}, no mainline repo, wtf?  (#{index+1}/#{project_count})"

      next

    end

    tree = `env GIT_DIR=#{git_dir} git ls-tree HEAD`

    if tree =~ /^\d+ (.+)\t(readme(\..*)?)$/i

      begin

        sha = $1

        readmename = $2

        puts "Creating Home in #{project.slug} based on #{$2} (#{$1}) (#{index+1}/#{project_count})"

        p = Page.find("Home", wiki_repo.git)

        readmeblob = project.mainline_repository.git.tree / readmename

        p.content = readmeblob.data

        p.user = project.user

        p.save

      rescue => e

        failure_msg "FAIL: #{project.slug} #{e.class.name}: #{e.message}"

      end

    end

  end

end