98 lines
2.9 KiB
Dart
98 lines
2.9 KiB
Dart
import 'dart:async';
|
|
import 'dart:ui';
|
|
|
|
import 'package:flutter_mozilla_components/flutter_mozilla_components.dart';
|
|
import 'package:lensai/domain/entities/equatable_image.dart';
|
|
import 'package:lensai/extensions/nullable.dart';
|
|
import 'package:lensai/features/geckoview/domain/entities/states/web_extension.dart';
|
|
import 'package:lensai/features/geckoview/domain/providers.dart';
|
|
import 'package:lensai/features/geckoview/utils/image_helper.dart';
|
|
import 'package:riverpod_annotation/riverpod_annotation.dart';
|
|
|
|
part 'web_extensions_state.g.dart';
|
|
|
|
@Riverpod(keepAlive: true)
|
|
class WebExtensionsState extends _$WebExtensionsState {
|
|
final _imageCache = <String, EquatableImage>{};
|
|
|
|
void _onExtensionUpdate(ExtensionDataEvent event) {
|
|
final ExtensionDataEvent(:extensionId, :data) = event;
|
|
|
|
if (data != null) {
|
|
final current =
|
|
state[extensionId] ??
|
|
WebExtensionState(
|
|
extensionId: extensionId,
|
|
icon: _imageCache[extensionId],
|
|
enabled: false,
|
|
);
|
|
|
|
state = {...state}
|
|
..[extensionId] = current.copyWith(
|
|
title: data.title,
|
|
enabled: data.enabled ?? current.enabled,
|
|
badgeText: data.badgeText,
|
|
badgeTextColor: data.badgeTextColor.mapNotNull(
|
|
(color) => Color(color),
|
|
),
|
|
badgeBackgroundColor: data.badgeBackgroundColor.mapNotNull(
|
|
(color) => Color(color),
|
|
),
|
|
);
|
|
} else {
|
|
if (state.containsKey(extensionId)) {
|
|
state = {...state}..remove(extensionId);
|
|
}
|
|
}
|
|
}
|
|
|
|
Future<void> _onIconChange(ExtensionIconEvent event) async {
|
|
final ExtensionIconEvent(:extensionId, :bytes) = event;
|
|
|
|
final image = await tryDecodeImage(bytes);
|
|
|
|
if (image != null) {
|
|
if (_imageCache[extensionId] != image) {
|
|
_imageCache[extensionId] = image;
|
|
|
|
if (state.containsKey(extensionId)) {
|
|
state = {...state}
|
|
..[extensionId] = state[extensionId]!.copyWith.icon(image);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@override
|
|
Map<String, WebExtensionState> build(WebExtensionActionType actionType) {
|
|
final addonService = ref.watch(addonServiceProvider);
|
|
|
|
final subscriptions = switch (actionType) {
|
|
WebExtensionActionType.browser => [
|
|
addonService.browserExtensionStream.listen((event) async {
|
|
_onExtensionUpdate(event);
|
|
}),
|
|
addonService.browserIconStream.listen((event) async {
|
|
await _onIconChange(event);
|
|
}),
|
|
],
|
|
WebExtensionActionType.page => [
|
|
addonService.pageExtensionStream.listen((event) async {
|
|
_onExtensionUpdate(event);
|
|
}),
|
|
addonService.pageIconStream.listen((event) async {
|
|
await _onIconChange(event);
|
|
}),
|
|
],
|
|
};
|
|
|
|
ref.onDispose(() {
|
|
for (final sub in subscriptions) {
|
|
unawaited(sub.cancel());
|
|
}
|
|
});
|
|
|
|
return {};
|
|
}
|
|
}
|