# Copyright 2009-2010 Gregory P. Ward

# Copyright 2009-2010 Intelerad Medical Systems Incorporated

# Copyright 2010-2011 Fog Creek Software

# Copyright 2010-2011 Unity Technologies

#

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

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

'''setup for largefiles extension: uisetup'''

from mercurial import archival, cmdutil, commands, extensions, filemerge, hg, \

    httppeer, localrepo, merge, scmutil, sshpeer, sshserver, wireproto

from mercurial.i18n import _

from mercurial.hgweb import hgweb_mod, protocol, webcommands

from mercurial.subrepo import hgsubrepo

import overrides

import proto

def uisetup(ui):

    # Disable auto-status for some commands which assume that all

    # files in the result are under Mercurial's control

    entry = extensions.wrapcommand(commands.table, 'add',

                                   overrides.overrideadd)

    addopt = [('', 'large', None, _('add as largefile')),

              ('', 'normal', None, _('add as normal file')),

              ('', 'lfsize', '', _('add all files above this size '

                                   '(in megabytes) as largefiles '

                                   '(default: 10)'))]

    entry[1].extend(addopt)

    # The scmutil function is called both by the (trivial) addremove command,

    # and in the process of handling commit -A (issue3542)

    entry = extensions.wrapfunction(scmutil, 'addremove',

                                    overrides.scmutiladdremove)

    entry = extensions.wrapcommand(commands.table, 'remove',

                                   overrides.overrideremove)

    entry = extensions.wrapcommand(commands.table, 'forget',

                                   overrides.overrideforget)

    # Subrepos call status function

    entry = extensions.wrapcommand(commands.table, 'status',

                                   overrides.overridestatus)

    entry = extensions.wrapfunction(hgsubrepo, 'status',

                                    overrides.overridestatusfn)

    entry = extensions.wrapcommand(commands.table, 'log',

                                   overrides.overridelog)

    entry = extensions.wrapcommand(commands.table, 'rollback',

                                   overrides.overriderollback)

    entry = extensions.wrapcommand(commands.table, 'verify',

                                   overrides.overrideverify)

    verifyopt = [('', 'large', None, _('verify largefiles')),

                 ('', 'lfa', None,

                     _('verify all revisions of largefiles not just current')),

                 ('', 'lfc', None,

                     _('verify largefile contents not just existence'))]

    entry[1].extend(verifyopt)

    entry = extensions.wrapcommand(commands.table, 'outgoing',

        overrides.overrideoutgoing)

    outgoingopt = [('', 'large', None, _('display outgoing largefiles'))]

    entry[1].extend(outgoingopt)

    entry = extensions.wrapcommand(commands.table, 'summary',

                                   overrides.overridesummary)

    summaryopt = [('', 'large', None, _('display outgoing largefiles'))]

    entry[1].extend(summaryopt)

    entry = extensions.wrapcommand(commands.table, 'update',

                                   overrides.overrideupdate)

    entry = extensions.wrapcommand(commands.table, 'pull',

                                   overrides.overridepull)

    pullopt = [('', 'all-largefiles', None,

                 _('download all pulled versions of largefiles'))]

    entry[1].extend(pullopt)

    entry = extensions.wrapcommand(commands.table, 'clone',

                                   overrides.overrideclone)

    cloneopt = [('', 'all-largefiles', None,

                 _('download all versions of all largefiles'))]

    entry[1].extend(cloneopt)

    entry = extensions.wrapfunction(hg, 'clone', overrides.hgclone)

    entry = extensions.wrapcommand(commands.table, 'cat',

                                   overrides.overridecat)

    entry = extensions.wrapfunction(merge, '_checkunknownfile',

                                    overrides.overridecheckunknownfile)

    entry = extensions.wrapfunction(merge, 'manifestmerge',

                                    overrides.overridemanifestmerge)

    entry = extensions.wrapfunction(filemerge, 'filemerge',

                                    overrides.overridefilemerge)

    entry = extensions.wrapfunction(cmdutil, 'copy',

                                    overrides.overridecopy)

    # Summary calls dirty on the subrepos

    entry = extensions.wrapfunction(hgsubrepo, 'dirty',

                                    overrides.overridedirty)

    # Backout calls revert so we need to override both the command and the

    # function

    entry = extensions.wrapcommand(commands.table, 'revert',

                                   overrides.overriderevert)

    entry = extensions.wrapfunction(commands, 'revert',

                                    overrides.overriderevert)

    # clone uses hg._update instead of hg.update even though they are the

    # same function... so wrap both of them)

    extensions.wrapfunction(hg, 'update', overrides.hgupdate)

    extensions.wrapfunction(hg, '_update', overrides.hgupdate)

    extensions.wrapfunction(hg, 'clean', overrides.hgclean)

    extensions.wrapfunction(hg, 'merge', overrides.hgmerge)

    extensions.wrapfunction(archival, 'archive', overrides.overridearchive)

    extensions.wrapfunction(hgsubrepo, 'archive', overrides.hgsubrepoarchive)

    extensions.wrapfunction(cmdutil, 'bailifchanged',

                            overrides.overridebailifchanged)

    # create the new wireproto commands ...

    wireproto.commands['putlfile'] = (proto.putlfile, 'sha')

    wireproto.commands['getlfile'] = (proto.getlfile, 'sha')

    wireproto.commands['statlfile'] = (proto.statlfile, 'sha')

    # ... and wrap some existing ones

    wireproto.commands['capabilities'] = (proto.capabilities, '')

    wireproto.commands['heads'] = (proto.heads, '')

    wireproto.commands['lheads'] = (wireproto.heads, '')

    # make putlfile behave the same as push and {get,stat}lfile behave

    # the same as pull w.r.t. permissions checks

    hgweb_mod.perms['putlfile'] = 'push'

    hgweb_mod.perms['getlfile'] = 'pull'

    hgweb_mod.perms['statlfile'] = 'pull'

    extensions.wrapfunction(webcommands, 'decodepath', overrides.decodepath)

    # the hello wireproto command uses wireproto.capabilities, so it won't see

    # our largefiles capability unless we replace the actual function as well.

    proto.capabilitiesorig = wireproto.capabilities

    wireproto.capabilities = proto.capabilities

    # these let us reject non-largefiles clients and make them display

    # our error messages

    protocol.webproto.refuseclient = proto.webprotorefuseclient

    sshserver.sshserver.refuseclient = proto.sshprotorefuseclient

    # can't do this in reposetup because it needs to have happened before

    # wirerepo.__init__ is called

    proto.ssholdcallstream = sshpeer.sshpeer._callstream

    proto.httpoldcallstream = httppeer.httppeer._callstream

    sshpeer.sshpeer._callstream = proto.sshrepocallstream

    httppeer.httppeer._callstream = proto.httprepocallstream

    # don't die on seeing a repo with the largefiles requirement

    localrepo.localrepository.supported |= set(['largefiles'])

    # override some extensions' stuff as well

    for name, module in extensions.extensions():

        if name == 'fetch':

            extensions.wrapcommand(getattr(module, 'cmdtable'), 'fetch',

                overrides.overridefetch)

        if name == 'purge':

            extensions.wrapcommand(getattr(module, 'cmdtable'), 'purge',

                overrides.overridepurge)

        if name == 'rebase':

            extensions.wrapcommand(getattr(module, 'cmdtable'), 'rebase',

                overrides.overriderebase)

        if name == 'transplant':

            extensions.wrapcommand(getattr(module, 'cmdtable'), 'transplant',

                overrides.overridetransplant)