education/classroom

View on GitHub
app/models/github_organization.rb

Summary

Maintainability
A
2 hrs
Test Coverage
# frozen_string_literal: true

class GitHubOrganization < GitHubResource
  def accept_membership(user_github_login)
    return if organization_member?(user_github_login)

    GitHub::Errors.with_error_handling do
      @client.update_organization_membership(login, state: "active")
    end
  end

  def add_membership(user_github_login)
    return if organization_member?(user_github_login)

    GitHub::Errors.with_error_handling do
      @client.update_organization_membership(login, user: user_github_login)
    end
  end

  def admin?(user_github_login)
    GitHub::Errors.with_error_handling do
      membership = @client.organization_membership(login, user: user_github_login)
      membership.role == "admin" && membership.state == "active"
    end
  rescue GitHub::Error
    false
  end

  def create_repository(repo_name, users_repo_options = {})
    repo_options = github_repo_default_options.merge(users_repo_options)

    repo = GitHub::Errors.with_error_handling do
      @client.create_repository(repo_name, repo_options)
    end

    GitHubRepository.new(@client, repo.id)
  end

  def create_repository_from_template(template_repo_id, repo_name, repo_options = {})
    repo = GitHub::Errors.with_error_handling(report_to_failbot: false) do
      @client.create_repository_from_template(template_repo_id, repo_name, repo_options)
    end

    GitHubRepository.new(@client, repo.id)
  end

  def delete_repository(repo_id)
    GitHub::Errors.with_error_handling do
      @client.delete_repository(repo_id)
    end
  end

  def create_team(team_name)
    github_team = GitHub::Errors.with_error_handling do
      @client.create_team(
        @id,
        description: "#{team_name} created by GitHub Classroom",
        name: team_name
      )
    end

    GitHubTeam.new(@client, github_team.id)
  end

  def delete_team(team_id)
    GitHub::Errors.with_error_handling do
      @client.delete_team(team_id)
    end
  end

  def github_avatar_url(size = 40)
    "#{avatar_url}&size=#{size}"
  end

  def organization_members(options = {})
    GitHub::Errors.with_error_handling { @client.organization_members(@id, options) }
  end

  def organization_member?(user_github_login)
    GitHub::Errors.with_error_handling { @client.organization_member?(@id, user_github_login) }
  end

  def plan
    GitHub::Errors.with_error_handling do
      organization = @client.organization(@id, headers: GitHub::APIHeaders.no_cache_no_store)

      if organization.owned_private_repos.present? && organization.plan.present?
        return { owned_private_repos: organization.owned_private_repos, private_repos: organization.plan.private_repos }
      end

      raise GitHub::Error, "Cannot retrieve this organizations repo plan, please reauthenticate your token."
    end
  end

  def remove_organization_member(user)
    github_user_login = user.github_user.login(use_cache: false)

    return if admin?(github_user_login)

    GitHub::Errors.with_error_handling do
      @client.remove_organization_member(@id, github_user_login)
    end
  end

  def team_invitations_url
    "#{GitHubClassroom.github_url}/orgs/#{login}/people"
  end

  def create_organization_webhook(config: {}, options: {})
    GitHub::Errors.with_error_handling do
      hook_config = { content_type: "json", secret: webhook_secret }.merge(config)

      hook_options = {
        # Send the [wildcard](https://developer.github.com/webhooks/#wildcard-event)
        # so that we don't have to upgrade the webhooks everytime we need something new.
        events: ["*"],
        active: true
      }.merge(options)

      @client.create_org_hook(@id, hook_config, hook_options)
    end
  end

  def activate_organization_webhook(webhook_id, config: {}, options: {})
    GitHub::Errors.with_error_handling do
      hook_config = { content_type: "json", secret: webhook_secret }.merge(config)

      hook_options = {
        # Send the [wildcard](https://developer.github.com/webhooks/#wildcard-event)
        # so that we don't have to upgrade the webhooks everytime we need something new.
        events: ["*"],
        active: true
      }.merge(options)

      @client.edit_org_hook(@id, webhook_id, hook_config, hook_options)
    end
  end

  def organization_webhooks
    GitHub::Errors.with_error_handling do
      @client.org_hooks(@id)
    end
  end

  def remove_organization_webhook(webhook_id)
    return if webhook_id.blank?
    GitHub::Errors.with_error_handling do
      @client.remove_org_hook(@login, webhook_id)
    end
  end

  def update_default_repository_permission!(default_repository_permission)
    GitHub::Errors.with_error_handling do
      @client.update_organization(
        @id,
        { default_repository_permission: default_repository_permission },
        accept: "application/vnd.github.korra-preview+json"
      )
    end
  end

  private

  def github_attributes
    %w[login avatar_url html_url name node_id]
  end

  def github_repo_default_options
    {
      has_issues:    true,
      has_wiki:      true,
      has_downloads: true,
      organization:  @id
    }
  end

  def webhook_secret
    Rails.application.secrets.webhook_secret
  end
end
# rubocop:enable Metrics/ClassLength