#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# creategroups.py
# Februar 2020
#
# create custom groups and add members using a csv file as input
# you can also add members to existing groups
#
# if the group already exists it will not be added and the http response is printed
# if the user does not exist it will not be added to the group and the http response is printed
# if the user is already a member of the group it will not be added to the group and the http response is printed
#
# none of the above conditions will prevent the processing of additional items in the csv
#
#
# Change History
#
# 03feb2020 Initial development
#
# Format of csv file is two columns
#  Column 1 group id 
#  Column 2 group name 
#  Column 3 is a description
#  Column 4 (optional) member id
#  Column 5 (optional) list of group names which are added to the group
#
# For example:
#group2,"Group 2","My Group 2"
#group3,"Group 3","My Group3",geladm
#group1,"Group 1","group 1"
#
# Copyright 2020, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
#
#  Licensed under the Apache License, Version 2.0 (the License);
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.
#
import argparse
import csv
import os
from sharedfunctions import callrestapi, getfolderid, file_accessible, getidsanduris


def getMembersOfGroup(id):

    reqval="/identities/groups/"+id+"/members"
    rst=callrestapi(reqval,'get')

    lst = set() 
    for member in rst['items']:
       lst.add(member['id'])

    return lst    

# setup command-line arguements    
parser = argparse.ArgumentParser(description="Create custom groups and establish membership")
parser.add_argument("-f","--file", help="Full path to csv file containing groups ",required='True')
args = parser.parse_args()
file=args.file

reqtype="post"

check=file_accessible(file,'r')

allgroups=callrestapi("/identities/groups","get")

# create a list of all groups

groupslist=[]
groupNameDic = {} # dictionary for lookup group name -> group id

if 'items' in allgroups:
    
        total_items=allgroups['count']
        returned_items=len(allgroups['items'])
        
         
        for i in range(0,returned_items): 
                                
           groupslist.append(allgroups['items'][i]['id'])
           groupNameDic[allgroups['items'][i]['name']] = allgroups['items'][i]['id']

 
# file can be read
if check:

    with open(file, 'rt') as f:
        
        filecontents = csv.reader(f)

        for row in filecontents:
            
            #print(row)

            cols=len(row)

            id=row[0]
            newgroup=row[1]
            description=row[2]

            data = {}
            data["id"]=id
            data['name'] = newgroup
            data['description'] = description
            data["state"]="active"

            if id in groupslist:
                 print("Note: group with name "+newgroup+"  and id "+id+" already exists." )

            else:
               
                print ("Note: Trying to creating Group: "+newgroup )

                reqtype='post'
                reqval="/identities/groups/"

                myresult=callrestapi(reqval,reqtype,data=data,stoponerror=0)

                if myresult != None: print("Note: Group: "+newgroup+" created" )
                      
            # 4th column is group membership (of user)
            if cols>=4 and row[3]!=None and row[3]!="":
         
                member=row[3]
                print("Note: Trying to add user "+ member+ " to group "+newgroup )

                reqval="/identities/groups/"+id+"/userMembers/"+member
                reqtype='put'
                myresult=callrestapi(reqval,reqtype,data=data,stoponerror=0)
                if myresult != None: print("Note: user: "+member+" added to group "+newgroup )

            if cols>=5 and row[4]!=None and row[4]!="":
                membergroups = row[4]
                currentMembers = getMembersOfGroup(id)

                for membergroup in membergroups.split(','):
                    membergroupid = groupNameDic[membergroup]
                    print ("Note: Trying to add group "+membergroup+" ("+membergroupid+") to group "+newgroup)

                    if membergroupid in currentMembers:
                        print ("  Group "+membergroup+" ("+membergroupid+") is already a member. Skiping")
                    else:
                        reqval="/identities/groups/"+id+"/groupMembers/"+membergroupid
                        reqtype='put'
                        myresult=callrestapi(reqval,reqtype,data=data,stoponerror=0)
                        if myresult != None: print("Note: group: "+membergroup+" added to group "+newgroup )

else:
    print("ERROR: cannot read "+file)