# 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'''

from . import (

    lfutil,

    remotestore,

)

class wirestore(remotestore.remotestore):

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

        cap = remote.capable(b'largefiles')

        if not cap:

            raise lfutil.storeprotonotcapable([])

        storetypes = cap.split(b',')

        if b'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."""

        with self.remote.commandexecutor() as e:

            fs = []

            for hash in hashes:

                fs.append(

                    (

                        hash,

                        e.callcommand(

                            b'statlfile',

                            {

                                b'sha': hash,

                            },

                        ),

                    )

                )

            return {hash: f.result() for hash, f in fs}