enable custom view widgets;

lib/country_code_picker.dart

@@ -17,6 +17,7 @@ class CountryCodePicker extends StatefulWidget {
   final InputDecoration searchDecoration;
   final TextStyle searchStyle;
   final WidgetBuilder emptySearchBuilder;
+  final Widget customWidget;
 
   /// shows the name of the country instead of the dialcode
   final bool showOnlyCountryWhenClosed;
@@ -43,7 +44,8 @@ class CountryCodePicker extends StatefulWidget {
     this.emptySearchBuilder,
     this.showOnlyCountryWhenClosed = false,
     this.alignLeft = false,
-    this.showFlag = true
+    this.showFlag = true,
+    this.customWidget,
   });
 
   @override
@@ -71,12 +73,17 @@ class _CountryCodePickerState extends State<CountryCodePicker> {
   _CountryCodePickerState(this.elements);
 
   @override
-  Widget build(BuildContext context) => FlatButton(
+  Widget build(BuildContext context) {
-        child: Flex(
+    Widget _widget;
+    if (widget.customWidget != null)
+      _widget = widget.customWidget;
+    else {
+      _widget = Flex(
         direction: Axis.horizontal,
         mainAxisSize: MainAxisSize.min,
         children: <Widget>[
-            widget.showFlag ? Flexible(
+          widget.showFlag
+              ? Flexible(
                   flex: widget.alignLeft ? 0 : 1,
                   fit: widget.alignLeft ? FlexFit.tight : FlexFit.loose,
                   child: Padding(
@@ -89,7 +96,8 @@ class _CountryCodePickerState extends State<CountryCodePicker> {
                       width: 32.0,
                     ),
                   ),
-            ) : Container(),
+                )
+              : Container(),
           Flexible(
             fit: widget.alignLeft ? FlexFit.tight : FlexFit.loose,
             child: Text(
@@ -100,10 +108,14 @@ class _CountryCodePickerState extends State<CountryCodePicker> {
             ),
           ),
         ],
-        ),
+      );
+    }
+    return FlatButton(
+      child: _widget,
       padding: widget.padding,
       onPressed: _showSelectionDialog,
     );
+  }
 
   @override
   initState() {
@@ -130,16 +142,12 @@ class _CountryCodePickerState extends State<CountryCodePicker> {
   void _showSelectionDialog() {
     showDialog(
       context: context,
-      builder: (_) =>
+      builder: (_) => SelectionDialog(elements, favoriteElements,
-        SelectionDialog(
-          elements,
-          favoriteElements,
           showCountryOnly: widget.showCountryOnly,
           emptySearchBuilder: widget.emptySearchBuilder,
           searchDecoration: widget.searchDecoration,
           searchStyle: widget.searchStyle,
-          showFlag: widget.showFlag
+          showFlag: widget.showFlag),
-        ),
     ).then((e) {
       if (e != null) {
         setState(() {