module Octokit

def client
  return @client if defined?(@client) && @client.same_options?(options)
  @client = Octokit::Client.new(options)
end

def enterprise_admin_client
  return @enterprise_admin_client if defined?(@enterprise_admin_client) && @enterprise_admin_client.same_options?(options)
  @enterprise_admin_client = Octokit::EnterpriseAdminClient.new(options)
end

def enterprise_management_console_client
  return @enterprise_management_console_client if defined?(@enterprise_management_console_client) && @enterprise_management_console_client.same_options?(options)
  @enterprise_management_console_client = Octokit::EnterpriseManagementConsoleClient.new(options)
end

def method_missing(method_name, *args, &block)
  if client.respond_to?(method_name)
    return client.send(method_name, *args, &block)
  elsif enterprise_admin_client.respond_to?(method_name)
    return enterprise_admin_client.send(method_name, *args, &block)
  elsif enterprise_management_console_client.respond_to?(method_name)
    return enterprise_management_console_client.send(method_name, *args, &block)
  end
  super
end

def respond_to_missing?(method_name, include_private=false)
  client.respond_to?(method_name, include_private) ||
  enterprise_admin_client.respond_to?(method_name, include_private) ||
  enterprise_management_console_client.respond_to?(method_name, include_private)
end