immich/mobile/lib/presentation/widgets/images/thumbnail.widget.dart

import 'package:flutter/material.dart';
import 'package:immich_mobile/domain/models/asset/base_asset.model.dart';
import 'package:immich_mobile/presentation/widgets/images/local_thumb_provider.dart';
import 'package:immich_mobile/presentation/widgets/images/remote_thumb_provider.dart';
import 'package:immich_mobile/presentation/widgets/images/thumb_hash_provider.dart';
import 'package:immich_mobile/widgets/asset_grid/thumbnail_placeholder.dart';
import 'package:immich_mobile/widgets/common/fade_in_placeholder_image.dart';
import 'package:logging/logging.dart';
import 'package:octo_image/octo_image.dart';

class Thumbnail extends StatelessWidget {
  const Thumbnail({
    this.asset,
    this.remoteId,
    this.size = const Size.square(256),
    this.fit = BoxFit.cover,
    super.key,
  }) : assert(
          asset != null || remoteId != null,
          'Either asset or remoteId must be provided',
        );

  final BaseAsset? asset;
  final String? remoteId;
  final Size size;
  final BoxFit fit;

  static ImageProvider imageProvider({
    BaseAsset? asset,
    String? remoteId,
    Size size = const Size.square(256),
  }) {
    assert(
      asset != null || remoteId != null,
      'Either asset or remoteId must be provided',
    );

    if (remoteId != null) {
      return RemoteThumbProvider(
        assetId: remoteId,
        height: size.height,
        width: size.width,
      );
    }

    if (asset is LocalAsset) {
      return LocalThumbProvider(
        asset: asset,
        height: size.height,
        width: size.width,
      );
    }

    if (asset is RemoteAsset) {
      return RemoteThumbProvider(
        assetId: asset.id,
        height: size.height,
        width: size.width,
      );
    }

    throw ArgumentError("Unsupported asset type: ${asset.runtimeType}");
  }

  @override
  Widget build(BuildContext context) {
    final thumbHash =
        asset is RemoteAsset ? (asset as RemoteAsset).thumbHash : null;
    final provider =
        imageProvider(asset: asset, remoteId: remoteId, size: size);

    return OctoImage.fromSet(
      image: provider,
      octoSet: OctoSet(
        placeholderBuilder: _blurHashPlaceholderBuilder(thumbHash, fit: fit),
        errorBuilder: _blurHashErrorBuilder(
          thumbHash,
          provider: provider,
          fit: fit,
          asset: asset,
        ),
      ),
      fadeOutDuration: const Duration(milliseconds: 100),
      fadeInDuration: Duration.zero,
      width: size.width,
      height: size.height,
      fit: fit,
      placeholderFadeInDuration: Duration.zero,
    );
  }
}

OctoPlaceholderBuilder _blurHashPlaceholderBuilder(
  String? thumbHash, {
  BoxFit? fit,
}) {
  return (context) => thumbHash == null
      ? const ThumbnailPlaceholder()
      : FadeInPlaceholderImage(
          placeholder: const ThumbnailPlaceholder(),
          image: ThumbHashProvider(thumbHash: thumbHash),
          fit: fit ?? BoxFit.cover,
        );
}

OctoErrorBuilder _blurHashErrorBuilder(
  String? blurhash, {
  BaseAsset? asset,
  ImageProvider? provider,
  BoxFit? fit,
}) =>
    (context, e, s) {
      Logger("ImThumbnail")
          .warning("Error loading thumbnail for ${asset?.name}", e, s);
      provider?.evict();
      return Stack(
        alignment: Alignment.center,
        children: [
          _blurHashPlaceholderBuilder(blurhash, fit: fit)(context),
          const Opacity(
            opacity: 0.75,
            child: Icon(Icons.error_outline_rounded),
          ),
        ],
      );
    };