proxypin/lib/ui/desktop/toolbar/setting/request_rewrite.dart

import 'package:flutter/material.dart';
import 'package:network_proxy/network/bin/server.dart';
import 'package:network_proxy/network/util/request_rewrite.dart';

class RequestRewrite extends StatefulWidget {
  final ProxyServer proxyServer;

  const RequestRewrite({super.key, required this.proxyServer});

  @override
  State<RequestRewrite> createState() => _RequestRewriteState();
}

class _RequestRewriteState extends State<RequestRewrite> {
  late RequestRuleList requestRuleList;
  late ValueNotifier<bool> enableNotifier;
  bool changed = false;

  @override
  void initState() {
    super.initState();
    requestRuleList = RequestRuleList(widget.proxyServer.requestRewrites);
    enableNotifier = ValueNotifier(widget.proxyServer.requestRewrites.enabled == true);
  }

  @override
  void dispose() {
    if (changed || enableNotifier.value != widget.proxyServer.requestRewrites.enabled) {
      widget.proxyServer.requestRewrites.enabled = enableNotifier.value;
      widget.proxyServer.flushRequestRewriteConfig();
    }

    enableNotifier.dispose();
    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    return Column(
      crossAxisAlignment: CrossAxisAlignment.start,
      children: [
        SizedBox(
            width: 280,
            child: ValueListenableBuilder(
                valueListenable: enableNotifier,
                builder: (_, bool v, __) {
                  return SwitchListTile(
                      contentPadding: const EdgeInsets.only(left: 2),
                      title: const Text('是否启用请求重写'),
                      dense: true,
                      value: enableNotifier.value,
                      onChanged: (value) {
                        enableNotifier.value = value;
                      });
                })),
        const SizedBox(height: 10),
        Row(children: [
          FilledButton.icon(
              icon: const Icon(Icons.add),
              onPressed: () {
                add();
              },
              label: const Text("增加")),
          const SizedBox(width: 10),
          OutlinedButton.icon(
              onPressed: () {
                var selectedIndex = requestRuleList.currentSelectedIndex();
                add(selectedIndex);
              },
              icon: const Icon(Icons.edit),
              label: const Text("编辑")),
          TextButton.icon(
              icon: const Icon(Icons.remove),
              label: const Text("删除"),
              onPressed: () {
                var removeSelected = requestRuleList.removeSelected();
                if (removeSelected.isEmpty) {
                  return;
                }

                changed = true;
                setState(() {
                  widget.proxyServer.requestRewrites.removeIndex(removeSelected);
                  requestRuleList.changeState();
                });
              })
        ]),
        const SizedBox(height: 10),
        requestRuleList,
      ],
    );
  }

  void add([int currentIndex = -1]) {
    showDialog(
        context: context,
        barrierDismissible: false,
        builder: (BuildContext context) {
          return RuleAddDialog(
              requestRewrites: widget.proxyServer.requestRewrites,
              currentIndex: currentIndex,
              onChange: () {
                changed = true;
                requestRuleList.changeState();
              });
        });
  }
}

///请求重写规则添加对话框
class RuleAddDialog extends StatelessWidget {
  final RequestRewrites requestRewrites;
  final int currentIndex;
  final Function onChange;

  const RuleAddDialog({super.key, required this.currentIndex, required this.onChange, required this.requestRewrites});

