fix bug by setting correctly the webview controller for next screen

Author: AdamMusa

lib/flutter_on_rails.dart

@@ -0,0 +1,2 @@
+export 'src/Widget/main/main_screen.dart';
+export 'src/minimal/notifier/init.dart';

lib/imports/exports.dart

@@ -1,4 +1,4 @@
-import 'package:flutterx/imports/exports.dart';
+import 'package:flutter_on_rails/src/imports/exports.dart';
 import '../assets/grouped_icons.dart';
 
 NavigationBar? buildBottomNavigationBar(AppNotifier provider, String url) {

lib/main_screen.dart

@@ -1,7 +1,7 @@
 // ignore_for_file: must_be_immutable
 
 import 'package:flutter/material.dart';
-import 'package:flutterx/widget_parser/entry_widget_parser.dart';
+import 'package:flutter_on_rails/src/widget_parser/entry_widget_parser.dart';
 
 class CustomAppBar extends StatelessWidget implements PreferredSizeWidget {
   CustomAppBar({super.key, required this.payload});

lib/Widget/bottom_navigation_bar.dart renamed to lib/src/Widget/bottom_navigation_bar.dart

@@ -1,28 +1,28 @@
 // ignore_for_file: library_private_types_in_public_api, deprecated_member_use
 
 import 'package:flutter/foundation.dart';
-import 'package:flutterx/Widget/bottom_navigation_bar.dart';
-import 'package:flutterx/Widget/custom_app_bar.dart';
-import 'package:flutterx/imports/exports.dart';
-import 'package:flutterx/navigation/handle_navigation.dart'
+import 'package:flutter_on_rails/src/Widget/bottom_navigation_bar.dart';
+import 'package:flutter_on_rails/src/Widget/custom_app_bar.dart';
+import 'package:flutter_on_rails/src/imports/exports.dart';
+import 'package:flutter_on_rails/src/navigation/handle_navigation.dart'
     show HandleNavigation;
-import 'package:flutterx/services/run_js.dart';
+import 'package:flutter_on_rails/src/services/run_js.dart';
 import 'next_screen.dart';
-import 'src/minimal/manager/manager.dart';
+import '../../minimal/manager/manager.dart';
 
-class FlutterX extends StatefulWidget {
-  FlutterX({super.key, this.url = "http://localhost:3000"});
+// ignore: must_be_immutable
+class MainScreen extends StatefulWidget {
+  MainScreen({super.key, this.url = "http://localhost:3000"});
   String? url = "";
   @override
-  _FlutterXState createState() => _FlutterXState();
+  _MainScreenState createState() => _MainScreenState();
 }
 
-class _FlutterXState extends State<FlutterX> {
+class _MainScreenState extends State<MainScreen> {
   PullToRefreshSettings pullToRefreshSettings = PullToRefreshSettings(
     color: Colors.blue,
   );
   PullToRefreshController? pullToRefreshController;
-  InAppWebViewController? webViewController;
   bool isNavigating = false;
   String? defaultUserAgent;
   bool isLoaded = false;
@@ -63,7 +63,7 @@ class _FlutterXState extends State<FlutterX> {
   }
 
   Future<void> _setUserAgent() async {
-    if (webViewController == null) return;
+    if (provider.state.controller == null) return;
 
     switch (defaultTargetPlatform) {
       case TargetPlatform.android:
@@ -131,13 +131,10 @@ class _FlutterXState extends State<FlutterX> {
                   ),
                   pullToRefreshController: pullToRefreshController,
                   onWebViewCreated: (controller) async {
-                    webViewController = controller;
-                    await _setUserAgent();
-                    // Store the controller in the provider
                     provider.setController(controller);
+                    await _setUserAgent();
                     // Add JavaScript handler to listen for postMessage calls
-
-                    controller.addJavaScriptHandler(
+                    provider.state.controller!.addJavaScriptHandler(
                       handlerName: 'Flutter',
                       callback: (args) {
                         // Args are the messages sent from JavaScript
@@ -150,7 +147,7 @@ class _FlutterXState extends State<FlutterX> {
                         }
                       },
                     );
-                    controller.addJavaScriptHandler(
+                    provider.state.controller!.addJavaScriptHandler(
                       handlerName: "actionHandler",
                       callback: (args) {
                         final actionJson = args[0]; // Your JSON string
@@ -190,7 +187,7 @@ class _FlutterXState extends State<FlutterX> {
                     isLoaded = true;
                     pullToRefreshController?.endRefreshing();
                     provider.setLoading(false);
-                    await controller.evaluateJavascript(
+                    await provider.state.controller!.evaluateJavascript(
                       source: """
                         (function() {
                           document.addEventListener("click", function(event) {
@@ -200,8 +197,8 @@ class _FlutterXState extends State<FlutterX> {
                             if (!link) return;
                             const action = link.getAttribute("data_frails_action");
                             console.log("Action:", action); 
-                            if (window.flutterx) {
-                              window.flutterx.callHandler('actionHandler', action);
+                            if (window.flutter_on_rails) {
+                              window.flutter_on_rails.callHandler('actionHandler', action);
                             }
                           });
                         })();

lib/Widget/custom_app_bar.dart renamed to lib/src/Widget/custom_app_bar.dart

@@ -4,9 +4,9 @@ import 'dart:convert';
 import 'package:flutter/foundation.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_inappwebview/flutter_inappwebview.dart';
-import 'package:flutterx/helpers/responds_to.dart';
-import 'package:flutterx/src/minimal/manager/manager.dart';
-import 'package:flutterx/services/run_js.dart';
+import 'package:flutter_on_rails/src/helpers/responds_to.dart';
+import 'package:flutter_on_rails/src/minimal/manager/manager.dart';
+import 'package:flutter_on_rails/src/services/run_js.dart';
 
 class NextPage extends StatelessWidget {
   NextPage({super.key, required this.url});
@@ -74,8 +74,7 @@ class NextPage extends StatelessWidget {
               onWebViewCreated: (
                 InAppWebViewController webViewController,
               ) async {
-                // _controller = webViewController;
-                provider.setController(webViewController);
+                webViewController = provider.state.controller!;
                 await _setUserAgent();
                 provider.state.controller!.addJavaScriptHandler(
                   handlerName: "inputFocus",