proxypin/lib/ui/desktop/request/list.dart

/*
 * Copyright 2023 Hongen Wang All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import 'dart:io';

import 'package:file_picker/file_picker.dart';
import 'package:flutter/material.dart';
import 'package:proxypin/l10n/app_localizations.dart';
import 'package:flutter_toastr/flutter_toastr.dart';
import 'package:proxypin/network/bin/server.dart';
import 'package:proxypin/network/channel/channel.dart';
import 'package:proxypin/network/channel/channel_context.dart';
import 'package:proxypin/network/channel/host_port.dart';
import 'package:proxypin/network/http/http.dart';
import 'package:proxypin/network/http/http_client.dart';
import 'package:proxypin/ui/component/widgets.dart';
import 'package:proxypin/ui/content/panel.dart';
import 'package:proxypin/ui/desktop/request/request_sequence.dart';
import 'package:proxypin/ui/desktop/request/request.dart';
import 'package:proxypin/ui/desktop/request/search.dart';
import 'package:proxypin/utils/har.dart';
import 'package:proxypin/utils/lang.dart';
import 'package:proxypin/utils/listenable_list.dart';

import '../../component/model/search_model.dart';
import 'domians.dart';
import 'package:proxypin/ui/desktop/request/report_servers.dart';

/// @author wanghongen
class DesktopRequestListWidget extends StatefulWidget {
  final ProxyServer proxyServer;
  final ListenableList<HttpRequest>? list;
  final NetworkTabController panel;

  const DesktopRequestListWidget({super.key, required this.proxyServer, this.list, required this.panel});

  @override
  State<StatefulWidget> createState() {
    return DesktopRequestListState();
  }
}

class DesktopRequestListState extends State<DesktopRequestListWidget> with AutomaticKeepAliveClientMixin {
  final GlobalKey<RequestSequenceState> requestSequenceKey = GlobalKey<RequestSequenceState>();
  final GlobalKey<DomainWidgetState> domainListKey = GlobalKey<DomainWidgetState>();
  final GlobalKey<SearchState> searchKey = GlobalKey<SearchState>();

  //请求列表容器
  ListenableList<HttpRequest> container = ListenableList();

  bool sortDesc = true;

  AppLocalizations get localizations => AppLocalizations.of(context)!;

  @override
  void initState() {
    super.initState();
    if (widget.list != null) {
      container = widget.list!;
    }
  }

  @override
  bool get wantKeepAlive => true;

  @override
  void dispose() {
    RequestWidget.removeAutoReadByIds(container.map((request) => request.requestId));
    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    super.build(context);

    List<Tab> tabs = [
      Tab(child: Text(localizations.domainList, style: const TextStyle(fontSize: 13))),
      Tab(child: Text(localizations.sequence, style: const TextStyle(fontSize: 13))),
    ];

    return DefaultTabController(
        length: tabs.length,
        child: Scaffold(
            appBar: AppBar(
              toolbarHeight: 40,
              title: SizedBox(height: 40, child: TabBar(tabs: tabs, dividerColor: Colors.transparent)),
              automaticallyImplyLeading: false,
              actions: [popupMenus()],
            ),
            bottomNavigationBar: Search(key: searchKey, onSearch: search),
            body: Padding(
                padding: const EdgeInsets.only(right: 5),
                child: TabBarView(physics: const NeverScrollableScrollPhysics(), children: [
                  DomainList(
                      key: domainListKey,
                      list: container,
                      panel: widget.panel,
                      proxyServer: widget.proxyServer,
                      onRemove: domainListRemove),
                  RequestSequence(
                      key: requestSequenceKey,
                      container: container,
                      proxyServer: widget.proxyServer,
                      onRemove: sequenceRemove),
                ]))));
  }

  Widget popupMenus() {
    return PopupMenuButton(
        offset: const Offset(0, 32),
        icon: const Icon(Icons.more_vert_outlined, size: 20),
        itemBuilder: (BuildContext context) {
          return <PopupMenuEntry>[
            CustomPopupMenuItem(
                height: 37,
                onTap: () => searchKey.currentState?.searchDialog(),
                child: IconText(
                    icon: const Icon(Icons.search, size: 17),
                    text: localizations.search,
                    textStyle: const TextStyle(fontSize: 13))),
            CustomPopupMenuItem(
                height: 37,
                onTap: () => export('ProxyPin_${DateTime.now().dateFormat()}.har'),
                child: IconText(
                    icon: const Icon(Icons.share, size: 16),
                    text: localizations.viewExport,
                    textStyle: const TextStyle(fontSize: 13))),
            CustomPopupMenuItem(
                height: 37,
                onTap: () => repeatAllRequests(),
                child: IconText(
                    icon: const Icon(Icons.repeat, size: 16),
                    text: localizations.repeatAllRequests,
                    textStyle: const TextStyle(fontSize: 13))),
            CustomPopupMenuItem(
                height: 37,
                onTap: () {
                  sortDesc = !sortDesc;
                  requestSequenceKey.currentState?.sort(sortDesc);
                  domainListKey.currentState?.sort(sortDesc);
                },
                child: IconText(
                    icon: const Icon(Icons.sort, size: 16),
                    text: sortDesc ? localizations.timeAsc : localizations.timeDesc,
                    textStyle: const TextStyle(fontSize: 13))),
            CustomPopupMenuItem(
                height: 37,
                onTap: () {
                  showReportServersDialog(context);
                },
                child: IconText(
                    icon: Icon(Icons.cloud_upload_outlined, size: 16),
                    text: localizations.reportServers,
                    textStyle: TextStyle(fontSize: 13))),
          ];
        });
  }

  ///添加请求
  void add(Channel channel, HttpRequest request) {
    container.add(request);
    domainListKey.currentState?.add(channel, request);
    requestSequenceKey.currentState?.add(request);
  }

  ///添加响应
  void addResponse(ChannelContext channelContext, HttpResponse response) {
    domainListKey.currentState?.addResponse(channelContext, response);
    requestSequenceKey.currentState?.addResponse(response);
  }

  ///移除
  void domainListRemove(List<HttpRequest> list) {
    container.removeWhere((element) => list.contains(element));
    requestSequenceKey.currentState?.remove(list);
    RequestWidget.removeAutoReadByIds(list.map((request) => request.requestId));
  }

  ///全部请求删除
  void sequenceRemove(List<HttpRequest> list) {
    container.removeWhere((element) => list.contains(element));
    domainListKey.currentState?.remove(list);
    RequestWidget.removeAutoReadByIds(list.map((request) => request.requestId));
  }

  void search(SearchModel searchModel) {
    domainListKey.currentState?.search(searchModel);
    requestSequenceKey.currentState?.search(searchModel);
  }

  List<HttpRequest>? currentView() {
    return domainListKey.currentState?.currentView();
  }

  ///清理
  void clean() {
    setState(() {
      RequestWidget.removeAutoReadByIds(container.map((request) => request.requestId));
      container.clear();
      domainListKey.currentState?.clean();
      requestSequenceKey.currentState?.clean();
      widget.panel.change(null, null);
    });
  }

  void cleanupEarlyData(int retain) {
    var list = container.source;
    if (list.length <= retain) {
      return;
    }

    var removeRange = container.removeRange(0, list.length - retain);

    domainListKey.currentState?.clean();
    requestSequenceKey.currentState?.clean();

    RequestWidget.removeAutoReadByIds(removeRange.map((request) => request.requestId));
  }

  ///导出
  Future<void> export(String fileName) async {
    var path = await FilePicker.platform.saveFile(fileName: fileName);
    if (path == null) {
      return;
    }

    //获取请求
    List<HttpRequest>? requests = currentView();
    if (requests == null) return;

    var file = await File(path).create();
    await Har.writeFile(requests, file, title: fileName);

    if (mounted) FlutterToastr.show(AppLocalizations.of(context)!.exportSuccess, context);
  }

  ///重发所有请求
  void repeatAllRequests() async {
    var requests = currentView();
    if (requests == null) return;

    var localizations = AppLocalizations.of(context);
    final proxyServer = widget.proxyServer;

    for (var request in requests) {
      var httpRequest = request.copy(uri: request.requestUrl);
      var proxyInfo = proxyServer.isRunning ? ProxyInfo.of("127.0.0.1", proxyServer.port) : null;
      try {
        await HttpClients.proxyRequest(httpRequest, proxyInfo: proxyInfo, timeout: const Duration(seconds: 3));
        if (mounted) {
          FlutterToastr.show(localizations!.reSendRequest, rootNavigator: true, context);
        }
      } catch (e) {
        if (mounted) {
          FlutterToastr.show('${localizations!.fail} $e', rootNavigator: true, context);
        }
      }
    }
  }
}