  @override
  Widget build(BuildContext context) {
    GlobalKey formKey = GlobalKey<FormState>();
    RequestRewriteRule? rule;
    if (currentIndex >= 0) {
      rule = requestRewrites.rules[currentIndex];
    }

    ValueNotifier<bool> enableNotifier = ValueNotifier(rule == null || rule.enabled);
    String? url = rule?.url;
    String? requestBody = rule?.requestBody;
    String? responseBody = rule?.responseBody;

    return AlertDialog(
        title: const Text("添加请求重写规则", style: TextStyle(fontSize: 16)),
        scrollable: true,
        content: Form(
            key: formKey,
            child: Column(
                mainAxisAlignment: MainAxisAlignment.start,
                crossAxisAlignment: CrossAxisAlignment.start,
                children: <Widget>[
                  ValueListenableBuilder(
                      valueListenable: enableNotifier,
                      builder: (_, bool enable, __) {
                        return SwitchListTile(
                            contentPadding: const EdgeInsets.only(left: 0),
                            title: const Text('是否启用', textAlign: TextAlign.start),
                            value: enable,
                            onChanged: (value) {
                              enableNotifier.value = value;
                            });
                      }),
                  TextFormField(
                      decoration: const InputDecoration(labelText: 'URL', hintText: '/api/v1/*'),
                      validator: (val) {
                        if (val == null || val.isEmpty) {
                          return 'URL不能为空';
                        }
                        return null;
                      },
                      initialValue: url,
                      onSaved: (val) => url = val),
                  TextFormField(
                      initialValue: requestBody,
                      decoration: const InputDecoration(labelText: '请求体替换为:'),
                      onSaved: (val) => requestBody = val),
                  TextFormField(
                      initialValue: responseBody,
                      minLines: 3,
                      maxLines: 15,
                      decoration: const InputDecoration(labelText: '响应体替换为:', hintText: '{"code":"200","data":{}}'),
                      onSaved: (val) => responseBody = val)
                ])),
        actions: [
          FilledButton(
              child: const Text("保存"),
              onPressed: () {
                if ((formKey.currentState as FormState).validate()) {
                  (formKey.currentState as FormState).save();

                  if (currentIndex >= 0) {
                    requestRewrites.rules[currentIndex] = RequestRewriteRule(enableNotifier.value, url!,
                        requestBody: requestBody, responseBody: responseBody);
                  } else {
                    requestRewrites.addRule(RequestRewriteRule(enableNotifier.value, url!,
                        requestBody: requestBody, responseBody: responseBody));
                  }

                  enableNotifier.dispose();
                  onChange.call();
                  Navigator.of(context).pop();
                }
              }),
          ElevatedButton(
              child: const Text("关闭"),
              onPressed: () {
                Navigator.of(context).pop();
              })
        ]);
  }
}

class RequestRuleList extends StatefulWidget {
  final RequestRewrites requestRewrites;

  RequestRuleList(this.requestRewrites) : super(key: GlobalKey<_RequestRuleListState>());

  @override
  State<RequestRuleList> createState() => _RequestRuleListState();

  List<int> removeSelected() {
    var state = (key as GlobalKey<_RequestRuleListState>).currentState;
    List<int> list = [];
    state?.selected.forEach((key, value) {
      if (value == true) {
        list.add(key);
      }
    });
    state?.selected.clear();
    return list;
  }

  int currentSelectedIndex() {
    var state = (key as GlobalKey<_RequestRuleListState>).currentState;
    return state?.currentSelectedIndex ?? -1;
  }

  changeState() {
    var state = (key as GlobalKey<_RequestRuleListState>).currentState;
    state?.changeState();
  }
}

class _RequestRuleListState extends State<RequestRuleList> {
  final Map<int, bool> selected = {};
  int currentSelectedIndex = -1;

  changeState() {
    setState(() {});
  }

  @override
  Widget build(BuildContext context) {
    return Container(
        padding: const EdgeInsets.only(top: 10),
        height: 300,
        constraints: const BoxConstraints(minWidth: 500),
        child: SingleChildScrollView(
            child: DataTable(
          columnSpacing: 36,
          dataRowMaxHeight: 100,
          border: TableBorder.symmetric(outside: BorderSide(width: 1, color: Theme.of(context).highlightColor)),
          columns: const <DataColumn>[
            DataColumn(label: Text('启用')),
            DataColumn(label: Text('URL')),
            DataColumn(label: Text('请求体')),
            DataColumn(label: Text('响应体')),
          ],
          rows: List.generate(
              widget.requestRewrites.rules.length,
              (index) => DataRow(
                      cells: [
                        DataCell(Text(widget.requestRewrites.rules[index].enabled ? "是" : "否")),
                        DataCell(ConstrainedBox(
                            constraints: const BoxConstraints(minWidth: 60),
                            child: Text(widget.requestRewrites.rules[index].url))),
                        DataCell(Container(
                          constraints: const BoxConstraints(maxWidth: 120),
                          padding: const EdgeInsetsDirectional.all(10),
                          child: SelectableText.rich(TextSpan(text: widget.requestRewrites.rules[index].requestBody),
                              style: const TextStyle(fontSize: 12)),
                        )),
                        DataCell(Container(
                          constraints: const BoxConstraints(maxWidth: 300),
                          padding: const EdgeInsetsDirectional.all(10),
                          child: SelectableText.rich(TextSpan(text: widget.requestRewrites.rules[index].responseBody),
                              style: const TextStyle(fontSize: 12)),
                        ))
                      ],
                      selected: selected[index] == true,
                      onSelectChanged: (value) {
                        setState(() {
                          selected[index] = value!;
                          currentSelectedIndex = index;
                        });
                      })),
        )));
  }
}