import sys
import bioblend
from bioblend import galaxy
from bioblend import *
from bioblend.galaxy import GalaxyInstance
from bioblend.galaxy.histories import HistoryClient
from bioblend.galaxy.tools import ToolClient
from bioblend.galaxy.workflows import WorkflowClient
from bioblend.galaxy.datasets import DatasetClient

def main():
    #USER INPUT
    GALAXY_URL = sys.argv[1]
    GALAXY_API_KEY = sys.argv[2]
    user_email = sys.argv[3]
    #user_id = sys.argv[4]

    gi = bioblend.galaxy.GalaxyInstance(GALAXY_URL, GALAXY_API_KEY)
    libraries = galaxy.libraries.LibraryClient(gi)
    already_existing_libraries = libraries.get_libraries(library_id=None, name="%s's library"%(user_email), deleted=False)
    library_id = ""
    lib_already_exists = False

    #Getting user_id
    users = bioblend.galaxy.users.UserClient(gi)
    user_list = users.get_users(deleted=False)
    for user_dic in user_list :
       if user_dic['email'] == user_email :
           print "============================================================================================"
           print 'user dictionnary : '
           print user_dic
           user_id = user_dic['id']
           print "============================================================================================"
           print 'value of user_id used to set permissions : '
           print user_id
           user_exists = True
           break

    if user_exists :
        for lib_dic in already_existing_libraries :
            if lib_dic['deleted'] == False and lib_dic['name'] == "%s's library"%(user_email) :
                lib_already_exists = True
                library_id = lib_dic['id']
                print "Library already exists, aborting"
                break

        #Creating data library
        if not lib_already_exists :
            libraries.create_library("%s's library"%(user_email), description="%s's library"%(user_email), synopsis="%s's library"%(user_email))
            existing_libraries = libraries.get_libraries(library_id=None, name="%s's library"%(user_email), deleted=False)
            for lib_dic in existing_libraries :
                if lib_dic['deleted'] == False and lib_dic['name'] == "%s's library"%(user_email) :
                    library_id = lib_dic['id']
                    break
            libraries.set_library_permissions(library_id, access_in=user_id, modify_in=user_id, add_in=user_id, manage_in=user_id)
            print "Data library created"
        ret = gi.libraries.get_library_permissions(library_id)
        print "============================================================================================"
        print "library permissions of current library : "
        print ret
        #assert(ret['access_library_role_list'] == user_id)
        #assert(ret['modify_library_role_list'] == user_id)
        #assert(ret['add_library_item_role_list']) == user_id)
        #assert(ret['manage_library_role_list']) == user_id)

    else :
        print "This user email does not exist, aborting"


if __name__ == "__main__" : main()