# Copyright 2010-2011 Fog Creek Software

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

'''largefile store working over Mercurial's wire protocol'''

import lfutil

import remotestore

class wirestore(remotestore.remotestore):

    def __init__(self, ui, repo, remote):

        cap = remote.capable('largefiles')

        if not cap:

            raise lfutil.storeprotonotcapable([])

        storetypes = cap.split(',')

        if 'serve' not in storetypes:

            raise lfutil.storeprotonotcapable(storetypes)

        self.remote = remote

        super(wirestore, self).__init__(ui, repo, remote.url())

    def _put(self, hash, fd):

        return self.remote.putlfile(hash, fd)

    def _get(self, hash):

        return self.remote.getlfile(hash)

    def _stat(self, hashes):

        '''For each hash, return 0 if it is available, other values if not.

        It is usually 2 if the largefile is missing, but might be 1 the server

        has a corrupted copy.'''

        batch = self.remote.batch()

        futures = {}

        for hash in hashes:

            futures[hash] = batch.statlfile(hash)

        batch.submit()

        retval = {}

        for hash in hashes:

            retval[hash] = futures[hash].value

        return retval