app/lib/common/io/aria2c.dart

142 lines
4.2 KiB
Dart
Raw Normal View History

2024-02-23 00:07:48 +08:00
import 'dart:convert';
import 'dart:io';
2024-02-24 18:30:44 +08:00
import 'dart:math';
2024-02-23 00:07:48 +08:00
import 'package:aria2/aria2.dart';
import 'package:flutter/foundation.dart';
import 'package:starcitizen_doctor/base/ui.dart';
import 'package:starcitizen_doctor/common/conf/app_conf.dart';
2024-02-24 17:41:44 +08:00
import 'package:starcitizen_doctor/common/conf/binary_conf.dart';
2024-02-23 00:07:48 +08:00
import 'package:starcitizen_doctor/common/helper/system_helper.dart';
class Aria2cManager {
2024-02-24 17:41:44 +08:00
static int? _daemonPID;
2024-02-23 00:07:48 +08:00
static final String _aria2cDir =
"${AppConf.applicationSupportDir}\\modules\\aria2c";
static Aria2c? _aria2c;
2024-02-24 17:41:44 +08:00
static Aria2c getClient() {
if (_aria2c != null) return _aria2c!;
throw "not connect!";
}
static bool get isAvailable => _daemonPID != null && _aria2c != null;
static Future checkLazyLoad() async {
try {
final sessionFile = File("$_aria2cDir\\aria2.session");
// 有下载任务则第一时间初始化
if (await sessionFile.exists() &&
(await sessionFile.readAsString()).trim().isNotEmpty) {
await launchDaemon();
}
} catch (e) {
dPrint("Aria2cManager.checkLazyLoad Error:$e");
2024-02-23 00:07:48 +08:00
}
}
static Future launchDaemon() async {
2024-02-24 17:41:44 +08:00
if (_daemonPID != null) return;
await BinaryModuleConf.extractModule(["aria2c"]);
2024-02-23 00:07:48 +08:00
/// skip for debug hot reload
if (kDebugMode) {
if ((await SystemHelper.getPID("aria2c")).isNotEmpty) {
dPrint("[Aria2cManager] debug skip for hot reload");
return;
}
}
final sessionFile = File("$_aria2cDir\\aria2.session");
if (!await sessionFile.exists()) {
await sessionFile.create(recursive: true);
}
final exePath = "$_aria2cDir\\aria2c.exe";
2024-02-24 18:30:44 +08:00
final port = await getFreePort();
final pwd = generateRandomPassword(16);
dPrint("Aria2cManager .----- aria2c start $port------");
2024-02-23 00:07:48 +08:00
final p = await Process.start(
exePath,
[
"-V",
"-c",
"-x 10",
"--dir=$_aria2cDir\\downloads",
"--disable-ipv6",
"--enable-rpc",
"--pause",
2024-02-24 18:30:44 +08:00
"--rpc-listen-port=$port",
"--rpc-secret=$pwd",
2024-02-23 00:07:48 +08:00
"--input-file=${sessionFile.absolute.path.trim()}",
"--save-session=${sessionFile.absolute.path.trim()}",
"--save-session-interval=60",
"--file-allocation=trunc",
2024-02-24 17:10:28 +08:00
"--seed-time=0",
2024-02-23 00:07:48 +08:00
],
2024-02-24 14:42:33 +08:00
workingDirectory: _aria2cDir);
2024-02-23 00:07:48 +08:00
p.stdout.transform(utf8.decoder).listen((event) {
if (event.trim().isEmpty) return;
2024-02-24 14:42:33 +08:00
dPrint("[aria2c]: ${event.trim()}");
2024-02-23 00:07:48 +08:00
if (event.contains("IPv4 RPC: listening on TCP port")) {
2024-02-24 17:41:44 +08:00
_daemonPID = p.pid;
2024-02-24 18:30:44 +08:00
_aria2c = Aria2c("ws://127.0.0.1:$port/jsonrpc", "websocket", pwd);
2024-02-24 17:41:44 +08:00
_aria2c!.getVersion().then((value) {
dPrint("Aria2cManager.connected! version == ${value.version}");
});
2024-02-23 00:07:48 +08:00
}
}, onDone: () {
dPrint("[aria2c] onDone: ");
2024-02-24 17:41:44 +08:00
_daemonPID = null;
2024-02-23 00:07:48 +08:00
}, onError: (e) {
dPrint("[aria2c] stdout ERROR: $e");
2024-02-24 17:41:44 +08:00
_daemonPID = null;
2024-02-23 00:07:48 +08:00
});
2024-02-24 17:41:44 +08:00
p.pid;
2024-02-23 00:07:48 +08:00
p.stderr.transform(utf8.decoder).listen((event) {
dPrint("[aria2c] stderr ERROR : $event");
});
while (true) {
2024-02-24 17:41:44 +08:00
if (_daemonPID != null) return;
2024-02-23 00:07:48 +08:00
await Future.delayed(const Duration(milliseconds: 100));
}
}
2024-02-24 17:10:28 +08:00
2024-02-24 18:30:44 +08:00
static Future<int> getFreePort() async {
final serverSocket = await ServerSocket.bind("127.0.0.1", 0);
final port = serverSocket.port;
await serverSocket.close();
return port;
}
static String generateRandomPassword(int length) {
const String charset =
"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
Random random = Random();
StringBuffer buffer = StringBuffer();
for (int i = 0; i < length; i++) {
int randomIndex = random.nextInt(charset.length);
buffer.write(charset[randomIndex]);
}
return buffer.toString();
}
2024-02-24 17:10:28 +08:00
static int textToByte(String text) {
if (text.length == 1) {
return 0;
}
if (int.tryParse(text) != null) {
return int.parse(text);
}
if (text.endsWith("k")) {
return int.parse(text.substring(0, text.length - 1)) * 1024;
}
if (text.endsWith("m")) {
return int.parse(text.substring(0, text.length - 1)) * 1024 * 1024;
}
return 0;
}
2024-02-23 00:07:48 +08:00
}