students_team action

lib/teachers_pet/actions/base.rb

@@ -2,6 +2,7 @@
 
 require 'configuration'
 require 'octokit'
+require 'highline/import'
 
 ## Common code for the edugit scripts.
 module TeachersPet
@@ -113,11 +114,11 @@ def get_team_member_logins(team_id)
       end
 
       def read_file(filename, type)
+        # Return a hash with team name as key and list of member(s) as value.
         map = Hash.new
         puts "Loading #{type}:"
         File.open(filename).each_line do |team|
-          # Team can be a single user, or a team name and multiple users
-          # Trim whitespace, otherwise issues occur
+          # Each line is a single user, or a team name and multiple users
           team.strip!
           items = team.split(' ')
           items.each do |item|
@@ -143,7 +144,6 @@ def read_file(filename, type)
       end
 
       def confirm(message, abort_on_no = true)
-        # confirm
         confirmed = false
         choose do |menu|
           menu.prompt = message

lib/teachers_pet/actions/clone_repos.rb

@@ -7,7 +7,6 @@
 
 require 'rubygems'
 require 'highline/question'
-require 'highline/import'
 require 'highline/compatibility'
 require 'teachers_pet/actions/base'
 

lib/teachers_pet/actions/create_repos.rb

@@ -5,7 +5,6 @@
 
 require 'rubygems'
 require 'highline/question'
-require 'highline/import'
 require 'highline/compatibility'
 require 'teachers_pet/actions/base'
 

lib/teachers_pet/actions/create_teams.rb

@@ -10,7 +10,6 @@
 
 require 'rubygems'
 require 'highline/question'
-require 'highline/import'
 require 'highline/compatibility'
 
 require 'teachers_pet/actions/base'

lib/teachers_pet/actions/fork_collab.rb

@@ -2,7 +2,6 @@
 
 require 'rubygems'
 require 'highline/question'
-require 'highline/import'
 require 'highline/compatibility'
 require 'teachers_pet/actions/base'
 

lib/teachers_pet/actions/open_issue.rb

@@ -5,7 +5,6 @@
 
 require 'rubygems'
 require 'highline/question'
-require 'highline/import'
 require 'teachers_pet/actions/base'
 
 module TeachersPet

lib/teachers_pet/actions/push_files.rb

@@ -2,7 +2,6 @@
 
 require 'rubygems'
 require 'highline/question'
-require 'highline/import'
 require 'highline/compatibility'
 require 'teachers_pet/actions/base'
 

lib/teachers_pet/actions/students_team.rb

@@ -0,0 +1,79 @@
+# Create a team from all users named in a students file
+
+$LOAD_PATH << File.join(File.dirname(__FILE__), '..', '..')
+
+require 'set'
+require 'rubygems'
+require 'highline/question'
+require 'highline/compatibility'
+
+require 'teachers_pet/actions/base'
+
+module TeachersPet
+  module Actions
+    class StudentsTeam < Base
+      def read_info
+        @organization = ask("What is the organization name?") { |q| q.default = TeachersPet::Configuration.organization }
+        @student_file = self.get_students_file_path
+      end
+
+      def load_files
+        @students = read_file(@student_file, 'Students')
+      end
+
+      def update_team
+        self.init_client
+
+        all_team_name='Students'
+        confirm("Add #{@students.size} members to '#{all_team_name}' team?")
+
+        # Get current members of org team
+        existing = Hash.new
+        teams = self.get_teams_by_name(@organization)
+        studentsteam = teams[all_team_name] || false
+        unless studentsteam
+          puts "\nCreating team #{all_team_name}..."
+          @client.create_team(@organization,
+            {
+              :name => all_team_name,
+              :permission => 'push'
+            })
+          studentsteam = self.get_teams_by_name(@organizaiton)['Students']
+        end
+
+        # Find and add new students to the org team
+        all_students = Set.new
+        @students.each do |team, members|
+          all_students.merge(members)
+        end
+        all_team_members = Set.new(self.get_team_member_logins(studentsteam[:id]))
+        new_students = all_students - all_team_members
+        old_students = all_team_members - all_students
+
+        puts "\nMembers of '#{studentsteam[:name]}':"
+        all_team_members.each do |member|
+          puts " -> #{member}"
+        end
+        puts "\nAdding members:"
+        new_students.each do |student|
+          @client.add_team_member(studentsteam[:id], student)
+          puts " -> #{student}"
+        end
+        puts "\nMembers not mentioned:"
+        old_students.each do |student|
+          puts " -> #{student}"
+        end
+
+        puts "\n#{all_team_members.length} previous members in '#{studentsteam[:name]}'."
+        puts "#{new_students.length} members added."
+        puts "#{(all_students - new_students).length} skipped."
+      end
+
+      def run
+        self.read_info
+        self.load_files
+        self.update_team
+      end
+    end
+  end
+end