import os, sys
import time
import json
from dataverse_api_link import DataverseAPILink

def msg(s): print s
def dashes(char='-'): msg(40*char)
def msgt(s): dashes(); msg(s); dashes()
def msgx(s): dashes('\/'); msg(s); dashes('\/'); sys.exit(0)

def get_dataverse_link_object(apikey='pete'):
    server_with_api = 'http://localhost:8080/'
    #server_with_api = 'https://dvn-build.hmdc.harvard.edu'
    return DataverseAPILink(server_with_api, use_https=False, apikey=apikey)

def check_dv():    
    dat = get_dataverse_link_object()
    dat.save_current_metadata('local-data')

    #add_and_publish_dataverses('local-data/dataverses_2014-0612_11.json','pete')
    add_and_publish_dataverses('demo-data/dataverses_2014-0609_16.json','pete')

def load_users_from_api_file(fname):
    """
    Given the JSON results of the list users command (/api/builtin-users):
        (a) Iterate through the list
        (b) Check if a user exists (by id)
        (c) If the user is not found, create the user
        
    :param fname: full path to a file with user info in JSON format
    """
    if not os.path.isfile(fname):
        msgx('File not found: %s' % fname)
        
    # Load the JSON file
    user_dict = json.loads(open(fname,'r').read())

    # Get a the DataverseAPILink object
    dv_lnk_obj = get_dataverse_link_object('pete')
    dv_lnk_obj.set_return_mode_python()
    
    # Iterate through json
    for user_info in user_dict.get('data', []):
        # check if user exists via api
        current_user_info = dv_lnk_obj.get_user_data(user_info.get('id', None))
        if current_user_info and current_user_info.get('status') == 'OK': 
            continue    # The user exist, loop to the next user

        user_info.pop('id')     # Use the param, except for the 'id'
        
        # Create the user, passing user params and a password
        #
        
        new_password = user_info.get('userName')
        dv_lnk_obj.create_user(user_info, new_password)
        
def add_and_publish_dataverses(fname, apikey):
    if not os.path.isfile(fname):
        msgx('File not found: %s' % fname)

    # Load the JSON file
    dv_dict = json.loads(open(fname,'r').read())

    # Get a the DataverseAPILink object
    dv_lnk_obj = get_dataverse_link_object(apikey)
    dv_lnk_obj.set_return_mode_python()

    # Iterate through json
    previous_alias = "root"
    for dv_info in dv_dict.get('data', []):
        # check if user exists via api
        current_dv_info = dv_lnk_obj.get_dataverse_by_id_or_alias(dv_info.get('id', None))
        
        # DV exists, continue loop
        if current_dv_info and current_dv_info.get('status') == 'OK': 
            msg('>>> FOUND IT')
            previous_alias = current_dv_info['data']['alias']
            continue    # The user exist, loop to the next user
            
        # No DV, create it
        keys_not_needed = ['id', 'ownerID', 'creationDate', 'creator']
        for key in keys_not_needed:
            if dv_info.has_key(key):
                dv_info.pop(key)
        
        msg('params to send: %s' % dv_info)
        # If created,  publish it
        json_resp = dv_lnk_obj.create_dataverse(previous_alias, dv_info)
        if json_resp.get('status') == 'OK': 
            new_dv_data = json_resp.get('data', {})
            new_id = new_dv_data.get('id', None)
            if new_id is not None:
                dv_lnk_obj.publish_dataverse(new_id)
        previous_alias = current_dv_info.get("alias", "root")
        #break
            
def add_dataverses(name, cnt=1, parent_dv_name_or_id=1, apikey='snoopy'):
    # get the DataverseAPILink
     dat = get_dataverse_link_object(apikey=apikey)
     dat.set_return_mode_python()
     
     for x in range(249, 260):      
         dat.publish_dataverse(x)
     return
     for x in range(0, cnt):        
        num  = x+1
        alias_str = "new_dv_%d" % num
        dv_params_str = """{ "alias":"%s",
                    "name":"%s %s",
                    "affiliation":"Affiliation value",
                    "contactEmail":"pete@malinator.com",
                    "permissionRoot":true,
                    "description":"More API testing"
                    }""" % (alias_str, name, num)
    
        dv_params = json.loads(dv_params_str)
        dat.create_dataverse(parent_dv_name_or_id, dv_params, )
        if x % 20 == 0: time.sleep(1)


def delete_dataverses_id_greather_than(id_num, apikey):
     if not type(id_num) == int:
         raise('id_num needs be an int--not a %s' % type(id_num))

     # get the DataverseAPILink
     dat = get_dataverse_link_object(apikey=apikey)
     dat.set_return_mode_python()
     
     # List the dataverses
     dv_json = dat.list_dataverses()
     print dv_json
     # Pull dataverse ids > 30
     dv_ids = [dv['id'] for dv in dv_json.get("data") if dv['id'] > id_num]

     # reverse order ids
     dv_ids.sort()
     dv_ids.reverse()

     # delete them
     for dv_id in dv_ids:
         print dat.delete_dataverse_by_id(dv_id)
     #print dat.list_datasets()
     
if __name__ == '__main__':
    check_dv()
    #load_users_from_api_file('demo-data/users_2014-0609_14.json')
    #load_users_from_api_file('demo-data/rp_users.json')
    #add_and_publish_dataverses('demo-data/dataverses_2014-0609_14.json', 'gromit')
    #add_and_publish_dataverses('demo-data/rp_dataverses.json', 'gromit')
    
    #add_dataverses('Other DV #', 17, 23, 'snoopy')
    #add_dataverses('Uma\'s Other Retricted DVs #', 7, 8, 'pete')
    #delete_dataverses_id_greather_than(177, 'pete')