import os
import clique
import glob

# Expect to find a web project structure modelled as such :
# <Project>/<Sequence[name]=test_sequence>/<Shot>/<Task>


from ftrack_api import Session

# set max frames to be published ad once...
max_frames = 50 # value suggested by the client who tested it.

# path where to find the frames
frames_path = '/mnt/zeus/storage/ftrack/projects_material/NissanXTrail/Footage/DPX/A016C019_101031_L1O3/*'


def generate_frames_chunks(frames, max_frames):
    # convert a sequence of filepath in a list of sequences of the given amount of *max_frames*
    results = []
    for i in xrange(0, len(frames), max_frames):
        results.append(clique.assemble(frames[i:i + max_frames],minimum_items=1)[0][0])

    return results


def publish_first_chunk_frames(session, collection, location):
    # simple code to publish an image sequence... 
    asset_type = session.query('AssetType where short is "geo"').first()

    shot = session.query('Shot where parent.name is "test_sequence"').one()
    task = session.query('Task where parent.parent.name is "test_sequence"').one()
    
    asset = session.query(
        'Asset where name is "something" and parent.id is "{}" and type.id is "{}"'.format(
            shot['id'], asset_type['id'])
    ).first()
    
    if not asset:
        # if the asset does not already exist create a new one...
        asset = session.create('Asset', {
            'parent': shot,
            'name': 'something',
            'type': asset_type
        })

    # Create a new version for the asset.
    status = session.query('Status').first()
    version = session.create('AssetVersion', {
        'asset': asset,
        'status': status,
        'comment': 'Added more leaves.',
        'task': task
    })
    session.commit()
    print 'publishing first set of frames.....'
    component = version.create_component(str(collection), location=location)
    return component


def update_component(session, sequence_component, new_collection, location):
    # update code extracted from session._create_component
    # https://bitbucket.org/ftrack/ftrack-python-api/src/1f20fcd0f68d2db5188c995d1b5a34791dd7c066/source/ftrack_api/session.py#lines-1973

    print 'Updating component', sequence_component
    print 'PRE', sequence_component['size'], len(sequence_component['members'])

    sequence_component_size = sequence_component['size']

    member_sizes = {}
    container_size = 0
    for item in new_collection:
        member_sizes[item] = session._get_filesystem_size(item)
        container_size += member_sizes[item]

    new_sequence_component_size = sequence_component_size + container_size

    # add missing frames to the sequence

    for member_path in new_collection:    
        member_data = {
            'name': new_collection.match(member_path).group('index'),
            'container': sequence_component,
            'size': member_sizes[member_path],
            'file_type': os.path.splitext(member_path)[-1]
        }

        component = session._create_component(
            'FileComponent', member_path, member_data, location=location
        )
        if component not in sequence_component['members']:
            sequence_component['members'].append(component)

    sequence_component['size'] = new_sequence_component_size
    print 'POST', sequence_component['size'], len(sequence_component['members'])
    session.commit()


def main():
    # gather all the files
    frames = glob.glob(frames_path)
    frame_sequences = generate_frames_chunks(frames, max_frames)
    
    session = Session()
    first_frames_part = frame_sequences[0]
    other_frames = frame_sequences[1:]
    location = session.pick_location()

    result_component = publish_first_chunk_frames(session, first_frames_part, location)
    for frames in other_frames:
        update_component(session, result_component, frames, location)

    # check the result...
    published_component = session.get('Component', result_component['id'])

    print published_component['size']
    print len(published_component['members'])


if __name__ == '__main__':
    main()