Highlight active option in AppBar

lib/ui/widgets/tablet_mode/appbar_custom_dropdown_widget.dart

@@ -30,16 +30,22 @@ class AppBarCustomDropDownWidget extends StatefulWidget {
 
 class _AppBarCustomDropDownWidgetState
     extends State<AppBarCustomDropDownWidget> {
+  bool _visible = false;
+
   @override
   Widget build(BuildContext context) {
     return IconButtonWithBadge(
       icon: widget.icon,
       onPressed: toggleOverlay,
       number: widget.notificationsBadge,
+      active: _visible,
     );
   }
 
   void toggleOverlay() async {
+    setState(() => _visible = !_visible);
+
+    if (_visible) {
       RenderBox renderBox = context.findRenderObject();
       final size = renderBox.size;
       final offset = renderBox.localToGlobal(Offset(size.width, size.height));
@@ -48,7 +54,10 @@ class _AppBarCustomDropDownWidgetState
         onBuild: widget.onBuildOverlay,
         showContext: context,
         offset: offset,
+        onDispose: () => setState(() => _visible = false),
       ));
+    } else
+      Navigator.of(context).pop();
   }
 }
 
@@ -56,13 +65,22 @@ class _AppBarCustomPopupRoute extends PopupRoute {
   final BuildContext showContext;
   final Widget Function(BuildContext) onBuild;
   final Offset offset;
+  final void Function() onDispose;
 
   _AppBarCustomPopupRoute({
     @required this.showContext,
     @required this.onBuild,
     @required this.offset,
+    @required this.onDispose,
   });
 
+  @override
+  void dispose() {
+    super.dispose();
+
+    onDispose();
+  }
+
   @override
   Color get barrierColor => null;