Merge pull request #19 from github/invite-to-org-and-add-to-team

Invite to org and add to team

Author: brntbeer

Gemfile

@@ -1,3 +1,3 @@
 source 'https://rubygems.org'
 
-gem 'octokit'
+gem 'octokit', '~> 4.7'

Gemfile.lock

@@ -1,21 +1,23 @@
 GEM
   remote: https://rubygems.org/
   specs:
-    addressable (2.3.8)
-    faraday (0.9.2)
+    addressable (2.5.1)
+      public_suffix (~> 2.0, >= 2.0.2)
+    faraday (0.12.2)
       multipart-post (>= 1.2, < 3)
     multipart-post (2.0.0)
-    octokit (4.2.0)
-      sawyer (~> 0.6.0, >= 0.5.3)
-    sawyer (0.6.0)
-      addressable (~> 2.3.5)
-      faraday (~> 0.8, < 0.10)
+    octokit (4.7.0)
+      sawyer (~> 0.8.0, >= 0.5.3)
+    public_suffix (2.0.5)
+    sawyer (0.8.1)
+      addressable (>= 2.3.5, < 2.6)
+      faraday (~> 0.8, < 1.0)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
-  octokit
+  octokit (~> 4.7)
 
 BUNDLED WITH
-   1.11.2
+   1.12.0

add-to-org.rb

@@ -0,0 +1,117 @@
+#!/usr/bin/env ruby
+# add USERNAME as member to ORGANIZATION and optionally to TEAM
+# assumes you created and stored the appropriate OAuth token as an ENV variable called GITHUBTEACHER_TOKEN
+
+require 'octokit'
+require 'optparse'
+
+# Make sure arguments are specified
+ARGV << '-h' if ARGV.empty?
+
+# Create options hash
+options = {}
+
+# Parse options
+OptionParser.new do |opts|
+  opts.banner = "Usage: add-to-org.rb [options]
+       NOTICE: This also assumes you have set up the variable GITHUBTEACHER_TOKEN with administrative privilages.\n\n"
+
+  opts.on("-u", "--username USERNAME", "Username for membership -- ex: githubstudent") do |u|
+    options[:username] = u
+  end
+
+  opts.on("-o", "--org ORGANIZATION", "Organization name -- ex: githubschool") do |o|
+    options[:org] = o
+  end
+
+  opts.on("-t", "--team TEAM_NAME", "[Optional] Team to add member to -- ex: developers") do |t|
+    options[:team] = t
+  end
+
+  opts.on_tail("-h", "--help", "Prints this help message") do |h|
+    puts opts
+    exit
+  end
+
+end.parse!
+
+
+TOKEN = ENV['GITHUBTEACHER_TOKEN']
+abort("Missing GITHUBTEACHER_TOKEN. Please set up an OAUTH token and set it in the environment by typing\n
+export GITHUBTEACHER_TOKEN=XXXXXXXXXXXXXXXXXXXXXXX\n
+and replace the Xs with your actual token. Reminder: This token needs admin privilages onto your organization in order to be inviting people.") unless TOKEN
+
+# Assign variables
+#  Team is optional, so you don't need to assign it if it's nil.
+username = options[:username]
+org = options[:org]
+team_name = options[:team] if !options[:team].nil?
+
+
+# Create a new Octokit Client
+Octokit.auto_paginate = true
+@client = Octokit::Client.new :access_token => TOKEN
+
+# If the team doesn't exist yet, create it
+#  otherwise, get the team_id from the list
+# Once we have the team_id we can add people to it
+#  regardless of them being a member of the organization yet
+def add_to_team_and_org(team_name, username, org)
+
+  team_id = if team_id.nil?
+    create_team(org, team_name, org)
+  else
+    get_team_id(team_name, org)
+  end
+
+  # Add username to team_id
+  # Slightly different than adding them as a member.
+  # If the username includes ","s, split it because it's a full list!
+  username.split(",").each do |name|
+    @client.add_team_membership(team_id, name)
+    puts "#{name} added to #{team_name}."
+  end
+
+end
+
+# Returns team_id
+def create_team(team_id, team_name, org)
+  response = @client.create_team(org, {:name => team_name})
+  team_id = response.id
+  puts "Team '#{team_name}' created at https://github.com/orgs/#{org}/teams" unless team_id.nil?
+
+  team_id
+end
+
+def get_team_id(team_name, org)
+  team_id = nil
+  team_list = @client.org_teams(org)
+
+  team_list.each do |team|
+    if team.name == team_name
+      team_id = team.id
+    end
+  end
+  team_id
+
+end
+
+# If they just want to add users as members, skip the team
+def add_to_org(org, username)
+  username.split(",").each do |name|
+    @client.update_organization_membership(org, {:user => name})
+    puts "#{name} added to #{org}."
+  end
+end
+
+begin
+  if !team_name.nil?
+    add_to_team_and_org(team_name, username, org)
+  else
+    add_to_org(org, username)
+  end
+rescue Octokit::Forbidden
+  abort "[403] - Unable to add member to organization. Check that the GITHUBTEACHER_TOKEN was created with administrative privilages so that you can add members to the organization"
+rescue Octokit::Unauthorized
+  abort "[401] - The credentials you've supplied are no longer valid. Please generate a new token."
+end