immich/mobile/lib/widgets/common/immich_thumbnail.dart

import 'dart:typed_data';

import 'package:flutter/material.dart';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:immich_mobile/providers/image/immich_local_thumbnail_provider.dart';
import 'package:immich_mobile/providers/image/immich_remote_thumbnail_provider.dart';
import 'package:immich_mobile/entities/asset.entity.dart';
import 'package:immich_mobile/utils/hooks/blurhash_hook.dart';
import 'package:immich_mobile/widgets/common/immich_image.dart';
import 'package:immich_mobile/widgets/common/thumbhash_placeholder.dart';
import 'package:octo_image/octo_image.dart';
import 'package:immich_mobile/providers/user.provider.dart';

class ImmichThumbnail extends HookConsumerWidget {
  const ImmichThumbnail({
    this.asset,
    this.width = 250,
    this.height = 250,
    this.fit = BoxFit.cover,
    super.key,
  });

  final Asset? asset;
  final double width;
  final double height;
  final BoxFit fit;

  /// Helper function to return the image provider for the asset thumbnail
  /// either by using the asset ID or the asset itself
  /// [asset] is the Asset to request, or else use [assetId] to get a remote
  /// image provider
  static ImageProvider imageProvider({
    Asset? asset,
    String? assetId,
    String? userId,
    int thumbnailSize = 256,
  }) {
    if (asset == null && assetId == null) {
      throw Exception('Must supply either asset or assetId');
    }

    if (asset == null) {
      return ImmichRemoteThumbnailProvider(
        assetId: assetId!,
      );
    }

    if (ImmichImage.useLocal(asset)) {
      return ImmichLocalThumbnailProvider(
        asset: asset,
        height: thumbnailSize,
        width: thumbnailSize,
        userId: userId,
      );
    } else {
      return ImmichRemoteThumbnailProvider(
        assetId: asset.remoteId!,
        height: thumbnailSize,
        width: thumbnailSize,
      );
    }
  }

  @override
  Widget build(BuildContext context, WidgetRef ref) {
    Uint8List? blurhash = useBlurHashRef(asset).value;
    final userId = ref.watch(currentUserProvider)?.id;

    if (asset == null) {
      return Container(
        color: Colors.grey,
        width: width,
        height: height,
        child: const Center(
          child: Icon(Icons.no_photography),
        ),
      );
    }

    final thumbnailProviderInstance = ImmichThumbnail.imageProvider(
      asset: asset,
      userId: userId,
    );

    customErrorBuilder(BuildContext ctx, Object error, StackTrace? stackTrace) {
      thumbnailProviderInstance.evict();

      final originalErrorWidgetBuilder =
          blurHashErrorBuilder(blurhash, fit: fit);
      return originalErrorWidgetBuilder(ctx, error, stackTrace);
    }

    return OctoImage.fromSet(
      placeholderFadeInDuration: Duration.zero,
      fadeInDuration: Duration.zero,
      fadeOutDuration: const Duration(milliseconds: 100),
      octoSet: OctoSet(
        placeholderBuilder: blurHashPlaceholderBuilder(blurhash, fit: fit),
        errorBuilder: customErrorBuilder,
      ),
      image: thumbnailProviderInstance,
      width: width,
      height: height,
      fit: fit,
    );
  }
}