# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@fb.com>
# Date 2014-04-17 03:55:59
# Node ID 2f8c4fa237f597929092b8f2cd356022a3df2840
# Parent  bef4a2adc532ddb43ea56e35b55b501ebc09090c

bundle2: adds a capabilities attribute on bundler20

This attribute conveys the capabilities supported by the destination of the
bundle. It is used to decide which parts to include in the bundle.

This is currently a set but will probably be turned into a dictionary to allow
capabilities with values.

diff --git a/mercurial/bundle2.py b/mercurial/bundle2.py
--- a/mercurial/bundle2.py
+++ b/mercurial/bundle2.py
@@ -340,10 +340,11 @@ class bundle20(object):
     populate it. Then call `getchunks` to retrieve all the binary chunks of
     data that compose the bundle2 container."""
 
-    def __init__(self, ui):
+    def __init__(self, ui, capabilities=()):
         self.ui = ui
         self._params = []
         self._parts = []
+        self.capabilities = set(capabilities)
 
     def addparam(self, name, value=None):
         """add a stream level parameter"""