Refactor WebBluetoothServiceClient in the web_bluetooth.mojom

third_party/WebKit/Source/modules/bluetooth/BluetoothRemoteGATTServer.h

 // Copyright 2015 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.
 
 #ifndef BluetoothRemoteGATTServer_h
 #define BluetoothRemoteGATTServer_h
 
 #include "bindings/core/v8/ScriptWrappable.h"
 #include "bindings/modules/v8/StringOrUnsignedLong.h"
 #include "modules/bluetooth/BluetoothDevice.h"
+#include "mojo/public/cpp/bindings/associated_binding.h"
 #include "platform/heap/Heap.h"
 #include "public/platform/modules/bluetooth/web_bluetooth.mojom-blink.h"
 #include "wtf/text/WTFString.h"
 
 namespace blink {
 
 class BluetoothDevice;
 class ScriptPromise;
 class ScriptPromiseResolver;
 class ScriptState;
 
 // BluetoothRemoteGATTServer provides a way to interact with a connected
 // bluetooth peripheral.
-class BluetoothRemoteGATTServer final
-    : public GarbageCollected<BluetoothRemoteGATTServer>,
-      public ScriptWrappable {
+class BluetoothRemoteGATTServer
+    : public GarbageCollectedFinalized<BluetoothRemoteGATTServer>,
+      public ScriptWrappable,
+      public mojom::blink::WebBluetoothServerClient {
+  USING_PRE_FINALIZER(BluetoothRemoteGATTServer, dispose);
   DEFINE_WRAPPERTYPEINFO();
 
  public:
   BluetoothRemoteGATTServer(BluetoothDevice*);
 
   static BluetoothRemoteGATTServer* create(BluetoothDevice*);
 
+  // mojom::blink::WebBluetoothServerClient:
+  void GATTServerDisconnected() override;
+
   void setConnected(bool connected) { m_connected = connected; }
 
   // The Active Algorithms set is maintained so that disconnection, i.e.
   // disconnect() method or the device disconnecting by itself, can be detected
   // by algorithms. They check via RemoveFromActiveAlgorithms that their
   // resolvers is still in the set of active algorithms.
   //
   // Adds |resolver| to the set of Active Algorithms. CHECK-fails if
   // |resolver| was already added.
   void AddToActiveAlgorithms(ScriptPromiseResolver*);
   // Removes |resolver| from the set of Active Algorithms if it was in the set
   // and returns true, false otherwise.
   bool RemoveFromActiveAlgorithms(ScriptPromiseResolver*);
   // Removes all ScriptPromiseResolvers from the set of Active Algorithms.
   void ClearActiveAlgorithms() { m_activeAlgorithms.clear(); }
 
+  // If gatt is connected then sets gatt.connected to false and disconnects.
+  // This function only performs the necessary steps to ensure a device
+  // disconnects therefore it should only be used when the object is being
+  // garbage collected or the context is being destroyed.
+  void disconnectIfConnected();
+
+  // Performs necessary cleanup when a device disconnects and fires
+  // gattserverdisconnected event.
+  void cleanupDisconnectedDeviceAndFireEvent();
+
+  void dispatchDisconnected();
+
+  // USING_PRE_FINALIZER interface.
+  // Called before the object gets garbage collected.
+  void dispose();
+
   // Interface required by Garbage Collectoin:
   DECLARE_VIRTUAL_TRACE();
 
   // IDL exposed interface:
   BluetoothDevice* device() { return m_device; }
   bool connected() { return m_connected; }
   ScriptPromise connect(ScriptState*);
   void disconnect(ScriptState*);
   ScriptPromise getPrimaryService(ScriptState*,
                                   const StringOrUnsignedLong& service,
                                   ExceptionState&);
   ScriptPromise getPrimaryServices(ScriptState*,
                                    const StringOrUnsignedLong& service,
                                    ExceptionState&);
   ScriptPromise getPrimaryServices(ScriptState*, ExceptionState&);
 
  private:
   ScriptPromise getPrimaryServicesImpl(
       ScriptState*,
       mojom::blink::WebBluetoothGATTQueryQuantity,
       String serviceUUID = String());
 
   void ConnectCallback(ScriptPromiseResolver*,
-                       mojom::blink::WebBluetoothResult);
+                       mojom::blink::WebBluetoothResult,
+                       mojom::blink::WebBluetoothServerClientAssociatedRequest);
   void GetPrimaryServicesCallback(
       const String& requestedServiceUUID,
       mojom::blink::WebBluetoothGATTQueryQuantity,
       ScriptPromiseResolver*,
       mojom::blink::WebBluetoothResult,
       Optional<Vector<mojom::blink::WebBluetoothRemoteGATTServicePtr>>
           services);
 
   // Contains a ScriptPromiseResolver corresponding to each active algorithm
   // using this server’s connection.
   HeapHashSet<Member<ScriptPromiseResolver>> m_activeAlgorithms;
 
+  mojo::AssociatedBinding<mojom::blink::WebBluetoothServerClient>
+      m_clientBinding;
+
   Member<BluetoothDevice> m_device;
   bool m_connected;
 };
 
 }  // namespace blink
 
 #endif  // BluetoothDevice_h