require 'logger'
require 'yaml'
require "octokit"
require "bicho"
require  File.join(File.expand_path("..", File.dirname(__FILE__)),"db","database")

module Nailed
  LOGGER = Logger.new(File.join(File.expand_path("..", File.dirname(__FILE__)),"log","nailed.log"))
  CONFIG_FILE =  File.join(File.expand_path("..", File.dirname(__FILE__)),"config","products.yml")
  PRODUCTS = YAML.load_file(CONFIG_FILE)

  class Bugzilla
    def initialize
      Bicho.client = Bicho::Client.new(Nailed::PRODUCTS["bugzilla"]["url"])
    end

    def get_bugs
      Nailed::PRODUCTS["products"].each do |product,values|
        values["versions"].each do |version|
          Bicho::Bug.where(:product => version).each do |bug|
            attributes = {
              :bug_id => bug.id,
              :summary => bug.summary,
              :status => bug.status,
              :is_open => bug.is_open,
              :product_name => bug.product,
              :component => bug.component,
              :severity => bug.severity,
              :priority => bug.priority,
              :whiteboard => bug.whiteboard,
              :assigned_to => bug.assigned_to,
              :creation_time => "#{bug.creation_time.to_date}T#{bug.creation_time.hour}:#{bug.creation_time.min}:#{bug.creation_time.sec}+00:00",
              :last_change_time => "#{bug.last_change_time.to_date}T#{bug.last_change_time.hour}:#{bug.last_change_time.min}:#{bug.last_change_time.sec}+00:00",
              :url => bug.url.gsub(/novell.com\//,'suse.com/show_bug.cgi?id=')
            }

            db_handler = (Bugreport.get(bug.id) || Bugreport.new).update(attributes)
          end
        end
      end
    end

    def write_bug_trends
      Nailed::PRODUCTS["products"].each do |product,values|
        values["versions"].each do |version|
          open = Bugreport.count(:is_open => true, :product_name => version)
          fixed = Bugreport.count(:status => "VERIFIED", :product_name => version) + \
                  Bugreport.count(:status => "RESOLVED", :product_name => version)
          db_handler = Bugtrend.first_or_create(
                       :time => Time.new.strftime("%Y-%m-%d %H:%M:%S"),
                       :open => open,
                       :fixed => fixed,
                       :product_name => version
                       )

          Nailed.save_state(db_handler)
        end
      end
    end

    def write_l3_trends
      open = Bugreport.count(:whiteboard.like => "%openL3%", :is_open => true)
      db_handler = L3Trend.first_or_create(
                   :time => Time.new.strftime("%Y-%m-%d %H:%M:%S"),
                   :open => open
                   )

      Nailed.save_state(db_handler)
    end
  end

  class Github
    attr_reader :client

    def initialize
      @client = Octokit::Client.new(:netrc => true)
    end

    def get_open_pulls
      Nailed::PRODUCTS["products"].each do |product,values|
        organization = values["organization"]
        repos = values["repos"]
        repos.each do |repo|
          if organization.nil?
            pulls = @client.pull_requests(repo)
          else
            pulls = @client.pull_requests("#{organization}/#{repo}")
          end
          pulls.each do |pr|
            db_handler = Pullrequest.first_or_create(
                         :pr_number => pr.number,
                         :title => pr.title,
                         :state => pr.state,
                         :url => pr.html_url,
                         :created_at => pr.created_at,
                         :repository_rname => repo
                         )

            Nailed.save_state(db_handler)
          end unless pulls.empty?
          write_pull_trends(repo)
        end unless repos.nil?
      end
    end

    def update_pull_states
      pulls = Pullrequest.all
      pulls.each do |db_pull|
        number = db_pull.pr_number
        repo = db_pull.repository_rname
        org = Repository.get(repo).organization_oname
        github_pull = @client.pull_request("#{org}/#{repo}", number)
        if github_pull.state == "closed"
          db_pull.destroy
        end
      end
    end

    def write_pull_trends(repo)
      open = Pullrequest.count(:repository_rname => repo)
      db_handler = Pulltrend.first_or_create(
                   :time => Time.new.strftime("%Y-%m-%d %H:%M:%S"),
                   :open => open,
                   :repository_rname => repo
                   )

      Nailed.save_state(db_handler)
    end
  end

  # Generic methods
  def Nailed.log(level,msg)
    if level == "error"
      Nailed::LOGGER.error(msg)
    else
      Nailed::LOGGER.info(msg)
    end
  end

  def Nailed.get_org_repos(github_client, org)
    all_repos = github_client.org_repos(org)
    all_repos.map(&:name)
  end

  def Nailed.fill_db_after_migration(github_client)
    Nailed::PRODUCTS["products"].each do |product,values|
      values["versions"].each do |version|
        db_handler = Product.first_or_create(:name => version)
        Nailed.save_state(db_handler)
      end unless values["versions"].nil?
      if values["organization"].nil?
        values["repos"].each do |repo|
          db_handler = Repository.first_or_create(:repo => repo)
          Nailed.save_state(db_handler)
        end unless values["repos"].nil?
      else
        db_handler = Organization.first_or_create(:oname => values["organization"])
        Nailed.save_state(db_handler)
        org_repos = Nailed.get_org_repos(github_client, values["organization"])
        org_repos.each do |org_repo|
          db_handler = Repository.first_or_create(:rname => org_repo, :organization_oname => values["organization"])
          Nailed.save_state(db_handler)
        end
      end
    end
  end

  def Nailed.save_state(db_handler)
    unless db_handler.save
      puts("ERROR: see logfile")
      log("error", db_handler.errors.inspect)
    end
  end
end