Mojo: Mojom: Add AST types for methods, interface bodies, and interfaces.

mojo/public/tools/bindings/pylib/mojom/parse/ast.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Node classes for the AST for a Mojo IDL file."""
 
 # Note: For convenience of testing, you probably want to define __eq__() methods
 # for all node types; it's okay to be slightly lax (e.g., not compare filename
 # and lineno). You may also define __repr__() to help with analyzing test
 # failures, especially for more complex types.
@@ skipping 167 matching lines @@
     return super(Import, self).__eq__(other) and \
            self.import_filename == other.import_filename
 
 
 class ImportList(NodeListBase):
   """Represents a list (i.e., sequence) of import statements."""
 
   _list_item_type = Import
 
 
+class Interface(Definition):
+  """Represents an interface definition."""
+
+  def __init__(self, name, attribute_list, body, **kwargs):
+    assert attribute_list is None or isinstance(attribute_list, AttributeList)
+    assert isinstance(body, InterfaceBody)
+    super(Interface, self).__init__(name, **kwargs)
+    self.attribute_list = attribute_list
+    self.body = body
+
+  def __eq__(self, other):
+    return super(Interface, self).__eq__(other) and \
+           self.attribute_list == other.attribute_list and \
+           self.body == other.body
+
+
+class Method(Definition):
+  """Represents a method definition."""
+
+  def __init__(self, name, ordinal, parameter_list, response_parameter_list,
+               **kwargs):
+    assert ordinal is None or isinstance(ordinal, Ordinal)
+    assert isinstance(parameter_list, ParameterList)
+    assert response_parameter_list is None or \
+           isinstance(response_parameter_list, ParameterList)
+    super(Method, self).__init__(name, **kwargs)
+    self.ordinal = ordinal
+    self.parameter_list = parameter_list
+    self.response_parameter_list = response_parameter_list
+
+  def __eq__(self, other):
+    return super(Method, self).__eq__(other) and \
+           self.ordinal == other.ordinal and \
+           self.parameter_list == other.parameter_list and \
+           self.response_parameter_list == other.response_parameter_list
+
+
+# This needs to be declared after |Method|.
+class InterfaceBody(NodeListBase):
+  """Represents the body of (i.e., list of definitions inside) an interface."""
+
+  _list_item_type = (Const, Enum, Method)
+
+
 class Module(NodeBase):
   """Represents a module statement."""
 
   def __init__(self, name, attribute_list, **kwargs):
     # |name| is either none or a "wrapped identifier".
     assert name is None or isinstance(name, tuple)
     assert attribute_list is None or isinstance(attribute_list, AttributeList)
     super(Module, self).__init__(**kwargs)
     self.name = name
     self.attribute_list = attribute_list
@@ skipping 101 matching lines @@
            self.ordinal == other.ordinal and \
            self.typename == other.typename and \
            self.default_value == other.default_value
 
 
 # This needs to be declared after |StructField|.
 class StructBody(NodeListBase):
   """Represents the body of (i.e., list of definitions inside) a struct."""
 
   _list_item_type = (Const, Enum, StructField)