Initial commit: Accessible SSH Terminal
This commit is contained in:
136
lib/services/ssh_service.dart
Normal file
136
lib/services/ssh_service.dart
Normal file
@@ -0,0 +1,136 @@
|
||||
import 'dart:async';
|
||||
import 'dart:convert';
|
||||
|
||||
import 'package:dartssh2/dartssh2.dart';
|
||||
|
||||
class SshService {
|
||||
SSHClient? _client;
|
||||
SSHSession? _session;
|
||||
Timer? _keepAliveTimer;
|
||||
|
||||
// Broadcast controller so multiple listeners (UI, loggers) can subscribe
|
||||
final StreamController<String> _outputController = StreamController<String>.broadcast();
|
||||
// Controller to notify about disconnection (true = clean exit/0, false = error/drop)
|
||||
final StreamController<bool> _disconnectController = StreamController.broadcast();
|
||||
|
||||
Stream<String> get outputStream => _outputController.stream;
|
||||
Stream<bool> get disconnectStream => _disconnectController.stream;
|
||||
|
||||
bool get isConnected => _client != null && !(_client?.isClosed ?? true);
|
||||
|
||||
Future<void> connect({
|
||||
required String host,
|
||||
required int port,
|
||||
required String username,
|
||||
String? password,
|
||||
String? privateKey,
|
||||
}) async {
|
||||
try {
|
||||
_outputController.add('Connecting to $host:$port as $username...\n');
|
||||
|
||||
final socket = await SSHSocket.connect(host, port);
|
||||
|
||||
_client = SSHClient(
|
||||
socket,
|
||||
username: username,
|
||||
onPasswordRequest: password != null ? () => password : null,
|
||||
identities: privateKey != null ? SSHKeyPair.fromPem(privateKey) : [],
|
||||
);
|
||||
|
||||
_outputController.add('Authenticating...\n');
|
||||
await _client!.authenticated;
|
||||
_outputController.add('Authenticated. Starting shell...\n');
|
||||
|
||||
_startKeepAlive();
|
||||
await _startShell();
|
||||
} catch (e) {
|
||||
_outputController.add('Connection failed: $e\n');
|
||||
await disconnect();
|
||||
rethrow;
|
||||
}
|
||||
}
|
||||
|
||||
void _startKeepAlive() {
|
||||
_keepAliveTimer?.cancel();
|
||||
_keepAliveTimer = Timer.periodic(const Duration(seconds: 30), (timer) async {
|
||||
if (!isConnected) {
|
||||
timer.cancel();
|
||||
return;
|
||||
}
|
||||
try {
|
||||
// Send a window change request with the same size.
|
||||
// This generates valid SSH traffic to keep the connection alive.
|
||||
_session?.resizeTerminal(80, 24);
|
||||
} catch (_) {
|
||||
// Fails silently; the main connection handlers will deal with actual drops.
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
Future<void> _startShell() async {
|
||||
if (_client == null) return;
|
||||
|
||||
try {
|
||||
_session = await _client!.shell(
|
||||
pty: const SSHPtyConfig(
|
||||
width: 80,
|
||||
height: 24,
|
||||
),
|
||||
);
|
||||
|
||||
// Listen for session completion (e.g. exit command)
|
||||
_session!.done.then((_) {
|
||||
final code = _session!.exitCode;
|
||||
disconnect(isClean: code == 0);
|
||||
});
|
||||
|
||||
// Use transform(utf8.decoder) to correctly handle multi-byte characters split across chunks
|
||||
_session!.stdout.cast<List<int>>().transform(const Utf8Decoder()).listen(
|
||||
(data) => _outputController.add(data),
|
||||
onError: (e) => _outputController.add('Error reading stdout: $e\n'),
|
||||
onDone: () {
|
||||
if (isConnected) _outputController.add('Session stdout closed.\n');
|
||||
},
|
||||
);
|
||||
|
||||
_session!.stderr.cast<List<int>>().transform(const Utf8Decoder()).listen(
|
||||
(data) => _outputController.add(data),
|
||||
onError: (e) => _outputController.add('Error reading stderr: $e\n'),
|
||||
);
|
||||
|
||||
// Wait for the connection to complete (unexpected drop or client close)
|
||||
_client!.done.then((_) {
|
||||
_outputController.add('Connection closed.\n');
|
||||
disconnect(isClean: false);
|
||||
});
|
||||
|
||||
} catch (e) {
|
||||
_outputController.add('Failed to start shell: $e\n');
|
||||
await disconnect();
|
||||
}
|
||||
}
|
||||
|
||||
void write(String data) {
|
||||
if (_session != null) {
|
||||
_session!.write(utf8.encode(data));
|
||||
}
|
||||
}
|
||||
|
||||
Future<void> disconnect({bool isClean = false}) async {
|
||||
_keepAliveTimer?.cancel();
|
||||
if (_client == null && _session == null) return;
|
||||
|
||||
_session?.close();
|
||||
_client?.close();
|
||||
_client = null;
|
||||
_session = null;
|
||||
|
||||
_disconnectController.add(isClean);
|
||||
}
|
||||
|
||||
void dispose() {
|
||||
_keepAliveTimer?.cancel();
|
||||
_outputController.close();
|
||||
_disconnectController.close();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user